Mercurial > hg > nsaunier > traffic-intelligence

comparison python/prediction.py @ 665:15e244d2a1b5

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
corrected bug with circular import for VideoFilenameAddable, moved to base module
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 20 May 2015 13:57:47 +0200
parents 72174e66aba5
children 9a258687af4c
comparison
equal deleted inserted replaced
664:455f9b93819c 665:15e244d2a1b5
1 #! /usr/bin/env python 1 #! /usr/bin/env python
2 '''Library for motion prediction methods''' 2 '''Library for motion prediction methods'''
3 3
4 import moving 4 import moving
5 import math 5 from utils import LCSS
6 import random 6
7 import math, random
7 import numpy as np 8 import numpy as np
8 from utils import LCSS 9
9 10
10 class PredictedTrajectory: 11 class PredictedTrajectory(object):
11 '''Class for predicted trajectories with lazy evaluation 12 '''Class for predicted trajectories with lazy evaluation
12 if the predicted position has not been already computed, compute it 13 if the predicted position has not been already computed, compute it
13 14
14 it should also have a probability''' 15 it should also have a probability'''
15 16
292 savePredictedTrajectoriesFigure(currentInstant, obj1, obj2, predictedTrajectories1, predictedTrajectories2, timeHorizon) 293 savePredictedTrajectoriesFigure(currentInstant, obj1, obj2, predictedTrajectories1, predictedTrajectories2, timeHorizon)
293 294
294 return currentInstant, collisionPoints, crossingZones 295 return currentInstant, collisionPoints, crossingZones
295 296
296 297
297 class PredictionParameters: 298 class PredictionParameters(object):
298 def __init__(self, name, maxSpeed): 299 def __init__(self, name, maxSpeed):
299 self.name = name 300 self.name = name
300 self.maxSpeed = maxSpeed 301 self.maxSpeed = maxSpeed
301 302
302 def __str__(self): 303 def __str__(self):