Remove DOFs from problem.
Remove DOFs from problem which are on entities on the given range and given field name. On the finite element level, DOFs can be still accessed however local PETSc indices and global PETSc indices are marked with the index -1.
2884 const Problem *prb_ptr;
2887 decltype(prb_ptr->numeredRowDofsPtr) numered_dofs[2] = {
2889 if (prb_ptr->numeredRowDofsPtr != prb_ptr->numeredColDofsPtr)
2890 numered_dofs[1] = prb_ptr->numeredColDofsPtr;
2892 int *nbdof_ptr[] = {&prb_ptr->nbDofsRow, &prb_ptr->nbDofsCol};
2893 int *local_nbdof_ptr[] = {&prb_ptr->nbLocDofsRow, &prb_ptr->nbLocDofsCol};
2894 int *ghost_nbdof_ptr[] = {&prb_ptr->nbGhostDofsRow, &prb_ptr->nbGhostDofsCol};
2896 const int nb_init_row_dofs = prb_ptr->getNbDofsRow();
2897 const int nb_init_col_dofs = prb_ptr->getNbDofsCol();
2898 const int nb_init_loc_row_dofs = prb_ptr->getNbLocalDofsRow();
2900 const int nb_init_ghost_row_dofs = prb_ptr->getNbGhostDofsRow();
2901 const int nb_init_ghost_col_dofs = prb_ptr->getNbGhostDofsCol();
2903 const std::array<int, 2> nb_init_dofs = {nb_init_row_dofs, nb_init_col_dofs};
2905 for (
int s = 0; s != 2; ++s)
2906 if (numered_dofs[s]) {
2908 typedef multi_index_container<
2910 NumeredDofEntity_multiIndex::iterator, indexed_by<sequenced<>>
2913 NumeredDofEntity_it_view_multiIndex;
2916 NumeredDofEntity_it_view_multiIndex dofs_it_view;
2919 for (
auto pit = ents.const_pair_begin(); pit != ents.const_pair_end();
2921 auto lo = numered_dofs[s]->get<Unique_mi_tag>().lower_bound(
2923 auto hi = numered_dofs[s]->get<Unique_mi_tag>().upper_bound(
2926 for (; lo != hi; ++lo)
2927 if ((*lo)->getDofCoeffIdx() >= lo_coeff &&
2928 (*lo)->getDofCoeffIdx() <= hi_coeff &&
2929 (*lo)->getDofOrder() >= lo_order &&
2930 (*lo)->getDofOrder() <= hi_order)
2931 dofs_it_view.emplace_back(numered_dofs[s]->project<0>(lo));
2935 for (
auto &dof : dofs_it_view)
2940 auto mod = NumeredDofEntity_part_and_all_indices_change(-1, -1, -1, -1);
2941 for (
auto dit : dofs_it_view) {
2942 bool success = numered_dofs[s]->modify(dit, mod);
2945 "can not set negative indices");
2949 std::vector<boost::weak_ptr<NumeredDofEntity>> dosf_weak_view;
2950 dosf_weak_view.reserve(dofs_it_view.size());
2951 for (
auto dit : dofs_it_view)
2952 dosf_weak_view.push_back(*dit);
2956 "Number of DOFs in multi-index %d and to delete %d\n",
2957 numered_dofs[s]->size(), dofs_it_view.size());
2960 for (
auto weak_dit : dosf_weak_view)
2961 if (
auto dit = weak_dit.lock()) {
2962 numered_dofs[s]->erase(dit->getLocalUniqueId());
2967 "Number of DOFs in multi-index after delete %d\n",
2968 numered_dofs[s]->size());
2971 int nb_global_dof = 0;
2972 int nb_local_dofs = 0;
2973 int nb_ghost_dofs = 0;
2975 for (
auto dit = numered_dofs[s]->begin(); dit != numered_dofs[s]->end();
2978 if ((*dit)->getDofIdx() >= 0) {
2980 if ((*dit)->getPetscLocalDofIdx() >= 0 &&
2981 (*dit)->getPetscLocalDofIdx() < *(local_nbdof_ptr[s]))
2983 else if ((*dit)->getPetscLocalDofIdx() >= *(local_nbdof_ptr[s]))
2991 MPI_Allreduce(&nb_local_dofs, nbdof_ptr[s], 1, MPI_INT, MPI_SUM,
2993 if (*(nbdof_ptr[s]) != nb_global_dof)
2995 "Number of local DOFs do not add up %d != %d",
2996 *(nbdof_ptr[s]), nb_global_dof);
2999 *(nbdof_ptr[s]) = nb_global_dof;
3000 *(local_nbdof_ptr[s]) = nb_local_dofs;
3001 *(ghost_nbdof_ptr[s]) = nb_ghost_dofs;
3004 auto get_indices_by_tag = [&](
auto tag) {
3005 std::vector<int> indices;
3006 indices.resize(nb_init_dofs[s], -1);
3007 for (
auto dit = numered_dofs[s]->get<Idx_mi_tag>().lower_bound(0);
3008 dit != numered_dofs[s]->get<Idx_mi_tag>().end(); ++dit) {
3009 indices[(*dit)->getDofIdx()] = decltype(tag)::get_index(dit);
3014 auto renumber = [&](
auto tag,
auto &indices) {
3017 for (
auto dit = numered_dofs[s]->get<decltype(tag)>().lower_bound(0);
3018 dit != numered_dofs[s]->get<decltype(tag)>().end(); ++dit) {
3019 indices[(*dit)->getDofIdx()] = idx++;
3024 auto get_sub_ao = [&](
auto sub_data) {
3026 return sub_data->getSmartRowMap();
3028 return sub_data->getSmartColMap();
3032 auto set_sub_is_and_ao = [&s, &prb_ptr](
auto sub_data,
auto is,
auto ao) {
3034 sub_data->rowIs = is;
3035 sub_data->rowMap = ao;
3037 sub_data->colIs = is;
3038 sub_data->colMap = ao;
3042 auto apply_symmetry = [&s, &prb_ptr](
auto sub_data) {
3044 if (prb_ptr->numeredRowDofsPtr == prb_ptr->numeredColDofsPtr) {
3045 sub_data->colIs = sub_data->getSmartRowIs();
3046 sub_data->colMap = sub_data->getSmartRowMap();
3051 auto set_sub_data = [&](
auto &indices) {
3053 if (
auto sub_data = prb_ptr->getSubData()) {
3056 &*indices.begin(), PETSC_COPY_VALUES);
3059 auto sub_ao = get_sub_ao(sub_data);
3060 CHKERR AOPetscToApplicationIS(sub_ao, sub_is);
3063 set_sub_is_and_ao(sub_data, sub_is, sub_ao);
3064 apply_symmetry(sub_data);
3066 prb_ptr->getSubData() = boost::make_shared<Problem::SubProblemData>();
3068 &*indices.begin(), PETSC_COPY_VALUES);
3071 set_sub_is_and_ao(prb_ptr->getSubData(), sub_is, sub_ao);
3072 apply_symmetry(prb_ptr->getSubData());
3077 auto global_indices = get_indices_by_tag(PetscGlobalIdx_mi_tag());
3078 auto local_indices = get_indices_by_tag(PetscLocalIdx_mi_tag());
3079 CHKERR set_sub_data(global_indices);
3080 CHKERR renumber(PetscGlobalIdx_mi_tag(), global_indices);
3081 CHKERR renumber(PetscLocalIdx_mi_tag(), local_indices);
3084 for (
auto dit = numered_dofs[s]->begin(); dit != numered_dofs[s]->end();
3086 auto idx = (*dit)->getDofIdx();
3088 auto mod = NumeredDofEntity_part_and_all_indices_change(
3089 (*dit)->getPart(),
i++, global_indices[idx], local_indices[idx]);
3090 bool success = numered_dofs[s]->modify(dit, mod);
3093 "can not set negative indices");
3095 auto mod = NumeredDofEntity_part_and_all_indices_change(
3096 (*dit)->getPart(), -1, -1, -1);
3097 bool success = numered_dofs[s]->modify(dit, mod);
3100 "can not set negative indices");
3105 for (
auto dof : (*numered_dofs[s])) {
3106 if (dof->getDofIdx() >= 0 && dof->getPetscGlobalDofIdx() < 0) {
3108 "Negative global idx");
3115 *(nbdof_ptr[1]) = *(nbdof_ptr[0]);
3116 *(local_nbdof_ptr[1]) = *(local_nbdof_ptr[0]);
3117 *(ghost_nbdof_ptr[1]) = *(ghost_nbdof_ptr[0]);
3125 "WORLD", Sev::inform,
3126 "Removed DOFs from problem %s dofs [%d / %d (before %d / %d) global]",
3127 prb_ptr->getName().c_str(), prb_ptr->getNbDofsRow(),
3128 prb_ptr->getNbDofsCol(), nb_init_row_dofs, nb_init_col_dofs);
3130 "Removed DOFs from problem %s dofs [ %d / %d "
3131 "(before %d / %d) local, %d / %d (before %d / %d)]",
3132 prb_ptr->getName().c_str(), prb_ptr->getNbLocalDofsRow(),
3133 prb_ptr->getNbLocalDofsCol(), nb_init_loc_row_dofs,
3134 nb_init_loc_row_dofs, prb_ptr->getNbGhostDofsRow(),
3135 prb_ptr->getNbGhostDofsCol(), nb_init_ghost_row_dofs,
3136 nb_init_ghost_col_dofs);
Public Member Functions inherited from MoFEM::UnknownInterface