Schur complement data storage. More...

Inheritance diagram for MoFEM::SchurElemMats:

Collaboration diagram for MoFEM::SchurElemMats:

Classes
struct	col_mi_tag

struct	uid_mi_tag

Public Member Functions
	SchurElemMats (const size_t idx, const UId uid_row, const UId uid_col)

virtual	~SchurElemMats ()=default

auto &	getMat () const

auto &	getRowInd () const

auto &	getColInd () const

Static Public Member Functions
static MoFEMErrorCode	MatSetValues (Mat M, const EntitiesFieldData::EntData &row_data, const EntitiesFieldData::EntData &col_data, const MatrixDouble &mat, InsertMode iora)

Public Attributes
const size_t	iDX

UId	uidRow

UId	uidCol

Protected Types
using	SchurElemStorage = multi_index_container< const SchurElemMats *, indexed_by< ordered_unique< tag< uid_mi_tag >, composite_key< SchurElemMats, member< SchurElemMats, const UId, &SchurElemMats::uidRow >, member< SchurElemMats, const UId, &SchurElemMats::uidCol > > >, ordered_non_unique< tag< col_mi_tag >, member< SchurElemMats, const UId, &SchurElemMats::uidCol > > > >

Static Protected Member Functions
static MoFEMErrorCode	assembleStorage (const EntitiesFieldData::EntData &row_data, const EntitiesFieldData::EntData &col_data, const MatrixDouble &mat, InsertMode iora)

Static Protected Attributes
static boost::ptr_vector< MatrixDouble >	locMats

static boost::ptr_vector< VectorInt >	rowIndices

static boost::ptr_vector< VectorInt >	colIndices

static boost::ptr_vector< SchurElemMats >	schurElemMats

static size_t	maxIndexCounter = 0

static SchurElemStorage	schurL2Storage

Friends
struct	OpSchurAssembleBegin

template<typename OP_SCHUR_ASSEMBLE_BASE >
struct	OpSchurAssembleEndImpl

Detailed Description

Schur complement data storage.

Definition at line 173 of file Schur.cpp.

Member Typedef Documentation

◆ SchurElemStorage

using MoFEM::SchurElemMats::SchurElemStorage = multi_index_container< const SchurElemMats *, indexed_by< ordered_unique< tag<uid_mi_tag>, composite_key< SchurElemMats, member<SchurElemMats, const UId, &SchurElemMats::uidRow>, member<SchurElemMats, const UId, &SchurElemMats::uidCol> > >, ordered_non_unique<tag<col_mi_tag>, member<SchurElemMats, const UId, &SchurElemMats::uidCol> > > >

protected

Definition at line 221 of file Schur.cpp.

Constructor & Destructor Documentation

◆ SchurElemMats()

MoFEM::SchurElemMats::SchurElemMats	(	const size_t	idx,
		const UId	uid_row,
		const UId	uid_col
	)

Definition at line 362 of file Schur.cpp.

364 : iDX(idx), uidRow(uid_row), uidCol(uid_col) {}

◆ ~SchurElemMats()

virtual MoFEM::SchurElemMats::~SchurElemMats ( )

virtualdefault

Member Function Documentation

◆ assembleStorage()

MoFEMErrorCode MoFEM::SchurElemMats::assembleStorage	(	const EntitiesFieldData::EntData &	row_data,
		const EntitiesFieldData::EntData &	col_data,
		const MatrixDouble &	mat,
		InsertMode	iora
	)

staticprotected

