From f9d909e9dd096896f41d55c0bc5c82e07955202c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Radek=20Hu=C5=A1ek?= <husek@iuuk.mff.cuni.cz>
Date: Thu, 29 Jul 2021 21:41:06 +0200
Subject: [PATCH] voltage_test_snarks.py
---
voltage_test_snarks.py | 58 ++++++++++++++++++++++++++++++++++++++++++
1 file changed, 58 insertions(+)
create mode 100755 voltage_test_snarks.py
diff --git a/voltage_test_snarks.py b/voltage_test_snarks.py
new file mode 100755
index 0000000..ae5c0cb
--- /dev/null
+++ b/voltage_test_snarks.py
@@ -0,0 +1,58 @@
+#!/usr/bin/python
+
+import os, sys
+import json
+import re
+
+from parmap import parmap
+
+def lists_to_tuples(x):
+ if isinstance(x, (list, tuple)):
+ return tuple( lists_to_tuples(e) for e in x )
+ if isinstance(x, dict):
+ return { k: lists_to_tuples(v) for k, v in x.items() }
+ return x
+
+L = set()
+
+for line in sys.stdin:
+ L.add(lists_to_tuples(eval(line))); continue
+
+ j = json.loads(line.rstrip())
+ n = int(re.match("Z_([0-9]+)", j["template_group"])[1])
+
+ a = j["assignment"]
+ for i in range(len(a)):
+ if a[i] >= n//2: a[i] -= n
+
+ s = sum(abs(x) for x in a)
+
+ L.add(lists_to_tuples((2*s, j["template_graph"], a)))
+
+L = list(enumerate(sorted(L)))
+
+nproc = int(os.getenv("P", "0"))
+
+def worker(T):
+# return json.dumps(T)
+
+ from graph_tools.all import EdgeColoring, voltage_graph_zn_to_sequence
+ from graph_tools.utils import stabilize_period
+ i, (s, t, a) = T
+ gs = voltage_graph_zn_to_sequence(t, a)
+ v = gs.graph(s).eval(EdgeColoring(3))
+ yield "%i G> %i %i" % (i, s, v)
+ X = gs.stabilize(EdgeColoring(3))
+ yield "%i S>" % (i,)
+ p, l = stabilize_period(X, shift=1)
+ yield json.dumps({
+ "id": i,
+ "template": t,
+ "assignment": a,
+ "repeats_from": p,
+ "has_three_coloring": l
+ })
+
+for x in parmap(worker, L, nproc or None, multimap=True, in_order=False, out_chunksize=1):
+ print(x)
+
--
GitLab