ProblemWithSolution.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/assembler/Problem.hpp>
#include <polyfem/assembler/Assembler.hpp>
 
#include <polyfem/utils/AutodiffTypes.hpp>
 
namespace polyfem
{
    namespace problem
    {
        class ProblemWithSolution : public assembler::Problem
        {
        public:
            ProblemWithSolution(const std::string &name);
 
            virtual void rhs(const assembler::Assembler &assembler, const Eigen::MatrixXd &pts, const double t, Eigen::MatrixXd &val) const override;
            virtual void dirichlet_bc(const mesh::Mesh &mesh, const Eigen::MatrixXi &global_ids, const Eigen::MatrixXd &uv, const Eigen::MatrixXd &pts, const double t, Eigen::MatrixXd &val) const override;
 
            virtual void exact(const Eigen::MatrixXd &pts, const double t, Eigen::MatrixXd &val) const override;
            virtual void exact_grad(const Eigen::MatrixXd &pts, const double t, Eigen::MatrixXd &val) const override;
 
            virtual bool has_exact_sol() const override { return true; }
            virtual bool is_rhs_zero() const override { return false; }
 
            virtual ~ProblemWithSolution() {}
 
        protected:
            virtual VectorNd eval_fun(const VectorNd &pt, double t) const = 0;
            virtual AutodiffGradPt eval_fun(const AutodiffGradPt &pt, double t) const = 0;
            virtual AutodiffHessianPt eval_fun(const AutodiffHessianPt &pt, double t) const = 0;
 
            virtual int size_for(const Eigen::MatrixXd &pts) const { return is_scalar() ? 1 : pts.cols(); }
        };
 
        class BilaplacianProblemWithSolution : public assembler::Problem
        {
        public:
            BilaplacianProblemWithSolution(const std::string &name);
 
            void rhs(const assembler::Assembler &assembler, const Eigen::MatrixXd &pts, const double t, Eigen::MatrixXd &val) const override;
            void dirichlet_bc(const mesh::Mesh &mesh, const Eigen::MatrixXi &global_ids, const Eigen::MatrixXd &uv, const Eigen::MatrixXd &pts, const double t, Eigen::MatrixXd &val) const override;
 
            void exact(const Eigen::MatrixXd &pts, const double t, Eigen::MatrixXd &val) const override;
            void exact_grad(const Eigen::MatrixXd &pts, const double t, Eigen::MatrixXd &val) const override;
 
            bool has_exact_sol() const override { return true; }
            bool is_rhs_zero() const override { return false; }
            bool is_scalar() const override { return true; }
 
            virtual ~BilaplacianProblemWithSolution() {}
        };
    } // namespace problem
} // namespace polyfem