Mercurial Hosting > traffic-intelligence

annotate trafficintelligence/tests/events.txt @ 1148:eb88d2984637

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
corrected interaction classification
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 12 May 2020 01:16:54 -0400
parents aafbc0bab925
children 14140b55e580
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1030
aafbc0bab925 moved method around to avoid cross-dependencies
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1028
diff changeset
1 >>> from trafficintelligence.events import *
aafbc0bab925 moved method around to avoid cross-dependencies
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1028
diff changeset
2 >>> from trafficintelligence.moving import MovingObject, TimeInterval, Point
aafbc0bab925 moved method around to avoid cross-dependencies
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1028
diff changeset
3 >>> from trafficintelligence.prediction import ConstantPredictionParameters
489
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
4
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
5 >>> objects = [MovingObject(num = i, timeInterval = TimeInterval(0,10)) for i in range(10)]
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
6 >>> interactions = createInteractions(objects)
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
7 >>> len([i for i in interactions if len(i.roadUserNumbers) == 1])
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
8 0
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
9 >>> objects2 = [MovingObject(num = i, timeInterval = TimeInterval(0,10)) for i in range(100, 110)]
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
10 >>> interactions = createInteractions(objects, objects2)
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
11 >>> len([i for i in interactions if len(i.roadUserNumbers) == 1])
000bddf84ad0 corrected bugs in safety analysis
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff changeset
12 0
662
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
13
918
3a06007a4bb7 modularized save trajectories, added slice to Trajectory, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 662
diff changeset
14 >>> o1 = MovingObject.generate(1, Point(-5.,0.), Point(1.,0.), TimeInterval(0,10))
3a06007a4bb7 modularized save trajectories, added slice to Trajectory, etc
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 662
diff changeset
15 >>> o2 = MovingObject.generate(2, Point(0.,-5.), Point(0.,1.), TimeInterval(0,10))
662
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
16 >>> inter = Interaction(roadUser1 = o1, roadUser2 = o2)
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
17 >>> inter.computeIndicators()
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
18 >>> predictionParams = ConstantPredictionParameters(10.)
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
19 >>> inter.computeCrossingsCollisions(predictionParams, 0.1, 10)
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
20 >>> ttc = inter.getIndicator("Time to Collision")
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
21 >>> ttc[0]
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
22 5.0
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
23 >>> ttc[1]
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
24 4.0
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
25 >>> (inter.collisionPoints[0][0] - Point(0.,0.)).norm2() < 0.0001
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
26 True
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
27 >>> (inter.collisionPoints[4][0] - Point(0.,0.)).norm2() < 0.0001
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
28 True
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
29 >>> inter.getIndicator(Interaction.indicatorNames[1])[4] < 0.000001 # collision angle
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
30 True
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
31 >>> inter.getIndicator(Interaction.indicatorNames[1])[5] is None
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
32 True
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
33 >>> inter.getIndicator(Interaction.indicatorNames[1])[6] # doctest:+ELLIPSIS
72174e66aba5 corrected bug that increased TTC by 1 frame and structure to store collision points and crossing zones
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 489
diff changeset
34 3.1415...
1148
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
35
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
36 >>> from collections import Counter
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
37 >>> from numpy import pi
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
38 >>> o1 = MovingObject.generate(0, Point(0,0), Point(1,0), TimeInterval(0,100))
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
39 >>> o2 = MovingObject.generate(0, Point(100,1), Point(-1,0), TimeInterval(0,100))
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
40 >>> inter12 = Interaction(roadUser1 = o1, roadUser2 = o2)
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
41 >>> inter12.computeIndicators()
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
42 >>> inter12.categorize(pi*20/180, pi*60/180)
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
43 >>> Counter(inter12.categories.values()).most_common()[0][0] # head on
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
44 0
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
45 >>> inter12.categories[max(inter12.categories.keys())] # then side
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
46 2
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
47 >>> o3 = MovingObject.generate(0, Point(0,2), Point(1,0), TimeInterval(0,100))
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
48 >>> inter13 = Interaction(roadUser1 = o1, roadUser2 = o3)
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
49 >>> inter13.computeIndicators()
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
50 >>> inter13.categorize(pi*20/180, pi*60/180)
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
51 >>> Counter(inter13.categories.values()).most_common()[0][0] # parallel
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
52 3
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
53 >>> len(Counter(inter13.categories.values()))
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
54 1
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
55 >>> o4 = MovingObject.generate(0, Point(100,20), Point(-1,0), TimeInterval(0,100))
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
56 >>> inter14 = Interaction(roadUser1 = o1, roadUser2 = o4)
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
57 >>> inter14.computeIndicators()
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
58 >>> inter14.categorize(pi*20/180, pi*60/180)
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
59 >>> Counter(inter14.categories.values()).most_common()[0][0] # side
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
60 2
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
61 >>> inter12.categories[0] # first head one
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
62 0
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
63 >>> inter12.categories[max(inter12.categories.keys())] # then side
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
64 2
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
65 >>> o5 = MovingObject.generate(0, Point(50,50), Point(0,-1), TimeInterval(0,100))
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
66 >>> inter15 = Interaction(roadUser1 = o1, roadUser2 = o5)
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
67 >>> inter15.computeIndicators()
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
68 >>> inter15.categorize(pi*20/180, pi*60/180)
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
69 >>> Counter(inter15.categories.values()).most_common()[0][0] # side
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
70 2
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
71 >>> len(Counter(inter15.categories.values()))
eb88d2984637 corrected interaction classification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents: 1030
diff changeset
72 1