AdolCHuHu.cpp

Go to the documentation of this file.

/**
 * @file AdolCHuHu.cpp
 * @author your name (you@domain.com)
 * @version 0.1
 * @date 2026-03-29
 * 
 * @copyright Copyright (c) 2026
 * 
 */
 
#include <MoFEM.hpp>
 
using namespace MoFEM;
 
#include "AdolCOps.hpp"
#include "AdolCHuHu.hpp"
 
namespace AdolCOps {
 
template <int DIM>
struct AdolCHuHu : public PhysicalEquations {
  using PhysicalEquations::PhysicalEquations;
 
  using A = PhysicalEquations;
 
  ForcesAndSourcesCore::UserDataOperator *
  createOp(boost::shared_ptr<PhysicalEquations> physical_ptr, bool eval_stress,
           bool eval_tangent) override;
 
  MoFEMErrorCode getOptions(MoFEM::Interface *m_field_ptr = nullptr) override {
    MoFEMFunctionBegin;
    MOFEM_LOG_CHANNEL("WORLD");
 
    PetscOptionsBegin(PETSC_COMM_WORLD, "huhu_", "", "none");
    CHKERR PetscOptionsScalar("-K", "Bulk modulus K", "", K, &K, PETSC_NULLPTR);
    PetscOptionsEnd();
 
    MOFEM_TAG_AND_LOG("WORLD", Sev::inform, "Default HuHu parameters:")
        << " K = " << K;
    defaultMaterialParameters = {K};
 
    std::string block_name = "MAT_HUHU";
 
    for (auto &m :
 
         m_field_ptr->getInterface<MeshsetsManager>()->
 
         getCubitMeshsetPtr(
             std::regex((boost::format("%s(.*)") % block_name).str()))
 
    ) {
 
      std::vector<double> block_data;
      CHKERR m->getAttributes(block_data);
      if (block_data.size() < 1) {
        block_data.push_back(
            K); // Use default K if not provided in block attributes
      }
      auto get_block_ents = [&]() {
        Range ents;
        CHK_MOAB_THROW(m_field_ptr->get_moab().get_entities_by_handle(
                           m->meshset, ents, true),
                       "can not get block entities");
        return ents;
      };
 
      A::paramVecByRange.push_back({get_block_ents(), {block_data[0]}});
 
      MOFEM_TAG_AND_LOG("WORLD", Sev::inform, "MatBlock for HuHu")
          << *m << " K = " << block_data[0];
    }
 
    MoFEMFunctionReturn(0);
  };
 
  MoFEMErrorCode setParams(EntityHandle ent, int gg) override {
    set_param_vec(A::tAg, defaultMaterialParameters.size(),
                  defaultMaterialParameters.data());
    return 0;
  }
 
  MoFEMErrorCode recordTape() override {
    MoFEMFunctionBegin;
 
    A::adolcDataPtr->insertCommonData("grad", MatrixDouble(DIM * DIM, 1));
    A::adolcDataPtr->insertCommonData("k", MatrixDouble(1, 1));
    A::adolcDataPtr->insertCommonData("k_dF", MatrixDouble(DIM * DIM, 1));
 
    A::adolcDataPtr->insertActiveData("F", MatrixDouble(DIM, DIM));
    A::adolcDataPtr->insertDependentData("k", MatrixDouble(1, 1));
    A::adolcDataPtr->insertDependentDerivativesData("k_dF",
                                                    MatrixDouble(DIM, DIM));
 
    auto t_F = getFTensor2FromPtr<DIM, DIM>(
        A::adolcDataPtr->getActiveDataPtr("F")->data().data());
    auto k = (*A::adolcDataPtr->getDependentDataPtr("k"))(0, 0);
 
    auto t_kd = FTensor::Kronecker_Delta<int>();
 
    FTENSOR_INDEX(DIM, i);
    FTENSOR_INDEX(DIM, j);
    FTENSOR_INDEX(DIM, I);
    FTENSOR_INDEX(DIM, J);
 
    t_F(i, J) = t_kd(i, J);
 
    FTensor::Tensor2<adouble, DIM, DIM> ta_F;
    adouble ta_k;
 
    adouble det_aF;
    FTensor::Tensor2<adouble, DIM, DIM> ta_invF;
 
    trace_on(A::tAg);
    auto p_K = mkparam(K);
    ta_F(i, J) <<= t_F(i, J);
    ta_k = p_K;
    ta_k >>= k;
 
    trace_off();
 
    MoFEMFunctionReturn(0);
  }
 
protected:
  double K = 1e-6;
  std::vector<double> defaultMaterialParameters = {K};
};
 
template <>
boost::shared_ptr<PhysicalEquations>
createAdolCPhysicalEquationsPtr<HUHU, MODEL_3D>(
    boost::shared_ptr<ADolCData> adolc_data_ptr, int tag) {
  return boost::make_shared<AdolCHuHu<3>>(adolc_data_ptr, tag);
}
 
template <>
boost::shared_ptr<PhysicalEquations>
createAdolCPhysicalEquationsPtr<HUHU, MODEL_2D_PLANE_STRAIN>(
    boost::shared_ptr<ADolCData> adolc_data_ptr, int tag) {
  return boost::make_shared<AdolCHuHu<2>>(adolc_data_ptr, tag);
}
 
template <int DIM, typename DomainEleOp>
struct OpEvalAdolCHuHuImpl : public DomainEleOp {
 
