controller.py

import time
import queue
from vehicle import *
#Controller class
class Controller:
	
	def __init__(self, p):
		self.sequence = p
		self.activePhase = 0
		self.sequence[self.activePhase].start()
	
	def advance(self):
		for i in range(len(self.sequence)):
			self.sequence[i].advance()
		if self.sequence[self.activePhase].isRed[-1]:
			self.activePhase = (self.activePhase + 1) % len(self.sequence)
			self.sequence[self.activePhase].start()
	
class Phase:
	def __init__(self,min, max, extender):
		self.ext = extender
		self.elapsed = 0.0
		self.extTime = 0.0
		self.approach = None
		self.isRed = [True]*7
		self.min = min
		self.max = max
	
	def extend(self):
		TYPE = 2 #0 = base, 1 = V2I, 2 = Traditional
		if self.ext and TYPE == 1:
			veh = self.approach.vehs[0]
			if (veh == None) or veh.pastPositions[-1] > veh.commRange:
				return False
			vp = veh.pastPositions[-1]
			
			timeLeft = self.max - self.elapsed
			ttt = timeLeft/10.0 * self.approach.speedLimit
			
			if ttt >= vp:
				return True
		elif self.ext and TYPE == 2:
			distBack = self.approach.speedLimit * 5
			for veh in self.approach.vehs:
				if veh.pastPositions[-1] > distBack and veh.pastPositions[-1] <= distBack + 2.44:
					self.extTime = self.elapsed
					break
				elif veh.pastPositions[-1] > distBack+2.44:
					break
			if self.elapsed - self.extTime < 5:
				return True
		else:
			return False
	
	def advance(self):
		if self.isRed[-1]:
			self.isRed.append(True)
		else:
			self.elapsed += 0.1
			if self.elapsed >= self.max:	#Reached max limit
				self.isRed.append(True)		#Turn red
			elif self.elapsed < self.min:	#Still in min time
				self.extTime = self.elapsed
				self.isRed.append(False)	#Stay green
			elif self.extend():				#Extension based on V2I
				self.isRed.append(False)	#Stay green
			else:							#No extension
				self.isRed.append(True)		#Turn red
		del self.isRed[0]
	
	def setApproach(self, approach):
		self.approach = approach
	
	def start(self):
		self.isRed[-1] = False
		self.elapsed = 0.0

class Approach:
	def __init__(self, phase, length, ts):
		self.vehs = []
		self.length = 0.0
		self.phase = phase
		self.speedLimit = 20.0
		self.phase.setApproach(self)
		self.length = length
		self.ts = ts
	
	def generateVeh(self, j):
		if j in self.ts:
			v = Vehicle(self, j)
			if len(self.vehs) > 0:
				v.vehInFront = self.vehs[-1]
			self.vehs.append(v)
	
	def advance(self, j):
		self.generateVeh(j)
		exited = [False] * len(self.vehs)
		for i in range(len(self.vehs)):
			exited[i] = self.vehs[i].advance(j)
		for i in range(len(exited)):
			if exited[i]:
				if len(self.vehs)>1:
					self.vehs[1].vehInFront = None
				del self.vehs[0]