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://www.centreleonberard.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 // vv
  20 #include "vvToolCropImage.h"
  21 #include "vvSlicer.h"
  22
  23 // clitk
  24 #include "clitkCropImage_ggo.h"
  25 #include "clitkCropImageGenericFilter.h"
  26 #include "clitkExceptionObject.h"
  27
  28 // qt
  29 #include <QComboBox>
  30 #include <QCursor>
  31 #include <QApplication>
  32 #include <QMessageBox>
  33 #include <QSignalMapper>
  34
  35 // vtk
  36 #include <vtkImageClip.h>
  37 #include <vtkImageTranslateExtent.h>
  38 #include <vtkImageData.h>
  39 #include <vtkSmartPointer.h>
  40
  41 //------------------------------------------------------------------------------
  42 // Create the tool and automagically (I like this word) insert it in
  43 // the main window menu.
  44 ADD_TOOL(vvToolCropImage);
  45 //------------------------------------------------------------------------------
  46
  47 //same order of int* returned by VtkImageData::WholeExtent
  48 enum sliderId {xmin, xmax, ymin, ymax, zmin, zmax, tmin, tmax};
  49 //------------------------------------------------------------------------------
  50 vvToolCropImage::vvToolCropImage(vvMainWindowBase * parent, Qt::WindowFlags f):
  51   vvToolWidgetBase(parent, f),
  52   vvToolBase<vvToolCropImage>(parent),
  53   Ui::vvToolCropImage(),mSliders(8)
  54 {
  55
  56   // GUI Initialization
  57   Ui_vvToolCropImage::setupUi(mToolWidget);
  58   // mTLabel2->setHidden(false);
  59   // tminSlider->setHidden(false);
  60   // tmaxSlider->setHidden(false);
  61   // spin_tmin->setHidden(false);
  62   // spin_tmax->setHidden(false);
  63   // mLabelTimeCropping->setHidden(false);
  64   mTLabel1->setHidden(true);
  65   mTLabel2->setHidden(true);
  66   tminSlider->setHidden(true);
  67   tmaxSlider->setHidden(true);
  68   spin_tmin->setHidden(true);
  69   spin_tmax->setHidden(true);
  70   mLabelTimeCropping->setHidden(true);
  71
  72   // Set how many inputs are needed for this tool
  73   AddInputSelector("Select one image");
  74
  75   mSliders[xmin]=xminSlider;
  76   mSliders[xmax]=xmaxSlider;
  77   mSliders[ymin]=yminSlider;
  78   mSliders[ymax]=ymaxSlider;
  79   mSliders[zmin]=zminSlider;
  80   mSliders[zmax]=zmaxSlider;
  81   mSliders[tmin]=tminSlider;
  82   mSliders[tmax]=tmaxSlider;
  83 }
  84 //------------------------------------------------------------------------------
  85
  86
  87 //------------------------------------------------------------------------------
  88 vvToolCropImage::~vvToolCropImage()
  89 {
  90
  91 }
  92 //------------------------------------------------------------------------------
  93
  94
  95 //------------------------------------------------------------------------------
  96 void vvToolCropImage::closeEvent(QCloseEvent *event)
  97 {
  98   if(mCurrentSlicerManager){
  99     // Reset extends
 100     for(int i=0; i<mExtentSize; i++) mReducedExtent[i] = mInitialExtent[i];
 101       UpdateExtent();
 102   }
 103   event->accept();
 104 }
 105 //------------------------------------------------------------------------------
 106
 107
 108 //------------------------------------------------------------------------------
 109 bool vvToolCropImage::close()
 110 {
 111   return vvToolWidgetBase::close();
 112 }
 113 //------------------------------------------------------------------------------
 114
 115
 116 //------------------------------------------------------------------------------
 117 void vvToolCropImage::reject()
 118 {
 119   for(int i=0; i<mExtentSize; i++) mReducedExtent[i] = mInitialExtent[i];
 120   UpdateExtent();
 121   return vvToolWidgetBase::reject();
 122 }
 123 //------------------------------------------------------------------------------
 124
 125
 126 //------------------------------------------------------------------------------
 127 void vvToolCropImage::sliderValueChanged(int dim)
 128 {
 129   int dimMin = dim;
 130   if(dim%2==0){//case we are minimum
 131     mSliders[dim+1]->setMinimum(mSliders[dim]->value());
 132   }else {
 133     mSliders[--dimMin]->setMaximum(mSliders[dim]->value());
 134   }
 135   mReducedExtent[dim] = mSliders[dim]->value() + mInitialExtent[dimMin];
 136   UpdateExtent();
 137 }
 138 //------------------------------------------------------------------------------
 139
 140
 141 //------------------------------------------------------------------------------
 142 void vvToolCropImage::UpdateExtent()
 143 {
 144   for(int i=0; i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
 145     mCurrentSlicerManager->GetSlicer(i)->SetReducedExtent(mReducedExtent);
 146     mCurrentSlicerManager->GetSlicer(i)->ForceUpdateDisplayExtent();
 147     mCurrentSlicerManager->GetSlicer(i)->Render();
 148   }
 149 }
 150 //------------------------------------------------------------------------------
 151
 152
 153 //------------------------------------------------------------------------------
 154 void vvToolCropImage::InputIsSelected(vvSlicerManager * slicer)
 155 {
 156   // Change interface according to number of dimension
 157   mExtentSize = 2*slicer->GetDimension();
 158    if (slicer->GetDimension()<4) {
 159      mTLabel1->setHidden(true);
 160      mTLabel2->setHidden(true);
 161      tminSlider->setHidden(true);
 162      tmaxSlider->setHidden(true);
 163      spin_tmin->setHidden(true);
 164      spin_tmax->setHidden(true);
 165      mLabelTimeCropping->setHidden(true);
 166    }
 167   if (slicer->GetDimension()<3) {
 168     mZLabel1->setHidden(true);
 169     mZLabel2->setHidden(true);
 170     zminSlider->setHidden(true);
 171     zmaxSlider->setHidden(true);
 172     spin_zmin->setHidden(true);
 173     spin_zmax->setHidden(true);
 174   }
 175
 176   // Record initial extend
 177   mReducedExtent = new int[mExtentSize];
 178   mInitialExtent = new int[mExtentSize];
 179   mReducedExtent = mCurrentSlicerManager->GetImage()->GetFirstVTKImageData()->GetWholeExtent();
 180   for(int i=0; i<mExtentSize; i++) mInitialExtent[i] = mReducedExtent[i];
 181   for(int i=0; i<mCurrentSlicerManager->GetNumberOfSlicers(); i++) {
 182     mCurrentSlicerManager->GetSlicer(i)->EnableReducedExtent(true);
 183   }
 184
 185   // Set initial sliders values
 186   std::vector<int> imsize = mCurrentSlicerManager->GetImage()->GetSize();
 187   for(int dim=0; dim<slicer->GetDimension() && dim<3; ++dim){
 188     mSliders[dim*2]->setMaximum(imsize[dim]-1);
 189     mSliders[dim*2+1]->setMaximum(imsize[dim]-1);
 190     mSliders[dim*2+1]->setValue(imsize[dim]-1);
 191   }
 192   spin_xmin->setMaximum(imsize[0]-1);
 193   spin_xmax->setMaximum(imsize[0]-1);
 194   spin_xmax->setValue(imsize[0]-1);
 195
 196   spin_ymin->setMaximum(imsize[1]-1);
 197   spin_ymax->setMaximum(imsize[1]-1);
 198   spin_ymax->setValue(imsize[1]-1);
 199
 200   if (slicer->GetDimension() >2) {
 201     spin_zmin->setMaximum(imsize[2]-1);
 202     spin_zmax->setMaximum(imsize[2]-1);
 203     spin_zmax->setValue(imsize[2]-1);
 204   }
 205
 206   if (slicer->GetDimension() >3) {
 207     spin_tmin->setMaximum(imsize[3]-1);
 208     spin_tmax->setMaximum(imsize[3]-1);
 209     spin_tmax->setValue(imsize[3]-1);
 210   }
 211
 212   QSignalMapper* signalMapper = new QSignalMapper(this);
 213   connect(signalMapper, SIGNAL(mapped(int)), this, SLOT(sliderValueChanged(int)));
 214   for(unsigned int i=0; i<mSliders.size(); ++i){
 215     signalMapper->setMapping(mSliders[i], i);
 216     connect(mSliders[i], SIGNAL(valueChanged(int)), signalMapper, SLOT(map()));
 217   }
 218
 219
 220
 221
 222   //  connect(mCurrentSlicerManager,SIGNAL(UpdateSlice(int,int)),this,SLOT(UpdateExtent()));
 223   //connect(mCurrentSlicerManager,SIGNAL(UpdateTSlice(int,int)),this,SLOT(UpdateExtent()));
 224
 225   //  connect(mIntensitySlider, SIGNAL(valueChanged(double)), this, SLOT(autoCropValueChanged(double)));
 226   UpdateExtent();
 227 }
 228 //------------------------------------------------------------------------------
 229
 230
 231 //------------------------------------------------------------------------------
 232 void vvToolCropImage::autoCropValueChanged(double v)
 233 {
 234   //TODO
 235 }
 236 //------------------------------------------------------------------------------
 237
 238
 239 //------------------------------------------------------------------------------
 240 void vvToolCropImage::apply()
 241 {
 242   if (!mCurrentSlicerManager) close();
 243   QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
 244
 245   // Typedef
 246   typedef args_info_clitkCropImage ArgsInfoType;
 247   typedef clitk::CropImageGenericFilter CropFilterType;
 248
 249   // Get options
 250   ArgsInfoType mArgsInfo;
 251   cmdline_parser_clitkCropImage_init(&mArgsInfo); // Initialisation to default
 252   int n = mCurrentSlicerManager->GetDimension()*2;  // 2D and 3D only
 253   mArgsInfo.boundingBox_given = n;
 254   mArgsInfo.boundingBox_arg = new int[n];
 255
 256   for(int dim=0; dim<mCurrentSlicerManager->GetDimension() && dim<3; ++dim){
 257     mArgsInfo.boundingBox_arg[dim*2] = mSliders[dim*2]->value();
 258     mArgsInfo.boundingBox_arg[dim*2+1] = mSliders[dim*2+1]->value();
 259   }
 260   if (n>6) { // 4D
 261     mArgsInfo.boundingBox_arg[6] = 0;
 262     mArgsInfo.boundingBox_arg[7] = mCurrentImage->GetSize()[3]-1;
 263   }
 264   // We MUST reset initial extend to input image before using the
 265   // filter to retrieve the correct image size
 266   for(int i=0; i<mExtentSize; i++) {
 267     mReducedExtent[i] = mInitialExtent[i];
 268   }
 269
 270   UpdateExtent();
 271   // Main filter
 272   CropFilterType::Pointer filter = CropFilterType::New();
 273   filter->SetInputVVImage(mCurrentImage);
 274   filter->SetArgsInfo(mArgsInfo);
 275
 276   // Go ! (not threaded)
 277   try{
 278     filter->Update();
 279   }
 280   catch(clitk::ExceptionObject & e) {
 281     DD(e.what());
 282     QApplication::restoreOverrideCursor();
 283     close();
 284   }
 285   std::ostringstream croppedImageName;
 286   croppedImageName << "Cropped_" << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd";
 287   // Retrieve result and display it
 288   vvImage::Pointer output = filter->GetOutputVVImage();
 289   AddImage(output,croppedImageName.str());
 290   // End
 291   QApplication::restoreOverrideCursor();
 292   close();
 293 }
 294 //------------------------------------------------------------------------------
 295