Speed up points layer inititalization

funkelab · Nov 4, 2024 · 7147eb7 · 7147eb7
1 parent 15e634f
commit 7147eb7
Showing 1 changed file with 2 additions and 14 deletions.
diff --git a/src/motile_plugin/data_views/views/layers/track_points.py b/src/motile_plugin/data_views/views/layers/track_points.py
@@ -28,13 +28,7 @@ def __init__(
     ):
         self.tracks_viewer = tracks_viewer
         self.nodes = list(tracks_viewer.tracks.graph.nodes)
-        self.node_index_dict = dict(
-            zip(
-                self.nodes,
-                [self.nodes.index(node) for node in self.nodes],
-                strict=False,
-            )
-        )
+        self.node_index_dict = {node: idx for idx, node in enumerate(self.nodes)}
 
         points = self.tracks_viewer.tracks.get_positions(self.nodes, incl_time=True)
         track_ids = [
@@ -104,13 +98,7 @@ def _refresh(self):
         )  # do not listen to new events until updates are complete
         self.nodes = list(self.tracks_viewer.tracks.graph.nodes)
 
-        self.node_index_dict = dict(
-            zip(
-                self.nodes,
-                [self.nodes.index(node) for node in self.nodes],
-                strict=False,
-            )
-        )
+        self.node_index_dict = {node: idx for idx, node in enumerate(self.nodes)}
 
         track_ids = [
             self.tracks_viewer.tracks.graph.nodes[node][NodeAttr.TRACK_ID.value]