mitkRegistrationWrapperMapperBase.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 _MITK_MITK_REGISTRATION_WRAPPER_MAPPER_BASE_H
#define _MITK_MITK_REGISTRATION_WRAPPER_MAPPER_BASE_H


#include <vtkSmartPointer.h>

#include <mitkVtkMapper.h>
#include <mitkLocalStorageHandler.h>

#include "MitkMatchPointRegistrationExports.h"

class vtkPropAssembly;
class vtkPolyDataMapper;
class vtkPolyData;
class vtkColorTransferFunction;
class vtkActor;

namespace mitk {

class MITKRegistrationWrapperMapperBase : public VtkMapper
{
public:

    mitkClassMacro(MITKRegistrationWrapperMapperBase, VtkMapper);

    //========== essential implementation for mapper ==========
    vtkProp *GetVtkProp(mitk::BaseRenderer *renderer) override;
    static void SetDefaultProperties(DataNode* node, BaseRenderer* renderer = nullptr, bool overwrite = false );
    void GenerateDataForRenderer(mitk::BaseRenderer* renderer) override;
    //=========================================================

    virtual bool GetGeometryDescription(mitk::BaseRenderer *renderer, mitk::BaseGeometry::ConstPointer& gridDesc, unsigned int& gridFrequ) const = 0;
    virtual bool RendererGeometryIsOutdated(mitk::BaseRenderer *renderer, const itk::TimeStamp& time) const = 0;

    class MITKMATCHPOINTREGISTRATION_EXPORT RegWrapperLocalStorage : public mitk::Mapper::BaseLocalStorage
    {
    public:
      vtkSmartPointer<vtkPolyData> m_DeformedGridData;
      vtkSmartPointer<vtkPolyData> m_StartGridData;

      vtkSmartPointer<vtkActor> m_DeformedGridActor;
      vtkSmartPointer<vtkPolyDataMapper> m_DeformedGridMapper;
      vtkSmartPointer<vtkActor> m_StartGridActor;
      vtkSmartPointer<vtkPolyDataMapper> m_StartGridMapper;

      vtkSmartPointer<vtkPropAssembly> m_RegAssembly;

      vtkSmartPointer<vtkColorTransferFunction>  m_LUT;

        itk::TimeStamp m_LastUpdateTime;
        RegWrapperLocalStorage();

        ~RegWrapperLocalStorage() override
        {
        }
    };

    mitk::LocalStorageHandler<RegWrapperLocalStorage> m_LSH;


protected:

    MITKRegistrationWrapperMapperBase();
    ~MITKRegistrationWrapperMapperBase() override;


private:

};

} // end namespace mitk




#endif /* MITKRegistrationWrapperMapperBase_H_HEADER_INCLUDED */