]> Creatis software - clitk.git/blobdiff - vv/vvToolBinarize.cxx
- autotool
[clitk.git] / vv / vvToolBinarize.cxx
index 251c0d68607cd4da37a300d0e406f3b147d0c0dd..4f91d19d2c20165c87bd0e1c7acfa23ac26f33de 100644 (file)
@@ -3,8 +3,8 @@
   Program:   vv
   Module:    $RCSfile: vvToolBinarize.cxx,v $
   Language:  C++
-  Date:      $Date: 2010/02/03 13:08:55 $
-  Version:   $Revision: 1.4 $
+  Date:      $Date: 2010/03/01 15:38:09 $
+  Version:   $Revision: 1.10 $
   Author :   David Sarrut (david.sarrut@creatis.insa-lyon.fr)
 
   Copyright (C) 2008
@@ -32,8 +32,9 @@
 
 #include "clitkBinarizeImageGenericFilter.h"
 
-#include "vtkImageActor.h"
-#include "vtkCamera.h"
+#include <vtkImageActor.h>
+#include <vtkCamera.h>
+#include <vtkImageClip.h>
 
 //------------------------------------------------------------------------------
 // Create the tool and automagically (I like this word) insert it in
@@ -43,32 +44,32 @@ ADD_TOOL(vvToolBinarize);
 
 
 //------------------------------------------------------------------------------
-vvToolBinarize::vvToolBinarize(QWidget * parent, Qt::WindowFlags f)
-  :QDialog(parent,f), vvToolBase<vvToolBinarize>(), Ui::vvToolBinarize() 
-{
-  // Set Modality : dialog is not modal but stay always on top because
-  // parent is set at construction
-  setModal(false);
-  setAttribute(Qt::WA_DeleteOnClose);
+void vvToolBinarize::Initialize() {
+  SetToolName("Binarize");
+  SetToolMenuName("Binarize");
+  SetToolIconFilename(":/new/prefix1/icons/binarize.png");
+  SetToolTip("Image interactive binarization with thresholds.");
+}
+//------------------------------------------------------------------------------
 
+
+//------------------------------------------------------------------------------
+vvToolBinarize::vvToolBinarize(vvMainWindowBase * parent, Qt::WindowFlags f)
+  :vvToolWidgetBase(parent,f), 
+   vvToolBase<vvToolBinarize>(parent), 
+   Ui::vvToolBinarize() 
+{
   // GUI Initialization
-  setupUi(this);
+  Ui_vvToolBinarize::setupUi(mToolWidget);
+  mInteractiveDisplayIsEnabled = mCheckBoxInteractiveDisplay->isChecked();
 
   // Connect signals & slots  
-  connect(mToolInputSelectionWidget, SIGNAL(accepted()), this, SLOT(InputIsSelected()));
-  connect(mToolInputSelectionWidget, SIGNAL(rejected()), this, SLOT(close()));
-  connect(buttonBox, SIGNAL(accepted()), this, SLOT(apply()));
-  connect(buttonBox, SIGNAL(rejected()), this, SLOT(close()));
   connect(mThresholdSlider1, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT1(double)));
   connect(mThresholdSlider2, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT2(double)));
-
   connect(mRadioButtonLowerThan, SIGNAL(toggled(bool)), this, SLOT(enableLowerThan(bool)));
