sequence.py

from parameters import *

class Sequence:
    def __init__(self) -> None:
        self.surface = pygame.Surface((RIGHT_SIDEBAR_WIDTH, SEQUENCE_HEIGHT))
        self.display_surface = pygame.display.get_surface()
        self.rect = self.surface.get_rect(topright=(WINDOW_WIDTH - WINDOW_PADDING, WINDOW_PADDING))
        self.font = pygame.font.Font(FONT_PATH, GAME_FONT_SIZE)

    def display_tetrominos(self, shapes: list[str]) -> None:
        for i, shape in enumerate(shapes):
            block_positions = TETROMINOS[shape]['shape']
            color = TETROMINOS[shape]['color']
            for pos in block_positions:
                x, y = pos
                x = (x + 2) * CELL_SIZE
                y = (y + 3 + (4 * i)) * CELL_SIZE
                if shape in ['I', 'O']:
                    x -= SEQUENCE_OFFSET_SIZE
                self.draw_block(x, y, color)

    def draw_block(self, x: int, y: int, color: str) -> None:
        line_color = COLORS['GAME_LINE_COLOR']
        pygame.draw.rect(self.surface, color, (x, y, CELL_SIZE, CELL_SIZE))
        pygame.draw.line(self.surface, line_color, start_pos=(x, y), end_pos=(x + CELL_SIZE, y), width=1)
        pygame.draw.line(self.surface, line_color, start_pos=(x, y), end_pos=(x, y + CELL_SIZE), width=1)
        pygame.draw.line(self.surface, line_color,
                        start_pos=(x + CELL_SIZE, y), end_pos=(x + CELL_SIZE, y + CELL_SIZE), width=1)
        pygame.draw.line(self.surface, line_color,
                        start_pos=(x, y + CELL_SIZE), end_pos=(x + CELL_SIZE, y + CELL_SIZE), width=1)
    
    def display_text(self) -> None:
        x = self.surface.get_width() * 0.5
        y = self.surface.get_height() * 0.05
        text_surface = self.font.render(TEXTS['SEQUENCE'], True, COLORS['TEXT'])
        text_rect = text_surface.get_rect(center = (x, y))
        self.surface.blit(text_surface, text_rect)

    def run(self, next_shapes: list[str]) -> None:
        self.surface.fill(COLORS['SEQUENCE_BACKGROUND'])
        self.display_text()
        self.display_tetrominos(next_shapes[0:3])
        self.display_surface.blit(self.surface, self.rect)
        pygame.draw.rect(self.display_surface, COLORS['BORDER_COLOR'], self.rect, 2, 2)