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

#include <vtkPiecewiseFunction.h>
#include <vtkColorTransferFunction.h>
#include <vtkVolumeProperty.h>

#include "cxImage.h"
#include "cxImageTF3D.h"


namespace cx
{

VolumeProperty::VolumeProperty() : QObject(),
        mOpacityTransferFunction(vtkPiecewiseFunctionPtr::New()),
        mColorTransferFunction(vtkColorTransferFunctionPtr::New()),
        mVolumeProperty(vtkVolumePropertyPtr::New())
{
        double maxVal = 255;
        mOpacityTransferFunction->AddPoint(0.0, 0.0);
        mOpacityTransferFunction->AddPoint(maxVal, 1.0);

        mColorTransferFunction->SetColorSpaceToRGB();
        mColorTransferFunction->AddRGBPoint(0.0, 0.0, 0.0, 0.0);
        mColorTransferFunction->AddRGBPoint(maxVal, 1.0, 1.0, 1.0);

        mVolumeProperty->SetColor(mColorTransferFunction);
        mVolumeProperty->SetScalarOpacity(mOpacityTransferFunction);
        mVolumeProperty->SetInterpolationTypeToLinear();

        // from snw
        mVolumeProperty->ShadeOff();
        mVolumeProperty->SetAmbient ( 0.2 );
        mVolumeProperty->SetDiffuse ( 0.9 );
        mVolumeProperty->SetSpecular ( 0.3 );
        mVolumeProperty->SetSpecularPower ( 15.0 );
        mVolumeProperty->SetScalarOpacityUnitDistance(0.8919);
}

vtkVolumePropertyPtr VolumeProperty::getVolumeProperty()
{
        return mVolumeProperty;
}

void VolumeProperty::setImage(ImagePtr image)
{
        if (image==mImage)
        {
                return;
        }

        if (mImage)
        {
                disconnect(mImage.get(), SIGNAL(vtkImageDataChanged()), this, SLOT(transferFunctionsChangedSlot()));
                disconnect(mImage.get(), SIGNAL(transferFunctionsChanged()), this, SLOT(transferFunctionsChangedSlot()));
        }

        mImage = image;

        if (mImage)
        {
                connect(mImage.get(), SIGNAL(vtkImageDataChanged()), this, SLOT(transferFunctionsChangedSlot()));
                connect(mImage.get(), SIGNAL(transferFunctionsChanged()), this, SLOT(transferFunctionsChangedSlot()));
        }

        this->transferFunctionsChangedSlot();
}

void VolumeProperty::transferFunctionsChangedSlot()
{
        if (!mImage)
                return;

        mVolumeProperty->SetColor(mImage->getTransferFunctions3D()->getColorTF());
        mVolumeProperty->SetScalarOpacity(mImage->getTransferFunctions3D()->getOpacityTF());
        mVolumeProperty->SetShade(mImage->getShadingOn());

        mVolumeProperty->SetAmbient(mImage->getShadingAmbient());
        mVolumeProperty->SetDiffuse(mImage->getShadingDiffuse());
        mVolumeProperty->SetSpecular(mImage->getShadingSpecular());
        mVolumeProperty->SetSpecularPower(mImage->getShadingSpecularPower());

        mVolumeProperty->SetInterpolationType(mImage->getInterpolationType());
}

} // namespace cx