Mercurial Hosting > traffic-intelligence
changeset 1279:9562f5e8edf8
corrected bug on adding curvilinear trajectories
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Wed, 03 Jul 2024 15:10:47 -0400 |
| parents | 8e61ff3cd503 |
| children | 2abeccdbb985 |
| files | trafficintelligence/storage.py |
| diffstat | 1 files changed, 6 insertions(+), 6 deletions(-) [+] |
line wrap: on
line diff
--- a/trafficintelligence/storage.py Thu Jun 27 15:31:36 2024 -0400 +++ b/trafficintelligence/storage.py Wed Jul 03 15:10:47 2024 -0400 @@ -359,9 +359,9 @@ '''Adds curvilinear positions (s_coordinate, y_coordinate, lane) 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} + tmpobjects = {obj.getNum(): obj for obj in objects} elif type(objects) == dict: - _objects = objects + tmpobjects = objects else: print('objects should be a list of dictionary of objects indexed by their num') return None @@ -380,12 +380,12 @@ for row in cursor: if objNum != row[0]: objNum = row[0] - if objNum in _objects: - _objects[objNum].curvilinearPositions = moving.CurvilinearTrajectory() + if objNum in tmpobjects: + tmpobjects[objNum].curvilinearPositions = moving.CurvilinearTrajectory() else: missingObjectNumbers.append(objNum) - if objNum in _objects: - objects[objNum].curvilinearPositions.addPositionSYL(row[2],row[3],row[4]) + if objNum in tmpobjects: + tmpobjects[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))