#!/usr/bin/python

##########################################
#
# 9.14.08 Joe Mariglio (www.joemariglio.com)
# 
# this should take a connectivity matrix 
# (stored in a file) and turn it into 
# xml data.
#
##########################################

import sys, string, os.path, getopt

docheader = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
docheader = docheader + "<graphml xmlns=\"http://graphml.graphdrawing.org/xmlns http://graphml.graphdrawing.org/xmlns/1.0/graphml.xsd\">"
docheader = docheader + "\n<graph id= \"G\">\n"

docfooter = "</graph>"
docfooter = docfooter+"\n</graphml>"

def mkedge(row, column):
	edge = "<edge source="+"\""+ str(row)+ "\" "+" target="+"\""+ str(column)+ "\" "+ "/>\n"
	return edge


def parse(infile):
	outdoc = docheader
	row = 0
	for line in infile:
		column = 0
		for char in line:
			if char == '0':
				column = column + 1
			elif char == '1':
				#make new edge connecting column to row
				outdoc = outdoc + mkedge(row, column)
				column = column + 1
		row = row + 1
	outdoc = outdoc + docfooter
	return outdoc
				

def usage():
	print "usage:"
	print "matrixToXML <input_filepath> <output_filepath>"

def main(argv):

	if len(sys.argv) < 3:
		usage()
	elif len(sys.argv) == 3:
		#input file
		inpath = sys.argv[1]
		infile = open(inpath, "r")
		#output file
		outpath = sys.argv[2]
		outfile = open(outpath, "w")
		doc = parse(infile)
		print doc
		outfile.write(doc)
		infile.close()
		outfile.close()

if __name__ == '__main__':
	import sys, string, os.path, getopt
	main(sys.argv[1:])