--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/dottoxml/src/dottoxml.py	Mon Jul 23 09:54:49 2012 +0200
@@ -0,0 +1,320 @@
+# coding: latin-1
+# Copyright (c) 2009,2010,2011,2012 Dirk Baechle.
+# www: http://www.mydarc.de/dl9obn/programming/python/dottoxml
+# mail: dl9obn AT darc.de
+#
+# This program is free software; you can redistribute it and/or modify it under
+# the terms of the GNU General Public License as published by the Free Software
+# Foundation; either version 2 of the License, or (at your option) any later
+# version.
+#
+# This program is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along with
+# this program; if not, write to the Free Software Foundation, Inc.,
+# 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+"""
+  %dottoxml.py [options] <infile.dot> <outfile.graphml>
+
+  convert a DOT file to Graphml XML (and various other formats)
+"""
+
+import sys
+import locale
+import optparse
+
+import dot
+
+# Usage message
+usgmsg = "Usage: dottoxml.py [options] infile.dot outfile.graphml"
+
+def usage():
+    print "dottoxml 1.3, 2012-04-04, Dirk Baechle\n"
+    print usgmsg
+    print "Hint: Try '-h' or '--help' for further infos!"
+
+def exportDot(o, nodes, edges, options):
+    o.write("graph [\n")
+
+    for k,nod in nodes.iteritems():
+        nod.exportDot(o,options)
+    for el in edges:
+        el.exportDot(o,nodes,options)
+
+def exportGML(o, nodes, edges, options):
+    o.write("graph [\n")
+    o.write("  comment \"Created by dottoxml.py\"\n")
+    o.write("  directed 1\n")
+    o.write("  IsPlanar 1\n")
+
+    for k,nod in nodes.iteritems():
+        nod.exportGML(o,options)
+    for el in edges:
+        el.exportGML(o,nodes,options)
+
+    o.write("]\n")
+
+def exportGraphml(o, nodes, edges, options):
+    import xml.dom.minidom
+    doc = xml.dom.minidom.Document()
+    root = doc.createElement(u'graphml')
+    root.setAttribute(u'xmlns',u'http://graphml.graphdrawing.org/xmlns/graphml')
+    root.setAttribute(u'xmlns:xsi',u'http://www.w3.org/2001/XMLSchema-instance')
+    root.setAttribute(u'xmlns:y',u'http://www.yworks.com/xml/graphml')
+    root.setAttribute(u'xsi:schemaLocation',u'http://graphml.graphdrawing.org/xmlns/graphml http://www.yworks.com/xml/schema/graphml/1.0/ygraphml.xsd')
+    doc.appendChild(root)        
+    key = doc.createElement(u'key')
+    key.setAttribute(u'for',u'node')    
+    key.setAttribute(u'id',u'd0')    
+    key.setAttribute(u'yfiles.type',u'nodegraphics')    
+    root.appendChild(key)
+
+    key = doc.createElement(u'key')
+    key.setAttribute(u'attr.name',u'description')    
+    key.setAttribute(u'attr.type',u'string')    
+    key.setAttribute(u'for',u'node')    
+    key.setAttribute(u'id',u'd1')    
+    root.appendChild(key)
+
+    key = doc.createElement(u'key')
+    key.setAttribute(u'for',u'edge')    
+    key.setAttribute(u'id',u'd2')    
+    key.setAttribute(u'yfiles.type',u'edgegraphics')    
+    root.appendChild(key)
+
+    key = doc.createElement(u'key')
+    key.setAttribute(u'attr.name',u'description')    
+    key.setAttribute(u'attr.type',u'string')    
+    key.setAttribute(u'for',u'edge')    
+    key.setAttribute(u'id',u'd3')    
+    root.appendChild(key)
+
+    key = doc.createElement(u'key')
+    key.setAttribute(u'for',u'node')    
+    key.setAttribute(u'id',u'd4')    
+    key.setAttribute(u'attr.name',u'url')    
+    key.setAttribute(u'attr.type',u'string')    
+    root.appendChild(key)
+
+    graph = doc.createElement(u'graph')
+    graph.setAttribute(u'edgedefault',u'directed')    
+    graph.setAttribute(u'id',u'G')    
+    graph.setAttribute(u'parse.edges',u'%d' % len(edges))   
+    graph.setAttribute(u'parse.nodes',u'%d' % len(nodes))
+    graph.setAttribute(u'parse.order', u'free')    
+    
+    for k,nod in nodes.iteritems():
+        nod.exportGraphml(doc, graph, options)
+    for el in edges:
+        el.exportGraphml(doc, graph, nodes, options)
+
+    root.appendChild(graph)
+    
+    data = doc.createElement(u'data')
+    data.setAttribute(u'key',u'd4')    
+    res = doc.createElement(u'y:Resources')
+    data.appendChild(res)    
+    root.appendChild(data)
+    
+    o.write(doc.toxml(encoding="utf-8"))
+
+def exportGDF(o, nodes, edges, options):
+    o.write("nodedef> name\n")
+    for k,nod in nodes.iteritems():
+        nod.exportGDF(o, options)
+    for el in edges:
+        el.exportGDF(o,nodes,options)
+    o.write("edgedef> node1,node2\n")
+
+def main():
+    parser = optparse.OptionParser(usage=usgmsg)
+    parser.add_option('-f', '--format',
+                      action='store', dest='format', default='Graphml',
+                      help='selects the output format (Graphml|GML|GDF) [default : %default]')
+    parser.add_option('-v', '--verbose',
+                      action='store_true', dest='verbose', default=False,
+                      help='enables messages (infos, warnings)')
+    parser.add_option('-s', '--sweep',
+                      action='store_true', dest='sweep', default=False,
+                      help='sweep nodes (remove nodes that are not connected)')
+    parser.add_option('--nn', '--no-nodes',
+                      action='store_false', dest='NodeLabels', default=True,
+                      help='do not output any node labels [Graphml]')
+    parser.add_option('--ne', '--no-edges',
+                      action='store_false', dest='EdgeLabels', default=True,
+                      help='do not output any edge labels [Graphml]')
+    parser.add_option('--nu', '--no-uml',
+                      action='store_false', dest='NodeUml', default=True,
+                      help='do not output any node methods/attributes in UML [Graphml]')
+    parser.add_option('--na', '--no-arrows',
+                      action='store_false', dest='Arrows', default=True,
+                      help='do not output any arrows [Graphml]')
+    parser.add_option('--nc', '--no-colors',
+                      action='store_false', dest='Colors', default=True,
+                      help='do not output any colors [Graphml]')
+    parser.add_option('--la', '--lump-attributes',
+                      action='store_true', dest='LumpAttributes', default=False,
+                      help='lump class attributes/methods together with the node label [Graphml]')
+    parser.add_option('--sc', '--separator-char',
+                      action='store', dest='SepChar', default='_', metavar='SEPCHAR',
+                      help='default separator char when lumping attributes/methods [default : "_"]')
+    parser.add_option('--ae', '--auto-edges',
+                      action='store_true', dest='EdgeLabelsAutoComplete', default=False,
+                      help='auto-complete edge labels')
+    parser.add_option('--ah', '--arrowhead',
+                      action='store', dest='DefaultArrowHead', default='none', metavar='TYPE',
+                      help='sets the default appearance of arrow heads for edges (normal|diamond|dot|...) [default : %default]')
+    parser.add_option('--at', '--arrowtail',
+                      action='store', dest='DefaultArrowTail', default='none', metavar='TYPE',
+                      help='sets the default appearance of arrow tails for edges (normal|diamond|dot|...) [default : %default]')
+    parser.add_option('--cn', '--color-nodes',
+                      action='store', dest='DefaultNodeColor', default='#CCCCFF', metavar='COLOR',
+                      help='default node color [default : "#CCCCFF"]')
+    parser.add_option('--ce', '--color-edges',
+                      action='store', dest='DefaultEdgeColor', default='#000000', metavar='COLOR',
+                      help='default edge color [default : "#000000"]')
+    parser.add_option('--cnt', '--color-nodes-text',
+                      action='store', dest='DefaultNodeTextColor', default='#000000', metavar='COLOR',
+                      help='default node text color for labels [default : "#000000"]')
+    parser.add_option('--cet', '--color-edges-text',
+                      action='store', dest='DefaultEdgeTextColor', default='#000000', metavar='COLOR',
+                      help='default edge text color for labels [default : "#000000"]')
+    parser.add_option('--ienc', '--input-encoding',
+                      action='store', dest='InputEncoding', default='', metavar='ENCODING',
+                      help='override encoding for input file [default : locale setting]')
+    parser.add_option('--oenc', '--output-encoding',
+                      action='store', dest='OutputEncoding', default='', metavar='ENCODING',
+                      help='override encoding for text output files [default : locale setting]')
+
+    options, args = parser.parse_args()
+    
+    if len(args) < 2:
+        usage()
+        sys.exit(1)
+
+    infile = args[0]
+    outfile = args[1]
+
+    options.DefaultNodeColor = dot.colorNameToRgb(options.DefaultNodeColor, '#CCCCFF')
+    options.DefaultEdgeColor = dot.colorNameToRgb(options.DefaultEdgeColor, '#000000')
+    options.DefaultNodeTextColor = dot.colorNameToRgb(options.DefaultNodeTextColor, '#000000')
+    options.DefaultEdgeTextColor = dot.colorNameToRgb(options.DefaultEdgeTextColor, '#000000')
+    
+    preferredEncoding = locale.getpreferredencoding()
+    if options.InputEncoding == "":
+        options.InputEncoding = preferredEncoding
+    if options.OutputEncoding == "":
+        options.OutputEncoding = preferredEncoding
+    
+    if options.verbose:
+        print "Input file: %s " % infile
+        print "Output file: %s " % outfile
+        print "Output format: %s" % options.format.lower()
+        print "Input encoding: %s" % options.InputEncoding
+        if options.format.lower() == "graphml":
+            print "Output encoding: utf-8 (fix for Graphml)"
+        else:
+            print "Output encoding: %s" % options.OutputEncoding
+
+    # Collect nodes and edges
+    nodes = {}
+    edges = []
+    default_edge = None
+    default_node = None
+    nid = 1
+    eid = 1
+    f = open(infile, 'r')
+    for l in f.readlines():
+        l = unicode(l, options.InputEncoding)
+        if l.find('->') >= 0:
+            # Process edge
+            e = dot.Edge()
+            e.initFromString(l)
+            e.id = eid
+            eid += 1
+            if default_edge:
+                e.complementAttributes(default_edge)
+            edges.append(e)
+        elif (l.find('[') > 0 and
+              l.find(']') > 0 and
+              l.find(';') > 0):
+            # Process node
+            n = dot.Node()
+            n.initFromString(l)
+            lowlabel = n.label.lower()
+            if (lowlabel != 'graph' and
+                lowlabel != 'edge' and
+                lowlabel != 'node'):
+                n.id = nid
+                nid += 1
+                if default_node:
+                    n.complementAttributes(default_node)
+                nodes[n.label] = n
+            else:
+                if lowlabel == 'edge':
+                    default_edge = n
+                elif lowlabel == 'node':
+                    default_node = n   
+        elif l.find('charset=') >=0:
+            # Pick up input encoding from DOT file
+            li = l.strip().split('=')
+            if len(li) == 2:
+                ienc = li[1].strip('"')
+                if ienc != "":
+                    options.InputEncoding = ienc
+                    if options.verbose:
+                        print "Info: Picked up input encoding '%s' from the DOT file." % ienc
+            
+    f.close()
+    
+    # Add single nodes, if required
+    for e in edges:
+        if not nodes.has_key(e.src):
+            n = dot.Node()
+            n.label = e.src
+            n.id = nid
+            nid += 1
+            nodes[e.src] = n
+        if not nodes.has_key(e.dest):
+            n = dot.Node()
+            n.label = e.dest
+            n.id = nid
+            nid += 1
+            nodes[e.dest] = n
+        nodes[e.src].referenced = True
+        nodes[e.dest].referenced = True
+
+    if options.verbose:
+        print "\nNodes: %d " % len(nodes)
+        print "Edges: %d " % len(edges)
+    
+    if options.sweep:
+        rnodes = {}
+        for key, n in nodes.iteritems():
+            if n.referenced:
+                rnodes[key] = n
+        nodes = rnodes
+        if options.verbose:
+            print "\nNodes after sweep: %d " % len(nodes)
+    
+    # Output
+    o = open(outfile, 'w')
+    format = options.format.lower()
+    if format == 'dot':
+        exportDot(o, nodes, edges, options)
+    elif format == 'graphml':
+        exportGraphml(o, nodes, edges, options)
+    elif format == 'gdf':
+        exportGDF(o, nodes, edges, options)
+    else: # GML
+        exportGML(o, nodes, edges, options)
+    o.close()
+
+    if options.verbose:
+        print "\nDone."
+
+if __name__ == '__main__':
+    main()
+