TreeToEdges Namespace Reference

Classes
class	Int

Functions
def	runme

Variables
tuple	cutoff = int(sys.argv[3])
list	infile = sys.argv[1]
list	outfile = sys.argv[2]

Function Documentation

def TreeToEdges.runme	(		infile,
			outfile,
			cutoff
	)

Definition at line 18 of file TreeToEdges.py.

def runme(infile,outfile,cutoff):
    fin = open(infile,'r')
    fout = open(outfile,'w')
    tree = {}
    count = 0
    
    for line in fin.xreadlines():

        a = line.split()
        id = int(a.pop(0))
        tot = int(a.pop(0))
        if tot < cutoff: break
        head = int(a.pop(0))
        
        for node in a:
            val = int(node)
            key = (head,val)
        
            n = tree.get(key)
            if n == None:
                tree[key] = Int(tot)
            else:
                n.inc(tot)
            head = val
            
        count += 1

    for node in tree.items():
        # print node
        print >>fout, node[1], ' ', node[0][0], ' ', node[0][1]