diff trafficintelligence/storage.py @ 1039:5621e4ad2428

removing prefix option to loadtrajectories from SQLite
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 04 Jul 2018 12:24:37 -0400
parents 6a6c37eb3a74
children 20799ac9524e
line wrap: on
line diff
--- a/trafficintelligence/storage.py	Wed Jul 04 00:30:40 2018 -0400
+++ b/trafficintelligence/storage.py	Wed Jul 04 12:24:37 2018 -0400
@@ -240,19 +240,15 @@
         userTypes[row[0]] = row[1]
     return userTypes
 
-def loadTrajectoriesFromSqlite(filename, trajectoryType, objectNumbers = None, withFeatures = False, timeStep = None, tablePrefix = None):
+def loadTrajectoriesFromSqlite(filename, trajectoryType, objectNumbers = None, withFeatures = False, timeStep = None):
     '''Loads the trajectories (in the general sense, 
     either features, objects (feature groups) or bounding box series) 
     The number loaded is either the first objectNumbers objects,
     or the indices in objectNumbers from the database'''
     objects = []
     with sqlite3.connect(filename) as connection:
-        if tablePrefix is None:
-            prefix = ''
-        else:
-            prefix = tablePrefix + '_'
-        objects = loadTrajectoriesFromTable(connection, prefix+'positions', trajectoryType, objectNumbers, timeStep)
-        objectVelocities = loadTrajectoriesFromTable(connection, prefix+'velocities', trajectoryType, objectNumbers, timeStep)
+        objects = loadTrajectoriesFromTable(connection, 'positions', trajectoryType, objectNumbers, timeStep)
+        objectVelocities = loadTrajectoriesFromTable(connection, 'velocities', trajectoryType, objectNumbers, timeStep)
 
         if len(objectVelocities) > 0:
             for o,v in zip(objects, objectVelocities):
@@ -324,6 +320,12 @@
             printDBError(error)
             return None
 
+def loadObjectTrajectoriesFromSqlite():
+    '''Loads object trajectories 
+    either simply objects or features (defaults to loadTrajectoriesFromSqlite) 
+    or the longest features for each object '''
+        
+
 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)'''
@@ -350,7 +352,7 @@
         if len(missingObjectNumbers) > 0:
             print('List of missing objects to attach corresponding curvilinear trajectories: {}'.format(missingObjectNumbers))
 
-def saveTrajectoriesToTable(connection, objects, trajectoryType, tablePrefix = None):
+def saveTrajectoriesToTable(connection, objects, trajectoryType):
     'Saves trajectories in table tableName'
     cursor = connection.cursor()
     # Parse feature and/or object structure and commit to DB
@@ -366,14 +368,10 @@
         elif trajectoryType == 'feature':
             features = objects
         # Setup feature queries
-        if tablePrefix is None:
-            prefix = ''
-        else:
-            prefix = tablePrefix+'_'
-        createTrajectoryTable(cursor, prefix+"positions")
-        createTrajectoryTable(cursor, prefix+"velocities")
-        positionQuery = insertTrajectoryQuery(prefix+"positions")
-        velocityQuery = insertTrajectoryQuery(prefix+"velocities")
+        createTrajectoryTable(cursor, "positions")
+        createTrajectoryTable(cursor, "velocities")
+        positionQuery = insertTrajectoryQuery("positions")
+        velocityQuery = insertTrajectoryQuery("velocities")
         # Setup object queries
         if trajectoryType == 'object':    
             createObjectsTable(cursor)