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