nsaunier/traffic-intelligence: python/storage.py comparison

comparison python/storage.py @ 491:343cfd185ca6

minor changes and reaarrangements

author	Nicolas Saunier <nicolas.saunier@polymtl.ca>
date	Wed, 16 Apr 2014 17:43:53 -0400
parents	f6415f012640
children	935430b1d408

comparison

equal deleted inserted replaced

-:60735bd452fc
+:343cfd185ca6
 #########################
 # Sqlite
 #########################
+# utils
+def printDBError(error):
+print('DB Error: {}'.format(error))
+def dropTables(connection, tableNames):
+'deletes the table with names in tableNames'
+try:
+cursor = connection.cursor()
+for tableName in tableNames:
+cursor.execute('DROP TABLE IF EXISTS '+tableName)
+except sqlite3.OperationalError as error:
+printDBError(error)
+# IO to sqlite
 def writeTrajectoriesToSqlite(objects, outFilename, trajectoryType, objectNumbers = -1):
 """
 This function writers trajectories to a specified sqlite file
 @param[in] objects -> a list of trajectories
 @param[in] trajectoryType -
 cursor = connection.cursor()
 try:
 cursor.execute('SELECT * from prototypes order by prototype_id, trajectory_id_matched')
 except sqlite3.OperationalError as error:
-utils.printDBError(error)
+printDBError(error)
 return []
 for row in cursor:
 matched_indexes.append((row[0],row[1]))
 cursor.execute(queryStatement)
 logging.debug(queryStatement)
 else:
 print('no trajectory type was chosen')
 except sqlite3.OperationalError as error:
-utils.printDBError(error)
+printDBError(error)
 return []
 objId = -1
 obj = None
 objects = []
 for obj in objects:
 obj.userType = userTypes[obj.getNum()]
 except sqlite3.OperationalError as error:
-utils.printDBError(error)
+printDBError(error)
 return []
 connection.close()
 return objects
 def removeFromSqlite(filename, dataType):
 'Removes some tables in the filename depending on type of data'
-connection = sqlite3.connect(filename)
+import os
-if dataType == 'object':
+if os.path.isfile(filename):
-utils.dropTables(connection, ['objects', 'objects_features'])
+connection = sqlite3.connect(filename)
-elif dataType == 'interaction':
+if dataType == 'object':
-utils.dropTables(connection, ['interactions', 'indicators'])
+dropTables(connection, ['objects', 'objects_features'])
-elif dataType == 'bb':
+elif dataType == 'interaction':
-utils.dropTables(connection, ['bounding_boxes'])
+dropTables(connection, ['interactions', 'indicators'])
+elif dataType == 'bb':
+dropTables(connection, ['bounding_boxes'])
+else:
+print('Unknown data type {} to delete from database'.format(dataType))
+connection.close()
 else:
-print('Unknown data type {} to delete from database'.format(dataType))
+print('{} does not exist'.format(filename))
-connection.close()
 def createInteractionTable(cursor):
 cursor.execute('CREATE TABLE IF NOT EXISTS interactions (id INTEGER PRIMARY KEY, object_id1 INTEGER, object_id2 INTEGER, first_frame_number INTEGER, last_frame_number INTEGER, FOREIGN KEY(object_id1) REFERENCES objects(id), FOREIGN KEY(object_id2) REFERENCES objects(id))')
 def createIndicatorTables(cursor):
 try:
 createInteractionTable(cursor)
 for inter in interactions:
 saveInteraction(cursor, inter)
 except sqlite3.OperationalError as error:
-utils.printDBError(error)
+printDBError(error)
 connection.commit()
 connection.close()
 def saveIndicator(cursor, interactionNum, indicator):
 for instant in indicator.getTimeInterval():
 for indicatorName in indicatorNames:
 indicator = inter.getIndicator(indicatorName)
 if indicator != None:
 saveIndicator(cursor, inter.getNum(), indicator)
 except sqlite3.OperationalError as error:
-utils.printDBError(error)
+printDBError(error)
 connection.commit()
 connection.close()
 def loadInteractions(filename):
 '''Loads interaction and their indicators
 indicatorName = events.Interaction.indicatorNames[indicatorTypeNum]
 tmpIndicators[indicatorName] = indicators.SeverityIndicator(indicatorName, indicatorValues)
 interactions.append(events.Interaction(interactionNum, moving.TimeInterval(row[3],row[4]), roadUserNumbers[0], roadUserNumbers[1]))
 interactions[-1].indicators = tmpIndicators
 except sqlite3.OperationalError as error:
-utils.printDBError(error)
+printDBError(error)
 return []
 connection.close()
 return interactions
 # load first and last object instants
 # CREATE TEMP TABLE IF NOT EXISTS object_instants AS SELECT OF.object_id, min(frame_number) as first_instant, max(frame_number) as last_instant from positions P, objects_features OF where P.trajectory_id = OF.trajectory_id group by OF.object_id order by OF.object_id
 cursor.execute('INSERT INTO bounding_boxes SELECT object_id, frame_number, min(x), min(y), max(x), max(y) from '
 '(SELECT object_id, frame_number, (x*{}+y*{}+{})/w as x, (x*{}+y*{}+{})/w as y from '
 '(SELECT OF.object_id, P.frame_number, P.x_coordinate as x, P.y_coordinate as y, P.x_coordinate*{}+P.y_coordinate*{}+{} as w from positions P, objects_features OF where P.trajectory_id = OF.trajectory_id)) '.format(invHomography[0,0], invHomography[0,1], invHomography[0,2], invHomography[1,0], invHomography[1,1], invHomography[1,2], invHomography[2,0], invHomography[2,1], invHomography[2,2])+
 'GROUP BY object_id, frame_number')
 except sqlite3.OperationalError as error:
-utils.printDBError(error)
+printDBError(error)
 connection.commit()
 connection.close()
 def loadBoundingBoxTable(filename):
 connection = sqlite3.connect(filename)
 #objId = -1
 for row in cursor:
 #if row[0] != objId:
 boundingBoxes.setdefault(row[1], []).append([moving.Point(row[2], row[3]), moving.Point(row[4], row[5])])
 except sqlite3.OperationalError as error:
-utils.printDBError(error)
+printDBError(error)
 return boundingBoxes
 connection.close()
 return boundingBoxes

Mercurial > hg > nsaunier > traffic-intelligence

comparison python/storage.py @ 491:343cfd185ca6