local cell density

Kapoorlabs-CAPED · Apr 27, 2024 · 3b6a736 · 3b6a736
1 parent b4a9668
commit 3b6a736
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/napatrackmater/Trackmate.py b/src/napatrackmater/Trackmate.py
@@ -477,7 +477,7 @@ def _create_generations(self, all_source_ids: list):
         root_leaf = []
         root_root = []
         root_splits = []
-        root_leaf_source = []
+
 
         # Get the root id
         for source_id in all_source_ids:
@@ -496,9 +496,10 @@ def _create_generations(self, all_source_ids: list):
                 root_leaf.append(target_target_id[0])
 
 
-        #if len(list(self.oneat_dividing_tracks.keys())) > 1:
-        #   for cell_id in list(self.oneat_dividing_tracks.keys()):
-        #          root_splits.append(cell_id) 
+        if len(list(self.oneat_dividing_tracks.keys())) > 1:
+           for cell_id in list(self.oneat_dividing_tracks.keys()):
+                  if cell_id in all_source_ids:
+                      root_splits.append(cell_id) 
 
 
         return root_root, root_splits, root_leaf