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

#include <sstream>
#include <vtkMath.h>
#include <vtkImageData.h>
#include <vtkDoubleArray.h>
#include <vtkVectorText.h>
#include <vtkFollower.h>
#include <vtkSphereSource.h>
#include <vtkPolyDataMapper.h>
#include <vtkProperty.h>
#include <vtkRenderer.h>
#include <vtkRenderWindow.h>
#include "cxView.h"
#include "cxLandmark.h"

#include "cxPatientModelService.h"
#include "cxTypeConversions.h"
#include "boost/bind.hpp"
#include "cxVtkHelperClasses.h"

namespace cx
{

PatientLandmarksSource::PatientLandmarksSource(PatientModelServicePtr dataManager) : mDataManager(dataManager)
{
        connect(mDataManager->getPatientLandmarks().get(), SIGNAL(landmarkAdded(QString)), this, SIGNAL(changed()));
        connect(mDataManager->getPatientLandmarks().get(), SIGNAL(landmarkRemoved(QString)), this, SIGNAL(changed()));
        connect(mDataManager.get(), SIGNAL(rMprChanged()), this, SIGNAL(changed()));
}
LandmarkMap PatientLandmarksSource::getLandmarks() const
{
        return mDataManager->getPatientLandmarks()->getLandmarks();
}
Transform3D PatientLandmarksSource::get_rMl() const
{
        return mDataManager->get_rMpr();
}
// --------------------------------------------------------
Vector3D PatientLandmarksSource::getTextPos(Vector3D p_l) const
{
        return p_l;
}
// --------------------------------------------------------

// --------------------------------------------------------
// --------------------------------------------------------
// --------------------------------------------------------

ImageLandmarksSource::ImageLandmarksSource()
{
}

DataPtr ImageLandmarksSource::getData()
{
        return mData;
}

void ImageLandmarksSource::setData(DataPtr image)
{
        if (image == mData)
                return;

        if (mData)
        {
                disconnect(mData->getLandmarks().get(), SIGNAL(landmarkAdded(QString)), this, SIGNAL(changed()));
                disconnect(mData->getLandmarks().get(), SIGNAL(landmarkRemoved(QString)), this, SIGNAL(changed()));
                disconnect(mData.get(), SIGNAL(transformChanged()), this, SIGNAL(changed()));
        }

        mData = image;

        if (mData)
        {
                connect(mData->getLandmarks().get(), SIGNAL(landmarkAdded(QString)), this, SIGNAL(changed()));
                connect(mData->getLandmarks().get(), SIGNAL(landmarkRemoved(QString)), this, SIGNAL(changed()));
                connect(mData.get(), SIGNAL(transformChanged()), this, SIGNAL(changed()));
        }

        emit changed();
}

LandmarkMap ImageLandmarksSource::getLandmarks() const
{
        if (!mData)
                return LandmarkMap();
        return mData->getLandmarks()->getLandmarks();
}

Transform3D ImageLandmarksSource::get_rMl() const
{
        if (!mData)
                return Transform3D::Identity();
        return mData->get_rMd();
}

Vector3D ImageLandmarksSource::getTextPos(Vector3D p_l) const
{
        Vector3D imageCenter = mData->boundingBox().center();
        Vector3D centerToSkinVector = (p_l - imageCenter).normal();
        Vector3D numberPosition = p_l + 10.0 * centerToSkinVector;
        return numberPosition;
}

// --------------------------------------------------------
// --------------------------------------------------------
// --------------------------------------------------------

LandmarkRepPtr LandmarkRep::New(PatientModelServicePtr dataManager, const QString& uid)
{
        return wrap_new(new LandmarkRep(dataManager), uid);
}

LandmarkRep::LandmarkRep(PatientModelServicePtr dataManager) :
        RepImpl(),
        mDataManager(dataManager),
        mInactiveColor(QColor::fromRgbF(0.5,0.5,0.5)),
        mColor(QColor(Qt::green)),
                                //  mSecondaryColor(0,0.6,0.8),
        mSecondaryColor(QColor::fromRgbF(0, 0.9, 0.5)),
        mShowLandmarks(true),
        mGraphicsSize(1),
        mLabelSize(2.5)
{
        connect(mDataManager.get(), SIGNAL(landmarkPropertiesChanged()), this, SLOT(internalUpdate()));

        mViewportListener.reset(new ViewportListener);
        mViewportListener->setCallback(boost::bind(&LandmarkRep::rescale, this));
}

LandmarkRep::~LandmarkRep()
{
}

void LandmarkRep::setPrimarySource(LandmarksSourcePtr primary)
{
//  std::cout << this <<  "  LandmarkRep::setPrimarySource " << primary.get() << std::endl;

        if (mPrimary)
                disconnect(mPrimary.get(), SIGNAL(changed()), this, SLOT(internalUpdate()));
        mPrimary = primary;
        if (mPrimary)
                connect(mPrimary.get(), SIGNAL(changed()), this, SLOT(internalUpdate()));

        this->internalUpdate();
}

void LandmarkRep::setSecondarySource(LandmarksSourcePtr secondary)
{
        if (mSecondary)
                disconnect(mSecondary.get(), SIGNAL(changed()), this, SLOT(internalUpdate()));
        mSecondary = secondary;
        if (mSecondary)
                connect(mSecondary.get(), SIGNAL(changed()), this, SLOT(internalUpdate()));

        this->internalUpdate();
}

void LandmarkRep::setColor(QColor color)
{
        mColor = color;
        this->internalUpdate();
}

void LandmarkRep::setSecondaryColor(QColor color)
{
        mSecondaryColor = color;
        this->internalUpdate();
}

void LandmarkRep::setGraphicsSize(double size)
{
        mGraphicsSize = size;
        this->internalUpdate();
}

void LandmarkRep::setLabelSize(double size)
{
        mLabelSize = size;
        this->internalUpdate();
}

void LandmarkRep::showLandmarks(bool on)
{
        if (on == mShowLandmarks)
                return;

        for (LandmarkGraphicsMapType::iterator iter = mGraphics.begin(); iter != mGraphics.end(); ++iter)
        {
                if (iter->second.mPrimaryPoint)
                        iter->second.mPrimaryPoint->getActor()->SetVisibility(on);
                if (iter->second.mSecondaryPoint)
                        iter->second.mSecondaryPoint->getActor()->SetVisibility(on);
                if (iter->second.mText)
                        iter->second.mText->getActor()->SetVisibility(on);
                if (iter->second.mLine)
                        iter->second.mLine->getActor()->SetVisibility(on);
        }
        mShowLandmarks = on;
}

void LandmarkRep::addAll()
{
//  std::cout << this << " LandmarkRep::addLandmark ADD ALL" << std::endl;

        LandmarkPropertyMap props = mDataManager->getLandmarkProperties();

        for (LandmarkPropertyMap::iterator it = props.begin(); it != props.end(); ++it)
        {
                this->addLandmark(it->first);
        }
}

void LandmarkRep::internalUpdate()
{
//  std::cout << this << " LandmarkRep::internalUpdate()" << std::endl;

        this->clearAll();
        this->addAll();
}

void LandmarkRep::clearAll()
{
//  std::cout << this << " LandmarkRep::addLandmark CLEAR ALL" << std::endl;
        mGraphics.clear();
}

void LandmarkRep::addRepActorsToViewRenderer(ViewPtr view)
{
        if (!view || !view->getRenderer())
                return;

        this->addAll();
        mViewportListener->startListen(view->getRenderer());
}

void LandmarkRep::removeRepActorsFromViewRenderer(ViewPtr view)
{
        this->clearAll();
        mViewportListener->stopListen();
}

void LandmarkRep::addLandmark(QString uid)
{
//  std::cout << this << " LandmarkRep::addLandmark init" << uid << std::endl;
        vtkRendererPtr renderer = this->getRenderer();

        LandmarkProperty property = mDataManager->getLandmarkProperties()[uid];
        if (property.getUid().isEmpty())
        {
//    std::cout << "LandmarkRep::addLandmark CLEAR" << uid << std::endl;
                mGraphics.erase(uid);
                return;
        }

        double radius = 2;
        QColor color = mColor;
        QColor secondaryColor = mSecondaryColor;

        if (!property.getActive())
        {
                radius = 1;
                color = mInactiveColor;
                secondaryColor = mInactiveColor;
        }

        LandmarkGraphics current;

        // primary point
        Landmark primary;
        Vector3D primary_r(0, 0, 0);
        if (mPrimary)
        {
//    std::cout << this << "   LandmarkRep::addLandmark found mPrimary" << uid << std::endl;
                primary = mPrimary->getLandmarks()[uid];
                if (!primary.getUid().isEmpty())
                {
//      std::cout << this << "  LandmarkRep::addLandmark" << uid << std::endl;

                        primary_r = mPrimary->get_rMl().coord(primary.getCoord());

                        current.mPrimaryPoint.reset(new GraphicalPoint3D(renderer));
                        current.mPrimaryPoint->setColor(color);
                        current.mPrimaryPoint->setRadius(radius);

                        current.mText.reset(new FollowerText3D(renderer));
                        current.mText->setText(property.getName());
                        current.mText->setSizeInNormalizedViewport(true, 0.025);
                        current.mText->setColor(color);

                        Vector3D text_r = mPrimary->get_rMl().coord(mPrimary->getTextPos(primary.getCoord()));

                        current.mPrimaryPoint->setValue(primary_r);
                        current.mText->setPosition(text_r);
                }
        }

        // secondary point
        Vector3D secondary_r(0, 0, 0);
        Landmark secondary;
        if (mSecondary)
        {
                secondary = mSecondary->getLandmarks()[uid];
                if (!secondary.getUid().isEmpty())
                {
                        secondary_r = mSecondary->get_rMl().coord(secondary.getCoord());

                        current.mSecondaryPoint.reset(new GraphicalPoint3D(renderer));
                        current.mSecondaryPoint->setColor(secondaryColor);
                        current.mSecondaryPoint->setRadius(radius);
                        current.mSecondaryPoint->setValue(secondary_r);
                }
        }

        // connecting line
        if (!secondary.getUid().isEmpty() && !secondary.getUid().isEmpty())
        {
                current.mLine.reset(new GraphicalLine3D(renderer));
                current.mLine->setColor(secondaryColor);
                current.mLine->setStipple(0x0F0F);

                current.mLine->setValue(primary_r, secondary_r);
        }

        mGraphics[uid] = current;
        this->rescale();
}

void LandmarkRep::rescale()
{
        if (!mViewportListener->isListening())
                return;

        for (LandmarkGraphicsMapType::iterator iter = mGraphics.begin(); iter != mGraphics.end(); ++iter)
        {
                GraphicalPoint3DPtr primaryPoint = iter->second.mPrimaryPoint;
                GraphicalPoint3DPtr secondaryPoint = iter->second.mSecondaryPoint;
                if (primaryPoint)
                {
                        double size = mViewportListener->getVpnZoom(primaryPoint->getValue());
                        double sphereSize = mGraphicsSize / 100 / size;
                        primaryPoint->setRadius(sphereSize);
                }
                if (secondaryPoint)
                {
                        double size = mViewportListener->getVpnZoom(secondaryPoint->getValue());
                        double sphereSize = mGraphicsSize / 100 / size;
                        secondaryPoint->setRadius(sphereSize);
                }
        }
}

} //namespace cx