Colony

#!/usr/bin/env python2

#

#Colony Graph Drawing

#Pipe stdin in the form

# <SOURCE NODE> <CONNECTED NODE 1> <CONNECTED NODE 2> ... <CONNECTED NODE N>

# (space separated)

######

######

#USES GRAPHVIZ LIBRARY

#USES PYTHON-GRAPH LIBRARY

#IF YOU DON'T HAVE THOSE, THIS WILL DIE

######

######

# Import graphviz

import sys

sys.path.append('..')

sys.path.append('/usr/lib/graphviz/python/')

sys.path.append('/usr/lib64/graphviz/python/')

import gv

# Import pygraph

from pygraph.classes.graph import graph

from pygraph.classes.digraph import digraph

from pygraph.algorithms.searching import breadth_first_search

from pygraph.readwrite.dot import write

# Graph creation

gr = digraph()

nodes = [];

def addNode(x):

        if(not(x in nodes)):

                nodes.append(x);

                gr.add_nodes([x]);

def drawGraph():

        dot = write(gr)

        gvv = gv.readstring(dot)

        gv.setv(gvv, 'overlap', 'none')

        gv.setv(gvv, 'splines', 'true')

        gv.setv(gvv, 'sep', '0.25')

        gv.setv(gvv, 'mode', 'maxent')

        gv.layout(gvv,'sfdp')

        gv.render(gvv,'png','colony.png')

print "Colony Graph Generator\n==================\n"

print "Pipe in space separated <node a> <node b> <node c> ... <node n>"

print "Where a connects to nodes b, c, ..., n"

print "Outputs as colony.png\n\n"

print "Everything is space-separated"

print "And newlines denote a new source node"

print "This code uses the python-graph library and is based on example code from that library's website"

print "\n"

l="asdf"

while(not(l=="")):

        l=sys.stdin.readline()

        l_s = l.split()

        map(addNode, l_s)

        for y in l_s[1:]:

                gr.add_edge((l_s[0],y))

        drawGraph()