Mass.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>
 
#include <memory>
 
namespace polyfem::assembler
{
    class Mass : public LinearAssembler
    {
    public:
        Mass();
        explicit Mass(std::shared_ptr<Density> density);
 
        using LinearAssembler::assemble;
 
        Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 9, 1>
        assemble(const LinearAssemblerData &data) const override;
 
        Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 3, 1> compute_rhs(const AutodiffHessianPt &pt) const override;
 
        void add_multimaterial(const int index, const json &params, const Units &units, const std::string &root_path) override;
 
        const Density &density() const { return *density_; }
 
        std::string name() const override { return "Mass"; }
        virtual std::map<std::string, ParamFunc> parameters() const override;
 
    private:
        // class that stores and compute density per point
        std::shared_ptr<Density> density_;
    };
 
    class HRZMass : public LinearAssembler
    {
    public:
        using LinearAssembler::assemble;
 
        Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 9, 1>
        assemble(const LinearAssemblerData &data) const override;
 
        virtual std::map<std::string, ParamFunc> parameters() const override
        {
            std::map<std::string, ParamFunc> res;
            return res;
        }
 
        std::string name() const override { return "HRZMass"; }
    };
} // namespace polyfem::assembler