Parsing fusion image from command line

[clitk.git] / vv / vv.cxx

diff --git a/vv/vv.cxx b/vv/vv.cxx
index 73e7792c8f2eb6bfdc44885889562e23bc883f2c..71974f1e6124dea74f279b35825a76edf7b369c2 100644 (file)
--- a/vv/vv.cxx
+++ b/vv/vv.cxx
@@ -1,7 +1,7 @@
 /*=========================================================================
   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
 
-  Authors belong to: 
+  Authors belong to:
   - University of LYON              http://www.universite-lyon.fr/
   - Léon Bérard cancer center       http://oncora1.lyon.fnclcc.fr
   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
@@ -27,69 +27,92 @@
 #include "vvMainWindow.h"
 #include "vvConstants.h"
 
+void load_image_first_error()
+{
+  std::cerr << "You need to load an image before adding an overlay!" << std::endl;
+  exit(1);
+}
+
 //------------------------------------------------------------------------------
 int main( int argc, char** argv )
 {
-#ifdef _WIN32
-    HWND hWnd = GetConsoleWindow();
-    ShowWindow( hWnd, SW_HIDE );
-#endif
+  CLITK_INIT;
 
-    CLITK_INIT;
+  QApplication app( argc, argv );
+  Q_INIT_RESOURCE(vvIcons);
+  //QPixmap pixmap(":/splashscreen.PNG");
+  QSplashScreen *splash = new QSplashScreen(QPixmap(QString::fromUtf8(":/new/prefix1/splashscreen.PNG")));
+  /*splash->showMessage("VV 1.0 developped by Léon Bérard c`ancer center http://oncora1.lyon.fnclcc.fr and CREATIS-LRMN http://www.creatis.insa-lyon.fr",(Qt::AlignRight | Qt::AlignBottom));*/
+  //  splash->show();
+  QTimer::singleShot(2000, splash, SLOT(close()));
+  while (!splash->isHidden())
+    app.processEvents();
 
-    QApplication app( argc, argv );
-    Q_INIT_RESOURCE(vvIcons);
-    //QPixmap pixmap(":/splashscreen.PNG");
-    QSplashScreen *splash = new QSplashScreen(QPixmap(QString::fromUtf8(":/new/prefix1/splashscreen.PNG")));
-    /*splash->showMessage("VV 1.0 developped by Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr and CREATIS-LRMN http://www.creatis.insa-lyon.fr",(Qt::AlignRight | Qt::AlignBottom));*/
-    //  splash->show();
-    QTimer::singleShot(2000, splash, SLOT(close()));
-    while (!splash->isHidden())
-        app.processEvents();
+  vvMainWindow window;
 
-    vvMainWindow window;
+  //Try to give the window a sensible default size
+  int width=QApplication::desktop()->width()*0.8;
+  int height=QApplication::desktop()->height()*0.9;
+  if (width> 1.5*height)
+    width=1.5*height;
+  window.resize(width,height);
 
-    //Try to give the window a sensible default size
-    int width=QApplication::desktop()->width()*0.8;
-    int height=QApplication::desktop()->height()*0.9;
-    if (width> 1.5*height)
-        width=1.5*height;
-    window.resize(width,height);
+  window.show();
 
-    window.show();
+  std::vector<std::string> sequence_filenames;
+  enum {P_NORMAL,P_SEQUENCE};
+  int parse_mode=P_NORMAL;
+  int n_image_loaded=0;
 
-    std::vector<std::string> filenames;
-    std::vector<std::pair<int ,std::string> > overlays;
-    std::vector<std::pair<int ,std::string> > vector_fields;
-    if (argc >1)
-    {
-        for (int i = 1; i < argc; i++)
-        {
-            std::string temp = argv[i];
-            if (temp=="--vf")
-            {
-                assert(filenames.size()>=1);
-                vector_fields.push_back(std::make_pair(filenames.size()-1,argv[i+1]));
-                i++; //skip vf name
-            }
-            else if (temp=="--overlay")
-            {
-                assert(filenames.size()>=1);
-                overlays.push_back(std::make_pair(filenames.size()-1,argv[i+1]));
-                i++; //skip overlay name
-            }
-            else
-                filenames.push_back(temp);
+  if (argc >1) {
+    for (int i = 1; i < argc; i++) {
+      std::string current = argv[i];
+      if (!current.compare(0,2,"--")) { //We are parsing an option
+        if (parse_mode == P_SEQUENCE) {//First finish the current sequence
+          window.LoadImages(sequence_filenames,MERGEDWITHTIME);
+          sequence_filenames.clear();
+          parse_mode=P_NORMAL;
         }
-        window.LoadImages(filenames,IMAGE);
-        for (std::vector<std::pair<int ,std::string> >::iterator i=overlays.begin();
-             i!=overlays.end();i++)
-          window.AddOverlayImage((*i).first,(*i).second.c_str());
-        for (std::vector<std::pair<int ,std::string> >::iterator i=vector_fields.begin();
-             i!=vector_fields.end();i++)
-          window.AddField((*i).second.c_str(), (*i).first);
-        
+        if (current=="--vf") {
+          if (!n_image_loaded) load_image_first_error();
+          window.AddField(argv[i+1],n_image_loaded-1);
+          i++; //skip vf name
+        } 
+        else if (current=="--overlay") {
+            if (!n_image_loaded) load_image_first_error();
+            window.AddOverlayImage(n_image_loaded-1,argv[i+1]);
+            i++; //skip overlay name
+          } 
+        else if (current=="--roi") {
+            if (!n_image_loaded) load_image_first_error();
+            window.AddROI(n_image_loaded-1,argv[i+1]);
+            i++; //skip roi name
+          }
+        else if (current=="--fusion") {
+            if (!n_image_loaded) load_image_first_error();
+            window.AddFusionImage(n_image_loaded-1,argv[i+1]);
+            i++; //skip fusion name
+          }
+        else if (current == "--sequence") {
+          n_image_loaded++; //count only one for the sequence
+          parse_mode=P_SEQUENCE; }
+      }
+      else if (parse_mode == P_SEQUENCE)
+        sequence_filenames.push_back(current);
+      else {
+        DD(current);
+        std::vector<std::string> image; image.push_back(current);
+        window.LoadImages(image,IMAGE);
+        n_image_loaded++;
+      }
     }
+    if (parse_mode == P_SEQUENCE) {//Finish any current sequence
+      window.LoadImages(sequence_filenames,MERGEDWITHTIME);
+      sequence_filenames.clear();
+      parse_mode=P_NORMAL;
+    }
+
+  }
 
-    return app.exec();
+  return app.exec();
 }