First commit of MAMMULT code

1 files changed, 43 insertions, 0 deletions

diff --git a/structure/metrics/edge_overlap.py b/structure/metrics/edge_overlap.py
new file mode 100644
index 0000000..1b27f55
--- /dev/null
+++ b/structure/metrics/edge_overlap.py
@@ -0,0 +1,43 @@
+####
+##
+## Compute the edge overlap of each edge of the multiplex, i.e. the
+## number of times that each edge appears in the multiplex
+##
+##
+
+import sys
+
+
+if len(sys.argv) < 2:
+    print "Usage: %s <layer1> [<layer2>...]" % sys.argv[0]
+    sys.exit(1)
+
+max_N = -1
+
+all_edges = {}
+
+layer_ID = -1
+
+for layer in sys.argv[1:]:
+    layer_ID += 1
+    with open(layer, "r") as lines:
+        for l in lines:
+            if l[0] == "#":
+                continue
+            s, d = [int(x) for x in l.strip(" \n").split(" ")[:2]]
+            if s > d:
+                tmp = s
+                s = d
+                d = tmp
+            if (s,d) in all_edges:
+                all_edges[(s,d)].append(layer_ID)
+            else:
+                all_edges[(s,d)] = [layer_ID]
+                
+K = len(all_edges.keys())
+
+for k in all_edges.keys():
+    val = len(set(all_edges[(k)]))
+    s,d = k
+    print s, d, val
+