Definition at line 367 of file Schur.cpp.

                                                                          {
   MoFEMFunctionBegin;
  
 #ifndef NDEBUG
   PetscLogEventBegin(SchurEvents::MOFEM_EVENT_schurMatSetValues, 0, 0, 0, 0);
 #endif // NDEBUG
  
   // get row indices, in case of store, get indices from storage
   // storage keeps marked indices to manage boundary conditions
   auto get_row_indices = [&]() -> const VectorInt & {
     if (auto e_ptr = row_data.getFieldEntities()[0]) {
       if (auto stored_data_ptr =
               e_ptr->getSharedStoragePtr<EssentialBcStorage>()) {
         return stored_data_ptr->entityIndices;
       }
     }
     return row_data.getIndices();
   };
  
   const auto &row_ind = get_row_indices();
   const auto &col_ind = col_data.getIndices();
  
   auto zero_rows_and_cols = [&](auto &mat, auto &row_ind, auto &col_ind) {
     MoFEMFunctionBeginHot;
     for (auto i = 0; i < row_ind.size(); ++i) {
       for (auto j = 0; j < col_ind.size(); ++j) {
         if (row_ind[i] < 0 || col_ind[j] < 0)
           mat(i, j) = 0;
       }
     }
     if (row_data.getFieldEntities()[0]->getLocalUniqueId() ==
         col_data.getFieldEntities()[0]->getLocalUniqueId()) {
       for (auto i = 0; i < row_ind.size(); ++i) {
         for (auto j = 0; j < col_ind.size(); ++j) {
           if (i == j && row_ind[i] < 0 && col_ind[j] < 0) {
             mat(i, j) = 1;
           }
         }
       }
     }
     MoFEMFunctionReturnHot(0);
   };
  
   const auto nb_rows = row_ind.size();
   const auto nb_cols = col_ind.size();
  
 #ifndef NDEBUG
   if (mat.size1() != nb_rows) {
     SETERRQ2(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
              "Wrong mat size %d != %d", mat.size1(), nb_rows);
   }
   if (mat.size2() != nb_cols) {
     SETERRQ2(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
              "Wrong mat size %d != %d", mat.size2(), nb_cols);
   }
 #endif // NDEBUG
  
   // get entity uid
   auto get_uid = [](auto &data) {
     if (data.getFieldEntities().size() == 1) {
  
       return data.getFieldEntities()[0]->getLocalUniqueId();
  
     } else {
  
       // Is assumed that sum of entities ids gives unique id, that is not true,
       // but corner case is improbable.
  
       // @todo: debug version should have test
  
       auto &uid0 = data.getFieldEntities()[0]->getLocalUniqueId();
       auto field_id0 = FieldEntity::getFieldBitNumberFromUniqueId(uid0);
       auto ent0 = FieldEntity::getHandleFromUniqueId(uid0);
       auto type0 = type_from_handle(ent0);
       auto id = id_from_handle(ent0);
  
       for (auto i = 1; i < data.getFieldEntities().size(); ++i) {
  
         // get entity id from ent
         id += id_from_handle(
  
             // get entity handle from unique uid
             FieldEntity::getHandleFromUniqueId(
                 data.getFieldEntities()[i]->getLocalUniqueId())
  
         );
       }
  
       return FieldEntity::getLocalUniqueIdCalculate(
           field_id0,
  
           ent_form_type_and_id(type0, id)
  
       );
     }
   };
  
   auto uid_row = get_uid(row_data);
   auto uid_col = get_uid(col_data);
  
   auto it =
       SchurElemMats::schurL2Storage.template get<SchurElemMats::uid_mi_tag>()
           .find(boost::make_tuple(uid_row, uid_col));
  
   if (it ==
       SchurElemMats::schurL2Storage.template get<SchurElemMats::uid_mi_tag>()
           .end()) {
  
     // get size of arrays of matrices
     const auto size = SchurElemMats::locMats.size();
  
     // expand memory allocation
     if (SchurElemMats::maxIndexCounter == size) {
       SchurElemMats::locMats.push_back(new MatrixDouble());
       SchurElemMats::rowIndices.push_back(new VectorInt());
       SchurElemMats::colIndices.push_back(new VectorInt());
       SchurElemMats::schurElemMats.push_back(
           new SchurElemMats(SchurElemMats::maxIndexCounter, uid_row, uid_col));
     } else {
       SchurElemMats::schurElemMats[SchurElemMats::maxIndexCounter].uidRow =
           uid_row;
       SchurElemMats::schurElemMats[SchurElemMats::maxIndexCounter].uidCol =
           uid_col;
     }
  
     // add matrix to storage
     auto p = SchurElemMats::schurL2Storage.emplace(
         &SchurElemMats::schurElemMats[SchurElemMats::maxIndexCounter++]);
 #ifndef NDEBUG
     if (!p.second) {
       SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY, "Failed to insert");
     }
 #endif // NDEBUG
  
     auto asmb = [&](auto &sm) {
       sm.resize(nb_rows, nb_cols, false);
       noalias(sm) = mat;
     };
  
     asmb((*p.first)->getMat());
     zero_rows_and_cols((*p.first)->getMat(), row_ind, col_ind);
  
     auto add_indices = [](auto &storage, auto &ind) {
       storage.resize(ind.size(), false);
       noalias(storage) = ind;
     };
  
     add_indices((*p.first)->getRowInd(), row_ind);
     add_indices((*p.first)->getColInd(), col_ind);
  
   } else {
     // entry (submatrix) already exists
  
     auto asmb = [&](auto &sm) {
       MoFEMFunctionBeginHot;
  
 #ifndef NDEBUG
       if (sm.size1() != nb_rows) {
         SETERRQ2(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
                  "Wrong mat or storage size %d != %d", sm.size1(), nb_rows);
       }
       if (sm.size2() != nb_cols) {
         SETERRQ2(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
                  "Wrong mat or storage size %d != %d", sm.size2(), nb_cols);
       }
 #endif // NDEBUG
  
       switch (iora) {
       case ADD_VALUES:
         sm += mat;
         break;
       case INSERT_VALUES:
         noalias(sm) = mat;
         break;
       default:
         SETERRQ(PETSC_COMM_SELF, MOFEM_NOT_IMPLEMENTED,
                 "Assembly type not implemented");
       }
       MoFEMFunctionReturnHot(0);
     };
  
     CHKERR asmb((*it)->getMat());
     CHKERR zero_rows_and_cols((*it)->getMat(), row_ind, col_ind);
  
     // no need to set indices
   }
  
 #ifndef NDEBUG
   PetscLogEventEnd(SchurEvents::MOFEM_EVENT_schurMatSetValues, 0, 0, 0, 0);
 #endif // NDEBUG
  
   MoFEMFunctionReturn(0);
 }

