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
 183   // Output is uchar ...
 184   mFGSlider->SetMaximum(255);
 185   mFGSlider->SetMinimum(0);
 186   mBGSlider->SetMaximum(255);
 187   mBGSlider->SetMinimum(0);
 188
 189   mFGSlider->SetValue(1);
 190   mBGSlider->SetValue(0);
 191   mFGSlider->SetSingleStep(1);
 192   mBGSlider->SetSingleStep(1);
 193
 194   // VTK objects for interactive display
 195   for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 196     mImageContour.push_back(new vvImageContour);
 197     mImageContour[i]->setSlicer(mCurrentSlicerManager->GetSlicer(i));
 198     mImageContour[i]->setColor(1.0, 0.0, 0.0);
 199     mImageContourLower.push_back(new vvImageContour);
 200     mImageContourLower[i]->setSlicer(mCurrentSlicerManager->GetSlicer(i));
 201     mImageContourLower[i]->setColor(0.0, 0.0, 1.0);
 202   }
 203   valueChangedT1(mThresholdSlider1->GetValue());
 204
 205   connect(mThresholdSlider1, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT1(double)));
 206   connect(mThresholdSlider2, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT2(double)));
 207
 208   connect(mCurrentSlicerManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
 209   connect(mCurrentSlicerManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
 210
 211   //  connect(mCurrentSlicerManager, SIGNAL(LeftButtonReleaseSignal(int)), SLOT(LeftButtonReleaseEvent(int)));
 212 }
 213 //------------------------------------------------------------------------------
 214
 215
 216 //------------------------------------------------------------------------------
 217 // void vvToolBinarize::LeftButtonReleaseEvent(int slicer) {
 218 //   DD("LeftButtonReleaseEvent");
 219 //   for(int i=0; i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 220 //     if (i == slicer);
 221 //     mCurrentSlicerManager->GetSlicer(i)->GetRenderWindow()->Render();
 222 //   }
 223 // }
 224 //------------------------------------------------------------------------------
 225
 226
 227 //------------------------------------------------------------------------------
 228 void vvToolBinarize::UpdateSlice(int slicer,int slices) {
 229   if (!mInteractiveDisplayIsEnabled) return;
 230   if (!mCurrentSlicerManager) close();
 231   mImageContour[slicer]->update(mThresholdSlider1->GetValue());
 232   if (mRadioButtonLowerThan->isChecked())
 233     mImageContourLower[slicer]->update(mThresholdSlider2->GetValue());
 234   //  mCurrentSlicerManager->GetSlicer(slicer)->Render();
 235 }
 236 //------------------------------------------------------------------------------
 237
 238
 239 //------------------------------------------------------------------------------
 240 void vvToolBinarize::GetArgsInfoFromGUI() {
 241
 242   /* //KEEP THIS FOR READING GGO FROM FILE
 243     int argc=1;
 244     std::string a = "toto";
 245     char * const* argv = new char*;
 246     //a.c_str();
 247     struct cmdline_parser_params p;
 248     p.check_required = 0;
 249     int good = cmdline_parser_ext(argc, argv, &args_info, &p);
 250   DD(good);
 251   */
 252
 253   mArgsInfo.imagetypes_flag = 0;
 254   mArgsInfo.upper_given = 0;
 255   mArgsInfo.lower_given = 0;
 256   bool inverseBGandFG = false;
 257
 258   mArgsInfo.lower_given = 1;
 259   mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
 260   // DD(mArgsInfo.lower_arg);
 261   if (mRadioButtonLowerThan->isChecked()) {
 262     mArgsInfo.upper_given = 1;
 263     mArgsInfo.upper_arg = mThresholdSlider2->GetValue();
 264     if (mArgsInfo.upper_arg<mArgsInfo.lower_arg) {
 265       mArgsInfo.upper_given = 0;
 266       DD("TODO : lower thres greater than greater thres ! Ignoring ");
 267     }
 268   }
 269
 270   mArgsInfo.fg_arg = mFGSlider->GetValue();
 271   mArgsInfo.bg_arg = mBGSlider->GetValue();
 272
 273   if (inverseBGandFG) {
 274     mArgsInfo.fg_arg = mFGSlider->GetValue();
 275     mArgsInfo.bg_arg = mBGSlider->GetValue();
 276   }
 277   mArgsInfo.fg_given = 1;
 278   mArgsInfo.bg_given = 1;
 279
 280   if (mCheckBoxUseBG->isChecked()) {
 281     if (mCheckBoxUseFG->isChecked()) mArgsInfo.mode_arg = (char*)"both";
 282     else mArgsInfo.mode_arg = (char*)"BG";
 283   }
 284   else mArgsInfo.mode_arg = (char*)"FG";
 285
 286   mArgsInfo.verbose_flag = false;
 287
 288   // Required (even if not used)
 289   mArgsInfo.input_given = 0;
 290   mArgsInfo.output_given = 0;
 291
 292   mArgsInfo.input_arg = new char;
 293   mArgsInfo.output_arg = new char;
 294 }
 295 //------------------------------------------------------------------------------
 296
 297
 298 //------------------------------------------------------------------------------
 299 void vvToolBinarize::apply() {
 300   if (!mCurrentSlicerManager) close();
 301   QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
 302   GetArgsInfoFromGUI();
 303
 304   // Main filter
 305   clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::Pointer filter =
 306      clitk::BinarizeImageGenericFilter<args_info_clitkBinarizeImage>::New();
 307   filter->SetInputVVImage(mCurrentImage);
 308   filter->SetArgsInfo(mArgsInfo);
 309   filter->EnableReadOnDisk(false);
 310   filter->Update();
 311
 312   // Output
 313   vvImage::Pointer output = filter->GetOutputVVImage();
 314   std::ostringstream osstream;
 315   osstream << "Binarized_" << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
 316   AddImage(output,osstream.str());
 317   QApplication::restoreOverrideCursor();
 318   close();
 319 }
 320 //------------------------------------------------------------------------------
 321
 322
 323 //------------------------------------------------------------------------------
 324 void vvToolBinarize::valueChangedT2(double v) {
 325   //  DD("valueChangedT2");
 326   if (mRadioButtonLowerThan->isChecked()) {
 327     mThresholdSlider1->SetMaximum(v);
 328     if (!mInteractiveDisplayIsEnabled) return;
 329     for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 330       mImageContourLower[i]->update(v);
 331     }
 332     // mCurrentSlicerManager->Render();
 333   }
 334 }
 335 //------------------------------------------------------------------------------
 336
 337
 338 //------------------------------------------------------------------------------
 339 void vvToolBinarize::valueChangedT1(double v) {
 340   //  DD("valueChangedT1");
 341   if (!mCurrentSlicerManager) close();
 342   mThresholdSlider2->SetMinimum(v);
 343   //  int m1 = (int)lrint(v);
 344   if (!mInteractiveDisplayIsEnabled) return;
 345   for(int i=0;i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 346     mImageContour[i]->update(v);
 347   }
 348   // mCurrentSlicerManager->Render();
 349 }
 350 //------------------------------------------------------------------------------