ForcesAndSourcesCore.hpp

Go to the documentation of this file.

/** \file ForcesAndSourcesCore.hpp
 * \brief Generic finite element implementation using operator pipelines
 *
 * This file provides the core framework for finite element computations in MoFEM
 * based on a pipeline of user data operators. Instead of monolithic finite element
 * implementations, ForcesAndSourcesCore treats finite elements as sequences of
 * modular operators that can be combined and reused.
 *
 * ## Core Concept: Operator Pipeline Architecture
 *
 * **Finite Element = Pipeline of Operators**
 * Rather than traditional finite element classes with hardcoded functionality,
 * MoFEM implements finite elements as pipelines of small, focused operators.
 * Each operator performs a specific task (calculate gradients, apply boundary
 * conditions, assemble matrices, etc.) and can be combined with others to
 * create complex finite element behaviors.
 *
 * **ForcesAndSourcesCore**: Generic base class that manages the operator pipeline
 * and provides the framework for element assembly. Users inherit from this class
 * and add specific operators to implement their particular problem.
 *
 * ## Operator Types and Usage:
 *
 * **OPROW**: Operators that work with test functions (rows of system matrix)
 * - Used for: Loading vectors, boundary conditions, source terms
 * - Example: Applying Neumann boundary conditions, body forces
 * - Operates during: Right-hand side vector assembly
 *
 * **OPCOL**: Operators that work with trial functions (columns of system matrix)
 * - Used for: Geometric operations, coordinate transformations
 * - Example: Calculating coordinates, setting up geometry
 * - Operates during: Matrix structure setup and geometric computations
 *
 * **OPROWCOL**: Operators that work with both test and trial functions
 * - Used for: Stiffness matrices, mass matrices, coupling terms
 * - Example: Linear elasticity stiffness, thermal conductivity
 * - Operates during: System matrix assembly (most common type)
 *
 * **OPSPACE**: Operators that work on the finite element space itself
 * - Used for: Basis function transformations, coordinate mappings
 * - Example: Piola transforms for Hdiv/Hcurl spaces, Jacobian operations
 * - Operates during: Pre-processing before matrix/vector assembly
 *
 *
 */
 
 
 
#ifndef __FORCES_AND_SOURCES_CORE__HPP__
#define __FORCES_AND_SOURCES_CORE__HPP__
 
using namespace boost::numeric;
 
namespace MoFEM {
 
/** \brief structure to get information from mofem into EntitiesFieldData
 * \ingroup mofem_forces_and_sources
 *
 */
struct ForcesAndSourcesCore : public FEMethod {
 
  Interface &mField;
 
  ForcesAndSourcesCore(Interface &m_field);
  typedef boost::function<int(int order_row, int order_col, int order_data)>
      RuleHookFun;
 
  typedef boost::function<MoFEMErrorCode(ForcesAndSourcesCore *fe_raw_ptr,
                                         int order_row, int order_col,
                                         int order_data)>
      GaussHookFun;
 
  /**
   * \brief Hook to get rule
   *
   * \todo check preferred format how works with gcc and clang,
   * see
   * <http://www.boost.org/doc/libs/1_64_0/doc/html/function/tutorial.html#idp247873024>
   */
  RuleHookFun getRuleHook;
 
  /**
   * @brief Set function to calculate integration rule
   *
   */
  GaussHookFun setRuleHook;
 
  /** \brief Data operator to do calculations at integration points.
  * \ingroup mofem_forces_and_sources
 
  Is inherited and implemented by user to do calculations. It can be used in
  many different ways but typically is used to integrate matrices (f.e.
  stiffness matrix) and the right hand vector (f.e. force vector).
 
  Note: It is assumed that operator is executed for symmetric problem. That
  means that is executed for not repeating entities on finite element. For
  example on triangle we have nodes, 3 edges and one face. Because of symmetry
  calculations are for: nodes-nodes, nodes-edge_0, nodes-edge_1, nodes-edge_2,
  nodes-face,
  edge_0-edge_0, edge_0-edge_1, edge_0-edge_2,
  edge_1-edge_1, edge_1-edge_2,
  edge_2-edge_2,
  edge_0-face, edge_1-face, edge_2-face,
  face - face
 
  In case of non symmetric problem in addition calculations of lower off
  diagonal terms. F.e. edge_1-edge_0, edge_2-edge_0, edge_2-edge_1,
 
  In that case class variable UserDataOperator::sYmm = false;
 
  Note: By default sYmm is set for symmetric problems
 
  */
  struct UserDataOperator;
 
  /** \brief Use to push back operator for row operator
 
   It can be used to calculate nodal forces or other quantities on the mesh.
 
   */
  boost::ptr_deque<UserDataOperator> &getOpPtrVector() { return opPtrVector; }
 
  /**
   * @brief Get the Entity Polynomial Base object
   *
   * @return boost::shared_ptr<BaseFunction>&&
   */
  auto &getElementPolynomialBase() { return elementPolynomialBasePtr; }
 
  /**
   * @brief Get the User Polynomial Base object
   *
   * @return boost::shared_ptr<BaseFunction>&
   */
  auto &getUserPolynomialBase() { return userPolynomialBasePtr; }
 
  /**
   * @brief Matrix of integration points
   *
   * Columns is equal to number of integration points, numver of rows
   * depends on dimension of finite element entity, for example for
   * tetrahedron rows are x,y,z,weight. Last row is integration weight.
   *
   * FIXME: that should be moved to private class data and acessed only by
   * member function
   */
  MatrixDouble gaussPts;
 
  virtual MoFEMErrorCode preProcess();
  virtual MoFEMErrorCode operator()();
  virtual MoFEMErrorCode postProcess();
 
public:
  /** \brief Get max order of approximation for data fields
 
  getMaxDataOrder() returns maximal order on entities, for
  all data on the element. So for example if finite element is triangle, and
  triangle base function have order 4 and on edges base function have order
  2, this function returns 4.
 
  If finite element has for example 2 or more approximated fields, for
  example Pressure (order 3) and displacement field (order 5), this function
  returns 5.
 
  */
  int getMaxDataOrder() const;
 
  /// \brief Get max order of approximation for field in rows
  int getMaxRowOrder() const;
 
