Mercurial Hosting > traffic-intelligence
comparison python/storage.py @ 218:b5772df11b37
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 [] |