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

#include <boost/bind.hpp>

#include <QDir>
#include <QStackedWidget>
#include <QPushButton>
#include <QFileDialog>

#include "vtkImageData.h"

#include "cxTime.h"
#include "cxLogger.h"
#include "cxProbeSector.h"
#include "cxRegistrationTransform.h"
#include "cxStringProperty.h"
#include "cxHelperWidgets.h"
#include "cxDataInterface.h"
#include "cxTrackingService.h"
#include "cxOptionsWidget.h"
#include "cxVideoService.h"
#include "cxPatientModelService.h"
#include "cxDetailedLabeledComboBoxWidget.h"
#include "cxStreamerService.h"
#include "cxVideoSource.h"
#include "cxViewService.h"
#include "cxImage.h"
#include "cxProfile.h"
#include "cxHelperWidgets.h"
#include "cxStringPropertyActiveVideoSource.h"

namespace cx
{

VideoConnectionWidget::VideoConnectionWidget(VisServicesPtr services, QWidget* parent) :
        BaseWidget(parent, "igt_link_widget", "Video Connection"),
        mServices(services),
        mUpdateConnectionInService(true)
{
        this->setToolTip("Connect to a video source");
        mOptions = profile()->getXmlSettings().descend("video");

        QString defaultConnection = mServices->video()->getConnectionMethod();
        mConnectionSelector = StringProperty::initialize("Connection", "", "Method for connecting to Video Server", defaultConnection, QStringList(), mOptions.getElement("video"));
        connect(mConnectionSelector.get(), SIGNAL(changed()), this, SLOT(setConnectionMethodAndUpdateGuiForConnectionMethodSlot()));

        connect(mServices->video().get(), &VideoService::connected, this, &VideoConnectionWidget::serverStatusChangedSlot);

        mStackedWidget = new QStackedWidget(this);
        mStackedWidgetFrame = this->wrapStackedWidgetInAFrame();
        mConnectButton = this->initializeConnectButton();
        mImportStreamImageButton = this->initializeImportStreamImageButton();
        mActiveVideoSourceSelector = this->initializeActiveVideoSourceSelector();
        mConnectionSelectionWidget = new DetailedLabeledComboBoxWidget(this, mConnectionSelector);

        mToptopLayout = new QVBoxLayout(this);
        mToptopLayout->addWidget(mConnectionSelectionWidget);
        mToptopLayout->addWidget(mStackedWidgetFrame);
        mToptopLayout->addWidget(mConnectButton);
        mToptopLayout->addWidget(mImportStreamImageButton);
        mToptopLayout->addWidget(sscCreateDataWidget(this, mActiveVideoSourceSelector));
        mToptopLayout->addStretch();

        connect(mServices->video().get(), SIGNAL(StreamerServiceAdded(StreamerService*)), this, SLOT(onServiceAdded(StreamerService*)));
        connect(mServices->video().get(), SIGNAL(StreamerServiceRemoved(StreamerService*)), this, SLOT(onServiceRemoved(StreamerService*)));
        connect(mServices->video().get(), &VideoService::connectionMethodChanged, this, &VideoConnectionWidget::connectionMethodChangedSlot);

        this->addExistingStreamerServices(); //Need to add StreamerServices already existing at this point, since we will only get signals when new Services are added

        this->setConnectionMethodAndUpdateGuiForConnectionMethodSlot();
}

VideoConnectionWidget::~VideoConnectionWidget()
{
        if (mServices->video())
        {
                disconnect(mServices->video().get(), SIGNAL(StreamerServiceAdded(StreamerService*)), this, SLOT(onServiceAdded(StreamerService*)));
                disconnect(mServices->video().get(), SIGNAL(StreamerServiceRemoved(StreamerService*)), this, SLOT(onServiceRemoved(StreamerService*)));
                disconnect(mServices->video().get(), &VideoService::connectionMethodChanged, this, &VideoConnectionWidget::connectionMethodChangedSlot);
        }
}

void VideoConnectionWidget::addExistingStreamerServices()
{
        QList<StreamerServicePtr> services = mServices->video()->getStreamerServices();
    foreach(StreamerServicePtr service, services)
        {
        this->onServiceAdded(service.get());
        }
}

void VideoConnectionWidget::onServiceAdded(StreamerService* service)
{
        QWidget* widget = this->createStreamerWidget(service);
        QWidget* serviceWidget = this->wrapVerticalStretch(widget);
        mStackedWidget->addWidget(serviceWidget);
        mStreamerServiceWidgets[service->getType()] = serviceWidget;

        this->addServiceToSelector(service);
}

QWidget* VideoConnectionWidget::createStreamerWidget(StreamerService* service)
{
//      QString serviceName = service->getName();
        QDomElement element = mOptions.getElement("video");
        std::vector<PropertyPtr> adapters = service->getSettings(element);

        OptionsWidget* widget = new OptionsWidget(mServices->view(), mServices->patient(), this);
        widget->setOptions(service->getType(), adapters, false);
        widget->setObjectName(service->getType());
        widget->setFocusPolicy(Qt::StrongFocus); // needed for help system: focus is used to display help text

        connect(mConnectionSelectionWidget, SIGNAL(detailsTriggered()), widget, SLOT(toggleAdvanced()));

        return widget;
}

void VideoConnectionWidget::onServiceRemoved(StreamerService *service)
{
        this->removeServiceFromSelector(service);
        this->removeServiceWidget(service->getType());
}

void VideoConnectionWidget::addServiceToSelector(StreamerService *service)
{
        QStringList range = mConnectionSelector->getValueRange();
        std::map<QString, QString> display = mConnectionSelector->getDisplayNames();

        range.append(service->getType());
        range.removeDuplicates();
        display[service->getType()] = service->getName();

        mConnectionSelector->setValueRange(range);
        mConnectionSelector->setDisplayNames(display);
}

void VideoConnectionWidget::removeServiceFromSelector(StreamerService *service)
{
        QStringList range = mConnectionSelector->getValueRange();

        int index = range.indexOf(service->getType());
        if(mConnectionSelector->getValue() == service->getType())
                mConnectionSelector->setValue(range[0]);
        range.removeAt(index);
        mConnectionSelector->setValueRange(range);
}

void VideoConnectionWidget::removeServiceWidget(QString name)
{
        QWidget* serviceWidget = mStreamerServiceWidgets[name];
        mStackedWidget->removeWidget(serviceWidget);
        delete serviceWidget;
        mStreamerServiceWidgets.erase(name);
}

StringPropertyActiveVideoSourcePtr VideoConnectionWidget::initializeActiveVideoSourceSelector()
{
        return StringPropertyActiveVideoSource::create(mServices->video());
}

QFrame* VideoConnectionWidget::wrapStackedWidgetInAFrame()
{
        QFrame* frame = new QFrame(this);
        frame->setFrameStyle(QFrame::StyledPanel | QFrame::Raised);
        frame->setSizePolicy(frame->sizePolicy().horizontalPolicy(), QSizePolicy::Fixed);
        QVBoxLayout* frameLayout = new QVBoxLayout(frame);
        frameLayout->addWidget(mStackedWidget);
        frame->setFocusPolicy(Qt::StrongFocus); // needed for help system: focus is used to display help text

        return frame;
}

QWidget* VideoConnectionWidget::wrapVerticalStretch(QWidget* input)
{
        QWidget* retval = new QWidget(this);
        QVBoxLayout* layout = new QVBoxLayout(retval);
        layout->addWidget(input);
        retval->setObjectName(input->objectName()); // help propagation
        layout->addStretch();
        layout->setMargin(0);
        layout->setSpacing(0);
        return retval;
}

void VideoConnectionWidget::setConnectionMethodAndUpdateGuiForConnectionMethodSlot()
{
        QString name = mConnectionSelector->getValue();
        //Need to set connection method in VideoConnectionManager before calling
        //useDirectLink(), useLocalServer() and useRemoteServer()
        if(mUpdateConnectionInService)
                mServices->video()->setConnectionMethod(name);
        mUpdateConnectionInService = true;

        QWidget* serviceWidget = mStreamerServiceWidgets[name];
        if(serviceWidget)
        {
                mStackedWidget->setCurrentWidget(serviceWidget);
                mStackedWidgetFrame->setObjectName(serviceWidget->objectName()); // for improved help
        }
}

void VideoConnectionWidget::connectionMethodChangedSlot()
{
        QString changedConnectionMethod = mServices->video()->getConnectionMethod();

        this->mUpdateConnectionInService = false;//Prevent infinite signal loop
        mConnectionSelector->setValue(changedConnectionMethod);
}

void VideoConnectionWidget::toggleConnectServer()
{
        if (mServices->video()->isConnected())
                mServices->video()->closeConnection();
        else
                mServices->video()->openConnection();
}

QPushButton* VideoConnectionWidget::initializeConnectButton()
{
        QPushButton* connectButton = new QPushButton("Connect", this);
        connectButton->setToolTip("Connect/disconnect to the video server using the seleted method");
        connect(connectButton, SIGNAL(clicked()), this, SLOT(toggleConnectServer()));
        return connectButton;
}

QPushButton* VideoConnectionWidget::initializeImportStreamImageButton()
{
        QPushButton* importstreamimagebutton = new QPushButton("Import image from stream", this);
        importstreamimagebutton->setToolTip("Import a single image/volume from the real time stream");
        importstreamimagebutton->setDisabled(true);
        connect(importstreamimagebutton, SIGNAL(clicked()), this, SLOT(importStreamImageSlot()));

        return importstreamimagebutton;
}

void VideoConnectionWidget::serverStatusChangedSlot()
{
        mImportStreamImageButton->setEnabled(mServices->video()->isConnected());
        if (mServices->video()->isConnected())
                mConnectButton->setText("Disconnect Server");
        else
                mConnectButton->setText("Connect Server");

        this->adjustSize();
}

void VideoConnectionWidget::importStreamImageSlot()
{
        if (!mServices->video()->isConnected())
        {
                reportWarning("Video is not connected");
                return;
        }
        Transform3D rMd = Transform3D::Identity();
        ToolPtr probe = mServices->tracking()->getFirstProbe();
        VideoSourcePtr videoSource;
        if (probe)
        {
                videoSource = probe->getProbe()->getRTSource();
                rMd = calculate_rMd_ForAProbeImage(probe);
        }
        else
                videoSource = mServices->video()->getActiveVideoSource();

        if (!videoSource)
        {
                reportWarning("No Video data source");
                return;
        }
        if (!videoSource->validData())
        {
                reportWarning("No valid video data");
                return;
        }

        vtkImageDataPtr input;
        input = videoSource->getVtkImageData();
        if (!input)
        {
                reportWarning("No Video data");
                return;
        }
        QString filename = generateFilename(videoSource);

        this->saveAndImportSnapshot(input, filename, rMd);

}

Transform3D VideoConnectionWidget::calculate_rMd_ForAProbeImage(ToolPtr probe)
{
        Transform3D rMd = Transform3D::Identity();
        Transform3D rMpr = mServices->patient()->get_rMpr();
        Transform3D prMt = probe->get_prMt();
        Transform3D tMu = probe->getProbe()->getSector()->get_tMu();
        Transform3D uMv = probe->getProbe()->getSector()->get_uMv();
        rMd = rMpr * prMt * tMu * uMv;
        return rMd;
}

QString VideoConnectionWidget::generateFilename(VideoSourcePtr videoSource)
{
        vtkImageDataPtr input = videoSource->getVtkImageData();
        int* extent = input->GetExtent();
        QString filename;
        QString format = timestampSecondsFormat();
        if (extent[5] - extent[4] > 0)
                filename = "3DRTSnapshot_";
        else
                filename = "2DRTSnapshot_";

        filename += videoSource->getName() + QDateTime::currentDateTime().toString(format);
        return filename;
}

void VideoConnectionWidget::saveAndImportSnapshot(vtkImageDataPtr input, QString filename, Transform3D rMd)
{
        vtkImageDataPtr copiedImage = vtkImageDataPtr::New();
        copiedImage->DeepCopy(input);

        ImagePtr output = mServices->patient()->createSpecificData<Image>(filename);
        output->setVtkImageData(copiedImage);
        output->get_rMd_History()->setRegistration(rMd);
        mServices->patient()->insertData(output);

        mServices->view()->autoShowData(output);
        report(QString("Saved snapshot %1 from active video source").arg(output->getName()));
}

} //end namespace cx