88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
|
from car import Car
|
||
|
|
import random
|
||
|
|
import turtle
|
||
|
|
|
||
|
|
COLORS = ["red", "orange", "yellow", "green", "blue", "purple"]
|
||
|
|
CAR_TYPES = [["BlueCarRight.gif", "BrightGreenCarRight.gif", "DarkBlueCarRight.gif",
|
||
|
|
"GreenCarRight.gif", "BMW-Z4Right.gif"],
|
||
|
|
["BlueCarLeft.gif", "BrightGreenCarLeft.gif", "DarkBlueCarLeft.gif", "GreenCarLeft.gif", "BMW-Z4Left.gif"]]
|
||
|
|
STARTING_POSITIONS = [(0, (-350, -205)), (0, (-350, -145)), (0, (-350, -80)),
|
||
|
|
(180, (350, 80)), (180, (350, 145)), (180, (350, 205))]
|
||
|
|
BASE_MOVE_DISTANCE = 5
|
||
|
|
MOVE_INCREMENT = 10
|
||
|
|
|
||
|
|
|
||
|
|
class CarManager:
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.cars = []
|
||
|
|
self.lanes = [0, 1, 2, 3, 4, 5]
|
||
|
|
self.last_lane = -1
|
||
|
|
for car_shape_list in CAR_TYPES:
|
||
|
|
for car_shape in car_shape_list:
|
||
|
|
car_shape = 'graphics/' + car_shape
|
||
|
|
turtle.register_shape(car_shape)
|
||
|
|
|
||
|
|
def create_car(self):
|
||
|
|
# Makes sure don't get 2 cars in a row in same lane
|
||
|
|
if len(self.lanes) == 0:
|
||
|
|
self.lanes = [0, 1, 2, 3, 4, 5]
|
||
|
|
lane = random.choice(self.lanes)
|
||
|
|
while lane == self.last_lane:
|
||
|
|
lane = random.choice(self.lanes)
|
||
|
|
self.lanes.remove(lane)
|
||
|
|
self.last_lane = lane
|
||
|
|
|
||
|
|
# Set up the car
|
||
|
|
starting_info = STARTING_POSITIONS[lane]
|
||
|
|
# Get car type (left or right) by using first index in starting_info, then set random shape (gif) accordingly
|
||
|
|
index = 0 if starting_info[0] == 0 else 1
|
||
|
|
shape = random.choice(CAR_TYPES[index])
|
||
|
|
shape = 'graphics/' + shape
|
||
|
|
|
||
|
|
# Create and add car
|
||
|
|
self.cars.append(Car(shape, starting_info[1], starting_info[0], BASE_MOVE_DISTANCE + random.randint(-1, 5)))
|
||
|
|
|
||
|
|
def update_cars(self):
|
||
|
|
# Remove cars that are finished going on road (off-screen and not coming back)
|
||
|
|
self.cars = [car for car in self.cars if (car.heading() == 0 and car.xcor() < 350)
|
||
|
|
or (car.heading() == 180 and car.xcor() > -350)]
|
||
|
|
# Iterate over remaining cars, and move them
|
||
|
|
for car in self.cars:
|
||
|
|
car.move()
|
||
|
|
|
||
|
|
def is_collision(self, other):
|
||
|
|
"""
|
||
|
|
|
||
|
|
:param other: A turtle object, or object of a child class of turtle
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
# Get edges of other
|
||
|
|
other_top_edge = other.ycor() + 12
|
||
|
|
other_bottom_edge = other.ycor() - 12
|
||
|
|
other_left_edge = other.xcor() - 12
|
||
|
|
other_right_edge = other.xcor() + 12
|
||
|
|
|
||
|
|
# Check each car for whether the distance between edges overlaps
|
||
|
|
for car in self.cars:
|
||
|
|
car_top_edge = car.ycor() + 11
|
||
|
|
car_bottom_edge = car.ycor() - 11
|
||
|
|
car_left_edge = car.xcor() - 20
|
||
|
|
car_right_edge = car.xcor() + 20
|
||
|
|
# It's mom's spaghetti (code)
|
||
|
|
if (
|
||
|
|
(
|
||
|
|
(other_top_edge - car_bottom_edge > 0 and car_top_edge - other_top_edge > 0)
|
||
|
|
or
|
||
|
|
(car_top_edge - other_bottom_edge > 0 and other_bottom_edge - car_bottom_edge > 0)
|
||
|
|
)
|
||
|
|
and
|
||
|
|
(
|
||
|
|
(other_left_edge - car_left_edge > 0 and car_right_edge - other_left_edge > 0)
|
||
|
|
or
|
||
|
|
(other_right_edge - car_left_edge > 0 and car_right_edge - other_right_edge > 0)
|
||
|
|
)
|
||
|
|
):
|
||
|
|
return True
|
||
|
|
return False
|