diff --git a/fixtracks/widgets/detectionview.py b/fixtracks/widgets/detectionview.py index 46d8649..1b86459 100644 --- a/fixtracks/widgets/detectionview.py +++ b/fixtracks/widgets/detectionview.py @@ -19,12 +19,6 @@ class Detection(QGraphicsEllipseItem): def mousePressEvent(self, event): self.signals.clicked.emit(self.data(0), QPointF(event.scenePos().x(), event.scenePos().y())) - # item = self.scene().itemAt(event.scenePos(), self.views()[0].transform()) - - # if item: - # item.setSelected(True) - - # print(f"Rectangle clicked at: {event.scenePos().x()}, {event.scenePos().y()}") def hoverEnterEvent(self, event): self.signals.hover.emit(self.data(0), QPointF(event.scenePos().x(), event.scenePos().y())) @@ -53,7 +47,6 @@ class DetectionScene(QGraphicsScene): def mouseMoveEvent(self, event): if self.selection_rect is not None: - # Update the selection rectangle as the mouse moves rect = QRectF(self.start_point, event.scenePos()).normalized() self.selection_rect.setRect(rect) super().mouseMoveEvent(event) @@ -64,7 +57,6 @@ class DetectionScene(QGraphicsScene): self.removeItem(self.selection_rect) self.selection_rect = None if rect.width() > 0.0: - # Find all items that intersect with the selection rectangle selected_items = self.items(rect, Qt.IntersectsItemShape) for item in selected_items: if not isinstance(item, Detection): @@ -158,8 +150,8 @@ class DetectionView(QWidget): def main(): def items_selected(items): - print("items selected", items) - + print("items selected") + import pickle import numpy as np from IPython import embed @@ -178,7 +170,7 @@ def main(): bg_coords = np.stack(df.keypoints[(df.track != 1) & (df.track != 2)].values,).astype(np.float32)[:,0,:] bg_tracks = df.track[(df.track != 1) & (df.track != 2)].values bg_ids = df.track[(df.track != 1) & (df.track != 2)].index.values - + scnd_coords = np.stack(df.keypoints[(df.track == 2)].values,).astype(np.float32)[:,0,:] scnd_tracks = df.track[df.track == 2].values scnd_ids = df.track[(df.track == 2)].index.values diff --git a/fixtracks/widgets/tracks.py b/fixtracks/widgets/tracks.py index bd1548c..eb69993 100644 --- a/fixtracks/widgets/tracks.py +++ b/fixtracks/widgets/tracks.py @@ -472,7 +472,7 @@ class FixTracks(QWidget): self._data_combo.addItem("Please select") self._data_combo.addItems(self.fileList) self._data_combo.setCurrentIndex(0) - + def populateKeypointCombo(self, num_keypoints): self._keypointcombo.clear() for i in range(num_keypoints):