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.
2886 const Problem *prb_ptr;
2889 decltype(prb_ptr->numeredRowDofsPtr) numered_dofs[2] = {
2891 if (prb_ptr->numeredRowDofsPtr != prb_ptr->numeredColDofsPtr)
2892 numered_dofs[1] = prb_ptr->numeredColDofsPtr;
2894 int *nbdof_ptr[] = {&prb_ptr->nbDofsRow, &prb_ptr->nbDofsCol};
2895 int *local_nbdof_ptr[] = {&prb_ptr->nbLocDofsRow, &prb_ptr->nbLocDofsCol};
2896 int *ghost_nbdof_ptr[] = {&prb_ptr->nbGhostDofsRow, &prb_ptr->nbGhostDofsCol};
2898 const int nb_init_row_dofs = prb_ptr->getNbDofsRow();
2899 const int nb_init_col_dofs = prb_ptr->getNbDofsCol();
2900 const int nb_init_loc_row_dofs = prb_ptr->getNbLocalDofsRow();
2902 const int nb_init_ghost_row_dofs = prb_ptr->getNbGhostDofsRow();
2903 const int nb_init_ghost_col_dofs = prb_ptr->getNbGhostDofsCol();
2905 const std::array<int, 2> nb_init_dofs = {nb_init_row_dofs, nb_init_col_dofs};
2907 for (
int s = 0; s != 2; ++s)
2908 if (numered_dofs[s]) {
2910 typedef multi_index_container<
2912 NumeredDofEntity_multiIndex::iterator, indexed_by<sequenced<>>
2915 NumeredDofEntity_it_view_multiIndex;
2918 NumeredDofEntity_it_view_multiIndex dofs_it_view;
2921 for (
auto pit = ents.const_pair_begin(); pit != ents.const_pair_end();
2923 auto lo = numered_dofs[s]->get<Unique_mi_tag>().lower_bound(
2925 auto hi = numered_dofs[s]->get<Unique_mi_tag>().upper_bound(
2928 for (; lo != hi; ++lo)
2929 if ((*lo)->getDofCoeffIdx() >= lo_coeff &&
2930 (*lo)->getDofCoeffIdx() <= hi_coeff &&
2931 (*lo)->getDofOrder() >= lo_order &&
2932 (*lo)->getDofOrder() <= hi_order)
2933 dofs_it_view.emplace_back(numered_dofs[s]->project<0>(lo));
2937 for (
auto &dof : dofs_it_view)
2942 auto mod = NumeredDofEntity_part_and_all_indices_change(-1, -1, -1, -1);
2943 for (
auto dit : dofs_it_view) {
2944 bool success = numered_dofs[s]->modify(dit, mod);
2947 "can not set negative indices");
2951 std::vector<boost::weak_ptr<NumeredDofEntity>> dosf_weak_view;
2952 dosf_weak_view.reserve(dofs_it_view.size());
2953 for (
auto dit : dofs_it_view)
2954 dosf_weak_view.push_back(*dit);
2958 "Number of DOFs in multi-index %d and to delete %d\n",
2959 numered_dofs[s]->size(), dofs_it_view.size());
2962 for (
auto weak_dit : dosf_weak_view)
2963 if (
auto dit = weak_dit.lock()) {
2964 numered_dofs[s]->erase(dit->getLocalUniqueId());
2969 "Number of DOFs in multi-index after delete %d\n",
2970 numered_dofs[s]->size());
2973 int nb_global_dof = 0;
2974 int nb_local_dofs = 0;
2975 int nb_ghost_dofs = 0;
2977 for (
auto dit = numered_dofs[s]->begin(); dit != numered_dofs[s]->end();
2980 if ((*dit)->getDofIdx() >= 0) {
2982 if ((*dit)->getPetscLocalDofIdx() >= 0 &&
2983 (*dit)->getPetscLocalDofIdx() < *(local_nbdof_ptr[s]))
2985 else if ((*dit)->getPetscLocalDofIdx() >= *(local_nbdof_ptr[s]))
2993 MPI_Allreduce(&nb_local_dofs, nbdof_ptr[s], 1, MPI_INT, MPI_SUM,
2995 if (*(nbdof_ptr[s]) != nb_global_dof)
2997 "Number of local DOFs do not add up %d != %d",
2998 *(nbdof_ptr[s]), nb_global_dof);
3001 *(nbdof_ptr[s]) = nb_global_dof;
3002 *(local_nbdof_ptr[s]) = nb_local_dofs;
3003 *(ghost_nbdof_ptr[s]) = nb_ghost_dofs;
3006 auto get_indices_by_tag = [&](
auto tag) {
3007 std::vector<int> indices;
3008 indices.resize(nb_init_dofs[s], -1);
3009 for (
auto dit = numered_dofs[s]->get<Idx_mi_tag>().lower_bound(0);
3010 dit != numered_dofs[s]->get<Idx_mi_tag>().end(); ++dit) {
3011 indices[(*dit)->getDofIdx()] = decltype(tag)::get_index(dit);
3016 auto renumber = [&](
auto tag,
auto &indices) {
3019 for (
auto dit = numered_dofs[s]->get<decltype(tag)>().lower_bound(0);
3020 dit != numered_dofs[s]->get<decltype(tag)>().end(); ++dit) {
3021 indices[(*dit)->getDofIdx()] = idx++;
3026 auto get_sub_ao = [&](
auto sub_data) {
3028 return sub_data->getSmartRowMap();
3030 return sub_data->getSmartColMap();
3034 auto set_sub_is_and_ao = [&s, &prb_ptr](
auto sub_data,
auto is,
auto ao) {
3036 sub_data->rowIs = is;
3037 sub_data->rowMap = ao;
3038 sub_data->colIs = is;
3039 sub_data->colMap = ao;
3041 sub_data->colIs = is;
3042 sub_data->colMap = ao;
3046 auto apply_symmetry = [&s, &prb_ptr](
auto sub_data) {
3048 if (prb_ptr->numeredRowDofsPtr == prb_ptr->numeredColDofsPtr) {
3049 sub_data->colIs = sub_data->getSmartRowIs();
3050 sub_data->colMap = sub_data->getSmartRowMap();
3055 auto set_sub_data = [&](
auto &indices) {
3057 if (
auto sub_data = prb_ptr->getSubData()) {
3060 &*indices.begin(), PETSC_COPY_VALUES);
3063 auto sub_ao = get_sub_ao(sub_data);
3064 CHKERR AOPetscToApplicationIS(sub_ao, sub_is);
3067 set_sub_is_and_ao(sub_data, sub_is, sub_ao);
3068 apply_symmetry(sub_data);
3070 prb_ptr->getSubData() = boost::make_shared<Problem::SubProblemData>();
3072 &*indices.begin(), PETSC_COPY_VALUES);
3075 set_sub_is_and_ao(prb_ptr->getSubData(), sub_is, sub_ao);
3076 apply_symmetry(prb_ptr->getSubData());
3081 auto global_indices = get_indices_by_tag(PetscGlobalIdx_mi_tag());
3082 auto local_indices = get_indices_by_tag(PetscLocalIdx_mi_tag());
3083 CHKERR set_sub_data(global_indices);
3084 CHKERR renumber(PetscGlobalIdx_mi_tag(), global_indices);
3085 CHKERR renumber(PetscLocalIdx_mi_tag(), local_indices);
3088 for (
auto dit = numered_dofs[s]->begin(); dit != numered_dofs[s]->end();
3090 auto idx = (*dit)->getDofIdx();
3092 auto mod = NumeredDofEntity_part_and_all_indices_change(
3093 (*dit)->getPart(),
i++, global_indices[idx], local_indices[idx]);
3094 bool success = numered_dofs[s]->modify(dit, mod);
3097 "can not set negative indices");
3099 auto mod = NumeredDofEntity_part_and_all_indices_change(
3100 (*dit)->getPart(), -1, -1, -1);
3101 bool success = numered_dofs[s]->modify(dit, mod);
3104 "can not set negative indices");
3109 for (
auto dof : (*numered_dofs[s])) {
3110 if (dof->getDofIdx() >= 0 && dof->getPetscGlobalDofIdx() < 0) {
3112 "Negative global idx");
3119 *(nbdof_ptr[1]) = *(nbdof_ptr[0]);
3120 *(local_nbdof_ptr[1]) = *(local_nbdof_ptr[0]);
3121 *(ghost_nbdof_ptr[1]) = *(ghost_nbdof_ptr[0]);
3129 "WORLD", Sev::inform,
3130 "Removed DOFs from problem %s dofs [%d / %d (before %d / %d) global]",
3131 prb_ptr->getName().c_str(), prb_ptr->getNbDofsRow(),
3132 prb_ptr->getNbDofsCol(), nb_init_row_dofs, nb_init_col_dofs);
3134 "Removed DOFs from problem %s dofs [ %d / %d "
3135 "(before %d / %d) local, %d / %d (before %d / %d)]",
3136 prb_ptr->getName().c_str(), prb_ptr->getNbLocalDofsRow(),
3137 prb_ptr->getNbLocalDofsCol(), nb_init_loc_row_dofs,
3138 nb_init_loc_row_dofs, prb_ptr->getNbGhostDofsRow(),
3139 prb_ptr->getNbGhostDofsCol(), nb_init_ghost_row_dofs,
3140 nb_init_ghost_col_dofs);
Public Member Functions inherited from MoFEM::UnknownInterface