Last active
May 12, 2018 10:50
-
-
Save matthiaseisen/3278cedcd53afe62c3f3 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import graphviz as gv | |
g1 = gv.Graph(format='svg') | |
g1.node('A') | |
g1.node('B') | |
g1.edge('A', 'B') | |
print(g1.source) | |
filename = g1.render(filename='img/g1') | |
print(filename) | |
g2 = gv.Digraph(format='svg') | |
g2.node('A') | |
g2.node('B') | |
g2.edge('A', 'B') | |
g2.render('img/g2') | |
import functools | |
graph = functools.partial(gv.Graph, format='svg') | |
digraph = functools.partial(gv.Digraph, format='svg') | |
g3 = graph() | |
nodes = ['A', 'B', ('C', {})] | |
edges = [ | |
('A', 'B'), | |
('B', 'C'), | |
(('A', 'C'), {}), | |
] | |
def add_nodes(graph, nodes): | |
for n in nodes: | |
if isinstance(n, tuple): | |
graph.node(n[0], **n[1]) | |
else: | |
graph.node(n) | |
return graph | |
def add_edges(graph, edges): | |
for e in edges: | |
if isinstance(e[0], tuple): | |
graph.edge(*e[0], **e[1]) | |
else: | |
graph.edge(*e) | |
return graph | |
add_edges( | |
add_nodes(digraph(), ['A', 'B', 'C']), | |
[('A', 'B'), ('A', 'C'), ('B', 'C')] | |
).render('img/g4') | |
add_edges( | |
add_nodes(digraph(), [ | |
('A', {'label': 'Node A'}), | |
('B', {'label': 'Node B'}), | |
'C' | |
]), | |
[ | |
(('A', 'B'), {'label': 'Edge 1'}), | |
(('A', 'C'), {'label': 'Edge 2'}), | |
('B', 'C') | |
] | |
).render('img/g5') | |
g6 = add_edges( | |
add_nodes(digraph(), [ | |
('A', {'label': 'Node A'}), | |
('B', {'label': 'Node B'}), | |
'C' | |
]), | |
[ | |
(('A', 'B'), {'label': 'Edge 1'}), | |
(('A', 'C'), {'label': 'Edge 2'}), | |
('B', 'C') | |
] | |
) | |
styles = { | |
'graph': { | |
'label': 'A Fancy Graph', | |
'fontsize': '16', | |
'fontcolor': 'white', | |
'bgcolor': '#333333', | |
'rankdir': 'BT', | |
}, | |
'nodes': { | |
'fontname': 'Helvetica', | |
'shape': 'hexagon', | |
'fontcolor': 'white', | |
'color': 'white', | |
'style': 'filled', | |
'fillcolor': '#006699', | |
}, | |
'edges': { | |
'style': 'dashed', | |
'color': 'white', | |
'arrowhead': 'open', | |
'fontname': 'Courier', | |
'fontsize': '12', | |
'fontcolor': 'white', | |
} | |
} | |
def apply_styles(graph, styles): | |
graph.graph_attr.update( | |
('graph' in styles and styles['graph']) or {} | |
) | |
graph.node_attr.update( | |
('nodes' in styles and styles['nodes']) or {} | |
) | |
graph.edge_attr.update( | |
('edges' in styles and styles['edges']) or {} | |
) | |
return graph | |
g6 = apply_styles(g6, styles) | |
g6.render('img/g6') | |
g7 = add_edges( | |
add_nodes(digraph(), [ | |
('A', {'label': 'Node A'}), | |
('B', {'label': 'Node B'}), | |
'C' | |
]), | |
[ | |
(('A', 'B'), {'label': 'Edge 1'}), | |
(('A', 'C'), {'label': 'Edge 2'}), | |
('B', 'C') | |
] | |
) | |
g8 = apply_styles( | |
add_edges( | |
add_nodes(digraph(), [ | |
('D', {'label': 'Node D'}), | |
('E', {'label': 'Node E'}), | |
'F' | |
]), | |
[ | |
(('D', 'E'), {'label': 'Edge 3'}), | |
(('D', 'F'), {'label': 'Edge 4'}), | |
('E', 'F') | |
] | |
), | |
{ | |
'nodes': { | |
'shape': 'square', | |
'style': 'filled', | |
'fillcolor': '#cccccc', | |
} | |
} | |
) | |
g7.subgraph(g8) | |
g7.edge('B', 'E', color='red', weight='2') | |
g7.render('img/g7') |
Hey this is really cool. I am trying to figure out how to a couple of things.
How to deal with rendering multiple subgraphs ?
How to create a subgraph border.
I am not sure also what diffence is between a subgraph and a cluster?
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thanks for the tutorial. Did you figure out a way to stylize final states (like double circle)?
edit: found the doc for it: use
attr
method andshape
kwarg.edit2: now I wonder how to stylize initial states, namely to get an edge without predecessor to point at it?