vv/vvToolCropImage.cxx

   1 /*=========================================================================
   2
   3   Program:   vv
   4   Module:    $RCSfile: vvToolCropImage.cxx,v $
   5   Language:  C++
   6   Date:      $Date: 2010/03/01 15:38:09 $
   7   Version:   $Revision: 1.3 $
   8   Author :   David Sarrut (david.sarrut@creatis.insa-lyon.fr)
   9
  10   Copyright (C) 2008
  11   Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr
  12   CREATIS-LRMN http://www.creatis.insa-lyon.fr
  13
  14   This program is free software: you can redistribute it and/or modify
  15   it under the terms of the GNU General Public License as published by
  16   the Free Software Foundation, version 3 of the License.
  17
  18   This program is distributed in the hope that it will be useful,
  19   but WITHOUT ANY WARRANTY; without even the implied warranty of
  20   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21   GNU General Public License for more details.
  22
  23   You should have received a copy of the GNU General Public License
  24   along with this program.  If not, see <http://www.gnu.org/licenses/>.
  25
  26   =========================================================================*/
  27
  28 #include "vvToolCropImage.h"
  29 #include <QComboBox>
  30 #include <QCursor>
  31 #include <QApplication>
  32 #include <vtkImageClip.h>
  33 #include <vtkImageTranslateExtent.h>
  34 #include <vtkImageData.h>
  35 #include <vtkSmartPointer.h>
  36 #include "vvSlicer.h"
  37
  38 //------------------------------------------------------------------------------
  39 // Create the tool and automagically (I like this word) insert it in
  40 // the main window menu.
  41 ADD_TOOL(vvToolCropImage);
  42 //------------------------------------------------------------------------------
  43
  44
  45 //------------------------------------------------------------------------------
  46 vvToolCropImage::vvToolCropImage(vvMainWindowBase * parent, Qt::WindowFlags f):
  47   vvToolWidgetBase(parent, f),
  48   vvToolBase<vvToolCropImage>(parent),
  49   Ui::vvToolCropImage() {
  50
  51   // GUI Initialization
  52   Ui_vvToolCropImage::setupUi(mToolWidget);
  53
  54 }
  55 //------------------------------------------------------------------------------
  56
  57
  58 //------------------------------------------------------------------------------
  59 vvToolCropImage::~vvToolCropImage() {
  60
  61 }
  62 //------------------------------------------------------------------------------
  63
  64
  65 //------------------------------------------------------------------------------
  66 bool vvToolCropImage::close() {
  67   for(int i=0; i<6; i++) mReducedExtent[i] = mInitialExtent[i];
  68   UpdateExtent();
  69   return vvToolWidgetBase::close();
  70 }
  71 //------------------------------------------------------------------------------
  72
  73
  74 //------------------------------------------------------------------------------
  75 void vvToolCropImage::reject() {
  76   for(int i=0; i<6; i++) mReducedExtent[i] = mInitialExtent[i];
  77   UpdateExtent();
  78   return vvToolWidgetBase::reject();
  79 }
  80 //------------------------------------------------------------------------------
  81
  82
  83 //------------------------------------------------------------------------------
  84 void vvToolCropImage::sliderXMinValueChanged(int s) {
  85   xmaxSlider->setMinimum(xminSlider->value());
  86   mReducedExtent[0] = xminSlider->value();
  87   UpdateExtent();
  88 }
  89 //------------------------------------------------------------------------------
  90
  91
  92 //------------------------------------------------------------------------------
  93 void vvToolCropImage::sliderXMaxValueChanged(int s) {
  94   xminSlider->setMaximum(xmaxSlider->value());
  95   mReducedExtent[1] = xmaxSlider->value();
  96   UpdateExtent();
  97 }
  98 //------------------------------------------------------------------------------
  99
 100
 101 //------------------------------------------------------------------------------
 102 void vvToolCropImage::sliderYMinValueChanged(int s) {
 103   ymaxSlider->setMinimum(yminSlider->value());
 104   mReducedExtent[2] = yminSlider->value();
 105   UpdateExtent();
 106 }
 107 //------------------------------------------------------------------------------
 108
 109
 110 //------------------------------------------------------------------------------
 111 void vvToolCropImage::sliderYMaxValueChanged(int s) {
 112   yminSlider->setMaximum(ymaxSlider->value());
 113   mReducedExtent[3] = ymaxSlider->value();
 114   UpdateExtent();
 115 }
 116 //------------------------------------------------------------------------------
 117
 118
 119 //------------------------------------------------------------------------------
 120 void vvToolCropImage::sliderZMinValueChanged(int s) {
 121   zmaxSlider->setMinimum(zminSlider->value());
 122   mReducedExtent[4] = zminSlider->value();
 123   UpdateExtent();
 124 }
 125 //------------------------------------------------------------------------------
 126
 127
 128 //------------------------------------------------------------------------------
 129 void vvToolCropImage::sliderZMaxValueChanged(int s) {
 130   zminSlider->setMaximum(zmaxSlider->value());
 131   mReducedExtent[5] = zmaxSlider->value();
 132   UpdateExtent();
 133 }
 134 //------------------------------------------------------------------------------
 135
 136
 137 //------------------------------------------------------------------------------
 138 // void vvToolCropImage::sliderTMinValueChanged(int s) {
 139 //   tmaxSlider->setMinimum(tminSlider->value());
 140 //   mReducedExtent[6] = tminSlider->value();
 141 //   UpdateExtent();
 142 // }
 143 //------------------------------------------------------------------------------
 144
 145
 146 //------------------------------------------------------------------------------
 147 // void vvToolCropImage::sliderTMaxValueChanged(int s) {
 148 //   tminSlider->setMaximum(tmaxSlider->value());
 149 //   mReducedExtent[7] = tmaxSlider->value();
 150 //   UpdateExtent();
 151 // }
 152 //------------------------------------------------------------------------------
 153
 154
 155 //------------------------------------------------------------------------------
 156 void vvToolCropImage::UpdateExtent() {
 157   for(int i=0; i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 158     mCurrentSlicerManager->GetSlicer(i)->SetReducedExtent(mReducedExtent);
 159     mCurrentSlicerManager->GetSlicer(i)->ForceUpdateDisplayExtent();
 160     mCurrentSlicerManager->GetSlicer(i)->Render();
 161   }
 162 }
 163 //------------------------------------------------------------------------------
 164
 165
 166 //------------------------------------------------------------------------------
 167 void vvToolCropImage::InputIsSelected(vvSlicerManager * slicer) {
 168
 169   // Change interface according to number of dimension
 170   mExtentSize = 2*slicer->GetDimension();
 171   // if (slicer->GetDimension()<4) {
 172   //   mTLabel1->setHidden(true);
 173   //   mTLabel2->setHidden(true);
 174   //   tminSlider->setHidden(true);
 175   //   tmaxSlider->setHidden(true);
 176   //   spin_tmin->setHidden(true);
 177   //   spin_tmax->setHidden(true);
 178   // }
 179   if (slicer->GetDimension()<3) {
 180     mZLabel1->setHidden(true);
 181     mZLabel2->setHidden(true);
 182     zminSlider->setHidden(true);
 183     zmaxSlider->setHidden(true);
 184     spin_zmin->setHidden(true);
 185     spin_zmax->setHidden(true);
 186   }
 187
 188   // Record initial extend
 189   mReducedExtent = new int[mExtentSize];
 190   mInitialExtent = new int[mExtentSize];
 191   mReducedExtent = mCurrentSlicerManager->GetImage()->GetFirstVTKImageData()->GetWholeExtent();
 192   for(int i=0; i<6; i++) mInitialExtent[i] = mReducedExtent[i];
 193   for(int i=0; i<mCurrentSlicerManager->NumberOfSlicers(); i++) {
 194     mCurrentSlicerManager->GetSlicer(i)->EnableReducedExtent(true);
 195   }
 196
 197   // Not now ....
 198   //  mIntensitySlider->SetImage(mCurrentImage);
 199   //mIntensitySlider->setEnabled(false);
 200
 201   // Set initial sliders values
 202   std::vector<int> imsize = mCurrentSlicerManager->GetImage()->GetSize();
 203   xminSlider->setMaximum(imsize[0]-1);
 204   xmaxSlider->setMaximum(imsize[0]-1);
 205   xmaxSlider->setValue(imsize[0]-1);
 206
 207   yminSlider->setMaximum(imsize[1]-1);
 208   ymaxSlider->setMaximum(imsize[1]-1);
 209   ymaxSlider->setValue(imsize[1]-1);
 210
 211   if (slicer->GetDimension() >2) {
 212     zminSlider->setMaximum(imsize[2]-1);
 213     zmaxSlider->setMaximum(imsize[2]-1);
 214     zmaxSlider->setValue(imsize[2]-1);
 215   }
 216
 217   // if (slicer->GetDimension() >3) {
 218   //   tminSlider->setMaximum(imsize[3]-1);
 219   //   tmaxSlider->setMaximum(imsize[3]-1);
 220   //   tmaxSlider->setValue(imsize[3]-1);
 221   // }
 222
 223   spin_xmin->setMaximum(imsize[0]-1);
 224   spin_xmax->setMaximum(imsize[0]-1);
 225   spin_xmax->setValue(imsize[0]-1);
 226
 227   spin_ymin->setMaximum(imsize[1]-1);
 228   spin_ymax->setMaximum(imsize[1]-1);
 229   spin_ymax->setValue(imsize[1]-1);
 230
 231   if (slicer->GetDimension() >2) {
 232     spin_zmin->setMaximum(imsize[2]-1);
 233     spin_zmax->setMaximum(imsize[2]-1);
 234     spin_zmax->setValue(imsize[2]-1);
 235   }
 236
 237   // if (slicer->GetDimension() >3) {
 238   //   spin_tmin->setMaximum(imsize[3]-1);
 239   //   spin_tmax->setMaximum(imsize[3]-1);
 240   //   spin_tmax->setValue(imsize[3]-1);
 241   // }
 242
 243   // Connect
 244   connect(xminSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderXMinValueChanged(int)));
 245   connect(xmaxSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderXMaxValueChanged(int)));
 246   connect(yminSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderYMinValueChanged(int)));
 247   connect(ymaxSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderYMaxValueChanged(int)));
 248   connect(zminSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderZMinValueChanged(int)));
 249   connect(zmaxSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderZMaxValueChanged(int)));
 250   // connect(tminSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderTMinValueChanged(int)));
 251   // connect(tmaxSlider, SIGNAL(valueChanged(int)), this, SLOT(sliderTMaxValueChanged(int)));
 252
 253   //  connect(mIntensitySlider, SIGNAL(valueChanged(double)), this, SLOT(autoCropValueChanged(double)));
 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