MatStVenantKirchhoff.cpp

Go to the documentation of this file.

/**
 * @file MatStVenantKirchhoff.cpp
 * @author your name (you@domain.com)
 * @brief 
 * @version 0.1
 * @date 2026-03-26
 * 
 * @copyright Copyright (c) 2026
 * 
 */
 
namespace MatOps {
 
template <int DIM>
struct MatStVenantKirchhoff : public MatElasticImpl<DIM> {
  using MatElasticImpl<DIM>::MatElasticImpl;
 
  using A = MatElasticImpl<DIM>;
 
  MoFEMErrorCode getOptions(MoFEM::Interface *m_field_ptr = nullptr) override {
    MoFEMFunctionBegin;
 
    MOFEM_LOG_CHANNEL("WORLD");
 
    if (!A::paramVecByRange.empty()) {
      MoFEMFunctionReturnHot(0);
    }
 
    PetscOptionsBegin(PETSC_COMM_WORLD, "stvenant_", "", "none");
    CHKERR PetscOptionsScalar("-young_modulus", "Young modulus", "", E, &E,
                              PETSC_NULLPTR);
    CHKERR PetscOptionsScalar("-poisson_ratio", "poisson ratio", "", nu, &nu,
                              PETSC_NULLPTR);
    PetscOptionsEnd();
 
    std::string block_name = "MAT_STVENANTKIRCHHOFF";
 
    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);
      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;
      };
 
      CHKERR addBlockParameters(*m_field_ptr, m->getName(),
                                m->getMeshsetId(), get_block_ents(),
                                block_data);
    }
 
    if (A::paramVecByRange.empty()) {
      commandLineParameters = {E, nu};
      MOFEM_TAG_AND_LOG("WORLD", Sev::inform,
                        "Command line StVenantKirchhoff parameters")
          << "E = " << E << " nu = " << nu;
    }
 
    MoFEMFunctionReturn(0);
  };
 
  MoFEMErrorCode
  addBlockParameters(MoFEM::Interface &, const std::string &block_name, int,
                     const Range &block_entities,
                     const std::vector<double> &block_data) override {
    MoFEMFunctionBegin;
 
    if (block_data.size() < 2) {
      SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
              "StVenantKirchhoff block %s has %d attributes; expected E "
              "and nu",
              block_name.c_str(), static_cast<int>(block_data.size()));
    }
    std::vector<double> params = {block_data[0], block_data[1]};
 
    A::paramVecByRange.push_back({block_entities, params});
 
    MOFEM_TAG_AND_LOG("WORLD", Sev::inform, "MatBlock for StVenantKirchhoff")
        << block_name << " E = " << params[0] << " nu = " << params[1];
 
    MoFEMFunctionReturn(0);
  }
 
  MoFEMErrorCode setParams(FEMethod *fe_ptr, int gg) override {
    (void)gg;
    const auto ent = fe_ptr->getFEEntityHandle();
    for (auto &[range, param_vec] : A::paramVecByRange) {
      if (std::find(range.begin(), range.end(), ent) != range.end()) {
        set_param_vec(A::tAg, param_vec.size(), param_vec.data());
        return 0;
      }
    }
    // command line overrides block parameters
    if (!commandLineParameters.empty()) {
      set_param_vec(A::tAg, commandLineParameters.size(),
                    commandLineParameters.data());
      return 0;
    }
    SETERRQ(PETSC_COMM_SELF, MOFEM_DATA_INCONSISTENCY,
            "StVenantKirchhoff material parameters not found for entity %lu",
            static_cast<unsigned long>(ent));
  }
 
  MoFEMErrorCode recordTape() override {
    MoFEMFunctionBegin;
 
    A::matOpsDataPtr->insertCommonData("grad", MatrixDouble());
    A::matOpsDataPtr->insertCommonData("P", MatrixDouble());
    A::matOpsDataPtr->insertCommonData("P_dF", MatrixDouble());
 
    A::matOpsDataPtr->insertActiveData("F", MatrixDouble());
    A::matOpsDataPtr->insertDependentData("P", MatrixDouble());
    A::matOpsDataPtr->insertDependentDerivativesData("P_dF", MatrixDouble());
 
    A::matOpsDataPtr->getActiveDataPtr("F")->resize(DIM, DIM, false);
    A::matOpsDataPtr->getDependentDataPtr("P")->resize(DIM, DIM, false);
 
 
    auto t_F = getFTensor2FromPtr<DIM, DIM>(
        A::matOpsDataPtr->getActiveDataPtr("F")->data().data());
    auto t_P = getFTensor2FromPtr<DIM, DIM>(
        A::matOpsDataPtr->getDependentDataPtr("P")->data().data());
 
    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) = 0;
 
    FTensor::Tensor2<adouble, DIM, DIM> ta_F;
    FTensor::Tensor2<adouble, DIM, DIM> ta_P;
    FTensor::Tensor2<adouble, DIM, DIM> ta_C;
    FTensor::Tensor2<adouble, DIM, DIM> ta_E;
    FTensor::Tensor2<adouble, DIM, DIM> ta_S;
 
    adouble trE;
 
    trace_on(A::tAg);
    auto p_E = mkparam(E);
    auto p_nu = mkparam(nu);
 
    const auto calc_lambda = [](const auto &young_modulus,
                                const auto &poisson_ratio) {
      const auto nu_value = 1 * poisson_ratio;
      return (young_modulus * nu_value) /
             ((1. + nu_value) * (1. - 2. * nu_value));
    };
 
    const auto calc_mu = [](const auto &young_modulus,
                            const auto &poisson_ratio) {
      const auto nu_value = 1 * poisson_ratio;
      return 0.5 * (young_modulus / (1. + nu_value));
    };
 
    auto lambda = calc_lambda(p_E, p_nu);
    auto mu = calc_mu(p_E, p_nu);
 
    ta_F(i, J) <<= t_F(i, J);
    // assume that gradient from approximated displacement, that why we add diagonal
    if (!MatElastic::useDeformationGradient) {
      ta_F(i, J) += t_kd(i, J);
    }
 
    ta_C(I, J) = ta_F(i, I) * ta_F(i, J);
    ta_E(I, J) = (ta_C(I, J) -  t_kd(I, J))/2.;
 
    // Stress Piola II
    trE = ta_E(I, I);
    ta_S(I, J) = (2 * mu) * ta_E(I, J) + lambda * trE * t_kd(I, J);
    // Stress Piola I
    ta_P(i, J) = ta_F(i, I) * ta_S(I, J);
 
    ta_P(i, I) >>= t_P(i, I);
 
    trace_off();
 
    MoFEMFunctionReturn(0);
  }
 
protected:
  double E = 1;
  double nu = 0.25;
  std::vector<double> commandLineParameters;
};
 
template <>
boost::shared_ptr<PhysicalEquations>
createMatOpsPhysicalEquationsPtr<ELASTICITY::STVENANTKIRCHHOFF, MODEL_3D>(
    boost::shared_ptr<MatOpsData> mat_ops_data_ptr, int tag) {
  return boost::make_shared<MatStVenantKirchhoff<3>>(mat_ops_data_ptr, tag);
}
 
template <>
boost::shared_ptr<PhysicalEquations>
createMatOpsPhysicalEquationsPtr<ELASTICITY::STVENANTKIRCHHOFF,
                                MODEL_2D_PLANE_STRAIN>(
    boost::shared_ptr<MatOpsData> mat_ops_data_ptr, int tag) {
  return boost::make_shared<MatStVenantKirchhoff<2>>(mat_ops_data_ptr, tag);
}
} // namespace MatOps