Mercurial Hosting > traffic-intelligence
diff python/storage.py @ 335:3950bfe22768
added functions to export trajectories to csv
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 14 Jun 2013 11:47:24 -0400 |
parents | 00800ebae698 |
children | 1046b7346886 |
line wrap: on
line diff
--- a/python/storage.py Fri Jun 14 10:34:11 2013 -0400 +++ b/python/storage.py Fri Jun 14 11:47:24 2013 -0400 @@ -290,6 +290,24 @@ out.close() +def writePositionsToCsv(f, obj): + timeInterval = obj.getTimeInterval() + positions = obj.getPositions() + curvilinearPositions = obj.getCurvilinearPositions() + for i in xrange(int(obj.length())): + p1 = positions[i] + s = '{},{},{},{}'.format(obj.num,timeInterval[i],p1.x,p1.y) + if curvilinearPositions != None: + p2 = curvilinearPositions[i] + s += ',{},{}'.format(p2[0],p2[1]) + f.write(s+'\n') + +def writeTrajectoriesToCsv(filename, objects): + f = open(filename, 'w') + for i,obj in enumerate(objects): + writePositionsToCsv(f, obj) + f.close() + if __name__ == "__main__": import doctest import unittest