# HG changeset patch # User Nicolas Saunier # Date 1720033847 14400 # Node ID 9562f5e8edf87f91afa0aab384fbcdcbba3745f4 # Parent 8e61ff3cd503dab81f7f443718551010072adf3a corrected bug on adding curvilinear trajectories diff -r 8e61ff3cd503 -r 9562f5e8edf8 trafficintelligence/storage.py --- 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))