  OpEvalAdolCHuHuImpl(boost::shared_ptr<PhysicalEquations> physical_ptr,
                      bool eval_stress, bool eval_tangent)
      : DomainEleOp("U", DomainEleOp::OPROW), physicalPtr(physical_ptr),
        evalStress(eval_stress), evalTangent(eval_tangent) {}
 
  MoFEMErrorCode doWork(int side, EntityType type, EntData &data);
 
protected:
  boost::shared_ptr<PhysicalEquations> physicalPtr;
  const bool evalStress;
  const bool evalTangent;
};
 
template <int DIM, typename DomainEleOp>
MoFEMErrorCode OpEvalAdolCHuHuImpl<DIM, DomainEleOp>::doWork(int side,
                                                             EntityType type,
                                                             EntData &data) {
  MoFEMFunctionBegin;
  int nb_dofs = data.getIndices().size();
  if (!nb_dofs)
    MoFEMFunctionReturnHot(0);
  int nb_integration_pts = data.getN().size1();
 
  constexpr int adolc_return_value = 0;
 
  auto physics_ptr = physicalPtr;
 
  auto get_tag = [&]() {
    if (physics_ptr->tagVsRangePtr) {
      for (const auto &tag_range_pair : *physics_ptr->tagVsRangePtr) {
        if (tag_range_pair.second.find(DomainEleOp::getFEEntityHandle()) !=
            tag_range_pair.second.end()) {
          return tag_range_pair.first;
        }
      }
    }
    return physics_ptr->tAg; // Default tag if no range matches
  };
 
  auto mat_grad_ptr = physics_ptr->adolcDataPtr->getCommonDataPtr("grad");
  auto mat_k_ptr = physics_ptr->adolcDataPtr->getCommonDataPtr("k");
  auto mat_k_dF_ptr = physics_ptr->adolcDataPtr->getCommonDataPtr("k_dF");
 
#ifndef NDEBUG
  if (!mat_grad_ptr || !mat_k_ptr || !mat_k_dF_ptr) {
    SETERRQ(PETSC_COMM_SELF, MOFEM_OPERATION_UNSUCCESSFUL,
            "Missing common data for ADOL-C evaluation");
  }
  if (mat_grad_ptr->size1() != DIM * DIM) {
    SETERRQ(PETSC_COMM_SELF, MOFEM_OPERATION_UNSUCCESSFUL,
            "Inconsistent size of gradient matrix for ADOL-C evaluation");
  }
  if (mat_grad_ptr->size2() != nb_integration_pts) {
    SETERRQ(PETSC_COMM_SELF, MOFEM_OPERATION_UNSUCCESSFUL,
            "Inconsistent size of gradient matrix data for ADOL-C evaluation");
  }
#endif
 
  if (evalStress) {
    FTENSOR_INDEX(DIM, i);
    FTENSOR_INDEX(DIM, J);
    FTENSOR_INDEX(DIM, K);
    FTENSOR_INDEX(DIM, L);
 
    auto t_grad_at_pts = getFTensor2FromMat<DIM, DIM>(*mat_grad_ptr);
    mat_k_ptr->resize(1, nb_integration_pts, false);
    auto t_k_at_pts = getFTensor1FromMat<1>(*mat_k_ptr);
 
    auto next = [&]() {
      ++t_grad_at_pts;
      ++t_k_at_pts;
    };
 
    auto t_kd = FTensor::Kronecker_Delta<int>();
 
    for (auto gg = 0; gg != nb_integration_pts; ++gg) {
      auto t_F = getFTensor2FromPtr<DIM, DIM>(
          physics_ptr->adolcDataPtr->getActiveDataPtr("F")->data().data());
      t_F(i, J) = t_grad_at_pts(i, J) + t_kd(i, J);
      CHKERR physics_ptr->setActiveContinuousVector();
      CHKERR physics_ptr->setDependentContinuousVector();
      CHKERR physics_ptr->setParams(this->getFEEntityHandle(), gg);
      int r = ::function(get_tag(), physics_ptr->dependentVariables.size(),
                         physics_ptr->activeVariables.size(),
                         physics_ptr->activeVariables.data(),
                         physics_ptr->dependentVariables.data());
      if (PetscUnlikely(r < adolc_return_value)) {
        SETERRQ(PETSC_COMM_SELF, MOFEM_OPERATION_UNSUCCESSFUL,
                "ADOL-C function evaluation with error");
      }
      CHKERR physics_ptr->getDependentContinuousVector();
      t_k_at_pts(0) =
          (*physics_ptr->adolcDataPtr->getDependentDataPtr("k"))(0, 0);
      next();
    }
  }
 
  if (evalTangent) {
    FTENSOR_INDEX(DIM, i);
    FTENSOR_INDEX(DIM, J);
    FTENSOR_INDEX(DIM, k);
    FTENSOR_INDEX(DIM, L);
 
    const int number_of_active_variables = physics_ptr->activeVariables.size();
    const int number_of_dependent_variables =
        physics_ptr->dependentVariables.size();
    CHKERR physics_ptr->setDependentDerivativesContinuousVector();
    std::vector<double *> jac_ptr(number_of_dependent_variables);
    for (unsigned int n = 0; n != number_of_dependent_variables; ++n) {
      jac_ptr[n] = &(
          physics_ptr
              ->dependentVariablesDerivatives[n * number_of_active_variables]);
    }
 
    auto t_grad_at_pts = getFTensor2FromMat<DIM, DIM>(*mat_grad_ptr);
    mat_k_dF_ptr->resize(DIM * DIM, nb_integration_pts, false);
    auto t_k_dF_at_pts = getFTensor2FromMat<DIM, DIM>(*mat_k_dF_ptr);
    auto t_kd = FTensor::Kronecker_Delta<int>();
 
    auto next = [&]() {
      ++t_grad_at_pts;
      ++t_k_dF_at_pts;
    };
 
 
    for (auto gg = 0; gg != nb_integration_pts; ++gg) {
      auto t_F = getFTensor2FromPtr<DIM, DIM>(
          physics_ptr->adolcDataPtr->getActiveDataPtr("F")->data().data());
      t_F(i, J) = t_grad_at_pts(i, J) + t_kd(i, J);
      CHKERR physics_ptr->setActiveContinuousVector();
      CHKERR physics_ptr->setDependentContinuousVector();
      CHKERR physics_ptr->setParams(this->getFEEntityHandle(), gg);
      int r = ::jacobian(get_tag(), number_of_dependent_variables,
                         number_of_active_variables,
                         physics_ptr->activeVariables.data(), jac_ptr.data());
      if (PetscUnlikely(r < adolc_return_value)) {
        SETERRQ(PETSC_COMM_SELF, MOFEM_OPERATION_UNSUCCESSFUL,
                "ADOL-C function evaluation with error");
      }
      CHKERR physics_ptr->getDependentDerivativesContinuousVector();
      auto t_k_dF = getFTensor2FromPtr<DIM, DIM>(
          physics_ptr->adolcDataPtr->getDependentDerivativesDataPtr("k_dF")
              ->data()
              .data());
      t_k_dF_at_pts(i, J) = t_k_dF(i, J);
      next();
    }
  }
 
  MoFEMFunctionReturn(0);
}
 
template <int DIM, typename EleOp>
static EleOp *createOpImpl(boost::shared_ptr<PhysicalEquations> physical_ptr,
                           bool eval_stress, bool eval_tangent) {
  return new OpEvalAdolCHuHuImpl<DIM, EleOp>(physical_ptr, eval_stress,
                                             eval_tangent);
}
 
template <>
ForcesAndSourcesCore::UserDataOperator *
AdolCHuHu<3>::createOp(boost::shared_ptr<PhysicalEquations> physical_ptr,
                       bool eval_stress, bool eval_tangent) {
  using EleOp = ForcesAndSourcesCore::UserDataOperator;
  return createOpImpl<3, EleOp>(physical_ptr, eval_stress, eval_tangent);
}
 
template <>
ForcesAndSourcesCore::UserDataOperator *
AdolCHuHu<2>::createOp(boost::shared_ptr<PhysicalEquations> physical_ptr,
                       bool eval_stress, bool eval_tangent) {
  using EleOp = ForcesAndSourcesCore::UserDataOperator;
  return createOpImpl<2, EleOp>(physical_ptr, eval_stress, eval_tangent);
}
 
} // namespace AdolCOps