import pygame as pg
import numpy as np

screen_height = 890
screen_width = 720
tile_size = 80

screen = pg.display.set_mode([screen_width, screen_height])
black_surface = pg.Surface((720, 170))
black_surface.fill((0, 0, 0))

# # agent
# agent_img = pg.image.load('tractor.png').convert_alpha()

# background
road = 0
ground = 1
wall = 2

bg_textures = {
    road: pg.image.load('road.png').convert(),
    ground: pg.image.load('ground.png').convert(),
    wall: pg.image.load('wall.png').convert()}

background = np.array([[1, 1, 1, 0, 0, 0, 1, 1, 1],
                       [1, 1, 1, 2, 0, 2, 1, 1, 1],
                       [1, 1, 1, 2, 0, 2, 1, 1, 1],
                       [0, 2, 2, 2, 0, 2, 2, 2, 0],
                       [0, 0, 0, 0, 0, 0, 0, 0, 0],
                       [0, 2, 2, 2, 0, 2, 2, 2, 0],
                       [1, 1, 1, 2, 0, 2, 1, 1, 1],
                       [1, 1, 1, 2, 0, 2, 1, 1, 1],
                       [1, 1, 1, 0, 0, 0, 1, 1, 1]])

# objects
rose = 3
tulip = 4
orchid = 5
sunflower = 6
barn = 7

ob_textures = {
    0: pg.image.load('none.png').convert_alpha(),
    rose: pg.image.load('rose.png').convert_alpha(),
    tulip: pg.image.load('tulip.png').convert_alpha(),
    orchid: pg.image.load('orchid.png').convert_alpha(),
    sunflower: pg.image.load('sunflower.png').convert_alpha(),
    barn: pg.image.load('barn.png').convert()}

objects = np.array([[3, 3, 3, 0, 0, 0, 4, 4, 4],
                    [3, 3, 3, 0, 0, 0, 4, 4, 4],
                    [3, 3, 3, 0, 0, 0, 4, 4, 4],
                    [0, 0, 0, 0, 0, 0, 0, 0, 0],
                    [0, 0, 0, 0, 7, 0, 0, 0, 0],
                    [0, 0, 0, 0, 0, 0, 0, 0, 0],
                    [5, 5, 5, 0, 0, 0, 6, 6, 6],
                    [5, 5, 5, 0, 0, 0, 6, 6, 6],
                    [5, 5, 5, 0, 0, 0, 6, 6, 6]])