import os import pygame import random import land import tractor import blocks import graph_search from pygame.locals import * class Game: cell_size = 50 cell_number = 15 # horizontally blocks_number = 15 def __init__(self): self.dead_leaf_body = [] self.green_leaf_body = [] self.stone_body = [] self.flower_body = [] self.dead_grass_body = [] self.grass_body = [] self.fawn_seed_body = [] self.fawn_wheat_body = [] self.black_earth_body = [] self.green_earth_body = [] self.fawn_soil_body = [] self.fen_soil_body = [] self.allBodyPos = [] self.entire_block = {} # initialize a window pygame.init() self.surface = pygame.display.set_mode((self.cell_size*self.cell_number, self.cell_size*self.cell_number)) # finds places for every type soil and grass self.black_earth = land.Land(self.surface, self.cell_size, self.cell_number, self.allBodyPos, 100) self.black_earth.locate_soil(self.black_earth_body) self.green_earth = land.Land(self.surface, self.cell_size, self.cell_number, self.allBodyPos, 100) self.green_earth.locate_soil(self.green_earth_body) self.fawn_soil = land.Land(self.surface, self.cell_size, self.cell_number, self.allBodyPos, 100) self.fawn_soil.locate_soil(self.fawn_soil_body) self.fen_soil = land.Land(self.surface, self.cell_size, self.cell_number, self.allBodyPos, 100) self.fen_soil.locate_soil(self.fen_soil_body) self.grass = land.Land(self.surface, self.cell_size, self.cell_number, self.allBodyPos, 100) self.blocks = blocks.Blocks(self.surface, self.cell_size) self.blocks.locate_blocks(self.blocks_number, self.cell_number, self.dead_leaf_body) self.blocks.locate_blocks(self.blocks_number, self.cell_number, self.stone_body) self.blocks.locate_blocks(self.blocks_number, self.cell_number, self.flower_body) # self.potato = blocks.Blocks(self.surface, self.cell_size) # self.potato.locate_soil('black earth', 6, 1, []) self.tractor = tractor.Tractor(self.surface, self.cell_size) self.tractor.draw() def run(self): # print(self.potato.get_soil_info().get_name()) # print(self.potato.get_soil_info().get_acidity()) # print(self.potato.get_soil_info().get_irrigation()) running = True clock = pygame.time.Clock() move_tractor_event = pygame.USEREVENT + 1 pygame.time.set_timer(move_tractor_event, 1000) # tractor moves every 1000 ms tractor_next_moves = [] graph_search_object = graph_search.Search(self.cell_size, self.cell_number) while running: clock.tick(60) # manual fps control not to overwork the computer for event in pygame.event.get(): if event.type == KEYDOWN: if pygame.key.get_pressed()[K_ESCAPE]: running = False # in case we want to use keyboard if pygame.key.get_pressed()[K_UP]: # self.tractor.move('up', self.cell_size, self.cell_number) self.tractor.move('move', self.cell_size, self.cell_number) # if pygame.key.get_pressed()[K_DOWN]: # self.tractor.move('down', self.cell_size, self.cell_number) if pygame.key.get_pressed()[K_LEFT]: self.tractor.move('left', self.cell_size, self.cell_number) if pygame.key.get_pressed()[K_RIGHT]: self.tractor.move('right', self.cell_size, self.cell_number) if pygame.key.get_pressed()[K_SPACE]: self.tractor.water(self.dead_leaf_body, self.green_leaf_body, self.cell_size) # self.tractor.water(self.grass_body, self.dead_grass_body, self.cell_size) if pygame.key.get_pressed()[K_q]: self.tractor.harvest(self.fawn_seed_body, self.fawn_wheat_body, self.cell_size) self.tractor.put_seed(self.fawn_soil_body, self.fawn_seed_body, self.cell_size) if event.type == move_tractor_event: if len(tractor_next_moves) == 0: random_x = random.randrange(0, self.cell_number * self.cell_size, 50) random_y = random.randrange(0, self.cell_number * self.cell_size, 50) print("Generated target: ",random_x, random_y) tractor_next_moves = graph_search_object.graphsearch( [self.tractor.x, self.tractor.y, self.tractor.angle], [random_x, random_y]) else: self.tractor.move(tractor_next_moves.pop(0)[0], self.cell_size, self.cell_number) elif event.type == QUIT: running = False self.surface.fill((123, 56, 51)) # background color self.grass.set_and_place_block_of_grass('good') self.black_earth.place_soil(self.black_earth_body, 'black_earth') self.green_earth.place_soil(self.green_earth_body, 'green_earth') self.fawn_soil.place_soil(self.fawn_soil_body, 'fawn_soil') self.fen_soil.place_soil(self.fen_soil_body, 'fen_soil') # plants examples self.blocks.place_blocks(self.surface, self.cell_size, self.dead_leaf_body, 'leaf') self.blocks.place_blocks(self.surface, self.cell_size, self.green_leaf_body, 'alive') self.blocks.place_blocks(self.surface, self.cell_size, self.stone_body, 'stone') self.blocks.place_blocks(self.surface, self.cell_size, self.flower_body, 'flower') # seeds self.blocks.place_blocks(self.surface, self.cell_size, self.fawn_seed_body, 'fawn_seed') # wheat self.blocks.place_blocks(self.surface, self.cell_size, self.fawn_wheat_body, 'fawn_wheat') self.tractor.draw() pygame.display.update() if __name__ == '__main__': game = Game() game.run()