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