File: graph_util.py

package info (click to toggle)
gamera 1:3.4.3-1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye, buster, sid
  • size: 15,912 kB
  • sloc: xml: 122,324; cpp: 50,730; python: 35,044; ansic: 258; makefile: 114; sh: 101
file content (65 lines) | stat: -rw-r--r-- 2,361 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# -*- mode: python; indent-tabs-mode: nil; tab-width: 3 -*-
# vim: set tabstop=3 shiftwidth=3 expandtab:
#
# Copyright (C) 2001-2005 Ichiro Fujinaga, Michael Droettboom,
#                          and Karl MacMillan
#
# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#

def graphviz_output(G, filename, string_function=str):
   """**graphviz_output** (*graph*, *filename*, *string_function* = str)

Writes a graph in the format used by the ``dot`` tool in the graphviz_ package for
graph visualisation.

.. _graphviz: http://www.graphviz.org/

*graph*
   A Gamera ``Graph`` object.

*filename*
   Filename to output to.

*string_function*
   A function to convert node values to strings.  By default this will use Python's
   standard ``str`` function.
"""
   fd = open(filename, 'w')
   if G.is_directed():
      fd.write("digraph G {\n")
      for node in G.get_nodes():
         fd.write('   "%s";\n' % 
                  string_function(node()))
         for edge in node.edges:
            fd.write('   "%s" -> "%s"' % (
               string_function(node()), string_function(edge.to_node())))
            if edge.label is not None:
               fd.write(' [ label = %f ]' % (edge.label))
            fd.write(';\n')
      fd.write("}\n")
   else:
      fd.write("graph G {\n")
      for node in G.get_nodes():
         fd.write('   "%s";\n' % 
                  string_function(node()))
      for edge in G.get_edges():
         fd.write('   "%s" -- "%s"' % (
            string_function(edge.from_node()), string_function(edge.to_node())))
         if edge.label is not None:
            fd.write(' [ label = %f ]' % (edge.label))
         fd.write(";\n")   
      fd.write("}\n")
   fd.close()