[feat] interactive: add stat_analysis feature preview

Author: bgallois

src/interactive.cpp

@@ -40,6 +40,8 @@ This file is part of Fast Track.
  */
 Interactive::Interactive(QWidget *parent) : QMainWindow(parent),
                                             ui(new Ui::Interactive),
+                                            video(new VideoReader()),
+                                            analysis(new StatAnalysis(this, false)),
                                             videoStatus(false),
                                             settingsFile(new QSettings(QStringLiteral("FastTrack"), QStringLiteral("FastTrackOrg"), this)) {
   ui->setupUi(this);
@@ -102,10 +104,12 @@ Interactive::Interactive(QWidget *parent) : QMainWindow(parent),
   connect(replayAction, &QAction::toggled, this, [this](bool isChecked) {
     if (isChecked) {
       ui->interactiveTab->addTab(replay, tr("Replay"));
+      ui->interactiveTab->addTab(analysis, tr("Analysis"));  // Preview feature. Will have a separate action latter if necessary.
       ui->interactiveTab->setCurrentIndex(1);
     }
     else {
-      ui->interactiveTab->removeTab(1);
+      ui->interactiveTab->removeTab(1);  // Remove Replay tab
+      ui->interactiveTab->removeTab(1);  // Remove Analysis tab
     }
   });
   ui->menuView->addAction(replayAction);
@@ -121,8 +125,11 @@ Interactive::Interactive(QWidget *parent) : QMainWindow(parent),
   });
 
   // Stay on the same frame when changing tab
-  connect(ui->interactiveTab, &QTabWidget::currentChanged, this, [this]() {
+  connect(ui->interactiveTab, &QTabWidget::currentChanged, this, [this](int index) {
     ui->slider->setValue(ui->slider->value());
+    if (index == 2) {
+      analysis->reload();
+    }
   });
 
   // Loads prefered style
@@ -378,12 +385,10 @@ Interactive::Interactive(QWidget *parent) : QMainWindow(parent),
   });
 
   // Replay tab
-  video = new VideoReader();
   replay = new Replay(this, false, ui->slider, video);
   connect(ui->interactiveTab, &QTabWidget::tabCloseRequested, this, [this](int index) {
     if (index != 0) {
-      ui->interactiveTab->removeTab(index);
-      replayAction->setChecked(false);
+      replayAction->setChecked(false);  // The replayAction takes care of the tabs removing
     }
   });
 
