From b43ba63596b8ef17c193c824d2befbd685eee90c Mon Sep 17 00:00:00 2001
From: Florian Ingels <florian.ingels@inria.fr>
Date: Thu, 25 Jun 2020 16:46:39 +0200
Subject: [PATCH] Started work en geomdag
---
src/geomtree/__init__.py | 3 ++-
src/geomtree/geomdag.py | 12 ++++++++++++
2 files changed, 14 insertions(+), 1 deletion(-)
create mode 100644 src/geomtree/geomdag.py
diff --git a/src/geomtree/__init__.py b/src/geomtree/__init__.py
index 50c4a56..61e80ce 100644
--- a/src/geomtree/__init__.py
+++ b/src/geomtree/__init__.py
@@ -12,4 +12,5 @@ __version__ = version.__version__
from geomtree.affine_transform import *
from geomtree.point_cloud import *
-from geomtree.geomtree import *
\ No newline at end of file
+from geomtree.geomtree import *
+from geomtree.geomdag import *
\ No newline at end of file
diff --git a/src/geomtree/geomdag.py b/src/geomtree/geomdag.py
new file mode 100644
index 0000000..a0e8fbc
--- /dev/null
+++ b/src/geomtree/geomdag.py
@@ -0,0 +1,12 @@
+def assign_ids_to_dag(dag, t, tree_type):
+ # a deplacer dans treex ?
+ node = t.get_attribute(tree_type + '_equivalence_class')
+ attr = dag.get_attribute(node)
+ if "tree_ids" in attr.keys():
+ tree_id = attr['tree_ids']
+ else:
+ tree_id = []
+ tree_id.append(t.my_id)
+ dag.add_attribute_to_node("tree_ids", tree_id, node)
+ for child in t.my_children:
+ assign_ids_to_dag(dag, child, tree_type)
--
GitLab