Calculate the grad-grad operator and assemble matrix. More...

#include <tutorials/cor-2to5/src/PoissonOperators.hpp>

Inheritance diagram for PoissonExample::OpK:

Collaboration diagram for PoissonExample::OpK:

Public Member Functions
	OpK (bool symm=true)

MoFEMErrorCode	doWork (int row_side, int col_side, EntityType row_type, EntityType col_type, EntitiesFieldData::EntData &row_data, EntitiesFieldData::EntData &col_data)
	Do calculations for give operator. More...

Public Member Functions inherited from MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator
int	getNumNodes ()
	get element number of nodes More...

const EntityHandle *	getConn ()
	get element connectivity More...

double	getVolume () const
	element volume (linear geometry) More...

double &	getVolume ()
	element volume (linear geometry) More...

FTensor::Tensor2< double *, 3, 3 > &	getJac ()
	get element Jacobian More...

FTensor::Tensor2< double *, 3, 3 > &	getInvJac ()
	get element inverse Jacobian More...

VectorDouble &	getCoords ()
	nodal coordinates More...

VolumeElementForcesAndSourcesCore *	getVolumeFE () const
	return pointer to Generic Volume Finite Element object More...

Public Member Functions inherited from MoFEM::ForcesAndSourcesCore::UserDataOperator
	UserDataOperator (const FieldSpace space, const char type=OPSPACE, const bool symm=true)

	UserDataOperator (const std::string field_name, const char type, const bool symm=true)

	UserDataOperator (const std::string row_field_name, const std::string col_field_name, const char type, const bool symm=true)

boost::shared_ptr< const NumeredEntFiniteElement >	getNumeredEntFiniteElementPtr () const
	Return raw pointer to NumeredEntFiniteElement. More...

EntityHandle	getFEEntityHandle () const
	Return finite element entity handle. More...

int	getFEDim () const
	Get dimension of finite element. More...

EntityType	getFEType () const
	Get dimension of finite element. More...

boost::weak_ptr< SideNumber >	getSideNumberPtr (const int side_number, const EntityType type)
	Get the side number pointer. More...

EntityHandle	getSideEntity (const int side_number, const EntityType type)
	Get the side entity. More...

int	getNumberOfNodesOnElement () const
	Get the number of nodes on finite element. More...

MoFEMErrorCode	getProblemRowIndices (const std::string filed_name, const EntityType type, const int side, VectorInt &indices) const
	Get row indices. More...

MoFEMErrorCode	getProblemColIndices (const std::string filed_name, const EntityType type, const int side, VectorInt &indices) const
	Get col indices. More...

const FEMethod *	getFEMethod () const
	Return raw pointer to Finite Element Method object. More...

int	getOpType () const
	Get operator types. More...

void	setOpType (const OpType type)
	Set operator type. More...

void	addOpType (const OpType type)
	Add operator type. More...

int	getNinTheLoop () const
	get number of finite element in the loop More...

int	getLoopSize () const
	get size of elements in the loop More...

std::string	getFEName () const
	Get name of the element. More...

ForcesAndSourcesCore *	getPtrFE () const

ForcesAndSourcesCore *	getSidePtrFE () const

ForcesAndSourcesCore *	getRefinePtrFE () const

const PetscData::Switches &	getDataCtx () const

const KspMethod::KSPContext	getKSPCtx () const

const SnesMethod::SNESContext	getSNESCtx () const

const TSMethod::TSContext	getTSCtx () const

Vec	getKSPf () const

Mat	getKSPA () const

Mat	getKSPB () const

Vec	getSNESf () const

Vec	getSNESx () const

Mat	getSNESA () const

Mat	getSNESB () const

Vec	getTSu () const

Vec	getTSu_t () const

Vec	getTSu_tt () const

Vec	getTSf () const

Mat	getTSA () const

Mat	getTSB () const

int	getTSstep () const

double	getTStime () const

double	getTStimeStep () const

double	getTSa () const

double	getTSaa () const

MatrixDouble &	getGaussPts ()
	matrix of integration (Gauss) points for Volume Element More...

auto	getFTensor0IntegrationWeight ()
	Get integration weights. More...

