import pygame

pygame.init()


class Smieciarka(pygame.sprite.Sprite):
    def __init__(self, x, y):
        self.x = x
        self.y = y
        self.image = pygame.image.load('resources/plansza/smieciarka.png')
        self.ruch = 0
        pygame.sprite.Sprite.__init__(self)
        self.rect = pygame.Rect(self.x * WIDTH + MARGIN * self.x + MARGIN, self.y * HEIGHT + MARGIN * self.y, WIDTH,
                                HEIGHT)

    def w_lewo(self):
        if self.x > 0:
            self.x -= 1
            self.rect.x = MARGIN + self.x * WIDTH + self.x * MARGIN
            if self.ruch == 2:
                self.image = pygame.image.load(
                    'resources/plansza/smieciarka.png')
            self.ruch = 1

    def w_prawo(self):
        if self.x < 14:
            self.x += 1
            self.rect.x = MARGIN + self.x * WIDTH + self.x * MARGIN
            if self.ruch == 1:
                self.image = pygame.transform.flip(
                    smieciarka.image, True, False)
            self.ruch = 2

    def w_gore(self):
        if self.y > 0:
            self.y -= 1
            self.rect.y = self.y * HEIGHT + self.y * MARGIN

    def w_dol(self):
        if self.y < 14:
            self.y += 1
            self.rect.y = self.y * HEIGHT + self.y * MARGIN


class Kontener(pygame.sprite.Sprite):
    def __init__(self, x, y):
        self.x = x
        self.y = y
        pygame.sprite.Sprite.__init__(self)
        self.rect = pygame.Rect(self.x * WIDTH + MARGIN * self.x + MARGIN, self.y * HEIGHT + MARGIN * self.y + MARGIN,
                                WIDTH, HEIGHT)


# Define some colors
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
GREEN = (0, 255, 0)
RED = (255, 0, 0)
BLUE = (0, 0, 255)
GREY = (128, 128, 128)

# This sets the WIDTH and HEIGHT of each grid location
WIDTH = 60
HEIGHT = 60

# This sets the margin between each cell
MARGIN = 5

# Create a 2 dimensional array. A two dimensional
# array is simply a list of lists.
grid = []
for row in range(100):
    # Add an empty array that will hold each cell
    # in this row
    grid.append([])
    for column in range(100):
        grid[row].append(0)  # Append a cell

# Set row 1, cell 5 to one. (Remember rows and
# column numbers start at zero.)
grid[0][0] = 1

# Set the HEIGHT and WIDTH of the screen
WINDOW_SIZE = [980, 980]
screen = pygame.display.set_mode(WINDOW_SIZE)

# Set title of screen
pygame.display.set_caption("Inteligentna śmieciarka")

# Loop until the user clicks the close button.
done = False

# Used to manage how fast the screen updates
clock = pygame.time.Clock()

smieciarka = Smieciarka(10, 10)
kontener_plastik = Kontener(0, 0)
kontener_plastik.image = pygame.image.load(
    "resources/plansza/pojemnik_plastik.png")
kontener_metal = Kontener(0, 4)
kontener_metal.image = pygame.image.load(
    "resources/plansza/pojemnik_metal.png")
kontener_organiczne = Kontener(2, 2)
kontener_organiczne.image = pygame.image.load(
    "resources/plansza/pojemnik_organiczne.png")
kontener_papier = Kontener(4, 0)
kontener_papier.image = pygame.image.load(
    "resources/plansza/pojemnik_papier.png")
kontener_szklo = Kontener(4, 4)
kontener_szklo.image = pygame.image.load(
    "resources/plansza/pojemnik_szklo.png")

kontener_list = pygame.sprite.Group()
smieciarka_list = pygame.sprite.Group()
all_sprites_list = pygame.sprite.Group()

kontener_list.add(kontener_plastik, kontener_metal,
                  kontener_organiczne, kontener_papier, kontener_szklo)
smieciarka_list.add(smieciarka)
all_sprites_list.add(kontener_plastik, kontener_metal,
                     kontener_organiczne, kontener_papier, kontener_szklo, smieciarka)

# -------- Main Program Loop -----------
while not done:
    for event in pygame.event.get():  # User did something
        if event.type == pygame.QUIT:  # If user clicked close
            done = True  # Flag that we are done so we exit this loop
        elif event.type == pygame.MOUSEBUTTONDOWN:
            # User clicks the mouse. Get the position
            pos = pygame.mouse.get_pos()
            # Change the x/y screen coordinates to grid coordinates
            column = pos[0] // (WIDTH + MARGIN)
            row = pos[1] // (HEIGHT + MARGIN)
            # Set that location to one
            grid[row][column] = 1
            print("Click ", pos, "Grid coordinates: ", row, column)
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT:
                smieciarka.w_lewo()
            if event.key == pygame.K_RIGHT:
                smieciarka.w_prawo()
            if event.key == pygame.K_UP:
                smieciarka.w_gore()
            if event.key == pygame.K_DOWN:
                smieciarka.w_dol()

    # Set the screen background
    screen.fill(BLACK)

    # Draw the grid
    for row in range(15):
        for column in range(15):
            color = GREY
            # if grid[row][column] == 1:
            #     color = BLUE
            pygame.draw.rect(screen,
                             color,
                             [(MARGIN + WIDTH) * column + MARGIN,
                              (MARGIN + HEIGHT) * row + MARGIN,
                              WIDTH,
                              HEIGHT])
    screen.blit(pygame.image.load("resources/plansza/wysypisko.jpg"), (5, 5))
    all_sprites_list.draw(screen)

    # Limit to 60 frames per second
    clock.tick(60)

    # Go ahead and update the screen with what we've drawn.
    pygame.display.flip()

# Be IDLE friendly. If you forget this line, the program will 'hang'
# on exit.
pygame.quit()