--- a/trafficintelligence/storage.py	Wed Apr 03 16:39:41 2024 -0400
+++ b/trafficintelligence/storage.py	Sun Apr 07 22:09:40 2024 -0400
@@ -357,7 +357,15 @@
 
 def addCurvilinearTrajectoriesFromSqlite(filename, objects):
     '''Adds curvilinear positions (s_coordinate, y_coordinate, lane)
-    from a database to an existing MovingObject dict (indexed by each objects's num)'''
+    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}
+    elif type(objects) == dict:
+        _objects = objects
+    else:
+        print('objects should be a list of dictionary of objects indexed by their num')
+        return None
+    
     with sqlite3.connect(filename) as connection:
         cursor = connection.cursor()
 
@@ -372,11 +380,11 @@
         for row in cursor:
             if objNum != row[0]:
                 objNum = row[0]
-                if objNum in objects:
-                    objects[objNum].curvilinearPositions = moving.CurvilinearTrajectory()
+                if objNum in _objects:
+                    _objects[objNum].curvilinearPositions = moving.CurvilinearTrajectory()
                 else:
                     missingObjectNumbers.append(objNum)
-            if objNum in objects:
+            if objNum in _objects:
                 objects[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))