  /// \brief Get max order of approximation for field in columns
  int getMaxColOrder() const;
 
  /**
   * @brief Get the entity data
   *
   * @param space
   * @param type
   * @param side
   * @return EntitiesFieldData::EntData&
   */
  auto &getEntData(const FieldSpace space, const EntityType type,
                   const int side) {
    return dataOnElement[space]->dataOnEntities[type][side];
  }
 
  /**
   * @brief Get data on entities and space
   * 
   * Entities data are stored by space, by entity type, and entity side.
   * 
   * @return std::array<boost::shared_ptr<EntitiesFieldData>, LASTSPACE>
   */
  auto &getDataOnElementBySpaceArray() { return dataOnElement; }
 
  /**
   * @brief Get derived data on entities and space
   *
   * Entities data are stored by space, by entity type, and entity side. Derived
   * data is used to store data on columns, so it shares information about shape
   * functions wih rows.
   *
   * @return std::array<boost::shared_ptr<EntitiesFieldData>, LASTSPACE>
   */
  auto &getDerivedDataOnElementBySpaceArray() { return derivedDataOnElement; }
 
protected:
  /**
   * \brief get sense (orientation) of entity
   * @param  type type of entity
   * @param  data entity data
   * @return      error code
   */
  MoFEMErrorCode getEntitySense(
      const EntityType type,
      boost::ptr_vector<EntitiesFieldData::EntData> &data) const;
 
  /**
   * @brief Get the entity data order
   *
   * @param type
   * @param space
   * @param data
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode getEntityDataOrder(
      const EntityType type, const FieldSpace space,
      boost::ptr_vector<EntitiesFieldData::EntData> &data) const;
 
  /**
   * @brief Get the entity sense (orientation)
   *
   * @tparam type
   * @param data
   * @return MoFEMErrorCode
   */
  template <EntityType type>
  inline MoFEMErrorCode getEntitySense(EntitiesFieldData &data) const {
    return getEntitySense(type, data.dataOnEntities[type]);
  }
 
  /**
   * @brief Get the entity data order for given space
   *
   * @tparam type
   * @param data
   * @param space
   * @return MoFEMErrorCode
   */
  template <EntityType type>
  inline MoFEMErrorCode getEntityDataOrder(EntitiesFieldData &data,
                                           const FieldSpace space) const {
    return getEntityDataOrder(type, space, data.dataOnEntities[type]);
  }
 
  /** \name Indices */
 
  /**@{*/
 
  /// \brief get node indices
  template <typename EXTRACTOR>
  MoFEMErrorCode
  getNodesIndices(const int bit_number,
                  FieldEntity_vector_view &ents_field, VectorInt &nodes_indices,
                  VectorInt &local_nodes_indices, EXTRACTOR &&extractor) const;
 
  /// \brief get row node indices from FENumeredDofEntity_multiIndex
  MoFEMErrorCode getRowNodesIndices(EntitiesFieldData &data,
                                    const int bit_number) const;
 
  /// \brief get col node indices from FENumeredDofEntity_multiIndex
  MoFEMErrorCode getColNodesIndices(EntitiesFieldData &data,
                                    const int bit_number) const;
 
  template <typename EXTRACTOR>
  MoFEMErrorCode getEntityIndices(EntitiesFieldData &data, const int bit_number,
                                  FieldEntity_vector_view &ents_field,
                                  const EntityType type_lo,
                                  const EntityType type_hi,
                                  EXTRACTOR &&extractor) const;
 
  MoFEMErrorCode
  getEntityRowIndices(EntitiesFieldData &data, const int bit_number,
                      const EntityType type_lo = MBVERTEX,
                      const EntityType type_hi = MBPOLYHEDRON) const;
 
  MoFEMErrorCode
  getEntityColIndices(EntitiesFieldData &data, const int bit_number,
                      const EntityType type_lo = MBVERTEX,
                      const EntityType type_hi = MBPOLYHEDRON) const;
 
 
  /// \brief get col NoField indices
  MoFEMErrorCode getNoFieldRowIndices(EntitiesFieldData &data,
                                      const int bit_number) const;
 
  /// \brief get col NoField indices
  MoFEMErrorCode getNoFieldColIndices(EntitiesFieldData &data,
                                      const int bit_number) const;
 
  /**@}*/
 
  /** \name Data */
 
  /**@{*/
 
  /** Get bit ref level in  entities, and set it to data
   */
  MoFEMErrorCode getBitRefLevelOnData();
 
  /**
   * \brief Get data on nodes
   * @param  data       Data structure
   * @param  field_name Field name
   * @return            Error code
   */
  MoFEMErrorCode getNodesFieldData(EntitiesFieldData &data,
                                   const int bit_number) const;
 
  MoFEMErrorCode
  getEntityFieldData(EntitiesFieldData &data, const int bit_number,
                     const EntityType type_lo = MBVERTEX,
                     const EntityType type_hi = MBPOLYHEDRON) const;
 
  /**
   * @brief Get field data on entities where no field is defined
   *
   * This is not standards, since op_data, dataOnElement[NOFIELD], has to be set
   * for rows and columns for each NOFIELD, also size of sides is not determined.
   *
   * @param data @param bit_number @return MoFEMErrorCode 
   */
  template <typename EXTRACTOR>
  MoFEMErrorCode getNoFieldEntityFieldData(EntitiesFieldData &data,
                                           const int bit_number,
                                           EXTRACTOR &&extractor) const;
 
  /**@}*/
 
  /// \brief Get nodes on faces
  MoFEMErrorCode getFaceNodes(EntitiesFieldData &data) const;
 
  /// \brief Get field approximation space and base on entities
  MoFEMErrorCode
  getSpacesAndBaseOnEntities(EntitiesFieldData &data) const;
 
  /** \name Data form NumeredDofEntity_multiIndex */
 
  /**@{*/
 
  /// \brief get indices of nodal indices which are declared for problem but
  /// not this particular element
  MoFEMErrorCode getProblemNodesIndices(const std::string &field_name,
                                        const NumeredDofEntity_multiIndex &dofs,
                                        VectorInt &nodes_indices) const;
 
