X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=vv%2FvvToolRigidReg.cxx;h=78e0d6625dd0640404044d247fa292b1eaa8a2cf;hb=c2e0628b1d9f0940ac192ff3683c3ede5d01ceb3;hp=3879adf474c4a06c7016f73ae2772d8fd98b4875;hpb=13105925771df98ddc6e3cd03283ed560c629704;p=clitk.git diff --git a/vv/vvToolRigidReg.cxx b/vv/vvToolRigidReg.cxx index 3879adf..78e0d66 100644 --- a/vv/vvToolRigidReg.cxx +++ b/vv/vvToolRigidReg.cxx @@ -3,7 +3,7 @@ Authors belong to: - University of LYON http://www.universite-lyon.fr/ - - Léon Bérard cancer center http://oncora1.lyon.fnclcc.fr + - Léon Bérard cancer center http://www.centreleonberard.fr - CREATIS CNRS laboratory http://www.creatis.insa-lyon.fr This software is distributed WITHOUT ANY WARRANTY; without even @@ -14,11 +14,10 @@ - BSD See included LICENSE.txt file - CeCILL-B http://www.cecill.info/licences/Licence_CeCILL-B_V1-en.html -======================================================================-====*/ +===========================================================================**/ // vv #include "vvToolRigidReg.h" -#include "vvImageReader.h" #include "vvSlicer.h" // vtk @@ -26,15 +25,17 @@ #include #include +// itk +#include + // clitk #include "clitkTransformUtilities.h" -#include "clitkAffineRegistrationGenericFilter.h" + // qt #include #include #include -#include -#include + //------------------------------------------------------------------------------ // Create the tool and automagically (I like this word) insert it in @@ -50,19 +51,28 @@ vvToolRigidReg::vvToolRigidReg(vvMainWindowBase * parent, Qt::WindowFlags f): { // GUI Initialization Ui_vvToolRigidReg::setupUi(mToolWidget); - // QSize qsize; -// qsize.setHeight(470); -// qsize.setWidth(850); -// mToolWidget->setFixedSize(qsize); - - - // Set how many inputs are needed for this tool - // mFilter = new clitk::AffineTransformGenericFilter; - mFilter = new clitk::AffineRegistrationGenericFilter; - + // Set how many inputs are needed for this tool - AddInputSelector("Select moving image",mFilter); - AddInputSelector("Select fixed image",mFilter); + AddInputSelector("Select moving image"); + + QFont font = transformationLabel->font(); + font.setStyleHint(QFont::TypeWriter); + transformationLabel->setFont(font); + + mInitialMatrix = vtkSmartPointer::New(); + + // Set slider ranges, assume degrees, will not be changed for radians + std::vector transSliders, rotSliders; + std::vector transSBs, rotSBs; + GetSlidersAndSpinBoxes(transSliders, rotSliders, transSBs, rotSBs); + for(int i=0; i<3; i++) { + transSliders[i]->setRange(-2000,2000); + rotSliders[i]->setRange(-360,360); + transSBs[i]->setRange(-2000,2000); + transSBs[i]->setDecimals(3); + rotSBs[i]->setRange(-360,360); + rotSBs[i]->setDecimals(3); + } } //------------------------------------------------------------------------------ @@ -73,653 +83,347 @@ vvToolRigidReg::~vvToolRigidReg() //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -bool vvToolRigidReg::close() +void vvToolRigidReg::Initialize() { - ResetTransform(); - RemoveOverlay(); - return vvToolWidgetBase::close(); + SetToolName("Register"); + SetToolMenuName("Register manually"); + SetToolIconFilename(":/common/icons/register.png"); + SetToolTip("Register manually."); + SetToolExperimental(false); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::reject() +void vvToolRigidReg::InputIsSelected(vvSlicerManager *input) { - - return vvToolWidgetBase::reject(); -} -//------------------------------------------------------------------------------ - -//------------------------------------------------------------------------------ -void vvToolRigidReg::GetArgsInfoFromGUI() -{ - QString str; - str.append("Configs Loaded for AutoRegister...Apply Filter..."); - tab2textedit->setText(str); - /* cmdline_parser_clitkAffineRegistration_configfile("/home/bharath/bin/AffineReg.txt",&mArgsInfo,1,1,0); - mArgsInfo.reference_arg=new char; - mArgsInfo.target_arg=new char; - mArgsInfo.output_arg=new char; - mArgsInfo.referenceMask_arg=new char; - mArgsInfo.targetMask_arg=new char; - mArgsInfo.initMatrix_arg=new char; - mArgsInfo.matrix_arg=new char;*/ - mArgsInfo.referenceMask_given=0; - mArgsInfo.reference_given=0; - mArgsInfo.reference_arg=new char; - mArgsInfo.target_given=0; - mArgsInfo.target_arg=new char; - mArgsInfo.output_given=0; - mArgsInfo.output_arg=new char; - mArgsInfo.checker_after_given=0; - mArgsInfo.checker_before_given=0; - mArgsInfo.after_given=0; - mArgsInfo.before_given=0; - // std::string strouput= "/home/bharath/bin/Lung3Doutput.mhd"; - //mArgsInfo.output_arg=const_cast(strouput.c_str()); - mArgsInfo.threads_given=1; - mArgsInfo.threads_arg=3; - mArgsInfo.normalize_flag=0; - DD("GetArgsInfoFromGUI_2"); - mArgsInfo.blur_arg=0.0; - mArgsInfo.referenceMask_arg=new char; - mArgsInfo.targetMask_arg=new char; - mArgsInfo.targetMask_given=0; - mArgsInfo.levels_given=1; - mArgsInfo.levels_arg=2; - mArgsInfo.moment_flag=1; - mArgsInfo.intThreshold_given=0; - mArgsInfo.intThreshold_arg=0.0; - mArgsInfo.transX_arg=0.0; - mArgsInfo.transY_arg=0.0; - mArgsInfo.transZ_arg=0.0; - mArgsInfo.transform_arg=2; - mArgsInfo.gradient_flag=1; - mArgsInfo.interp_given=1; - mArgsInfo.interp_arg=1; - mArgsInfo.interpOrder_given=1; - mArgsInfo.interpOrder_arg=3; - mArgsInfo.interpSF_given=1; - mArgsInfo.interpSF_arg=20;//default - mArgsInfo.metric_given=1; - mArgsInfo.metric_arg=0; - mArgsInfo.samples_arg=1;//default - mArgsInfo.stdDev_arg=0.4; - mArgsInfo.step_arg=2.0; - mArgsInfo.relax_arg=0.7; - mArgsInfo.valueTol_arg=0.01; - mArgsInfo.stepTol_arg=0.1; - mArgsInfo.gradTol_arg=1e-5; - mArgsInfo.lineAcc_arg=0.9; - mArgsInfo.convFactor_arg=1e+12; - mArgsInfo.maxIt_arg=500; - mArgsInfo.maxLineIt_arg=50; - mArgsInfo.maxEval_arg=500; - mArgsInfo.maxCorr_arg=5; - mArgsInfo.selectBound_arg=0; - mArgsInfo.inc_arg=1.2; - mArgsInfo.dec_arg=4; - mArgsInfo.optimizer_arg=1; - mArgsInfo.initMatrix_given=0; - mArgsInfo.initMatrix_arg=new char; - mArgsInfo.tWeight_given=1; - mArgsInfo.tWeight_arg=1.0; - mArgsInfo.rWeight_given=1.0; - mArgsInfo.rWeight_arg=50.0; - // std::string str="/home/bharath/bin/matrix.txt"; - // mArgsInfo.initMatrix_arg=const_cast(str.c_str()); - mArgsInfo.matrix_given=0; -} -//------------------------------------------------------------------------------ - -//------------------------------------------------------------------------------ -void vvToolRigidReg::InputIsSelected(std::vector & l) -{ - //inputs - mInput1 = l[0]; - mInput2 = l[1]; - - UpdateTextEditor(mCurrentSlicerManager->GetImage()->GetTransform()->GetMatrix(),textEdit_2); - - for(int i =0;i<4;i++) - for(int j=0;j<4;j++) - mInitialMatrix[i*4+j]=mCurrentSlicerManager->GetImage()->GetTransform()->GetMatrix()->GetElement(i,j); - - if(mInput1->GetFileName()==mInput2->GetFileName()) - { - QMessageBox::information(this, "Warning","Your Reference and Target Images are the same"); - } - mTwoInputs = true; - SetOverlay(); - mImageSize=mInput1->GetImage()->GetSize(); - SetRotationCenter(); - SetSliderRanges(); - - //connect all sigs to slots - connect(resetbutton, SIGNAL(pressed()), this, SLOT(ResetTransform())); - connect(tab2loadbutton, SIGNAL(pressed()), this, SLOT(GetArgsInfoFromGUI())); - connect(tab2applybutton, SIGNAL(pressed()), this, SLOT(AutoRegister())); - - - connect(Xval, SIGNAL(editingFinished()), this, SLOT(SetXvalue())); - connect(Yval, SIGNAL(editingFinished()), this, SLOT(SetYvalue())); - connect(Zval, SIGNAL(editingFinished()), this, SLOT(SetZvalue())); - - connect(xtrans_slider, SIGNAL(valueChanged(int)), this, SLOT(UpdateTransform_sliders())); - connect(ytrans_slider, SIGNAL(valueChanged(int)), this, SLOT(UpdateTransform_sliders())); - connect(ztrans_slider, SIGNAL(valueChanged(int)), this, SLOT(UpdateTransform_sliders())); - - - connect(xrot_slider, SIGNAL(valueChanged(int)), this, SLOT(UpdateTransform_sliders())); - connect(yrot_slider, SIGNAL(valueChanged(int)), this, SLOT(UpdateTransform_sliders())); - connect(zrot_slider, SIGNAL(valueChanged(int)), this, SLOT(UpdateTransform_sliders())); - - connect(xtrans_sb, SIGNAL(valueChanged(double)), this, SLOT(UpdateTransform_sb())); - connect(ytrans_sb, SIGNAL(valueChanged(double)), this, SLOT(UpdateTransform_sb())); - connect(ztrans_sb, SIGNAL(valueChanged(double)), this, SLOT(UpdateTransform_sb())); - connect(xrot_sb, SIGNAL(valueChanged(double)), this, SLOT(UpdateTransform_sb())); - connect(yrot_sb, SIGNAL(valueChanged(double)), this, SLOT(UpdateTransform_sb())); - connect(zrot_sb, SIGNAL(valueChanged(double)), this, SLOT(UpdateTransform_sb())); - - connect(loadbutton, SIGNAL(pressed()), this, SLOT(ReadFile())); - connect(savebutton, SIGNAL(pressed()), this, SLOT(SaveFile())); - + mInput = input; + HideInputSelector(); + QTabWidget * tab = dynamic_cast(mMainWindow)->GetTab(); + move(tab->mapToGlobal(tab->pos())); + resize(tab->width(), 0); + + //default image rotation center is the center of the image + QString xcord,ycord,zcord; + std::vector imageorigin; + imageorigin=mInput->GetImage()->GetOrigin(); + std::vector imageSize = mInput->GetImage()->GetSize(); + std::vector imageSpacing = mInput->GetImage()->GetSpacing(); + xcord=xcord.setNum(imageorigin[0]+imageSize[0]*imageSpacing[0]/2, 'g', 3); + ycord=ycord.setNum(imageorigin[1]+imageSize[1]*imageSpacing[1]/2, 'g', 3); + zcord=zcord.setNum(imageorigin[2]+imageSize[2]*imageSpacing[2]/2, 'g', 3); + Xval->setText(xcord); + Yval->setText(ycord); + Zval->setText(zcord); + + //backup original matrix + for(int j=0; j<4; j++) + for(int i=0; i<4; i++) + mInitialMatrix->SetElement(i,j, mCurrentSlicerManager->GetImage()->GetTransform()->GetMatrix()->GetElement(i,j)); + QString origTransformString = dynamic_cast(mMainWindow)->Get4x4MatrixDoubleAsString(mInitialMatrix); + transformationLabel->setText(origTransformString); + SetTransform(mInitialMatrix); + + //connect all sigs to slots + connect(resetbutton, SIGNAL(pressed()), this, SLOT(ResetTransform())); + connect(loadbutton, SIGNAL(pressed()), this, SLOT(LoadFile())); + connect(savebutton, SIGNAL(pressed()), this, SLOT(SaveFile())); + + connect(xtrans_slider, SIGNAL(valueChanged(int)), this, SLOT(SliderChange(int))); + connect(ytrans_slider, SIGNAL(valueChanged(int)), this, SLOT(SliderChange(int))); + connect(ztrans_slider, SIGNAL(valueChanged(int)), this, SLOT(SliderChange(int))); + connect(xrot_slider, SIGNAL(valueChanged(int)), this, SLOT(SliderChange(int))); + connect(yrot_slider, SIGNAL(valueChanged(int)), this, SLOT(SliderChange(int))); + connect(zrot_slider, SIGNAL(valueChanged(int)), this, SLOT(SliderChange(int))); + connect(xtrans_sb, SIGNAL(valueChanged(double)), this, SLOT(SpinBoxChange(double))); + connect(ytrans_sb, SIGNAL(valueChanged(double)), this, SLOT(SpinBoxChange(double))); + connect(ztrans_sb, SIGNAL(valueChanged(double)), this, SLOT(SpinBoxChange(double))); + connect(xrot_sb, SIGNAL(valueChanged(double)), this, SLOT(SpinBoxChange(double))); + connect(yrot_sb, SIGNAL(valueChanged(double)), this, SLOT(SpinBoxChange(double))); + connect(zrot_sb, SIGNAL(valueChanged(double)), this, SLOT(SpinBoxChange(double))); + + connect(stepTransSpinBox, SIGNAL(valueChanged(double)), this, SLOT(SetTranslationStep(double))); + connect(stepRotSpinBox, SIGNAL(valueChanged(double)), this, SLOT(SetRotationStep(double))); + + connect(checkBoxDegrees, SIGNAL(stateChanged(int)), this, SLOT(ToggleSpinBoxAnglesUnit())); + + connect(Xval, SIGNAL(editingFinished()), this, SLOT(ChangeOfRotationCenter())); + connect(Yval, SIGNAL(editingFinished()), this, SLOT(ChangeOfRotationCenter())); + connect(Zval, SIGNAL(editingFinished()), this, SLOT(ChangeOfRotationCenter())); + + // Init step modifiers + stepTransSpinBox->setValue(1.); + stepRotSpinBox->setValue(1.); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ void vvToolRigidReg::apply() { - RemoveOverlay(); + vvToolWidgetBase::close(); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::SetOverlay() +bool vvToolRigidReg::close() { - for (int i =0; iNumberOfSlicers(); i++) { - mCurrentSlicerManager->GetSlicer(i)->SetOverlay(mInput2->GetImage()); - mCurrentSlicerManager->GetSlicer(i)->SetActorVisibility("overlay",0,true); - mCurrentSlicerManager->SetColorMap(); - mCurrentSlicerManager->Render(); - } + QString warning = "Are you sure you want to reset the original transform?"; + QMessageBox msgBox(QMessageBox::Warning, tr("Reset transform"),warning, 0, this); + msgBox.addButton(tr("Yes"), QMessageBox::AcceptRole); + msgBox.addButton(tr("No"), QMessageBox::RejectRole); + if (msgBox.exec() == QMessageBox::AcceptRole) { + SetTransform(mInitialMatrix); + return vvToolWidgetBase::close(); } -//------------------------------------------------------------------------------ - -//------------------------------------------------------------------------------ -void vvToolRigidReg::RemoveOverlay() -{ - for(int i=0;iNumberOfSlicers();i++) - { - mInput1->RemoveActor("overlay",0); - mInput1->SetColorMap(0); - mInput1->Render(); - hide(); - } + return false; } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::SetXvalue() +void vvToolRigidReg::reject() { - QString xstr = Xval->text(); + return vvToolWidgetBase::reject(); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::SetYvalue() +void vvToolRigidReg::SetTranslationStep(double v) { - QString ystr = Yval->text(); + xtrans_sb->setSingleStep(v); + ytrans_sb->setSingleStep(v); + ztrans_sb->setSingleStep(v); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::SetZvalue() +void vvToolRigidReg::SetRotationStep(double v) { - QString zstr = Zval->text(); + xrot_sb->setSingleStep(v); + yrot_sb->setSingleStep(v); + zrot_sb->setSingleStep(v); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::SetTransform(double tX, double tY, double tZ, double aX, double aY, double aZ,bool update) +void vvToolRigidReg::SliderChange(int newVal) { - vtkSmartPointer transform = mInput1->GetImage()->GetTransform(); - transform->PostMultiply(); - //Rotations - if (aX!=0 || aY!=0 || aZ!=0) { - double x, y ,z; - x= Xval->text().toDouble(); - y= Yval->text().toDouble(); - z= Zval->text().toDouble(); - transform->Translate(-x,-y,-z); - if (aX!=0) transform->RotateX(aX); - if (aY!=0) transform->RotateY(aY); - if (aZ!=0) transform->RotateZ(aZ); - transform->Translate(x,y,z); + std::vector transSliders, rotSliders; + std::vector transSBs, rotSBs; + GetSlidersAndSpinBoxes(transSliders, rotSliders, transSBs, rotSBs); + for(int i=0; i<3; i++) { + if(transSliders[i] == QObject::sender()) { + transSBs[i]->setValue(newVal); + } + if(rotSliders[i] == QObject::sender()) { + double rad = (checkBoxDegrees->checkState()==Qt::Unchecked)?itk::Math::pi/180.:1.; + rotSBs[i]->setValue(newVal*rad); + } } - //Translations - if (tX!=0||tY!=0||tZ!=0) - transform->Translate(tX*mInput1->GetImage()->GetSpacing()[0],tY*mInput1->GetImage()->GetSpacing()[1],tZ*mInput1->GetImage()->GetSpacing()[2]); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::SetTransform(vtkMatrix4x4 *matrix) +void vvToolRigidReg::SpinBoxChange(double newVal) { - vtkSmartPointer transform=vtkSmartPointer::New(); - for(int i=0; i<4;i++) - for(int j=0;j<4;j++) - mCurrentSlicerManager->GetImage()->GetTransform()->GetMatrix()->SetElement(i,j,matrix->GetElement(i,j)); - - Render(); -} -//------------------------------------------------------------------------------ - -//------------------------------------------------------------------------------ -void vvToolRigidReg::Render() -{ - for (int i=0; iNumberOfSlicers(); i++) { - mCurrentSlicerManager->GetSlicer(i)->ForceUpdateDisplayExtent(); - mCurrentSlicerManager->GetSlicer(i)->Render(); - } -} -//------------------------------------------------------------------------------ - -//------------------------------------------------------------------------------ - void vvToolRigidReg::UpdateTextEditor(vtkMatrix4x4 *matrix,QTextEdit* textEdit) -{ - QFont font=QFont("Times New Roman",11); - textEdit->setCurrentFont(font); - textEdit->update(); - - QString str1,str2,str3; - QColor color; - textEdit->clear(); - textEdit->setAcceptRichText(true); - str2=textEdit->toPlainText(); - str2.append("#Rotation Center(mm): \n#"); - textEdit->setText(str2); - - str2=textEdit->toPlainText(); - textEdit->setTextColor(QColor(255,0,0)); - str2.append(str3.append(Xval->text())); - textEdit->setText(str2); - str3.clear(); - - str2=textEdit->toPlainText(); - str2.append("\t"); - textEdit->setText(str2); - - str2=textEdit->toPlainText(); - str2.append(str3.append(Yval->text())); - textEdit->setText(str2); - str3.clear(); - - str2=textEdit->toPlainText(); - str2.append("\t"); - textEdit->setText(str2); - - - str2=textEdit->toPlainText(); - str2.append(str3.append(Zval->text())); - textEdit->setText(str2); - - - str2=textEdit->toPlainText(); - str2.append("\n"); - textEdit->setText(str2); - - - str2=textEdit->toPlainText(); - str2.append("#Transformation Matrix(mm):\n"); - textEdit->setText(str2); - for(int i=0;i<4;i++) - { - for(int j=0;j<4;j++) - { - str2=textEdit->toPlainText(); - // str2.append("\t"+str1.setNum(matrix->Element[i][j])); - str2.append(QString("%1\t").arg(str1.setNum(matrix->Element[i][j]),2)); - textEdit->setText(str2); - } - str2=textEdit->toPlainText(); - str2.append("\n"); - textEdit->setText(str2); + std::vector transSliders, rotSliders; + std::vector transSBs, rotSBs; + GetSlidersAndSpinBoxes(transSliders, rotSliders, transSBs, rotSBs); + for(int i=0; i<3; i++) { + if(transSBs[i] == QObject::sender()) { + transSliders[i]->blockSignals(true); + transSliders[i]->setValue(itk::Math::Round(newVal)); + transSliders[i]->blockSignals(false); + } + if(rotSBs[i] == QObject::sender()) { + double rad = (checkBoxDegrees->checkState()==Qt::Unchecked)?180./itk::Math::pi:1.; + rotSliders[i]->blockSignals(true); + rotSliders[i]->setValue(itk::Math::Round(newVal*rad)); + rotSliders[i]->blockSignals(false); } - //QString str = QFileDialog::getOpenFileName(); - textEdit->setTextColor(QColor(255,0,0)); - textEdit->setFont(QFont("courrier new",12,4,true)); - textEdit->toPlainText().toAscii(); + } - str2=textEdit->toPlainText(); - textEdit->setText(str2); + // Compute transform and set + vtkSmartPointer transform_final=mInput->GetImage()->GetTransform(); + transform_final->Identity(); + transform_final->PostMultiply(); + + // Rotations + double x=0, y=0 ,z=0; + x= Xval->text().toDouble(); + y= Yval->text().toDouble(); + z= Zval->text().toDouble(); + double rad = (checkBoxDegrees->checkState()==Qt::Unchecked)?180./itk::Math::pi:1.; + transform_final->Translate(-x,-y,-z); + transform_final->RotateY(yrot_sb->value()*rad); + transform_final->RotateX(xrot_sb->value()*rad); + transform_final->RotateZ(zrot_sb->value()*rad); + transform_final->Translate(x,y,z); + + // Translation + transform_final->Translate(xtrans_sb->value(), + ytrans_sb->value(), + ztrans_sb->value()); + transform_final->Update(); + SetTransform(transform_final->GetMatrix()); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::UpdateTransform_sliders() +void vvToolRigidReg::ToggleSpinBoxAnglesUnit() { - InitializeSliders(xtrans_slider->value()*mInput1->GetImage()->GetSpacing()[0], - ytrans_slider->value()*mInput1->GetImage()->GetSpacing()[1], - ztrans_slider->value()*mInput1->GetImage()->GetSpacing()[2], - xrot_slider->value(),yrot_slider->value(),zrot_slider->value(),false); - UpdateTransform(true); - // Render(); + double rad = (checkBoxDegrees->checkState()==Qt::Unchecked)?itk::Math::pi/180.:180./itk::Math::pi; + std::vector transSliders, rotSliders; + std::vector transSBs, rotSBs; + GetSlidersAndSpinBoxes(transSliders, rotSliders, transSBs, rotSBs); + for(int i=0; i<3; i++) { + rotSBs[i]->blockSignals(true); + rotSBs[i]->setValue(rotSBs[i]->value()*rad); + rotSBs[i]->blockSignals(false); + } } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::UpdateTransform_sb() +void vvToolRigidReg::SaveFile() { - InitializeSliders(xtrans_sb->value(), - ytrans_sb->value(), - ztrans_sb->value(), - xrot_sb->value(),yrot_sb->value(),zrot_sb->value(),false); - DD(xtrans_sb->value()); - UpdateTransform(false); - // Render(); -} -//------------------------------------------------------------------------------ - -//------------------------------------------------------------------------------ -void vvToolRigidReg::AutoRegister() -{ - if (!mCurrentSlicerManager) close(); - QApplication::setOverrideCursor(QCursor(Qt::WaitCursor)); - - std::vector inputs; - // Input - inputs.push_back(mInput1->GetImage()); - inputs.push_back(mInput2->GetImage()); - // Check input type - // Main filter - clitk::AffineRegistrationGenericFilter::Pointer filter = - clitk::AffineRegistrationGenericFilter::New(); - filter->SetInputVVImages(inputs); - filter->SetArgsInfo(mArgsInfo); - DD("ArgsInfo given in"); - filter->EnableReadOnDisk(false); - filter->Update(); - DD("I am done...! Updated"); - vvImage::Pointer output = filter->GetOutputVVImage(); - DD("filter getoutput done..."); - std::ostringstream osstream; - osstream << "Registered" << "_ " - << mCurrentSlicerManager->GetSlicer(0)->GetFileName() << ".mhd"; - AddImage(output,osstream.str()); - - QApplication::restoreOverrideCursor(); - close(); + //Write the Transformation Matrix + std::string absPath = mCurrentSlicerManager->GetFileName(); + absPath = itksys::SystemTools::GetFilenameWithoutExtension(absPath) + std::string(".mat"); + QString filename = QFileDialog::getSaveFileName(this, tr("Save Transformation Matrix File"), + absPath.c_str(), + tr("Text (*.mat *.txt *.doc *.rtf)")); + + QFile file(filename); + if (file.open(QFile::WriteOnly | QFile::Truncate)) { + vtkMatrix4x4* matrix = mCurrentSlicerManager->GetImage()->GetTransform()->GetMatrix(); + QString matrixStr = dynamic_cast(mMainWindow)->Get4x4MatrixDoubleAsString(matrix,16); + QTextStream out(&file); + out << matrixStr; + } + else + { + QMessageBox::information(this,"Error","Unable to open file for writing"); + } } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::UpdateTransform(bool slider_enabled) +void vvToolRigidReg::LoadFile() { - vtkSmartPointer transform_final=vtkSmartPointer::New(); - transform_final->SetMatrix(mInitialMatrix); - transform_final->PostMultiply(); - //Rotations - double x=0, y=0 ,z=0; - x= Xval->text().toDouble(); - y= Yval->text().toDouble(); - z= Zval->text().toDouble(); - transform_final->Translate(-x,-y,-z); - if(slider_enabled){ - transform_final->RotateY(yrot_slider->value()); - transform_final->RotateX(xrot_slider->value()); - transform_final->RotateZ(zrot_slider->value()); - } - else{ - transform_final->RotateY(yrot_sb->value()); - transform_final->RotateX(xrot_sb->value()); - transform_final->RotateZ(zrot_sb->value()); - } - transform_final->Translate(x,y,z); - transform_final->PreMultiply(); - if(slider_enabled){ - transform_final->Translate(xtrans_slider->value()*mInput1->GetImage()->GetSpacing()[0],0,0); - transform_final->Translate(0,ytrans_slider->value()*mInput1->GetImage()->GetSpacing()[1],0); - transform_final->Translate(0,0,ztrans_slider->value()*mInput1->GetImage()->GetSpacing()[2]); - } - else{ - transform_final->Translate(xtrans_sb->value(),0,0); - transform_final->Translate(0,ytrans_sb->value(),0); - transform_final->Translate(0,0,ztrans_sb->value()); - } - SetTransform(transform_final->GetMatrix()); - UpdateTextEditor(transform_final->GetMatrix(),textEdit); + //Open File to read the transformation parameters + QString file = QFileDialog::getOpenFileName( + this, + "Choose the filename for the transformation matrix", + vtksys::SystemTools::GetFilenamePath(mCurrentSlicerManager->GetFileName()).c_str(), + "Text (*.mat *.txt *.rtf *.doc)"); + if (file.isEmpty()) + return; + + + itk::Matrix itkMat = clitk::ReadMatrix3D(file.toStdString()); + vtkSmartPointer matrix = vtkSmartPointer::New(); + matrix->Identity(); + for(int j=0; j<4; j++) + for(int i=0; i<4; i++) + matrix->SetElement(j,i,itkMat[j][i]); + SetTransform(matrix); } //------------------------------------------------------------------------------ - - + //------------------------------------------------------------------------------ -void vvToolRigidReg::SaveFile() +void vvToolRigidReg::ChangeOfRotationCenter() { - //Write the Transformation Matrix - QString f1 = QFileDialog::getSaveFileName(this, tr("Save Transformation Matrix File"), - mMainWindow->GetInputPathName(), - tr("Text (*.mat *.txt *.doc *.rtf)")); - QFile file1(f1); - std::vector transparameters; - QString line1; - - for(int i =0;i<4;i++) - for(int j=0;j<4;j++) - transparameters.push_back(line1.setNum(mCurrentSlicerManager->GetImage()->GetTransform()->GetMatrix()->Element[i][j])); - - if (file1.open(QFile::WriteOnly | QFile::Truncate)) { - QTextStream out1(&file1); - for(int i =0;i<4;i++){ - for(int j=0;j<4;j++) { - out1<GetImage()->GetTransform()->GetMatrix()); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::ReadFile() +void vvToolRigidReg::ResetTransform() { - std::string x; - QString center; - double * orientations=new double[3]; - double * translations=new double[3]; - vtkMatrix4x4 *matrix=vtkMatrix4x4::New(); - vtkSmartPointer transform = mCurrentSlicerManager->GetImage()->GetTransform(); - - //Open File to read the transformation parameters - QString file1 = QFileDialog::getOpenFileName( - this, - "Choose the Transformation Parameters file", - mMainWindow->GetInputPathName(), - "Text (*.mat *.txt *.rtf *.doc)"); - if (file1.isEmpty()) - return; - QFile Qfile1(file1); - // ifstream readfile; - std::string transfile= file1.toStdString(); - std::string filename1(transfile); - std::ifstream f1(filename1.c_str()); - if(f1.is_open()) - { - f1.close(); - itk::Matrix itkMat = clitk::ReadMatrix3D(transfile); - for(int j=0; j<4; j++) - for(int i=0; i<4; i++) - matrix->SetElement(i,j,itkMat[i][j]); - } - UpdateTextEditor(matrix,textEdit); - transform->SetMatrix(matrix); - transform->GetOrientation(orientations); - transform->PostMultiply(); - - //Obtain the Rotation Center , set it to origin - Xval->setText(center.setNum(0)); - Yval->setText(center.setNum(0)); - Zval->setText(center.setNum(0)); - - //In the Order or Y X Z // - //now postmultiply for the rotations - SetTransform(0,0,0,0,0,-rint(orientations[2]),false); - SetTransform(0,0,0,-rint(orientations[0]),0,0,false); - SetTransform(0,0,0,0,-rint(orientations[1]),0,false); - - transform->GetPosition(translations); - transform->Identity(); - - DD(translations[0]/mInput1->GetImage()->GetSpacing()[0]); - DD(translations[1]/mInput1->GetImage()->GetSpacing()[1]); - DD(translations[2]/mInput1->GetImage()->GetSpacing()[2]); - DD(orientations[0]); - DD(orientations[1]); - DD(orientations[2]); - //set the sliders and spin box values - InitializeSliders(rint(translations[0]),rint(translations[1]) - ,rint(translations[2]),rint(orientations[0]),rint(orientations[1]),rint(orientations[2]),true); - SetTransform(matrix); - + SetTransform(mInitialMatrix); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::ResetTransform() +void vvToolRigidReg::SetTransform(vtkMatrix4x4 *matrix) { - vtkMatrix4x4 *matrix = vtkMatrix4x4::New(); - for(int i=0;i<4;i++) - for(int j=0;j<4;j++) - matrix->SetElement(i,j,mInitialMatrix[i*4+j]); - SetTransform(matrix); - SetRotationCenter(); - SetSliderRanges(); - UpdateTextEditor(matrix,textEdit); + vtkSmartPointer transform=vtkSmartPointer::New(); + mCurrentSlicerManager->GetImage()->GetTransform()->SetMatrix(matrix); + transform->Update(); + Render(); + dynamic_cast(mMainWindow)->ImageInfoChanged(); + + // Compute parameters from transfer using itk Euler transform + itk::Euler3DTransform::CenterType center; + center[0] = Xval->text().toDouble(); + center[1] = Yval->text().toDouble(); + center[2] = Zval->text().toDouble(); + itk::Euler3DTransform::MatrixType rotMat; + itk::Euler3DTransform::OutputVectorType transVec; + for(int i=0; i<3; i++) { + transVec[i] = matrix->GetElement(i,3); + for(int j=0; j<3; j++) + rotMat[i][j] = matrix->GetElement(i,j); + } + itk::Euler3DTransform::Pointer euler; + euler = itk::Euler3DTransform::New(); + euler->SetCenter(center); + euler->SetMatrix(rotMat); + euler->SetOffset(transVec); + + + // Modify GUI according to the new parameters + std::vector transSliders, rotSliders; + std::vector transSBs, rotSBs; + GetSlidersAndSpinBoxes(transSliders, rotSliders, transSBs, rotSBs); + for(int i=0; i<3; i++) { + transSBs[i]->blockSignals(true); + transSBs[i]->setValue( euler->GetParameters()[i+3] ); + transSBs[i]->blockSignals(false); + transSliders[i]->blockSignals(true); + transSliders[i]->setValue( itk::Math::Round(euler->GetParameters()[i+3]) ); + transSliders[i]->blockSignals(false); + double rad = (checkBoxDegrees->checkState()==Qt::Checked)?180./itk::Math::pi:1.; + rotSBs[i]->blockSignals(true); + rotSBs[i]->setValue( euler->GetParameters()[i]*rad ); + rotSBs[i]->blockSignals(false); + rotSliders[i]->blockSignals(true); + rotSliders[i]->setValue( itk::Math::Round(euler->GetParameters()[i]*180./itk::Math::pi) ); + rotSliders[i]->blockSignals(false); + } } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::SetRotationCenter() +// Just an helper function to shorten the code with loops on sliders and spinboxes +void vvToolRigidReg::GetSlidersAndSpinBoxes(std::vector&transSliders, std::vector&rotSliders, + std::vector&transSBs, std::vector&rotSBs) { - //default image rotation center is the center of the image - QString xcord,ycord,zcord; - std::vector imageorigin; - imageorigin=mInput1->GetImage()->GetOrigin(); - DD("before bug"); - xcord=xcord.setNum(imageorigin[0]+mImageSize[0]*mInput1->GetImage()->GetSpacing()[0]/2, 'g', 3); - ycord=ycord.setNum(imageorigin[1]+mImageSize[1]*mInput1->GetImage()->GetSpacing()[1]/2, 'g', 3); - zcord=zcord.setNum(imageorigin[2]+mImageSize[2]*mInput1->GetImage()->GetSpacing()[2]/2, 'g', 3); - - Xval->setText(xcord); - Yval->setText(ycord); - Zval->setText(zcord); - InitializeSliders(0,0,0,0,0,0,true); + transSliders.push_back(xtrans_slider); + transSliders.push_back(ytrans_slider); + transSliders.push_back(ztrans_slider); + + rotSliders.push_back(xrot_slider); + rotSliders.push_back(yrot_slider); + rotSliders.push_back(zrot_slider); + + transSBs.push_back(xtrans_sb); + transSBs.push_back(ytrans_sb); + transSBs.push_back(ztrans_sb); + + rotSBs.push_back(xrot_sb); + rotSBs.push_back(yrot_sb); + rotSBs.push_back(zrot_sb); } //------------------------------------------------------------------------------ //------------------------------------------------------------------------------ -void vvToolRigidReg::InitializeSliders(double xtrans,double ytrans, double ztrans, double xrot, double yrot, double zrot,bool sliders) +void vvToolRigidReg::Render() { - - xtrans_sb->blockSignals(true); - xtrans_sb->setSingleStep(mInput1->GetImage()->GetSpacing()[0]); - xtrans_sb->setValue(xtrans); - xtrans_sb->blockSignals(false); - ytrans_sb->blockSignals(true); - ytrans_sb->setSingleStep(mInput1->GetImage()->GetSpacing()[1]); - ytrans_sb->setValue(ytrans); - ytrans_sb->blockSignals(false); - ztrans_sb->blockSignals(true); - ztrans_sb->setSingleStep(mInput1->GetImage()->GetSpacing()[2]); - ztrans_sb->setValue(ztrans); - ztrans_sb->blockSignals(false); - - if(sliders){ - xtrans_slider->blockSignals(true); - xtrans_slider->setValue(rint(xtrans)); - xtrans_slider->blockSignals(false); - ytrans_slider->blockSignals(true); - ytrans_slider->setValue(rint(ytrans)); - ytrans_slider->blockSignals(false); - ztrans_slider->blockSignals(true); - ztrans_slider->setValue(rint(ztrans)); - ztrans_slider->blockSignals(false); + for (int i=0; iGetNumberOfSlicers(); i++) + { + mCurrentSlicerManager->GetSlicer(i)->ForceUpdateDisplayExtent(); + mCurrentSlicerManager->GetSlicer(i)->Render(); } - xrot_sb->blockSignals(true); - xrot_sb->setValue(xrot); - xrot_sb->blockSignals(false); - yrot_sb->blockSignals(true); - yrot_sb->setValue(yrot); - yrot_sb->blockSignals(false); - zrot_sb->blockSignals(true); - zrot_sb->setValue(zrot); - zrot_sb->blockSignals(false); - xrot_slider->blockSignals(true); - xrot_slider->setValue(xrot); - xrot_slider->blockSignals(false); - yrot_slider->blockSignals(true); - yrot_slider->setValue(yrot); - yrot_slider->blockSignals(false); - zrot_slider->blockSignals(true); - zrot_slider->setValue(zrot); - zrot_slider->blockSignals(false); } //------------------------------------------------------------------------------ -//------------------------------------------------------------------------------ -void vvToolRigidReg::SetSliderRanges() -{ - xtrans_slider->blockSignals(true); - xtrans_slider->setRange(-2000,2000); - xtrans_slider->blockSignals(false); - - ytrans_slider->blockSignals(true); - ytrans_slider->setRange(-2000,2000); - ytrans_slider->blockSignals(false); - - ztrans_slider->blockSignals(true); - ztrans_slider->setRange(-2000,2000); - ztrans_slider->blockSignals(false); - - xtrans_sb->blockSignals(true); - xtrans_sb->setRange(-2000,2000); - xtrans_sb->setDecimals(3); - xtrans_sb->blockSignals(false); - - ytrans_sb->blockSignals(true); - ytrans_sb->setRange(-2000,2000); - ytrans_sb->setDecimals(3); - ytrans_sb->blockSignals(false); - - ztrans_sb->blockSignals(true); - ztrans_sb->setRange(-2000,2000); - ztrans_sb->setDecimals(3); - ztrans_sb->blockSignals(false); - - xrot_slider->blockSignals(true); - xrot_slider->setRange(-360,360); - xrot_slider->blockSignals(false); - - yrot_slider->blockSignals(true); - yrot_slider->setRange(-360,360); - yrot_slider->blockSignals(false); - - zrot_slider->blockSignals(true); - zrot_slider->setRange(-360,360); - zrot_slider->blockSignals(false); - - xrot_sb->blockSignals(true); - xrot_sb->setRange(-360,360); - xrot_sb->blockSignals(false); - - yrot_sb->blockSignals(true); - yrot_sb->setRange(-360,360); - yrot_sb->blockSignals(false); - - zrot_sb->blockSignals(true); - zrot_sb->setRange(-360,360); - zrot_sb->blockSignals(false); -}