cxPatientLandMarksWidget.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 "cxPatientLandMarksWidget.h"

#include <QPushButton>
#include <QTableWidget>
#include <QLabel>
#include <QCheckBox>

#include "cxActiveToolProxy.h"
#include "cxLandmarkListener.h"
#include "cxSettings.h"
#include "cxPatientModelService.h"
#include "cxTrackingService.h"
#include "cxRegistrationService.h"
#include "cxViewService.h"
#include "cxTypeConversions.h"
#include "cxReporter.h"
#include "cxLandmark.h"
#include "cxImage.h"
#include "cxActiveData.h"

namespace cx
{

PatientLandMarksWidget::PatientLandMarksWidget(RegServicesPtr services,
        QWidget* parent, QString objectName, QString windowTitle) :
        LandmarkRegistrationWidget(services, parent, objectName, windowTitle),
        mToolSampleButton(new QPushButton("Sample Tool", this))
{
        mLandmarkListener->useI2IRegistration(false);

        connect(services->patient().get(), &PatientModelService::rMprChanged, this, &PatientLandMarksWidget::setModified);

        //buttons
        mToolSampleButton->setDisabled(true);
        connect(mToolSampleButton, SIGNAL(clicked()), this, SLOT(toolSampleButtonClickedSlot()));

        mRemoveLandmarkButton = new QPushButton("Clear", this);
        mRemoveLandmarkButton->setToolTip("Clear selected landmark");
        connect(mRemoveLandmarkButton, &QPushButton::clicked, this, &PatientLandMarksWidget::removeLandmarkButtonClickedSlot);

        //toolmanager
        mActiveToolProxy = ActiveToolProxy::New(services->tracking());
        connect(mActiveToolProxy.get(), SIGNAL(toolVisible(bool)), this, SLOT(updateToolSampleButton()));
        connect(mActiveToolProxy.get(), SIGNAL(activeToolChanged(const QString&)), this, SLOT(updateToolSampleButton()));

        connect(settings(), &Settings::valueChangedFor, this, &PatientLandMarksWidget::globalConfigurationFileChangedSlot);

        //layout
        mVerticalLayout->addWidget(mLandmarkTableWidget);
        mVerticalLayout->addWidget(mToolSampleButton);
        mVerticalLayout->addWidget(mAvarageAccuracyLabel);
        mVerticalLayout->addWidget(mRemoveLandmarkButton);

        mMouseClickSample->setText("Sample with mouse clicks in anyplane view.");
        mMouseClickSample->show();
        connect(mMouseClickSample, &QCheckBox::stateChanged, this, &PatientLandMarksWidget::mouseClickSampleStateChanged);

        mVerticalLayout->addWidget(mMouseClickSample);

        this->updateToolSampleButton();
}

PatientLandMarksWidget::~PatientLandMarksWidget()
{
}

void PatientLandMarksWidget::globalConfigurationFileChangedSlot(QString key)
{
        if (key == "giveManualToolPhysicalProperties")
                this->updateToolSampleButton();
}

void PatientLandMarksWidget::updateToolSampleButton()
{
        ToolPtr tool = mServices->tracking()->getActiveTool();

        bool enabled = tool && tool->getVisible() && (!tool->hasType(Tool::TOOL_MANUAL) || settings()->value("giveManualToolPhysicalProperties").toBool()); // enable only for non-manual tools.
        mToolSampleButton->setEnabled(enabled);

        if (mServices->tracking()->getActiveTool())
                mToolSampleButton->setText("Sample " + qstring_cast(tool->getName()));
        else
                mToolSampleButton->setText("No tool");
}

void PatientLandMarksWidget::toolSampleButtonClickedSlot()
{
        ToolPtr tool = mServices->tracking()->getActiveTool();

        if (!tool)
        {
                reportError("mToolToSample is NULL!");
                return;
        }
        //TODO What if the reference frame isnt visible?
        Transform3D lastTransform_prMt = tool->get_prMt();
        Vector3D p_pr = lastTransform_prMt.coord(Vector3D(0, 0, tool->getTooltipOffset()));

        // TODO: do we want to allow sampling points not defined in image??
        if (mActiveLandmark.isEmpty() && !mServices->patient()->getLandmarkProperties().empty())
                mActiveLandmark = mServices->patient()->getLandmarkProperties().begin()->first;

        mServices->patient()->getPatientLandmarks()->setLandmark(Landmark(mActiveLandmark, p_pr));
        reporter()->playSampleSound();

        this->activateLandmark(this->getNextLandmark());

        this->performRegistration(); // automatic when sampling in physical patient space (Mantis #0000674)s
}

void PatientLandMarksWidget::showEvent(QShowEvent* event)
{
        mServices->view()->setRegistrationMode(rsPATIENT_REGISTRATED);
        LandmarkRegistrationWidget::showEvent(event);
        mMouseClickSample->setChecked(true);
}

void PatientLandMarksWidget::hideEvent(QHideEvent* event)
{
        mServices->view()->setRegistrationMode(rsNOT_REGISTRATED);
        LandmarkRegistrationWidget::hideEvent(event);
}

void PatientLandMarksWidget::removeLandmarkButtonClickedSlot()
{
        QString next = this->getNextLandmark();
        mServices->patient()->getPatientLandmarks()->removeLandmark(mActiveLandmark);
        this->activateLandmark(next);
}

void PatientLandMarksWidget::cellClickedSlot(int row, int column)
{
        LandmarkRegistrationWidget::cellClickedSlot(row, column);

        mRemoveLandmarkButton->setEnabled(true);
}

void PatientLandMarksWidget::prePaintEvent()
{
        LandmarkRegistrationWidget::prePaintEvent();

        std::vector<Landmark> landmarks = this->getAllLandmarks();
        mRemoveLandmarkButton->setEnabled(!landmarks.empty() && !mActiveLandmark.isEmpty());
}

LandmarkMap PatientLandMarksWidget::getTargetLandmarks() const
{
        return mServices->patient()->getPatientLandmarks()->getLandmarks();
}

Transform3D PatientLandMarksWidget::getTargetTransform() const
{
        Transform3D rMpr = mServices->patient()->get_rMpr();
        return rMpr;
}

void PatientLandMarksWidget::setTargetLandmark(QString uid, Vector3D p_target)
{
        mServices->patient()->getPatientLandmarks()->setLandmark(Landmark(uid, p_target));
        reporter()->playSampleSound();
}

void PatientLandMarksWidget::performRegistration()
{
        ActiveDataPtr activeData = mServices->patient()->getActiveData();
        if (!mServices->registration()->getFixedData())
                mServices->registration()->setFixedData(activeData->getActive<Image>());

        if (mServices->patient()->getPatientLandmarks()->getLandmarks().size() < 3)
                return;

        mServices->registration()->doPatientRegistration();

        this->updateAverageAccuracyLabel();
}

QString PatientLandMarksWidget::getTargetName() const
{
        return "Patient";
}

void PatientLandMarksWidget::pointSampled(Vector3D p_r)
{
        QTableWidgetItem* item = getLandmarkTableItem();
        if(!item)
        {
                CX_LOG_WARNING() << "PatientLandMarksWidget::pointSampled() Cannot get item from mLandmarkTableWidget";
                return;
        }
        QString uid = item->data(Qt::UserRole).toString();

        Transform3D rMtarget = this->getTargetTransform();
        Vector3D p_target = rMtarget.inv().coord(p_r);

        this->setTargetLandmark(uid, p_target);
        this->activateLandmark(this->getNextLandmark());
        this->performRegistration();
}
} //cx