  /// \brief get indices by type (generic function) which are declared for
  /// problem but not this particular element
  MoFEMErrorCode getProblemTypeIndices(const std::string &field_name,
                                       const NumeredDofEntity_multiIndex &dofs,
                                       EntityType type, int side_number,
                                       VectorInt &indices) const;
 
  MoFEMErrorCode getProblemNodesRowIndices(const std::string &field_name,
                                           VectorInt &nodes_indices) const;
  MoFEMErrorCode getProblemTypeRowIndices(const std::string &field_name,
                                          EntityType type, int side_number,
                                          VectorInt &indices) const;
  MoFEMErrorCode getProblemNodesColIndices(const std::string &field_name,
                                           VectorInt &nodes_indices) const;
  MoFEMErrorCode getProblemTypeColIndices(const std::string &field_name,
                                          EntityType type, int side_number,
                                          VectorInt &indices) const;
 
  /**@}*/
 
  /**
   * \brief another variant of getRule
   * @param  order_row  order of base function on row
   * @param  order_col  order of base function on columns
   * @param  order_data order of base function approximating data
   * @return            integration rule
   *
   * This function is overloaded by the user. The integration rule
   * is set such that specific operator implemented by the user is
   * integrated accurately. For example if user implement bilinear operator
   * \f[
   * b(u,v) =
   * \int_\mathcal{T}
   * \frac{\partial u_i}{\partial x_j}\frac{\partial v_i}{\partial x_j}
   * \textrm{d}\mathcal{T}
   * \f]
   * then if \f$u\f$ and \f$v\f$ are polynomial of given \em order, then
   * exact integral would be \code int getRule(int order) { return
   * 2*(order-1); }; \endcode
   *
   * The integration points and weights are set appropriately for given
   * entity type and integration rule from \ref quad.c
   *
   * Method \ref ForcesAndSourcesCore::getRule takes at argument takes
   * maximal polynomial order set on the element on all fields defined on
   * the element. If a user likes to have more control, another variant of
   * this function can be called which distinguishing between field orders
   * on rows, columns and data, the i.e. first argument of a bilinear form,
   * the second argument of bilinear form and field coefficients on the
   * element.
   *
   * \note If user set rule to -1 or any other negative integer, then method
   * \ref ForcesAndSourcesCore::setGaussPts is called. In that method user
   * can implement own (specific) integration method.
   *
   * \bug this function should be const
   */
  virtual int getRule(int order_row, int order_col, int order_data);
 
  /** \brief set user specific integration rule
 
    This function allows for user defined integration rule. The key is to
    called matrix gaussPts, which is used by other MoFEM procedures. Matrix
    has number of rows equal to problem dimension plus one, where last index
    is used to store weight values. %Number of columns is equal to number of
    integration points.
 
    \note This function is called if method \ref
    ForcesAndSourcesCore::getRule is returning integer -1 or any other
    negative integer.
 
    User sets
    \code
    MatrixDouble gaussPts;
    \endcode
    where
    \code
    gaussPts.resize(dim+1,nb_gauss_pts);
    \endcode
    number rows represents local coordinates of integration points
    in reference element, where last index in row is for integration weight.
 
    */
  virtual MoFEMErrorCode setGaussPts(int order_row, int order_col,
                                     int order_data);
 
  /**
   * \brief Calculate base functions
   * @return Error code
   */
  MoFEMErrorCode
  calHierarchicalBaseFunctionsOnElement(const FieldApproximationBase b);
 
  /**
   * \brief Calculate base functions
   * @return Error code
   */
  MoFEMErrorCode calHierarchicalBaseFunctionsOnElement();
 
  /**
   * @brief Calculate Bernstein-Bezier base
   *
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode calBernsteinBezierBaseFunctionsOnElement();
 
  /**
   * @brief Create a entity data on element object
   *
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode createDataOnElement(EntityType type);
 
  /**
   * @brief Iterate user data operators
   *
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode loopOverOperators();
 
  /**@{*/
 
  /** \name Deprecated (do not use) */
 
  /** \deprecated Use getRule(int row_order, int col_order, int data order)
   */
  virtual int getRule(int order);
 
  /** \deprecated setGaussPts(int row_order, int col_order, int data order);
   */
  virtual MoFEMErrorCode setGaussPts(int order);
 
  /**@}*/
 
  /**
   * @brief Entity data on element entity rows fields
   *
   */
  const std::array<boost::shared_ptr<EntitiesFieldData>, LASTSPACE>
      dataOnElement;
 
  /**
   * @brief Entity data on element entity columns fields
   *
  */
  const std::array<boost::shared_ptr<EntitiesFieldData>, LASTSPACE>
      derivedDataOnElement;
 
  EntitiesFieldData &dataNoField;
  EntitiesFieldData &dataH1;
  EntitiesFieldData &dataHcurl;
  EntitiesFieldData &dataHdiv;
  EntitiesFieldData &dataL2;
 
  /**
   * @brief Vector of finite element users data operators
   *
   */
  boost::ptr_deque<UserDataOperator> opPtrVector;
 
  friend class UserDataOperator;
 
protected:
  /**
   * @brief Last evaluated type of element entity
   *
   */
  EntityType lastEvaluatedElementEntityType;
 
private:
  /**
   * @brief Pointer to entity polynomial base
   *
   */
  boost::shared_ptr<BaseFunction> elementPolynomialBasePtr;
 
  /**
   * @brief Pointer to user polynomial base
   */
  boost::shared_ptr<BaseFunction> userPolynomialBasePtr;
 
  /**
   * @brief Element to integrate on the sides
   *
   */
  ForcesAndSourcesCore *sidePtrFE;
 
  /**
   * @brief Set the pointer to face element on the side
   *
   * \note Function is is used by face element, while it iterates over
   * elements on the side
   *
   * @param side_fe_ptr
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode setSideFEPtr(const ForcesAndSourcesCore *side_fe_ptr);
 
  /**u
   * @brief Element to integrate parent or child
   *
   */
  ForcesAndSourcesCore *refinePtrFE;
 
