# HG changeset patch
# User Nicolas Saunier <nicolas.saunier@polymtl.ca>
# Date 1512508309 18000
# Node ID 40af0f20ee2dbe2fa315b78ba2cc00870c16467b
# Parent  eec549e9ff2e21097b8f5fb9a80b98571b254ac9
reworking the functions

diff -r eec549e9ff2e -r 40af0f20ee2d python/sumo.py
--- a/python/sumo.py	Tue Dec 05 11:52:59 2017 -0500
+++ b/python/sumo.py	Tue Dec 05 16:11:49 2017 -0500
@@ -5,7 +5,7 @@
 
 import csv
 
-def convertTazEdges(inFilename, outFilename):
+def loadTazEdges(inFilename):
     '''Converts list of OSM edges per OSM edge and groups per TAZ'''
     data = []
     tazs = {}
@@ -26,13 +26,15 @@
                                 tazs[tazID].append(edge)
                         else:
                             tazs[tazID] = [edge]
-
+    return tazs
+        
+def saveTazEdges(outFilename, tazs):
     with open(outFilename,'w') as out:
 	out.write('<tazs>\n')				
 	for tazID in tazs:
 	    out.write('<taz id="{}" edges="'.format(tazID)+' '.join(tazs[tazID])+'"/>\n')
 	out.write('</tazs>\n')
-
+    
 # TODO add utils from process-cyber.py?
         
 # if __name__ == "__main__":