1 // Copyright (c) 2017-2023, Lawrence Livermore National Security, LLC and other CEED contributors. 2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3 // 4 // SPDX-License-Identifier: BSD-2-Clause 5 // 6 // This file is part of CEED: http://github.com/ceed 7 8 #include "../qfunctions/grid_anisotropy_tensor.h" 9 10 #include <petscdmplex.h> 11 12 #include "../navierstokes.h" 13 14 PetscErrorCode GridAnisotropyTensorProjectionSetupApply(Ceed ceed, User user, CeedData ceed_data, CeedElemRestriction *elem_restr_grid_aniso, 15 CeedVector *grid_aniso_vector) { 16 NodalProjectionData grid_aniso_proj; 17 OperatorApplyContext mass_matop_ctx, l2_rhs_ctx; 18 CeedOperator op_rhs_assemble, op_mass; 19 CeedQFunction qf_rhs_assemble, qf_mass; 20 CeedBasis basis_grid_aniso; 21 PetscInt dim; 22 CeedInt q_data_size; 23 MPI_Comm comm = PetscObjectComm((PetscObject)user->dm); 24 KSP ksp; 25 26 PetscFunctionBeginUser; 27 PetscCall(PetscNew(&grid_aniso_proj)); 28 29 // -- Create DM for Anisotropic tensor L^2 projection 30 grid_aniso_proj->num_comp = 7; 31 PetscCall(DMClone(user->dm, &grid_aniso_proj->dm)); 32 PetscCall(DMGetDimension(grid_aniso_proj->dm, &dim)); 33 PetscCall(PetscObjectSetName((PetscObject)grid_aniso_proj->dm, "Grid Anisotropy Tensor Projection")); 34 35 { // -- Setup DM 36 PetscFE fe; 37 PetscSection section; 38 PetscInt q_order = user->app_ctx->degree + user->app_ctx->q_extra; 39 PetscCall(PetscFECreateLagrange(PETSC_COMM_SELF, dim, grid_aniso_proj->num_comp, PETSC_FALSE, user->app_ctx->degree, q_order, &fe)); 40 PetscCall(PetscObjectSetName((PetscObject)fe, "Grid Anisotropy Tensor Projection")); 41 PetscCall(DMAddField(grid_aniso_proj->dm, NULL, (PetscObject)fe)); 42 PetscCall(DMCreateDS(grid_aniso_proj->dm)); 43 PetscCall(DMPlexSetClosurePermutationTensor(grid_aniso_proj->dm, PETSC_DETERMINE, NULL)); 44 45 PetscCall(DMGetLocalSection(grid_aniso_proj->dm, §ion)); 46 PetscCall(PetscSectionSetFieldName(section, 0, "")); 47 PetscCall(PetscSectionSetComponentName(section, 0, 0, "KMGridAnisotropyTensorXX")); 48 PetscCall(PetscSectionSetComponentName(section, 0, 1, "KMGridAnisotropyTensorYY")); 49 PetscCall(PetscSectionSetComponentName(section, 0, 2, "KMGridAnisotropyTensorZZ")); 50 PetscCall(PetscSectionSetComponentName(section, 0, 3, "KMGridAnisotropyTensorYZ")); 51 PetscCall(PetscSectionSetComponentName(section, 0, 4, "KMGridAnisotropyTensorXZ")); 52 PetscCall(PetscSectionSetComponentName(section, 0, 5, "KMGridAnisotropyTensorXY")); 53 PetscCall(PetscSectionSetComponentName(section, 0, 6, "GridAnisotropyTensorFrobNorm")); 54 55 PetscCall(PetscFEDestroy(&fe)); 56 } 57 58 // -- Get Pre-requisite things 59 PetscCallCeed(ceed, CeedElemRestrictionGetNumComponents(ceed_data->elem_restr_qd_i, &q_data_size)); 60 61 PetscCall(GetRestrictionForDomain(ceed, grid_aniso_proj->dm, 0, 0, 0, 0, -1, grid_aniso_proj->num_comp, elem_restr_grid_aniso, NULL, NULL)); 62 PetscCall(CreateBasisFromPlex(ceed, grid_aniso_proj->dm, 0, 0, 0, 0, &basis_grid_aniso)); 63 64 // -- Build RHS operator 65 PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, AnisotropyTensorProjection, AnisotropyTensorProjection_loc, &qf_rhs_assemble)); 66 PetscCallCeed(ceed, CeedQFunctionAddInput(qf_rhs_assemble, "qdata", q_data_size, CEED_EVAL_NONE)); 67 PetscCallCeed(ceed, CeedQFunctionAddOutput(qf_rhs_assemble, "v", grid_aniso_proj->num_comp, CEED_EVAL_INTERP)); 68 69 PetscCallCeed(ceed, CeedOperatorCreate(ceed, qf_rhs_assemble, NULL, NULL, &op_rhs_assemble)); 70 PetscCallCeed(ceed, CeedOperatorSetField(op_rhs_assemble, "qdata", ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, ceed_data->q_data)); 71 PetscCallCeed(ceed, CeedOperatorSetField(op_rhs_assemble, "v", *elem_restr_grid_aniso, basis_grid_aniso, CEED_VECTOR_ACTIVE)); 72 73 PetscCall(OperatorApplyContextCreate(user->dm, grid_aniso_proj->dm, ceed, op_rhs_assemble, CEED_VECTOR_NONE, NULL, NULL, NULL, &l2_rhs_ctx)); 74 75 // -- Build Mass Operator 76 PetscCall(CreateMassQFunction(ceed, grid_aniso_proj->num_comp, q_data_size, &qf_mass)); 77 PetscCallCeed(ceed, CeedOperatorCreate(ceed, qf_mass, NULL, NULL, &op_mass)); 78 PetscCallCeed(ceed, CeedOperatorSetField(op_mass, "u", *elem_restr_grid_aniso, basis_grid_aniso, CEED_VECTOR_ACTIVE)); 79 PetscCallCeed(ceed, CeedOperatorSetField(op_mass, "qdata", ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, ceed_data->q_data)); 80 PetscCallCeed(ceed, CeedOperatorSetField(op_mass, "v", *elem_restr_grid_aniso, basis_grid_aniso, CEED_VECTOR_ACTIVE)); 81 82 { // -- Setup KSP for L^2 projection 83 Mat mat_mass; 84 PetscCall(OperatorApplyContextCreate(grid_aniso_proj->dm, grid_aniso_proj->dm, ceed, op_mass, NULL, NULL, NULL, NULL, &mass_matop_ctx)); 85 PetscCall(CreateMatShell_Ceed(mass_matop_ctx, &mat_mass)); 86 87 PetscCall(KSPCreate(comm, &ksp)); 88 PetscCall(KSPSetOptionsPrefix(ksp, "grid_anisotropy_tensor_projection_")); 89 { 90 PC pc; 91 PetscCall(KSPGetPC(ksp, &pc)); 92 PetscCall(PCSetType(pc, PCJACOBI)); 93 PetscCall(PCJacobiSetType(pc, PC_JACOBI_DIAGONAL)); 94 PetscCall(KSPSetType(ksp, KSPCG)); 95 PetscCall(KSPSetNormType(ksp, KSP_NORM_NATURAL)); 96 PetscCall(KSPSetTolerances(ksp, 1e-10, PETSC_DEFAULT, PETSC_DEFAULT, PETSC_DEFAULT)); 97 } 98 PetscCall(KSPSetOperators(ksp, mat_mass, mat_mass)); 99 PetscCall(KSPSetFromOptions(ksp)); 100 } 101 102 { // -- Project anisotropy data and store in CeedVector 103 Vec Grid_Anisotropy, grid_anisotropy_loc; 104 105 // Get L^2 Projection RHS 106 PetscCall(DMGetGlobalVector(grid_aniso_proj->dm, &Grid_Anisotropy)); 107 108 PetscCall(ApplyCeedOperatorLocalToGlobal(NULL, Grid_Anisotropy, l2_rhs_ctx)); 109 110 // Solve projection problem 111 PetscCall(KSPSolve(ksp, Grid_Anisotropy, Grid_Anisotropy)); 112 113 // Copy anisotropy tensor data to CeedVector 114 PetscCall(DMGetLocalVector(grid_aniso_proj->dm, &grid_anisotropy_loc)); 115 PetscCallCeed(ceed, CeedElemRestrictionCreateVector(*elem_restr_grid_aniso, grid_aniso_vector, NULL)); 116 PetscCall(DMGlobalToLocal(grid_aniso_proj->dm, Grid_Anisotropy, INSERT_VALUES, grid_anisotropy_loc)); 117 PetscCall(VecCopyP2C(grid_anisotropy_loc, *grid_aniso_vector)); 118 PetscCall(DMRestoreLocalVector(grid_aniso_proj->dm, &grid_anisotropy_loc)); 119 PetscCall(DMRestoreGlobalVector(grid_aniso_proj->dm, &Grid_Anisotropy)); 120 } 121 122 // -- Cleanup 123 PetscCall(NodalProjectionDataDestroy(grid_aniso_proj)); 124 PetscCall(OperatorApplyContextDestroy(l2_rhs_ctx)); 125 PetscCall(OperatorApplyContextDestroy(mass_matop_ctx)); 126 PetscCallCeed(ceed, CeedQFunctionDestroy(&qf_rhs_assemble)); 127 PetscCallCeed(ceed, CeedQFunctionDestroy(&qf_mass)); 128 PetscCallCeed(ceed, CeedBasisDestroy(&basis_grid_aniso)); 129 PetscCallCeed(ceed, CeedOperatorDestroy(&op_rhs_assemble)); 130 PetscCallCeed(ceed, CeedOperatorDestroy(&op_mass)); 131 PetscCall(KSPDestroy(&ksp)); 132 PetscFunctionReturn(PETSC_SUCCESS); 133 } 134 135 PetscErrorCode GridAnisotropyTensorCalculateCollocatedVector(Ceed ceed, User user, CeedData ceed_data, CeedElemRestriction *elem_restr_grid_aniso, 136 CeedVector *aniso_colloc_ceed, PetscInt *num_comp_aniso) { 137 CeedInt q_data_size, num_nodes; 138 CeedQFunction qf_colloc; 139 CeedOperator op_colloc; 140 141 PetscFunctionBeginUser; 142 *num_comp_aniso = 7; 143 PetscCallCeed(ceed, CeedBasisGetNumNodes(ceed_data->basis_q, &num_nodes)); 144 PetscCallCeed(ceed, CeedElemRestrictionGetNumComponents(ceed_data->elem_restr_qd_i, &q_data_size)); 145 PetscCall(GetRestrictionForDomain(ceed, user->dm, 0, 0, 0, 0, num_nodes, *num_comp_aniso, NULL, NULL, elem_restr_grid_aniso)); 146 147 // -- Build collocation operator 148 PetscCallCeed(ceed, CeedQFunctionCreateInterior(ceed, 1, AnisotropyTensorCollocate, AnisotropyTensorCollocate_loc, &qf_colloc)); 149 PetscCallCeed(ceed, CeedQFunctionAddInput(qf_colloc, "qdata", q_data_size, CEED_EVAL_NONE)); 150 PetscCallCeed(ceed, CeedQFunctionAddOutput(qf_colloc, "v", *num_comp_aniso, CEED_EVAL_NONE)); 151 152 PetscCallCeed(ceed, CeedOperatorCreate(ceed, qf_colloc, NULL, NULL, &op_colloc)); 153 PetscCallCeed(ceed, CeedOperatorSetField(op_colloc, "qdata", ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, ceed_data->q_data)); 154 PetscCallCeed(ceed, CeedOperatorSetField(op_colloc, "v", *elem_restr_grid_aniso, CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE)); 155 156 PetscCallCeed(ceed, CeedElemRestrictionCreateVector(*elem_restr_grid_aniso, aniso_colloc_ceed, NULL)); 157 158 PetscCallCeed(ceed, CeedOperatorApply(op_colloc, CEED_VECTOR_NONE, *aniso_colloc_ceed, CEED_REQUEST_IMMEDIATE)); 159 160 PetscCallCeed(ceed, CeedQFunctionDestroy(&qf_colloc)); 161 PetscCallCeed(ceed, CeedOperatorDestroy(&op_colloc)); 162 PetscFunctionReturn(PETSC_SUCCESS); 163 } 164