MassLinearElement.h

Go to the documentation of this file.

#ifndef included_AMP_MassLinearElement
#define included_AMP_MassLinearElement
 
#include <vector>
 
#include <memory>
 
// AMP files
#include "AMP/operators/libmesh/MassElement.h"
 
namespace AMP::Operator {
 
 
class MassLinearElement : public MassElement
{
public:
    explicit MassLinearElement( std::shared_ptr<const ElementOperationParameters> params )
        : MassElement( params ),
          d_elementMassMatrix( NULL ),
          d_equation( MassDensityModel::MassEquation::UnknownMassEquation )
    {
        d_densityAtGauss = params->d_db->getWithDefault<bool>( "DensityAtGaussPoints", true );
    }
 
    virtual ~MassLinearElement() {}
 
    void setElementMassMatrix( std::vector<std::vector<double>> &elementMassMatrix )
    {
        d_elementMassMatrix = &( elementMassMatrix );
    }
 
    void setElementVectors( const std::vector<double> &localTemp,
                            const std::vector<double> &localConc,
                            const std::vector<double> &localBurn )
    {
        d_LocalTemperature   = localTemp;
        d_LocalConcentration = localConc;
        d_LocalBurnup        = localBurn;
    }
 
    void apply() override;
 
protected:
    std::vector<std::vector<double>> *d_elementMassMatrix;
 
    bool d_densityAtGauss;
 
    std::vector<double> d_LocalTemperature;
    std::vector<double> d_LocalConcentration;
    std::vector<double> d_LocalBurnup;
 
    MassDensityModel::MassEquation d_equation;
 
private:
};
} // namespace AMP::Operator
 
#endif