diff options
Diffstat (limited to 'tools/bw_graph.py')
| -rwxr-xr-x | tools/bw_graph.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/tools/bw_graph.py b/tools/bw_graph.py index ab772d4..d283713 100755 --- a/tools/bw_graph.py +++ b/tools/bw_graph.py @@ -26,6 +26,9 @@ def get_color(value, max): else: val = math.floor(multiplier - 1) + if value < 0: + val = math.floor(multiplier - 1) + color = color_range[val] return color.get_hex() @@ -34,7 +37,7 @@ def get_penwidth(value, max): if max == 0: return "1.0" - r = (4.0 * float(value) / (0.5 * max)) + 1 + r = (4.0 * float(abs(value)) / (0.5 * max)) + 1 return str(r) description = "CSV to bandwidth graph script" @@ -63,10 +66,8 @@ bandwidth = {} nodes = set() for row in reader: nodes.add(row[0]) - if (row[1], row[0]) in bandwidth: - bandwidth[row[1], row[0]] += int(row[2]) - else: - bandwidth[row[0], row[1]] = int(row[2]) + nodes.add(row[1]) + bandwidth[row[0], row[1]] = int(row[2]) node_bandwidth = {} for key, value in bandwidth.items(): @@ -103,7 +104,8 @@ for key, value in bandwidth.items(): pw = get_penwidth(value, max_b) edge = pydot.Edge(key[0], key[1], color = color, - penwidth=pw) + penwidth=pw, + label=round(value * 8 / 1000000, 2)) gvizg.add_edge(edge) except Exception as e: print('Failed to create pydot Edge: ' + str(e)) |