Mercurial > hg > nsaunier > traffic-intelligence

comparison python/moving.py @ 62:290fceb125d2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
moved road user types and added plotting for all road users
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 04 Nov 2010 17:40:07 -0400
parents 40e1508380ed
children c75bcdaed00f
comparison
equal deleted inserted replaced
61:a8c6d544f015 62:290fceb125d2
244 244
245 def getTrajectoryInPolygon(self, polygon): 245 def getTrajectoryInPolygon(self, polygon):
246 'Returns the set of points inside the polygon' 246 'Returns the set of points inside the polygon'
247 # use shapely polygon contains 247 # use shapely polygon contains
248 pass 248 pass
249
250 ##################
251 # Moving Objects
252 ##################
253
254 userTypeNames = ['car',
255 'pedestrian',
256 'twowheels',
257 'bus'
258 'truck']
249 259
250 class MovingObject(STObject): 260 class MovingObject(STObject):
251 '''Class for moving objects 261 '''Class for moving objects
252 i.e. with a trajectory and a geometry (volume) (constant) 262 i.e. with a trajectory and a geometry (volume) (constant)
253 and a usertype (e.g. road user) 263 and a usertype (e.g. road user)
318 instants.append(self.timeInterval[i]*(1-ratio)+ratio*self.timeInterval[i+1]) 328 instants.append(self.timeInterval[i]*(1-ratio)+ratio*self.timeInterval[i+1])
319 return instants 329 return instants
320 330
321 # def computeVelocities(self): 331 # def computeVelocities(self):
322 332
333 def plotRoadUsers(objects, colors):
334 '''Colors is a PlottingPropertyValues instance'''
335 from matplotlib.pyplot import figure, axis
336 figure()
337 for obj in objects:
338 obj.draw(colors.get(obj.userType))
339 axis('equal')
340
341
323 # need for a class representing the indicators, their units, how to print them in graphs... 342 # need for a class representing the indicators, their units, how to print them in graphs...
324 class TemporalIndicator: 343 class TemporalIndicator:
325 '''Class for temporal indicators 344 '''Class for temporal indicators
326 i.e. indicators that take a value at specific instants 345 i.e. indicators that take a value at specific instants
327 346