tools: Add script to subtract two bandwidth csv

This adds a script to subtract two csv with bandwidth results from
each other. It can then be passed to bw_graph.py.

2 files changed, 56 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))
diff --git a/tools/subtract.py b/tools/subtract.py
new file mode 100755
index 0000000..6929b61
--- /dev/null
+++ b/tools/subtract.py
@@ -0,0 +1,48 @@
+#!/usr/bin/env python
+
+import argparse
+import csv
+import math
+
+description = "Subtracts two CSV from each other"
+epilog = "2018 Sander Vrijders <[email protected]>"
+
+parser = argparse.ArgumentParser(description=description,
+                                 epilog=epilog)
+
+parser.add_argument('-o', '--output', type=str, required=True,
+                    help='Output CSV filename.')
+
+parser.add_argument('-f', '--first', type=str, required=True,
+                    help='First input CSV filename.')
+
+parser.add_argument('-s', '--second', type=str, required=True,
+                    help='Second input CSV filename.')
+
+args = parser.parse_args()
+
+csvfile = open(args.second, 'r')
+reader = csv.reader(csvfile, delimiter=';', quotechar='|')
+
+# Set second measurement
+bandwidth = {}
+for row in reader:
+    if (row[1], row[0]) in bandwidth:
+        bandwidth[row[1], row[0]] += int(row[2])
+    else:
+        bandwidth[row[0], row[1]] = int(row[2])
+
+# Subtract first measurement
+csvfile2 = open(args.first, 'r')
+reader2 = csv.reader(csvfile2, delimiter=';', quotechar='|')
+for row in reader2:
+    if (row[1], row[0]) in bandwidth:
+        bandwidth[row[1], row[0]] -= int(row[2])
+    else:
+        bandwidth[row[0], row[1]] -= int(row[2])
+
+
+csvfile3 = open(args.output, 'w')
+writer = csv.writer(csvfile3, delimiter=';', quotechar='|')
+for key, value in bandwidth.items():
+    writer.writerow([key[0], key[1], value])