MatUmat.cpp

Go to the documentation of this file.

/**
 * @file MatUmat.cpp
 * @brief UMAT material model within MoFEM material impl.
 */
 
#include <MoFEM.hpp>
 
using namespace MoFEM;
 
#include "MatOps.hpp"
#include "MatUmatImpl.hpp"
 
namespace MatOps {
 
template <int DIM, typename EleOp>
static EleOp *createOpImpl(boost::shared_ptr<PhysicalEquations> physical_ptr,
                           bool eval_stress, bool eval_tangent, bool update);
 
template <int DIM, int MODEL_TYPE>
ForcesAndSourcesCore::UserDataOperator *
MatUmatImpl<DIM, MODEL_TYPE>::createOp(
    boost::shared_ptr<PhysicalEquations> physical_ptr, bool eval_stress,
    bool eval_tangent, bool update) {
  using EleOp = ForcesAndSourcesCore::UserDataOperator;
  return createOpImpl<DIM, EleOp>(physical_ptr, eval_stress, eval_tangent,
                                  update);
}
 
template <>
boost::shared_ptr<PhysicalEquations>
createMatOpsPhysicalEquationsPtr<UMAT, MODEL_3D>(
    boost::shared_ptr<MatOpsData> mat_ops_data_ptr, int tag) {
  return boost::make_shared<MatUmatImpl<3, MODEL_3D>>(mat_ops_data_ptr, tag);
}
 
template <>
boost::shared_ptr<PhysicalEquations>
createMatOpsPhysicalEquationsPtr<UMAT, MODEL_2D_PLANE_STRAIN>(
    boost::shared_ptr<MatOpsData> mat_ops_data_ptr, int tag) {
  return boost::make_shared<MatUmatImpl<2, MODEL_2D_PLANE_STRAIN>>(
      mat_ops_data_ptr, tag);
}
 
template <>
boost::shared_ptr<PhysicalEquations>
createMatOpsPhysicalEquationsPtr<UMAT, MODEL_2D_PLANE_STRESS>(
    boost::shared_ptr<MatOpsData> mat_ops_data_ptr, int tag) {
  return boost::make_shared<MatUmatImpl<2, MODEL_2D_PLANE_STRESS>>(
      mat_ops_data_ptr, tag);
}
 
template <>
boost::shared_ptr<PhysicalEquations>
createMatOpsPhysicalEquationsPtr<UMAT, MODEL_AXISYMMETRIC>(
    boost::shared_ptr<MatOpsData> mat_ops_data_ptr, int tag) {
  return boost::make_shared<MatUmatImpl<3, MODEL_AXISYMMETRIC>>(
      mat_ops_data_ptr, tag);
}
 
template <int DIM, typename DomainEleOp>
struct OpEvalMatUmatMaterialImpl : public DomainEleOp {
 
  using OP = DomainEleOp;
 
  OpEvalMatUmatMaterialImpl(boost::shared_ptr<PhysicalEquations> physical_ptr,
                            bool eval_stress, bool eval_tangent, bool update)
      : DomainEleOp(NOSPACE, DomainEleOp::OPSPACE), physicalPtr(physical_ptr),
        evalStress(eval_stress), evalTangent(eval_tangent),
        updateState(update) {}
 
