Mercurial > hg > nsaunier > traffic-intelligence

comparison python/storage.py @ 759:a05b70f307dd dev

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added function to count vehicles per VISSIM link
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 04 Nov 2015 17:12:06 -0500
parents a73f43aac00e
children f8e0a8ea8402
comparison
equal deleted inserted replaced
757:61248f2cd459 759:a05b70f307dd
759 cursor.execute(queryStatement) 759 cursor.execute(queryStatement)
760 return [row[0] for row in cursor] 760 return [row[0] for row in cursor]
761 except sqlite3.OperationalError as error: 761 except sqlite3.OperationalError as error:
762 printDBError(error) 762 printDBError(error)
763 763
764 def getNObjectsInLinkFromVissimFile(filename, linkIds):
765 '''Returns the number of objects that traveled through the link ids'''
766 connection = sqlite3.connect(filename)
767 cursor = connection.cursor()
768 queryStatement = 'SELECT link_id, COUNT(DISTINCT trajectory_id) FROM curvilinear_positions where link_id IN ('+','.join([str(id) for id in linkIds])+') GROUP BY link_id'
769 try:
770 cursor.execute(queryStatement)
771 return {row[0]:row[1] for row in cursor}
772 except sqlite3.OperationalError as error:
773 printDBError(error)
764 774
765 def loadTrajectoriesFromVissimFile(filename, simulationStepsPerTimeUnit, objectNumbers = None, warmUpLastInstant = None, usePandas = False, nDecimals = 2, lowMemory = True): 775 def loadTrajectoriesFromVissimFile(filename, simulationStepsPerTimeUnit, objectNumbers = None, warmUpLastInstant = None, usePandas = False, nDecimals = 2, lowMemory = True):
766 '''Reads data from VISSIM .fzp trajectory file 776 '''Reads data from VISSIM .fzp trajectory file
767 simulationStepsPerTimeUnit is the number of simulation steps per unit of time used by VISSIM (second) 777 simulationStepsPerTimeUnit is the number of simulation steps per unit of time used by VISSIM (second)
768 for example, there seems to be 10 simulation steps per simulated second in VISSIM, 778 for example, there seems to be 10 simulation steps per simulated second in VISSIM,