From b3e0a1464d6bd94d0a05c75e7eaf12b616fe5efb Mon Sep 17 00:00:00 2001
From: vihuhol <semichev9@gmail.com>
Date: Sat, 27 Apr 2019 18:37:26 +0300
Subject: [PATCH] Choosing tracks file by file dialog

---
 .../include/markup_backend/markup.h           | 34 ++++++++++++++++---
 .../markup_frontend/maincontrolpanel.h        |  3 +-
 .../include/markup_frontend/markupwidget.h    |  5 +--
 markup_frontend/src/maincontrolpanel.cpp      |  5 +--
 markup_frontend/src/markupwidget.cpp          | 26 ++++++++------
 5 files changed, 54 insertions(+), 19 deletions(-)

diff --git a/markup_backend/include/markup_backend/markup.h b/markup_backend/include/markup_backend/markup.h
index 23069d0..e78237f 100644
--- a/markup_backend/include/markup_backend/markup.h
+++ b/markup_backend/include/markup_backend/markup.h
@@ -16,7 +16,7 @@ struct PipelineRunParams {
     std::string video_path = "";
     std::string tracker_model_path = "";
     std::string tmp_video_dir = "/tmp/markup_video";
-    std::string debug_gt_tacks = "../markup_tool/data/test/MOT16-04/gt/gt.txt";
+    std::string tracks_path = "";
     std::string tmp_img_extention = ".png";
     size_t tmp_img_path_pad2length = 10;
     std::string tracks_filename = "tracks.txt";
@@ -72,6 +72,26 @@ class MarkUp {
         video_ = std::make_unique<Video>(params_.video_path);
     }
 
+    bool set_tracks(const std::string& filepath) {
+        if (filepath.empty()) {
+            return false;
+        }
+
+        if (!boost::filesystem::exists(filepath)) {
+            // TODO: Exceptions
+            std::cout << "File not found:" << filepath << std::endl;
+            return false;
+        }
+
+        if (params_.tracks_path == filepath) {
+            return true;
+        }
+
+        track_container_.reset(nullptr);
+
+        params_.tracks_path = filepath;
+    }
+
     bool run() {
         track_container_.reset(nullptr);
 
@@ -81,15 +101,21 @@ class MarkUp {
             return false;
         }
 
+        if (params_.tracks_path.empty()) {
+            // TODO: Exceptions
+            std::cout << "File with tracks are not choosen" << std::endl;
+            return false;
+        }
+
         // track_container_ = this->run_pipeline(*video_);
 
-        /* if (boost::filesystem::exists(params_.debug_gt_tacks)) {
+        /* if (boost::filesystem::exists(params_.tracks_path)) {
             // TODO: Exceptions
-            std::cout << "No debug tracks:" << params_.debug_gt_tacks << std::endl;
+            std::cout << "No debug tracks:" << params_.tracks_path << std::endl;
             return false;
         }*/
 
-        track_container_ = read_dummy_trackcontainer(params_.debug_gt_tacks);
+        track_container_ = read_dummy_trackcontainer(params_.tracks_path);
 
         return true;
     }
diff --git a/markup_frontend/include/markup_frontend/maincontrolpanel.h b/markup_frontend/include/markup_frontend/maincontrolpanel.h
index 877ebca..ed41ac8 100644
--- a/markup_frontend/include/markup_frontend/maincontrolpanel.h
+++ b/markup_frontend/include/markup_frontend/maincontrolpanel.h
@@ -13,7 +13,8 @@ class MainControlPanel : public QWidget
     explicit MainControlPanel(QWidget *parent = nullptr);
 
 signals:
-    void send_path(QDir dir);
+    void send_video_path(QDir dir);
+    void send_tracks_path(QString path);
     void send_run();
 
 private slots:
diff --git a/markup_frontend/include/markup_frontend/markupwidget.h b/markup_frontend/include/markup_frontend/markupwidget.h
index 5259bc7..007477f 100644
--- a/markup_frontend/include/markup_frontend/markupwidget.h
+++ b/markup_frontend/include/markup_frontend/markupwidget.h
@@ -19,14 +19,15 @@ class MarkupWidget : public QWidget
 
 public slots:
     void slot_set_video_path(QDir path);
+    void slot_set_tracks_path(QString path);
     void slot_framechanged(FrameWithControl *fwc);
     void slot_run();
 
 private:
     MainControlPanel *maincontrol = nullptr;
 
-    FrameWithControl *fwcup = nullptr;
-    FrameWithControl *fwcdn = nullptr;
+    FrameWithControl *upper_frame_panel = nullptr;
+    FrameWithControl *down_frame_panel = nullptr;
 
     // проверить возможность удаления...
     // QDir path;
diff --git a/markup_frontend/src/maincontrolpanel.cpp b/markup_frontend/src/maincontrolpanel.cpp
index 08dc528..a34e7f3 100644
--- a/markup_frontend/src/maincontrolpanel.cpp
+++ b/markup_frontend/src/maincontrolpanel.cpp
@@ -38,7 +38,7 @@ MainControlPanel::MainControlPanel(QWidget *parent) : QWidget(parent)
 void MainControlPanel::slot_loadvideo() {
     auto path = QFileDialog::getExistingDirectory(this, "Open Folder with frames");
     QDir dir(path);
-    emit send_path(dir);
+    emit send_video_path(dir);
     // call some backend-function
 }
 
@@ -51,7 +51,8 @@ void MainControlPanel::slot_loadmodel() {
 
 void MainControlPanel::slot_loadtracks() {
     auto path = QFileDialog::getOpenFileName(this, "Open Tracks", "",
-                                             "*.tracks");
+                                             "*.txt");
+    emit send_tracks_path(path);
     qDebug() << "tracks path : " << path;
     // call some backend-function
 }
diff --git a/markup_frontend/src/markupwidget.cpp b/markup_frontend/src/markupwidget.cpp
index 4404628..2b7304d 100644
--- a/markup_frontend/src/markupwidget.cpp
+++ b/markup_frontend/src/markupwidget.cpp
@@ -10,8 +10,8 @@ MarkupWidget::MarkupWidget(QWidget *parent)
         : QWidget(parent)
 {
     maincontrol = new MainControlPanel(this);
-    fwcup = new FrameWithControl(this);
-    fwcdn = new FrameWithControl(this);
+    upper_frame_panel = new FrameWithControl(this);
+    down_frame_panel = new FrameWithControl(this);
 
     QHBoxLayout *hlayout = new QHBoxLayout(this);
     QVBoxLayout* fwclayout = new QVBoxLayout;
@@ -19,25 +19,31 @@ MarkupWidget::MarkupWidget(QWidget *parent)
     hlayout->addWidget(maincontrol);
     hlayout->addLayout(fwclayout);
 
-    fwclayout->addWidget(fwcup);
-    fwclayout->addWidget(fwcdn);
+    fwclayout->addWidget(upper_frame_panel);
+    fwclayout->addWidget(down_frame_panel);
 
-    connect(maincontrol, &MainControlPanel::send_path, fwcup, &FrameWithControl::setPath);
-    connect(maincontrol, &MainControlPanel::send_path, fwcdn, &FrameWithControl::setPath);
-    connect(maincontrol, &MainControlPanel::send_path, this, &MarkupWidget::slot_set_video_path);
+    connect(maincontrol, &MainControlPanel::send_video_path, upper_frame_panel, &FrameWithControl::setPath);
+    connect(maincontrol, &MainControlPanel::send_video_path, down_frame_panel, &FrameWithControl::setPath);
+    connect(maincontrol, &MainControlPanel::send_video_path, this, &MarkupWidget::slot_set_video_path);
+
+    connect(maincontrol, &MainControlPanel::send_tracks_path, this, &MarkupWidget::slot_set_tracks_path);
 
     connect(maincontrol, &MainControlPanel::send_run, this, &MarkupWidget::slot_run);
 
-    connect(fwcup, &FrameWithControl::send_framechanged, this, &MarkupWidget::slot_framechanged);
-    connect(fwcdn, &FrameWithControl::send_framechanged, this, &MarkupWidget::slot_framechanged);
+    connect(upper_frame_panel, &FrameWithControl::send_framechanged, this, &MarkupWidget::slot_framechanged);
+    connect(down_frame_panel, &FrameWithControl::send_framechanged, this, &MarkupWidget::slot_framechanged);
 }
 
-// тут нужно передавать путь в бекэнд... но сейчас некуда...
 void MarkupWidget::slot_set_video_path(QDir path) {
     markup.set_video(std::string(path.path().toUtf8().constData()));
     path = path;
 }
 
+void MarkupWidget::slot_set_tracks_path(QString path) {
+    markup.set_tracks(std::string(path.toUtf8().constData()));
+    path = path;
+}
+
 void MarkupWidget::slot_run() {
     qDebug() << "Rewrite using new MarkUp interface";
     markup.run();