vv/vvResamplerDialog.cxx

   1 #ifndef _vvResamplerDialog_CXX
   2 #define _vvResamplerDialog_CXX
   3
   4 /*=========================================================================
   5
   6 Program:   vv
   7 Language:  C++
   8 Author :   David Sarrut (david.sarrut@gmail.com)
   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 "vvResamplerDialog.h"
  29 #include "clitkImageResampleGenericFilter.h"
  30 #include "vvSlicer.h"
  31
  32 #include <QFileInfo>
  33 #include <QMessageBox>
  34
  35 #define COLUMN_IMAGE_NAME 7
  36
  37 //====================================================================
  38 vvResamplerDialog::vvResamplerDialog(QWidget * parent, Qt::WindowFlags f)
  39         :QDialog(parent,f), Ui::vvResamplerDialog() {
  40
  41     // initialization
  42     setupUi(this);
  43     Init();
  44
  45     // Connect signals & slots
  46     connect(this, SIGNAL(accepted()), this, SLOT(Resample()));
  47     connect(mImagesComboBox, SIGNAL(currentIndexChanged(QString)), this, SLOT(UpdateCurrentInputImage()));
  48
  49     connect(sizeRadioButton, SIGNAL(clicked()), this, SLOT(UpdateControlSizeAndSpacing()));
  50     connect(scaleSizeRadioButton, SIGNAL(clicked()), this, SLOT(UpdateControlSizeAndSpacing()));
  51     connect(isoSizeRadioButton, SIGNAL(clicked()), this, SLOT(UpdateControlSizeAndSpacing()));
  52     connect(spacingRadioButton, SIGNAL(clicked()), this, SLOT(UpdateControlSizeAndSpacing()));
  53     connect(scaleSpacingRadioButton, SIGNAL(clicked()), this, SLOT(UpdateControlSizeAndSpacing()));
  54     connect(isoSpacingRadioButton, SIGNAL(clicked()), this, SLOT(UpdateControlSizeAndSpacing()));
  55
  56     connect(xSizeLineEdit, SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSpacingFromSize()));
  57     connect(ySizeLineEdit, SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSpacingFromSize()));
  58     connect(zSizeLineEdit, SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSpacingFromSize()));
  59     connect(xSpacingLineEdit, SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSizeFromSpacing()));
  60     connect(ySpacingLineEdit, SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSizeFromSpacing()));
  61     connect(zSpacingLineEdit, SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSizeFromSpacing()));
  62     connect(scaleSizeLineEdit,SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSizeFromScale()));
  63     connect(scaleSpacingLineEdit,SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSpacingFromScale()));
  64     connect(isoSizeLineEdit,SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSizeFromIso()));
  65     connect(isoSpacingLineEdit,SIGNAL(textEdited(QString)),this,SLOT(ComputeNewSpacingFromIso()));
  66
  67     connect(gaussianFilterCheckBox,SIGNAL(stateChanged(int)),this,SLOT(UpdateGaussianFilter()));
  68     connect(interpolationComboBox,SIGNAL(currentIndexChanged(QString)),this,SLOT(UpdateInterpolation()));
  69 }
  70 //====================================================================
  71
  72 //====================================================================
  73 void vvResamplerDialog::Init() {
  74     mLastError ="";
  75
  76     mInputFileFormat = "";
  77     ComponentType = "";
  78     mPixelType = "";
  79
  80     mInputSize.resize(0);
  81     mInputSpacing.resize(0);
  82     mInputOrigin.resize(0);
  83     mOutputSize.resize(0);
  84     mOutputSpacing.resize(0);
  85
  86     xSizeLineEdit->setText("");
  87     ySizeLineEdit->setText("");
  88     zSizeLineEdit->setText("");
  89     xSpacingLineEdit->setText("");
  90     ySpacingLineEdit->setText("");
  91     zSpacingLineEdit->setText("");
  92     scaleSizeLineEdit->setText("");
  93     scaleSpacingLineEdit->setText("");
  94
  95     sizeRadioButton->setChecked(0);
  96     scaleSizeRadioButton->setChecked(0);
  97     isoSizeRadioButton->setChecked(0);
  98     spacingRadioButton->setChecked(0);
  99     scaleSpacingRadioButton->setChecked(0);
 100     isoSpacingRadioButton->setChecked(0);
 101
 102     gaussianFilterCheckBox->setCheckState(Qt::Unchecked);
 103
 104     xGaussianLineEdit->hide();
 105     yGaussianLineEdit->hide();
 106     zGaussianLineEdit->hide();
 107     gaussianFilterLabel->hide();
 108
 109     bSplineLabel->hide();
 110     bSplineOrderSpinBox->hide();
 111     bLUTFactorLabel->hide();
 112     bLUTSpinBox->hide();
 113     mDimension = -1;
 114
 115     QPalette qPalette;
 116     qPalette.setColor(QPalette::Foreground, QColor(Qt::blue));
 117     mInputFormatLabel->setPalette(qPalette);
 118     mInputDimLabel->setPalette(qPalette);
 119     mInputPixelTypeLabel->setPalette(qPalette);
 120     mInputSizeLabel->setPalette(qPalette);
 121     mInputSpacingLabel->setPalette(qPalette);
 122
 123     UpdateCurrentInputImage();
 124 }
 125 //====================================================================
 126
 127 //====================================================================
 128 void vvResamplerDialog::UpdateCurrentInputImage() {
 129     // Set current image & index
 130     mCurrentIndex = mImagesComboBox->currentIndex();
 131     if (mCurrentIndex == -1) {
 132         mCurrentImage = NULL;
 133         return ;
 134     }
 135     mCurrentImage = mSlicerManagers[mCurrentIndex]->GetSlicer(0)->GetImage();
 136     if (mCurrentImage.IsNull()) return;
 137     mInputFileName = mSlicerManagers[mCurrentIndex]->GetFileName().c_str();
 138
 139     // Set current information
 140     mPixelType = mCurrentImage->GetScalarTypeAsString().c_str();
 141     //ds      ComponentType = mCurrentImageGetNumberOfScalarComponents();
 142     mDimension = mCurrentImage->GetNumberOfDimensions();
 143
 144     // Copy size, spacing ...
 145     mInputOrigin.resize(mDimension);
 146     mInputSpacing.resize(mDimension);
 147     mInputSize.resize(mDimension);
 148     for (int i = 0; i < mDimension;i++) {
 149         mInputOrigin[i] = mCurrentImage->GetOrigin()[i];
 150         mInputSpacing[i] = mCurrentImage->GetSpacing()[i];
 151         mInputSize[i] = mCurrentImage->GetSize()[i];
 152     }
 153
 154     // Get file format
 155     mInputFileFormat = itksys::SystemTools::GetFilenameLastExtension(mInputFileName.toStdString()).c_str();
 156
 157     // Display infos
 158     mInputFormatLabel->setText(mInputFileFormat);
 159     mInputSizeLabel->setText(GetVectorIntAsString(mInputSize));
 160     mInputDimLabel->setText(QString::number(mDimension)+"D");
 161     mInputSpacingLabel->setText(GetVectorDoubleAsString(mInputSpacing));
 162     mInputPixelTypeLabel->setText(mPixelType);
 163     mInputMemoryLabel->setText(GetSizeInBytes(mInputSize));
 164
 165     // Set current size
 166     scaleSizeRadioButton->setChecked(true);
 167     UpdateControlSizeAndSpacing();
 168     scaleSizeLineEdit->setText("100");
 169     ComputeNewSizeFromScale();
 170
 171     // Update output
 172     UpdateOutputInfo();
 173 }
 174 //====================================================================
 175
 176 //====================================================================
 177 void vvResamplerDialog::UpdateOutputInfo() {
 178     mOutputSizeLabel->setText(GetVectorIntAsString(mOutputSize));
 179     mOutputSpacingLabel->setText(GetVectorDoubleAsString(mOutputSpacing));
 180     mOutputMemoryLabel->setText(GetSizeInBytes(mOutputSize));
 181 }
 182 //====================================================================
 183
 184 //====================================================================
 185 QString vvResamplerDialog::GetSizeInBytes(std::vector<int> & size) {
 186     int t = 1;
 187     for (unsigned int i=0; i<size.size(); i++) t *= size[i];
 188     t *= mCurrentImage->GetScalarSize()*mCurrentImage->GetNumberOfScalarComponents();
 189     QString result = QString::number(t);
 190     result += " bytes (";
 191     if (t > 1000000000) {
 192         t /= 1000000000;
 193         result += QString::number(t);
 194         result += " GB)";
 195     }
 196     else if (t > 1000000) {
 197         t /= 1000000;
 198         result += QString::number(t);
 199         result += " MB)";
 200     }
 201     else if (t > 1000) {
 202         t /= 1000;
 203         result += QString::number(t);
 204         result += " KB)";
 205     }
 206     return result;
 207 }
 208 //====================================================================
 209
 210 //====================================================================
 211 QString vvResamplerDialog::GetVectorDoubleAsString(std::vector<double> vectorDouble) {
 212     QString result;
 213     for (unsigned int i= 0; i<vectorDouble.size(); i++) {
 214         if (i != 0)
 215             result += " x ";
 216         result += QString::number(vectorDouble[i]);
 217     }
 218     return result;
 219 }
 220 //====================================================================
 221
 222 //====================================================================
 223 QString vvResamplerDialog::GetVectorIntAsString(std::vector<int> vectorInt) {
 224     QString result;
 225     for (unsigned int i= 0; i<vectorInt.size(); i++) {
 226         if (i != 0)
 227             result += " x ";
 228         result += QString::number(vectorInt[i]);
 229     }
 230     return result;
 231 }
 232 //====================================================================
 233
 234 //====================================================================
 235 void vvResamplerDialog::FillSizeEdit(std::vector<int> size) {
 236     xSizeLineEdit->setText(QString::number(size[0]));
 237     ySizeLineEdit->setText(QString::number(size[1]));
 238     if (size.size() > 2)
 239         zSizeLineEdit->setText(QString::number(size[2]));
 240 }
 241 //====================================================================
 242
 243 //====================================================================
 244 void vvResamplerDialog::FillSpacingEdit(std::vector<double> spacing) {
 245     xSpacingLineEdit->setText(QString::number(spacing[0]));
 246     ySpacingLineEdit->setText(QString::number(spacing[1]));
 247     if (spacing.size() > 2)
 248         zSpacingLineEdit->setText(QString::number(spacing[2]));
 249 }
 250 //====================================================================
 251
 252 //====================================================================
 253 void vvResamplerDialog::UpdateOutputSizeAndSpacing() {
 254     mOutputSize.resize(mDimension);
 255     mOutputSize = mInputSize;
 256     mOutputSpacing.resize(mDimension);
 257     mOutputSpacing = mInputSpacing;
 258     mOutputSize[0] = (int)xSizeLineEdit->text().toDouble();
 259     mOutputSize[1] = (int)ySizeLineEdit->text().toDouble();
 260     if (mDimension > 2)
 261         mOutputSize[2] = (int)zSizeLineEdit->text().toDouble();
 262
 263     mOutputSpacing[0] = xSpacingLineEdit->text().toDouble();
 264     mOutputSpacing[1] = ySpacingLineEdit->text().toDouble();
 265     if (mDimension > 2)
 266         mOutputSpacing[2] = zSpacingLineEdit->text().toDouble();
 267
 268     UpdateOutputInfo();
 269 }
 270 //====================================================================
 271
 272 //====================================================================
 273 void vvResamplerDialog::UpdateControlSizeAndSpacing() {
 274     scaleSizeLineEdit->setText("");
 275     scaleSpacingLineEdit->setText("");
 276     isoSizeLineEdit->setText("");
 277     isoSpacingLineEdit->setText("");
 278
 279     xSizeLineEdit->setReadOnly(1);
 280     ySizeLineEdit->setReadOnly(1);
 281     zSizeLineEdit->setReadOnly(1);
 282     scaleSizeLineEdit->setReadOnly(1);
 283     isoSizeLineEdit->setReadOnly(1);
 284
 285     xSpacingLineEdit->setReadOnly(1);
 286     ySpacingLineEdit->setReadOnly(1);
 287     zSpacingLineEdit->setReadOnly(1);
 288     scaleSpacingLineEdit->setReadOnly(1);
 289     isoSpacingLineEdit->setReadOnly(1);
 290
 291     if (sizeRadioButton->isChecked()) {
 292         xSizeLineEdit->setReadOnly(0);
 293         ySizeLineEdit->setReadOnly(0);
 294         if (mDimension > 2)
 295             zSizeLineEdit->setReadOnly(0);
 296     }
 297     else {
 298         if (spacingRadioButton->isChecked()) {
 299             xSpacingLineEdit->setReadOnly(0);
 300             ySpacingLineEdit->setReadOnly(0);
 301             if (mDimension > 2)
 302                 zSpacingLineEdit->setReadOnly(0);
 303         }
 304         else if (scaleSizeRadioButton->isChecked())
 305             scaleSizeLineEdit->setReadOnly(0);
 306         else if (scaleSpacingRadioButton->isChecked())
 307             scaleSpacingLineEdit->setReadOnly(0);
 308         else if (isoSizeRadioButton->isChecked())
 309             isoSizeLineEdit->setReadOnly(0);
 310         else if (isoSpacingRadioButton->isChecked())
 311             isoSpacingLineEdit->setReadOnly(0);
 312     }
 313 }
 314 //====================================================================
 315
 316 //====================================================================
 317 void vvResamplerDialog::ComputeNewSpacingFromSize() {
 318     double newSpacing = mInputSpacing[0]*mInputSize[0];
 319     xSpacingLineEdit->setText(QString::number(newSpacing/xSizeLineEdit->text().toDouble()));
 320     newSpacing = mInputSpacing[1]*mInputSize[1];
 321     ySpacingLineEdit->setText(QString::number(newSpacing/ySizeLineEdit->text().toDouble()));
 322     if (mDimension > 2)
 323     {
 324         newSpacing = mInputSpacing[2]*mInputSize[2];
 325         zSpacingLineEdit->setText(QString::number(newSpacing/zSizeLineEdit->text().toDouble()));
 326     }
 327     UpdateOutputSizeAndSpacing();
 328 }
 329 //====================================================================
 330
 331 //====================================================================
 332 void vvResamplerDialog::ComputeNewSizeFromSpacing() {
 333     double newSize = mInputSpacing[0]*mInputSize[0];
 334     xSizeLineEdit->setText(QString::number(newSize/xSpacingLineEdit->text().toDouble()));
 335     newSize = mInputSpacing[1]*mInputSize[1];
 336     ySizeLineEdit->setText(QString::number(newSize/ySpacingLineEdit->text().toDouble()));
 337     if (mDimension > 2)
 338     {
 339         newSize = mInputSpacing[2]*mInputSize[2];
 340         zSizeLineEdit->setText(QString::number(newSize/zSpacingLineEdit->text().toDouble()));
 341     }
 342     UpdateOutputSizeAndSpacing();
 343 }
 344 //====================================================================
 345
 346 //====================================================================
 347 void vvResamplerDialog::ComputeNewSpacingFromScale() {
 348     xSpacingLineEdit->setText(QString::number(mInputSpacing[0]*scaleSpacingLineEdit->text().toDouble()/100));
 349     ySpacingLineEdit->setText(QString::number(mInputSpacing[1]*scaleSpacingLineEdit->text().toDouble()/100));
 350     if (mDimension > 2)
 351         zSpacingLineEdit->setText(QString::number(mInputSpacing[2]*scaleSpacingLineEdit->text().toDouble()/100));
 352     ComputeNewSizeFromSpacing();
 353 }
 354 //====================================================================
 355
 356 //====================================================================
 357 void vvResamplerDialog::ComputeNewSizeFromScale() {
 358     xSizeLineEdit->setText(QString::number(mInputSize[0]*scaleSizeLineEdit->text().toDouble()/100));
 359     ySizeLineEdit->setText(QString::number(mInputSize[1]*scaleSizeLineEdit->text().toDouble()/100));
 360     if (mDimension > 2)
 361         zSizeLineEdit->setText(QString::number(mInputSize[2]*scaleSizeLineEdit->text().toDouble()/100));
 362     ComputeNewSpacingFromSize();
 363 }
 364 //====================================================================
 365
 366 //====================================================================
 367 void vvResamplerDialog::ComputeNewSpacingFromIso() {
 368     xSpacingLineEdit->setText(QString::number(isoSpacingLineEdit->text().toDouble()));
 369     ySpacingLineEdit->setText(QString::number(isoSpacingLineEdit->text().toDouble()));
 370     if (mDimension > 2)
 371         zSpacingLineEdit->setText(QString::number(isoSpacingLineEdit->text().toDouble()));
 372     ComputeNewSizeFromSpacing();
 373 }
 374 //====================================================================
 375
 376 //====================================================================
 377 void vvResamplerDialog::ComputeNewSizeFromIso() {
 378     xSizeLineEdit->setText(QString::number(isoSizeLineEdit->text().toDouble()));
 379     ySizeLineEdit->setText(QString::number(isoSizeLineEdit->text().toDouble()));
 380     if (mDimension > 2)
 381         zSizeLineEdit->setText(QString::number(isoSizeLineEdit->text().toDouble()));
 382     ComputeNewSpacingFromSize();
 383 }
 384 //====================================================================
 385
 386 //====================================================================
 387 void vvResamplerDialog::UpdateInterpolation() {
 388     if (interpolationComboBox->currentText() == "BSpline") {
 389         bSplineLabel->show();
 390         bSplineOrderSpinBox->show();
 391         bLUTFactorLabel->hide();
 392         bLUTSpinBox->hide();
 393     }
 394     else if (interpolationComboBox->currentText() == "Blut (faster BSpline)")   {
 395         bSplineLabel->show();
 396         bSplineOrderSpinBox->show();
 397         bLUTFactorLabel->show();
 398         bLUTSpinBox->show();
 399     }
 400     else {
 401         bSplineLabel->hide();
 402         bSplineOrderSpinBox->hide();
 403         bLUTFactorLabel->hide();
 404         bLUTSpinBox->hide();
 405     }
 406 }
 407 //====================================================================
 408
 409 //====================================================================
 410 void vvResamplerDialog::UpdateGaussianFilter() {
 411     if (gaussianFilterCheckBox->isChecked()) {
 412         gaussianFilterLabel->show();
 413         xGaussianLineEdit->show();
 414         yGaussianLineEdit->show();
 415         if (mDimension > 2)
 416             zGaussianLineEdit->show();
 417     }
 418     else {
 419         gaussianFilterLabel->hide();
 420         xGaussianLineEdit->hide();
 421         yGaussianLineEdit->hide();
 422         zGaussianLineEdit->hide();
 423     }
 424 }
 425 //====================================================================
 426
 427 //====================================================================
 428 void vvResamplerDialog::SetSlicerManagers(std::vector<vvSlicerManager*> & m,int current_image_index) {
 429     mSlicerManagers = m;
 430     for (unsigned int i = 0; i < mSlicerManagers.size(); i++) {
 431         mImagesComboBox->addItem(mSlicerManagers[i]->GetFileName().c_str());
 432     }
 433     mImagesComboBox->setCurrentIndex(current_image_index);
 434 }
 435 //====================================================================
 436
 437 //====================================================================
 438 void vvResamplerDialog::Resample() {
 439
 440     // Get resampler options
 441     std::vector<double> sigma;
 442     sigma.push_back(xGaussianLineEdit->text().toDouble());
 443     sigma.push_back(yGaussianLineEdit->text().toDouble());
 444     if (mDimension > 2) sigma.push_back(zGaussianLineEdit->text().toDouble());
 445
 446
 447     // Create resampler filter
 448     clitk::ImageResampleGenericFilter::Pointer filter = clitk::ImageResampleGenericFilter::New();
 449     filter->SetOutputSize(mOutputSize);
 450     filter->SetOutputSpacing(mOutputSpacing);
 451     filter->SetInterpolationName(interpolationComboBox->currentText().toLower().toStdString());
 452
 453     if (interpolationComboBox->currentText() == "BSpline")
 454         filter->SetBSplineOrder(bSplineOrderSpinBox->value());
 455     else if (interpolationComboBox->currentText() == "Blut (faster BSpline)") {
 456         filter->SetInterpolationName("blut");
 457         filter->SetBSplineOrder(bSplineOrderSpinBox->value());
 458         filter->SetBLUTSampling(bLUTSpinBox->value());
 459     }
 460     if (gaussianFilterCheckBox->isChecked())
 461         filter->SetGaussianSigma(sigma);
 462     //  filter->SetOutputFileName(OutputFileName.toStdString());
 463     filter->SetDefaultPixelValue(defaultPixelValueLineEdit->text().toDouble());
 464     filter->SetInputVVImage(mCurrentImage);
 465
 466     // Go !
 467     filter->Update();
 468     mOutput = filter->GetOutputVVImage();
 469 }
 470 //====================================================================
 471 std::string vvResamplerDialog::GetOutputFileName()
 472 {
 473     QFileInfo info(mImagesComboBox->currentText());
 474     return (info.path().toStdString() + "/resampled_" + info.fileName().toStdString());
 475 }
 476
 477 #endif /* end #define _vvResamplerDialog_CXX */
 478