imstkVTKSurfaceNormalRenderDelegate.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 "imstkVTKPolyDataRenderDelegate.h"

class vtkDataArray;
class vtkPolyData;

namespace imstk
{
class PointSet;
template<typename T, int N> class VecDataArray;

class VTKSurfaceNormalRenderDelegate : public VTKPolyDataRenderDelegate
{
public:
    VTKSurfaceNormalRenderDelegate();
    ~VTKSurfaceNormalRenderDelegate() override = default;

    void processEvents() override;

protected:
    void init() override;

    void vertexDataModified(Event* e);

    void geometryModified(Event* e);

    void updateRenderProperties() override;

    std::shared_ptr<VecDataArray<double, 3>> computeTriangleCenters(
        std::shared_ptr<VecDataArray<double, 3>> verticesPtr,
        std::shared_ptr<VecDataArray<int, 3>> indicesPtr);

    std::shared_ptr<VecDataArray<double, 3>> computeTriangleNormals(
        std::shared_ptr<VecDataArray<double, 3>> verticesPtr,
        std::shared_ptr<VecDataArray<int, 3>> indicesPtr);

    std::shared_ptr<VecDataArray<double, 3>> m_surfMeshVertices;
    std::shared_ptr<VecDataArray<int, 3>>    m_surfMeshIndices;

    std::shared_ptr<VecDataArray<double, 3>> m_triangleCenterVertices;
    std::shared_ptr<VecDataArray<double, 3>> m_triangleNormals;

    vtkSmartPointer<vtkPolyData> m_polydata;
    vtkSmartPointer<vtkPolyData> m_glyphPolyData;

    vtkSmartPointer<vtkDataArray> m_mappedVertexArray;  
    vtkSmartPointer<vtkDataArray> m_mappedNormalsArray; 
};
} // namespace imstk