import pygame
from . import sprites


class Surface(pygame.Surface):
    def __init__(self, window_size, resources_dir, game_fps):
        super().__init__(window_size, pygame.SRCALPHA)
        self.name = 'egg_select'
        self.running = True
        self.next_surface = None

        self.bg = pygame.image.load(resources_dir + '/images/bg.png').convert_alpha()
        self.sprites = pygame.sprite.Group()

        egg_list = [
            'dev_egg',
            'blue',
            'rainbow'
        ]
        eggs = []
        for egg in egg_list:
            eggs.append(sprites.SelectionEgg(egg, resources_dir))

        eggs_per_row = 3
        distance_between_eggs = 36 / eggs_per_row

        # Count the total rows.
        total_rows = -(-len(eggs) // eggs_per_row)
        distance_between_rows = 32 / eggs_per_row

        # Determine the location of each egg.
        for egg in eggs:
            current_row = eggs.index(egg) // eggs_per_row
            rows_after = total_rows - (current_row + 1)
            egg_in_row = eggs.index(egg) % eggs_per_row
            eggs_after = min(len(eggs) - (current_row * eggs_per_row), eggs_per_row) - (egg_in_row + 1)

            x_offset = 32
            y_offset = 30

            # The x coordinate of an egg is determined by which egg in the row it is, and how many eggs
            # are in that row. If there is only 1 egg in a row, it is in the middle of the screen. If
            # there are two, they're on equal halves and so on.
            x = x_offset - (eggs_after * distance_between_eggs) + (egg_in_row * distance_between_eggs)
            y = y_offset - (rows_after * distance_between_rows) + (current_row * distance_between_rows)

            egg.rect.x = x
            egg.rect.y = y

            # Add the egg to the sprite list.
            self.sprites.add(egg)

    def update(self):
        self.blit(self.bg, (0, 0))
        self.sprites.update()
        self.sprites.draw(self)