PerSimplexAutodiffFunction.hpp

Go to the documentation of this file.

#pragma once
 
#include "PerSimplexFunction.hpp"
 
#include <wmtk/function/utils/autodiff.h>
#include <wmtk/function/utils/AutoDiffRAII.hpp>
#include <wmtk/function/utils/AutoDiffUtils.hpp>
#include <wmtk/simplex/utils/SimplexComparisons.hpp>
 
#include <optional>
 
namespace wmtk::function {
class PerSimplexAutodiffFunction : public PerSimplexFunction
{
public:
    using DScalar = DScalar2<double, Eigen::Matrix<double, -1, 1>, Eigen::Matrix<double, -1, -1>>;
    using Scalar = typename DScalar::Scalar;
    using DSVec = Eigen::VectorX<DScalar>;
 
    static_assert(
        std::is_same_v<Scalar, double>); // MTAO: i'm leaving scalar here but is it ever not double?
    PerSimplexAutodiffFunction(
        const Mesh& mesh,
        const PrimitiveType primitive_type,
        const attribute::MeshAttributeHandle& variable_attribute_handle);
 
    ~PerSimplexAutodiffFunction();
 
 
    double get_value(const simplex::Simplex& domain_simplex) const override;
 
    Eigen::VectorXd get_gradient(
        const simplex::Simplex& domain_simplex,
        const simplex::Simplex& variable_simplex) const override;
 
    Eigen::MatrixXd get_hessian(
        const simplex::Simplex& domain_simplex,
        const simplex::Simplex& variable_simplex) const override;
 
protected:
    std::vector<DSVec> get_coordinates(
        const simplex::Simplex& domain_simplex,
        const std::optional<simplex::Simplex>& variable_simplex_opt = {}) const;
 
    std::vector<DSVec> get_coordinates(
        const attribute::Accessor<double>& accessor,
        const simplex::Simplex& domain_simplex,
        const std::optional<simplex::Simplex>& variable_simplex_opt = {}) const;
 
 
    virtual DScalar eval(
        const simplex::Simplex& domain_simplex,
        const std::vector<DSVec>& coordinates) const = 0;
};
 
} // namespace wmtk::function