figure.py

import random


class Figure:
    # (x, y)
    figures = {
        1: [(-1, 0), (0, 0), (1, 0), (1, 1)],  # L-shape left-handed
        2: [(-1, 0), (0, 0), (1, 0), (1, -1)],  # L-shape right-handed
        3: [(0, 0), (0, -1), (1, -1), (-1, 0)],  # S-shape left-handed
        4: [(0, 0), (0, 1), (1, 1), (-1, 0)],  # S-shape right-handed
        5: [(0, 0), (0, -1), (0, 1), (1, 0)],  # T-shape
        6: [(0, 0), (0, 1), (1, 0), (1, 1)],  # square
        7: [(0, -1), (0, 0), (0, 1), (0, 2)],  # straight bar
    }
    figure = None
    is_falling = None

    def __init__(self, max_x, max_y, config):
        self.max_x = max_x
        self.max_y = max_y
        self.figure = self.figures[random.randint(1, len(self.figures))]
        self.figure_height = self.figure[len(self.figure) - 1][1] + 1
        self.figure_width = self.figure[len(self.figure) - 1][0] + 1
        self.config = config
        self.is_dead = False
        self.x = random.randint(0, max_x - self.figure_width)
        self.y = 0
        self.pushback_to_grid()
        if self.is_colliding():
            raise Exception("Initial collision")

    def update_dimensions(self):
        self.figure_height = self.figure[len(self.figure) - 1][1] + 1
        self.figure_width = self.figure[len(self.figure) - 1][0] + 1

    def destroy_me(self):
        self.y -= 1
        for block in self.figure:
            self.config.grid[self.y + block[1]][self.x + block[0]] = "X"
        self.is_dead = True

    def pushback_to_grid(self):
        """push the tetromino back to the grid if partially or fully outside of boundaries"""
        is_outside_boundaries = True
        while is_outside_boundaries:
            for block in self.figure:
                if self.x + block[0] < 0:
                    self.x += 1
                    continue
                if self.x + block[0] >= self.max_x:
                    self.x -= 1
                    continue
                if self.y + block[1] >= self.max_y:
                    self.y -= 1
                    continue
                if self.y + block[1] < 0:
                    self.y += 1
                    continue
                is_outside_boundaries = False

    def is_colliding(self):
        """check for collision with settled tetrominos on the grid"""
        for block in self.figure:
            if self.config.grid[self.y + block[1]][self.x + block[0]] == "X":
                return True
        return False

    def move_step_down(self):
        if not self.is_dead:
            self.pushback_to_grid()
            self.y += 1
            for block in self.figure:
                if self.y + block[1] >= self.max_y:
                    self.destroy_me()
                    break
            if not self.is_dead:
                if self.is_colliding():
                    self.destroy_me()

    def move_to_bottom(self):
        while not self.is_dead:
            self.move_step_down()

    def move_right(self):
        self.x += 1
        self.pushback_to_grid()
        if self.is_colliding():
            self.x -= 1

    def move_left(self):
        self.x -= 1
        self.pushback_to_grid()
        if self.is_colliding():
            self.x += 1

    def rotate_right(self):
        if self.y + 1 >= self.figure[len(self.figure) - 1][0] + 1:
            self.figure = [
                (self.figure[i][1], -self.figure[i][0]) for i in range(len(self.figure))
            ]
            self.pushback_to_grid()
            if self.is_colliding():
                # force rotate left
                self.figure = [
                    (-self.figure[i][1], self.figure[i][0]) for i in range(len(self.figure))
                ]
                self.pushback_to_grid()

    def rotate_left(self):
        if self.y + 1 >= self.figure[len(self.figure) - 1][0] + 1:
            self.figure = [
                (-self.figure[i][1], self.figure[i][0]) for i in range(len(self.figure))
            ]
            self.pushback_to_grid()
            if self.is_colliding():
                # force rotate right
                self.figure = [
                    (self.figure[i][1], -self.figure[i][0]) for i in range(len(self.figure))
                ]
                self.pushback_to_grid()

    def draw(self, grid):
        for pos in self.figure:
            grid[self.y + pos[1]][self.x + pos[0]] = "X"
        return grid