subtree(users/wpcarro): docking briefcase at '24f5a642'
git-subtree-dir: users/wpcarro git-subtree-mainline:464bbcb15cgit-subtree-split:24f5a642afChange-Id: I6105b3762b79126b3488359c95978cadb3efa789
This commit is contained in:
commit
019f8fd211
766 changed files with 175420 additions and 0 deletions
|
|
@ -0,0 +1,39 @@
|
|||
from graphviz import Digraph
|
||||
from collections import deque
|
||||
from fixtures import weighted_graph
|
||||
|
||||
# There are three ways to model a graph:
|
||||
# 1. Edge list: [(Vertex, Vertex)]
|
||||
# 2. Neighbors table: Map(Vertex, [Vertex])
|
||||
# 3. Adjacency matrix: [[Boolean]]
|
||||
#
|
||||
# The following graph is a neighbors table.
|
||||
|
||||
|
||||
# to_graphviz :: Vertex -> Map(Vertex, [(Vertex, Weight)]) -> String
|
||||
def to_graphviz(start, g):
|
||||
"""Compiles the graph into GraphViz."""
|
||||
d = Digraph()
|
||||
q = deque()
|
||||
seen = set()
|
||||
|
||||
q.append(start)
|
||||
|
||||
while q:
|
||||
v = q.popleft()
|
||||
if v in seen:
|
||||
continue
|
||||
d.node(v, label=v)
|
||||
|
||||
for w, x in g[v]:
|
||||
d.edge(v, x, label=str(w))
|
||||
q.append(x)
|
||||
seen.add(v)
|
||||
|
||||
return d.source
|
||||
|
||||
|
||||
with open('/tmp/test.gv', 'w') as f:
|
||||
src = to_graphviz('a', weighted_graph)
|
||||
f.write(src)
|
||||
print('/tmp/test.gv created!')
|
||||
Loading…
Add table
Add a link
Reference in a new issue