mirror of
https://github.com/FliegendeWurst/hypergraph-drawing.git
synced 2024-09-07 21:21:48 +00:00
168 lines
3.7 KiB
Python
Executable File
168 lines
3.7 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
import sys
|
|
import itertools
|
|
import pydot # tested with 1.4.2
|
|
|
|
if len(sys.argv) != 2:
|
|
print("Usage: dot_hypergraph_to_asymptote.py <dot file>")
|
|
sys.exit(1)
|
|
|
|
# output size
|
|
width = None # 400
|
|
height = 639
|
|
|
|
# size of node and edge labels
|
|
fontsize = "20pt"
|
|
|
|
# specify minimum text width
|
|
# (useful if your labels contain wide unicode characters)
|
|
text_width_base = 35 # 40
|
|
text_width_factor = 5.5 # 9
|
|
|
|
# draw edges reversed
|
|
reverse_edges = True
|
|
|
|
# output \begin{asy} ... \end{asy}
|
|
output_latex = False
|
|
|
|
graphs = pydot.graph_from_dot_file(sys.argv[1])
|
|
graph = graphs[0]
|
|
|
|
nodes = dict()
|
|
node_list = []
|
|
|
|
for node in graph.get_nodes():
|
|
pos = node.get_pos()
|
|
if pos is None:
|
|
continue
|
|
pos = pos[1:-1]
|
|
nodes[node.get_name()] = tuple(map(float, pos.split(",")))
|
|
node_list.append(node.get_name())
|
|
|
|
edges = dict()
|
|
|
|
for edge in graph.get_edges():
|
|
label = edge.get_label()
|
|
if label is None:
|
|
continue
|
|
label = label.replace('"', "")
|
|
src = edge.get_source()
|
|
dest = edge.get_destination()
|
|
if label not in edges:
|
|
edges[label] = [(src, dest)]
|
|
else:
|
|
edges[label].append((src, dest))
|
|
|
|
hyperedges = []
|
|
|
|
for label in edges:
|
|
xmin = min(itertools.chain(*map(lambda x: [nodes[x[0]][0], nodes[x[1]][0]], edges[label])))
|
|
xmax = max(itertools.chain(*map(lambda x: [nodes[x[0]][0], nodes[x[1]][0]], edges[label])))
|
|
ymin = min(map(lambda x: nodes[x[0]][1], edges[label]))
|
|
ymax = max(map(lambda x: nodes[x[1]][1], edges[label]))
|
|
x = (xmax + xmin) / 2
|
|
y = (ymax + ymin) / 2
|
|
#print(label, xmin, xmax, ymin, ymax, x, y)
|
|
hyperedges.append(((x, y), label))
|
|
#input("")
|
|
|
|
if output_latex:
|
|
print("\\begin{asy}")
|
|
|
|
# code based on https://tex.stackexchange.com/a/108099/185782 (g.kov)
|
|
|
|
if width is None:
|
|
width = 0
|
|
if height is None:
|
|
height = 0
|
|
print("size(", width, ",", height, ");")
|
|
print("import flowchart;")
|
|
if fontsize is not None:
|
|
print("defaultpen(fontsize(", fontsize, "));")
|
|
|
|
print("pair[]pv={")
|
|
|
|
for node in node_list:
|
|
print(nodes[node], ",")
|
|
|
|
print("};")
|
|
|
|
print("string[]pv_labels={")
|
|
|
|
for name in node_list:
|
|
if not name.startswith('"'):
|
|
name = '"' + name
|
|
if not name[-1] == '"':
|
|
name = name + '"'
|
|
print(name, ',')
|
|
|
|
print("};")
|
|
|
|
print("real[]pv_size={")
|
|
|
|
for node in node_list:
|
|
print(text_width_base + len(node[1:-1])*text_width_factor, ',')
|
|
|
|
print("};")
|
|
|
|
print("pair[][] pe={")
|
|
|
|
for x in hyperedges:
|
|
print("{", x[0], ",E},")
|
|
|
|
print("};")
|
|
|
|
print("string[] pe_labels={")
|
|
|
|
for x in hyperedges:
|
|
print('"', x[1].replace("_", "\\_"), '",')
|
|
|
|
print("};")
|
|
|
|
print("""
|
|
pen nodeFill=white;
|
|
pen nodeLine=darkblue+1.2pt+opacity(0.5);
|
|
|
|
block[] v=new block[pv.length];
|
|
block[] e=new block[pe.length];
|
|
|
|
|
|
for(int i=0;i<pv.length;++i){
|
|
v[i]=rectangle(pv_labels[i],pv[i],nodeFill,nodeLine,minwidth=pv_size[i]);
|
|
draw(v[i]);
|
|
}
|
|
|
|
string s;
|
|
for(int i=0;i<pe.length;++i){
|
|
s=pe_labels[i];
|
|
e[i]=circle(s,pe[i][0]);
|
|
label(s,pe[i][0],pe[i][1]);
|
|
dot(pe[i][0]);
|
|
}
|
|
|
|
add(
|
|
new void(picture pic, transform t) {
|
|
blockconnector operator --=blockconnector(pic,t);
|
|
real tg;
|
|
pen linePen=darkblue+1.2pt;
|
|
currentpen=linePen;
|
|
arrowfactor=4;
|
|
""")
|
|
|
|
for n, edge in enumerate(hyperedges):
|
|
label = edge[1]
|
|
for e in edges[label]:
|
|
i1 = node_list.index(e[0])
|
|
i2 = node_list.index(e[1])
|
|
if reverse_edges:
|
|
print("tg=90;")
|
|
print("draw(pic,v[%d].top(t){dir(90)} .. {dir(tg)}t*e[%d].center{dir(tg)} .. v[%d].bottom(t){dir(90)}, Arrow(HookHead));" % (i2, n, i1))
|
|
else:
|
|
print("tg=270;")
|
|
print("draw(pic,v[%d].bottom(t){dir(270)} .. {dir(tg)}t*e[%d].center{dir(tg)} .. v[%d].top(t){dir(270)},Arrow(HookHead));" % (i1, n, i2))
|
|
|
|
print("});")
|
|
if output_latex:
|
|
print("\\end{asy}")
|