Mercurial Hosting > traffic-intelligence
annotate python/events.py @ 398:3399bd48cb40
Ajout d'une méthode pour obtenir le nombre de FPS
Méthode de capture des trames vidéos plus résistante aux erreur
Utilisation d'un dictionnaire pour les fichier de configuration afin de garder le nom des sections
author | Jean-Philippe Jodoin <jpjodoin@gmail.com> |
---|---|
date | Mon, 29 Jul 2013 13:46:07 -0400 |
parents | 6e0dedd34920 |
children | 365d8dee44f3 |
rev | line source |
---|---|
56
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
1 #! /usr/bin/env python |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
2 '''Libraries for events |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
3 Interactions, pedestrian crossing...''' |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
4 |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
5 import numpy as np |
306
93d851d0d21e
bug correction, minor work on indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
299
diff
changeset
|
6 from numpy import arccos |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
7 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
8 import multiprocessing |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
9 import itertools |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
10 |
341
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
11 import moving, prediction, indicators, utils |
56
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
12 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
13 __metaclass__ = type |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
14 |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
15 class Interaction(moving.STObject): |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
16 '''Class for an interaction between two road users |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
17 or a road user and an obstacle |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
18 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
19 link to the moving objects |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
20 contains the indicators in a dictionary with the names as keys |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
21 ''' |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
22 |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
23 categories = {'Head On': 0, |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
24 'rearend': 1, |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
25 'side': 2, |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
26 'parallel': 3} |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
27 |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
28 indicatorNames = ['Collision Course Dot Product', |
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
29 'Collision Course Angle', |
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
30 'Distance', |
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
31 'Minimum Distance', |
341
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
32 'Velocity Angle', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
33 'Speed Differential', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
34 'Collision Probability', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
35 'Time to Collision', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
36 'Probability of Successful Evasive Action', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
37 'predicted Post Encroachment Time'] |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
38 |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
39 indicatorNameToIndices = utils.inverseEnumeration(indicatorNames) |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
40 |
388
6e0dedd34920
minor name change
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
387
diff
changeset
|
41 indicatorShortNames = ['CCDP', |
341
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
42 'CCAng', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
43 'Dist', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
44 'MinDist', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
45 'VA', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
46 'SD', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
47 'PoC', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
48 'TTC', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
49 'P(SEA)', |
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
50 'pPET'] |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
51 |
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
52 def __init__(self, num = None, timeInterval = None, roaduserNum1 = None, roaduserNum2 = None, roadUser1 = None, roadUser2 = None, categoryNum = None): |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
53 moving.STObject.__init__(self, num, timeInterval) |
339 | 54 self.roadUserNumbers = set([roaduserNum1, roaduserNum2]) |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
55 self.roadUser1 = roadUser1 |
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
56 self.roadUser2 = roadUser2 |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
57 self.categoryNum = categoryNum |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
58 self.indicators = {} |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
59 |
340
1046b7346886
work in progress on storing indicator values
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
339
diff
changeset
|
60 def getRoadUserNumbers(self): |
1046b7346886
work in progress on storing indicator values
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
339
diff
changeset
|
61 return self.roadUserNumbers |
1046b7346886
work in progress on storing indicator values
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
339
diff
changeset
|
62 |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
63 def getIndicator(self, indicatorName): |
321
a5e40bd04cf4
rearranged LCSS indicator functions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
320
diff
changeset
|
64 return self.indicators.get(indicatorName, None) |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
65 |
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
66 def addIndicator(self, indicator): |
321
a5e40bd04cf4
rearranged LCSS indicator functions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
320
diff
changeset
|
67 if indicator: |
a5e40bd04cf4
rearranged LCSS indicator functions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
320
diff
changeset
|
68 self.indicators[indicator.name] = indicator |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
69 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
70 def computeIndicators(self): |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
71 '''Computes the collision course cosine only if the cosine is positive''' |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
72 collisionCourseDotProducts = {}#[0]*int(self.timeInterval.length()) |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
73 collisionCourseAngles = {} |
325
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
74 velocityAngles = {} |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
75 distances = {}#[0]*int(self.timeInterval.length()) |
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
76 speedDifferentials = {} |
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
77 for instant in self.timeInterval: |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
78 deltap = self.roadUser1.getPositionAtInstant(instant)-self.roadUser2.getPositionAtInstant(instant) |
325
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
79 v1 = self.roadUser1.getVelocityAtInstant(instant) |
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
80 v2 = self.roadUser2.getVelocityAtInstant(instant) |
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
81 deltav = v2-v1 |
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
82 velocityAngles[instant] = arccos(moving.Point.dot(v1, v2)/(v1.norm2()*v2.norm2())) |
299
7e5fb4abd070
renaming event to events and correcting errors in indicator computation
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
294
diff
changeset
|
83 collisionCourseDotProducts[instant] = moving.Point.dot(deltap, deltav) |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
84 distances[instant] = deltap.norm2() |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
85 speedDifferentials[instant] = deltav.norm2() |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
86 #if collisionCourseDotProducts[instant] > 0: |
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
87 collisionCourseAngles[instant] = arccos(collisionCourseDotProducts[instant]/(distances[instant]*speedDifferentials[instant])) |
306
93d851d0d21e
bug correction, minor work on indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
299
diff
changeset
|
88 |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
89 # todo shorten the time intervals based on the interaction definition |
387
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
90 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[0], collisionCourseDotProducts)) |
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
91 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[1], collisionCourseAngles)) |
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
92 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[2], distances)) |
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
93 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[4], velocityAngles)) |
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
94 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[5], speedDifferentials)) |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
95 |
306
93d851d0d21e
bug correction, minor work on indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
299
diff
changeset
|
96 # todo test for interaction instants and interval, compute indicators |
93d851d0d21e
bug correction, minor work on indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
299
diff
changeset
|
97 |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
98 # if we have features, compute other indicators |
378
3805b9639647
added tests for movement orientation, object size
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
358
diff
changeset
|
99 if self.roadUser1.features != None and self.roadUser2.features != None: |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
100 minDistance={} |
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
101 for instant in self.timeInterval: |
382
ba813f148ade
development for clustering
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
378
diff
changeset
|
102 minDistance[instant] = moving.MovingObject.minDistance(self.roadUser1, self.roadUser2, instant) |
387
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
103 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[3], minDistance)) |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
104 |
338
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
105 def computeCrossingsCollisions(self, predictionParameters, collisionDistanceThreshold, timeHorizon, computeCZ = False, debug = False, timeInterval = None): |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
106 '''Computes all crossing and collision points at each common instant for two road users. ''' |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
107 self.collisionPoints={} |
339 | 108 self.crossingZones={} |
338
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
109 TTCs = {} |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
110 |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
111 if timeInterval: |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
112 commonTimeInterval = timeInterval |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
113 else: |
338
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
114 commonTimeInterval = self.timeInterval |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
115 for i in list(commonTimeInterval)[:-1]: # do not look at the 1 last position/velocities, often with errors |
358
c41ff9f3c263
moved current method for collision points and crossing zones computation into prediction parameters (put expectedindicator in SafetyPoint)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
357
diff
changeset
|
116 self.collisionPoints[i], self.crossingZones[i] = predictionParameters.computeCrossingsCollisionsAtInstant(i, self.roadUser1, self.roadUser2, collisionDistanceThreshold, timeHorizon, computeCZ, debug) |
357
e5fe0e6d48a1
corrected bug computing TTC (resp. pPET) if there is no collision point (resp. crossing zone)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
341
diff
changeset
|
117 if len(self.collisionPoints[i]) > 0: |
358
c41ff9f3c263
moved current method for collision points and crossing zones computation into prediction parameters (put expectedindicator in SafetyPoint)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
357
diff
changeset
|
118 TTCs[i] = prediction.SafetyPoint.computeExpectedIndicator(self.collisionPoints[i]) |
341
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
119 # add probability of collision, and probability of successful evasive action |
387
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
120 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[7], TTCs)) |
338
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
121 |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
122 if computeCZ: |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
123 pPETs = {} |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
124 for i in list(commonTimeInterval)[:-1]: |
357
e5fe0e6d48a1
corrected bug computing TTC (resp. pPET) if there is no collision point (resp. crossing zone)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
341
diff
changeset
|
125 if len(self.crossingZones[i]) > 0: |
358
c41ff9f3c263
moved current method for collision points and crossing zones computation into prediction parameters (put expectedindicator in SafetyPoint)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
357
diff
changeset
|
126 pPETs[i] = prediction.SafetyPoint.computeExpectedIndicator(self.crossingZones[i]) |
387
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
127 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[9], pPETs)) |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
128 |
310
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
129 def addVideoFilename(self,videoFilename): |
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
130 self.videoFilename= videoFilename |
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
131 |
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
132 def addInteractionType(self,interactionType): |
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
133 ''' interaction types: conflict or collision if they are known''' |
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
134 self.interactionType= interactionType |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
135 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
136 def createInteractions(objects): |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
137 '''Create all interactions of two co-existing road users |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
138 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
139 todo add test to compute categories?''' |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
140 interactions = [] |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
141 num = 0 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
142 for i in xrange(len(objects)): |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
143 for j in xrange(i): |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
144 commonTimeInterval = objects[i].commonTimeInterval(objects[j]) |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
145 if not commonTimeInterval.empty(): |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
146 interactions.append(Interaction(num, commonTimeInterval, objects[i].num, objects[j].num, objects[i], objects[j])) |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
147 num += 1 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
148 return interactions |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
149 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
150 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
151 # TODO: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
152 #http://stackoverflow.com/questions/3288595/multiprocessing-using-pool-map-on-a-function-defined-in-a-class |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
153 #http://www.rueckstiess.net/research/snippets/show/ca1d7d90 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
154 def calculateIndicatorPipe(pairs, predParam, timeHorizon=75,collisionDistanceThreshold=1.8): |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
155 collisionPoints, crossingZones = prediction.computeCrossingsCollisions(pairs.roadUser1, pairs.roadUser2, predParam, collisionDistanceThreshold, timeHorizon) |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
156 #print pairs.num |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
157 # Ignore empty collision points |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
158 empty = 1 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
159 for i in collisionPoints: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
160 if(collisionPoints[i] != []): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
161 empty = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
162 if(empty == 1): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
163 pairs.hasCP = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
164 else: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
165 pairs.hasCP = 1 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
166 pairs.CP = collisionPoints |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
167 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
168 # Ignore empty crossing zones |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
169 empty = 1 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
170 for i in crossingZones: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
171 if(crossingZones[i] != []): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
172 empty = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
173 if(empty == 1): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
174 pairs.hasCZ = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
175 else: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
176 pairs.hasCZ = 1 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
177 pairs.CZ = crossingZones |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
178 return pairs |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
179 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
180 def calculateIndicatorPipe_star(a_b): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
181 """Convert `f([1,2])` to `f(1,2)` call.""" |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
182 return calculateIndicatorPipe(*a_b) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
183 |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
184 class VehPairs(): |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
185 '''Create a veh-pairs object from objects list''' |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
186 def __init__(self,objects): |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
187 self.pairs = createInteractions(objects) |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
188 self.interactionCount = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
189 self.CPcount = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
190 self.CZcount = 0 |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
191 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
192 # Process indicator calculation with support for multi-threading |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
193 def calculateIndicators(self,predParam,threads=1,timeHorizon=75,collisionDistanceThreshold=1.8): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
194 if(threads > 1): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
195 pool = multiprocessing.Pool(threads) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
196 self.pairs = pool.map(calculateIndicatorPipe_star, itertools.izip(self.pairs, itertools.repeat(predParam))) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
197 pool.close() |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
198 else: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
199 #prog = Tools.ProgressBar(0, len(self.pairs), 77) #Removed in traffic-intelligenc port |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
200 for j in xrange(len(self.pairs)): |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
201 #prog.updateAmount(j) #Removed in traffic-intelligenc port |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
202 collisionPoints, crossingZones = prediction.computeCrossingsCollisions(self.pairs[j].roadUser1, self.pairs[j].roadUser2, predParam, collisionDistanceThreshold, timeHorizon) |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
203 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
204 # Ignore empty collision points |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
205 empty = 1 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
206 for i in collisionPoints: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
207 if(collisionPoints[i] != []): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
208 empty = 0 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
209 if(empty == 1): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
210 self.pairs[j].hasCP = 0 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
211 else: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
212 self.pairs[j].hasCP = 1 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
213 self.pairs[j].CP = collisionPoints |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
214 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
215 # Ignore empty crossing zones |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
216 empty = 1 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
217 for i in crossingZones: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
218 if(crossingZones[i] != []): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
219 empty = 0 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
220 if(empty == 1): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
221 self.pairs[j].hasCZ = 0 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
222 else: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
223 self.pairs[j].hasCZ = 1 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
224 self.pairs[j].CZ = crossingZones |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
225 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
226 for j in self.pairs: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
227 self.interactionCount = self.interactionCount + len(j.CP) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
228 self.CPcount = len(self.getCPlist()) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
229 self.Czcount = len(self.getCZlist()) |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
230 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
231 |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
232 def getPairsWCP(self): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
233 lists = [] |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
234 for j in self.pairs: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
235 if(j.hasCP): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
236 lists.append(j.num) |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
237 return lists |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
238 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
239 def getPairsWCZ(self): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
240 lists = [] |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
241 for j in self.pairs: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
242 if(j.hasCZ): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
243 lists.append(j.num) |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
244 return lists |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
245 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
246 def getCPlist(self,indicatorThreshold=99999): |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
247 lists = [] |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
248 for j in self.pairs: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
249 if(j.hasCP): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
250 for k in j.CP: |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
251 if(j.CP[k] != [] and j.CP[k][0].indicator < indicatorThreshold): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
252 lists.append([k,j.CP[k][0]]) |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
253 return lists |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
254 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
255 def getCZlist(self,indicatorThreshold=99999): |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
256 lists = [] |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
257 for j in self.pairs: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
258 if(j.hasCZ): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
259 for k in j.CZ: |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
260 if(j.CZ[k] != [] and j.CZ[k][0].indicator < indicatorThreshold): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
261 lists.append([k,j.CZ[k][0]]) |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
262 return lists |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
263 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
264 def genIndicatorHistogram(self, CPlist=False, bins=range(0,100,1)): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
265 if(not CPlist): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
266 CPlist = self.getCPlist() |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
267 if(not CPlist): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
268 return False |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
269 TTC_list = [] |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
270 for i in CPlist: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
271 TTC_list.append(i[1].indicator) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
272 histo = np.histogram(TTC_list,bins=bins) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
273 histo += (histo[0].astype(float)/np.sum(histo[0]),) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
274 return histo |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
275 |
59
f955e83da499
developed indicators in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
56
diff
changeset
|
276 class Crossing(moving.STObject): |
56
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
277 '''Class for the event of a street crossing |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
278 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
279 TODO: detecter passage sur la chaussee |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
280 identifier origines et destination (ou uniquement chaussee dans FOV) |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
281 carac traversee |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
282 detecter proximite veh (retirer si trop similaire simultanement |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
283 carac interaction''' |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
284 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
285 def __init__(self, roaduserNum = None, num = None, timeInterval = None): |
59
f955e83da499
developed indicators in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
56
diff
changeset
|
286 moving.STObject.__init__(self, num, timeInterval) |
56
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
287 self.roaduserNum = roaduserNum |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
288 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
289 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
290 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
291 if __name__ == "__main__": |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
292 import doctest |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
293 import unittest |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
294 #suite = doctest.DocFileSuite('tests/moving.txt') |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
295 suite = doctest.DocTestSuite() |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
296 unittest.TextTestRunner().run(suite) |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
297 |