  /**
   * @brief Set the pointer to face element refined
   *
   * @param refine_fe_ptr
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode setRefineFEPtr(const ForcesAndSourcesCore *refine_fe_ptr);
 
  friend class VolumeElementForcesAndSourcesCoreOnSide;
  friend class FaceElementForcesAndSourcesCoreOnSide;
  friend class FaceElementForcesAndSourcesCoreOnChildParent;
  friend class EdgeElementForcesAndSourcesCoreOnChildParent;
  friend class VolumeElementForcesAndSourcesCoreOnContactPrismSide;
 
  template <int DIM> friend struct OpCopyGeomDataToE;
  template <typename E> friend struct OpBrokenLoopSide; 
 
protected:
  MatrixDouble coordsAtGaussPts; ///< coordinated at gauss points
  double elementMeasure; ///< Depending on dimension of elements, stores length,
                         ///< area or volume of element.
};
 
struct ForcesAndSourcesCore::UserDataOperator : public DataOperator {
 
  /**
   * \brief Controls loop over entities on element
   *
   * - OPROW is used if row vector is assembled
   * - OPCOL is usually used if column vector is assembled
   * - OPROWCOL is usually used for assemble matrices.
   * - OPSPACE no field is defined for such operator. Is usually used to modify
   * base
   * 
   *
   * For typical problem like Bubnov-Galerkin OPROW and OPCOL are the same. In
   * more general case for example for non-square matrices columns and rows
   * could have different numeration and/or different set of fields.
   *
   */
  enum OpType {
    OPROW = 1 << 0,    ///< operator doWork function is executed on FE rows
    OPCOL = 1 << 1,    ///< operator doWork function is executed on FE columns
    OPROWCOL = 1 << 2, ///< operator doWork is executed on FE rows &columns
    OPSPACE = 1 << 3,  ///< operator do Work is execute on space data
    OPLAST = 1 << 3    ///< @deprecated would be removed
  };
 
  static const char *const OpTypeNames[];
 
  char opType;
  std::string rowFieldName;
  std::string colFieldName;
  FieldSpace sPace;
 
  /**
   * @brief Constructor for operators working on finite element spaces
   *
   * This constructor is used when operators modify basis functions on
   * approximation spaces. The operator runs for all data on the specified
   * space without access to specific field data. Commonly used for:
   * - Coordinate transformations (Piola transforms)
   * - Integration weight modifications
   * - Basis function scaling or rotation
   *
   * @param space Field space to operate on (H1, HDIV, HCURL, L2, etc.)
   * @param type Operator type (OPSPACE, OPROW, OPCOL, OPROWCOL)
   * @param symm Whether the operator maintains matrix symmetry (default: true)
   */
  UserDataOperator(const FieldSpace space, const char type = OPSPACE,
                   const bool symm = true);
 
  /**
   * @brief Constructor for operators working on a single field
   *
   * This constructor creates operators that work with data from a specific
   * field. Used for operations like:
   * - Calculating field values at integration points
   * - Computing field gradients or derivatives
   * - Applying boundary conditions to specific fields
   * - Post-processing field data
   *
   * @param field_name Name of the field to operate on
   * @param type Operator type:
   *             - OPROW: Works with test functions (RHS assembly)
   *             - OPCOL: Works with trial functions (geometry setup)
   *             - OPROWCOL: Works with both (matrix assembly)
   * @param symm Whether the operator maintains matrix symmetry (default: true)
   */
  UserDataOperator(const std::string field_name, const char type,
                   const bool symm = true);
 
  /**
   * @brief Constructor for operators working on two fields (bilinear forms)
   *
   * This constructor creates operators that couple two different fields,
   * essential for:
   * - Mixed formulations (pressure-velocity coupling)
   * - Multi-physics problems (thermal-mechanical coupling)
   * - Contact mechanics (displacement-Lagrange multiplier)
   * - Interface problems between different field types
   *
   * @param row_field_name Name of the field for test functions (rows)
   * @param col_field_name Name of the field for trial functions (columns)
   * @param type Operator type (typically OPROWCOL for coupling)
   * @param symm Whether the operator maintains matrix symmetry (default: true)
   *             Note: Coupling between different fields is often non-symmetric
   */
  UserDataOperator(const std::string row_field_name,
                   const std::string col_field_name, const char type,
                   const bool symm = true);
 
  /** \brief Return raw pointer to NumeredEntFiniteElement
   */
  inline boost::shared_ptr<const NumeredEntFiniteElement>
  getNumeredEntFiniteElementPtr() const;
 
  /**
   * \brief Return finite element entity handle
   * @return Finite element entity handle
   */
  inline EntityHandle getFEEntityHandle() const;
 
  /**
   * @brief Get dimension of finite element
   *
   * @return int
   */
  inline int getFEDim() const;
 
  /**
   * @brief Get dimension of finite element
   *
   * @return int
   */
  inline EntityType getFEType() const;
 
  /**
   * @brief Get the side number pointer
   *
   * \note For vertex is expectation. Side basses in argument of function doWork
   * is zero. For other entity types side can be used as argument of this
   * function.
   *
   * @param side_number
   * @param type
   * @return boost::weak_ptr<SideNumber>
   */
  inline boost::weak_ptr<SideNumber> getSideNumberPtr(const int side_number,
                                                      const EntityType type);
 
  /**
   * @brief Get the side entity
   *
   * \note For vertex is expectation. Side basses in argument of function
   * doWork is zero. For other entity types side can be used as argument of
   * this function.
   *
   * \code
   * MoFEMErrorCode doWork(int side, EntityType type,
   *                     EntitiesFieldData::EntData &data) {
   *  MoFEMFunctionBegin;
   *
   *  if (type == MBVERTEX) {
   *    for (int n = 0; n != number_of_nodes; ++n)
   *      EntityHandle ent = getSideEntity(n, type);
   *
   *      // Do somthing
   *
   *  } else {
   *    EntityHandle ent = getSideEntity(side, type);
   *
   *    // Do somthing
   *
   *  }
   *
   *  MoFEMFunctionReturn(0);
   * }
   * \endcode
   *
   * @param side_number
   * @param type
   */
  inline EntityHandle getSideEntity(const int side_number,
                                    const EntityType type);
 
