From c231b52876f42b759ec102ec67a10ba0ff529cba Mon Sep 17 00:00:00 2001 From: Jan Grewe Date: Fri, 21 Feb 2025 10:59:57 +0100 Subject: [PATCH] [trackingdata] more logging, cleanup --- fixtracks/utils/trackingdata.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/fixtracks/utils/trackingdata.py b/fixtracks/utils/trackingdata.py index 7592f1f..a41c120 100644 --- a/fixtracks/utils/trackingdata.py +++ b/fixtracks/utils/trackingdata.py @@ -61,7 +61,7 @@ class TrackingData(QObject): self._indices = np.where((self._data[col] >= self._start) & (self._data[col] < self._stop))[0] def selectedData(self, col): - return self._data[col][self._indices] + return self[col][self._indices] def setUserSelection(self, ids): """ @@ -84,14 +84,20 @@ class TrackingData(QObject): userFlag : bool Should the "userlabeled" state of the detections be set to True or False? """ - self._data["track"][self._user_selections] = track_id + self["track"][self._user_selections] = track_id self.setAssignmentStatus(userFlag) def setAssignmentStatus(self, isTrue: bool): - self._data["userlabeled"][self._user_selections] = isTrue + logging.debug("TrackingData:Re-setting assignment status of user selected data to %s", str(isTrue)) + self["userlabeled"][self._user_selections] = isTrue def revertAssignmentStatus(self): - self._data["userlabeled"][:] = False + logging.debug("TrackingData:Un-setting assignment status of all data!") + self["userlabeled"][:] = False + + def revertTrackAssignments(self): + logging.debug("TrackingData: Reverting all track assignments!") + self["track"][:] = -1 def deleteDetections(self): # from IPython import embed @@ -116,7 +122,7 @@ class TrackingData(QObject): None """ if len(tracks) != self.numDetections: - logging.error("DataController: Size of passed tracks does not match data!") + logging.error("Trackingdata: Size of passed tracks does not match data!") return self._data["track"] = tracks