jgrewe/fixtracks
1
0
Fork 0
Code Issues 4 Pull Requests Actions Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Jan Grewe 2025-02-20 18:00:29 +01:00
parent 33b46af8d0
commit 20b2915b6b
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
fixtracks/fixtracks.py
View File

@ -1,11 +1,9 @@
from PySide6.QtCore import QSize, Qt from PySide6.QtCore import QSize, Qt
from PySide6.QtWidgets import QMainWindow, QWidget, QToolBar, QMenu, QMenuBar, QSizePolicy, QFileDialog from PySide6.QtWidgets import QMainWindow, QWidget, QToolBar, QSizePolicy, QFileDialog
from PySide6.QtWidgets import QDialog, QVBoxLayout from PySide6.QtGui import QKeySequence, QAction, QIcon
from PySide6.QtGui import QKeySequence, QAction, QIcon, QPalette
from fixtracks.widgets.centralwidget import CentralWidget from fixtracks.widgets.centralwidget import CentralWidget
from fixtracks.dialogs.previewdialog import PreviewDialog from fixtracks.dialogs.previewdialog import PreviewDialog
from fixtracks.utils.reader import ImageReader, DataFrameReader
from fixtracks.dialogs.about import AboutDialog from fixtracks.dialogs.about import AboutDialog
from fixtracks.dialogs.help import HelpDialog from fixtracks.dialogs.help import HelpDialog
import fixtracks.resources import fixtracks.resources

4
fixtracks/widgets/classifier.py
View File

@ -239,7 +239,7 @@ class SizeClassifier(QWidget):
def setCoordinates(self, coordinates): def setCoordinates(self, coordinates):
self._coordinates = 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) n, e = self.estimate_histogram(self._sizes)
plot = self._plot_widget.addPlot() plot = self._plot_widget.addPlot()
bgi = pg.BarGraphItem(x0=e[:-1], x1=e[1:], height=n, pen='w', brush=(0,0,255,150)) 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) df = pickle.load(f)
data = TrackingData() data = TrackingData()
data.setData(as_dict(df)) data.setData(as_dict(df))
coords = data.coordinates()
app = QApplication([]) app = QApplication([])
window = QWidget() window = QWidget()
@ -610,6 +611,7 @@ def main():
# else: # else:
w = ClassifierWidget() w = ClassifierWidget()
w.setData(data) w.setData(data)
w.size_classifier.setCoordinates(coords)
layout = QVBoxLayout() layout = QVBoxLayout()
layout.addWidget(w) layout.addWidget(w)