commit 5a40f4eb8175eb399fbf4c51b5d2994aa919a0c3 Author: Szymon Komosinski Date: Wed Jun 17 21:38:13 2020 +0200 First commit diff --git a/barn.png b/barn.png new file mode 100644 index 0000000..7b1e516 Binary files /dev/null and b/barn.png differ diff --git a/config.py b/config.py new file mode 100644 index 0000000..8462345 --- /dev/null +++ b/config.py @@ -0,0 +1,52 @@ +import pygame as pg +import numpy as np + +screen_height = 560 +screen_width = 700 +tile_size = 80 + +screen = pg.display.set_mode([screen_width, screen_height]) + +# agent +agent_img = pg.image.load('tractor.png').convert() + +# background +road = 0 +ground = 1 + +bg_textures = { + road: pg.image.load('road.png').convert(), + ground: pg.image.load('ground.png').convert()} + +background = np.array([[1, 1, 1, 0, 1, 1, 1], + [1, 1, 1, 0, 1, 1, 1], + [1, 1, 1, 0, 1, 1, 1], + [0, 0, 0, 0, 0, 0, 0], + [1, 1, 1, 0, 1, 1, 1], + [1, 1, 1, 0, 1, 1, 1], + [1, 1, 1, 0, 1, 1, 1]]) + +# objects +rose = 2 +tulip = 3 +orchid = 4 +sunflower = 5 +barn = 6 +disease = 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(), + disease: pg.image.load('disease.png').convert_alpha()} + +objects = np.array([[2, 2, 2, 0, 3, 3, 3], + [2, 2, 2, 0, 3, 3, 3], + [2, 2, 2, 0, 3, 3, 3], + [0, 0, 0, 6, 0, 0, 0], + [4, 4, 4, 0, 5, 5, 5], + [4, 4, 4, 0, 5, 5, 5], + [4, 4, 4, 0, 5, 5, 5]]) diff --git a/disease.png b/disease.png new file mode 100644 index 0000000..4a20d61 Binary files /dev/null and b/disease.png differ diff --git a/ground.png b/ground.png new file mode 100644 index 0000000..a740e0f Binary files /dev/null and b/ground.png differ diff --git a/main.py b/main.py new file mode 100644 index 0000000..09011b5 --- /dev/null +++ b/main.py @@ -0,0 +1,38 @@ +import pygame as pg +import sys +import numpy as np +from tile import Tile +from config import * + +done = False + +clock = pg.time.Clock() +pg.display.set_caption('Intelligent Tractor') +tiles = [] + +for y in range(0, 7): + for x in range(0, 7): + t = Tile(background[x][y], objects[x][y], x, y) + tiles.append(t) + + +def display_background(): + for tile in tiles: + screen.blit(bg_textures[tile.ground], (tile.x * tile_size, tile.y * tile_size)) + + +def display_objects(): + for tile in tiles: + screen.blit(ob_textures[tile.object], (tile.x * tile_size, tile.y * tile_size)) + pg.display.flip() + + +while not done: + for event in pg.event.get(): + if event.type == pg.QUIT: + done = True + display_background() + display_objects() + clock.tick(60) + +pg.quit() \ No newline at end of file diff --git a/none.png b/none.png new file mode 100644 index 0000000..cd8241c Binary files /dev/null and b/none.png differ diff --git a/orchid.png b/orchid.png new file mode 100644 index 0000000..632bb5d Binary files /dev/null and b/orchid.png differ diff --git a/road.png b/road.png new file mode 100644 index 0000000..2e696bc Binary files /dev/null and b/road.png differ diff --git a/rose.png b/rose.png new file mode 100644 index 0000000..15a1866 Binary files /dev/null and b/rose.png differ diff --git a/sunflower.png b/sunflower.png new file mode 100644 index 0000000..3d15758 Binary files /dev/null and b/sunflower.png differ diff --git a/tile.py b/tile.py new file mode 100644 index 0000000..760f417 --- /dev/null +++ b/tile.py @@ -0,0 +1,7 @@ +class Tile: + def __init__(self, a, b, x, y): + self.ground = a + self.object = b + self.x = x + self.y = y + self.h = 0 \ No newline at end of file diff --git a/tractor.png b/tractor.png new file mode 100644 index 0000000..3a71b60 Binary files /dev/null and b/tractor.png differ diff --git a/tulip.png b/tulip.png new file mode 100644 index 0000000..e757934 Binary files /dev/null and b/tulip.png differ diff --git a/water.png b/water.png new file mode 100644 index 0000000..1d68564 Binary files /dev/null and b/water.png differ