MatrixDouble &	getCoordsAtGaussPts ()
	Gauss points and weight, matrix (nb. of points x 3) More...

auto	getFTensor1CoordsAtGaussPts ()
	Get coordinates at integration points assuming linear geometry. More...

double	getMeasure () const
	get measure of element More...

double &	getMeasure ()
	get measure of element More...

MoFEMErrorCode	loopSide (const string &fe_name, ForcesAndSourcesCore side_fe, const size_t dim, const EntityHandle ent_for_side=0, const int verb=QUIET, const LogManager::SeverityLevel sev=Sev::noisy, AdjCache adj_cache=nullptr)
	User calls this function to loop over elements on the side of face. This function calls finite element with its operator to do calculations. More...

MoFEMErrorCode	loopThis (const string &fe_name, ForcesAndSourcesCore *this_fe, const int verb=QUIET, const LogManager::SeverityLevel sev=Sev::noisy)
	User calls this function to loop over the same element using a different set of integration points. This function calls finite element with its operator to do calculations. More...

MoFEMErrorCode	loopParent (const string &fe_name, ForcesAndSourcesCore *parent_fe, const int verb=QUIET, const LogManager::SeverityLevel sev=Sev::noisy)
	User calls this function to loop over parent elements. This function calls finite element with its operator to do calculations. More...

MoFEMErrorCode	loopChildren (const string &fe_name, ForcesAndSourcesCore *child_fe, const int verb=QUIET, const LogManager::SeverityLevel sev=Sev::noisy)
	User calls this function to loop over parent elements. This function calls finite element with its operator to do calculations. More...

Public Member Functions inherited from MoFEM::DataOperator
	DataOperator (const bool symm=true)

virtual	~DataOperator ()=default

virtual MoFEMErrorCode	opLhs (EntitiesFieldData &row_data, EntitiesFieldData &col_data)

virtual MoFEMErrorCode	doWork (int side, EntityType type, EntitiesFieldData::EntData &data)
	Operator for linear form, usually to calculate values on right hand side. More...

virtual MoFEMErrorCode	opRhs (EntitiesFieldData &data, const bool error_if_no_base=false)

bool	getSymm () const
	Get if operator uses symmetry of DOFs or not. More...

void	setSymm ()
	set if operator is executed taking in account symmetry More...

void	unSetSymm ()
	unset if operator is executed for non symmetric problem More...

Protected Member Functions
virtual MoFEMErrorCode	iNtegrate (EntitiesFieldData::EntData &row_data, EntitiesFieldData::EntData &col_data)
	Integrate grad-grad operator. More...

virtual MoFEMErrorCode	aSsemble (EntitiesFieldData::EntData &row_data, EntitiesFieldData::EntData &col_data)
	Assemble local entity block matrix. More...

Protected Member Functions inherited from MoFEM::VolumeElementForcesAndSourcesCore::UserDataOperator
MoFEMErrorCode	setPtrFE (ForcesAndSourcesCore *ptr)

Protected Attributes
int	nbRows
	< error code More...

int	nbCols
	number if dof on column More...

int	nbIntegrationPts
	number of integration points More...

bool	isDiag
	true if this block is on diagonal More...

FTensor::Index< 'i', 3 >	i
	summit Index More...

MatrixDouble	locMat
	local entity block matrix More...

Protected Attributes inherited from MoFEM::ForcesAndSourcesCore::UserDataOperator
ForcesAndSourcesCore *	ptrFE

Additional Inherited Members
Public Types inherited from MoFEM::ForcesAndSourcesCore::UserDataOperator
enum	OpType { OPROW = 1 << 0, OPCOL = 1 << 1, OPROWCOL = 1 << 2, OPSPACE = 1 << 3, OPLAST = 1 << 3 }
	Controls loop over entities on element. More...

using	AdjCache = std::map< EntityHandle, std::vector< boost::weak_ptr< NumeredEntFiniteElement > >>

Public Types inherited from MoFEM::DataOperator
using	DoWorkLhsHookFunType = boost::function< MoFEMErrorCode(DataOperator *op_ptr, int row_side, int col_side, EntityType row_type, EntityType col_type, EntitiesFieldData::EntData &row_data, EntitiesFieldData::EntData &col_data)>

