1 // SPDX-FileCopyrightText: Copyright (c) 2017-2024, HONEE contributors. 2 // SPDX-License-Identifier: Apache-2.0 OR BSD-2-Clause 3 4 #include <navierstokes.h> 5 6 PetscClassId HONEE_CLASSID; 7 8 // @brief Initalize `HONEE` class. 9 static PetscErrorCode HoneeInitClass_Private() { 10 static PetscBool registered = PETSC_FALSE; 11 12 PetscFunctionBeginUser; 13 if (registered) PetscFunctionReturn(PETSC_SUCCESS); 14 PetscCall(PetscClassIdRegister("Honee", &HONEE_CLASSID)); 15 registered = PETSC_TRUE; 16 PetscFunctionReturn(PETSC_SUCCESS); 17 } 18 19 /** 20 @brief Initialize `Honee` object 21 22 @param[in] comm `MPI_Comm` for the object 23 @param[out] honee The initialized `Honee` object 24 **/ 25 PetscErrorCode HoneeInit(MPI_Comm comm, Honee *honee) { 26 Honee honee_; 27 AppCtx app_ctx; 28 ProblemData problem; 29 Physics phys_ctx; 30 Units units; 31 32 PetscFunctionBeginUser; 33 PetscCall(HoneeInitClass_Private()); 34 PetscCall(PetscHeaderCreate(honee_, HONEE_CLASSID, "Honee", "HONEE", "Honee", comm, HoneeDestroy, NULL)); 35 PetscCall(PetscNew(&app_ctx)); 36 PetscCall(PetscNew(&problem)); 37 PetscCall(PetscNew(&phys_ctx)); 38 PetscCall(PetscNew(&units)); 39 honee_->app_ctx = app_ctx; 40 honee_->units = units; 41 honee_->phys = phys_ctx; 42 honee_->problem_data = problem; 43 problem->set_bc_from_ics = PETSC_TRUE; 44 honee_->comm = PETSC_COMM_WORLD; 45 honee_->start_time = time(NULL); 46 47 PetscCall(RegisterLogEvents()); 48 49 *honee = honee_; 50 PetscFunctionReturn(PETSC_SUCCESS); 51 } 52 53 /** 54 @brief Destroy `Honee` object 55 56 @param[in] honee Object to be destroyed 57 **/ 58 PetscErrorCode HoneeDestroy(Honee *honee) { 59 Honee honee_ = *honee; 60 Ceed ceed; 61 62 PetscFunctionBeginUser; 63 if (!honee_) PetscFunctionReturn(PETSC_SUCCESS); 64 PetscValidHeaderSpecific(honee_, HONEE_CLASSID, 1); 65 66 ceed = honee_->ceed; 67 MPI_Comm comm = honee_->comm; 68 AppCtx app_ctx = honee_->app_ctx; 69 ProblemData problem = honee_->problem_data; 70 71 PetscCall(SgsDDDataDestroy(honee_->sgs_dd_data)); 72 PetscCall(DifferentialFilterDataDestroy(honee_->diff_filter)); 73 PetscCall(SGS_DD_TrainingDataDestroy(honee_->sgs_dd_train)); 74 PetscCall(SmartSimDataDestroy(honee_->smartsim)); 75 PetscCall(QDataClearStoredData()); 76 PetscCall(DivDiffFluxProjectionDataDestroy(honee_->diff_flux_proj)); 77 78 // -- Vectors 79 PetscCallCeed(ceed, CeedVectorDestroy(&honee_->x_coord)); 80 PetscCallCeed(ceed, CeedVectorDestroy(&honee_->q_ceed)); 81 PetscCallCeed(ceed, CeedVectorDestroy(&honee_->q_dot_ceed)); 82 PetscCallCeed(ceed, CeedVectorDestroy(&honee_->g_ceed)); 83 84 // -- Bases 85 PetscCallCeed(ceed, CeedBasisDestroy(&honee_->basis_q)); 86 PetscCallCeed(ceed, CeedBasisDestroy(&honee_->basis_x)); 87 88 // -- Restrictions 89 PetscCallCeed(ceed, CeedElemRestrictionDestroy(&honee_->elem_restr_q)); 90 PetscCallCeed(ceed, CeedElemRestrictionDestroy(&honee_->elem_restr_x)); 91 92 // Destroy QFunction contexts after using 93 { 94 PetscCallCeed(ceed, CeedQFunctionContextDestroy(&problem->ics.qfctx)); 95 PetscCallCeed(ceed, CeedQFunctionContextDestroy(&problem->apply_vol_rhs.qfctx)); 96 PetscCallCeed(ceed, CeedQFunctionContextDestroy(&problem->apply_vol_ifunction.qfctx)); 97 PetscCallCeed(ceed, CeedQFunctionContextDestroy(&problem->apply_vol_ijacobian.qfctx)); 98 } 99 100 // -- Operators 101 PetscCall(OperatorApplyContextDestroy(honee_->op_ics_ctx)); 102 PetscCall(OperatorApplyContextDestroy(honee_->op_rhs_ctx)); 103 PetscCall(OperatorApplyContextDestroy(honee_->op_strong_bc_ctx)); 104 PetscCallCeed(ceed, CeedOperatorDestroy(&honee_->op_ifunction)); 105 106 // -- Ceed 107 PetscCheck(CeedDestroy(&ceed) == CEED_ERROR_SUCCESS, comm, PETSC_ERR_LIB, "Destroying Ceed object failed"); 108 109 // --------------------------------------------------------------------------- 110 // Clean up PETSc 111 // --------------------------------------------------------------------------- 112 // -- Vectors 113 PetscCall(VecDestroy(&honee_->Q_loc)); 114 PetscCall(VecDestroy(&honee_->Q_dot_loc)); 115 116 PetscCall(KSPDestroy(&honee_->mass_ksp)); 117 118 // -- Matrices 119 PetscCall(MatDestroy(&honee_->interp_viz)); 120 PetscCall(MatDestroy(&honee_->mat_ijacobian)); 121 122 // -- DM 123 PetscCall(DMDestroy(&honee_->dm_viz)); 124 125 // -- Function list 126 PetscCall(PetscFunctionListDestroy(&app_ctx->problems)); 127 128 PetscCall(PetscFree(app_ctx->amat_type)); 129 PetscCall(PetscFree(app_ctx->wall_forces.walls)); 130 PetscCall(PetscViewerDestroy(&app_ctx->wall_forces.viewer)); 131 132 // -- Structs 133 for (PetscInt i = 0; i < problem->num_bc_defs; i++) { 134 PetscCall(BCDefinitionDestroy(&problem->bc_defs[i])); 135 } 136 PetscCall(PetscFree(problem->bc_defs)); 137 PetscCall(PetscFree(honee_->units)); 138 PetscCall(PetscFree(problem)); 139 PetscCall(PetscFree(honee_->phys)); 140 PetscCall(PetscFree(app_ctx)); 141 PetscCall(PetscHeaderDestroy(&honee_)); 142 *honee = NULL; 143 PetscFunctionReturn(PETSC_SUCCESS); 144 } 145