import pygame as pg
import sys
import numpy as np
from tile import Tile
from config import *
from agent import Agent
from Astar import *

done = False

clock = pg.time.Clock()
pg.display.set_caption('Intelligent Tractor')
pg.init()
pg.font.init()


class Game:
    def __init__(self):
        self.tiles = []
        self.water_lvl = []
        self.day = 1

        for y in range(0, 9):
            for x in range(0, 9):
                pos = (x, y)
                t = Tile(background[x][y], objects[x][y], pos)
                t.water_usage()
                self.tiles.append(t)

    def display_background(self):
        for tile in self.tiles:
            screen.blit(bg_textures[tile.ground], (tile.x * tile_size, tile.y * tile_size))

    def display_objects(self):
        for tile in self.tiles:
            screen.blit(ob_textures[tile.object], (tile.x * tile_size, tile.y * tile_size))

    def display_legend(self):
        screen.blit(black_surface, (0, 720))
        self.water_lvl = []
        for tile in self.tiles:
            if tile.pos == (1, 1):
                rose_w = tile.w
                self.water_lvl.append((rose_w, (0, 3)))
            if tile.pos == (7, 1):
                orchid_w = tile.w
                self.water_lvl.append((orchid_w, (5, 0)))
            if tile.pos == (1, 7):
                tulip_w = tile.w
                self.water_lvl.append((tulip_w, (3, 8)))
            if tile.pos == (7, 7):
                sunflower_w = tile.w
                self.water_lvl.append((sunflower_w, (8, 5)))
        font = pg.font.SysFont('ubuntumono', 40)
        text = font.render("water levels", 1, (255, 255, 255))
        text1 = font.render("roses: " + str(rose_w) + "%", 1, (255, 255, 255))
        text2 = font.render("orchids: " + str(orchid_w) + "%", 1, (255, 255, 255))
        text3 = font.render("tulips: " + str(tulip_w) + "%", 1, (255, 255, 255))
        text4 = font.render("sunflowers: " + str(sunflower_w) + "%", 1, (255, 255, 255))
        text5 = font.render("day: " + str(self.day), 1, (255, 255, 255))
        screen.blit(text, (10, 720))
        screen.blit(text1, (10, 760))
        screen.blit(text2, (360, 760))
        screen.blit(text3, (10, 800))
        screen.blit(text4, (360, 800))
        screen.blit(text5, (10, 840))

    def water_target(self):
        o = None
        for lvl in self.water_lvl:
            if lvl[0] < 10:
                o = lvl[1]
                break
        return o

    def watering(self, plants):
        if plants == (0, 3):
            for tile in self.tiles:
                if tile.object == rose:
                    tile.w = 100
        if plants == (5, 0):
            for tile in self.tiles:
                if tile.object == orchid:
                    tile.w = 100
        if plants == (3, 8):
            for tile in self.tiles:
                if tile.object == tulip:
                    tile.w = 100
        if plants == (8, 5):
            for tile in self.tiles:
                if tile.object == sunflower:
                    tile.w = 100

    def add_day(self):
        self.day = self.day + 1
        for tile in self.tiles:
            tile.w = tile.w - tile.wu
            if tile.w < 0:
                tile.w = 0


def game_loop():
    path = None
    g.display_background()
    g.display_objects()
    g.display_legend()
    if g.water_target() is None:
        g.add_day()
    else:
        path = astar(background, a.pos, g.water_target())
    if path is not None:
        if len(path) == 1:
            g.watering(path[0])
            path = None
        else:
            a.display(path[1])
    a.display()
    clock.tick(60)


g = Game()
a = Agent()
while not done:
    for event in pg.event.get():
        if event.type == pg.QUIT:
            done = True
    game_loop()

pg.quit()