changeset 1135:342701cdac30

bug fix for concatenate
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 05 Mar 2020 16:53:48 -0500
parents 4b2a55d570c1
children 30171d4fd3df e9c12982ed28
files trafficintelligence/moving.py trafficintelligence/utils.py
diffstat 2 files changed, 10 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/trafficintelligence/moving.py	Thu Mar 05 16:20:29 2020 -0500
+++ b/trafficintelligence/moving.py	Thu Mar 05 16:53:48 2020 -0500
@@ -1326,7 +1326,7 @@
             featurePositions = Trajectory()
             featureVelocities = Trajectory()
             p = firstObject.getPositionAtInstant(emptyInterval.first)+v
-            for t in range(emptyInterval.first+1, emptyInterval.last):
+            for t in range(emptyInterval.first+1, emptyInterval.last+1):
             	positions.addPosition(p)
             	velocities.addPosition(v)
             	featurePositions.addPosition(p)
--- a/trafficintelligence/utils.py	Thu Mar 05 16:20:29 2020 -0500
+++ b/trafficintelligence/utils.py	Thu Mar 05 16:53:48 2020 -0500
@@ -88,6 +88,15 @@
         result[x] = i
     return result
 
+def findElement(l, num):
+    i = 0
+    while l[i].getNum() != num:
+        i += 1
+    if i < len(l):
+        return l[i]
+    else:
+        return None
+
 #########################
 # Simple statistics
 #########################