GeodesicsInHeat.h

#pragma once

#include <vector>
#include <Eigen/SparseCholesky>
#include <Euclid/MeshUtil/MeshDefs.h>
#include <Euclid/Util/Memory.h>

namespace Euclid
{
template<typename Mesh>
class GeodesicsInHeat
{
public:
    using FT = FT_t<Mesh>;
    using Vector_3 = Vector_3_t<Mesh>;
    using SpMat = Eigen::SparseMatrix<FT>;

public:
    void build(const Mesh& mesh,
               float scale = 1.0f,
               FT resolution = 0,
               const SpMat* cot_mat = nullptr,
               const SpMat* mass_mat = nullptr);

    void scale(float scale);

    template<typename T>
    void compute(
        const typename boost::graph_traits<const Mesh>::vertex_descriptor& v,
        std::vector<T>& geodesics);

public:
    const Mesh* mesh = nullptr;

    FT resolution = 0.0;

    ProPtr<const SpMat> cot_mat = nullptr;

    ProPtr<const SpMat> mass_mat = nullptr;

    Eigen::SimplicialLDLT<SpMat> heat_solver;

    Eigen::SimplicialLDLT<SpMat> poisson_solver;
};

} // namespace Euclid

#include "src/GeodesicsInHeat.cpp"