@@ -485,6 +490,7 @@ void Interactive::openFolder() {
   ui->display->clear();
   video->release();
   replay->clear();
+  analysis->clear();
   ui->backgroundProgressBar->setValue(0);
   ui->isBin->setCheckable(false);
   ui->isSub->setCheckable(false);
@@ -533,6 +539,7 @@ void Interactive::openFolder() {
       // Load replay
       replay->loadReplay(dir);
       if (!replay->trackingData->isEmpty) {
+        analysis->openTrackingData(replay->trackingData, true);
         replayAction->setChecked(true);
       }
 
@@ -853,7 +860,7 @@ void Interactive::getParameters() {
 }
 
 /**
- * @brief Does a tracing analysis on a sub-part of the image sequence defined by the user. Triggered when previewButton is clicked.
+ * @brief Does a tracking analysis on a sub-part of the image sequence defined by the user. Triggered when previewButton is clicked.
  */
 void Interactive::previewTracking() {
   if (videoStatus) {
@@ -863,6 +870,7 @@ void Interactive::previewTracking() {
     ui->trackButton->setDisabled(true);
     replayAction->setChecked(false);
     replay->clear();  // Avoid mixing 2 subsequent analysy
+    analysis->clear();
 
     QThread *thread = new QThread;
     Tracking *tracking = new Tracking(memoryDir.toStdString(), background, ui->startImage->value(), ui->startImage->value() + ui->stopImage->value());
@@ -878,13 +886,15 @@ void Interactive::previewTracking() {
       ui->previewButton->setDisabled(false);
       ui->trackButton->setDisabled(false);
       replay->loadReplay(dir);
+      analysis->openTrackingData(replay->trackingData, true);
       replayAction->setChecked(true);
     });
     connect(tracking, &Tracking::forceFinished, this, [this](const QString &errorMessage) {
       ui->slider->setDisabled(false);
       ui->previewButton->setDisabled(false);
       ui->trackButton->setDisabled(false);
       replay->loadReplay(dir);
+      analysis->openTrackingData(replay->trackingData, true);
       replayAction->setChecked(true);
       emit message(errorMessage);
     });
@@ -912,6 +922,7 @@ void Interactive::track() {
     ui->trackButton->setDisabled(true);
     replayAction->setChecked(false);
     replay->clear();  // Avoid mixing 2 subsequent analysy
+    analysis->clear();
 
     QThread *thread = new QThread;
     Tracking *tracking = new Tracking(memoryDir.toStdString(), background);
@@ -931,6 +942,7 @@ void Interactive::track() {
       ui->previewButton->setDisabled(false);
       ui->trackButton->setDisabled(false);
       replay->loadReplay(dir);
+      analysis->openTrackingData(replay->trackingData, true);
       replayAction->setChecked(true);
       logMap->insert(QStringLiteral("status"), errorMessage);
       emit log(*logMap);
@@ -943,6 +955,7 @@ void Interactive::track() {
       ui->previewButton->setDisabled(false);
       ui->trackButton->setDisabled(false);
       replay->loadReplay(dir);
+      analysis->openTrackingData(replay->trackingData, true);
       replayAction->setChecked(true);
       logMap->insert(QStringLiteral("status"), QStringLiteral("Done"));
       emit log(*logMap);

src/interactive.h

@@ -60,6 +60,7 @@ This file is part of Fast Track.
 #include "opencv2/highgui/highgui.hpp"
 #include "opencv2/imgproc/imgproc.hpp"
 #include "replay.h"
+#include "statanalysis.h"
 #include "tracking.h"
 #include "videoreader.h"
 
@@ -129,8 +130,9 @@ class Interactive : public QMainWindow {
   QPixmap resizedPix;
   vector<Point3i> colorMap;
   double currentZoom;
-  Replay *replay;
   VideoReader *video;
+  Replay *replay;
+  StatAnalysis *analysis;
   bool videoStatus;
 
   QString style;

src/statanalysis.cpp

@@ -23,7 +23,8 @@ StatAnalysis::StatAnalysis(QWidget* parent, bool isStandalone) : QMainWindow(par
                                                                  isStandalone(isStandalone),
                                                                  trackingData(new Data()),
                                                                  settingsFile(new QSettings(QStringLiteral("FastTrack"), QStringLiteral("FastTrackOrg"), this)),
-                                                                 ruler(1) {
+                                                                 ruler(1),
+                                                                 isData(false) {
   ui->setupUi(this);
 
   // Loads settings
@@ -96,28 +97,30 @@ void StatAnalysis::openTrackingData() {
   memoryDir = file;
 }
 
-void StatAnalysis::openTrackingData(const QString& file) {
+void StatAnalysis::openTrackingData(const QString& file, bool loadOnDemand) {
   clear();
   QString dir = QFileInfo(file).dir().path();  // Data takes the tracking folder.
   QApplication::setOverrideCursor(Qt::WaitCursor);
   trackingData->setPath(dir);
-  if (!trackingData->isEmpty) {
+  if (!trackingData->isEmpty && !loadOnDemand) {  // If loadOnDemand is true, need to call reload to actually load data.
     loadObjectList();
     QList<int> objects = trackingData->getId(0, trackingData->maxFrameIndex);
     initPlots(objects);
   }
+  isData = !trackingData->isEmpty;
   QApplication::restoreOverrideCursor();
 }
 
-void StatAnalysis::openTrackingData(Data* data) {
+void StatAnalysis::openTrackingData(Data* data, bool loadOnDemand) {
   clear();
   QApplication::setOverrideCursor(Qt::WaitCursor);
   trackingData = new Data(*data);
-  if (!trackingData->isEmpty) {
+  if (!trackingData->isEmpty && !loadOnDemand) {  // If loadOnDemand is true, need to call reload to actually load data.
     loadObjectList();
     QList<int> objects = trackingData->getId(0, trackingData->maxFrameIndex);
     initPlots(objects);
   }
+  isData = !trackingData->isEmpty;
   QApplication::restoreOverrideCursor();
 }
 
@@ -145,6 +148,19 @@ QList<int> StatAnalysis::getSelectedObjects() {
   return objects;
 }
 
+void StatAnalysis::reload() {
+  if (isData) {
+    QApplication::setOverrideCursor(Qt::WaitCursor);
+    ui->objectList->setRowCount(0);  // Remove all rows
+    clearPlots();
+    ruler = 1;
+    loadObjectList();
+    QList<int> objects = trackingData->getId(0, trackingData->maxFrameIndex);
+    initPlots(objects);
+    QApplication::restoreOverrideCursor();
+  }
+}
+
 void StatAnalysis::refresh() {
   QList<int> objects = getSelectedObjects();
   refreshPlots(objects);

src/statanalysis.h

@@ -57,8 +57,9 @@ class StatAnalysis : public QMainWindow {
   StatAnalysis &operator=(StatAnalysis &&T) = delete;
   StatAnalysis(StatAnalysis &&T) = delete;
   ~StatAnalysis();
-  void openTrackingData(const QString &file);
-  void openTrackingData(Data *data);
+  void openTrackingData(const QString &file, bool loadOnDemand = false);
+  void openTrackingData(Data *data, bool loadOnDemand = false);
+  void reload();
   void clear();
 
  private slots:
@@ -79,6 +80,7 @@ class StatAnalysis : public QMainWindow {
   QSettings *settingsFile;
   QString memoryDir;
   double ruler;
+  bool isData;
 };
 
 #endif  // STATANALYSIS_H