MultiModel.cpp

Go to the documentation of this file.

#include "MultiModel.hpp"
 
// #include <polyfem/basis/Basis.hpp>
// #include <polyfem/autogen/auto_elasticity_rhs.hpp>
 
// #include <igl/Timer.h>
 
namespace polyfem::assembler
{
    void MultiModel::set_size(const int size)
    {
        Assembler::set_size(size);
 
        all_elastic_materials_.set_size(size);
    }
    void MultiModel::set_size(const int size) {…}
 
    void MultiModel::add_multimaterial(const int index, const json &params, const Units &units)
    {
        assert(size() == 2 || size() == 3);
 
        all_elastic_materials_.add_multimaterial(index, params, units);
    }
    void MultiModel::add_multimaterial(const int index, const json &params, const Units &units) {…}
 
    Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 3, 1>
    MultiModel::compute_rhs(const AutodiffHessianPt &pt) const
    {
        assert(pt.size() == size());
        Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 3, 1> res;
        assert(false);
 
        return res;
    }
    MultiModel::compute_rhs(const AutodiffHessianPt &pt) const {…}
 
    Eigen::VectorXd
    MultiModel::assemble_gradient(const NonLinearAssemblerData &data) const
    {
        const int el_id = data.vals.element_id;
        const std::string model = multi_material_models_[el_id];
        const auto assembler = all_elastic_materials_.get_assembler(model);
        return assembler->assemble_gradient(data);
    }
    MultiModel::assemble_gradient(const NonLinearAssemblerData &data) const {…}
 
    Eigen::MatrixXd
    MultiModel::assemble_hessian(const NonLinearAssemblerData &data) const
    {
        const int el_id = data.vals.element_id;
        const std::string model = multi_material_models_[el_id];
        const auto assembler = all_elastic_materials_.get_assembler(model);
        return assembler->assemble_hessian(data);
    }
    MultiModel::assemble_hessian(const NonLinearAssemblerData &data) const {…}
 
    double MultiModel::compute_energy(const NonLinearAssemblerData &data) const
    {
        const int el_id = data.vals.element_id;
        const std::string model = multi_material_models_[el_id];
        const auto assembler = all_elastic_materials_.get_assembler(model);
        return assembler->compute_energy(data);
    }
    double MultiModel::compute_energy(const NonLinearAssemblerData &data) const {…}
 
    void MultiModel::assign_stress_tensor(
        const OutputData &data,
        const int all_size,
        const ElasticityTensorType &type,
        Eigen::MatrixXd &all,
        const std::function<Eigen::MatrixXd(const Eigen::MatrixXd &)> &fun) const
    {
        const std::string model = multi_material_models_[data.el_id];
        const auto assembler = all_elastic_materials_.get_assembler(model);
        auto elasticity_assembler = std::dynamic_pointer_cast<assembler::ElasticityAssembler>(assembler);
        assert(elasticity_assembler && "Failed to cast assembler to ElasticityAssembler.");
        elasticity_assembler->assign_stress_tensor(data, all_size, type, all, fun);
    }
    void MultiModel::assign_stress_tensor( {…}
 
    std::map<std::string, Assembler::ParamFunc> MultiModel::parameters() const
    {
        return all_elastic_materials_.parameters();
    }
    std::map<std::string, Assembler::ParamFunc> MultiModel::parameters() const {…}
} // namespace polyfem::assembler