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