def test():
car1 = Car(30,50,3,3,100,"blue")
car2 = Car(20,80,3,3,100,"green")
road = Road(1,2)
road.add_car(car1)
road.add_car(car2)
road.test_for_stop()
road.plot()
car1 = Car(30,40,1.5,2.5,100,"blue")
road = Road(1,2)
road.add_car(car1)
road.test_for_stop()
road.plot()
def play():
n = int(input("Number of cars: "))
ds = float(input("Crossroad length: "))
dt = float(input("Yellow light time: "))
road = Road(ds,dt)
prev = 0
for i in range(n):
x0 = 0
v = float(input("Initial velocity: "))
maxv = float(input("Maximum velocity: "))
ap = float(input("Positive acceleration: "))
an = float(input("Negative acceleration: "))
if i == 0:
x0 = prev + float(input("Initial distance: "))
else:
x0 = prev + float(input("Distance: from previous car"))
prev = x0
color = input("Color of the car: ")
road.add_car(Car(x0,v,an,ap,maxv,color))
road.test_for_stop()
road.plot()
