def write_graph(graph, output_file, filename, output_type='gr'):
output = cStringIO.StringIO()
symtab = labels = False
if output_type == 'gr':
from utils.graph_output import write_gr as write_graph
write_graph(graph,
symtab=symtab,
labels=labels,
output=output,
gtfs_filename=filename)
elif output_type == 'gml':
from utils.graph_output import write_gml as write_graph
write_graph(graph,
symtab=symtab,
labels=labels,
output=output,
gtfs_filename=filename)
if output_file != '-':
#logging.warning('Writing output to file')
with open(output_file, 'w') as f:
f.write(output.getvalue())
f.flush()
#logging.warning('Output written to: %s' % output_file)
else:
print output.getvalue()
labels=labels,
output=output,
gtfs_filename=filename)
if output_file != '-':
#logging.warning('Writing output to file')
with open(output_file, 'w') as f:
f.write(output.getvalue())
f.flush()
#logging.warning('Output written to: %s' % output_file)
else:
print output.getvalue()
for filename in os.listdir(path):
# if not filename.endswith('.csvdata'):
# continue
# if filename.startswith('vertices'):
# continue
if not filename in ('out.opsahl-powergrid'):
continue
filepath = os.path.join(path, filename)
print 'filepath=', filepath
basename = os.path.basename(filepath)
g = create_graph(filepath)
#write_graph(graph=g,filename='test.gr')
write_graph(output_file='%s.gr' % filename,
graph=g,
filename=basename,
output_type='gr')
#write_graph(output_file='%s.gml'%filename, graph=g,filename=basename, output_type='gml')