opening/creating projects, running reconstruction from CL

Author: UladBohdan

CMakeLists.txt

@@ -28,7 +28,8 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${Qt5Widgets_EXECUTABLE_COMPILE_FLAGS}")
 # The order below is important:
 add_subdirectory(libs/QtAwesome/QtAwesome)
 add_subdirectory(src)
+add_subdirectory(cli)
 add_subdirectory(ui)
 
 add_executable(${PROJECT_NAME} main.cpp)
-target_link_libraries(${PROJECT_NAME} src ui QtAwesome)
+target_link_libraries(${PROJECT_NAME} src cli ui QtAwesome)

cli/CMakeLists.txt

@@ -0,0 +1,12 @@
+set(SOURCE_FILES
+    cli.cpp
+    )
+
+set(HEADERS
+    cli.h
+    )
+
+add_library(cli ${SOURCE_FILES} ${HEADERS})
+
+# Never include ui target here.
+target_link_libraries(cli src Qt5::Widgets)

cli/cli.cpp

@@ -0,0 +1,108 @@
+// Copyright 2017 Lybros.
+
+#include "cli.h"
+
+#include <QDateTime>
+#include <QDir>
+#include <QString>
+
+#include "gflags/gflags.h"
+#include "glog/logging.h"
+
+#include "../src/project.h"
+
+/* The minimum sets of parameters in CLI mode (to pass verification) are:
+ *   new + project_name + project_path + images_path + action
+ *   open + action
+ *   temp + images_path + action
+ */
+
+DEFINE_string(cli_mode, "",
+              "In CLI mode you may either create a new project "
+              "or open an existent one. Choose between open, new or temp "
+              "(temp means new + removal after processing).");
+DEFINE_string(project_path, "",
+              "If you're in 'open' cli_mode, you must specify the project.");
+DEFINE_string(project_name, "",
+              "You must provide a project name in order to create a new one.");
+DEFINE_string(images_path, "",
+              "You must provide a dataset in order to create a new project.");
+
+DEFINE_string(action, "",
+              "Choose if you want to extract features, match them or run "
+              "reconstruction. Visualization is only available in 'ui' mode");
+
+bool VerifyCommandlineArguments() {
+  if (FLAGS_cli_mode != "new" &&
+      FLAGS_cli_mode != "open" &&
+      FLAGS_cli_mode != "temp") {
+    LOG(ERROR) << "Wrong cli_mode value.";
+    return false;
+  }
+
+  // TODO(uladbohdan): to support more actions.
+  if (FLAGS_action != "reconstruction") {
+    LOG(ERROR) << "You must pass a valid action to proceed.";
+    return false;
+  }
+
+  if (FLAGS_cli_mode == "new") {
+    if (FLAGS_project_name == "" ||
+        FLAGS_project_path == "" ||
+        FLAGS_images_path == "") {
+      LOG(ERROR) << "You must provide project_name, project_path and "
+                    "images_path to create a new project.";
+      return false;
+    }
+  }
+
+  if (FLAGS_cli_mode == "open") {
+    if (FLAGS_project_path == "") {
+      LOG(ERROR) << "You must provide project_path to open a project.";
+      return false;
+    }
+  }
+
+  if (FLAGS_cli_mode == "temp") {
+    if (FLAGS_images_path == "") {
+      LOG(ERROR) << "You must provide images_path to create a temp project";
+      return false;
+    }
+  }
+
+  return true;
+}
+
+void RunCLI() {
+  LOG(INFO) << "Running in CLI mode.";
+
+  if (!VerifyCommandlineArguments()) {
+    LOG(ERROR) << "CL arguments verification failed.";
+    return;
+  }
+
+  Project* project;
+
+  if (FLAGS_cli_mode == "open") {
+    project = new Project(QString::fromStdString(FLAGS_project_path));
+  } else if (FLAGS_cli_mode == "new") {
+    project = new Project(QString::fromStdString(FLAGS_project_name),
+                          QString::fromStdString(FLAGS_project_path),
+                          QString::fromStdString(FLAGS_images_path));
+  } else if (FLAGS_cli_mode == "temp") {
+    project = new Project(QDateTime::currentDateTime().toString(Qt::ISODate),
+                          QDir::tempPath(),
+                          QString::fromStdString(FLAGS_images_path));
+  }
+
+  if (FLAGS_action == "reconstruction") {
+    LOG(INFO) << "Starting reconstruction...";
+    project->BuildModelToBinary();
+  }
+
+  if (FLAGS_cli_mode == "temp") {
+    project->RemoveProject();
+  }
+
+  delete project;
+}

cli/cli.h

@@ -0,0 +1,8 @@
+// Copyright 2017 Lybros.
+
+#ifndef CLI_CLI_H_
+#define CLI_CLI_H_
+
+void RunCLI();
+
+#endif  // CLI_CLI_H_

main.cpp

@@ -1,20 +1,42 @@
 // Copyright 2017 Lybros.
 
+#include <gflags/gflags.h>
+#include <glog/logging.h>
+
+#include "cli/cli.h"
 #include "ui/mainwindow.h"
 
 #include <QApplication>
 
+DEFINE_string(mode, "ui", "Application mode. Choose between ui and cli.");
+
 int main(int argc, char* argv[]) {
-  QApplication a(argc, argv);
-  QtAwesome* awesome = new QtAwesome(&a);
-  awesome->initFontAwesome();
+  gflags::ParseCommandLineFlags(&argc, &argv, true);
 
   google::InitGoogleLogging(argv[0]);
   LOG(INFO) << "Logging is enabled.";
 
-  MainWindow w;
-  w.set_icons(awesome);
-  w.show();
+  if (FLAGS_mode == "ui") {
+    LOG(INFO) << "Running UI.";
+    QApplication a(argc, argv);
+    QtAwesome* awesome = new QtAwesome(&a);
+    awesome->initFontAwesome();
+
+    MainWindow w;
+    w.set_icons(awesome);
+    w.show();
+
+    return a.exec();
+
+  } else if (FLAGS_mode == "cli") {
+    // Running in CLI mode means no need in using any code from ui/, as ui/ is
+    // only a wrapper for a Project object.
+    // In CLI mode we may simply create a Project and work with it directly.
+    RunCLI();
+    return 0;
+  }
 
-  return a.exec();
+  LOG(ERROR) << "'mode' flag must be either 'ui' or 'cli'. "
+                "If not provided, default value is 'ui'.";
+  return 0;
 }

src/project.cpp

@@ -171,6 +171,10 @@ Options* Project::GetOptions() {
   return options_;
 }
 
+bool Project::RemoveProject() {
+  return true;
+}
+
 Project::~Project() {
   WriteConfigurationFile();
   delete options_;

src/project.h

@@ -83,6 +83,9 @@ class Project {
 
   Options* GetOptions();
 
+  // Method completely removes Project from filesystem.
+  bool RemoveProject();
+
   ~Project();
 
  private: