Started work en geomdag

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 *
+from geomtree.geomtree import *
\ No newline at end of file
+from geomtree.geomdag import *
\ No newline at end of file

src/geomtree/geomdag.py

+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)