diff --git a/main.py b/main.py index d2a9e50..40d2f2d 100644 --- a/main.py +++ b/main.py @@ -1,16 +1,37 @@ import sys import pygame + pygame.init() #SETUP -WIDTH, HEIGHT = 800, 800 -WINDOW = pygame.display.set_mode((WIDTH, HEIGHT)) +WIDTH, HEIGHT = 800, 600 +screen = pygame.display.set_mode((WIDTH, HEIGHT)) pygame.display.set_caption("Traffic Controller") #CONSTANTS FPS = 60 -# Game loop +#CLOCK clock = pygame.time.Clock() + +#COLOURS + +ROAD_COLOUR = (50, 50, 50) +LINE_COLOUR = (255, 255, 0) +CAR_COLOUR = (200, 0, 0) +BG_COLOUR = (30, 150, 30) + +#CAR +car_width = 40 +car_height = 70 +car_x = WIDTH // 2 - car_width // 2 +car_y = HEIGHT - car_height - 20 +car_speed = 4 +lane_speed = 5 + +#LANE +lane_y = 0 + +#LOOP running = True while running: clock.tick(FPS) @@ -20,3 +41,31 @@ while running: pygame.quit() sys.exit() + #INPUT + keys = pygame.key.get_pressed() + if keys[pygame.K_LEFT]: + car_x -= car_speed + if keys[pygame.K_RIGHT]: + car_x += car_speed + + #LANE SPEED + lane_y += lane_speed + if lane_y > HEIGHT: + lane_y = 0 + + #BG + screen.fill(BG_COLOUR) + + #ROAD + pygame.draw.rect(screen, ROAD_COLOUR, (200, 0, 400, HEIGHT)) + + #LANE DIVIDER + for i in range(-1, 10): + pygame.draw.rect(screen, LINE_COLOUR, + (WIDTH//2 - 5, lane_y + i * 80, 10, 40)) + + #CAR + pygame.draw.rect(screen, CAR_COLOUR, (car_x, car_y, car_width, car_height)) + + pygame.display.flip() + clock.tick(FPS)