Mercurial Hosting > traffic-intelligence
annotate python/events.py @ 614:5e09583275a4
Merged Nicolas/trafficintelligence into default
author | Mohamed Gomaa <eng.m.gom3a@gmail.com> |
---|---|
date | Fri, 05 Dec 2014 12:13:53 -0500 |
parents | 07b1bd0785cd |
children | 84690dfe5560 |
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', |
408
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
42 'CCA', |
341
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 |
408
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
52 indicatorUnits = ['', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
53 'rad', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
54 'm', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
55 'm', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
56 'rad', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
57 'm/s', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
58 '', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
59 's', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
60 '', |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
61 ''] |
365d8dee44f3
last changes for TRB14
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
388
diff
changeset
|
62 |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
63 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
|
64 moving.STObject.__init__(self, num, timeInterval) |
566
07b1bd0785cd
simplifications to interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
559
diff
changeset
|
65 if timeInterval == None and roadUser1 != None and roadUser2 != None: |
07b1bd0785cd
simplifications to interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
559
diff
changeset
|
66 self.timeInterval = roadUser1.commonTimeInterval(roadUser2) |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
67 self.roadUser1 = roadUser1 |
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
68 self.roadUser2 = roadUser2 |
566
07b1bd0785cd
simplifications to interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
559
diff
changeset
|
69 if roaduserNum1 != None and roaduserNum2 != None: |
07b1bd0785cd
simplifications to interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
559
diff
changeset
|
70 self.roadUserNumbers = set([roaduserNum1, roaduserNum2]) |
07b1bd0785cd
simplifications to interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
559
diff
changeset
|
71 elif roadUser1 != None and roadUser2 != None: |
07b1bd0785cd
simplifications to interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
559
diff
changeset
|
72 self.roadUserNumbers = set(roadUser1.getNum(), roadUser2.getNum()) |
07b1bd0785cd
simplifications to interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
559
diff
changeset
|
73 else: |
07b1bd0785cd
simplifications to interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
559
diff
changeset
|
74 self.roadUserNumbers = None |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
75 self.categoryNum = categoryNum |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
76 self.indicators = {} |
451
cd342a774806
Point/CurvilinearTrajectory/Interaction utiles
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
436
diff
changeset
|
77 self.interactionInterval = None |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
78 |
340
1046b7346886
work in progress on storing indicator values
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
339
diff
changeset
|
79 def getRoadUserNumbers(self): |
1046b7346886
work in progress on storing indicator values
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
339
diff
changeset
|
80 return self.roadUserNumbers |
1046b7346886
work in progress on storing indicator values
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
339
diff
changeset
|
81 |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
82 def getIndicator(self, indicatorName): |
321
a5e40bd04cf4
rearranged LCSS indicator functions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
320
diff
changeset
|
83 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
|
84 |
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
85 def addIndicator(self, indicator): |
321
a5e40bd04cf4
rearranged LCSS indicator functions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
320
diff
changeset
|
86 if indicator: |
a5e40bd04cf4
rearranged LCSS indicator functions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
320
diff
changeset
|
87 self.indicators[indicator.name] = indicator |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
88 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
89 def computeIndicators(self): |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
90 '''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
|
91 collisionCourseDotProducts = {}#[0]*int(self.timeInterval.length()) |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
92 collisionCourseAngles = {} |
325
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
93 velocityAngles = {} |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
94 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
|
95 speedDifferentials = {} |
452
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
96 interactionInstants = [] |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
97 for instant in self.timeInterval: |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
98 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
|
99 v1 = self.roadUser1.getVelocityAtInstant(instant) |
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
100 v2 = self.roadUser2.getVelocityAtInstant(instant) |
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
101 deltav = v2-v1 |
6c9c7c956926
added velocity angle in indicators
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
321
diff
changeset
|
102 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
|
103 collisionCourseDotProducts[instant] = moving.Point.dot(deltap, deltav) |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
104 distances[instant] = deltap.norm2() |
294
1f253f218b9f
evolution of indicators and their computation in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
293
diff
changeset
|
105 speedDifferentials[instant] = deltav.norm2() |
452
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
106 if collisionCourseDotProducts[instant] > 0: |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
107 interactionInstants.append(instant) |
320
419f30491a4b
renamed fields movingObject to roadUser, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
317
diff
changeset
|
108 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
|
109 |
452
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
110 if len(interactionInstants) >= 2: |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
111 self.interactionInterval = moving.TimeInterval(interactionInstants[0], interactionInstants[-1]) |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
112 else: |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
113 self.interactionInterval = moving.TimeInterval() |
387
91679eb2ff2c
cleaning up safety analysis and the new traditional constant velocity method
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
382
diff
changeset
|
114 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
|
115 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
|
116 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
|
117 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
|
118 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
|
119 |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
120 # if we have features, compute other indicators |
559
806df5f61c03
adapted safety-analysis script to use multi-threading
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
558
diff
changeset
|
121 if len(self.roadUser1.features) != 0 and len(self.roadUser2.features) != 0: |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
122 minDistance={} |
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
123 for instant in self.timeInterval: |
382
ba813f148ade
development for clustering
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
378
diff
changeset
|
124 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
|
125 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[3], minDistance)) |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
126 |
558
a80ef6931fd8
updated safety-analysis to test multiprocessing
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
526
diff
changeset
|
127 def computeCrossingsCollisions(self, predictionParameters, collisionDistanceThreshold, timeHorizon, computeCZ = False, debug = False, timeInterval = None, nProcesses = 1): |
338
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
128 '''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
|
129 self.collisionPoints={} |
339 | 130 self.crossingZones={} |
338
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
131 TTCs = {} |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
132 |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
133 if timeInterval: |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
134 commonTimeInterval = timeInterval |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
135 else: |
338
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
136 commonTimeInterval = self.timeInterval |
559
806df5f61c03
adapted safety-analysis script to use multi-threading
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
558
diff
changeset
|
137 self.collisionPoints, self.crossingZones = prediction.computeCrossingsCollisions(predictionParameters, self.roadUser1, self.roadUser2, collisionDistanceThreshold, timeHorizon, computeCZ, debug, commonTimeInterval, nProcesses) |
558
a80ef6931fd8
updated safety-analysis to test multiprocessing
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
526
diff
changeset
|
138 for i, cp in self.collisionPoints.iteritems(): |
a80ef6931fd8
updated safety-analysis to test multiprocessing
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
526
diff
changeset
|
139 TTCs[i] = prediction.SafetyPoint.computeExpectedIndicator(cp) |
341
2f39c4ed0b62
first version of indicator saving to sqlite
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
340
diff
changeset
|
140 # 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
|
141 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[7], TTCs)) |
338
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
142 |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
143 if computeCZ: |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
144 pPETs = {} |
f3aceea2afbb
first safety analysis script
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
325
diff
changeset
|
145 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
|
146 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
|
147 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
|
148 self.addIndicator(indicators.SeverityIndicator(Interaction.indicatorNames[9], pPETs)) |
317
d280b881e860
added indicator min distance
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
310
diff
changeset
|
149 |
310
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
150 def addVideoFilename(self,videoFilename): |
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
151 self.videoFilename= videoFilename |
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
152 |
f7ca78a11ea6
add functions to add videofilename and interaction type in Interaction class
Mohamed Gomaa
parents:
306
diff
changeset
|
153 def addInteractionType(self,interactionType): |
454
62d05436099d
corrected indentation bug
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
452
diff
changeset
|
154 ''' interaction types: conflict or collision if they are known''' |
451
cd342a774806
Point/CurvilinearTrajectory/Interaction utiles
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
436
diff
changeset
|
155 self.interactionType= interactionType |
cd342a774806
Point/CurvilinearTrajectory/Interaction utiles
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
436
diff
changeset
|
156 |
487
e04b22ce2fcd
generalized createInteractions to 2 lists of objects (for cars and pedestrians for example)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
454
diff
changeset
|
157 def createInteractions(objects, _others = None): |
e04b22ce2fcd
generalized createInteractions to 2 lists of objects (for cars and pedestrians for example)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
454
diff
changeset
|
158 '''Create all interactions of two co-existing road users''' |
e04b22ce2fcd
generalized createInteractions to 2 lists of objects (for cars and pedestrians for example)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
454
diff
changeset
|
159 if _others != None: |
e04b22ce2fcd
generalized createInteractions to 2 lists of objects (for cars and pedestrians for example)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
454
diff
changeset
|
160 others = _others |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
161 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
162 interactions = [] |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
163 num = 0 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
164 for i in xrange(len(objects)): |
487
e04b22ce2fcd
generalized createInteractions to 2 lists of objects (for cars and pedestrians for example)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
454
diff
changeset
|
165 if _others == None: |
e04b22ce2fcd
generalized createInteractions to 2 lists of objects (for cars and pedestrians for example)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
454
diff
changeset
|
166 others = objects[:i] |
e04b22ce2fcd
generalized createInteractions to 2 lists of objects (for cars and pedestrians for example)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
454
diff
changeset
|
167 for j in xrange(len(others)): |
489
000bddf84ad0
corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
487
diff
changeset
|
168 commonTimeInterval = objects[i].commonTimeInterval(others[j]) |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
169 if not commonTimeInterval.empty(): |
489
000bddf84ad0
corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
487
diff
changeset
|
170 interactions.append(Interaction(num, commonTimeInterval, objects[i].num, others[j].num, objects[i], others[j])) |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
171 num += 1 |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
172 return interactions |
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
173 |
430
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
174 def prototypeCluster(interactions, similarityMatrix, alignmentMatrix, indicatorName, minSimilarity): |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
175 '''Finds exemplar indicator time series for all interactions |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
176 Returns the prototype indices (in the interaction list) and the label of each indicator (interaction) |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
177 |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
178 if an indicator profile (time series) is different enough (<minSimilarity), |
452
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
179 it will become a new prototype. |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
180 Non-prototype interactions will be assigned to an existing prototype''' |
430
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
181 |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
182 # sort indicators based on length |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
183 indices = range(similarityMatrix.shape[0]) |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
184 def compare(i, j): |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
185 if len(interactions[i].getIndicator(indicatorName)) > len(interactions[j].getIndicator(indicatorName)): |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
186 return -1 |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
187 elif len(interactions[i].getIndicator(indicatorName)) == len(interactions[j].getIndicator(indicatorName)): |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
188 return 0 |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
189 else: |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
190 return 1 |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
191 indices.sort(compare) |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
192 # go through all indicators |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
193 prototypeIndices = [indices[0]] |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
194 for i in indices[1:]: |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
195 if similarityMatrix[i][prototypeIndices].max() < minSimilarity: |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
196 prototypeIndices.append(i) |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
197 |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
198 # assignment |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
199 labels = [-1]*similarityMatrix.shape[0] |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
200 indices = [i for i in range(similarityMatrix.shape[0]) if i not in prototypeIndices] |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
201 for i in prototypeIndices: |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
202 labels[i] = i |
436
b64ff7fe7b45
corrected clustering bug (in indicator assignment)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
430
diff
changeset
|
203 for i in indices: |
430
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
204 prototypeIndex = similarityMatrix[i][prototypeIndices].argmax() |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
205 labels[i] = prototypeIndices[prototypeIndex] |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
206 |
fb3654a9127d
integrating indicator clustering code
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
408
diff
changeset
|
207 return prototypeIndices, labels |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
208 |
452
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
209 def prototypeMultivariateCluster(interactions, similarityMatrics, indicatorNames, minSimilarities, minClusterSize): |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
210 '''Finds exmaple indicator time series (several indicators) for all interactions |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
211 |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
212 if any interaction indicator time series is different enough (<minSimilarity), |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
213 it will become a new prototype. |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
214 Non-prototype interactions will be assigned to an existing prototype if all indicators are similar enough''' |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
215 pass |
c59a47ce0209
reorganized interactioninterval (in compute indicators) and comments
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
451
diff
changeset
|
216 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
217 # TODO: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
218 #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
|
219 #http://www.rueckstiess.net/research/snippets/show/ca1d7d90 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
220 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
|
221 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
|
222 #print pairs.num |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
223 # Ignore empty collision points |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
224 empty = 1 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
225 for i in collisionPoints: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
226 if(collisionPoints[i] != []): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
227 empty = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
228 if(empty == 1): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
229 pairs.hasCP = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
230 else: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
231 pairs.hasCP = 1 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
232 pairs.CP = collisionPoints |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
233 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
234 # Ignore empty crossing zones |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
235 empty = 1 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
236 for i in crossingZones: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
237 if(crossingZones[i] != []): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
238 empty = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
239 if(empty == 1): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
240 pairs.hasCZ = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
241 else: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
242 pairs.hasCZ = 1 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
243 pairs.CZ = crossingZones |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
244 return pairs |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
245 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
246 def calculateIndicatorPipe_star(a_b): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
247 """Convert `f([1,2])` to `f(1,2)` call.""" |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
248 return calculateIndicatorPipe(*a_b) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
249 |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
250 class VehPairs(): |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
251 '''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
|
252 def __init__(self,objects): |
293
ee3302528cdc
rearranged new code by Paul (works now)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
292
diff
changeset
|
253 self.pairs = createInteractions(objects) |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
254 self.interactionCount = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
255 self.CPcount = 0 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
256 self.CZcount = 0 |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
257 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
258 # Process indicator calculation with support for multi-threading |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
259 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
|
260 if(threads > 1): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
261 pool = multiprocessing.Pool(threads) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
262 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
|
263 pool.close() |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
264 else: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
265 #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
|
266 for j in xrange(len(self.pairs)): |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
267 #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
|
268 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
|
269 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
270 # Ignore empty collision points |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
271 empty = 1 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
272 for i in collisionPoints: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
273 if(collisionPoints[i] != []): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
274 empty = 0 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
275 if(empty == 1): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
276 self.pairs[j].hasCP = 0 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
277 else: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
278 self.pairs[j].hasCP = 1 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
279 self.pairs[j].CP = collisionPoints |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
280 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
281 # Ignore empty crossing zones |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
282 empty = 1 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
283 for i in crossingZones: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
284 if(crossingZones[i] != []): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
285 empty = 0 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
286 if(empty == 1): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
287 self.pairs[j].hasCZ = 0 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
288 else: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
289 self.pairs[j].hasCZ = 1 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
290 self.pairs[j].CZ = crossingZones |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
291 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
292 for j in self.pairs: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
293 self.interactionCount = self.interactionCount + len(j.CP) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
294 self.CPcount = len(self.getCPlist()) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
295 self.Czcount = len(self.getCZlist()) |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
296 |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
297 |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
298 def getPairsWCP(self): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
299 lists = [] |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
300 for j in self.pairs: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
301 if(j.hasCP): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
302 lists.append(j.num) |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
303 return lists |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
304 |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
305 def getPairsWCZ(self): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
306 lists = [] |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
307 for j in self.pairs: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
308 if(j.hasCZ): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
309 lists.append(j.num) |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
310 return lists |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
311 |
526
21bdeb29f855
corrected bug in initialization of lists and loading trajectories from vissim files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
489
diff
changeset
|
312 def getCPlist(self,indicatorThreshold=float('Inf')): |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
313 lists = [] |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
314 for j in self.pairs: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
315 if(j.hasCP): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
316 for k in j.CP: |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
317 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
|
318 lists.append([k,j.CP[k][0]]) |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
319 return lists |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
320 |
526
21bdeb29f855
corrected bug in initialization of lists and loading trajectories from vissim files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
489
diff
changeset
|
321 def getCZlist(self,indicatorThreshold=float('Inf')): |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
322 lists = [] |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
323 for j in self.pairs: |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
324 if(j.hasCZ): |
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
325 for k in j.CZ: |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
326 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
|
327 lists.append([k,j.CZ[k][0]]) |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
328 return lists |
292
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
329 |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
330 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
|
331 if(not CPlist): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
332 CPlist = self.getCPlist() |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
333 if(not CPlist): |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
334 return False |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
335 TTC_list = [] |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
336 for i in CPlist: |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
337 TTC_list.append(i[1].indicator) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
338 histo = np.histogram(TTC_list,bins=bins) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
339 histo += (histo[0].astype(float)/np.sum(histo[0]),) |
8b2c8a4015f1
class VehPairs updated. Now supports primitive multithreading.
Paul@BEAST-III
parents:
291
diff
changeset
|
340 return histo |
291
9f81218e497a
class VehPairs subsumes createInteractions(objects); legacy code remains
Paul@BEAST-III
parents:
283
diff
changeset
|
341 |
59
f955e83da499
developed indicators in interactions
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
56
diff
changeset
|
342 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
|
343 '''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
|
344 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
345 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
|
346 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
|
347 carac traversee |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
348 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
|
349 carac interaction''' |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
350 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
351 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
|
352 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
|
353 self.roaduserNum = roaduserNum |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
354 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
355 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
356 |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
357 if __name__ == "__main__": |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
358 import doctest |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
359 import unittest |
489
000bddf84ad0
corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
487
diff
changeset
|
360 suite = doctest.DocFileSuite('tests/events.txt') |
000bddf84ad0
corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
487
diff
changeset
|
361 #suite = doctest.DocTestSuite() |
56
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
362 unittest.TextTestRunner().run(suite) |
61fe73df2d36
created new package, moved Interaction class and created Crossing class
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
363 |