SzybciorSmartTraktor/main.py
Mateusz Cwojdziński 35dac32d41 Add AI agent movement
2022-03-09 18:59:50 +01:00

44 lines
1.1 KiB
Python

import pygame
WIDTH, HEIGHT = 800, 800
WIN = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Sztuczna Inteligencja")
FPS = 30
AGENT_IMG = pygame.image.load("dot-22-433567.png")
AGENT = pygame.transform.scale(AGENT_IMG, (50, 50))
def agent_movement(keys_pressed, agent):
if keys_pressed[pygame.K_LEFT] and agent.x > 0:
agent.x -= 50
if keys_pressed[pygame.K_RIGHT] and agent.x < 750:
agent.x += 50
if keys_pressed[pygame.K_UP] and agent.y > 0:
agent.y -= 50
if keys_pressed[pygame.K_DOWN] and agent.y < 750:
agent.y += 50
def draw_window(agent):
WIN.fill((255,255,255))
WIN.blit(AGENT, (agent.x, agent.y))
pygame.display.update()
def main():
clock = pygame.time.Clock()
run = True
agent = pygame.Rect(0, 0, 50, 50)
while run:
clock.tick(FPS)
for event in pygame.event.get():
if event.type == pygame.QUIT:
run = False
draw_window(agent)
keys_pressed = pygame.key.get_pressed()
agent_movement(keys_pressed, agent)
pygame.quit()
if __name__ == "__main__":
main()