vv/vvToolBinarize.cxx

   1 /*=========================================================================
   2   Program:   vv                     http://www.creatis.insa-lyon.fr/rio/vv
   3
   4   Authors belong to:
   5   - University of LYON              http://www.universite-lyon.fr/
   6   - Léon Bérard cancer center       http://oncora1.lyon.fnclcc.fr
   7   - CREATIS CNRS laboratory         http://www.creatis.insa-lyon.fr
   8
   9   This software is distributed WITHOUT ANY WARRANTY; without even
  10   the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  11   PURPOSE.  See the copyright notices for more information.
  12
  13   It is distributed under dual licence
  14
  15   - BSD        See included LICENSE.txt file
  16   - CeCILL-B   http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html
  17 ======================================================================-====*/
  18
  19 #include "vvToolBinarize.h"
  20 #include "vvSlicerManager.h"
  21 #include "vvSlicer.h"
  22 #include "vvToolInputSelectorWidget.h"
  23
  24 #include "clitkBinarizeImageGenericFilter.h"
  25
  26 #include <vtkImageActor.h>
  27 #include <vtkCamera.h>
  28 #include <vtkImageClip.h>
  29 #include <vtkRenderWindow.h>
  30
  31 //------------------------------------------------------------------------------
  32 // Create the tool and automagically (I like this word) insert it in
  33 // the main window menu.
  34 ADD_TOOL(vvToolBinarize);
  35 //------------------------------------------------------------------------------
  36
  37
  38 //------------------------------------------------------------------------------
  39 void vvToolBinarize::Initialize() {
  40   SetToolName("Binarize");
  41   SetToolMenuName("Binarize");
  42   SetToolIconFilename(":/common/icons/binarize.png");
  43   SetToolTip("Image interactive binarization with thresholds.");
  44 }
  45 //------------------------------------------------------------------------------
  46
  47
  48 //------------------------------------------------------------------------------
  49 vvToolBinarize::vvToolBinarize(vvMainWindowBase * parent, Qt::WindowFlags f)
  50   :vvToolWidgetBase(parent,f),
  51    vvToolBase<vvToolBinarize>(parent),
  52    Ui::vvToolBinarize()
  53 {
  54   // GUI Initialization
  55   Ui_vvToolBinarize::setupUi(mToolWidget);
  56   mInteractiveDisplayIsEnabled = mCheckBoxInteractiveDisplay->isChecked();
  57
  58   // Connect signals & slots
  59   connect(mRadioButtonLowerThan, SIGNAL(toggled(bool)), this, SLOT(enableLowerThan(bool)));
  60   connect(mCheckBoxUseFG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
  61   connect(mCheckBoxUseBG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
  62   connect(mCheckBoxInteractiveDisplay, SIGNAL(toggled(bool)), this, SLOT(InteractiveDisplayToggled(bool)));
  63
  64   // Initialize some widget
  65   mThresholdSlider1->SetText("");
  66   mThresholdSlider2->SetText("");
  67   mFGSlider->SetText("Foreground value");
  68   mBGSlider->SetText("Background value");
  69
  70   // Main filter
  71   mFilter = new clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>;
  72
  73   // Set how many inputs are needed for this tool
  74   AddInputSelector("Select one image", mFilter);
  75 }
  76 //------------------------------------------------------------------------------
  77
  78
  79 //------------------------------------------------------------------------------
  80 vvToolBinarize::~vvToolBinarize() {
  81 }
  82 //------------------------------------------------------------------------------
  83
  84
  85 //------------------------------------------------------------------------------
  86 void vvToolBinarize::InteractiveDisplayToggled(bool b) {
  87   mInteractiveDisplayIsEnabled = b;
  88   if (!mInteractiveDisplayIsEnabled) {
  89     RemoveVTKObjects();
  90   }
  91   else {
  92     for(unsigned int i=0; i<mImageContour.size(); i++) {
  93       mImageContour[i]->showActors();
  94       if (mRadioButtonLowerThan->isChecked())
  95         mImageContourLower[i]->showActors();
  96     }
  97     if (mCurrentSlicerManager)
  98       mCurrentSlicerManager->Render();
  99   }
 100 }
 101 //------------------------------------------------------------------------------
 102
 103
 104 //------------------------------------------------------------------------------
 105 void vvToolBinarize::RemoveVTKObjects() {
 106   for(unsigned int i=0; i<mImageContour.size(); i++) {
 107     mImageContour[i]->hideActors();
 108     mImageContourLower[i]->hideActors();
 109   }
 110   if (mCurrentSlicerManager)
 111     mCurrentSlicerManager->Render();
 112 }
 113 //------------------------------------------------------------------------------
 114
 115
 116 //------------------------------------------------------------------------------
 117 bool vvToolBinarize::close() {
 118   RemoveVTKObjects();
 119   return vvToolWidgetBase::close();
 120 }
 121 //------------------------------------------------------------------------------
 122
 123
 124 //------------------------------------------------------------------------------
 125 void vvToolBinarize::reject() {
 126   // DD("vvToolBinarize::reject");
 127   RemoveVTKObjects();
 128   return vvToolWidgetBase::reject();
 129 }
 130 //------------------------------------------------------------------------------
 131
 132
 133 //------------------------------------------------------------------------------
 134 void vvToolBinarize::enableLowerThan(bool b) {
 135   if (!b) {
 136     mThresholdSlider1->resetMaximum();
 137     for(unsigned int i=0; i<mImageContour.size(); i++) {
 138       mImageContourLower[i]->hideActors();
 139     }
 140     mCurrentSlicerManager->Render();
 141   }
 142   else {
 143     valueChangedT1(mThresholdSlider1->GetValue());
 144     valueChangedT2(mThresholdSlider2->GetValue());
 145     for(unsigned int i=0; i<mImageContour.size(); i++) {
 146       mImageContourLower[i]->showActors();
 147     }
 148     mCurrentSlicerManager->Render();
 149   }
 150 }
 151 //------------------------------------------------------------------------------
 152
 153
 154 //------------------------------------------------------------------------------
 155 void vvToolBinarize::useFGBGtoggled(bool) {
 156   if (!mCheckBoxUseBG->isChecked() && !mCheckBoxUseFG->isChecked())
 157     mCheckBoxUseBG->toggle();
 158 }
 159 //------------------------------------------------------------------------------
 160
 161
 162 //------------------------------------------------------------------------------
 163 // void vvToolBinarize::InputIsSelected(std::vector<vvSlicerManager *> & m) {
 164 //   DD("vvToolBinarize::InputIsSelected vector in binarize");
 165 //   DD(m.size());
 166 // }
 167 void vvToolBinarize::InputIsSelected(vvSlicerManager * m) {
 168   mCurrentSlicerManager = m;
 169
 170   // Specific for this gui
 171   mThresholdSlider1->SetValue(0);
 172   mThresholdSlider2->SetValue(0);
 173   mThresholdSlider1->SetImage(mCurrentImage);
 174   mThresholdSlider2->SetImage(mCurrentImage);
 175   mFGSlider->SetImage(mCurrentImage);
 176   mBGSlider->SetImage(mCurrentImage);
 177   //  DD(mCurrentSlicerManager->GetFileName().c_str());
 178   mFGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
 179   mFGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
 180   mBGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
 181   mBGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
 182   mFGSlider->SetValue(1);
 183   mBGSlider->SetValue(0);
 184   mFGSlider->SetSingleStep(1);
 185   mBGSlider->SetSingleStep(1);
 186
 187   // VTK objects for interactive display
 188   for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 189     mImageContour.push_back(new vvImageContour);
 190     mImageContour[i]->setSlicer(mCurrentSlicerManager->GetSlicer(i));
 191     mImageContour[i]->setColor(1.0, 0.0, 0.0);
 192     mImageContourLower.push_back(new vvImageContour);
 193     mImageContourLower[i]->setSlicer(mCurrentSlicerManager->GetSlicer(i));
 194     mImageContourLower[i]->setColor(0.0, 0.0, 1.0);
 195   }
 196   valueChangedT1(mThresholdSlider1->GetValue());
 197
 198   connect(mThresholdSlider1, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT1(double)));
 199   connect(mThresholdSlider2, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT2(double)));
 200
 201   connect(mCurrentSlicerManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
 202   connect(mCurrentSlicerManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
 203
 204   //  connect(mCurrentSlicerManager, SIGNAL(LeftButtonReleaseSignal(int)), SLOT(LeftButtonReleaseEvent(int)));
 205 }
 206 //------------------------------------------------------------------------------
 207
 208
 209 //------------------------------------------------------------------------------
 210 // void vvToolBinarize::LeftButtonReleaseEvent(int slicer) {
 211 //   DD("LeftButtonReleaseEvent");
 212 //   for(int i=0; i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 213 //     if (i == slicer);
 214 //     mCurrentSlicerManager->GetSlicer(i)->GetRenderWindow()->Render();
 215 //   }
 216 // }
 217 //------------------------------------------------------------------------------
 218
 219
 220 //------------------------------------------------------------------------------
 221 void vvToolBinarize::UpdateSlice(int slicer,int slices) {
 222   if (!mInteractiveDisplayIsEnabled) return;
 223   if (!mCurrentSlicerManager) close();
 224   for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 225     //    DD(i);
 226     mImageContour[i]->update(mThresholdSlider1->GetValue());
 227     if (mRadioButtonLowerThan->isChecked())
 228       mImageContourLower[i]->update(mThresholdSlider2->GetValue());
 229
 230   }
 231   mCurrentSlicerManager->Render();
 232 }
 233 //------------------------------------------------------------------------------
 234
 235
 236 //------------------------------------------------------------------------------
 237 void vvToolBinarize::GetArgsInfoFromGUI() {
 238
 239   /* //KEEP THIS FOR READING GGO FROM FILE
 240     int argc=1;
 241     std::string a = "toto";
 242     char * const* argv = new char*;
 243     //a.c_str();
 244     struct cmdline_parser_params p;
 245     p.check_required = 0;
 246     int good = cmdline_parser_ext(argc, argv, &args_info, &p);
 247   DD(good);
 248   */
 249
 250   mArgsInfo.imagetypes_flag = 0;
 251   mArgsInfo.upper_given = 0;
 252   mArgsInfo.lower_given = 0;
 253   bool inverseBGandFG = false;
 254
 255   mArgsInfo.lower_given = 1;
 256   mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
 257   // DD(mArgsInfo.lower_arg);
 258   if (mRadioButtonLowerThan->isChecked()) {
 259     mArgsInfo.upper_given = 1;
 260     mArgsInfo.upper_arg = mThresholdSlider2->GetValue();
 261     if (mArgsInfo.upper_arg<mArgsInfo.lower_arg) {
 262       mArgsInfo.upper_given = 0;
 263       DD("TODO : lower thres greater than greater thres ! Ignoring ");
 264     }
 265   }
 266
 267   mArgsInfo.fg_arg = mFGSlider->GetValue();
 268   mArgsInfo.bg_arg = mBGSlider->GetValue();
 269
 270   if (inverseBGandFG) {
 271     mArgsInfo.fg_arg = mFGSlider->GetValue();
 272     mArgsInfo.bg_arg = mBGSlider->GetValue();
 273   }
 274   mArgsInfo.fg_given = 1;
 275   mArgsInfo.bg_given = 1;
 276
 277   if (mCheckBoxUseBG->isChecked()) {
 278     if (mCheckBoxUseFG->isChecked()) mArgsInfo.mode_arg = (char*)"both";
 279     else mArgsInfo.mode_arg = (char*)"BG";
 280   }
 281   else mArgsInfo.mode_arg = (char*)"FG";
 282
 283   mArgsInfo.verbose_flag = false;
 284
 285   // Required (even if not used)
 286   mArgsInfo.input_given = 0;
 287   mArgsInfo.output_given = 0;
 288
 289   mArgsInfo.input_arg = new char;
 290   mArgsInfo.output_arg = new char;
 291 }
 292 //------------------------------------------------------------------------------
 293
 294
 295 //------------------------------------------------------------------------------
 296 void vvToolBinarize::apply() {
 297   if (!mCurrentSlicerManager) close();
 298   QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
 299   GetArgsInfoFromGUI();
 300
 301   // Main filter
 302   clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::Pointer filter =
 303      clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::New();
 304   filter->SetInputVVImage(mCurrentImage);
 305   filter->SetArgsInfo(mArgsInfo);
 306   filter->EnableReadOnDisk(false);
 307   filter->Update();
 308
 309   // Output
 310   vvImage::Pointer output = filter->GetOutputVVImage();
 311   std::ostringstream osstream;
 312   osstream << "Binarized_" << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
 313   AddImage(output,osstream.str());
 314   QApplication::restoreOverrideCursor();
 315   close();
 316 }
 317 //------------------------------------------------------------------------------
 318
 319
 320 //------------------------------------------------------------------------------
 321 void vvToolBinarize::valueChangedT2(double v) {
 322   //  DD("valueChangedT2");
 323   if (mRadioButtonLowerThan->isChecked()) {
 324     mThresholdSlider1->SetMaximum(v);
 325     if (!mInteractiveDisplayIsEnabled) return;
 326     for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 327       mImageContourLower[i]->update(v);
 328     }
 329     // mCurrentSlicerManager->Render();
 330   }
 331 }
 332 //------------------------------------------------------------------------------
 333
 334
 335 //------------------------------------------------------------------------------
 336 void vvToolBinarize::valueChangedT1(double v) {
 337   //  DD("valueChangedT1");
 338   if (!mCurrentSlicerManager) close();
 339   mThresholdSlider2->SetMinimum(v);
 340   //  int m1 = (int)lrint(v);
 341   if (!mInteractiveDisplayIsEnabled) return;
 342   for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 343     mImageContour[i]->update(v);
 344   }
 345   // mCurrentSlicerManager->Render();
 346 }
 347 //------------------------------------------------------------------------------