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://www.centreleonberard.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 // vv
  20 #include "vvToolBinarize.h"
  21 #include "vvSlicerManager.h"
  22 #include "vvSlicer.h"
  23 #include "vvToolInputSelectorWidget.h"
  24
  25 // clitk
  26 #include "clitkBinarizeImageGenericFilter.h"
  27
  28 // vtk
  29 #include <vtkImageActor.h>
  30 #include <vtkCamera.h>
  31 #include <vtkImageClip.h>
  32 #include <vtkRenderWindow.h>
  33
  34 //------------------------------------------------------------------------------
  35 // Create the tool and automagically (I like this word) insert it in
  36 // the main window menu.
  37 ADD_TOOL(vvToolBinarize);
  38 //------------------------------------------------------------------------------
  39
  40
  41 //------------------------------------------------------------------------------
  42 void vvToolBinarize::Initialize()
  43 { cout << __func__ << endl;
  44   SetToolName("Binarize");
  45   SetToolMenuName("Binarize");
  46   SetToolIconFilename(":/common/icons/binarize.png");
  47   SetToolTip("Image interactive binarization with thresholds.");
  48 }
  49 //------------------------------------------------------------------------------
  50
  51
  52 //------------------------------------------------------------------------------
  53 vvToolBinarize::vvToolBinarize(vvMainWindowBase * parent, Qt::WindowFlags f)
  54   :vvToolWidgetBase(parent,f),
  55    vvToolBase<vvToolBinarize>(parent),
  56    Ui::vvToolBinarize()
  57 { cout << __func__ << endl;
  58   // GUI Initialization
  59   Ui_vvToolBinarize::setupUi(mToolWidget);
  60   mInteractiveDisplayIsEnabled = mCheckBoxInteractiveDisplay->isChecked();
  61
  62   // Connect signals & slots
  63   connect(mRadioButtonLowerThan, SIGNAL(toggled(bool)), this, SLOT(enableLowerThan(bool)));
  64   connect(mCheckBoxUseFG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
  65   connect(mCheckBoxUseBG, SIGNAL(toggled(bool)), this, SLOT(useFGBGtoggled(bool)));
  66   connect(mCheckBoxInteractiveDisplay, SIGNAL(toggled(bool)), this, SLOT(InteractiveDisplayToggled(bool)));
  67
  68   // Initialize some widget
  69   mThresholdSlider1->SetText("");
  70   mThresholdSlider2->SetText("");
  71   mFGSlider->SetText("Foreground value");
  72   mBGSlider->SetText("Background value");
  73
  74   // Main filter
  75   mFilter = clitk::BinarizeImageGenericFilter::New();
  76
  77   // Set how many inputs are needed for this tool
  78   AddInputSelector("Select one image", mFilter);
  79 }
  80 //------------------------------------------------------------------------------
  81
  82
  83 //------------------------------------------------------------------------------
  84 vvToolBinarize::~vvToolBinarize()
  85 { cout << __func__ << endl;
  86 }
  87 //------------------------------------------------------------------------------
  88
  89
  90 //------------------------------------------------------------------------------
  91 void vvToolBinarize::InteractiveDisplayToggled(bool b)
  92 { cout << __func__ << endl;
  93   mInteractiveDisplayIsEnabled = b;
  94   if (!mInteractiveDisplayIsEnabled) {
  95     RemoveVTKObjects();
  96   } else {
  97     for(unsigned int i=0; i<mImageContour.size(); i++) {
  98       mImageContour[i]->ShowActors();
  99       mImageContour[i]->Print(cout);
 100       if (mRadioButtonLowerThan->isChecked())
 101         mImageContourLower[i]->ShowActors();
 102     }
 103     if (mCurrentSlicerManager)
 104       mCurrentSlicerManager->Render();
 105   }
 106 }
 107 //------------------------------------------------------------------------------
 108
 109
 110 //------------------------------------------------------------------------------
 111 void vvToolBinarize::RemoveVTKObjects()
 112 { cout << __func__ << endl;
 113   for(unsigned int i=0; i<mImageContour.size(); i++) {
 114     mImageContour[i]->HideActors();
 115     mImageContourLower[i]->HideActors();
 116   }
 117   if (mCurrentSlicerManager)
 118     mCurrentSlicerManager->Render();
 119 }
 120 //------------------------------------------------------------------------------
 121
 122
 123 //------------------------------------------------------------------------------
 124 bool vvToolBinarize::close()
 125 { cout << __func__ << endl;
 126   // RemoveVTKObjects();
 127   return vvToolWidgetBase::close();
 128 }
 129 //------------------------------------------------------------------------------
 130
 131
 132 //------------------------------------------------------------------------------
 133 void vvToolBinarize::closeEvent(QCloseEvent *event)
 134 { cout << __func__ << endl;
 135   RemoveVTKObjects();
 136   event->accept();
 137 }
 138 //------------------------------------------------------------------------------
 139
 140
 141 //------------------------------------------------------------------------------
 142 void vvToolBinarize::reject()
 143 { cout << __func__ << endl;
 144   // DD("vvToolBinarize::reject");
 145   RemoveVTKObjects();
 146   return vvToolWidgetBase::reject();
 147 }
 148 //------------------------------------------------------------------------------
 149
 150
 151 //------------------------------------------------------------------------------
 152 void vvToolBinarize::enableLowerThan(bool b)
 153 { cout << __func__ << endl;
 154   if (!b) {
 155     mThresholdSlider1->resetMaximum();
 156     for(unsigned int i=0; i<mImageContour.size(); i++) {
 157       mImageContourLower[i]->HideActors();
 158     }
 159     mCurrentSlicerManager->Render();
 160   } else {
 161     valueChangedT1(mThresholdSlider1->GetValue());
 162     valueChangedT2(mThresholdSlider2->GetValue());
 163     for(unsigned int i=0; i<mImageContour.size(); i++) {
 164       mImageContourLower[i]->ShowActors();
 165     }
 166     mCurrentSlicerManager->Render();
 167   }
 168 }
 169 //------------------------------------------------------------------------------
 170
 171
 172 //------------------------------------------------------------------------------
 173 void vvToolBinarize::useFGBGtoggled(bool)
 174 { cout << __func__ << endl;
 175   if (!mCheckBoxUseBG->isChecked() && !mCheckBoxUseFG->isChecked())
 176     mCheckBoxUseBG->toggle();
 177 }
 178 //------------------------------------------------------------------------------
 179
 180
 181 //------------------------------------------------------------------------------
 182 // void vvToolBinarize::InputIsSelected(std::vector<vvSlicerManager *> & m) {
 183 //   DD("vvToolBinarize::InputIsSelected vector in binarize");
 184 //   DD(m.size());
 185 // }
 186 void vvToolBinarize::InputIsSelected(vvSlicerManager * m)
 187 { cout << __func__ << endl;
 188   mCurrentSlicerManager = m;
 189
 190   // Specific for this gui
 191   mThresholdSlider1->SetValue(0);
 192   mThresholdSlider2->SetValue(0);
 193   mThresholdSlider1->SetImage(mCurrentImage);
 194   mThresholdSlider2->SetImage(mCurrentImage);
 195   mFGSlider->SetImage(mCurrentImage);
 196   mBGSlider->SetImage(mCurrentImage);
 197   //  DD(mCurrentSlicerManager->GetFileName().c_str());
 198   //  mFGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
 199   //   mFGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
 200   //   mBGSlider->SetMaximum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMax());
 201   //   mBGSlider->SetMinimum(mCurrentImage->GetFirstVTKImageData()->GetScalarTypeMin());
 202
 203   // Output is uchar ...
 204   mFGSlider->SetMaximum(255);
 205   mFGSlider->SetMinimum(0);
 206   mBGSlider->SetMaximum(255);
 207   mBGSlider->SetMinimum(0);
 208
 209   mFGSlider->SetValue(1);
 210   mBGSlider->SetValue(0);
 211   mFGSlider->SetSingleStep(1);
 212   mBGSlider->SetSingleStep(1);
 213
 214   // VTK objects for interactive display
 215   for(int i=0; i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
 216     mImageContour.push_back(vvImageContour::New());
 217     mImageContour[i]->SetSlicer(mCurrentSlicerManager->GetSlicer(i));
 218     mImageContour[i]->SetColor(1.0, 0.0, 0.0);
 219     mImageContour[i]->SetDepth(100); // to be in front of (whe used with ROI tool)
 220     mImageContourLower.push_back(vvImageContour::New());
 221     mImageContourLower[i]->SetSlicer(mCurrentSlicerManager->GetSlicer(i));
 222     mImageContourLower[i]->SetColor(0.0, 0.0, 1.0);
 223     mImageContourLower[i]->SetDepth(100); // to be in front of (whe used with ROI tool)
 224     mImageContour[i]->Print(cout);
 225   }
 226   valueChangedT1(mThresholdSlider1->GetValue());
 227
 228   connect(mThresholdSlider1, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT1(double)));
 229   connect(mThresholdSlider2, SIGNAL(valueChanged(double)), this, SLOT(valueChangedT2(double)));
 230
 231   connect(mCurrentSlicerManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
 232   connect(mCurrentSlicerManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateSlice(int, int)));
 233
 234   connect(mCurrentSlicerManager,SIGNAL(UpdateOrientation(int,int)),this,SLOT(UpdateOrientation(int, int)));
 235
 236   //  connect(mCurrentSlicerManager, SIGNAL(LeftButtonReleaseSignal(int)), SLOT(LeftButtonReleaseEvent(int)));
 237   cout << mInteractiveDisplayIsEnabled << endl;
 238   InteractiveDisplayToggled(mInteractiveDisplayIsEnabled);
 239 }
 240 //------------------------------------------------------------------------------
 241
 242
 243 //------------------------------------------------------------------------------
 244 // void vvToolBinarize::LeftButtonReleaseEvent(int slicer) {
 245 //   DD("LeftButtonReleaseEvent");
 246 //   for(int i=0; i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
 247 //     if (i == slicer);
 248 //     mCurrentSlicerManager->GetSlicer(i)->GetRenderWindow()->Render();
 249 //   }
 250 // }
 251 //------------------------------------------------------------------------------
 252
 253
 254 //------------------------------------------------------------------------------
 255 void vvToolBinarize::UpdateOrientation(int slicer,int orientation)
 256 { cout << __func__ << endl;
 257   Update(slicer);
 258 }
 259 //------------------------------------------------------------------------------
 260
 261 //------------------------------------------------------------------------------
 262 void vvToolBinarize::UpdateSlice(int slicer,int slices)
 263 { cout << __func__ << endl;
 264   Update(slicer);
 265 }
 266 //------------------------------------------------------------------------------
 267
 268 //------------------------------------------------------------------------------
 269 void vvToolBinarize::Update(int slicer)
 270 { cout << __func__ << endl;
 271   if (!mInteractiveDisplayIsEnabled) return;
 272   if (!mCurrentSlicerManager) close();
 273   mImageContour[slicer]->Update(mThresholdSlider1->GetValue());
 274   if (mRadioButtonLowerThan->isChecked())
 275     mImageContourLower[slicer]->Update(mThresholdSlider2->GetValue());
 276 }
 277 //------------------------------------------------------------------------------
 278
 279 //------------------------------------------------------------------------------
 280 void vvToolBinarize::GetArgsInfoFromGUI()
 281 { cout << __func__ << endl;
 282
 283   /* //KEEP THIS FOR READING GGO FROM FILE
 284      int argc=1;
 285      std::string a = "toto";
 286      char * const* argv = new char*;
 287      //a.c_str();
 288      struct cmdline_parser_params p;
 289      p.check_required = 0;
 290      int good = cmdline_parser_ext(argc, argv, &args_info, &p);
 291      DD(good);
 292   */
 293   cmdline_parser_clitkBinarizeImage_init(&mArgsInfo); // Initialisation to default
 294   bool inverseBGandFG = false;
 295
 296   mArgsInfo.lower_given = 1;
 297   mArgsInfo.lower_arg = mThresholdSlider1->GetValue();
 298   if (mRadioButtonLowerThan->isChecked()) {
 299     mArgsInfo.upper_given = 1;
 300     mArgsInfo.upper_arg = mThresholdSlider2->GetValue();
 301     if (mArgsInfo.upper_arg<mArgsInfo.lower_arg) {
 302       mArgsInfo.upper_given = 0;
 303       DD("TODO : lower thres greater than greater thres ! Ignoring ");
 304     }
 305   }
 306
 307   mArgsInfo.fg_arg = mFGSlider->GetValue();
 308   mArgsInfo.bg_arg = mBGSlider->GetValue();
 309
 310   if (inverseBGandFG) {
 311     mArgsInfo.fg_arg = mFGSlider->GetValue();
 312     mArgsInfo.bg_arg = mBGSlider->GetValue();
 313   }
 314   mArgsInfo.fg_given = 1;
 315   mArgsInfo.bg_given = 1;
 316
 317   if (mCheckBoxUseBG->isChecked()) {
 318     if (mCheckBoxUseFG->isChecked()) mArgsInfo.mode_arg = (char*)"both";
 319     else mArgsInfo.mode_arg = (char*)"BG";
 320   } else mArgsInfo.mode_arg = (char*)"FG";
 321
 322   mArgsInfo.verbose_flag = false;
 323
 324   // // Required (even if not used)
 325   // mArgsInfo.input_given = 0;
 326   // mArgsInfo.output_given = 0;
 327
 328   // mArgsInfo.input_arg = new char;
 329   // mArgsInfo.output_arg = new char;
 330 }
 331 //------------------------------------------------------------------------------
 332
 333
 334 //------------------------------------------------------------------------------
 335 void vvToolBinarize::apply()
 336 { cout << __func__ << endl;
 337   if (!mCurrentSlicerManager) close();
 338   QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
 339   GetArgsInfoFromGUI();
 340
 341   // Main filter
 342   clitk::BinarizeImageGenericFilter::Pointer filter =
 343     clitk::BinarizeImageGenericFilter::New();
 344   filter->SetInputVVImage(mCurrentImage);
 345   filter->SetArgsInfo(mArgsInfo);
 346   filter->Update();
 347
 348   // Output
 349   vvImage::Pointer output = filter->GetOutputVVImage();
 350   std::ostringstream osstream;
 351   osstream << "Binarized_" << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
 352   AddImage(output,osstream.str());
 353   QApplication::restoreOverrideCursor();
 354   close();
 355 }
 356 //------------------------------------------------------------------------------
 357
 358
 359 //------------------------------------------------------------------------------
 360 void vvToolBinarize::valueChangedT2(double v)
 361 { cout << __func__ << endl;
 362   //  DD("valueChangedT2");
 363   if (mRadioButtonLowerThan->isChecked()) {
 364     mThresholdSlider1->SetMaximum(v);
 365     if (!mInteractiveDisplayIsEnabled) return;
 366     for(int i=0;i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
 367       mImageContourLower[i]->Update(v);
 368     }
 369     mCurrentSlicerManager->Render();
 370   }
 371 }
 372 //------------------------------------------------------------------------------
 373
 374
 375 //------------------------------------------------------------------------------
 376 void vvToolBinarize::valueChangedT1(double v)
 377 { cout << __func__ << endl;
 378   //  DD("valueChangedT1");
 379   if (!mCurrentSlicerManager) close();
 380   mThresholdSlider2->SetMinimum(v);
 381   //  int m1 = (int)lrint(v);
 382   if (!mInteractiveDisplayIsEnabled) return;
 383   for(int i=0;i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
 384     mImageContour[i]->Update(v);
 385   }
 386   mCurrentSlicerManager->Render();
 387 }
 388 //------------------------------------------------------------------------------