PoissonExample::OpK Struct Reference

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.

Protected Member Functions
virtual MoFEMErrorCode	iNtegrate (EntitiesFieldData::EntData &row_data, EntitiesFieldData::EntData &col_data)
	Integrate grad-grad operator.
virtual MoFEMErrorCode	aSsemble (EntitiesFieldData::EntData &row_data, EntitiesFieldData::EntData &col_data)
	Assemble local entity block matrix.

Protected Attributes
int	nbRows
	< error code
int	nbCols
	number if dof on column
int	nbIntegrationPts
	number of integration points
bool	isDiag
	true if this block is on diagonal
FTensor::Index< 'i', 3 >	i
	summit Index
MatrixDouble	locMat
	local entity block matrix

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.

      : VolumeElementForcesAndSourcesCore::UserDataOperator("U", "U", OPROWCOL,
                                                            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

Examples

PoissonOperators.hpp.

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_NULLPTR ? 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 )

inline

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

Examples

PoissonOperators.hpp.

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.

Examples

PoissonOperators.hpp.

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

Examples

PoissonOperators.hpp.

Definition at line 82 of file PoissonOperators.hpp.

isDiag

bool PoissonExample::OpK::isDiag

protected

true if this block is on diagonal

Examples

PoissonOperators.hpp.

Definition at line 80 of file PoissonOperators.hpp.

locMat

MatrixDouble PoissonExample::OpK::locMat

protected

local entity block matrix

Examples

PoissonOperators.hpp.

Definition at line 83 of file PoissonOperators.hpp.

nbCols

int PoissonExample::OpK::nbCols

protected

number if dof on column

Examples

PoissonOperators.hpp.

Definition at line 78 of file PoissonOperators.hpp.

nbIntegrationPts

int PoissonExample::OpK::nbIntegrationPts

protected

number of integration points

Examples

PoissonOperators.hpp.

Definition at line 79 of file PoissonOperators.hpp.

nbRows

int PoissonExample::OpK::nbRows

protected

< error code

number of dofs on rows

Examples

PoissonOperators.hpp.

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