using	DoWorkRhsHookFunType = boost::function< MoFEMErrorCode(DataOperator *op_ptr, int side, EntityType type, EntitiesFieldData::EntData &data)>

Public Attributes inherited from MoFEM::ForcesAndSourcesCore::UserDataOperator
char	opType

std::string	rowFieldName

std::string	colFieldName

FieldSpace	sPace

Public Attributes inherited from MoFEM::DataOperator
DoWorkLhsHookFunType	doWorkLhsHook

DoWorkRhsHookFunType	doWorkRhsHook

bool	sYmm
	If true assume that matrix is symmetric structure. More...

std::array< bool, MBMAXTYPE >	doEntities
	If true operator is executed for entity. More...

bool &	doVertices
	\deprectaed If false skip vertices More...

bool &	doEdges
	\deprectaed If false skip edges More...

bool &	doQuads
	\deprectaed More...

bool &	doTris
	\deprectaed More...

bool &	doTets
	\deprectaed More...

bool &	doPrisms
	\deprectaed More...

Static Public Attributes inherited from MoFEM::ForcesAndSourcesCore::UserDataOperator
static const char *const	OpTypeNames []

Detailed Description

Calculate the grad-grad operator and assemble matrix.

Calculate

\[ \mathbf{K}=\int_\Omega \nabla \boldsymbol\phi \cdot \nabla \boldsymbol\phi \textrm{d}\Omega \]

and assemble to global matrix.

This operator is executed on element for each unique combination of entities.

Definition at line 28 of file PoissonOperators.hpp.

Constructor & Destructor Documentation

◆ OpK()

PoissonExample::OpK::OpK ( bool symm = true )

inline

Definition at line 30 of file PoissonOperators.hpp.

31 : VolumeElementForcesAndSourcesCore::UserDataOperator("U", "U", OPROWCOL,

32 symm) {}

Member Function Documentation

◆ aSsemble()

virtual MoFEMErrorCode PoissonExample::OpK::aSsemble	(	EntitiesFieldData::EntData &	row_data,
		EntitiesFieldData::EntData &	col_data
	)

inlineprotectedvirtual

Assemble local entity block matrix.

Parameters

row_data	row data (consist base functions on row entity)
col_data	column data (consist base functions on column entity)

Returns: error code

Definition at line 137 of file PoissonOperators.hpp.

                                                                       {
     MoFEMFunctionBegin;
     // get pointer to first global index on row
     const int *row_indices = &*row_data.getIndices().data().begin();
     // get pointer to first global index on column
     const int *col_indices = &*col_data.getIndices().data().begin();
     Mat B = getFEMethod()->ksp_B != PETSC_NULL ? getFEMethod()->ksp_B
                                                : getFEMethod()->snes_B;
     // assemble local matrix
     CHKERR MatSetValues(B, nbRows, row_indices, nbCols, col_indices,
                         &*locMat.data().begin(), ADD_VALUES);
  
     if (!isDiag && sYmm) {
       // if not diagonal term and since global matrix is symmetric assemble
       // transpose term.
       locMat = trans(locMat);
       CHKERR MatSetValues(B, nbCols, col_indices, nbRows, row_indices,
                           &*locMat.data().begin(), ADD_VALUES);
     }
     MoFEMFunctionReturn(0);
   }

◆ doWork()

MoFEMErrorCode PoissonExample::OpK::doWork	(	int	row_side,
		int	col_side,
		EntityType	row_type,
		EntityType	col_type,
		EntitiesFieldData::EntData &	row_data,
		EntitiesFieldData::EntData &	col_data
	)

inlinevirtual

Do calculations for give operator.

Parameters

row_side	row side number (local number) of entity on element
col_side	column side number (local number) of entity on element
row_type	type of row entity MBVERTEX, MBEDGE, MBTRI or MBTET
col_type	type of column entity MBVERTEX, MBEDGE, MBTRI or MBTET
row_data	data for row
col_data	data for column

Returns: error code

Reimplemented from MoFEM::DataOperator.

