import pygame

from util.ConfigurationManager import ConfigurationManager

''' 
    For information on the Tetris piece Tetromino go here:
    https://tetris.fandom.com/wiki/Tetromino
'''
class Piece: # TODO game objects base class / interface? 
        
    def __init__(self, shape, position, color):
        self.color = color
        self.center, self.points = self.__get_points(shape, position)

    def __get_points(self, shape, position):
        tile_size = ConfigurationManager.configuration["engine"]["tile-size"]

        center = ()
        points = []
        for square in shape:
            sub_points = []
            for vertex in square:
                # convert shape vertex into game surface point
                point = [vertex[0] * tile_size + position[0], vertex[1] * tile_size + position[1]]
                sub_points.append(point)

                # store center of gravity
                if len(vertex) > 2 and vertex[2]:
                    center = point

            points.append(sub_points)

        return center, points

    def draw(self, surface):
        tile_size = ConfigurationManager.configuration["engine"]["tile-size"]
        hex_color = ConfigurationManager.configuration["color"]["border"] # TODO Should abstract out color call?

        base_color = pygame.Color(self.color) 
        border_color = pygame.Color(hex_color)

        for sub_points in self.points:
            pygame.draw.polygon(surface, base_color, sub_points, 0)
            pygame.draw.polygon(surface, border_color, sub_points, max(tile_size // 6, 1))

    def move(self, vector):
        for sub_points in self.points:
            for point in sub_points:
                point[0] += vector[0]
                point[1] += vector[1]
 
    # rotate the piece clockwise
    def rotate(self):
        for sub_points in self.points:
            for point in sub_points:
                h = point[0] - self.center[0]
                k = point[1] - self.center[1]

                point[0] = k + self.center[0]
                point[1] = (h * -1) + self.center[1]

    # shape attributes
    I_SHAPE = (((0, 0), (1, 0), (1, 1), (0, 1)), ((1, 0), (2, 0), (2, 1), (1, 1)), ((2, 0, True), (3, 0), (3, 1), (2, 1)), ((3, 0), (4, 0), (4, 1), (3, 1)))