Simulation_dataset: added nhx_of_convergence_tree

lib/simulation_pipeline.ml

@@ -15,6 +15,27 @@ type tree =
       branch_length2 : float ;
     }
 
+let nhx_of_convergence_tree tree =
+  let f = fun%workflow dest ->
+    let open Phylogenetics in
+    let open Codepitk in
+    let tree =
+      [%eval tree]
+      |> Tree.map
+        ~node:(fun () -> { Newick_ast.name = None })
+        ~leaf:(fun s  -> { Newick_ast.name = Some s })
+        ~branch:Convergence_tree.(fun bi ->
+            let tags = match bi.condition with
+              | `Ancestral -> ["Condition", "0"]
+              | `Convergent -> ["Condition", "1" ; "Transition", "1"]
+            in
+            { Newick_ast.length = Some bi.length ; tags }
+          )
+    in
+    Newick.to_file (Tree tree) dest
+  in
+  Workflow.path_plugin ~descr:"nhx_of_convergence_tree" f
+
 module type S = sig
   type query
 

lib/simulation_pipeline.mli

@@ -9,6 +9,10 @@ type tree =
       branch_length2 : float;
     }
 
+val nhx_of_convergence_tree :
+  Codepitk.Convergence_tree.t workflow ->
+  nhx file
+
 module type S = sig
   type query