AssemblerUtils.hpp

Go to the documentation of this file.

#pragma once
 
#include <polyfem/assembler/Assembler.hpp>
#include <polyfem/basis/ElementBases.hpp>
#include <polyfem/utils/AutodiffTypes.hpp>
#include <polyfem/utils/MatrixUtils.hpp>
 
namespace polyfem::assembler
{
    class AssemblerUtils
    {
    public:
        static std::string other_assembler_name(const std::string &formulation);
 
        static std::shared_ptr<Assembler> make_assembler(const std::string &formulation);
        static std::shared_ptr<MixedAssembler> make_mixed_assembler(const std::string &formulation);
        static std::shared_ptr<MixedNLAssembler> make_mixed_nl_assembler(const std::string &formulation);
 
        enum class BasisType
        {
            SIMPLEX_LAGRANGE,
            CUBE_LAGRANGE,
            PRISM_LAGRANGE,
            PYRAMID_LAGRANGE,
            SPLINE,
            POLY
        };
 
        AssemblerUtils() = delete;
 
        static void merge_mixed_matrices(
            const int n_bases, const int n_pressure_bases, const int problem_dim, const bool add_average,
            const StiffnessMatrix &velocity_stiffness, const StiffnessMatrix &mixed_stiffness, const StiffnessMatrix &pressure_stiffness,
            StiffnessMatrix &stiffness);
 
        static int quadrature_order(const std::string &assembler, const int basis_degree, const BasisType &b_type, const int dim);
 
        static bool is_elastic_material(const std::string &material);
        static std::vector<std::string> elastic_materials();
    };
 
    class AllElasticMaterials
    {
 
    public:
        AllElasticMaterials();
 
        void set_size(const int size);
        void add_multimaterial(const int index, const json &params, const Units &units, const std::string &root_path);
        std::shared_ptr<assembler::ElasticityNLAssembler> get_assembler(const std::string &name) const;
 
        std::map<std::string, Assembler::ParamFunc> parameters() const;
 
    private:
        std::unordered_map<std::string, std::shared_ptr<assembler::ElasticityNLAssembler>> elastic_material_map_;
    };
} // namespace polyfem::assembler