From 2737fed192091465a85d23177a975912dc7c8091 Mon Sep 17 00:00:00 2001
From: Jan Grewe <jan.grewe@g-node.org>
Date: Tue, 18 Feb 2025 09:18:32 +0100
Subject: [PATCH] [trackingdata] add column to mark user-labeled detections

---
 fixtracks/utils/trackingdata.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/fixtracks/utils/trackingdata.py b/fixtracks/utils/trackingdata.py
index 3d0ca19..98a732a 100644
--- a/fixtracks/utils/trackingdata.py
+++ b/fixtracks/utils/trackingdata.py
@@ -19,6 +19,7 @@ class TrackingData(QObject):
     def setData(self, datadict):
         assert isinstance(datadict, dict)
         self._data = datadict
+        self._data["userlabeled"] = np.zeros_like(self["frame"], dtype=bool)
         self._columns = [k for k in self._data.keys()]
 
     @property
@@ -81,6 +82,7 @@ class TrackingData(QObject):
             The new track id for the user-selected detections
         """
         self._data["track"][self._user_selections] = track_id
+        self._data["userlabeled"][self._user_selections] = True
 
     def assignTracks(self, tracks):
         """assignTracks _summary_