Zaktualizuj 'basic_grid.py'

This commit is contained in:
Cezary Adamczak 2021-05-10 14:35:41 +02:00
parent bb05f4b310
commit 2830efa247

View File

@ -1,3 +1,5 @@
### TO DO: PLANT SPRITES, PLANT GROWTH ###
# Import the pygame module
import pygame
@ -5,20 +7,20 @@ import pygame
# Updated to conform to flake8 and black standards
from pygame.locals import (
K_UP,
K_DOWN,
K_LEFT,
K_RIGHT,
K_ESCAPE,
K_SPACE,
KEYDOWN,
QUIT
)
from field import *
from tractor import *
from plant import *
from colors import *
from dimensions import *
import field as F
import tractor as T
import plant as P
import colors as C
import dimensions as D
import node as N
import mapschema as maps
# Initialize pygame
pygame.init()
@ -28,32 +30,46 @@ pygame.display.set_caption("Inteligentny Traktor")
# Create the screen object
# The size is determined by the constant SCREEN_WIDTH and SCREEN_HEIGHT
screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
screen = pygame.display.set_mode((D.SCREEN_WIDTH, D.SCREEN_HEIGHT))
#define map
mapschema = maps.createField()
# Create field array
field = []
for row in range(GSIZE):
for row in range(D.GSIZE):
field.append([])
for column in range(GSIZE):
fieldbit = Field(row, column)
for column in range(D.GSIZE):
fieldbit = F.Field(row, column, mapschema[column][row])
field[row].append(fieldbit)
tractor = Tractor(field[0][0])
plant = Plant(field[1][1], "wheat")
tractor = T.Tractor(field[0][0])
mapschema = maps.createPlants()
plants = []
for row in range(D.GSIZE):
plants.append([])
for column in range(D.GSIZE):
if mapschema[column][row] != 0:
plantbit = P.Plant(field[row][column], mapschema[column][row])
plants[row].append(plantbit)
print(tractor.rect)
print(field[0][0].rect)
path = []
# Variable to keep the main loop running
RUNNING = True
ticker = 0
TICKER = 0
clock = pygame.time.Clock()
# Main loop
while RUNNING:
@ -68,37 +84,62 @@ while RUNNING:
elif event.type == QUIT:
RUNNING = False
processor = N.Node(field, tractor.position, tractor.direction)
if path is None or len(path) == 0:
path = processor.findPathToPlant()
if path is not None:
if path[0] == "move":
tractor.move()
path.pop(0)
elif path[0] =="left":
tractor.rotate_left()
path.pop(0)
elif path[0] == "right":
tractor.rotate_right()
path.pop(0)
elif path[0] == "hydrate":
tractor.hydrate(field)
path.pop(0)
else:
path.pop(0)
# Get all keys pressed at a time
pressed_keys = pygame.key.get_pressed()
# Update the state of tractor
tractor.update(pressed_keys)
tractor.hydrate(field, pressed_keys)
if pressed_keys[K_UP]:
tractor.move()
elif pressed_keys[K_LEFT]:
tractor.rotate_left()
elif pressed_keys[K_RIGHT]:
tractor.rotate_right()
# Set the screen background
screen.fill(DBROWN)
screen.fill(C.DBROWN)
# Draw the field
for row in range(GSIZE):
for column in range(GSIZE):
for row in range(D.GSIZE):
for column in range(D.GSIZE):
screen.blit(field[row][column].surf, field[row][column].rect)
screen.blit(tractor.surf, tractor.rect)
# Draw the player on the screen
for row in plants:
for plant in row:
if TICKER % 10 == 0:
plant.grow()
screen.blit(plant.surf, plant.rect)
if ticker == 0:
for row in range(GSIZE):
for column in range(GSIZE):
if TICKER == 0:
for row in range(D.GSIZE):
for column in range(D.GSIZE):
field[row][column].dehydrate()
plant.grow()
ticker = (ticker + 1)%4
TICKER = (TICKER + 1)%100
# Update the screen
pygame.display.flip()
# Ensure program maintains a rate of 15 frames per second
clock.tick(4)
clock.tick(8)