From 53d83244c1bf36af86e90ce5fe758a369f73563e Mon Sep 17 00:00:00 2001
From: "serena.spinoso" <serena.spinoso@polito.it>
Date: Sat, 25 Feb 2017 12:00:55 +0100
Subject: Add verigraph code base

JIRA: PARSER-111

Change-Id: Ie76e14fabbb6c388ebc89d9a15dd3021b25fa892
Signed-off-by: serena.spinoso <serena.spinoso@polito.it>
---
 .../j-verigraph-generator/routing_generator.py     | 80 ++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 verigraph/service/src/tests/j-verigraph-generator/routing_generator.py

(limited to 'verigraph/service/src/tests/j-verigraph-generator/routing_generator.py')

diff --git a/verigraph/service/src/tests/j-verigraph-generator/routing_generator.py b/verigraph/service/src/tests/j-verigraph-generator/routing_generator.py
new file mode 100644
index 0000000..c8956f2
--- /dev/null
+++ b/verigraph/service/src/tests/j-verigraph-generator/routing_generator.py
@@ -0,0 +1,80 @@
+#!/usr/bin/python
+
+##############################################################################
+# Copyright (c) 2017 Politecnico di Torino and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+from pprint import pprint
+import sys, getopt
+import os
+from utility import *
+
+# used by test_class_generator
+def generate_routing_from_chain(chain):
+    routing = {}
+    routing["routing_table"] = {}
+
+    chain = chain["nodes"]
+    for i in range(0, len(chain)):
+        routing["routing_table"][chain[i]["name"]] = {}
+        for j in range(i-1, -1, -1):
+            routing["routing_table"][chain[i]["name"]][chain[j]["address"]] = chain[i-1]["name"]
+        for k in range (i+1, len(chain)):
+            routing["routing_table"][chain[i]["name"]][chain[k]["address"]] = chain[i+1]["name"]
+    pprint(routing)
+    return routing
+
+def generate_routing_from_chains_file(chains_file, chain_number):
+    routing = {}
+    routing["routing_table"] = {}
+
+    chains = convert_unicode_to_ascii(parse_json_file(chains_file))
+    chain = None
+    for chn in chains["chains"]:
+         if chn["id"] == chain_number:
+             chain = chn["nodes"]
+             break
+    if chain == None:
+         return routing
+
+    for i in range(0, len(chain)):
+        routing["routing_table"][chain[i]["name"]] = {}
+        for j in range(i-1, -1, -1):
+            routing["routing_table"][chain[i]["name"]][chain[j]["address"]] = chain[i-1]["name"]
+        for k in range (i+1, len(chain)):
+            routing["routing_table"][chain[i]["name"]][chain[k]["address"]] = chain[i+1]["name"]
+    pprint(routing)
+    return routing
+
+def main(argv):
+    if len(argv) < 4:
+        print 'routing_generator.py -c <chains_file> -n <chain_number>'
+        sys.exit(2)
+    chains_file = ""
+    chain_number = ""
+    try:
+        opts, args = getopt.getopt(argv,"hc:n:",["chains=","id="])
+    except getopt.GetoptError:
+        print 'routing_generator.py -c <chains_file> -n <chain_number>'
+        sys.exit(2)
+    for opt, arg in opts:
+        if opt == '-h':
+            print 'routing_generator.py -c <chains_file> -n <chain_number>'
+            sys.exit()
+        elif opt in ("-c", "--chains"):
+            chains_file = arg
+        elif opt in ("-n", "--id"):
+            chain_number = arg
+
+    print "Chains file is " + chains_file
+    print "Chain id is " + chain_number
+
+    return generate_routing_from_chains_file(chains_file, chain_number)
+
+if __name__ == '__main__':
+    main(sys.argv[1:])
\ No newline at end of file
-- 
cgit 1.2.3-korg