  MoFEMErrorCode doWork(int side, EntityType type, EntData &data);
 
protected:
  boost::shared_ptr<PhysicalEquations> physicalPtr;
  const bool evalStress;
  const bool evalTangent;
  const bool updateState;
};
 
template <int DIM, typename DomainEleOp>
MoFEMErrorCode
OpEvalMatUmatMaterialImpl<DIM, DomainEleOp>::doWork(int side, EntityType type,
                                                    EntData &data) {
  MoFEMFunctionBegin;
 
  auto umat_op_data_ptr = boost::make_shared<UMatOpData>(this);
  physicalPtr->matOpsDataPtr->setUserDataPtr(umat_op_data_ptr);
 
  int nb_integration_pts = OP::getGaussPts().size2();
 
  auto get_tag = [&]() {
    if (physicalPtr->tagVsRangePtr) {
      for (const auto &tag_range_pair : *(physicalPtr->tagVsRangePtr)) {
        if (tag_range_pair.second.find(DomainEleOp::getFEEntityHandle()) !=
            tag_range_pair.second.end()) {
          return tag_range_pair.first;
        }
      }
    }
#ifndef NDEBUG
    if (MatOpsTagsRegistry::getTagName(physicalPtr->tAg).empty()) {
      CHK_THROW_MESSAGE(MOFEM_OPERATION_UNSUCCESSFUL,
                        "ADOL-C tag not found " +
                            std::to_string(physicalPtr->tAg));
    }
#endif
    return physicalPtr->tAg; // Default tag if no range matches
  };
 
  const int current_tag = get_tag();
  auto *fe_ptr = const_cast<FEMethod *>(this->getFEMethod());
  const auto ent = fe_ptr->getFEEntityHandle();
 
  physicalPtr->matOpsDataPtr->getActiveDataPtr("F")->resize(DIM, DIM, false);
  physicalPtr->matOpsDataPtr->getDependentDataPtr("P")->resize(DIM, DIM, false);
  physicalPtr->matOpsDataPtr->getDependentDerivativesDataPtr("P_dF")->resize(
      DIM * DIM, DIM * DIM, false);
  auto mat_grad_ptr = physicalPtr->matOpsDataPtr->getCommonDataPtr("grad");
  auto mat_P_ptr = physicalPtr->matOpsDataPtr->getCommonDataPtr("P");
  auto mat_P_dF_ptr = physicalPtr->matOpsDataPtr->getCommonDataPtr("P_dF");
 
#ifndef NDEBUG
  if (!mat_grad_ptr || !mat_P_ptr || !mat_P_dF_ptr) {
    SETERRQ(PETSC_COMM_SELF, MOFEM_OPERATION_UNSUCCESSFUL,
            "Missing common data for ADOL-C evaluation");
  }
  if (mat_grad_ptr->size2() != DIM * DIM) {
    SETERRQ(PETSC_COMM_SELF, MOFEM_OPERATION_UNSUCCESSFUL,
            "Inconsistent size of gradient matrix for ADOL-C evaluation");
  }
  if (mat_grad_ptr->size1() != nb_integration_pts) {
    SETERRQ(PETSC_COMM_SELF, MOFEM_OPERATION_UNSUCCESSFUL,
            "Inconsistent size of gradient matrix data for ADOL-C evaluation "
            "%zu != %d",
            mat_grad_ptr->size1(), nb_integration_pts);
  }
#endif
 
  using DL = DataLayoutTraits<DataLayout::GaussByCoeffs>;
  auto get_grad_at_pts =
      MatrixSizeHelper<GetFTensor2FromMatType<DIM, DIM, -1, DL>, DL>::get(
          *mat_grad_ptr, nb_integration_pts);
  auto get_P_at_pts =
      MatrixSizeHelper<GetFTensor2FromMatType<DIM, DIM, -1, DL>, DL>::size(
          *mat_P_ptr, nb_integration_pts);
  auto get_P_dF_at_pts =
      MatrixSizeHelper<GetFTensor4FromMatType<DIM, DIM, DIM, DIM, -1, DL>,
                       DL>::size(*mat_P_dF_ptr, nb_integration_pts);
 
  if (evalStress) {
    FTENSOR_INDEX(DIM, i);
    FTENSOR_INDEX(DIM, J);
 
    auto t_grad_at_pts = get_grad_at_pts();
    auto t_P_at_pts = get_P_at_pts();
 
    auto next = [&]() {
      ++t_grad_at_pts;
      ++t_P_at_pts;
    };
 
    auto t_kd = FTensor::Kronecker_Delta<int>();
 
    for (auto gg = 0; gg != nb_integration_pts; ++gg) {
      auto t_F = getFTensor2FromPtr<DIM, DIM>(
          physicalPtr->matOpsDataPtr->getActiveDataPtr("F")->data().data());
      t_F(i, J) = t_grad_at_pts(i, J);
      CHKERR physicalPtr->setParams(fe_ptr, gg);
      CHKERR physicalPtr->evaluateVariable(current_tag, ent, gg);
      auto t_P = getFTensor2FromPtr<DIM, DIM>(
          physicalPtr->matOpsDataPtr->getDependentDataPtr("P")->data().data());
      t_P_at_pts(i, J) = t_P(i, J);
      next();
    }
  }
 
  if (evalTangent) {
    FTENSOR_INDEX(DIM, i);
    FTENSOR_INDEX(DIM, J);
    FTENSOR_INDEX(DIM, k);
    FTENSOR_INDEX(DIM, L);
 
    auto t_grad_at_pts = get_grad_at_pts();
    auto t_P_dF_at_pts = get_P_dF_at_pts();
    auto t_kd = FTensor::Kronecker_Delta<int>();
 
    auto next = [&]() {
      ++t_grad_at_pts;
      ++t_P_dF_at_pts;
    };
 
    for (auto gg = 0; gg != nb_integration_pts; ++gg) {
      auto t_F = getFTensor2FromPtr<DIM, DIM>(
          physicalPtr->matOpsDataPtr->getActiveDataPtr("F")->data().data());
      t_F(i, J) = t_grad_at_pts(i, J);
      CHKERR physicalPtr->setParams(fe_ptr, gg);
      CHKERR physicalPtr->evaluateDerivatives(current_tag, ent, gg);
      auto t_P_dF = getFTensor4FromPtr<DIM, DIM, DIM, DIM>(
          physicalPtr->matOpsDataPtr->getDependentDerivativesDataPtr("P_dF")
              ->data()
              .data());
      t_P_dF_at_pts(i, J, k, L) = t_P_dF(i, J, k, L);
      next();
    }
  }
 
  if (updateState) {
    FTENSOR_INDEX(DIM, i);
    FTENSOR_INDEX(DIM, J);
 
    auto t_grad_at_pts = get_grad_at_pts();
    auto next = [&]() { ++t_grad_at_pts; };
 
    for (int gg = 0; gg != nb_integration_pts; ++gg) {
      auto t_F = getFTensor2FromPtr<DIM, DIM>(
          physicalPtr->matOpsDataPtr->getActiveDataPtr("F")->data().data());
      t_F(i, J) = t_grad_at_pts(i, J);
      CHKERR physicalPtr->setParams(fe_ptr, gg);
      CHKERR physicalPtr->updateState(current_tag, ent, gg);
      next();
    }
  }
 
  MoFEMFunctionReturn(0);
}
 
template <int DIM, typename EleOp>
static EleOp *createOpImpl(boost::shared_ptr<PhysicalEquations> physical_ptr,
                           bool eval_stress, bool eval_tangent, bool update) {
  return new OpEvalMatUmatMaterialImpl<DIM, EleOp>(physical_ptr, eval_stress,
                                                   eval_tangent, update);
}
 
template ForcesAndSourcesCore::UserDataOperator *
MatUmatImpl<3, MODEL_3D>::createOp(
  boost::shared_ptr<PhysicalEquations> physical_ptr, bool eval_stress,
  bool eval_tangent, bool update);
 
template ForcesAndSourcesCore::UserDataOperator *
MatUmatImpl<2, MODEL_2D_PLANE_STRAIN>::createOp(
  boost::shared_ptr<PhysicalEquations> physical_ptr, bool eval_stress,
  bool eval_tangent, bool update);
 
template ForcesAndSourcesCore::UserDataOperator *
MatUmatImpl<2, MODEL_2D_PLANE_STRESS>::createOp(
  boost::shared_ptr<PhysicalEquations> physical_ptr, bool eval_stress,
  bool eval_tangent, bool update);
 
template ForcesAndSourcesCore::UserDataOperator *
MatUmatImpl<3, MODEL_AXISYMMETRIC>::createOp(
  boost::shared_ptr<PhysicalEquations> physical_ptr, bool eval_stress,
  bool eval_tangent, bool update);
 
} // namespace MatOps