  /**
   * @brief Get the number of nodes on finite element
   *
   * @return int
   */
  inline int getNumberOfNodesOnElement() const;
 
  /** \brief Get row indices
 
  Field could be or not declared for this element but is declared for
  problem
 
  \param field_name
  \param type entity type
  \param side side number, any number if type is MBVERTEX
  \return indices
 
  NOTE: Using those indices to assemble matrix will result in error if new
  non-zero values need to be created.
 
  */
  MoFEMErrorCode getProblemRowIndices(const std::string filed_name,
                                      const EntityType type, const int side,
                                      VectorInt &indices) const;
 
  /** \brief Get col indices
 
  Field could be or not declared for this element but is declared for
  problem
 
  \param field_name
  \param type entity type
  \param side side number, any number if type is MBVERTEX
  \return indices
 
  NOTE: Using those indices to assemble matrix will result in error if new
  non-zero values need to be created.
 
  */
  MoFEMErrorCode getProblemColIndices(const std::string filed_name,
                                      const EntityType type, const int side,
                                      VectorInt &indices) const;
 
  /** \brief Return raw pointer to Finite Element Method object
   */
  inline const FEMethod *getFEMethod() const;
 
  /**
   * \brief Get operator types
   * @return Return operator type
   */
  inline int getOpType() const;
 
  /**
   * \brief Set operator type
   * @param Operator type
   */
  inline void setOpType(const OpType type);
 
  /**
   * \brief Add operator type
   */
  inline void addOpType(const OpType type);
 
  /**
   * \brief get number of finite element in the loop
   * @return number of finite element
   */
  inline int getNinTheLoop() const;
 
  /**
   * \brief get size of elements in the loop
   * @return loop size
   */
  inline int getLoopSize() const;
 
  /** \brief Get name of the element
   */
  inline std::string getFEName() const;
 
  /** \name Accessing KSP */
 
  /**@{*/
 
  inline const PetscData::Switches &getDataCtx() const;
 
  inline const KspMethod::KSPContext getKSPCtx() const;
 
  inline const SnesMethod::SNESContext getSNESCtx() const;
 
  inline const TSMethod::TSContext getTSCtx() const;
 
  /**@}*/
 
  /**@{*/
 
  inline Vec getKSPf() const;
 
  inline Mat getKSPA() const;
 
  inline Mat getKSPB() const;
 
  /**@}*/
 
  /** \name Accessing SNES */
 
  /**@{*/
 
  inline Vec getSNESf() const;
 
  inline Vec getSNESx() const;
 
  inline Mat getSNESA() const;
 
  inline Mat getSNESB() const;
 
  /**@}*/
 
  /** \name Accessing TS */
 
  /**@{*/
 
  inline Vec getTSu() const;
 
  inline Vec getTSu_t() const;
 
  inline Vec getTSu_tt() const;
 
  inline Vec getTSf() const;
 
  inline Mat getTSA() const;
 
  inline Mat getTSB() const;
 
  inline int getTSstep() const;
 
  inline double getTStime() const;
 
  inline double getTStimeStep() const;
 
  inline double getTSa() const;
 
  inline double getTSaa() const;
 
  /**@}*/
 
  /**@{*/
 
    /**@}*/
 
  /** \name Base functions and integration points */
 
  /**@{*/
 
  /** \brief matrix of integration (Gauss) points for Volume Element
   *
   * For triangle: columns 0,1 are x,y coordinates respectively and column
   * 2 is a weight value for example getGaussPts()(1,13) returns y
   * coordinate of 13th Gauss point on particular volume element
   *
   * For tetrahedron: columns 0,1,2 are x,y,z coordinates respectively and
   * column 3 is a weight value for example getGaussPts()(1,13) returns y
   * coordinate of 13th Gauss point on particular volume element
   *
   */
  inline MatrixDouble &getGaussPts();
 
  /**
   * @brief Get integration weights
   *
   * \code
   * auto t_w = getFTensor0IntegrationWeight();
   * for(int gg = 0; gg!=getGaussPts.size2(); ++gg) {
   *  // integrate something
   *  ++t_w;
   * }
   * \endcode
   *
   * @return FTensor::Tensor0<FTensor::PackPtr<double *, 1>>
   */
  inline auto getFTensor0IntegrationWeight();
 
  /**@}*/
 
  /** \name Coordinates and access to internal data */
 
  /**@{*/
 
  /** \brief Gauss points and weight, matrix (nb. of points x 3)
 
    Column 0-2 integration points coordinate x, y and z, respectively. At rows
    are integration points.
 
  */
  inline MatrixDouble &getCoordsAtGaussPts();
 
  /**
   * \brief Get coordinates at integration points assuming linear geometry
   *
   * \code
   * auto t_coords = getFTensor1CoordsAtGaussPts();
   * for(int gg = 0;gg!=nb_int_ptrs;gg++) {
   *   // do something
   *   ++t_coords;
   * }
   * \endcode
   *
   */
  inline auto getFTensor1CoordsAtGaussPts();
 
  /**@}*/
 
  /**@{*/
 
  /** \name Measures (area, volume, length, etc.) */
 
  /**
   * \brief get measure of element
   * @return volume
   */
  inline double getMeasure() const;
 
  /**
   * \brief get measure of element
   * @return volume
   */
  inline double &getMeasure();
 
  /**}*/
 
  /**@{*/
 
  /** \name Loops */
 
  using AdjCache =
      std::map<EntityHandle,
               std::vector<boost::weak_ptr<NumeredEntFiniteElement>>>;
 
