class UDPInputWithDirPeds(Input):
"""UDPInput is a barebones UDP Input class. It takes any data it receives and adds it to the
'data' element of the response dict. It also notes the 'address'. Specify:
<Port> -- the Port to listen on.
<Mode> -- [Raw] or JSON -- if the mode is set to JSON, packets will be parsed as JSON"""
def inputInit(self):
HOST = '' # Symbolic name meaning all available interfaces
PORT = self.argDict['Port'] # Arbitrary non-privileged port
self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.sock.bind((HOST, PORT))
self.dirPeds = SynchDirPeds({})
self.sensNetLoc = SynchSenseNetLoc({'SensorSpacing':45, 'Mode':'SensorNetwork',
'IPIndexTable':self['IPIndexTable']})
def socketLoop(self):
(data,address) = self.sock.recvfrom(1024)
while data:
if not self['Mode'] or self['Mode'] == 'Raw':
dataDict = {'data':data, 'address':address}
else:
dataDict = json.loads(data)
dataDict['Address'] = address
pedData = self.processPed(dataDict)
self.respond(pedData)
#print 'data'
(data, address) = self.sock.recvfrom(1024)
def processPed(self, data):
data = self.sensNetLoc.processInput(data)
data = self.dirPeds.processInput(data)
return data
def run(self):
self.socketLoop()
class UDPInputWithDirPeds(Input):
"""UDPInput is a barebones UDP Input class. It takes any data it receives and adds it to the
'data' element of the response dict. It also notes the 'address'. Specify:
<Port> -- the Port to listen on.
<Mode> -- [Raw] or JSON -- if the mode is set to JSON, packets will be parsed as JSON"""
def inputInit(self):
HOST = '' # Symbolic name meaning all available interfaces
PORT = self.argDict['Port'] # Arbitrary non-privileged port
self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.sock.bind((HOST, PORT))
self.dirPeds = SynchDirPeds({})
self.sensNetLoc = SynchSenseNetLoc({
'SensorSpacing': 45,
'Mode': 'SensorNetwork',
'IPIndexTable': self['IPIndexTable']
})
def socketLoop(self):
(data, address) = self.sock.recvfrom(1024)
while data:
if not self['Mode'] or self['Mode'] == 'Raw':
dataDict = {'data': data, 'address': address}
else:
dataDict = json.loads(data)
dataDict['Address'] = address
pedData = self.processPed(dataDict)
self.respond(pedData)
#print 'data'
(data, address) = self.sock.recvfrom(1024)
def processPed(self, data):
data = self.sensNetLoc.processInput(data)
data = self.dirPeds.processInput(data)
return data
def run(self):
self.socketLoop()
class PedestrianSimulator(Input):
"""Simulates pedestrians inputing to simulated motion sensors.
Params:
MaxX
Velocity
NumSensors
SensorSpacing
NumPeds
"""
def inputInit(self):
self.peds = []
self.responses = []
self.lock = thread.allocate_lock() #TODO: refactor to LockingInput
self.sensors = []
self.initPedestrians(self['NumPeds'], self['Velocity'], self['MaxX'])
self.initSensors()
self.dirPeds = SynchDirPeds({})
self.sensNetLoc = SynchSenseNetLoc({'SensorSpacing':45})
#TODO: homogenize
def initSensors(self):
x = 0
for i in range(self['NumSensors']):
sensor = MotionSensorSimulator({'Id':str(i), 'parentScope':self,'DataHook':self,
'RefreshInterval':500, 'Location':x})
x += self['SensorSpacing']
self.sensors.append(sensor)
print 'max sensor x', x
print 'maxx', self['MaxX']
def initPedestrians(self,numPeds, vel, maxX):
for i in xrange(numPeds):
if random.random() > .5:
vel *= -1
self.peds.append({'Loc':random.randint(0, maxX), 'Vel':vel})
def getLocs(self):
return [ped['Loc'] for ped in self.peds]
def addPedestrian(self, loc, vel):
self.peds.append({'Loc':loc,'Vel':vel+random.randint(-5,5)})
def evaluateSensors(self):
for s in self.sensors:
s.sensingLoop()
def sensingLoop(self):
self.evaluateSensors()
self.movePedestrians(self['RefreshInterval'])
if self.responses:
pedData = self.processPed(self.responses)
self.respond(pedData)
self.responses = []
def processPed(self, data):
data = self.sensNetLoc.processInput(data)
data = self.dirPeds.processInput(data)
return data
def processResponse(self, sensorInput):
"""Queue up response"""
self.responses.append(sensorInput)
def movePedestrians(self, dt):
for ped in self.peds:
ped['Vel'] += random.randint(-5,5) #random velocity changes
ped['Loc'] += ped['Vel'] * dt/1000
#Bounce pedestrians that turn
if ped['Loc'] < 0:
ped['Vel'] = -ped['Vel']
ped['Loc'] = 0
if ped['Loc'] > self['MaxX']:
ped['Vel'] = -ped['Vel']
ped['Loc'] = self['MaxX']