-
-  // VTK objects
-  mClipper = vtkImageClip::New();
-  mSquares1 = vtkMarchingSquares::New();
-  mSquaresMapper1 = vtkPolyDataMapper::New();
-  mSquaresActor1 = vtkActor::New();
+  connect(mCheckBoxUseFG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
+  connect(mCheckBoxUseBG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
+  connect(mCheckBoxInteractiveDisplay, SIGNAL(toggled(bool)), this, SLOT(InteractiveDisplayToggled(bool)));
 
   // Initialize some widget
   mThresholdSlider1->SetText("");
@@ -76,14 +77,8 @@ vvToolBinarize::vvToolBinarize(QWidget * parent, Qt::WindowFlags f)
   mFGSlider->SetText("Foreground value");
   mBGSlider->SetText("Background value");
 
-  // Disable main widget while input image is not selected
-  toolMainWidget->setEnabled(false);
-
   // Main filter 
   mFilter = new clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>; //DS PUT IN BASECLASS ?
-
-  // Initialize the input selection (mFilter should be create before !)
-  InitializeListOfInputImages(mToolInputSelectionWidget, mFilter);
 }
 //------------------------------------------------------------------------------
 
@@ -94,6 +89,48 @@ vvToolBinarize::~vvToolBinarize() {
 //------------------------------------------------------------------------------
 
 
+//------------------------------------------------------------------------------
+void vvToolBinarize::InteractiveDisplayToggled(bool b) {
+  mInteractiveDisplayIsEnabled = b;
+  if (!mInteractiveDisplayIsEnabled) {
+    RemoveVTKObjects();
+  }
+  else {
+    for(unsigned int i=0; i<mImageContour.size(); i++)
+      mImageContour[i]->showActors();
+    if (mCurrentSlicerManager)
+      mCurrentSlicerManager->Render();
+  }
+}
+//------------------------------------------------------------------------------
+
+
+//------------------------------------------------------------------------------
+void vvToolBinarize::RemoveVTKObjects() { 
+  for(unsigned int i=0; i<mImageContour.size(); i++)
+    mImageContour[i]->hideActors();
+  if (mCurrentSlicerManager)
+    mCurrentSlicerManager->Render();
+}
+//------------------------------------------------------------------------------
+
+
+//------------------------------------------------------------------------------
+bool vvToolBinarize::close() { 
+  RemoveVTKObjects();
+  return vvToolWidgetBase::close(); 
+}
+//------------------------------------------------------------------------------
+
+
+//------------------------------------------------------------------------------
+void vvToolBinarize::reject() { 
+  RemoveVTKObjects();
+  return vvToolWidgetBase::reject(); 
+}
+//------------------------------------------------------------------------------
+
+
 //------------------------------------------------------------------------------
 void vvToolBinarize::enableLowerThan(bool b) {
   if (!b) {
@@ -107,116 +144,53 @@ void vvToolBinarize::enableLowerThan(bool b) {
 
 
 //------------------------------------------------------------------------------
-void vvToolBinarize::InputIsSelected() {
+void vvToolBinarize::useFGBGtoggled(bool) {
+  if (!mCheckBoxUseBG->isChecked() && !mCheckBoxUseFG->isChecked()) 
+    mCheckBoxUseBG->toggle();
+}
+//------------------------------------------------------------------------------
 
-  // Common
-  int index = mToolInputSelectionWidget->GetSelectedInputIndex();
-  mCurrentSliceManager = mSlicerManagersCompatible[index];
-  mCurrentImage = mCurrentSliceManager->GetImage();
-  toolMainWidget->setEnabled(true);
+
+//------------------------------------------------------------------------------
+void vvToolBinarize::InputIsSelected(vvSlicerManager * m) {
+  mCurrentSlicerManager = m;
 
   // Specific for this gui
+  mThresholdSlider1->SetValue(0);
+  mThresholdSlider2->SetValue(0);
   mThresholdSlider1->SetImage(mCurrentImage);
   mThresholdSlider2->SetImage(mCurrentImage);
   mFGSlider->SetImage(mCurrentImage);
   mBGSlider->SetImage(mCurrentImage);
+  //  DD(mCurrentSlicerManager->GetFileName().c_str());
+  mFGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
+  mFGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
+  mBGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
+  mBGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
   mFGSlider->SetValue(1);
   mBGSlider->SetValue(0);
-  DD(mCurrentSliceManager->GetFileName().c_str());
-  
-  DD("VTK");
-  DD(mCurrentSliceManager->NumberOfSlicers());
-  //    mClipper->SetInput(mCurrentSliceManager->GetSlicer(0)->GetInput());
-  DD(mCurrentImage->GetFirstVTKImageData());
-  DD(mClipper);
-  DD(mCurrentSliceManager->GetSlicer(0));
-  mClipper->SetInput(mCurrentImage->GetFirstVTKImageData());
-  
-  mSquares1->SetInput(mClipper->GetOutput());
-  mSquaresMapper1->SetInput(mSquares1->GetOutput());
-  mSquaresActor1->SetMapper(mSquaresMapper1);
-  mSquaresActor1->GetProperty()->SetColor(1.0,0,0);
-  mSquaresActor1->SetPickable(0);
-  mCurrentSliceManager->GetSlicer(0)->GetRenderer()->AddActor(mSquaresActor1);
-  mSquares1->Update();
-  
-  
-  DD("VTK end");
-  
-  //    connect(mCurrentSliceManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
-  connect(mCurrentSliceManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
-  //connect(mCurrentSliceManager,SIGNAL(UpdateSliceRange(int,int,int,int,int)),this,SLOT(UpdateSlice(int, int)));
-  //    connect(mCurrentSliceManager,SIGNAL(LandmarkAdded()),this,SLOT(InsertSeed()));
   
+  // VTK objects for interactive display
+  for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
+    mImageContour.push_back(new vvImageContour);
+    mImageContour[i]->setSlicer(mCurrentSlicerManager->GetSlicer(i));
+  }
+  valueChangedT1(mThresholdSlider1->GetValue());
+  connect(mCurrentSlicerManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
+  connect(mCurrentSlicerManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
+
 }
 //------------------------------------------------------------------------------
 
 
 //------------------------------------------------------------------------------
 void vvToolBinarize::UpdateSlice(int slicer,int slices) {
-  // A METTRE SUR TOUT LES SLICES ! PAS QUE 0
-
-  // !! signal update slice pas tjs quand move slicer ???
-
-
-   int slice = mCurrentSliceManager->GetSlicer(0)->GetSlice();
-    //int tslice = mCurrentSliceManager->GetSlicer(0)->GetTSlice();
-    mClipper->SetInput(mCurrentSliceManager->GetSlicer(0)->GetInput());
-    int* extent = mCurrentSliceManager->GetSlicer(0)->GetImageActor()->GetDisplayExtent();
-    mClipper->SetOutputWholeExtent(extent[0],extent[1],extent[2],extent[3],extent[4],extent[5]);
-    int i;
-    for (i = 0; i < 6;i = i+2)
-    {
-        if (extent[i] == extent[i+1])
-        {
-            break;
-        }
-    }
-
-    switch (i)
-    {
-    case 0:
-        if (mCurrentSliceManager->GetSlicer(0)->GetRenderer()->GetActiveCamera()->GetPosition()[0] > slice)
-        {
-            mSquaresActor1->SetPosition(1,0,0);
-            // mSquaresActor2->SetPosition(1,0,0);
-        }
-        else
-        {
-            mSquaresActor1->SetPosition(-1,0,0);
-            // mSquaresActor2->SetPosition(-1,0,0);
-        }
-        break;
-    case 2:
-        if (mCurrentSliceManager->GetSlicer(0)->GetRenderer()->GetActiveCamera()->GetPosition()[1] > slice)
-        {
-            mSquaresActor1->SetPosition(0,1,0);
-          //   mSquaresActor2->SetPosition(0,1,0);
-        }
-        else
-        {
-            mSquaresActor1->SetPosition(0,-1,0);
-            // mSquaresActor2->SetPosition(0,-1,0);
-        }
-        break;
-    case 4:
-        if (mCurrentSliceManager->GetSlicer(0)->GetRenderer()->GetActiveCamera()->GetPosition()[2] > slice)
-        {
-            mSquaresActor1->SetPosition(0,0,1);
-            // mSquaresActor2->SetPosition(0,0,1);
-        }
-        else
-        {
-            mSquaresActor1->SetPosition(0,0,-1);
-            // mSquaresActor2->SetPosition(0,0,-1);
-        }
-        break;
-    }
-    mSquares1->Update();
-   //  mSquares2->Update();
-
-    mCurrentSliceManager->Render(); 
+  if (!mInteractiveDisplayIsEnabled) return;
+  if (!mCurrentSlicerManager) close();
+  for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
+    mImageContour[i]->update(mThresholdSlider1->GetValue());
+  }
+  mCurrentSlicerManager->Render(); 
 }
 //------------------------------------------------------------------------------
 
@@ -234,43 +208,26 @@ void vvToolBinarize::GetArgsInfoFromGUI() {
   DD(good);
   */
 
+  mArgsInfo.imagetypes_flag = 0;
   mArgsInfo.upper_given = 0;
   mArgsInfo.lower_given = 0;
   bool inverseBGandFG = false;
 
-  if (mRadioButtonGreaterThan->isChecked()) { // Greater Than (and Lower Than)
-    mArgsInfo.lower_given = 1;
-    mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
-    DD(mArgsInfo.lower_arg);
-    if (mRadioButtonLowerThan->isChecked()) {
-      mArgsInfo.upper_given = 1;
-      mArgsInfo.upper_arg = mThresholdSlider2->GetValue();
-      if (mArgsInfo.upper_arg<mArgsInfo.lower_arg) {
-        mArgsInfo.upper_given = 0;
-        DD("TODO : lower thres greater than greater thres ! Ignoring ");
-      }
-    }
-  }
-  else {
-    if (mRadioButtonEqualThan->isChecked()) {
-      mArgsInfo.lower_given = 1;
-      mArgsInfo.upper_given = 1;
-      mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
-      mArgsInfo.upper_arg = mThresholdSlider1->GetValue();
-    }
-    else {
-      mArgsInfo.lower_given = 1;
-      mArgsInfo.upper_given = 1;
-      mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
-      mArgsInfo.upper_arg = mThresholdSlider1->GetValue();
-      inverseBGandFG = true;
+  mArgsInfo.lower_given = 1;
+  mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
+  // DD(mArgsInfo.lower_arg);
+  if (mRadioButtonLowerThan->isChecked()) {
+    mArgsInfo.upper_given = 1;
+    mArgsInfo.upper_arg = mThresholdSlider2->GetValue();
+    if (mArgsInfo.upper_arg<mArgsInfo.lower_arg) {
+      mArgsInfo.upper_given = 0;
+      DD("TODO : lower thres greater than greater thres ! Ignoring ");
     }
   }
 
   mArgsInfo.fg_arg = mFGSlider->GetValue();
   mArgsInfo.bg_arg = mBGSlider->GetValue();
 
-  DD(inverseBGandFG);
   if (inverseBGandFG) {
     mArgsInfo.fg_arg = mFGSlider->GetValue();
     mArgsInfo.bg_arg = mBGSlider->GetValue();
@@ -278,13 +235,13 @@ void vvToolBinarize::GetArgsInfoFromGUI() {
   mArgsInfo.fg_given = 1;
   mArgsInfo.bg_given = 1;
 
-  mArgsInfo.useBG_flag = mCheckBoxUseBG->isChecked();
-  mArgsInfo.useFG_flag = mCheckBoxUseFG->isChecked();
-
-  DD(mArgsInfo.useBG_flag);
-  DD(mArgsInfo.useFG_flag);
+  if (mCheckBoxUseBG->isChecked()) {
+    if (mCheckBoxUseFG->isChecked()) mArgsInfo.mode_arg = (char*)"both";
+    else mArgsInfo.mode_arg = (char*)"BG";
+  }
+  else mArgsInfo.mode_arg = (char*)"FG";
 
-  mArgsInfo.verbose_flag = true;
+  mArgsInfo.verbose_flag = false;
 
   // Required (even if not used)
   mArgsInfo.input_given = 0;
@@ -298,27 +255,23 @@ void vvToolBinarize::GetArgsInfoFromGUI() {
 
 //------------------------------------------------------------------------------
 void vvToolBinarize::apply() {
-  DD("Apply");
-
+  if (!mCurrentSlicerManager) close();
+  QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
   GetArgsInfoFromGUI();
 
-  DD(mArgsInfo.lower_arg);
-  // cmdline_parser2(argc, argv, &args_info, 1, 1, 0);                 
-  //   if (args_info.config_given)     
-  // cmdline_parser_configfile ("toto.config", &args_info, 0, 0, 1);
-  //   else cmdline_parser(argc, argv, &args_info);
-
   // Main filter
   clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::Pointer filter = 
-    clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::New();
+     clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::New();
   filter->SetArgsInfo(mArgsInfo);
   filter->SetInputVVImage(mCurrentImage);
   filter->Update();
 
-  // Output ???
+  // Output
   vvImage::Pointer output = filter->GetOutputVVImage();
-  DD(output->GetScalarTypeAsString());
-  CREATOR(vvToolBinarize)->mMainWindow->AddImage(output,"toto.mhd"); 
+  std::ostringstream osstream;
+  osstream << "Binarized_" << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
+  AddImage(output,osstream.str()); 
+  QApplication::restoreOverrideCursor();
   close();
 }
 //------------------------------------------------------------------------------
@@ -333,70 +286,13 @@ void vvToolBinarize::valueChangedT2(double v) {
 
 //------------------------------------------------------------------------------
 void vvToolBinarize::valueChangedT1(double v) {
+  if (!mCurrentSlicerManager) close();
   mThresholdSlider2->SetMinimum(v);
-  DD(v);
-  int m1 = (int)lrint(v);
-  DD(m1);  
-  int* extent = mCurrentSliceManager->GetSlicer(0)->GetImageActor()->GetDisplayExtent();
-  mClipper->SetOutputWholeExtent(extent[0],extent[1],extent[2],extent[3],extent[4],extent[5]);
-
-
-int slice = mCurrentSliceManager->GetSlicer(0)->GetSlice();
-
- int i;
-for (i = 0; i < 6;i = i+2)
-    {
-        if (extent[i] == extent[i+1])
-        {
-            break;
-        }
-    }
-
-    switch (i)
-    {
-    case 0:
-        if (mCurrentSliceManager->GetSlicer(0)->GetRenderer()->GetActiveCamera()->GetPosition()[0] > slice)
-        {
-            mSquaresActor1->SetPosition(1,0,0);
-            // mSquaresActor2->SetPosition(1,0,0);
-        }
-        else
-        {
-            mSquaresActor1->SetPosition(-1,0,0);
-            // mSquaresActor2->SetPosition(-1,0,0);
-        }
-        break;
-    case 2:
-        if (mCurrentSliceManager->GetSlicer(0)->GetRenderer()->GetActiveCamera()->GetPosition()[1] > slice)
-        {
-            mSquaresActor1->SetPosition(0,1,0);
-           //  mSquaresActor2->SetPosition(0,1,0);
-        }
-        else
-        {
-            mSquaresActor1->SetPosition(0,-1,0);
-            // mSquaresActor2->SetPosition(0,-1,0);
-        }
-        break;
-    case 4:
-        if (mCurrentSliceManager->GetSlicer(0)->GetRenderer()->GetActiveCamera()->GetPosition()[2] > slice)
-        {
-            mSquaresActor1->SetPosition(0,0,1);
-            // mSquaresActor2->SetPosition(0,0,1);
-        }
-        else
-        {
-            mSquaresActor1->SetPosition(0,0,-1);
-            // mSquaresActor2->SetPosition(0,0,-1);
-        }
-        break;
-    }
-
-
-
-
-  mSquares1->SetValue(0,m1);
-  mSquares1->Update();
-  mCurrentSliceManager->Render();
+  int m1 = (int)lrint(v);  
+  if (!mInteractiveDisplayIsEnabled) return;
+  for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
+    mImageContour[i]->update(m1);
+  }
+  mCurrentSlicerManager->Render();
 }
 //------------------------------------------------------------------------------