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, q_data_size, num_qpts_1d, num_nodes_1d; 22 MPI_Comm comm = PetscObjectComm((PetscObject)user->dm); 23 KSP ksp; 24 25 PetscFunctionBeginUser; 26 PetscCall(PetscNew(&grid_aniso_proj)); 27 28 // -- Create DM for Anisotropic tensor L^2 projection 29 grid_aniso_proj->num_comp = 7; 30 PetscCall(DMClone(user->dm, &grid_aniso_proj->dm)); 31 PetscCall(DMGetDimension(grid_aniso_proj->dm, &dim)); 32 PetscCall(PetscObjectSetName((PetscObject)grid_aniso_proj->dm, "Grid Anisotropy Tensor Projection")); 33 34 { // -- Setup DM 35 PetscFE fe; 36 PetscSection section; 37 PetscCall(PetscFECreateLagrange(PETSC_COMM_SELF, dim, grid_aniso_proj->num_comp, PETSC_FALSE, user->app_ctx->degree, PETSC_DECIDE, &fe)); 38 PetscCall(PetscObjectSetName((PetscObject)fe, "Grid Anisotropy Tensor Projection")); 39 PetscCall(DMAddField(grid_aniso_proj->dm, NULL, (PetscObject)fe)); 40 PetscCall(DMCreateDS(grid_aniso_proj->dm)); 41 PetscCall(DMPlexSetClosurePermutationTensor(grid_aniso_proj->dm, PETSC_DETERMINE, NULL)); 42 43 PetscCall(DMGetLocalSection(grid_aniso_proj->dm, §ion)); 44 PetscCall(PetscSectionSetFieldName(section, 0, "")); 45 PetscCall(PetscSectionSetComponentName(section, 0, 0, "KMGridAnisotropyTensorXX")); 46 PetscCall(PetscSectionSetComponentName(section, 0, 1, "KMGridAnisotropyTensorYY")); 47 PetscCall(PetscSectionSetComponentName(section, 0, 2, "KMGridAnisotropyTensorZZ")); 48 PetscCall(PetscSectionSetComponentName(section, 0, 3, "KMGridAnisotropyTensorYZ")); 49 PetscCall(PetscSectionSetComponentName(section, 0, 4, "KMGridAnisotropyTensorXZ")); 50 PetscCall(PetscSectionSetComponentName(section, 0, 5, "KMGridAnisotropyTensorXY")); 51 PetscCall(PetscSectionSetComponentName(section, 0, 6, "GridAnisotropyTensorFrobNorm")); 52 53 PetscCall(PetscFEDestroy(&fe)); 54 } 55 56 // -- Get Pre-requisite things 57 CeedBasisGetNumQuadraturePoints1D(ceed_data->basis_q, &num_qpts_1d); 58 CeedBasisGetNumNodes1D(ceed_data->basis_q, &num_nodes_1d); 59 CeedElemRestrictionGetNumComponents(ceed_data->elem_restr_qd_i, &q_data_size); 60 61 PetscCall(GetRestrictionForDomain(ceed, grid_aniso_proj->dm, 0, 0, 0, num_qpts_1d, grid_aniso_proj->num_comp, elem_restr_grid_aniso, NULL, NULL)); 62 CeedBasisCreateTensorH1Lagrange(ceed, dim, grid_aniso_proj->num_comp, num_nodes_1d, num_qpts_1d, CEED_GAUSS, &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 CeedQFunctionDestroy(&qf_rhs_assemble); 126 CeedQFunctionDestroy(&qf_mass); 127 CeedBasisDestroy(&basis_grid_aniso); 128 CeedOperatorDestroy(&op_rhs_assemble); 129 CeedOperatorDestroy(&op_mass); 130 PetscCall(KSPDestroy(&ksp)); 131 PetscFunctionReturn(0); 132 } 133 134 PetscErrorCode GridAnisotropyTensorCalculateCollocatedVector(Ceed ceed, User user, CeedData ceed_data, CeedElemRestriction *elem_restr_grid_aniso, 135 CeedVector *aniso_colloc_ceed, PetscInt *num_comp_grid_aniso) { 136 PetscInt dim, q_data_size, num_qpts_1d, num_nodes_1d, loc_num_elem; 137 CeedQFunction qf_colloc; 138 CeedOperator op_colloc; 139 CeedBasis basis_grid_aniso; 140 141 PetscFunctionBeginUser; 142 // -- Get Pre-requisite things 143 *num_comp_grid_aniso = 7; 144 PetscCall(DMGetDimension(user->dm, &dim)); 145 CeedBasisGetNumQuadraturePoints1D(ceed_data->basis_q, &num_qpts_1d); 146 CeedBasisGetNumNodes1D(ceed_data->basis_q, &num_nodes_1d); 147 CeedElemRestrictionGetNumComponents(ceed_data->elem_restr_qd_i, &q_data_size); 148 149 PetscCall(GetRestrictionForDomain(ceed, user->dm, 0, 0, 0, num_qpts_1d, *num_comp_grid_aniso, NULL, NULL, elem_restr_grid_aniso)); 150 151 CeedInt Q_dim = CeedIntPow(num_qpts_1d, dim); 152 CeedElemRestrictionGetNumElements(ceed_data->elem_restr_q, &loc_num_elem); 153 CeedElemRestrictionCreateStrided(ceed, loc_num_elem, Q_dim, *num_comp_grid_aniso, *num_comp_grid_aniso * loc_num_elem * Q_dim, CEED_STRIDES_BACKEND, 154 elem_restr_grid_aniso); 155 156 CeedBasisCreateTensorH1Lagrange(ceed, dim, *num_comp_grid_aniso, num_nodes_1d, num_qpts_1d, CEED_GAUSS, &basis_grid_aniso); 157 158 // -- Build collocation operator 159 CeedQFunctionCreateInterior(ceed, 1, AnisotropyTensorCollocate, AnisotropyTensorCollocate_loc, &qf_colloc); 160 CeedQFunctionAddInput(qf_colloc, "qdata", q_data_size, CEED_EVAL_NONE); 161 CeedQFunctionAddOutput(qf_colloc, "v", *num_comp_grid_aniso, CEED_EVAL_NONE); 162 163 CeedOperatorCreate(ceed, qf_colloc, NULL, NULL, &op_colloc); 164 CeedOperatorSetField(op_colloc, "qdata", ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, ceed_data->q_data); 165 CeedOperatorSetField(op_colloc, "v", *elem_restr_grid_aniso, CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE); 166 CeedOperatorSetNumQuadraturePoints(op_colloc, CeedIntPow(num_qpts_1d, dim)); 167 168 CeedElemRestrictionCreateVector(*elem_restr_grid_aniso, aniso_colloc_ceed, NULL); 169 170 CeedOperatorApply(op_colloc, CEED_VECTOR_NONE, *aniso_colloc_ceed, CEED_REQUEST_IMMEDIATE); 171 172 PetscFunctionReturn(0); 173 } 174