vv/vvToolCropImage.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 "vvToolCropImage.h"
  20 #include <QComboBox>
  21 #include <QCursor>
  22 #include <QApplication>
  23 #include <vtkImageClip.h>
  24 #include <vtkImageTranslateExtent.h>
  25 #include <vtkImageData.h>
  26 #include <vtkSmartPointer.h>
  27 #include "vvSlicer.h"
  28
  29 //------------------------------------------------------------------------------
  30 // Create the tool and automagically (I like this word) insert it in
  31 // the main window menu.
  32 ADD_TOOL(vvToolCropImage);
  33 //------------------------------------------------------------------------------
  34
  35
  36 //------------------------------------------------------------------------------
  37 vvToolCropImage::vvToolCropImage(vvMainWindowBase * parent, Qt::WindowFlags f):
  38   vvToolWidgetBase(parent, f),
  39   vvToolBase<vvToolCropImage>(parent),
  40   Ui::vvToolCropImage()
  41 {
  42
  43   // GUI Initialization
  44   Ui_vvToolCropImage::setupUi(mToolWidget);
  45
  46   // Set how many inputs are needed for this tool
  47   AddInputSelector("Select one image");
  48 }
  49 //------------------------------------------------------------------------------
  50
  51
  52 //------------------------------------------------------------------------------
  53 vvToolCropImage::~vvToolCropImage()
  54 {
  55
  56 }
  57 //------------------------------------------------------------------------------
  58
  59
  60 //------------------------------------------------------------------------------
  61 bool vvToolCropImage::close()
  62 {
  63   for(int i=0; i<mExtentSize; i++) mReducedExtent[i] = mInitialExtent[i];
  64   UpdateExtent();
  65   return vvToolWidgetBase::close();
  66 }
  67 //------------------------------------------------------------------------------
  68
  69
  70 //------------------------------------------------------------------------------
  71 void vvToolCropImage::reject()
  72 {
  73   for(int i=0; i<mExtentSize; i++) mReducedExtent[i] = mInitialExtent[i];
  74   UpdateExtent();
  75   return vvToolWidgetBase::reject();
  76 }
  77 //------------------------------------------------------------------------------
  78
  79
  80 //------------------------------------------------------------------------------
  81 void vvToolCropImage::sliderXMinValueChanged(int s)
  82 {
  83   xmaxSlider->setMinimum(xminSlider->value());
  84   mReducedExtent[0] = xminSlider->value();
  85   UpdateExtent();
  86 }
  87 //------------------------------------------------------------------------------
  88
  89
  90 //------------------------------------------------------------------------------
  91 void vvToolCropImage::sliderXMaxValueChanged(int s)
  92 {
  93   xminSlider->setMaximum(xmaxSlider->value());
  94   mReducedExtent[1] = xmaxSlider->value();
  95   UpdateExtent();
  96 }
  97 //------------------------------------------------------------------------------
  98
  99
 100 //------------------------------------------------------------------------------
 101 void vvToolCropImage::sliderYMinValueChanged(int s)
 102 {
 103   ymaxSlider->setMinimum(yminSlider->value());
 104   mReducedExtent[2] = yminSlider->value();
 105   UpdateExtent();
 106 }
 107 //------------------------------------------------------------------------------
 108
 109
 110 //------------------------------------------------------------------------------
 111 void vvToolCropImage::sliderYMaxValueChanged(int s)
 112 {
 113   yminSlider->setMaximum(ymaxSlider->value());
 114   mReducedExtent[3] = ymaxSlider->value();
 115   UpdateExtent();
 116 }
 117 //------------------------------------------------------------------------------
 118
 119
 120 //------------------------------------------------------------------------------
 121 void vvToolCropImage::sliderZMinValueChanged(int s)
 122 {
 123   zmaxSlider->setMinimum(zminSlider->value());
 124   mReducedExtent[4] = zminSlider->value();
 125   UpdateExtent();
 126 }
 127 //------------------------------------------------------------------------------
 128
 129
 130 //------------------------------------------------------------------------------
 131 void vvToolCropImage::sliderZMaxValueChanged(int s)
 132 {
 133   zminSlider->setMaximum(zmaxSlider->value());
 134   mReducedExtent[5] = zmaxSlider->value();
 135   UpdateExtent();
 136 }
 137 //------------------------------------------------------------------------------
 138
 139
 140 //------------------------------------------------------------------------------
 141 // void vvToolCropImage::sliderTMinValueChanged(int s) {
 142 //   tmaxSlider->setMinimum(tminSlider->value());
 143 //   mReducedExtent[6] = tminSlider->value();
 144 //   UpdateExtent();
 145 // }
 146 //------------------------------------------------------------------------------
 147
 148
 149 //------------------------------------------------------------------------------
 150 // void vvToolCropImage::sliderTMaxValueChanged(int s) {
 151 //   tminSlider->setMaximum(tmaxSlider->value());
 152 //   mReducedExtent[7] = tmaxSlider->value();
 153 //   UpdateExtent();
 154 // }
 155 //------------------------------------------------------------------------------
 156
 157
 158 //------------------------------------------------------------------------------
 159 void vvToolCropImage::UpdateExtent()
 160 {
 161   for(int i=0; i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 162     mCurrentSlicerManager->GetSlicer(i)->SetReducedExtent(mReducedExtent);
 163     mCurrentSlicerManager->GetSlicer(i)->ForceUpdateDisplayExtent();
 164     mCurrentSlicerManager->GetSlicer(i)->Render();
 165   }
 166 }
 167 //------------------------------------------------------------------------------
 168
 169
 170 //------------------------------------------------------------------------------
 171 void vvToolCropImage::InputIsSelected(vvSlicerManager * slicer)
 172 {
 173
 174   // Change interface according to number of dimension
 175   mExtentSize = 2*slicer->GetDimension();
 176   // if (slicer->GetDimension()<4) {
 177   //   mTLabel1->setHidden(true);
 178   //   mTLabel2->setHidden(true);
 179   //   tminSlider->setHidden(true);
 180   //   tmaxSlider->setHidden(true);
 181   //   spin_tmin->setHidden(true);
 182   //   spin_tmax->setHidden(true);
 183   // }
 184   if (slicer->GetDimension()<3) {
 185     mZLabel1->setHidden(true);
 186     mZLabel2->setHidden(true);
 187     zminSlider->setHidden(true);
 188     zmaxSlider->setHidden(true);
 189     spin_zmin->setHidden(true);
 190     spin_zmax->setHidden(true);
 191   }
 192
 193   // Record initial extend
 194   mReducedExtent = new int[mExtentSize];
 195   mInitialExtent = new int[mExtentSize];
 196   mReducedExtent = mCurrentSlicerManager->GetImage()->GetFirstVTKImageData()->GetWholeExtent();
 197   for(int i=0; i<mExtentSize; i++) mInitialExtent[i] = mReducedExtent[i];
 198   for(int i=0; i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 199     //DD(i);
 200     //DD(mReducedExtent[i]);
 201     //DD(mInitialExtent[i]);
 202     mCurrentSlicerManager->GetSlicer(i)->EnableReducedExtent(true);
 203   }
 204
 205   // Not now ....
 206   //  mIntensitySlider->SetImage(mCurrentImage);
 207   //mIntensitySlider->setEnabled(false);
 208
 209   // Set initial sliders values
 210   std::vector<int> imsize = mCurrentSlicerManager->GetImage()->GetSize();
 211   xminSlider->setMaximum(imsize[0]-1);
 212   xmaxSlider->setMaximum(imsize[0]-1);
 213   xmaxSlider->setValue(imsize[0]-1);
 214
 215   yminSlider->setMaximum(imsize[1]-1);
 216   ymaxSlider->setMaximum(imsize[1]-1);
 217   ymaxSlider->setValue(imsize[1]-1);
 218
 219   if (slicer->GetDimension() >2) {
 220     zminSlider->setMaximum(imsize[2]-1);
 221     zmaxSlider->setMaximum(imsize[2]-1);
 222     zmaxSlider->setValue(imsize[2]-1);
 223   }
 224
 225   // if (slicer->GetDimension() >3) {
 226   //   tminSlider->setMaximum(imsize[3]-1);
 227   //   tmaxSlider->setMaximum(imsize[3]-1);
 228   //   tmaxSlider->setValue(imsize[3]-1);
 229   // }
 230
 231   spin_xmin->setMaximum(imsize[0]-1);
 232   spin_xmax->setMaximum(imsize[0]-1);
 233   spin_xmax->setValue(imsize[0]-1);
 234
 235   spin_ymin->setMaximum(imsize[1]-1);
 236   spin_ymax->setMaximum(imsize[1]-1);
 237   spin_ymax->setValue(imsize[1]-1);
 238
 239   if (slicer->GetDimension() >2) {
 240     spin_zmin->setMaximum(imsize[2]-1);
 241     spin_zmax->setMaximum(imsize[2]-1);
 242     spin_zmax->setValue(imsize[2]-1);
 243   }
 244
 245   // if (slicer->GetDimension() >3) {
 246   //   spin_tmin->setMaximum(imsize[3]-1);
 247   //   spin_tmax->setMaximum(imsize[3]-1);
 248   //   spin_tmax->setValue(imsize[3]-1);
 249   // }
 250
 251   // Connect
 252   connect(xminSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderXMinValueChanged(int)));
 253   connect(xmaxSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderXMaxValueChanged(int)));
 254   connect(yminSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderYMinValueChanged(int)));
 255   connect(ymaxSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderYMaxValueChanged(int)));
 256   connect(zminSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderZMinValueChanged(int)));
 257   connect(zmaxSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderZMaxValueChanged(int)));
 258   // connect(tminSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderTMinValueChanged(int)));
 259   // connect(tmaxSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderTMaxValueChanged(int)));
 260
 261   //  connect(mCurrentSlicerManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateExtent()));
 262   //connect(mCurrentSlicerManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateExtent()));
 263
 264   //  connect(mIntensitySlider, SIGNAL(valueChanged(double)), this, SLOT(autoCropValueChanged(double)));
 265   UpdateExtent();
 266 }
 267 //------------------------------------------------------------------------------
 268
 269
 270 //------------------------------------------------------------------------------
 271 void vvToolCropImage::autoCropValueChanged(double v)
 272 {
 273   // DD(v);
 274   //  vvImageToITKImageVector -> TODO a generic method
 275   // then sliceriterator on each dimension from in to max
 276   // ==> make a clitkGenericFilter even for the regular apply ...
 277 }
 278 //------------------------------------------------------------------------------
 279
 280
 281 //------------------------------------------------------------------------------
 282 void vvToolCropImage::apply()
 283 {
 284
 285   QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
 286   vvImage::Pointer mResult=vvImage::New();
 287   vvImage::Pointer image= mCurrentSlicerManager->GetImage();
 288   for (std::vector<vtkImageData*>::const_iterator i=image->GetVTKImages().begin();
 289        i!=image->GetVTKImages().end(); i++) {
 290     vtkSmartPointer<vtkImageClip> filter=vtkSmartPointer<vtkImageClip>::New();
 291     ///Vtk is very weird, you need to "translate the extent" to get the correct origin
 292     //http://markmail.org/message/vndc2tr6kcabiakp#query:vtkImageClip%20origin+page:1+mid:6na7y57floutklvz+state:results
 293     vtkSmartPointer<vtkImageTranslateExtent> translate=vtkSmartPointer<vtkImageTranslateExtent>::New();
 294     filter->SetInput(*i);
 295     filter->SetOutputWholeExtent(xminSlider->value(),xmaxSlider->value(),
 296                                  yminSlider->value(),ymaxSlider->value(),
 297                                  zminSlider->value(),zmaxSlider->value());
 298     translate->SetTranslation(-xminSlider->value(),-yminSlider->value(),-zminSlider->value());
 299     translate->SetInput(filter->GetOutput());
 300     filter->ClipDataOn(); //Really create a cropped copy of the image
 301     translate->Update();
 302     vtkImageData* output=vtkImageData::New();
 303     output->ShallowCopy(translate->GetOutput());
 304     mResult->AddImage(output);
 305   }
 306   QApplication::restoreOverrideCursor();
 307   std::ostringstream osstream;
 308   osstream << "Crop_" << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
 309   AddImage(mResult, osstream.str());
 310   close();
 311 }
 312 //------------------------------------------------------------------------------
 313