import os

import pygame
from pygame.rect import Rect

from game.player import Player
from game.quad_tree import QuadTree
from game.stone import Stone
from game.tile import Tile


class GameMap:

    def __init__(self, width, height):
        self.width = width
        self.height = height
        self.game_objects = []
        self.player = Player()
        self.game_objects.append(self.player)
        self.tiles = [[Tile() for x in range(width)] for y in range(height)]
        self.textures = [pygame.image.load(os.path.join('..', 'assets', 'map1.png')),
                         pygame.image.load(os.path.join('..', 'assets', 'map2.png')),
                         pygame.image.load(os.path.join('..', 'assets', 'map3.png')),
                         pygame.image.load(os.path.join('..', 'assets', 'map4.png'))]
        self.game_objects.append(Stone([100, 200]))
        self.quad_tree = QuadTree(0, Rect(0, 0, width * 32, height * 32))

    def draw(self, window):
        for y in range(self.height):
            for x in range(self.width):
                window.blit(self.textures[self.tiles[y][x].background_id], (x * 32, y * 32))

        for game_object in self.game_objects:
            game_object.draw(window)

    def update(self, pressed_keys):
        self.quad_tree.clear()

        for game_object in self.game_objects:
            self.quad_tree.insert(game_object)

        self.player.update(pressed_keys)

        for game_object in self.game_objects:
            possible_colliders = []
            self.quad_tree.retrieve(possible_colliders, game_object)
            for collider in possible_colliders:
                if game_object.get_rect().colliderect(collider.get_rect()) and game_object != collider:
                    game_object.velocity = [0, 0]
                    game_object.pos = game_object.last_pos