SourceNonlinearElement.h

Go to the documentation of this file.

#ifndef included_AMP_SourceNonlinearElement
#define included_AMP_SourceNonlinearElement
 
#include <vector>
 
#include <memory>
 
// AMP files
#include "AMP/operators/ElementOperation.h"
#include "AMP/operators/libmesh/SourcePhysicsModel.h"
 
// Libmesh headers
DISABLE_WARNINGS
#include "libmesh/libmesh_config.h"
#undef LIBMESH_ENABLE_REFERENCE_COUNTING
#include "libmesh/elem.h"
#include "libmesh/fe_base.h"
#include "libmesh/fe_type.h"
#include "libmesh/quadrature_gauss.h"
ENABLE_WARNINGS
 
 
namespace AMP::Operator {
 
class SourceNonlinearElement : public ElementOperation
{
public:
    explicit SourceNonlinearElement( std::shared_ptr<const ElementOperationParameters> params );
 
    virtual ~SourceNonlinearElement() {}
 
    void initializeForCurrentElement( const libMesh::Elem *elem,
                                      std::shared_ptr<SourcePhysicsModel> sourceTransportModel );
 
    void setElementInputVector( const std::vector<std::vector<double>> &elementInputVector )
    {
        d_elementInputVector = elementInputVector;
    }
 
    void setElementVectors( const std::vector<std::vector<double>> &elementInputVector,
                            const std::vector<std::vector<double>> &elementAuxVector,
                            std::vector<double> &elementOutputVector )
    {
        d_elementInputVector  = elementInputVector;
        d_elementAuxVector    = elementAuxVector;
        d_elementOutputVector = &( elementOutputVector );
    }
 
    void setElementFlags( const std::string &inputVariableType )
    {
        d_isInputType = inputVariableType;
    }
 
    void apply() override;
 
    std::shared_ptr<libMesh::FEBase> getFEBase() { return d_fe; }
 
    unsigned int getNumberOfGaussPoints() { return ( d_qrule->n_points() ); }
 
 
protected:
    std::vector<std::vector<double>> d_elementInputVector;
 
    std::vector<std::vector<double>> d_elementAuxVector;
 
    std::vector<double> *d_elementOutputVector;
 
    std::vector<std::vector<double>> d_elementOtherVectors;
 
    std::shared_ptr<libMesh::FEType> d_feType;
 
    std::shared_ptr<libMesh::FEBase> d_fe;
 
    std::shared_ptr<libMesh::QBase> d_qrule;
 
    const std::vector<libMesh::Real> *d_JxW;
 
    const std::vector<std::vector<libMesh::Real>> *d_phi;
 
    const std::vector<std::vector<libMesh::RealGradient>> *d_dphi;
 
    std::string d_isInputType;
 
    const libMesh::Elem *d_elem;
 
    bool d_integrateVolume;
 
    std::shared_ptr<SourcePhysicsModel> d_sourcePhysicsModel;
 
private:
};
} // namespace AMP::Operator
 
#endif