imstkVTKRenderDelegate.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkEventObject.h"
#include "imstkMacros.h"
#include "imstkTextureManager.h"
#include "imstkVTKTextureDelegate.h"

#include <vtkSmartPointer.h>

class vtkAbstractMapper;
class vtkProp;
class vtkTransform;
class vtkTexture;

namespace imstk
{
class Texture;
class RenderMaterial;
class VisualModel;

class VTKRenderDelegate : public EventObject, public std::enable_shared_from_this<VTKRenderDelegate>
{
public:
    ~VTKRenderDelegate() override = default;

    static std::shared_ptr<VTKRenderDelegate> makeDelegate(std::shared_ptr<VisualModel> visualModel);

    void initialize(std::shared_ptr<VisualModel> model);

    std::shared_ptr<VisualModel> getVisualModel() const { return m_visualModel; }

    vtkSmartPointer<vtkProp> getVtkActor() const { return m_actor; }

    void setTextureManager(std::weak_ptr<TextureManager<VTKTextureDelegate>> textureManager) { m_textureManager = textureManager; }

    void update();

    virtual void processEvents();

    virtual void updateRenderProperties() = 0;

protected:
    VTKRenderDelegate();

    virtual void init() = 0;

    void materialModified(Event* imstkNotUsed(e)) { updateRenderProperties(); }

    void visualModelModified(Event* e);

    vtkSmartPointer<vtkTexture> getVTKTexture(std::shared_ptr<Texture> texture);

    vtkSmartPointer<vtkTransform> m_transform;

    // VTK data members used to create the rendering pipeline
    vtkSmartPointer<vtkProp> m_actor;
    vtkSmartPointer<vtkAbstractMapper> m_mapper;

    std::shared_ptr<VisualModel>    m_visualModel; 
    std::shared_ptr<RenderMaterial> m_material;

    std::weak_ptr<TextureManager<VTKTextureDelegate>> m_textureManager;
};
} // namespace imstk