  /**
   * @brief User calls this function to loop over elements on the side of
   * face. This function calls finite element with its operator to do
   * calculations.
   *
   * @param fe_name       name of the side element
   * @param side_fe       pointer to the side element instance
   * @param dim           dimension the of side element
   * @param ent_for_side  entity handle for which adjacent volume or face will
   * be accessed
   * @param verb
   * @param sev
   * @param adj_cache
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode loopSide(const string &fe_name, ForcesAndSourcesCore *side_fe,
                          const size_t dim, const EntityHandle ent_for_side = 0,
                          boost::shared_ptr<Range> fe_range = nullptr,
                          const int verb = QUIET,
                          const LogManager::SeverityLevel sev = Sev::noisy,
                          AdjCache *adj_cache = nullptr
 
  );
 
  /**
   * @brief  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.
   *
   * @param fe_name
   * @param this_fe
   * @param verb
   * @param sev
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode loopThis(const string &fe_name,
                          ForcesAndSourcesCore *this_fe,
                          const int verb = QUIET,
                          const LogManager::SeverityLevel sev = Sev::noisy);
 
  /**
   * @brief  User calls this function to loop over parent elements. This function
   * calls finite element with its operator to do calculations.
   *
   * @param fe_name
   * @param parent_fe
   * @param verb
   * @param sev
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode loopParent(const string &fe_name,
                            ForcesAndSourcesCore *parent_fe,
                            const int verb = QUIET,
                            const LogManager::SeverityLevel sev = Sev::noisy);
 
  /**
   * @brief  User calls this function to loop over parent elements. This function
   * calls finite element with its operator to do calculations.
   *
   * @param fe_name
   * @param child_fe
   * @param verb
   * @param sev
   * @return MoFEMErrorCode
   */
  MoFEMErrorCode loopChildren(const string &fe_name,
                              ForcesAndSourcesCore *child_fe,
                              const int verb = QUIET,
                              const LogManager::SeverityLevel sev = Sev::noisy);
 
  /**
   * @brief Iterate over range of elements
   *
   * @param fe_name
   * @param range_fe
   * @param fe_range
   * @param verb
   * @param sev
   * @return * MoFEMErrorCode
   */
  MoFEMErrorCode loopRange(const string &fe_name,
                           ForcesAndSourcesCore *range_fe,
                           boost::shared_ptr<Range> fe_range,
                           const int verb = QUIET,
                           const LogManager::SeverityLevel sev = Sev::noisy);
 
  /**@}*/
 
  inline ForcesAndSourcesCore *getPtrFE() const;
 
  inline ForcesAndSourcesCore *getSidePtrFE() const;
 
  inline ForcesAndSourcesCore *getRefinePtrFE() const;
 
  /**@{*/
 
  /** \name Database */
 
  inline MoFEM::Interface &getMField() { return ptrFE->mField; }
  inline moab::Interface &getMoab() { return getMField().get_moab(); }
 
  /**@}*/
 
  /**@{ */
 
  /** \name Pipelines */
 
  virtual boost::weak_ptr<ForcesAndSourcesCore> getSubPipelinePtr() const {
    return boost::weak_ptr<ForcesAndSourcesCore>();
  };
 
  /**@} */
 
protected:
  ForcesAndSourcesCore *ptrFE;
 
