mitkTubeGraphVtkMapper3D.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 mitkTubeGraphVtkMapper3D_h
#define mitkTubeGraphVtkMapper3D_h
 
#include <MitkTubeGraphExports.h>
#include <mitkLocalStorageHandler.h>
 
#include "mitkCircularProfileTubeElement.h"
#include "mitkTubeGraph.h"
#include "mitkTubeGraphProperty.h"
#include "mitkVtkMapper.h"
 
#include <vtkActor.h>
#include <vtkAppendPolyData.h>
#include <vtkAssembly.h>
#include <vtkPolyData.h>
#include <vtkSmartPointer.h>
 
namespace mitk
{
  class MITKTUBEGRAPH_EXPORT TubeGraphVtkMapper3D : public VtkMapper
  {
  public:
    /* Typedefs */
    typedef TubeGraph::EdgeDescriptorType EdgeDescriptorType;
    typedef TubeGraph::VertexDescriptorType VertexDescriptorType;
 
    mitkClassMacro(TubeGraphVtkMapper3D, VtkMapper);
    itkNewMacro(Self);
    virtual const TubeGraph *GetInput();
    vtkProp *GetVtkProp(mitk::BaseRenderer *renderer) override;
 
  protected:
    TubeGraphVtkMapper3D();
    ~TubeGraphVtkMapper3D() override;
 
    void GenerateDataForRenderer(mitk::BaseRenderer *renderer) override;
 
    virtual void GenerateTubeGraphData(mitk::BaseRenderer *renderer);
 
    virtual void RenderTubeGraphPropertyInformation(mitk::BaseRenderer *renderer);
 
    void GeneratePolyDataForTube(TubeGraphEdge &edge,
                                 const TubeGraph::Pointer &graph,
                                 const TubeGraphProperty::Pointer &graphProperty,
                                 mitk::BaseRenderer *renderer);
    void GeneratePolyDataForFurcation(TubeGraphVertex &vertex,
                                      const TubeGraph::Pointer &graph,
                                      mitk::BaseRenderer *renderer);
    void ClipPolyData(TubeGraphVertex &vertex,
                      const TubeGraph::Pointer &graph,
                      const TubeGraphProperty::Pointer &graphProperty,
                      mitk::BaseRenderer *renderer);
 
  private:
    bool ClipStructures();
 
    class LocalStorage : public mitk::Mapper::BaseLocalStorage
    {
    public:
      vtkSmartPointer<vtkAssembly> m_vtkTubeGraphAssembly;
      std::map<TubeGraph::TubeDescriptorType, vtkSmartPointer<vtkActor>> m_vtkTubesActorMap;
      std::map<TubeGraph::VertexDescriptorType, vtkSmartPointer<vtkActor>> m_vtkSpheresActorMap;
 
      itk::TimeStamp m_lastGenerateDataTime;
      itk::TimeStamp m_lastRenderDataTime;
 
      LocalStorage() { m_vtkTubeGraphAssembly = vtkSmartPointer<vtkAssembly>::New(); }
      ~LocalStorage() override {}
    };
 
    LocalStorageHandler<LocalStorage> m_LSH;
  };
} // namespace
 
#endif