developer_doc/nightly/cx_landmark_image2_image_registration_widget_8cpp_source.html

/*=========================================================================

This file is part of CustusX, an Image Guided Therapy Application.


Copyright (c) SINTEF Department of Medical Technology.

All rights reserved.


CustusX is released under a BSD 3-Clause license.


See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.

=========================================================================*/

#include "cxLandmarkImage2ImageRegistrationWidget.h"


#include <sstream>

#include <QVBoxLayout>

#include <QPushButton>

#include <QTableWidget>

#include <QTableWidgetItem>

#include <QHeaderView>

#include <QLabel>

#include <QSlider>

#include <QCheckBox>

#include <vtkDoubleArray.h>

#include <vtkImageData.h>


#include "cxLabeledComboBoxWidget.h"

#include "cxSettings.h"

#include "cxLandmarkRep.h"

#include "cxView.h"

#include "cxRegistrationProperties.h"

#include "cxRegistrationService.h"

#include "cxViewService.h"

#include "cxRepContainer.h"

#include "cxLandmarkListener.h"


namespace cx

{

LandmarkImage2ImageRegistrationWidget::LandmarkImage2ImageRegistrationWidget(RegServicesPtr services,

        QWidget* parent, QString objectName, QString windowTitle) :

        LandmarkRegistrationWidget(services, parent, objectName, windowTitle)

{

        mLandmarkTableWidget->hide();


        mLandmarkListener->useI2IRegistration();


        mFixedProperty.reset(new StringPropertyRegistrationFixedImage(services->registration(), services->patient()));

        mMovingProperty.reset(new StringPropertyRegistrationMovingImage(services->registration(), services->patient()));


        mTranslationCheckBox = new QCheckBox("Translation only", this);

        mTranslationCheckBox->setChecked(settings()->value("registration/I2ILandmarkTranslation", false).toBool());

        connect(mTranslationCheckBox, SIGNAL(toggled(bool)), this, SLOT(translationCheckBoxChanged()));


        mRegisterButton = new QPushButton("Register", this);

        mRegisterButton->setToolTip("Perform registration");

        connect(mRegisterButton, SIGNAL(clicked()), this, SLOT(registerSlot()));


        mVerticalLayout->addWidget(new LabeledComboBoxWidget(this, mFixedProperty));

        mVerticalLayout->addWidget(new LabeledComboBoxWidget(this, mMovingProperty));

        mVerticalLayout->addWidget(mTranslationCheckBox);

        mVerticalLayout->addWidget(mAvarageAccuracyLabel);


        QHBoxLayout* regLayout = new QHBoxLayout;

        regLayout->addWidget(mRegisterButton);

        mVerticalLayout->addLayout(regLayout);

        mVerticalLayout->addStretch();

}


void LandmarkImage2ImageRegistrationWidget::translationCheckBoxChanged()

{

        settings()->setValue("registration/I2ILandmarkTranslation", mTranslationCheckBox->isChecked());

}


void LandmarkImage2ImageRegistrationWidget::registerSlot()

{

        this->performRegistration();

}


LandmarkImage2ImageRegistrationWidget::~LandmarkImage2ImageRegistrationWidget()

{

}


QString LandmarkImage2ImageRegistrationWidget::defaultWhatsThis() const

{

        return "<html>"

                "<h3>Landmark based image to image registration.</h3>"

                "<p>Register moving image to fixed image. </p>"

                "</html>";

}


void LandmarkImage2ImageRegistrationWidget::showEvent(QShowEvent* event)

{

        mServices->view()->setRegistrationMode(rsIMAGE_REGISTRATED);

        LandmarkRegistrationWidget::showEvent(event);

}


void LandmarkImage2ImageRegistrationWidget::hideEvent(QHideEvent* event)

{

        mServices->view()->setRegistrationMode(rsNOT_REGISTRATED);

        LandmarkRegistrationWidget::hideEvent(event);

}


void LandmarkImage2ImageRegistrationWidget::prePaintEvent()

{

}


LandmarkMap LandmarkImage2ImageRegistrationWidget::getTargetLandmarks() const

{

        ImagePtr moving = boost::dynamic_pointer_cast<Image>(mServices->registration()->getMovingData());


        if (moving)

                return moving->getLandmarks()->getLandmarks();

        else

                return LandmarkMap();

}


void LandmarkImage2ImageRegistrationWidget::performRegistration()

{

        mServices->registration()->doImageRegistration(mTranslationCheckBox->isChecked());

        this->updateAverageAccuracyLabel();

}


Transform3D LandmarkImage2ImageRegistrationWidget::getTargetTransform() const

{

        if (!mServices->registration()->getMovingData())

                return Transform3D::Identity();

        return mServices->registration()->getMovingData()->get_rMd();

}


void LandmarkImage2ImageRegistrationWidget::setTargetLandmark(QString uid, Vector3D p_target)

{

        ImagePtr image = boost::dynamic_pointer_cast<Image>(mServices->registration()->getMovingData());

        if (!image)

                return;

        image->getLandmarks()->setLandmark(Landmark(uid, p_target));

}


QString LandmarkImage2ImageRegistrationWidget::getTargetName() const

{

        DataPtr image = mServices->registration()->getMovingData();

        if (!image)

                return "None";

        return image->getName();

}


}//namespace cx