35 lines
1.8 KiB
Python
35 lines
1.8 KiB
Python
import definitions
|
|
import graph
|
|
import map
|
|
import plant
|
|
import pygame
|
|
import station
|
|
import tractor
|
|
pygame.display.set_caption("Smart Tractor")
|
|
def main():
|
|
#tworzenie podstawowych obiektów
|
|
map1 = map.Map([])
|
|
map1.create_base_map()
|
|
amount_of_seeds_dict = {"beetroot": definitions.TRACTOR_AMOUNT_OF_SEEDS_EACH_TYPE, "carrot": definitions.TRACTOR_AMOUNT_OF_SEEDS_EACH_TYPE, "potato": definitions.TRACTOR_AMOUNT_OF_SEEDS_EACH_TYPE, "wheat": definitions.TRACTOR_AMOUNT_OF_SEEDS_EACH_TYPE}
|
|
collected_plants_dict = {"beetroot": 0, "carrot": 0, "potato": 0, "wheat": 0}
|
|
fertilizer_dict = {"beetroot": definitions.TRACTOR_FERTILIZER, "carrot": definitions.TRACTOR_FERTILIZER, "potato": definitions.TRACTOR_FERTILIZER, "wheat": definitions.TRACTOR_FERTILIZER}
|
|
station1 = station.Station(collected_plants_dict)
|
|
tractor1 = tractor.Tractor(amount_of_seeds_dict, collected_plants_dict, definitions.TRACTOR_DIRECTION_EAST, fertilizer_dict, definitions.TRACTOR_FUEL, definitions.TRACTOR_WATER_LEVEL, 0, 0)
|
|
tractor1_rect = pygame.Rect(tractor1.get_x(), tractor1.get_y(), definitions.BLOCK_SIZE, definitions.BLOCK_SIZE)
|
|
clock = pygame.time.Clock()
|
|
run = True
|
|
while run: #pętla główna programu
|
|
clock.tick(definitions.FPS)
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
run = False
|
|
istate = graph.Istate(definitions.TRACTOR_DIRECTION_EAST, 0, 0)
|
|
istate1 = graph.Istate(None, 4, 4)
|
|
print(graph.graphsearch([], [], istate, graph.succ, istate1))
|
|
map1.draw_window(tractor1, tractor1_rect)
|
|
tractor1.tractor1_handle_movement(tractor1_rect)
|
|
tractor1.do_work(map1, station1, tractor1_rect)
|
|
plant.Plant.grow_plants(map1)
|
|
pygame.quit()
|
|
if __name__ == "__main__":
|
|
main() |