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 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 CeedQFunctionCreateInterior(ceed, 1, AnisotropyTensorProjection, AnisotropyTensorProjection_loc, &qf_rhs_assemble); 66 CeedQFunctionAddInput(qf_rhs_assemble, "qdata", q_data_size, CEED_EVAL_NONE); 67 CeedQFunctionAddOutput(qf_rhs_assemble, "v", grid_aniso_proj->num_comp, CEED_EVAL_INTERP); 68 69 CeedOperatorCreate(ceed, qf_rhs_assemble, NULL, NULL, &op_rhs_assemble); 70 CeedOperatorSetField(op_rhs_assemble, "qdata", ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, ceed_data->q_data); 71 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 CeedOperatorCreate(ceed, qf_mass, NULL, NULL, &op_mass); 78 CeedOperatorSetField(op_mass, "u", *elem_restr_grid_aniso, basis_grid_aniso, CEED_VECTOR_ACTIVE); 79 CeedOperatorSetField(op_mass, "qdata", ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, ceed_data->q_data); 80 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 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 CeedQFunctionDestroy(&qf_rhs_assemble); 127 CeedQFunctionDestroy(&qf_mass); 128 CeedBasisDestroy(&basis_grid_aniso); 129 CeedOperatorDestroy(&op_rhs_assemble); 130 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 CeedBasisGetNumNodes(ceed_data->basis_q, &num_nodes); 144 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 CeedQFunctionCreateInterior(ceed, 1, AnisotropyTensorCollocate, AnisotropyTensorCollocate_loc, &qf_colloc); 149 CeedQFunctionAddInput(qf_colloc, "qdata", q_data_size, CEED_EVAL_NONE); 150 CeedQFunctionAddOutput(qf_colloc, "v", *num_comp_aniso, CEED_EVAL_NONE); 151 152 CeedOperatorCreate(ceed, qf_colloc, NULL, NULL, &op_colloc); 153 CeedOperatorSetField(op_colloc, "qdata", ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, ceed_data->q_data); 154 CeedOperatorSetField(op_colloc, "v", *elem_restr_grid_aniso, CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE); 155 156 CeedElemRestrictionCreateVector(*elem_restr_grid_aniso, aniso_colloc_ceed, NULL); 157 158 CeedOperatorApply(op_colloc, CEED_VECTOR_NONE, *aniso_colloc_ceed, CEED_REQUEST_IMMEDIATE); 159 160 CeedQFunctionDestroy(&qf_colloc); 161 CeedOperatorDestroy(&op_colloc); 162 PetscFunctionReturn(PETSC_SUCCESS); 163 } 164