X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=vv%2Fvv.cxx;h=59cacb58911b5bb3ba8d51c18a49d46ae0840fee;hb=ff1d5b8af965e75ecdbc1b0cbff0dad6f3b3c511;hp=b30e97c47148fb0904f83e4c8f8f185869a18206;hpb=2b4229b2e82be6bd48817b0171afb2d69a9e9758;p=clitk.git

diff --git a/vv/vv.cxx b/vv/vv.cxx
index b30e97c..59cacb5 100644
--- a/vv/vv.cxx
+++ b/vv/vv.cxx
@@ -17,16 +17,51 @@
 ======================================================================-====*/
 #include <utility>
 #include <cassert>
+#include <ctime>
+#include <string>
+#include <ostream>
+#include <sstream>
 #include <QApplication>
 #include <QPixmap>
 #include <QSplashScreen>
 #include <QTimer>
 #include <QDesktopWidget>
+#include <QDir>
 
 #include "clitkIO.h"
 #include "vvMainWindow.h"
 #include "vvConstants.h"
 
+#include <vtkFileOutputWindow.h>
+#include <vtkSmartPointer.h>
+#include <itkFileOutputWindow.h>
+#include <itkSmartPointer.h>
+#include <itksys/SystemTools.hxx>
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include  <errno.h>
+
+void load_image_first_error()
+{
+  std::cerr << "You need to load an image before adding an overlay!" << std::endl;
+  exit(1);
+}
+
+std::string create_timed_string()
+{
+  time_t t;
+  time(&t);
+
+  struct tm* pt = localtime(&t);
+
+  const int size = 64;
+  char st[size];
+  strftime(st, size, "%Y%m%d-%H%M%S", pt);
+
+  return st;
+}
+
 //------------------------------------------------------------------------------
 int main( int argc, char** argv )
 {
@@ -53,50 +88,93 @@ int main( int argc, char** argv )
 
   window.show();
 
-  std::vector<std::string> filenames;
-  std::vector<std::pair<int ,std::string> > overlays;
-  std::vector<std::pair<int ,std::string> > rois;
-  std::vector<std::pair<int ,std::string> > vector_fields;
+  std::vector<std::string> sequence_filenames;
+  enum {P_NORMAL,P_SEQUENCE,P_WINDOW,P_LEVEL};
+  int parse_mode=P_NORMAL;
+  int n_image_loaded=0;
+  std::string win(""), lev("");
+
   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 {
-	  if (temp=="--roi") {
-	    assert(filenames.size()>=1);
-	    rois.push_back(std::make_pair(filenames.size()-1,argv[i+1]));
-	    i++; //skip overlay name
-	  } 
-	  else {
-	    filenames.push_back(temp);
-	  }
-	}
+      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;
+        }
+        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 (current == "--window") {
+          parse_mode=P_WINDOW;
+        } else if (current == "--level") {
+          parse_mode=P_LEVEL;
+        } else if (current == "--log") {
+          std::string log_dir = QDir::tempPath().toStdString() + std::string("/vv-log");
+
+          if(itksys::SystemTools::FileExists(log_dir.c_str()) &&
+              !itksys::SystemTools::FileIsDirectory(log_dir.c_str())) {
+            itkGenericExceptionMacro(<< "Error creating log directory, file exists and is not a directory.");
+          } else if(!itksys::SystemTools::MakeDirectory(log_dir.c_str())) {
+            itkGenericExceptionMacro(<< "Error creating log directory.");
+          }
+
+          std::string log_file = log_dir + "/" + create_timed_string() + ".log";
+
+          itk::SmartPointer<itk::FileOutputWindow> itk_log = itk::FileOutputWindow::New();
+          itk_log->SetFileName(log_file.c_str());
+          itk_log->FlushOn();
+          itk_log->AppendOn();
+          itk::OutputWindow::SetInstance(itk_log);
+
+          vtkSmartPointer<vtkFileOutputWindow> vtk_log = vtkFileOutputWindow::New();
+          vtk_log->SetFileName(log_file.c_str());
+          vtk_log->FlushOn();
+          vtk_log->AppendOn();
+          vtkOutputWindow::SetInstance(vtk_log);
+        }
+      } else if (parse_mode == P_SEQUENCE) {
+        sequence_filenames.push_back(current);
+      } else if (parse_mode == P_WINDOW) {
+        win=current;
+        parse_mode=P_NORMAL;
+      } else if (parse_mode == P_LEVEL) {
+        lev=current;
+        parse_mode=P_NORMAL;
+      } else {
+        std::vector<std::string> image;
+        image.push_back(current);
+        window.LoadImages(image,IMAGE);
+        n_image_loaded++;
       }
     }
-    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);
-    for (std::vector<std::pair<int ,std::string> >::iterator i=rois.begin();
-         i!=rois.end(); i++) {
-      DD((*i).second.c_str());
-      DD((*i).first);
-      // window.AddROI((*i).second.c_str(), (*i).first);
+    if (parse_mode == P_SEQUENCE) { //Finish any current sequence
+      window.LoadImages(sequence_filenames,MERGEDWITHTIME);
+      sequence_filenames.clear();
+      parse_mode=P_NORMAL;
     }
+  }
 
+  if(win!="" && lev!="") {
+    window.WindowLevelChanged(atof(win.c_str()), atof(lev.c_str()), 6, 0);
+    window.ApplyWindowLevelToAllImages();
   }
 
   return app.exec();