Laplacian.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/assembler/Assembler.hpp>
#include <polyfem/assembler/MatParams.hpp>
#include <polyfem/utils/AutodiffTypes.hpp>
 
// local assembler for laplace equation
namespace polyfem
{
    namespace assembler
    {
        class Laplacian : public LinearAssembler, public NLAssembler
        {
        public:
            explicit Laplacian(const std::string &conductivity_param_name = "");
 
            using LinearAssembler::assemble;
            using NLAssembler::assemble_energy;
            using NLAssembler::assemble_gradient;
            using NLAssembler::assemble_hessian;
 
            std::string name() const override { return "Laplacian"; }
            std::map<std::string, ParamFunc> parameters() const override;
            void add_multimaterial(const int index, const json &params, const Units &units, const std::string &root_path) override;
 
            Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 9, 1> assemble(const LinearAssemblerData &data) const override;
 
            double compute_energy(const NonLinearAssemblerData &data) const override;
            Eigen::VectorXd assemble_gradient(const NonLinearAssemblerData &data) const override;
            Eigen::MatrixXd assemble_hessian(const NonLinearAssemblerData &data) const override;
 
            VectorNd compute_rhs(const AutodiffHessianPt &pt) const override;
 
            void compute_stress_grad_multiply_mat(const OptAssemblerData &data,
                                                  const Eigen::MatrixXd &mat,
                                                  Eigen::MatrixXd &stress,
                                                  Eigen::MatrixXd &result) const override;
 
            void compute_stiffness_value(const double t,
                                         const assembler::ElementAssemblyValues &vals,
                                         const Eigen::MatrixXd &local_pts,
                                         const Eigen::MatrixXd &displacement,
                                         Eigen::MatrixXd &tensor) const override;
 
            Eigen::Matrix<AutodiffScalarGrad, Eigen::Dynamic, 1, 0, 3, 1> kernel(const int dim, const AutodiffGradPt &rvect, const AutodiffScalarGrad &r) const override;
 
            bool is_linear() const override { return true; }
 
        private:
            double conductivity(const RowVectorNd &uv, const RowVectorNd &p, double t, int element_id) const;
 
            std::string conductivity_param_name_;
            GenericMatParam conductivity_;
        };
    } // namespace assembler
} // namespace polyfem