imstkVTKPointSetRenderDelegate.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 vtkDoubleArray;
class vtkPolyData;

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

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

    void processEvents() override;

// Callbacks for modifications, when an element changes the user or API must post the modified event
// to inform that this happened, if the actual buffer on the geometry is swapped then geometry
// modified would instead be called
protected:
    void init() override;

    void vertexDataModified(Event* e);
    void vertexScalarsModified(Event* e);

    void geometryModified(Event* e);

    void setVertexBuffer(std::shared_ptr<VecDataArray<double, 3>> vertices);
    void setVertexScalarBuffer(std::shared_ptr<AbstractDataArray> scalars);

    std::shared_ptr<PointSet> m_geometry;
    std::shared_ptr<VecDataArray<double, 3>> m_vertices;
    std::shared_ptr<AbstractDataArray>       m_vertexScalars;

    vtkSmartPointer<vtkPolyData> m_polydata;

    vtkSmartPointer<vtkDoubleArray> m_mappedVertexArray;       
    vtkSmartPointer<vtkDataArray>   m_mappedVertexScalarArray; 
};
} // namespace imstk