nsaunier/traffic-intelligence

corrected bug on adding curvilinear trajectories

Commit 9562f5e8edf8 · Nicolas Saunier · 2024-07-03 15:10 -0400

Changeset

9562f5e8edf87f91afa0aab384fbcdcbba3745f4

View source at this commit

Comments

No comments yet.

Log in to comment

Diff

diff --git a/trafficintelligence/storage.py b/trafficintelligence/storage.py
--- a/trafficintelligence/storage.py
+++ b/trafficintelligence/storage.py
@@ -359,9 +359,9 @@
     '''Adds curvilinear positions (s_coordinate, y_coordinate, lane)
     from a database to an existing MovingObject dict (indexed by each objects's num) or list'''
     if type(objects) == list:
-        _objects = {obj.getNum(): obj for obj in objects}
+        tmpobjects = {obj.getNum(): obj for obj in objects}
     elif type(objects) == dict:
-        _objects = objects
+        tmpobjects = objects
     else:
         print('objects should be a list of dictionary of objects indexed by their num')
         return None
@@ -380,12 +380,12 @@
         for row in cursor:
             if objNum != row[0]:
                 objNum = row[0]
-                if objNum in _objects:
-                    _objects[objNum].curvilinearPositions = moving.CurvilinearTrajectory()
+                if objNum in tmpobjects:
+                    tmpobjects[objNum].curvilinearPositions = moving.CurvilinearTrajectory()
                 else:
                     missingObjectNumbers.append(objNum)
-            if objNum in _objects:
-                objects[objNum].curvilinearPositions.addPositionSYL(row[2],row[3],row[4])
+            if objNum in tmpobjects:
+                tmpobjects[objNum].curvilinearPositions.addPositionSYL(row[2],row[3],row[4])
         if len(missingObjectNumbers) > 0:
             print('List of missing objects to attach corresponding curvilinear trajectories: {}'.format(missingObjectNumbers))