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

#include "cxRecordSessionWidget.h"
#include "cxRecordSession.h"
#include "cxToolRep3D.h"
#include "cxToolTracer.h"
#include "cxLogger.h"
#include "cxPatientModelService.h"
#include "cxViewService.h"
#include "cxStringProperty.h"
#include "cxLabeledComboBoxWidget.h"
#include "cxTrackingService.h"
#include "cxProfile.h"
#include "cxHelperWidgets.h"
#include "cxRepContainer.h"
#include "cxWidgetObscuredListener.h"
#include "cxStringPropertySelectTool.h"
#include "cxAcquisitionService.h"
#include "cxRecordSessionSelector.h"
#include "cxBoolProperty.h"


namespace cx
{

RecordTrackingWidget::RecordTrackingWidget(XmlOptionFile options,
                                                                                   AcquisitionServicePtr acquisitionService,
                                                                                   VisServicesPtr services,
                                                                                   QString category,
                                                                                   QWidget* parent) :
        QWidget(parent),
        mServices(services),
        mOptions(options),
        mAcquisitionService(acquisitionService)
{
        QVBoxLayout* mVerticalLayout = new QVBoxLayout(this);

        mToolSelector = StringPropertySelectTool::New(services->tracking());

        mSelectRecordSession.reset(new SelectRecordSession(mOptions, acquisitionService, services));
        connect(mSelectRecordSession->getSessionSelector().get(), &StringProperty::changed, this, &RecordTrackingWidget::onMergeChanged);

        mMergeWithExistingSession = BoolProperty::initialize("mergerecording", "Merge",
                                                                                                                 "Merge new recording with selected recorded session",
                                                                                                                 false, QDomNode());
        connect(mMergeWithExistingSession.get(), &BoolProperty::changed, this, &RecordTrackingWidget::onMergeChanged);

        AcquisitionService::TYPES context(AcquisitionService::tTRACKING);
        mRecordSessionWidget = new RecordSessionWidget(mAcquisitionService, this, context, category);

        mVerticalLayout->setMargin(0);

        mToolSelectorWidget = sscCreateDataWidget(this, mToolSelector);
        mVerticalLayout->addWidget(mToolSelectorWidget);
        mVerticalLayout->addWidget(mRecordSessionWidget);
        mVerticalLayout->addWidget(sscCreateDataWidget(this, mMergeWithExistingSession));
        mVerticalLayout->addWidget(new LabeledComboBoxWidget(this, mSelectRecordSession->getSessionSelector()));

        mObscuredListener.reset(new WidgetObscuredListener(this));
        connect(mObscuredListener.get(), SIGNAL(obscured(bool)), this, SLOT(obscuredSlot(bool)));
}

void RecordTrackingWidget::displayToolSelector(bool on)
{
        mToolSelectorWidget->setVisible(on);
}

StringPropertyPtr RecordTrackingWidget::getSessionSelector()
{
        return mSelectRecordSession->getSessionSelector();
}

void RecordTrackingWidget::acquisitionStarted()
{
        mRecordingTool = this->getSuitableRecordingTool();

        ToolRep3DPtr activeRep3D = this->getToolRepIn3DView();
        if (activeRep3D)
        {
                if (!mMergeWithExistingSession->getValue())
                        activeRep3D->getTracer()->clear();
                activeRep3D->getTracer()->setColor(QColor("magenta"));
                activeRep3D->getTracer()->start();
        }
}

void RecordTrackingWidget::acquisitionStopped()
{
        QString newUid = mAcquisitionService->getLatestSession()->getUid();
        mSelectRecordSession->getSessionSelector()->setValue(newUid);

        mServices->patient()->autoSave();

        ToolRep3DPtr activeRep3D = this->getToolRepIn3DView();
        if (activeRep3D)
        {
                activeRep3D->getTracer()->stop();
        }
        mRecordingTool.reset();

        emit acquisitionCompleted();
}

void RecordTrackingWidget::acquisitionCancelled()
{
        ToolRep3DPtr activeRep3D = this->getToolRepIn3DView();
        if (activeRep3D)
        {
                activeRep3D->getTracer()->stop();
                activeRep3D->getTracer()->clear();
        }
        mRecordingTool.reset();
}

void RecordTrackingWidget::onMergeChanged()
{
        QString mergeSession = "";
        if (mMergeWithExistingSession->getValue())
                mergeSession = mSelectRecordSession->getSessionSelector()->getValue();

        mRecordSessionWidget->setCurrentSession(mergeSession);
}

ToolRep3DPtr RecordTrackingWidget::getToolRepIn3DView()
{
        return mServices->view()->get3DReps(0, 0)->findFirst<ToolRep3D>(mRecordingTool);
}

void RecordTrackingWidget::obscuredSlot(bool obscured)
{
        if (obscured)
                mAcquisitionService->cancelRecord();

        if (!obscured)
        {
                connect(mAcquisitionService.get(), &AcquisitionService::started, this, &RecordTrackingWidget::acquisitionStarted);
                connect(mAcquisitionService.get(), &AcquisitionService::acquisitionStopped, this, &RecordTrackingWidget::acquisitionStopped, Qt::QueuedConnection);
                connect(mAcquisitionService.get(), &AcquisitionService::cancelled, this, &RecordTrackingWidget::acquisitionCancelled);
                connect(mToolSelector.get(), &StringPropertySelectTool::changed, this, &RecordTrackingWidget::onToolChanged);
                connect(mServices->tracking().get(), &TrackingService::activeToolChanged, this, &RecordTrackingWidget::onToolChanged);
        }
        else
        {
                disconnect(mAcquisitionService.get(), &AcquisitionService::started, this, &RecordTrackingWidget::acquisitionStarted);
                disconnect(mAcquisitionService.get(), &AcquisitionService::acquisitionStopped, this, &RecordTrackingWidget::acquisitionStopped);
                disconnect(mAcquisitionService.get(), &AcquisitionService::cancelled, this, &RecordTrackingWidget::acquisitionCancelled);
                disconnect(mToolSelector.get(), &StringPropertySelectTool::changed, this, &RecordTrackingWidget::onToolChanged);
                disconnect(mServices->tracking().get(), &TrackingService::activeToolChanged, this, &RecordTrackingWidget::onToolChanged);
        }

        mSelectRecordSession->setVisible(!obscured);
}

void RecordTrackingWidget::onToolChanged()
{
        mSelectRecordSession->setTool(mToolSelector->getTool());
}


ToolPtr RecordTrackingWidget::getSuitableRecordingTool()
{
        ToolPtr retval = mToolSelector->getTool();
        if(!retval)
                retval = mServices->tracking()->getActiveTool();
        return retval;
}

TimedTransformMap RecordTrackingWidget::getRecordedTrackerData_prMt()
{
        return mSelectRecordSession->getRecordedTrackerData_prMt();
}

} //namespace cx