from pygame import *
from random import randint

BLACK = (0,0,0)
CELL_WIDTH = 8
CELL_HEIGHT= 8

class Tank:
    def __init__(self, direction, x, y, file_image):
        self.direction = direction
        self.health = 100
        self.x = x
        self.y = y
        self.speed = [0, 0]
        self.image = image.load(file_image).convert()
        self.init_image = image.load(file_image).convert()
        self.rect = self.image.get_rect()

    def draw(self):
        global screen
        screen.blit(self.image, (self.x, self.y))

class Level:
    def __init__(self, file, wall_image):
        self.level = open(file).read().rstrip().split("\n")
        self.image = image.load(wall_image).convert()
        self.walls = []
        x = 0
        y = 0
        for line in self.level:
            for cell in line:
                if cell == "*":
                    self.walls += [Rect(x, y, CELL_WIDTH, CELL_HEIGHT)]
                x += CELL_WIDTH
            y += CELL_HEIGHT
            x = 0
    def draw(self):
        x = 0
        y = 0
        for line in self.level:
            for cell in line:
                if cell == "*":
                    screen.blit(self.image,(x,y))
                x += CELL_WIDTH
            y += CELL_HEIGHT
            x = 0

class Game:
    def __init__(self):
        global screen
        init()
        display.set_caption("Battle City")
        screen = display.set_mode((480, 416))
        self.clock = time.Clock()

        self.level = Level("level01.txt", "images/wall1.png")
        self.player = Tank(0, 100, 100, "images/mytank.png")

    def draw(self):
        global screen

        screen.fill(BLACK)
        self.player.draw()
        self.level.draw()
        # Bullets
        # Score/Health

        display.flip()

    def start(self):
        while True:
            time_passed = self.clock.tick(60)

            for e in event.get():
                if e.type == QUIT:
                    quit()
                elif e.type == KEYDOWN:
                    if e.key == K_LEFT:
                        self.player.speed = [-1, 0]
                        self.player.image = transform.rotate(self.player.init_image, 90)

                    if e.key == K_RIGHT:
                        self.player.speed = [1, 0]
                        self.player.image = transform.rotate(self.player.init_image, -90)

                    if e.key == K_UP:
                        self.player.speed = [0, -1]
                        self.player.image = self.player.init_image
                    if e.key == K_DOWN:
                        self.player.speed = [0, 1]
                        self.player.image = transform.rotate(self.player.init_image, 180)
                elif e.type == KEYUP:
                    self.player.speed = [0, 0]
            self.player.x += self.player.speed[0]
            self.player.y += self.player.speed[1]
            self.draw()


screen = None
game = Game()
game.start()