mitkRegistrationWrapperMapperBase.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/


#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 ==========
    virtual vtkProp *GetVtkProp(mitk::BaseRenderer *renderer);
    static void SetDefaultProperties(DataNode* node, BaseRenderer* renderer = NULL, bool overwrite = false );
    static void SetVtkMapperImmediateModeRendering(vtkMapper *mapper);
    virtual void GenerateDataForRenderer(mitk::BaseRenderer* renderer);
    //=========================================================

    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()
        {
        }
    };

    mitk::LocalStorageHandler<RegWrapperLocalStorage> m_LSH;


protected:

    MITKRegistrationWrapperMapperBase();
    virtual ~MITKRegistrationWrapperMapperBase();


private:

};

} // end namespace mitk




#endif /* MITKRegistrationWrapperMapperBase_H_HEADER_INCLUDED */