Definition at line 44 of file PoissonOperators.hpp.

                                                             {
     MoFEMFunctionBegin;
     // get number of dofs on row
     nbRows = row_data.getIndices().size();
     // if no dofs on row, exit that work, nothing to do here
     if (!nbRows)
       MoFEMFunctionReturnHot(0);
     // get number of dofs on column
     nbCols = col_data.getIndices().size();
     // if no dofs on Columbia, exit nothing to do here
     if (!nbCols)
       MoFEMFunctionReturnHot(0);
     // get number of integration points
     nbIntegrationPts = getGaussPts().size2();
     // check if entity block is on matrix diagonal
     if (row_side == col_side && row_type == col_type) {
       isDiag = true; // yes, it is
     } else {
       isDiag = false;
     }
     // integrate local matrix for entity block
     CHKERR iNtegrate(row_data, col_data);
     // assemble local matrix
     CHKERR aSsemble(row_data, col_data);
     MoFEMFunctionReturn(0);
   }

◆ iNtegrate()

virtual MoFEMErrorCode PoissonExample::OpK::iNtegrate	(	EntitiesFieldData::EntData &	row_data,
		EntitiesFieldData::EntData &	col_data
	)

inlineprotectedvirtual

Integrate grad-grad operator.

Parameters

row_data	row data (consist base functions on row entity)
col_data	column data (consist base functions on column entity)

Returns: error code

Reimplemented in PoissonExample::OpKt.

Definition at line 91 of file PoissonOperators.hpp.

                                                                        {
     MoFEMFunctionBegin;
     // set size of local entity bock
     locMat.resize(nbRows, nbCols, false);
     // clear matrix
     locMat.clear();
     // get element volume
     double vol = getVolume();
     // get integration weights
     auto t_w = getFTensor0IntegrationWeight();
     // get base function gradient on rows
     auto t_row_grad = row_data.getFTensor1DiffN<3>();
     // loop over integration points
     for (int gg = 0; gg != nbIntegrationPts; gg++) {
       // take into account Jacobean
       const double alpha = t_w * vol;
       // noalias(locMat) +=
       // alpha*prod(row_data.getDiffN(gg),trans(col_data.getDiffN(gg))); take
       // fist element to local matrix
       FTensor::Tensor0<double *> a(&*locMat.data().begin());
       // loop over rows base functions
       for (int rr = 0; rr != nbRows; rr++) {
         // get column base functions gradient at gauss point gg
         auto t_col_grad = col_data.getFTensor1DiffN<3>(gg, 0);
         // loop over columns
         for (int cc = 0; cc != nbCols; cc++) {
           // calculate element of local matrix
           a += alpha * (t_row_grad(i) * t_col_grad(i));
           ++t_col_grad; // move to another gradient of base function on column
           ++a;          // move to another element of local matrix in column
         }
         ++t_row_grad; // move to another element of gradient of base function on
                       // row
       }
       ++t_w; // move to another integration weight
     }
     MoFEMFunctionReturn(0);
   }

Member Data Documentation

◆ i

FTensor::Index<'i', 3> PoissonExample::OpK::i

protected

summit Index

Definition at line 82 of file PoissonOperators.hpp.

◆ isDiag

bool PoissonExample::OpK::isDiag

protected

true if this block is on diagonal

Definition at line 80 of file PoissonOperators.hpp.

◆ locMat

MatrixDouble PoissonExample::OpK::locMat

protected

local entity block matrix

Definition at line 83 of file PoissonOperators.hpp.

◆ nbCols

int PoissonExample::OpK::nbCols

protected

number if dof on column

Definition at line 78 of file PoissonOperators.hpp.

◆ nbIntegrationPts

int PoissonExample::OpK::nbIntegrationPts

protected

number of integration points

Definition at line 79 of file PoissonOperators.hpp.

◆ nbRows

int PoissonExample::OpK::nbRows

protected

< error code

number of dofs on rows

Definition at line 77 of file PoissonOperators.hpp.

The documentation for this struct was generated from the following file:

tutorials/cor-2to5/src/PoissonOperators.hpp

Public Member Functions

Protected Member Functions

Protected Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ OpK()

Member Function Documentation

◆ aSsemble()

◆ doWork()

◆ iNtegrate()

Member Data Documentation

◆ i

◆ isDiag

◆ locMat

◆ nbCols

◆ nbIntegrationPts

◆ nbRows