59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
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]])
|