added trafficSim.py

Author: David Bailey

trafficSim.py

@@ -0,0 +1,166 @@
+from math import sqrt
+
+class Person:
+ def __init__(self,route):
+  self.arrived = False
+  self.travelTime = 0
+  self.route = route
+  self.currentRouteSegment = self.route.pop()
+  self.positionOnCurrentRouteSegment = 0
+  self.width = 2
+  self.length = 2
+ def Walk(self):
+  self.travelTime += 1
+  if isinstance(self.currentRouteSegment,FourWayStopSignIntersection):
+   pass
+  elif isinstance(self.currentRouteSegment,FourWaySignalIntersection):
+   pass
+  elif isinstance(self.currentRouteSegment,RoadSegment):
+   self.positionOnCurrentRouteSegment += 3.7
+   if self.positionOnCurrentRouteSegment >= self.currentRouteSegment.length: # we reached the end of the segment and are at a signal
+    if self.currentRouteSegment.signal.color == 'green': # if green we go
+     if self.route: # more routeSegments in our route
+      self.routePosition = self.routePosition - self.routeSegment.length
+      self.routeSegment = self.route.pop()
+     else:
+      self.arrived = True
+    else: # red/yellow we stop
+     self.routePosition = self.routeSegment.length
+  elif isinstance(self.currentRouteSegment,Crosswalk):
+   pass
+
+class Vehicle:
+ def __init__(self,driver):
+  self.driver = driver
+  self.speed = 0.0
+  self.acceleration = 0.0
+
+class Bicycle(Vehicle):
+ def __init__(self,driver):
+  Vehicle.__init__(self,driver)
+  self.width = 3
+  self.lenght = 6
+
+class Car(Vehicle):
+ def __init__(self,driver):
+  Vehicle.__init__(self,driver)
+
+class Carpool(Car):
+ def __init__(self,driver):
+  Car.__init__(self,driver)
+
+class Bus(Vehicle):
+ pass
+
+class Train:
+ pass
+
+class Ferry:
+ pass
+
+class Place:
+ def __init__(self, x, y):
+  self.x = x
+  self.y = y
+
+class Signal: # red = stop, yellow = stop if able, green = go
+ def __init__(self,color,greenTime,redTime,yellowTime):
+  self.color = color
+  self.greenTime = greenTime
+  self.redTime = redTime
+  self.yellowTime = yellowTime
+ def countdown(self):
+  if self.color == 'red':
+   if self.redTime:
+    self.redTime -= 1
+   else: self.color = 'green'
+  elif self.color == 'green':
+   if self.greenTime:
+    self.greenTime -= 1
+   else: self.color = 'yellow'
+  elif self.color == 'yellow':
+   if self.yellowTime:
+    self.yellowTime -= 1
+   else: self.color = 'red'
+
+class FourWayStopSign: # this direction stop, then yeild, then go
+ def __init__(self):
+  pass
+
+class FourWayStopSignIntersection:
+ def __init__(self,nRoadSegment,sRoadSegment,eRoadSegment,wRoadSegment):
+  self.stopSign = FourWayStopSign()
+  self.nRoadSegment = nRoadSegment
+  self.sRoadSegment = sRoadSegment
+  self.eRoadSegment = eRoadSegment
+  self.wRoadSegment = wRoadSegment
+  nQueque = False
+  sQueque = False
+  eQueque = False
+  wQueque = False
+
+class TwoWayStopSign: # this direction stop, then yeild, then go
+ def __init__(self):
+
+
+class YeildSign: # yeild, then go
+ def __init__(self):
+  pass
+
+class = Crosswalk: #
+ def __init__(self):
+  pedestrian = False
+
+Signals = []
+
+class FourWaySignal:
+ def __init__(self,majorTime,minorTime,yellowTime):
+  self.major = Signal('green',majorTime,minorTime+yellowTime,yellowTime)
+  Signals.append(self.major)
+  self.minor = Signal('red',minorTime,majorTime+yellowTime,yellowTime)
+  Signals.append(self.minor)
+
+class FourWaySignalIntersection:
+ def __init__(self)
+  self.pedestrianSignal1 = FourWaySignal(13,3,12)
+  self.trafficSignal = FourWaySignal(20,10,5)
+  self.pedestrianSignal2 = FourWaySignal(13,3,12)
+
+class RoadSegment:
+ def __init__(self, origin, destination, signal, type):
+  self.origin = origin
+  self.destination = destination
+  self.length = sqrt((self.origin.x+self.destination.x)**2+(self.origin.y+self.destination.y)**2)
+  self.signal = signal
+  self.type = type
+
+class TwoLaneRoad:
+ def __init__(self,end1,end2):
+  self.sidewalk1a = RoadSegment(end1,end2,'person')
+  self.sidewalk1b = RoadSegment(end2,end1,'person')
+  self.roada = RoadSegment(end1,end2,'vehicle')
+  self.roadb = RoadSegment(end2,end1,'vehicle')
+  self.sidewalk2a = RoadSegment(end1,end2,'person')
+  self.sidewalk2b = RoadSegment(end2,end1,'person')
+
+p1 = Place(0,0)
+p2 = Place(10,10)
+p3 = Place(20,20)
+
+rs1 = RoadSegment(p1,p2,s1)
+rs2 = RoadSegment(p2,p3,s2)
+
+rs3 = RoadSegment(p3,p2,s1)
+rs4 = RoadSegment(p2,p1,s2)
+
+P1 = Person([rs2,rs1])
+P2 = Person([rs4,rs3])
+People = [P1,P2]
+
+while P1.arrived == False:
+ for S in Signals:
+  print S.color
+  S.countdown()
+ for P in People:
+  print str(P.arrived) + " " + str(P.travelTime) + " " + str(P.routeSegment) + " " + str(P.routePosition) + "\n"
+  P.Walk()