cxBronchoscopyImage2ImageRegistrationWidget.cpp

Go to the documentation of this file.

/*=========================================================================
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 "cxBronchoscopyImage2ImageRegistrationWidget.h"
#include <vtkPolyData.h>
#include "cxTransform3D.h"
#include "cxDataSelectWidget.h"
#include "cxMesh.h"
#include "cxSelectDataStringProperty.h"
#include "cxView.h"
#include "cxBronchoscopyRegistration.h"
#include "cxLogger.h"
#include "cxTypeConversions.h"
#include "cxPatientModelService.h"
#include "cxRegistrationService.h"
#include "cxViewService.h"
#include "cxStringProperty.h"
#include "cxLabeledComboBoxWidget.h"
#include "cxDoubleProperty.h"
#include "cxProfile.h"
#include "cxHelperWidgets.h"
#include "cxBoolProperty.h"
#include "cxWidgetObscuredListener.h"
#include "cxViewGroupData.h"
#include "cxHelperWidgets.h"
#include "cxRegServices.h"

namespace cx
{
BronchoscopyImage2ImageRegistrationWidget::BronchoscopyImage2ImageRegistrationWidget(RegServicesPtr services, QWidget* parent) :
    RegistrationBaseWidget(services, parent, "org_custusx_registration_method_bronchoscopy_image2image_widget",
                           "Bronchoscopy Registration"),
    mBronchoscopyRegistration(new BronchoscopyRegistration())
{
    mVerticalLayout = new QVBoxLayout(this);
}

void BronchoscopyImage2ImageRegistrationWidget::prePaintEvent()
{
    this->setup();
}

void BronchoscopyImage2ImageRegistrationWidget::setup()
{
    mOptions = profile()->getXmlSettings().descend("bronchoscopyregistrationimage2imagewidget");

    mSelectMeshFixedWidget = StringPropertySelectMesh::New(mServices->patient());
    mSelectMeshFixedWidget->setValueName("Centerline fixed image: ");

    mSelectMeshMovingWidget = StringPropertySelectMesh::New(mServices->patient());
    mSelectMeshMovingWidget->setValueName("Centerline moving image: ");

    connect(mServices->patient().get(),&PatientModelService::patientChanged,this,&BronchoscopyImage2ImageRegistrationWidget::clearDataOnNewPatient);

    mRegisterButton = new QPushButton("Register");
    connect(mRegisterButton, SIGNAL(clicked()), this, SLOT(registerSlot()));
    mRegisterButton->setToolTip(this->defaultWhatsThis());

    mVerticalLayout->setMargin(0);
    mVerticalLayout->addWidget(new DataSelectWidget(mServices->view(), mServices->patient(), this, mSelectMeshFixedWidget));
    mVerticalLayout->addWidget(new DataSelectWidget(mServices->view(), mServices->patient(), this, mSelectMeshMovingWidget));

    mVerticalLayout->addWidget(mRegisterButton);

    mVerticalLayout->addStretch();
}

QString BronchoscopyImage2ImageRegistrationWidget::defaultWhatsThis() const
{
    return QString("Registration of CT images based on airway centerlines");
}

void BronchoscopyImage2ImageRegistrationWidget::registerSlot()
{

    if(!mSelectMeshFixedWidget->getMesh())
    {
        reportError("Fixed centerline missing");
        return;
    }
    if(!mSelectMeshMovingWidget->getMesh())
    {
        reportError("Moving centerline missing");
        return;
    }

    vtkPolyDataPtr centerlineFixed = mSelectMeshFixedWidget->getMesh()->getVtkPolyData();//input
    Transform3D rMdFixed = mSelectMeshFixedWidget->getMesh()->get_rMd();

    vtkPolyDataPtr centerlineMoving = mSelectMeshMovingWidget->getMesh()->getVtkPolyData();//input
    Transform3D rMdMoving = mSelectMeshMovingWidget->getMesh()->get_rMd();

    Transform3D d_M_d = Transform3D(mBronchoscopyRegistration->runBronchoscopyRegistrationImage2Image(centerlineFixed, centerlineMoving));

    DataPtr fixedData = mSelectMeshFixedWidget->getData();
    mServices->registration()->setFixedData(fixedData);
    DataPtr movingData = mSelectMeshMovingWidget->getData();
    mServices->registration()->setMovingData(movingData);

    Transform3D delta_pre_rMd = rMdFixed * d_M_d * rMdMoving.inv();//output
    mServices->registration()->addImage2ImageRegistration(delta_pre_rMd, "Bronchoscopy: centerline to centerline");
}

void BronchoscopyImage2ImageRegistrationWidget::clearDataOnNewPatient()
{
    mMeshFixed.reset();
    mMeshMoving.reset();
}

} //namespace cx