imstkPointSet.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 "imstkGeometry.h"
#include "imstkTypes.h"

#include <unordered_map>

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

class PointSet : public Geometry
{
public:
    PointSet();
    ~PointSet() override = default;

    IMSTK_TYPE_NAME(PointSet)

    
    void initialize(std::shared_ptr<VecDataArray<double, 3>> positions);

    virtual void clear();

    virtual void print() const override;

    virtual void computeBoundingBox(Vec3d& lowerCorner, Vec3d& upperCorner, const double paddingPercent = 0.0) override;

// Accessors
    void setInitialVertexPositions(std::shared_ptr<VecDataArray<double, 3>> vertices);

    std::shared_ptr<VecDataArray<double, 3>> getInitialVertexPositions() const { return m_initialVertexPositions; }

    Vec3d& getInitialVertexPosition(const size_t vertNum);

    void setVertexPositions(std::shared_ptr<VecDataArray<double, 3>> positions);

    std::shared_ptr<VecDataArray<double, 3>> getVertexPositions(DataType type = DataType::PostTransform) const;

    void setVertexPosition(const size_t vertNum, const Vec3d& pos);

    const Vec3d& getVertexPosition(const size_t vertNum, DataType type = DataType::PostTransform) const;
    Vec3d& getVertexPosition(const size_t vertNum, DataType type       = DataType::PostTransform);

    int getNumVertices() const;

// Attributes
    void setVertexAttribute(const std::string& arrayName, std::shared_ptr<AbstractDataArray> arr);

    std::shared_ptr<AbstractDataArray> getVertexAttribute(const std::string& arrayName) const;

    bool hasVertexAttribute(const std::string& arrayName) const;

    void setVertexAttributes(std::unordered_map<std::string, std::shared_ptr<AbstractDataArray>> attributes) { m_vertexAttributes = attributes; }

    const std::unordered_map<std::string, std::shared_ptr<AbstractDataArray>>& getVertexAttributes() const { return m_vertexAttributes; }

    void setVertexScalars(const std::string& arrayName, std::shared_ptr<AbstractDataArray> scalars);
    void setVertexScalars(const std::string& arrayName);
    std::string getActiveVertexScalars() const { return m_activeVertexScalars; }
    std::shared_ptr<AbstractDataArray> getVertexScalars() const;

    void setVertexNormals(const std::string& arrayName, std::shared_ptr<VecDataArray<double, 3>> normals);
    void setVertexNormals(const std::string& arrayName);
    std::string getActiveVertexNormals() const { return m_activeVertexNormals; }
    std::shared_ptr<VecDataArray<double, 3>> getVertexNormals() const;

    void setVertexTangents(const std::string& arrayName, std::shared_ptr<VecDataArray<float, 3>> tangents);
    void setVertexTangents(const std::string& arrayName);
    std::string getActiveVertexTangents() const { return m_activeVertexTangents; }
    std::shared_ptr<VecDataArray<float, 3>> getVertexTangents() const;

    void setVertexTCoords(const std::string& arrayName, std::shared_ptr<VecDataArray<float, 2>> tcoords);
    void setVertexTCoords(const std::string& arrayName);
    std::string getActiveVertexTCoords() const { return m_activeVertexTCoords; }
    std::shared_ptr<VecDataArray<float, 2>> getVertexTCoords() const;

    void updatePostTransformData() const override;

    std::unique_ptr<PointSet> clone()
    {
        return std::unique_ptr<PointSet>(cloneImplementation());
    }

protected:
    void applyTransform(const Mat4d& m) override;

    void setActiveVertexAttribute(std::string& activeAttributeName, const std::string attributeName,
                                  const int expectedNumComponents, const ScalarTypeId expectedScalarType);

    std::shared_ptr<VecDataArray<double, 3>> m_initialVertexPositions;
    std::shared_ptr<VecDataArray<double, 3>> m_vertexPositions;

    std::unordered_map<std::string, std::shared_ptr<AbstractDataArray>> m_vertexAttributes;
    std::string m_activeVertexNormals  = "";
    std::string m_activeVertexScalars  = "";
    std::string m_activeVertexTangents = "";
    std::string m_activeVertexTCoords  = "";

    Vec3d m_lowerCorner;
    Vec3d m_upperCorner;

private:
    PointSet* cloneImplementation() const;
};
} // namespace imstk