  virtual MoFEMErrorCode setPtrFE(ForcesAndSourcesCore *ptr);
 
private:
  friend class ForcesAndSourcesCore;
  friend class EdgeElementForcesAndSourcesCore;
  friend class FaceElementForcesAndSourcesCore;
  friend class ContactPrismElementForcesAndSourcesCore;
};
 
/// \deprecated Used ForcesAndSourcesCore instead
DEPRECATED typedef ForcesAndSourcesCore ForcesAndSurcesCore;
 
boost::shared_ptr<const NumeredEntFiniteElement>
ForcesAndSourcesCore::UserDataOperator::getNumeredEntFiniteElementPtr() const {
  return ptrFE->numeredEntFiniteElementPtr;
};
 
EntityHandle ForcesAndSourcesCore::UserDataOperator::getFEEntityHandle() const {
  return getNumeredEntFiniteElementPtr()->getEnt();
}
 
int ForcesAndSourcesCore::UserDataOperator::getFEDim() const {
  return dimension_from_handle(getFEEntityHandle());
};
 
EntityType ForcesAndSourcesCore::UserDataOperator::getFEType() const {
  return type_from_handle(getFEEntityHandle());
};
 
boost::weak_ptr<SideNumber>
ForcesAndSourcesCore::UserDataOperator::getSideNumberPtr(
    const int side_number, const EntityType type) {
  auto &side_table_by_side_and_type =
      ptrFE->numeredEntFiniteElementPtr->getSideNumberTable().get<1>();
  auto side_it =
      side_table_by_side_and_type.find(boost::make_tuple(type, side_number));
  if (side_it != side_table_by_side_and_type.end())
    return *side_it;
  else
    return boost::weak_ptr<SideNumber>();
}
 
EntityHandle
ForcesAndSourcesCore::UserDataOperator::getSideEntity(const int side_number,
                                                      const EntityType type) {
  if (auto side_ptr = getSideNumberPtr(side_number, type).lock())
    return side_ptr->ent;
  else
    return 0;
}
 
int ForcesAndSourcesCore::UserDataOperator::getNumberOfNodesOnElement() const {
  return ptrFE->getNumberOfNodes();
}
 
const FEMethod *ForcesAndSourcesCore::UserDataOperator::getFEMethod() const {
  return ptrFE;
}
 
int ForcesAndSourcesCore::UserDataOperator::getOpType() const { return opType; }
 
void ForcesAndSourcesCore::UserDataOperator::setOpType(const OpType type) {
  opType = type;
}
 
void ForcesAndSourcesCore::UserDataOperator::addOpType(const OpType type) {
  opType |= type;
}
 
int ForcesAndSourcesCore::UserDataOperator::getNinTheLoop() const {
  return getFEMethod()->getNinTheLoop();
}
 
int ForcesAndSourcesCore::UserDataOperator::getLoopSize() const {
  return getFEMethod()->getLoopSize();
}
 
std::string ForcesAndSourcesCore::UserDataOperator::getFEName() const {
  return getFEMethod()->getFEName();
}
 
const PetscData::Switches &
ForcesAndSourcesCore::UserDataOperator::getDataCtx() const {
  return getFEMethod()->data_ctx;
}
 
const KspMethod::KSPContext
ForcesAndSourcesCore::UserDataOperator::getKSPCtx() const {
  return getFEMethod()->ksp_ctx;
}
 
const SnesMethod::SNESContext
ForcesAndSourcesCore::UserDataOperator::getSNESCtx() const {
  return getFEMethod()->snes_ctx;
}
 
const TSMethod::TSContext
ForcesAndSourcesCore::UserDataOperator::getTSCtx() const {
  return getFEMethod()->ts_ctx;
}
 
Vec ForcesAndSourcesCore::UserDataOperator::getKSPf() const {
#ifndef NDEBUG
  if (getFEMethod()->ksp_f == PETSC_NULLPTR)
    THROW_MESSAGE("KSP not set F vector");
#endif
  return getFEMethod()->ksp_f;
}
 
Mat ForcesAndSourcesCore::UserDataOperator::getKSPA() const {
#ifndef NDEBUG
  if (getFEMethod()->ksp_A == PETSC_NULLPTR)
    THROW_MESSAGE("KSP not set A vector");
#endif
  return getFEMethod()->ksp_A;
}
 
Mat ForcesAndSourcesCore::UserDataOperator::getKSPB() const {
#ifndef NDEBUG
  if (getFEMethod()->ksp_B == PETSC_NULLPTR)
    THROW_MESSAGE("KSP not set B vector");
#endif
  return getFEMethod()->ksp_B;
}
 
Vec ForcesAndSourcesCore::UserDataOperator::getSNESf() const {
#ifndef NDEBUG
  if (getFEMethod()->snes_f == PETSC_NULLPTR)
    THROW_MESSAGE("SNES not set F vector");
#endif
  return getFEMethod()->snes_f;
}
 
Vec ForcesAndSourcesCore::UserDataOperator::getSNESx() const {
#ifndef NDEBUG
  if (getFEMethod()->snes_x == PETSC_NULLPTR)
    THROW_MESSAGE("SNESnot set X vector");
#endif
  return getFEMethod()->snes_x;
}
 
Mat ForcesAndSourcesCore::UserDataOperator::getSNESA() const {
#ifndef NDEBUG
  if (getFEMethod()->snes_A == PETSC_NULLPTR)
    THROW_MESSAGE("SNES not set A vector");
#endif
  return getFEMethod()->snes_A;
}
 
Mat ForcesAndSourcesCore::UserDataOperator::getSNESB() const {
#ifndef NDEBUG
  if (getFEMethod()->snes_B == PETSC_NULLPTR)
    THROW_MESSAGE("SNES not set A matrix");
#endif
  return getFEMethod()->snes_B;
}
 
Vec ForcesAndSourcesCore::UserDataOperator::getTSu() const {
#ifndef NDEBUG
  if (getFEMethod()->ts_u == PETSC_NULLPTR)
    THROW_MESSAGE("TS not set U vector");
#endif
  return getFEMethod()->ts_u;
}
 
Vec ForcesAndSourcesCore::UserDataOperator::getTSu_t() const {
#ifndef NDEBUG
  if (getFEMethod()->ts_u_t == PETSC_NULLPTR)
    THROW_MESSAGE("TS not set U_t vector");
#endif
  return getFEMethod()->ts_u_t;
}
 
Vec ForcesAndSourcesCore::UserDataOperator::getTSu_tt() const {
#ifndef NDEBUG
  if (getFEMethod()->ts_u_tt == PETSC_NULLPTR)
    THROW_MESSAGE("TS not set U_tt vector");
#endif
  return getFEMethod()->ts_u_tt;
}
 
Vec ForcesAndSourcesCore::UserDataOperator::getTSf() const {
#ifndef NDEBUG
  if (getFEMethod()->ts_F == PETSC_NULLPTR)
    THROW_MESSAGE("TS not set F vector");
#endif
  return getFEMethod()->ts_F;
}
 
Mat ForcesAndSourcesCore::UserDataOperator::getTSA() const {
#ifndef NDEBUG
  if (getFEMethod()->ts_A == PETSC_NULLPTR)
    THROW_MESSAGE("TS not set A matrix");
#endif
  return getFEMethod()->ts_A;
}
 
Mat ForcesAndSourcesCore::UserDataOperator::getTSB() const {
#ifndef NDEBUG
  if (getFEMethod()->ts_B == PETSC_NULLPTR)
    THROW_MESSAGE("TS not set B matrix");
#endif
  return getFEMethod()->ts_B;
}
 
int ForcesAndSourcesCore::UserDataOperator::getTSstep() const {
#ifndef NDEBUG
  if ((getFEMethod()->data_ctx & PetscData::PetscData::CtxSetTime).none())
    THROW_MESSAGE("TS not set time");
#endif
  return getFEMethod()->ts_step;
}
 
double ForcesAndSourcesCore::UserDataOperator::getTStime() const {
#ifndef NDEBUG
  if ((getFEMethod()->data_ctx & PetscData::PetscData::CtxSetTime).none())
    THROW_MESSAGE("TS not set time");
#endif
  return getFEMethod()->ts_t;
}
 
double ForcesAndSourcesCore::UserDataOperator::getTStimeStep() const {
#ifndef NDEBUG
  if ((getFEMethod()->data_ctx & PetscData::PetscData::CtxSetTime).none())
    THROW_MESSAGE("TS not set time");
#endif
  return getFEMethod()->ts_dt;
}
 
double ForcesAndSourcesCore::UserDataOperator::getTSa() const {
#ifndef NDEBUG
  if ((getFEMethod()->data_ctx & (PetscData::CtxSetA | PetscData::CtxSetB))
          .none() ||
      (getFEMethod()->data_ctx & (PetscData::CtxSetX_T)).none())
    THROW_MESSAGE("TS not set B matrix");
#endif
  return getFEMethod()->ts_a;
}
 
double ForcesAndSourcesCore::UserDataOperator::getTSaa() const {
#ifndef NDEBUG
  if ((getFEMethod()->data_ctx & (PetscData::CtxSetA | PetscData::CtxSetB))
          .none() ||
      (getFEMethod()->data_ctx & (PetscData::CtxSetX_TT)).none())
    THROW_MESSAGE("TS not set B matrix");
#endif
  return getFEMethod()->ts_aa;
}
 
MatrixDouble &ForcesAndSourcesCore::UserDataOperator::getGaussPts() {
  return static_cast<ForcesAndSourcesCore *>(ptrFE)->gaussPts;
}
 
auto ForcesAndSourcesCore::UserDataOperator::getFTensor0IntegrationWeight() {
  return FTensor::Tensor0<FTensor::PackPtr<double *, 1>>(
      &(getGaussPts()(getGaussPts().size1() - 1, 0)));
}
 
ForcesAndSourcesCore *ForcesAndSourcesCore::UserDataOperator::getPtrFE() const {
  return ptrFE;
}
 
ForcesAndSourcesCore *
ForcesAndSourcesCore::UserDataOperator::getSidePtrFE() const {
  return ptrFE->sidePtrFE;
}
 
ForcesAndSourcesCore *
ForcesAndSourcesCore::UserDataOperator::getRefinePtrFE() const {
  return ptrFE->refinePtrFE;
}
 
MatrixDouble &ForcesAndSourcesCore::UserDataOperator::getCoordsAtGaussPts() {
  return static_cast<ForcesAndSourcesCore *>(ptrFE)->coordsAtGaussPts;
}
 
auto ForcesAndSourcesCore::UserDataOperator::getFTensor1CoordsAtGaussPts() {
  return FTensor::Tensor1<FTensor::PackPtr<double *, 3>, 3>(
      &getCoordsAtGaussPts()(0, 0), &getCoordsAtGaussPts()(0, 1),
      &getCoordsAtGaussPts()(0, 2));
}
 
double ForcesAndSourcesCore::UserDataOperator::getMeasure() const {
  return static_cast<ForcesAndSourcesCore *>(ptrFE)->elementMeasure;
}
 
double &ForcesAndSourcesCore::UserDataOperator::getMeasure() {
  return static_cast<ForcesAndSourcesCore *>(ptrFE)->elementMeasure;
}
 
/**
 * @brief Element used to execute operators on side of the element
 * 
 * @tparam E template for side element type
 * 
 */
template <typename E>
struct OpLoopSide : public ForcesAndSourcesCore::UserDataOperator {
 
