From 29c8d0c127e40b78873ca489338129cef33bcfe1 Mon Sep 17 00:00:00 2001
From: Jean Sirmai <jean@a-lec.org>
Date: Sat, 8 Jan 2022 19:41:35 +0100
Subject: [PATCH] WIP: plus simple, plus propre.  ( ;- ))

---
 src/__pycache__/model.cpython-39.pyc | Bin 19517 -> 19477 bytes
 src/model.py                         |   8 +++---
 src/rules_tree.py                    |  37 ++++++++++++---------------
 3 files changed, 20 insertions(+), 25 deletions(-)

diff --git a/src/__pycache__/model.cpython-39.pyc b/src/__pycache__/model.cpython-39.pyc
index 330ff4efc6d5a4200b9104cdf7db72f945a1c0e0..d4dd9304d428f0f87453132ebc8152edfb25ed43 100644
GIT binary patch
delta 229
zcmdlxgK_E%M&3kTUM>b8_<8nbBKt;OHDf*Q^wg60;?$h9_{<bPMh1prCXhU1m8f%m
zUP@+3W`3R?l8VXyWuzuI7)nl_XRNBtU5wBgh+Qj8#pFh2k;xy76ejDK$WMMC&Z=CH
mSd^Mq0<<#(NuxN5os(-#k~DZhCcqVkV^<7RvsutoM;rjOh(<~P

delta 241
zcmbO_gK_T+M&3kTUM>b8a7w<JD7}$a&DcO7J+&l0FFz$UzBn}}Ej}~FkCB0)m<gna
zu}aE0KQAS-Br`ux52kE#q;Z0QKrx!8K<t`e$|m18jxZ8}84+KQSd^Mq0<<dxNt+B7
ZyW&lfjD%o@Ahd;J*9KF#`K5`DH~`b)OaTA@

diff --git a/src/model.py b/src/model.py
index 454e984..e3986b7 100644
--- a/src/model.py
+++ b/src/model.py
@@ -331,16 +331,16 @@ class Condition:
     def set_weight(self, weight):
         self.node.attrib["weight"] = str(weight)
 
-    def get_node_self_id(self):
+    def get_self_id(self):
         return int(self.node.get("node_id"))
 
-    def set_node_self_id(self, node_id):
+    def set_self_id(self, node_id):
         self.node.attrib["node_id"] = str(node_id)
 
-    def get_node_parent_id(self):
+    def get_parent_id(self):
         return int(self.node.get("parent"))
 
-    def set_node_parent_id(self, parent):
+    def set_parent_id(self, parent):
         self.node.attrib["parent"] = str(parent)
 
 
diff --git a/src/rules_tree.py b/src/rules_tree.py
index 5f998db..cb231e7 100644
--- a/src/rules_tree.py
+++ b/src/rules_tree.py
@@ -8,44 +8,46 @@ import model
 
 
 class CondTreeNode(NodeMixin):
-    def __init__(self, id, x, site, weight, parent_node_id):
+    def __init__(self, id, parent_id, x, site, weight):
         super(CondTreeNode, self).__init__()
         self.id = id
-        self.parent_node_id = parent_node_id
+        self.parent_id = parent_id
         self.parent = None
         self.x = x   # TODO  self.y = y,z...
         self.site = site
         self.weight = weight
 
     def add_parent_node(self, parent_node):
-        print('orphan node ', self.id, ' is adopted by node ', parent_node.id)
+        # print('orphan node ', self.id, ' is adopted by node ', parent_node.id)
         self.parent = parent_node
 
 
 class TransTreeNode(NodeMixin):
-    def __init__(self, parent_node_id, id):
+    def __init__(self, id, parent_id):
         super(TransTreeNode, self).__init__()
-        self.parent_node_id = parent_node_id
+        self.parent_id = parent_id
         self.parent = None
         self.id = id
 
     def add_parent_leaf_node(self, parent_node):
-        #  print('orphan transition is adopted by leaf node', parent_node.id)
+        # print('orphan transition is adopted by leaf node', parent_node.id)
         self.parent = parent_node
 
 
 conditions_set = set()
 transitions_set = set()
-root = 0
+root = None
 
 for i in range(0, len(model.testmodel.get_conditions().get_all())):
     condition = model.testmodel.get_conditions().get_one(i)
-    conditions_set.add(CondTreeNode(
-                       condition.get_node_self_id(),
-                       condition.get_node_parent_id(),
+    tmp = CondTreeNode(condition.get_self_id(),
+                       condition.get_parent_id(),
                        condition.get_coord("x"),
                        condition.get_site(),
-                       condition.get_weight()))
+                       condition.get_weight())
+    conditions_set.add(tmp)
+    if (tmp.parent_id == 0):
+        root = tmp
 
 for i in range(0, len(model.testmodel.get_transitions().get_all())):
     transition = model.testmodel.get_transitions().get_one(i)
@@ -53,22 +55,15 @@ for i in range(0, len(model.testmodel.get_transitions().get_all())):
                         transition.get_id(),
                         transition.get_parent()))
 
-#                      0          1          2          3      4
-#        <condition site="1" weight="0" node_id="3" parent="2" x="2"/>
-
-for i in conditions_set:
-    if (i.parent_node_id == 0):
-        root = i
-
 for i in conditions_set:
     for j in conditions_set:
-        if ((i != j) & (i.parent_node_id == j.id)):
+        if ((i != j) & (i.parent_id == j.id)):
             i.add_parent_node(j)
 
 for condition_node in LevelOrderIter(root):
     if condition_node.is_leaf:
         for transition_node in transitions_set:
-            if (str(condition_node.id) == transition_node.parent):
+            if (str(condition_node.id) == transition_node.parent_id):
                 transition_node.add_parent_leaf_node(condition_node)
 
 for pre, fill, node in RenderTree(root):
@@ -76,7 +71,7 @@ for pre, fill, node in RenderTree(root):
     if (type(node) == CondTreeNode):
         print(treestr.ljust(16), node.weight, ' arrow(s) at [site', node.site,
               'in cell', node.x, end='')
-        if (not node.depth == 2):
+        if (not node.depth == 2):  # tree.depth()
             print(end=']\n')
         else:
             print(']   >  ', node.children[0].id, end='\n')