cxNeuroTrainingWidget.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 "cxNeuroTrainingWidget.h"
#include <boost/bind.hpp>
#include "cxApplication.h"
#include "cxSettings.h"
#include "cxLogger.h"
#include "cxStateService.h"
#include "cxRegistrationWidget.h"
#include "cxStreamerServiceUtilities.h"
#include "cxRegistrationService.h"
#include "cxMesh.h"
#include "cxRegServices.h"
#include "cxTrackingService.h"

namespace cx
{

NeuroTrainingWidget::NeuroTrainingWidget(RegServicesPtr services, ctkPluginContext* context, QWidget *parent) :
        TrainingWidget(services, "NeuroSimulatorWidget", "Neuro Simulator", parent),
        mPluginContext(context)
{
        func_t transitionToStep1 = boost::bind(&NeuroTrainingWidget::onImport, this);
        func_t transitionToStep2 = boost::bind(&NeuroTrainingWidget::onRegisterStep, this);
        func_t transitionToStep3 = boost::bind(&NeuroTrainingWidget::onUse2DUSStep, this);
        func_t transitionToStep4 = boost::bind(&NeuroTrainingWidget::on3DUSAcqStep, this);
        func_t transitionToStep5 = boost::bind(&NeuroTrainingWidget::onShowAllUSStep, this);

        TrainingWidget::registrateTransition(transitionToStep1);
        TrainingWidget::registrateTransition(transitionToStep2);
        TrainingWidget::registrateTransition(transitionToStep3);
        TrainingWidget::registrateTransition(transitionToStep4);
        TrainingWidget::registrateTransition(transitionToStep5);
}

void NeuroTrainingWidget::onImport()
{
        this->setUSSimulatorInput(this->getFirstUSVolume());
//      this->makeUnavailable("Kaisa");
        this->makeUnavailable("US", true);

        this->changeWorkflowToImport();
}

void NeuroTrainingWidget::setUSSimulatorInput(QString usUid)
{
        if(usUid.isEmpty())
        {
                CX_LOG_DEBUG() << "NeuroTrainingWidget::setUSSimulatorInput: usUid is empty, not changing existing.";
                return;
        }
        cx::StreamerService* streamerService = cx::StreamerServiceUtilities::getStreamerServiceFromType("ussimulator_streamer", mPluginContext);
        SimulatedStreamerService* simulatorStreamerService = dynamic_cast<SimulatedStreamerService*>(streamerService);
        if(simulatorStreamerService)
        {
                CX_LOG_INFO() << "Setting US simulator input to: " << usUid;
                simulatorStreamerService->setImageToStream(usUid);
        }
        else
                CX_LOG_WARNING() << "Cannot find SimulatedImageStreamerService";
}

void NeuroTrainingWidget::changeImageToPatientRegistrationToFast()
{
        QWidget* registrationWidget = findMainWindowChildWithObjectName<QWidget*>("org_custusx_registration_gui_widget");
        if(registrationWidget)
        {
                RegistrationWidget* widget =  dynamic_cast<RegistrationWidget*>(registrationWidget);
                if(widget)
                {
                        widget->selectRegistrationMethod("ImageToPatient", "Fast");
                }
        }
}

void NeuroTrainingWidget::onRegisterStep()
{
        this->startTracking();

        this->changeWorkflowToRegistration();
        this->setSurfaceForPointCloudRegistration("Kaisa");
        this->changeImageToPatientRegistrationToFast();
}

void NeuroTrainingWidget::setSurfaceForPointCloudRegistration(QString uidPart)
{
        MeshPtr mesh = this->getMesh(uidPart);
        if(mesh)
                CX_LOG_INFO() << "Setting registration fixed data to: " << mesh->getUid();
        mServices->registration()->setFixedData(mesh);
}

void NeuroTrainingWidget::onUse2DUSStep()
{
        this->changeWorkflowToUSAcquisition();
}

void NeuroTrainingWidget::on3DUSAcqStep()
{
        this->changeWorkflowToUSAcquisition();
}

void NeuroTrainingWidget::onShowAllUSStep()
{
        this->makeAvailable("US", true);
        this->changeWorkflowToNavigation();
}

void NeuroTrainingWidget::startTracking()
{
        mServices->tracking()->setState(Tool::tsTRACKING);;
}

void NeuroTrainingWidget::changeWorkflowToImport()
{
        mServices->state()->setWorkFlowState("PatientDataUid");
}

void NeuroTrainingWidget::changeWorkflowToRegistration()
{
        mServices->state()->setWorkFlowState("RegistrationUid");
}

void NeuroTrainingWidget::changeWorkflowToUSAcquisition()
{
        mServices->state()->setWorkFlowState("IntraOpImagingUid");
}

void NeuroTrainingWidget::changeWorkflowToNavigation()
{
        mServices->state()->setWorkFlowState("NavigationUid");
}

} // cx