  using UserDataOperator = ForcesAndSourcesCore::UserDataOperator;
 
  OpLoopSide(MoFEM::Interface &m_field, const std::string fe_name,
             const int side_dim, boost::shared_ptr<Range> fe_range,
             const LogManager::SeverityLevel sev = Sev::noisy,
             boost::shared_ptr<AdjCache> adj_cache = nullptr)
      : UserDataOperator(NOSPACE, OPSPACE), sideFEPtr(new E(m_field)),
        sideFEName(fe_name), sideDim(side_dim), sevLevel(sev),
        adjCache(adj_cache), feRange(fe_range) {}
 
  /**
   * @brief Construct a new Op Loop Side object
   * 
   * @param m_field 
   * @param fe_name name of side (domain element)
   * @param side_dim dimension
   * @param sev severity level
   * @param adj_cache if set to null cache of the element is used
   */
  OpLoopSide(MoFEM::Interface &m_field, const std::string fe_name,
             const int side_dim,
             const LogManager::SeverityLevel sev = Sev::noisy,
             boost::shared_ptr<AdjCache> adj_cache = nullptr)
      : OpLoopSide(m_field, fe_name, side_dim, nullptr, sev, adj_cache) {}
 
  MoFEMErrorCode doWork(int side, EntityType type,
                        EntitiesFieldData::EntData &data) {
    return loopSide(sideFEName, sideFEPtr.get(), sideDim, 0, feRange, VERBOSE,
                    sevLevel, adjCache.get());
  };
 
  boost::ptr_deque<UserDataOperator> &getOpPtrVector() {
    return sideFEPtr->getOpPtrVector();
  }
 
  boost::shared_ptr<E> &getSideFEPtr() { return sideFEPtr; }
 
  boost::weak_ptr<ForcesAndSourcesCore> getSubPipelinePtr() const override {
    return sideFEPtr;
  }
 
protected:
  const std::string sideFEName;
  const int sideDim;
  boost::shared_ptr<E> sideFEPtr;
  const LogManager::SeverityLevel sevLevel;
  boost::shared_ptr<AdjCache> adjCache;
  boost::shared_ptr<Range> feRange;
};
 
/**
 * @brief Iterate over range of (sub)elements
 * 
 * @tparam E 
 */
template <typename E>
struct OpLoopRange : public ForcesAndSourcesCore::UserDataOperator {
 
  using UserDataOperator = ForcesAndSourcesCore::UserDataOperator;
 
  OpLoopRange(MoFEM::Interface &m_field, const std::string fe_name,
              boost::shared_ptr<Range> fe_range,
              const LogManager::SeverityLevel sev = Sev::noisy)
      : UserDataOperator(NOSPACE, OPSPACE), rangeFEPtr(new E(m_field)),
        rangeFEName(fe_name), sevLevel(sev), feRange(fe_range) {}
 
  MoFEMErrorCode doWork(int side, EntityType type,
                        EntitiesFieldData::EntData &data) {
    return loopRange(rangeFEName, rangeFEPtr.get(), feRange, VERBOSE, sevLevel);
  };
 
  boost::ptr_deque<UserDataOperator> &getOpPtrVector() {
    return rangeFEPtr->getOpPtrVector();
  }
 
  boost::shared_ptr<E> &getRangeFEPtr() { return rangeFEPtr; }
 
  boost::weak_ptr<ForcesAndSourcesCore> getSubPipelinePtr() const override {
    return rangeFEPtr;
  }
 
protected:
  const std::string rangeFEName;
  boost::shared_ptr<E> rangeFEPtr;
  const LogManager::SeverityLevel sevLevel;
  boost::shared_ptr<Range> feRange;
};
 
/**
 * \brief Copy geometry-related data from one element to other
 *
 * That can be used to copy high order geometry data from coarse element to
 * children. That is often a case when higher order geometry is defined only on
 * coarse elements.
 *
 * \note Integration points have to be located at the same geometric positions
 *
 * FIXME: Write atom test
 */
template <int DIM>
struct OpCopyGeomDataToE;
 
} // namespace MoFEM
 
#endif //__FORCES_AND_SOURCES_CORE__HPP__
 
/**
 * \defgroup mofem_forces_and_sources Forces and sources
 * \ingroup mofem
 *
 * \brief Manages complexities related to assembly of vector and matrices at
 * single finite element level.
 *
 **/