mitkMultiLabelSegmentationVtkMapper3D.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkMultiLabelSegmentationVtkMapper3D_h
#define mitkMultiLabelSegmentationVtkMapper3D_h
 
// MITK
#include "MitkMultilabelExports.h"
#include "mitkCommon.h"
 
// MITK Rendering
#include "mitkBaseRenderer.h"
#include "mitkExtractSliceFilter.h"
#include "mitkLabelSetImage.h"
#include "mitkVtkMapper.h"
 
// VTK
#include <vtkSmartPointer.h>
 
class vtkPolyDataMapper;
class vtkImageData;
class vtkLookupTable;
class vtkVolumeProperty;
class vtkVolume;
class vtkSmartVolumeMapper;
class vtkColorTransferFunction;
class vtkPiecewiseFunction;
 
namespace mitk
{
  class MultiLabelSegmentationGroupMapping;
  class IPreferences;
 
  class MITKMULTILABEL_EXPORT MultiLabelSegmentationVtkMapper3D : public VtkMapper
  {
  public:
    mitkClassMacro(MultiLabelSegmentationVtkMapper3D, VtkMapper);
 
    itkNewMacro(Self);
 
    void Update(mitk::BaseRenderer *renderer) override;
 
    //### methods of MITK-VTK rendering pipeline
    vtkProp *GetVtkProp(mitk::BaseRenderer *renderer) override;
    //### end of methods of MITK-VTK rendering pipeline
 
    class MITKMULTILABEL_EXPORT LocalStorage : public mitk::Mapper::BaseLocalStorage
    {
    public:
      vtkSmartPointer<vtkPropAssembly> m_Actors;
 
      std::map<const Image*, std::unique_ptr<MultiLabelSegmentationGroupMapping>> m_GroupPipelines;
 
      vtkSmartPointer<vtkColorTransferFunction> m_TransferFunction;
      vtkSmartPointer<vtkPiecewiseFunction> m_OpacityTransferFunction;
      vtkSmartPointer<vtkPiecewiseFunction> m_FadedOpacityTransferFunction;
 
      bool m_UseFadedPipeline;
 
      itk::TimeStamp m_LastDataUpdateTime;
      itk::TimeStamp m_LastPropertyUpdateTime;
 
      mitk::TimeStepType m_LastUpdateTimeStep;
 
      mitk::LookupTable::Pointer m_LabelLookupTable;
 
      std::optional<bool> m_GPUCheckSuccessfull;
      bool m_3DRenderingPreference;
 
      IPreferences* m_SegPreferences;
 
      LocalStorage();
      ~LocalStorage() override;
    };
 
    LocalStorage* GetLocalStorage(mitk::BaseRenderer* renderer);
 
    static void SetDefaultProperties(mitk::DataNode* node, mitk::BaseRenderer* renderer = nullptr, bool overwrite = false);
 
  protected:
    MultiLabelSegmentationVtkMapper3D();
    ~MultiLabelSegmentationVtkMapper3D() override;
 
    void GenerateDataForRenderer(mitk::BaseRenderer *renderer) override;
 
    void UpdateLookupTable(LocalStorage* localStorage);
 
    using OutdatedGroupVectorType = std::vector<std::pair<mitk::MultiLabelSegmentation::GroupIndexType, const mitk::Image*>>;
    OutdatedGroupVectorType CheckForOutdatedGroups(mitk::MultiLabelSegmentationVtkMapper3D::LocalStorage* ls,
      mitk::MultiLabelSegmentation* seg, bool fadedPipelineChanged);
 
    void UpdateVolumeMapping(LocalStorage* localStorage, const OutdatedGroupVectorType& outdatedData);
 
    mitk::LocalStorageHandler<LocalStorage> m_LSH;
  };
 
} // namespace mitk
 
#endif