SVolume.hpp

/* ***** BEGIN LICENSE BLOCK *****
 * FW4SPL - Copyright (C) IRCAD, 2009-2018.
 * Distributed under the terms of the GNU Lesser General Public License (LGPL) as
 * published by the Free Software Foundation.
 * ****** END LICENSE BLOCK ****** */

#pragma once

#include "visuVTKVRAdaptor/config.hpp"

#include <fwData/Image.hpp>

#include <fwDataTools/helper/MedicalImageAdaptor.hpp>

#include <fwRenderVTK/IAdaptor.hpp>
#include <fwRenderVTK/SRender.hpp>

class vtkSmartVolumeMapper;
class vtkBoxWidget2;
class vtkColorTransferFunction;
class vtkCommand;
class vtkPiecewiseFunction;
class vtkPlaneCollection;
class vtkVolume;
class vtkVolumeProperty;

namespace visuVTKVRAdaptor
{

class VISUVTKVRADAPTOR_CLASS_API SVolume : public ::fwDataTools::helper::MedicalImageAdaptor,
                                           public ::fwRenderVTK::IAdaptor
{

public:

    fwCoreServiceClassDefinitionsMacro( (SVolume)(::fwRenderVTK::IAdaptor) );

    VISUVTKVRADAPTOR_API SVolume() noexcept;

    VISUVTKVRADAPTOR_API virtual ~SVolume() noexcept;

    static const ::fwServices::IService::KeyType s_IMAGE_INOUT;
    static const ::fwServices::IService::KeyType s_TF_INOUT;

    VISUVTKVRADAPTOR_API void setClippingPlanesId( ::fwRenderVTK::SRender::VtkObjectIdType id );

    VISUVTKVRADAPTOR_API void setVtkClippingPlanes( vtkPlaneCollection* planes );

    VISUVTKVRADAPTOR_API virtual KeyConnectionsMap getAutoConnections() const override;

    VISUVTKVRADAPTOR_API void crop();

    VISUVTKVRADAPTOR_API void updateTransform();

    VISUVTKVRADAPTOR_API void updateCropBoxTransform();

protected:

    VISUVTKVRADAPTOR_API void configuring() override;
    VISUVTKVRADAPTOR_API void starting() override;
    VISUVTKVRADAPTOR_API void stopping() override;
    VISUVTKVRADAPTOR_API void updating() override;
    VISUVTKVRADAPTOR_API void swapping(const KeyType& key) override;

    VISUVTKVRADAPTOR_API virtual void updateTFPoints() override;

    VISUVTKVRADAPTOR_API virtual void updateTFWindowing(double window, double level) override;

    void resetBoxWidget();

    void activateBoxClipping( bool activate );

    void show( bool isVisible );

    void buildPipeline();

    void updateImage( ::fwData::Image::sptr image  );

    void updateVolumeTransferFunction( ::fwData::Image::sptr image );

    vtkPlaneCollection* m_clippingPlanes;

    ::fwRenderVTK::SRender::VtkObjectIdType m_clippingPlanesId;

    vtkSmartVolumeMapper* m_volumeMapper;
    vtkVolumeProperty* m_volumeProperty;
    vtkVolume* m_volume;

    vtkPiecewiseFunction* m_opacityTransferFunction;
    vtkColorTransferFunction* m_colorTransferFunction;

    vtkCommand* m_abortCommand;

    vtkBoxWidget2* m_boxWidget;
    vtkCommand* m_croppingCommand;
    vtkCommand* m_transformCommand;

    bool m_croppingBoxDefaultState;

    std::string m_cropBoxTransformID;

    std::string m_blendMode;

    vtkTransform* m_cropBoxTransform;

private:

    bool m_autoResetCamera;

    double m_reductionFactor;
};

} //namespace visuVTKVRAdaptor