Mercurial > hg > nsaunier > traffic-intelligence

comparison python/storage.py @ 218:b5772df11b37

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
corrected bugs to load objects and display trajectories over videos
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Thu, 21 Jun 2012 18:04:11 -0400
parents 5e2983b05d4e
children 584613399513
comparison
equal deleted inserted replaced
217:ba71924cadf5 218:b5772df11b37
79 try: 79 try:
80 if trajectoryType == 'feature': 80 if trajectoryType == 'feature':
81 if type(objectNumbers) == int: 81 if type(objectNumbers) == int:
82 if objectNumbers == -1: 82 if objectNumbers == -1:
83 cursor.execute('SELECT * from positions order by trajectory_id, frame_number') 83 cursor.execute('SELECT * from positions order by trajectory_id, frame_number')
84 else: 84 else:
85 cursor.execute('SELECT * from positions where trajectory_id between 0 and {0} order by trajectory_id, frame_number'.format(objectNumbers)) 85 cursor.execute('SELECT * from positions where trajectory_id between 0 and {0} order by trajectory_id, frame_number'.format(objectNumbers))
86 elif type(objectNumbers) == list: 86 elif type(objectNumbers) == list:
87 cursor.execute('SELECT * from positions where trajectory_id in ('+', '.join([str(n) for n in objectNumbers])+') order by trajectory_id, frame_number') 87 cursor.execute('SELECT * from positions where trajectory_id in ('+', '.join([str(n) for n in objectNumbers])+') order by trajectory_id, frame_number')
88 elif trajectoryType == 'object': 88 elif trajectoryType == 'object':
89 if type(objectNumbers) == int: 89 if type(objectNumbers) == int:
90 if objectNumbers == -1: 90 if objectNumbers == -1:
91 cursor.execute('SELECT OF.object_id, P.frame_number, avg(P.x_coordinate), avg(P.y_coordinate) from positions P, objects_features OF where P.trajectory_id = OF.trajectory_id group by object_id, frame_number') 91 cursor.execute('SELECT OF.object_id, P.frame_number, avg(P.x_coordinate), avg(P.y_coordinate) from positions P, objects_features OF where P.trajectory_id = OF.trajectory_id group by object_id, frame_number')
92 else: 92 else:
93 cursor.execute('SELECT OF.object_id, P.frame_number, avg(P.x_coordinate), avg(P.y_coordinate) from positions P, objects_features OF where P.trajectory_id = OF.trajectory_id and trajectory_id between 0 and {0} group by object_id, frame_number'.format(objectNumbers)) 93 cursor.execute('SELECT OF.object_id, P.frame_number, avg(P.x_coordinate), avg(P.y_coordinate) from positions P, objects_features OF where P.trajectory_id = OF.trajectory_id and OF.object_id between 0 and {0} group by object_id, frame_number'.format(objectNumbers))
94 elif type(objectNumbers) == list: 94 elif type(objectNumbers) == list:
95 cursor.execute('SELECT OF.object_id, P.frame_number, avg(P.x_coordinate), avg(P.y_coordinate) from positions P, objects_features OF where P.trajectory_id = OF.trajectory_id and trajectory_id in ('+', '.join([str(n) for n in objectNumbers])+') group by object_id, frame_number') 95 cursor.execute('SELECT OF.object_id, P.frame_number, avg(P.x_coordinate), avg(P.y_coordinate) from positions P, objects_features OF where P.trajectory_id = OF.trajectory_id and OF.object_id in ('+', '.join([str(n) for n in objectNumbers])+') group by object_id, frame_number')
96 else: 96 else:
97 print('no trajectory type was chosen') 97 print('no trajectory type was chosen')
98 except sqlite3.OperationalError as err: 98 except sqlite3.OperationalError as err:
99 print('DB Error: {0}'.format(err)) 99 print('DB Error: {0}'.format(err))
100 return [] 100 return []