imstkTetrahedralMesh.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 "imstkMacros.h"
#include "imstkVolumetricMesh.h"
#include "imstkVecDataArray.h"

#include <array>

namespace imstk
{
class TetrahedralMesh : public VolumetricMesh<4>
{
public:
    TetrahedralMesh() = default;
    ~TetrahedralMesh() override = default;

    static std::string StrainParameterName;

    IMSTK_TYPE_NAME(TetrahedralMesh)

    
    std::shared_ptr<SurfaceMesh> extractSurfaceMesh() override;

    Vec4d computeBarycentricWeights(const int tetId, const Vec3d& pos) const override;

    void computeTetrahedronBoundingBox(const size_t& tetId, Vec3d& min, Vec3d& max) const;

// Accessors
    void setTetrahedraAsRemoved(const unsigned int tetId)
    {
        m_removedMeshElems.reserve(getNumTetrahedra());
        m_removedMeshElems.push_back(tetId);
    }

    const std::vector<int>& getRemovedTetrahedra() const { return m_removedMeshElems; }

    double getVolume() override;

    int getNumTetrahedra() const { return getNumCells(); }
    void setTetrahedraIndices(std::shared_ptr<VecDataArray<int, 4>> indices) { setCells(indices); }
    std::shared_ptr<VecDataArray<int, 4>> getTetrahedraIndices() const { return getCells(); }

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

    std::shared_ptr<VecDataArray<double, 3>> getStrainParameters();
    void setStrainParameters(std::shared_ptr<VecDataArray<double, 3>> strainParameters);

protected:
    std::vector<int> m_removedMeshElems;

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