-
Notifications
You must be signed in to change notification settings - Fork 1
/
format_for_klapaukh.py
113 lines (93 loc) · 2.14 KB
/
format_for_klapaukh.py
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# format_for_klapaukh.py
#
# Input: graphfile layoutfile scale [outfile]
# Where the graph and layout are stored as SVGs
#
# Output: An SVG with an adjacency matrix and node locations
# padded with a bunch of boilerplate for an experiment
# on force-directed layout that Klapaukh was running,
# but we are not.
import os
import sys
import networkx as nx
########
# MATH #
########
def scale_layout(layout, scale):
out = []
for pt in layout:
out.append(map(lambda x: (x+1)/2*scale, pt))
return out
###################
# STRING BUILDING #
###################
def generate_boilerplate(w=800,h=800):
out = """<?xml version="1.0" encoding="ISO-8859-1" standalone="no"?>
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 20010904//EN"
"http://www.w3.org/TR/2001/REC-SVG-2010904/DTD/svg10.dtd">
<svg xmlns="http://www.w3.org/2000/svg"
xmlns:xlink="http://www.w3.org/1999/xlink" xml:space="preserve"
width="{0}"
height="{1}"
viewBox = "0 0 {0} {1}"
zoomAndPan="disable">
<!--
elapsed: 1
filename: test.xml
width: {0}
height: {1}
iterations: 1
forcemode: 1
ke: 0
kh: 0
kl: 0
kw: 0
mass: 0
time: 0
coefficientOfRestitution: 0
mus: 0
muk: 0
kg: 0
wellMass: 0
edgeCharge: 0
finalKineticEnergy: 0
nodeWidth: 0
nodeHeight: 0
nodeCharge: 0
-
Start Graph:
""".format(w,h)
return out
def format_layout_data_networkx(G):
nodes = sorted(nx.nodes(G))
edges = nx.edges(G)
all_pos = nx.get_node_attributes(G, 'pos')
# print(all_pos)
out = str(len(nodes)) + "\n"
for i in range(0, len(nodes)):
source = nodes[i]
curr_node_x = all_pos[source].split(",")[0]
curr_node_y = all_pos[source].split(",")[1]
line = str(curr_node_x) + " " + str(curr_node_y)
for j in range(0, len(nodes)):
target = nodes[j]
if i == j:
line += " 0"
continue
if ((source,target) in edges) or ((target,source) in edges):
line += " 1"
else:
line += " 0"
out += line + "\n"
return out
def build_svg_networkx(G, scale):
svg = generate_boilerplate(scale, scale)
svg += format_layout_data_networkx(G)
svg += "</svg>"
return svg
def write_svg(filename, svg):
if filename != None:
with open(filename, 'w') as f:
f.write(svg)
else:
print(svg)