add function to get path to root length

seung-lab · Nov 2, 2017 · 903fedf · 903fedf
1 parent 8af50c0
commit 903fedf
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/src/BranchNets.jl b/src/BranchNets.jl
@@ -294,6 +294,22 @@ function get_edge_list( self::BranchNet )
     edgeList 
 end 
 
+function get_path_to_root_length(self::BranchNet, branchIndex::Integer)
+    path2RootLength = 0.0
+    branchPathLengthList = get_branch_path_length_list( self )
+    while true 
+        path2RootLength += branchPathLengthList[ branchIndex ]
+        parentBranchIndex = get_parent_branch_index(self, branchIndex )
+        if parentBranchIndex < 1 
+            # root branch do not have parent 
+            break 
+        else
+            branchIndex = parentBranchIndex 
+        end 
+    end
+    path2RootLength 
+end 
+
 """
     get_branch_path_length_list(self::BranchNet)
 get euclidean path length of each branch 

diff --git a/test/BranchNets.jl b/test/BranchNets.jl
@@ -40,7 +40,10 @@ end
 
     println("remove subtree in soma...")
     @time newBranchNet = BranchNets.remove_subtree_in_soma(branchNet)
-
+
+    println("get path to root length ...")
+    @time path2RootLength = BranchNets.get_path_to_root_length( branchNet, 5 )
+
     println("sholl analysis ...")
     @time shollNumList = BranchNets.get_sholl_number_list(branchNet, 10000 )