BlockMatData.hpp

Go to the documentation of this file.

/* This file is part of MoFEM.
 * MoFEM is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the
 * Free Software Foundation, either version 3 of the License, or (at your
 * option) any later version.
 *
 * MoFEM is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
 * License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with MoFEM. If not, see <http://www.gnu.org/licenses/>. */
 
#pragma once
 
namespace MoFEM {
 
// Not really data at integration points, used to calculate Schur complement
struct BlockMatData {
 
  std::string rowField;
  std::string colField;
  EntityType rowType;
  EntityType colType;
  int rowSide;
  int colSide;
  MatrixDouble M;
  VectorInt rowInd;
  VectorInt colInd;
 
  bool setAtElement;
 
  BlockMatData(const std::string row_field, const std::string col_field,
               EntityType row_type, EntityType col_type, int row_side,
               int col_side, const MatrixDouble &m, const VectorInt row_ind,
               VectorInt col_ind)
      : rowField(row_field), colField(col_field), rowType(row_type),
        colType(col_type), rowSide(row_side), colSide(col_side),
        setAtElement(true) {
 
    M.resize(m.size1(), m.size2(), false);
    noalias(M) = m;
    rowInd.resize(row_ind.size(), false);
    noalias(rowInd) = row_ind;
    colInd.resize(col_ind.size(), false);
    noalias(colInd) = col_ind;
  }
 
  void setInd(const VectorInt &row_ind, const VectorInt &col_ind) const {
    auto &const_row_ind = const_cast<VectorInt &>(rowInd);
    auto &const_col_ind = const_cast<VectorInt &>(colInd);
    const_row_ind.resize(row_ind.size(), false);
    noalias(const_row_ind) = row_ind;
    const_col_ind.resize(col_ind.size(), false);
    noalias(const_col_ind) = col_ind;
  }
 
  void setMat(const MatrixDouble &m) const {
    auto &const_m = const_cast<MatrixDouble &>(M);
    const_m.resize(m.size1(), m.size2(), false);
    noalias(const_m) = m;
  }
 
  void addMat(const MatrixDouble &m) const {
    auto &const_m = const_cast<MatrixDouble &>(M);
    const_m += m;
  }
 
  void clearMat() const {
    auto &const_m = const_cast<MatrixDouble &>(M);
    const_m.clear();
  }
 
  void setSetAtElement() const {
    bool &set = const_cast<bool &>(setAtElement);
    set = true;
  }
 
  void unSetAtElement() const {
    bool &set = const_cast<bool &>(setAtElement);
    set = false;
  }
};
 
typedef multi_index_container<
    BlockMatData,
    indexed_by<
 
        ordered_unique<
 
            composite_key<
                BlockMatData,
 
                member<BlockMatData, std::string, &BlockMatData::rowField>,
                member<BlockMatData, std::string, &BlockMatData::colField>,
                member<BlockMatData, EntityType, &BlockMatData::rowType>,
                member<BlockMatData, EntityType, &BlockMatData::colType>,
                member<BlockMatData, int, &BlockMatData::rowSide>,
                member<BlockMatData, int, &BlockMatData::colSide>
 
                >>,
        ordered_non_unique<
 
            composite_key<
                BlockMatData,
 
                member<BlockMatData, std::string, &BlockMatData::rowField>,
                member<BlockMatData, std::string, &BlockMatData::colField>,
                member<BlockMatData, EntityType, &BlockMatData::rowType>,
                member<BlockMatData, EntityType, &BlockMatData::colType>
 
                >>,
        ordered_non_unique<
 
            composite_key<
                BlockMatData,
 
                member<BlockMatData, std::string, &BlockMatData::rowField>,
                member<BlockMatData, std::string, &BlockMatData::colField>
 
                >>,
        ordered_non_unique<
            member<BlockMatData, std::string, &BlockMatData::rowField>>,
        ordered_non_unique<
            member<BlockMatData, std::string, &BlockMatData::colField>>
 
        >>
    BlockMatContainer;
    
} // namespace MoFEM