NonlinearFEOperator.h

Go to the documentation of this file.

#ifndef included_AMP_NonlinearFEOperator
#define included_AMP_NonlinearFEOperator
 
// AMP headers
#include "AMP/mesh/Mesh.h"
#include "AMP/mesh/MeshElement.h"
#include "AMP/operators/Operator.h"
#include "AMP/operators/libmesh/FEOperatorParameters.h"
 
// Libmesh headers
DISABLE_WARNINGS
#include "libmesh/libmesh_config.h"
#undef LIBMESH_ENABLE_REFERENCE_COUNTING
#include "libmesh/elem.h"
ENABLE_WARNINGS
 
#include <vector>
 
 
namespace AMP::Operator {
 
class NonlinearFEOperator : public Operator
{
public:
    explicit NonlinearFEOperator( std::shared_ptr<const OperatorParameters> params );
 
    virtual ~NonlinearFEOperator();
 
    std::string type() const override { return "NonlinearFEOperator"; }
 
    virtual void apply( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                        AMP::LinearAlgebra::Vector::shared_ptr f ) override;
 
protected:
    virtual void preAssembly( AMP::LinearAlgebra::Vector::const_shared_ptr u,
                              AMP::LinearAlgebra::Vector::shared_ptr r ) = 0;
 
    virtual void postAssembly() = 0;
 
    virtual void preElementOperation( const AMP::Mesh::MeshElement & ) = 0;
 
    virtual void postElementOperation() = 0;
 
    void createLibMeshElementList();
 
    void destroyLibMeshElementList();
 
    std::vector<libMesh::Elem *> d_currElemPtrs;
 
    size_t d_currElemIdx;
 
    std::shared_ptr<ElementOperation> d_elemOp; 
};
} // namespace AMP::Operator
 
#endif