◆ getColInd()

auto& MoFEM::SchurElemMats::getColInd ( ) const

inline

Definition at line 184 of file Schur.cpp.

184 { return colIndices[iDX]; }

◆ getMat()

auto& MoFEM::SchurElemMats::getMat ( ) const

inline

Definition at line 182 of file Schur.cpp.

182 { return locMats[iDX]; }

◆ getRowInd()

auto& MoFEM::SchurElemMats::getRowInd ( ) const

inline

Definition at line 183 of file Schur.cpp.

183 { return rowIndices[iDX]; }

◆ MatSetValues()

MoFEMErrorCode MoFEM::SchurElemMats::MatSetValues	(	Mat	M,
		const EntitiesFieldData::EntData &	row_data,
		const EntitiesFieldData::EntData &	col_data,
		const MatrixDouble &	mat,
		InsertMode	iora
	)

static

Definition at line 2762 of file Schur.cpp.

                                                                       {
   MoFEMFunctionBegin;
   CHKERR assembleStorage(row_data, col_data, mat, iora);
   CHKERR SchurBackendMatSetValuesPtr::matSetValuesPtr(M, row_data, col_data,
                                                       mat, iora);
   MoFEMFunctionReturn(0);
 }

Friends And Related Function Documentation

◆ OpSchurAssembleBegin

friend struct OpSchurAssembleBegin

friend

Definition at line 197 of file Schur.cpp.

◆ OpSchurAssembleEndImpl

template<typename OP_SCHUR_ASSEMBLE_BASE >

friend struct OpSchurAssembleEndImpl

friend

Definition at line 199 of file Schur.cpp.

Member Data Documentation

◆ colIndices

boost::ptr_vector< VectorInt > MoFEM::SchurElemMats::colIndices

staticprotected

Definition at line 225 of file Schur.cpp.

◆ iDX

const size_t MoFEM::SchurElemMats::iDX

Definition at line 178 of file Schur.cpp.

◆ locMats

boost::ptr_vector< MatrixDouble > MoFEM::SchurElemMats::locMats

staticprotected

Definition at line 223 of file Schur.cpp.

◆ maxIndexCounter

size_t MoFEM::SchurElemMats::maxIndexCounter = 0

staticprotected

Definition at line 227 of file Schur.cpp.

◆ rowIndices

boost::ptr_vector< VectorInt > MoFEM::SchurElemMats::rowIndices

staticprotected

Definition at line 224 of file Schur.cpp.

◆ schurElemMats

boost::ptr_vector< SchurElemMats > MoFEM::SchurElemMats::schurElemMats

staticprotected

Definition at line 226 of file Schur.cpp.

◆ schurL2Storage

SchurElemMats::SchurElemStorage MoFEM::SchurElemMats::schurL2Storage

staticprotected

Definition at line 229 of file Schur.cpp.

◆ uidCol

UId MoFEM::SchurElemMats::uidCol

Definition at line 180 of file Schur.cpp.

◆ uidRow

UId MoFEM::SchurElemMats::uidRow

Definition at line 179 of file Schur.cpp.

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

src/finite_elements/impl/Schur.cpp

Classes

Public Member Functions

Static Public Member Functions

Public Attributes

Protected Types

Static Protected Member Functions

Static Protected Attributes

Friends

Detailed Description

Member Typedef Documentation

◆ SchurElemStorage

Constructor & Destructor Documentation

◆ SchurElemMats()

◆ ~SchurElemMats()

Member Function Documentation

◆ assembleStorage()

◆ getColInd()

◆ getMat()

◆ getRowInd()

◆ MatSetValues()

Friends And Related Function Documentation

◆ OpSchurAssembleBegin

◆ OpSchurAssembleEndImpl

Member Data Documentation

◆ colIndices

◆ iDX

◆ locMats

◆ maxIndexCounter

◆ rowIndices

◆ schurElemMats

◆ schurL2Storage

◆ uidCol

◆ uidRow