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