75 lines
1.8 KiB
Python
75 lines
1.8 KiB
Python
import pygame
|
|
import sys
|
|
from elephant import Elephant
|
|
from agent import Agent
|
|
|
|
BLACK = (0, 0, 0)
|
|
|
|
GRID_SIZE = 75
|
|
GRID_WIDTH = 20
|
|
GRID_HEIGHT = 10
|
|
|
|
pygame.init()
|
|
|
|
WINDOW_SIZE = (GRID_WIDTH * GRID_SIZE, GRID_HEIGHT * GRID_SIZE)
|
|
screen = pygame.display.set_mode(WINDOW_SIZE)
|
|
pygame.display.set_caption("Mini Zoo")
|
|
|
|
|
|
|
|
background_image = pygame.image.load('tło.jpg')
|
|
background_image = pygame.transform.scale(background_image, WINDOW_SIZE)
|
|
|
|
exclamation_image = pygame.image.load('exclamation.png')
|
|
exclamation_image = pygame.transform.scale(exclamation_image, (GRID_SIZE,GRID_SIZE))
|
|
|
|
an1 = Elephant(10, 2)
|
|
an2 = Elephant(12, 2)
|
|
an3 = Elephant(14, 7)
|
|
old_an1 = Elephant(3, 6, adult=True)
|
|
|
|
Animals = [an1, an2, an3,old_an1]
|
|
|
|
def draw_grid():
|
|
for y in range(0, GRID_HEIGHT * GRID_SIZE, GRID_SIZE):
|
|
for x in range(0, GRID_WIDTH * GRID_SIZE, GRID_SIZE):
|
|
rect = pygame.Rect(x, y, GRID_SIZE, GRID_SIZE)
|
|
pygame.draw.rect(screen, BLACK, rect, 1)
|
|
|
|
def draw_exclamation(x, y):
|
|
screen.blit(exclamation_image, (x*GRID_SIZE, y*GRID_SIZE - GRID_SIZE))
|
|
|
|
def draw_Animals():
|
|
for Animal in Animals:
|
|
Animal.draw(screen, GRID_SIZE)
|
|
if Animal.feed() == 'True':
|
|
draw_exclamation(Animal.x, Animal.y)
|
|
|
|
|
|
|
|
def main():
|
|
agent = Agent(0, 0, 'avatar.png', GRID_SIZE)
|
|
clock = pygame.time.Clock()
|
|
|
|
while True:
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
pygame.quit()
|
|
sys.exit()
|
|
agent.handle_event(event,GRID_HEIGHT,GRID_WIDTH,Animals)
|
|
|
|
|
|
|
|
screen.blit(background_image,(0,0))
|
|
draw_grid()
|
|
draw_Animals()
|
|
|
|
|
|
|
|
agent.draw(screen)
|
|
pygame.display.flip()
|
|
clock.tick(10)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|