From 20b2915b6bd2dc26daff4f8002f4472eac5478b8 Mon Sep 17 00:00:00 2001
From: Jan Grewe <jan.grewe@g-node.org>
Date: Thu, 20 Feb 2025 18:00:29 +0100
Subject: [PATCH] cleanup

---
 fixtracks/fixtracks.py          | 6 ++----
 fixtracks/widgets/classifier.py | 4 +++-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/fixtracks/fixtracks.py b/fixtracks/fixtracks.py
index e979479..1e3ae91 100644
--- a/fixtracks/fixtracks.py
+++ b/fixtracks/fixtracks.py
@@ -1,11 +1,9 @@
 from PySide6.QtCore import QSize, Qt
-from PySide6.QtWidgets import QMainWindow, QWidget, QToolBar, QMenu, QMenuBar, QSizePolicy, QFileDialog
-from PySide6.QtWidgets import QDialog, QVBoxLayout
-from PySide6.QtGui import QKeySequence, QAction, QIcon, QPalette
+from PySide6.QtWidgets import QMainWindow, QWidget, QToolBar, QSizePolicy, QFileDialog
+from PySide6.QtGui import QKeySequence, QAction, QIcon
 
 from fixtracks.widgets.centralwidget import CentralWidget
 from fixtracks.dialogs.previewdialog import PreviewDialog
-from fixtracks.utils.reader import ImageReader, DataFrameReader
 from fixtracks.dialogs.about import AboutDialog
 from fixtracks.dialogs.help import HelpDialog
 import fixtracks.resources
diff --git a/fixtracks/widgets/classifier.py b/fixtracks/widgets/classifier.py
index 813f435..bb4e7f1 100644
--- a/fixtracks/widgets/classifier.py
+++ b/fixtracks/widgets/classifier.py
@@ -239,7 +239,7 @@ class SizeClassifier(QWidget):
 
     def setCoordinates(self, coordinates):
         self._coordinates = coordinates
-        self._sizes = self.estimate_length(coordinates)
+        self._sizes = self.estimate_length(coordinates, bodyaxis=[0, 1, 2, 5])
         n, e = self.estimate_histogram(self._sizes)
         plot = self._plot_widget.addPlot()
         bgi = pg.BarGraphItem(x0=e[:-1], x1=e[1:], height=n, pen='w', brush=(0,0,255,150))
@@ -600,6 +600,7 @@ def main():
         df = pickle.load(f)
     data = TrackingData()
     data.setData(as_dict(df))
+    coords = data.coordinates()
 
     app = QApplication([])
     window = QWidget()
@@ -610,6 +611,7 @@ def main():
     # else:
     w = ClassifierWidget()
     w.setData(data)
+    w.size_classifier.setCoordinates(coords)
 
     layout = QVBoxLayout()
     layout.addWidget(w)