13d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 377841947SLeila Ghaffari // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 577841947SLeila Ghaffari // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 777841947SLeila Ghaffari 877841947SLeila Ghaffari /// @file 977841947SLeila Ghaffari /// Setup libCEED for Navier-Stokes example using PETSc 1077841947SLeila Ghaffari 1177841947SLeila Ghaffari #include "../navierstokes.h" 1277841947SLeila Ghaffari 1377841947SLeila Ghaffari // Utility function - essential BC dofs are encoded in closure indices as -(i+1). 1477841947SLeila Ghaffari PetscInt Involute(PetscInt i) { 1577841947SLeila Ghaffari return i >= 0 ? i : -(i+1); 1677841947SLeila Ghaffari } 1777841947SLeila Ghaffari 1877841947SLeila Ghaffari // Utility function to create local CEED restriction 197ed3e4cdSJeremy L Thompson PetscErrorCode CreateRestrictionFromPlex(Ceed ceed, DM dm, CeedInt height, 207ed3e4cdSJeremy L Thompson DMLabel domain_label, CeedInt value, CeedElemRestriction *elem_restr) { 217ed3e4cdSJeremy L Thompson PetscInt num_elem, elem_size, num_dof, num_comp, *elem_restr_offsets; 2277841947SLeila Ghaffari PetscErrorCode ierr; 237ed3e4cdSJeremy L Thompson 2477841947SLeila Ghaffari PetscFunctionBeginUser; 257ed3e4cdSJeremy L Thompson ierr = DMPlexGetLocalOffsets(dm, domain_label, value, height, 0, &num_elem, 267ed3e4cdSJeremy L Thompson &elem_size, &num_comp, &num_dof, &elem_restr_offsets); 277ed3e4cdSJeremy L Thompson CHKERRQ(ierr); 2877841947SLeila Ghaffari 297ed3e4cdSJeremy L Thompson CeedElemRestrictionCreate(ceed, num_elem, elem_size, num_comp, 307ed3e4cdSJeremy L Thompson 1, num_dof, CEED_MEM_HOST, CEED_COPY_VALUES, 317ed3e4cdSJeremy L Thompson elem_restr_offsets, elem_restr); 327ed3e4cdSJeremy L Thompson ierr = PetscFree(elem_restr_offsets); CHKERRQ(ierr); 3377841947SLeila Ghaffari 3477841947SLeila Ghaffari PetscFunctionReturn(0); 3577841947SLeila Ghaffari } 3677841947SLeila Ghaffari 3777841947SLeila Ghaffari // Utility function to get Ceed Restriction for each domain 3877841947SLeila Ghaffari PetscErrorCode GetRestrictionForDomain(Ceed ceed, DM dm, CeedInt height, 3977841947SLeila Ghaffari DMLabel domain_label, PetscInt value, 407ed3e4cdSJeremy L Thompson CeedInt Q, CeedInt q_data_size, 4177841947SLeila Ghaffari CeedElemRestriction *elem_restr_q, 4277841947SLeila Ghaffari CeedElemRestriction *elem_restr_x, 4377841947SLeila Ghaffari CeedElemRestriction *elem_restr_qd_i) { 4477841947SLeila Ghaffari DM dm_coord; 4577841947SLeila Ghaffari CeedInt dim, loc_num_elem; 4677841947SLeila Ghaffari CeedInt Q_dim; 472534dcc8SJed Brown CeedElemRestriction elem_restr_tmp; 4877841947SLeila Ghaffari PetscErrorCode ierr; 4977841947SLeila Ghaffari PetscFunctionBeginUser; 5077841947SLeila Ghaffari 5177841947SLeila Ghaffari ierr = DMGetDimension(dm, &dim); CHKERRQ(ierr); 5277841947SLeila Ghaffari dim -= height; 5377841947SLeila Ghaffari Q_dim = CeedIntPow(Q, dim); 542534dcc8SJed Brown ierr = CreateRestrictionFromPlex(ceed, dm, height, domain_label, value, 552534dcc8SJed Brown &elem_restr_tmp); 562534dcc8SJed Brown CHKERRQ(ierr); 572534dcc8SJed Brown if (elem_restr_q) *elem_restr_q = elem_restr_tmp; 582534dcc8SJed Brown if (elem_restr_x) { 5977841947SLeila Ghaffari ierr = DMGetCoordinateDM(dm, &dm_coord); CHKERRQ(ierr); 6077841947SLeila Ghaffari ierr = DMPlexSetClosurePermutationTensor(dm_coord, PETSC_DETERMINE, NULL); 6177841947SLeila Ghaffari CHKERRQ(ierr); 627ed3e4cdSJeremy L Thompson ierr = CreateRestrictionFromPlex(ceed, dm_coord, height, domain_label, value, 637ed3e4cdSJeremy L Thompson elem_restr_x); 647ed3e4cdSJeremy L Thompson CHKERRQ(ierr); 652534dcc8SJed Brown } 662534dcc8SJed Brown if (elem_restr_qd_i) { 672534dcc8SJed Brown CeedElemRestrictionGetNumElements(elem_restr_tmp, &loc_num_elem); 6877841947SLeila Ghaffari CeedElemRestrictionCreateStrided(ceed, loc_num_elem, Q_dim, 6977841947SLeila Ghaffari q_data_size, q_data_size*loc_num_elem*Q_dim, 7077841947SLeila Ghaffari CEED_STRIDES_BACKEND, elem_restr_qd_i); 712534dcc8SJed Brown } 722534dcc8SJed Brown if (!elem_restr_q) CeedElemRestrictionDestroy(&elem_restr_tmp); 7377841947SLeila Ghaffari PetscFunctionReturn(0); 7477841947SLeila Ghaffari } 7577841947SLeila Ghaffari 7677841947SLeila Ghaffari // Utility function to create CEED Composite Operator for the entire domain 7777841947SLeila Ghaffari PetscErrorCode CreateOperatorForDomain(Ceed ceed, DM dm, SimpleBC bc, 7877841947SLeila Ghaffari CeedData ceed_data, Physics phys, 79e334ad8fSJed Brown CeedOperator op_apply_vol, 80e334ad8fSJed Brown CeedOperator op_apply_ijacobian_vol, 81e334ad8fSJed Brown CeedInt height, 82e334ad8fSJed Brown CeedInt P_sur, CeedInt Q_sur, 83e334ad8fSJed Brown CeedInt q_data_size_sur, CeedInt jac_data_size_sur, 84e334ad8fSJed Brown CeedOperator *op_apply, CeedOperator *op_apply_ijacobian) { 852fe7aee7SLeila Ghaffari //CeedInt dim; 8677841947SLeila Ghaffari DMLabel domain_label; 8777841947SLeila Ghaffari PetscErrorCode ierr; 8877841947SLeila Ghaffari PetscFunctionBeginUser; 8977841947SLeila Ghaffari 9077841947SLeila Ghaffari // Create Composite Operaters 9177841947SLeila Ghaffari CeedCompositeOperatorCreate(ceed, op_apply); 92e334ad8fSJed Brown if (op_apply_ijacobian) 93e334ad8fSJed Brown CeedCompositeOperatorCreate(ceed, op_apply_ijacobian); 9477841947SLeila Ghaffari 9577841947SLeila Ghaffari // --Apply Sub-Operator for the volume 9677841947SLeila Ghaffari CeedCompositeOperatorAddSub(*op_apply, op_apply_vol); 97e334ad8fSJed Brown if (op_apply_ijacobian) 98e334ad8fSJed Brown CeedCompositeOperatorAddSub(*op_apply_ijacobian, op_apply_ijacobian_vol); 9977841947SLeila Ghaffari 10077841947SLeila Ghaffari // -- Create Sub-Operator for in/outflow BCs 10188626eedSJames Wright if (phys->has_neumann || 1) { 10277841947SLeila Ghaffari // --- Setup 10377841947SLeila Ghaffari ierr = DMGetLabel(dm, "Face Sets", &domain_label); CHKERRQ(ierr); 1042fe7aee7SLeila Ghaffari //ierr = DMGetDimension(dm, &dim); CHKERRQ(ierr); 10577841947SLeila Ghaffari 10677841947SLeila Ghaffari // --- Get number of quadrature points for the boundaries 10777841947SLeila Ghaffari CeedInt num_qpts_sur; 10877841947SLeila Ghaffari CeedBasisGetNumQuadraturePoints(ceed_data->basis_q_sur, &num_qpts_sur); 10977841947SLeila Ghaffari 1102fe7aee7SLeila Ghaffari // --- Create Sub-Operator for inflow boundaries 1112fe7aee7SLeila Ghaffari for (CeedInt i=0; i < bc->num_inflow; i++) { 11277841947SLeila Ghaffari CeedVector q_data_sur; 113e334ad8fSJed Brown CeedOperator op_setup_sur, op_apply_inflow, 114e334ad8fSJed Brown op_apply_inflow_jacobian = NULL; 1152fe7aee7SLeila Ghaffari CeedElemRestriction elem_restr_x_sur, elem_restr_q_sur, elem_restr_qd_i_sur; 11677841947SLeila Ghaffari 11777841947SLeila Ghaffari // ---- CEED Restriction 1182fe7aee7SLeila Ghaffari ierr = GetRestrictionForDomain(ceed, dm, height, domain_label, bc->inflows[i], 1192fe7aee7SLeila Ghaffari Q_sur, q_data_size_sur, &elem_restr_q_sur, &elem_restr_x_sur, 1202fe7aee7SLeila Ghaffari &elem_restr_qd_i_sur); 12177841947SLeila Ghaffari CHKERRQ(ierr); 12277841947SLeila Ghaffari 12377841947SLeila Ghaffari // ---- CEED Vector 12477841947SLeila Ghaffari PetscInt loc_num_elem_sur; 12577841947SLeila Ghaffari CeedElemRestrictionGetNumElements(elem_restr_q_sur, &loc_num_elem_sur); 12677841947SLeila Ghaffari CeedVectorCreate(ceed, q_data_size_sur*loc_num_elem_sur*num_qpts_sur, 12777841947SLeila Ghaffari &q_data_sur); 12877841947SLeila Ghaffari 12977841947SLeila Ghaffari // ---- CEED Operator 13077841947SLeila Ghaffari // ----- CEED Operator for Setup (geometric factors) 13177841947SLeila Ghaffari CeedOperatorCreate(ceed, ceed_data->qf_setup_sur, NULL, NULL, &op_setup_sur); 13277841947SLeila Ghaffari CeedOperatorSetField(op_setup_sur, "dx", elem_restr_x_sur, 133e6225c47SLeila Ghaffari ceed_data->basis_x_sur, CEED_VECTOR_ACTIVE); 13477841947SLeila Ghaffari CeedOperatorSetField(op_setup_sur, "weight", CEED_ELEMRESTRICTION_NONE, 13577841947SLeila Ghaffari ceed_data->basis_x_sur, CEED_VECTOR_NONE); 136a61c78d6SJeremy L Thompson CeedOperatorSetField(op_setup_sur, "surface qdata", elem_restr_qd_i_sur, 13777841947SLeila Ghaffari CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE); 13877841947SLeila Ghaffari 13977841947SLeila Ghaffari // ----- CEED Operator for Physics 1402fe7aee7SLeila Ghaffari CeedOperatorCreate(ceed, ceed_data->qf_apply_inflow, NULL, NULL, 1412fe7aee7SLeila Ghaffari &op_apply_inflow); 1422fe7aee7SLeila Ghaffari CeedOperatorSetField(op_apply_inflow, "q", elem_restr_q_sur, 14377841947SLeila Ghaffari ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE); 1442fe7aee7SLeila Ghaffari CeedOperatorSetField(op_apply_inflow, "surface qdata", elem_restr_qd_i_sur, 14577841947SLeila Ghaffari CEED_BASIS_COLLOCATED, q_data_sur); 1462fe7aee7SLeila Ghaffari CeedOperatorSetField(op_apply_inflow, "x", elem_restr_x_sur, 14777841947SLeila Ghaffari ceed_data->basis_x_sur, ceed_data->x_coord); 1482fe7aee7SLeila Ghaffari CeedOperatorSetField(op_apply_inflow, "v", elem_restr_q_sur, 14977841947SLeila Ghaffari ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE); 15077841947SLeila Ghaffari 151e334ad8fSJed Brown if (ceed_data->qf_apply_inflow_jacobian) { 152e334ad8fSJed Brown CeedOperatorCreate(ceed, ceed_data->qf_apply_inflow_jacobian, NULL, NULL, 153e334ad8fSJed Brown &op_apply_inflow_jacobian); 154e334ad8fSJed Brown CeedOperatorSetField(op_apply_inflow_jacobian, "dq", elem_restr_q_sur, 155e334ad8fSJed Brown ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE); 156e334ad8fSJed Brown CeedOperatorSetField(op_apply_inflow_jacobian, "surface qdata", 157e334ad8fSJed Brown elem_restr_qd_i_sur, 158e334ad8fSJed Brown CEED_BASIS_COLLOCATED, q_data_sur); 159e334ad8fSJed Brown CeedOperatorSetField(op_apply_inflow_jacobian, "x", elem_restr_x_sur, 160e334ad8fSJed Brown ceed_data->basis_x_sur, ceed_data->x_coord); 161e334ad8fSJed Brown CeedOperatorSetField(op_apply_inflow_jacobian, "v", elem_restr_q_sur, 162e334ad8fSJed Brown ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE); 163e334ad8fSJed Brown } 164e334ad8fSJed Brown 16577841947SLeila Ghaffari // ----- Apply CEED operator for Setup 16677841947SLeila Ghaffari CeedOperatorApply(op_setup_sur, ceed_data->x_coord, q_data_sur, 16777841947SLeila Ghaffari CEED_REQUEST_IMMEDIATE); 16877841947SLeila Ghaffari 1692fe7aee7SLeila Ghaffari // ----- Apply Sub-Operator for Physics 1702fe7aee7SLeila Ghaffari CeedCompositeOperatorAddSub(*op_apply, op_apply_inflow); 171e334ad8fSJed Brown if (op_apply_ijacobian) 172e334ad8fSJed Brown CeedCompositeOperatorAddSub(*op_apply_ijacobian, op_apply_inflow_jacobian); 17377841947SLeila Ghaffari 17477841947SLeila Ghaffari // ----- Cleanup 17577841947SLeila Ghaffari CeedVectorDestroy(&q_data_sur); 17677841947SLeila Ghaffari CeedElemRestrictionDestroy(&elem_restr_q_sur); 17777841947SLeila Ghaffari CeedElemRestrictionDestroy(&elem_restr_x_sur); 17877841947SLeila Ghaffari CeedElemRestrictionDestroy(&elem_restr_qd_i_sur); 17977841947SLeila Ghaffari CeedOperatorDestroy(&op_setup_sur); 1802fe7aee7SLeila Ghaffari CeedOperatorDestroy(&op_apply_inflow); 181e334ad8fSJed Brown CeedOperatorDestroy(&op_apply_inflow_jacobian); 18277841947SLeila Ghaffari } 18377841947SLeila Ghaffari 1842fe7aee7SLeila Ghaffari // --- Create Sub-Operator for outflow boundaries 1852fe7aee7SLeila Ghaffari for (CeedInt i=0; i < bc->num_outflow; i++) { 186e334ad8fSJed Brown CeedVector q_data_sur, jac_data_sur; 187e334ad8fSJed Brown CeedOperator op_setup_sur, op_apply_outflow, 188e334ad8fSJed Brown op_apply_outflow_jacobian = NULL; 189e334ad8fSJed Brown CeedElemRestriction elem_restr_x_sur, elem_restr_q_sur, elem_restr_qd_i_sur, 190e334ad8fSJed Brown elem_restr_jd_i_sur; 1912fe7aee7SLeila Ghaffari 1922fe7aee7SLeila Ghaffari // ---- CEED Restriction 1932fe7aee7SLeila Ghaffari ierr = GetRestrictionForDomain(ceed, dm, height, domain_label, bc->outflows[i], 1942fe7aee7SLeila Ghaffari Q_sur, q_data_size_sur, &elem_restr_q_sur, &elem_restr_x_sur, 1952fe7aee7SLeila Ghaffari &elem_restr_qd_i_sur); 1962fe7aee7SLeila Ghaffari CHKERRQ(ierr); 197*39c69132SJed Brown if (jac_data_size_sur > 0) { 198*39c69132SJed Brown // State-dependent data will be passed from residual to Jacobian. This will be collocated. 199e334ad8fSJed Brown ierr = GetRestrictionForDomain(ceed, dm, height, domain_label, bc->outflows[i], 200e334ad8fSJed Brown Q_sur, jac_data_size_sur, NULL, NULL, 201e334ad8fSJed Brown &elem_restr_jd_i_sur); 202e334ad8fSJed Brown CHKERRQ(ierr); 203*39c69132SJed Brown CeedElemRestrictionCreateVector(elem_restr_jd_i_sur, &jac_data_sur, NULL); 204*39c69132SJed Brown } else { 205*39c69132SJed Brown elem_restr_jd_i_sur = NULL; 206*39c69132SJed Brown jac_data_sur = NULL; 207*39c69132SJed Brown } 2082fe7aee7SLeila Ghaffari 2092fe7aee7SLeila Ghaffari // ---- CEED Vector 2102fe7aee7SLeila Ghaffari PetscInt loc_num_elem_sur; 2112fe7aee7SLeila Ghaffari CeedElemRestrictionGetNumElements(elem_restr_q_sur, &loc_num_elem_sur); 2122fe7aee7SLeila Ghaffari CeedVectorCreate(ceed, q_data_size_sur*loc_num_elem_sur*num_qpts_sur, 2132fe7aee7SLeila Ghaffari &q_data_sur); 2142fe7aee7SLeila Ghaffari 2152fe7aee7SLeila Ghaffari // ---- CEED Operator 2162fe7aee7SLeila Ghaffari // ----- CEED Operator for Setup (geometric factors) 2172fe7aee7SLeila Ghaffari CeedOperatorCreate(ceed, ceed_data->qf_setup_sur, NULL, NULL, &op_setup_sur); 2182fe7aee7SLeila Ghaffari CeedOperatorSetField(op_setup_sur, "dx", elem_restr_x_sur, 2192fe7aee7SLeila Ghaffari ceed_data->basis_x_sur, CEED_VECTOR_ACTIVE); 2202fe7aee7SLeila Ghaffari CeedOperatorSetField(op_setup_sur, "weight", CEED_ELEMRESTRICTION_NONE, 2212fe7aee7SLeila Ghaffari ceed_data->basis_x_sur, CEED_VECTOR_NONE); 2222fe7aee7SLeila Ghaffari CeedOperatorSetField(op_setup_sur, "surface qdata", elem_restr_qd_i_sur, 2232fe7aee7SLeila Ghaffari CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE); 2242fe7aee7SLeila Ghaffari 2252fe7aee7SLeila Ghaffari // ----- CEED Operator for Physics 2262fe7aee7SLeila Ghaffari CeedOperatorCreate(ceed, ceed_data->qf_apply_outflow, NULL, NULL, 2272fe7aee7SLeila Ghaffari &op_apply_outflow); 2282fe7aee7SLeila Ghaffari CeedOperatorSetField(op_apply_outflow, "q", elem_restr_q_sur, 2292fe7aee7SLeila Ghaffari ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE); 2302fe7aee7SLeila Ghaffari CeedOperatorSetField(op_apply_outflow, "surface qdata", elem_restr_qd_i_sur, 2312fe7aee7SLeila Ghaffari CEED_BASIS_COLLOCATED, q_data_sur); 2322fe7aee7SLeila Ghaffari CeedOperatorSetField(op_apply_outflow, "x", elem_restr_x_sur, 2332fe7aee7SLeila Ghaffari ceed_data->basis_x_sur, ceed_data->x_coord); 2342fe7aee7SLeila Ghaffari CeedOperatorSetField(op_apply_outflow, "v", elem_restr_q_sur, 2352fe7aee7SLeila Ghaffari ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE); 236*39c69132SJed Brown if (elem_restr_jd_i_sur) 237e334ad8fSJed Brown CeedOperatorSetField(op_apply_outflow, "surface jacobian data", 238e334ad8fSJed Brown elem_restr_jd_i_sur, 239e334ad8fSJed Brown CEED_BASIS_COLLOCATED, jac_data_sur); 240e334ad8fSJed Brown 241e334ad8fSJed Brown if (ceed_data->qf_apply_outflow_jacobian) { 242e334ad8fSJed Brown CeedOperatorCreate(ceed, ceed_data->qf_apply_outflow_jacobian, NULL, NULL, 243e334ad8fSJed Brown &op_apply_outflow_jacobian); 244e334ad8fSJed Brown CeedOperatorSetField(op_apply_outflow_jacobian, "dq", elem_restr_q_sur, 245e334ad8fSJed Brown ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE); 246e334ad8fSJed Brown CeedOperatorSetField(op_apply_outflow_jacobian, "surface qdata", 247e334ad8fSJed Brown elem_restr_qd_i_sur, 248e334ad8fSJed Brown CEED_BASIS_COLLOCATED, q_data_sur); 249e334ad8fSJed Brown CeedOperatorSetField(op_apply_outflow_jacobian, "surface jacobian data", 250e334ad8fSJed Brown elem_restr_jd_i_sur, 251e334ad8fSJed Brown CEED_BASIS_COLLOCATED, jac_data_sur); 252e334ad8fSJed Brown CeedOperatorSetField(op_apply_outflow_jacobian, "v", elem_restr_q_sur, 253e334ad8fSJed Brown ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE); 254e334ad8fSJed Brown } 2552fe7aee7SLeila Ghaffari 2562fe7aee7SLeila Ghaffari // ----- Apply CEED operator for Setup 2572fe7aee7SLeila Ghaffari CeedOperatorApply(op_setup_sur, ceed_data->x_coord, q_data_sur, 2582fe7aee7SLeila Ghaffari CEED_REQUEST_IMMEDIATE); 2592fe7aee7SLeila Ghaffari 2602fe7aee7SLeila Ghaffari // ----- Apply Sub-Operator for Physics 2612fe7aee7SLeila Ghaffari CeedCompositeOperatorAddSub(*op_apply, op_apply_outflow); 262e334ad8fSJed Brown if (op_apply_ijacobian) 263e334ad8fSJed Brown CeedCompositeOperatorAddSub(*op_apply_ijacobian, op_apply_outflow_jacobian); 2642fe7aee7SLeila Ghaffari 2652fe7aee7SLeila Ghaffari // ----- Cleanup 2662fe7aee7SLeila Ghaffari CeedVectorDestroy(&q_data_sur); 267e334ad8fSJed Brown CeedVectorDestroy(&jac_data_sur); 2682fe7aee7SLeila Ghaffari CeedElemRestrictionDestroy(&elem_restr_q_sur); 2692fe7aee7SLeila Ghaffari CeedElemRestrictionDestroy(&elem_restr_x_sur); 2702fe7aee7SLeila Ghaffari CeedElemRestrictionDestroy(&elem_restr_qd_i_sur); 271e334ad8fSJed Brown CeedElemRestrictionDestroy(&elem_restr_jd_i_sur); 2722fe7aee7SLeila Ghaffari CeedOperatorDestroy(&op_setup_sur); 2732fe7aee7SLeila Ghaffari CeedOperatorDestroy(&op_apply_outflow); 274e334ad8fSJed Brown CeedOperatorDestroy(&op_apply_outflow_jacobian); 2752fe7aee7SLeila Ghaffari } 2762fe7aee7SLeila Ghaffari } 27711436a05SJames Wright 27811436a05SJames Wright // ----- Get Context Labels for Operator 27911436a05SJames Wright CeedOperatorContextGetFieldLabel(*op_apply, "solution time", 28011436a05SJames Wright &phys->solution_time_label); 28188626eedSJames Wright CeedOperatorContextGetFieldLabel(*op_apply, "timestep size", 28288626eedSJames Wright &phys->timestep_size_label); 28311436a05SJames Wright 28477841947SLeila Ghaffari PetscFunctionReturn(0); 28577841947SLeila Ghaffari } 28677841947SLeila Ghaffari 28777841947SLeila Ghaffari PetscErrorCode SetupLibceed(Ceed ceed, CeedData ceed_data, DM dm, User user, 288c95f9967SJed Brown AppCtx app_ctx, ProblemData *problem, SimpleBC bc) { 28977841947SLeila Ghaffari PetscErrorCode ierr; 29077841947SLeila Ghaffari PetscFunctionBeginUser; 29177841947SLeila Ghaffari 29277841947SLeila Ghaffari // ***************************************************************************** 29377841947SLeila Ghaffari // Set up CEED objects for the interior domain (volume) 29477841947SLeila Ghaffari // ***************************************************************************** 29577841947SLeila Ghaffari const PetscInt num_comp_q = 5; 29677841947SLeila Ghaffari const CeedInt dim = problem->dim, 29777841947SLeila Ghaffari num_comp_x = problem->dim, 29877841947SLeila Ghaffari q_data_size_vol = problem->q_data_size_vol, 299e334ad8fSJed Brown jac_data_size_vol = num_comp_q + 6 + 3, 30077841947SLeila Ghaffari P = app_ctx->degree + 1, 30177841947SLeila Ghaffari Q = P + app_ctx->q_extra; 302a3ae0734SJed Brown CeedElemRestriction elem_restr_jd_i; 303a3ae0734SJed Brown CeedVector jac_data; 30477841947SLeila Ghaffari 30577841947SLeila Ghaffari // ----------------------------------------------------------------------------- 30677841947SLeila Ghaffari // CEED Bases 30777841947SLeila Ghaffari // ----------------------------------------------------------------------------- 30877841947SLeila Ghaffari CeedBasisCreateTensorH1Lagrange(ceed, dim, num_comp_q, P, Q, CEED_GAUSS, 30977841947SLeila Ghaffari &ceed_data->basis_q); 31077841947SLeila Ghaffari CeedBasisCreateTensorH1Lagrange(ceed, dim, num_comp_x, 2, Q, CEED_GAUSS, 31177841947SLeila Ghaffari &ceed_data->basis_x); 31277841947SLeila Ghaffari CeedBasisCreateTensorH1Lagrange(ceed, dim, num_comp_x, 2, P, 31377841947SLeila Ghaffari CEED_GAUSS_LOBATTO, &ceed_data->basis_xc); 31477841947SLeila Ghaffari 31577841947SLeila Ghaffari // ----------------------------------------------------------------------------- 31677841947SLeila Ghaffari // CEED Restrictions 31777841947SLeila Ghaffari // ----------------------------------------------------------------------------- 31877841947SLeila Ghaffari // -- Create restriction 3197ed3e4cdSJeremy L Thompson ierr = GetRestrictionForDomain(ceed, dm, 0, 0, 0, Q, q_data_size_vol, 3207ed3e4cdSJeremy L Thompson &ceed_data->elem_restr_q, &ceed_data->elem_restr_x, 32177841947SLeila Ghaffari &ceed_data->elem_restr_qd_i); CHKERRQ(ierr); 322a3ae0734SJed Brown 323a3ae0734SJed Brown ierr = GetRestrictionForDomain(ceed, dm, 0, 0, 0, Q, jac_data_size_vol, 324a3ae0734SJed Brown NULL, NULL, 325a3ae0734SJed Brown &elem_restr_jd_i); CHKERRQ(ierr); 32677841947SLeila Ghaffari // -- Create E vectors 32777841947SLeila Ghaffari CeedElemRestrictionCreateVector(ceed_data->elem_restr_q, &user->q_ceed, NULL); 32877841947SLeila Ghaffari CeedElemRestrictionCreateVector(ceed_data->elem_restr_q, &user->q_dot_ceed, 32977841947SLeila Ghaffari NULL); 33077841947SLeila Ghaffari CeedElemRestrictionCreateVector(ceed_data->elem_restr_q, &user->g_ceed, NULL); 33177841947SLeila Ghaffari 33277841947SLeila Ghaffari // ----------------------------------------------------------------------------- 33377841947SLeila Ghaffari // CEED QFunctions 33477841947SLeila Ghaffari // ----------------------------------------------------------------------------- 33577841947SLeila Ghaffari // -- Create QFunction for quadrature data 33691e5af17SJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->setup_vol.qfunction, 33791e5af17SJed Brown problem->setup_vol.qfunction_loc, 33877841947SLeila Ghaffari &ceed_data->qf_setup_vol); 339841e4c73SJed Brown if (problem->setup_vol.qfunction_context) { 340841e4c73SJed Brown CeedQFunctionSetContext(ceed_data->qf_setup_vol, 341841e4c73SJed Brown problem->setup_vol.qfunction_context); 342841e4c73SJed Brown CeedQFunctionContextDestroy(&problem->setup_vol.qfunction_context); 343841e4c73SJed Brown } 34477841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_setup_vol, "dx", num_comp_x*dim, 34577841947SLeila Ghaffari CEED_EVAL_GRAD); 34677841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_setup_vol, "weight", 1, CEED_EVAL_WEIGHT); 347a61c78d6SJeremy L Thompson CeedQFunctionAddOutput(ceed_data->qf_setup_vol, "qdata", q_data_size_vol, 34877841947SLeila Ghaffari CEED_EVAL_NONE); 34977841947SLeila Ghaffari 35077841947SLeila Ghaffari // -- Create QFunction for ICs 35191e5af17SJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->ics.qfunction, 35291e5af17SJed Brown problem->ics.qfunction_loc, 35377841947SLeila Ghaffari &ceed_data->qf_ics); 354841e4c73SJed Brown CeedQFunctionSetContext(ceed_data->qf_ics, problem->ics.qfunction_context); 355841e4c73SJed Brown CeedQFunctionContextDestroy(&problem->ics.qfunction_context); 35677841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_ics, "x", num_comp_x, CEED_EVAL_INTERP); 35777841947SLeila Ghaffari CeedQFunctionAddOutput(ceed_data->qf_ics, "q0", num_comp_q, CEED_EVAL_NONE); 35877841947SLeila Ghaffari 35977841947SLeila Ghaffari // -- Create QFunction for RHS 36091e5af17SJed Brown if (problem->apply_vol_rhs.qfunction) { 36191e5af17SJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->apply_vol_rhs.qfunction, 36291e5af17SJed Brown problem->apply_vol_rhs.qfunction_loc, &ceed_data->qf_rhs_vol); 363841e4c73SJed Brown CeedQFunctionSetContext(ceed_data->qf_rhs_vol, 364841e4c73SJed Brown problem->apply_vol_rhs.qfunction_context); 365841e4c73SJed Brown CeedQFunctionContextDestroy(&problem->apply_vol_rhs.qfunction_context); 36677841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_rhs_vol, "q", num_comp_q, CEED_EVAL_INTERP); 367a3ae0734SJed Brown CeedQFunctionAddInput(ceed_data->qf_rhs_vol, "Grad_q", num_comp_q*dim, 36877841947SLeila Ghaffari CEED_EVAL_GRAD); 369a61c78d6SJeremy L Thompson CeedQFunctionAddInput(ceed_data->qf_rhs_vol, "qdata", q_data_size_vol, 37077841947SLeila Ghaffari CEED_EVAL_NONE); 37177841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_rhs_vol, "x", num_comp_x, CEED_EVAL_INTERP); 37277841947SLeila Ghaffari CeedQFunctionAddOutput(ceed_data->qf_rhs_vol, "v", num_comp_q, 37377841947SLeila Ghaffari CEED_EVAL_INTERP); 374a3ae0734SJed Brown CeedQFunctionAddOutput(ceed_data->qf_rhs_vol, "Grad_v", num_comp_q*dim, 37577841947SLeila Ghaffari CEED_EVAL_GRAD); 37677841947SLeila Ghaffari } 37777841947SLeila Ghaffari 37877841947SLeila Ghaffari // -- Create QFunction for IFunction 37991e5af17SJed Brown if (problem->apply_vol_ifunction.qfunction) { 38091e5af17SJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->apply_vol_ifunction.qfunction, 38191e5af17SJed Brown problem->apply_vol_ifunction.qfunction_loc, &ceed_data->qf_ifunction_vol); 382841e4c73SJed Brown CeedQFunctionSetContext(ceed_data->qf_ifunction_vol, 383841e4c73SJed Brown problem->apply_vol_ifunction.qfunction_context); 384841e4c73SJed Brown CeedQFunctionContextDestroy(&problem->apply_vol_ifunction.qfunction_context); 38577841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "q", num_comp_q, 38677841947SLeila Ghaffari CEED_EVAL_INTERP); 387a3ae0734SJed Brown CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "Grad_q", num_comp_q*dim, 38877841947SLeila Ghaffari CEED_EVAL_GRAD); 389a61c78d6SJeremy L Thompson CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "q dot", num_comp_q, 39077841947SLeila Ghaffari CEED_EVAL_INTERP); 391a61c78d6SJeremy L Thompson CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "qdata", q_data_size_vol, 39277841947SLeila Ghaffari CEED_EVAL_NONE); 39377841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "x", num_comp_x, 39477841947SLeila Ghaffari CEED_EVAL_INTERP); 39577841947SLeila Ghaffari CeedQFunctionAddOutput(ceed_data->qf_ifunction_vol, "v", num_comp_q, 39677841947SLeila Ghaffari CEED_EVAL_INTERP); 397a3ae0734SJed Brown CeedQFunctionAddOutput(ceed_data->qf_ifunction_vol, "Grad_v", num_comp_q*dim, 39877841947SLeila Ghaffari CEED_EVAL_GRAD); 399a3ae0734SJed Brown CeedQFunctionAddOutput(ceed_data->qf_ifunction_vol, "jac_data", 400a3ae0734SJed Brown jac_data_size_vol, CEED_EVAL_NONE); 40177841947SLeila Ghaffari } 40277841947SLeila Ghaffari 403e334ad8fSJed Brown CeedQFunction qf_ijacobian_vol = NULL; 404e334ad8fSJed Brown if (problem->apply_vol_ijacobian.qfunction) { 405e334ad8fSJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->apply_vol_ijacobian.qfunction, 406e334ad8fSJed Brown problem->apply_vol_ijacobian.qfunction_loc, &qf_ijacobian_vol); 407e334ad8fSJed Brown CeedQFunctionSetContext(qf_ijacobian_vol, 408e334ad8fSJed Brown problem->apply_vol_ijacobian.qfunction_context); 409e334ad8fSJed Brown CeedQFunctionContextDestroy(&problem->apply_vol_ijacobian.qfunction_context); 410e334ad8fSJed Brown CeedQFunctionAddInput(qf_ijacobian_vol, "dq", num_comp_q, 411e334ad8fSJed Brown CEED_EVAL_INTERP); 412e334ad8fSJed Brown CeedQFunctionAddInput(qf_ijacobian_vol, "Grad_dq", num_comp_q*dim, 413e334ad8fSJed Brown CEED_EVAL_GRAD); 414e334ad8fSJed Brown CeedQFunctionAddInput(qf_ijacobian_vol, "qdata", q_data_size_vol, 415e334ad8fSJed Brown CEED_EVAL_NONE); 416e334ad8fSJed Brown CeedQFunctionAddInput(qf_ijacobian_vol, "x", num_comp_x, 417e334ad8fSJed Brown CEED_EVAL_INTERP); 418e334ad8fSJed Brown CeedQFunctionAddInput(qf_ijacobian_vol, "jac_data", 419e334ad8fSJed Brown jac_data_size_vol, CEED_EVAL_NONE); 420e334ad8fSJed Brown CeedQFunctionAddOutput(qf_ijacobian_vol, "v", num_comp_q, 421e334ad8fSJed Brown CEED_EVAL_INTERP); 422e334ad8fSJed Brown CeedQFunctionAddOutput(qf_ijacobian_vol, "Grad_v", num_comp_q*dim, 423e334ad8fSJed Brown CEED_EVAL_GRAD); 424e334ad8fSJed Brown } 425e334ad8fSJed Brown 42677841947SLeila Ghaffari // --------------------------------------------------------------------------- 42777841947SLeila Ghaffari // Element coordinates 42877841947SLeila Ghaffari // --------------------------------------------------------------------------- 42977841947SLeila Ghaffari // -- Create CEED vector 43077841947SLeila Ghaffari CeedElemRestrictionCreateVector(ceed_data->elem_restr_x, &ceed_data->x_coord, 43177841947SLeila Ghaffari NULL); 43277841947SLeila Ghaffari 43377841947SLeila Ghaffari // -- Copy PETSc vector in CEED vector 43477841947SLeila Ghaffari Vec X_loc; 43577841947SLeila Ghaffari const PetscScalar *X_loc_array; 43677841947SLeila Ghaffari ierr = DMGetCoordinatesLocal(dm, &X_loc); CHKERRQ(ierr); 4371864f1c2SLeila Ghaffari ierr = VecScale(X_loc, problem->dm_scale); CHKERRQ(ierr); 43877841947SLeila Ghaffari ierr = VecGetArrayRead(X_loc, &X_loc_array); CHKERRQ(ierr); 43977841947SLeila Ghaffari CeedVectorSetArray(ceed_data->x_coord, CEED_MEM_HOST, CEED_COPY_VALUES, 44077841947SLeila Ghaffari (PetscScalar *)X_loc_array); 44177841947SLeila Ghaffari ierr = VecRestoreArrayRead(X_loc, &X_loc_array); CHKERRQ(ierr); 44277841947SLeila Ghaffari 44377841947SLeila Ghaffari // ----------------------------------------------------------------------------- 44477841947SLeila Ghaffari // CEED vectors 44577841947SLeila Ghaffari // ----------------------------------------------------------------------------- 44677841947SLeila Ghaffari // -- Create CEED vector for geometric data 44777841947SLeila Ghaffari CeedInt num_qpts_vol; 44877841947SLeila Ghaffari PetscInt loc_num_elem_vol; 44977841947SLeila Ghaffari CeedBasisGetNumQuadraturePoints(ceed_data->basis_q, &num_qpts_vol); 45077841947SLeila Ghaffari CeedElemRestrictionGetNumElements(ceed_data->elem_restr_q, &loc_num_elem_vol); 45177841947SLeila Ghaffari CeedVectorCreate(ceed, q_data_size_vol*loc_num_elem_vol*num_qpts_vol, 45277841947SLeila Ghaffari &ceed_data->q_data); 45377841947SLeila Ghaffari 454a3ae0734SJed Brown CeedElemRestrictionCreateVector(elem_restr_jd_i, &jac_data, NULL); 45577841947SLeila Ghaffari // ----------------------------------------------------------------------------- 45677841947SLeila Ghaffari // CEED Operators 45777841947SLeila Ghaffari // ----------------------------------------------------------------------------- 45877841947SLeila Ghaffari // -- Create CEED operator for quadrature data 45977841947SLeila Ghaffari CeedOperatorCreate(ceed, ceed_data->qf_setup_vol, NULL, NULL, 46077841947SLeila Ghaffari &ceed_data->op_setup_vol); 46177841947SLeila Ghaffari CeedOperatorSetField(ceed_data->op_setup_vol, "dx", ceed_data->elem_restr_x, 46277841947SLeila Ghaffari ceed_data->basis_x, CEED_VECTOR_ACTIVE); 46377841947SLeila Ghaffari CeedOperatorSetField(ceed_data->op_setup_vol, "weight", 464e6225c47SLeila Ghaffari CEED_ELEMRESTRICTION_NONE, ceed_data->basis_x, CEED_VECTOR_NONE); 465a61c78d6SJeremy L Thompson CeedOperatorSetField(ceed_data->op_setup_vol, "qdata", 466e6225c47SLeila Ghaffari ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE); 46777841947SLeila Ghaffari 46877841947SLeila Ghaffari // -- Create CEED operator for ICs 46977841947SLeila Ghaffari CeedOperatorCreate(ceed, ceed_data->qf_ics, NULL, NULL, &ceed_data->op_ics); 47077841947SLeila Ghaffari CeedOperatorSetField(ceed_data->op_ics, "x", ceed_data->elem_restr_x, 47177841947SLeila Ghaffari ceed_data->basis_xc, CEED_VECTOR_ACTIVE); 47277841947SLeila Ghaffari CeedOperatorSetField(ceed_data->op_ics, "q0", ceed_data->elem_restr_q, 47377841947SLeila Ghaffari CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE); 474841e4c73SJed Brown CeedOperatorContextGetFieldLabel(ceed_data->op_ics, "evaluation time", 475841e4c73SJed Brown &user->phys->ics_time_label); 47677841947SLeila Ghaffari 47777841947SLeila Ghaffari // Create CEED operator for RHS 47877841947SLeila Ghaffari if (ceed_data->qf_rhs_vol) { 47977841947SLeila Ghaffari CeedOperator op; 48077841947SLeila Ghaffari CeedOperatorCreate(ceed, ceed_data->qf_rhs_vol, NULL, NULL, &op); 48177841947SLeila Ghaffari CeedOperatorSetField(op, "q", ceed_data->elem_restr_q, ceed_data->basis_q, 48277841947SLeila Ghaffari CEED_VECTOR_ACTIVE); 483a3ae0734SJed Brown CeedOperatorSetField(op, "Grad_q", ceed_data->elem_restr_q, ceed_data->basis_q, 48477841947SLeila Ghaffari CEED_VECTOR_ACTIVE); 485a61c78d6SJeremy L Thompson CeedOperatorSetField(op, "qdata", ceed_data->elem_restr_qd_i, 486e6225c47SLeila Ghaffari CEED_BASIS_COLLOCATED, ceed_data->q_data); 48777841947SLeila Ghaffari CeedOperatorSetField(op, "x", ceed_data->elem_restr_x, ceed_data->basis_x, 48877841947SLeila Ghaffari ceed_data->x_coord); 48977841947SLeila Ghaffari CeedOperatorSetField(op, "v", ceed_data->elem_restr_q, ceed_data->basis_q, 49077841947SLeila Ghaffari CEED_VECTOR_ACTIVE); 491a3ae0734SJed Brown CeedOperatorSetField(op, "Grad_v", ceed_data->elem_restr_q, ceed_data->basis_q, 49277841947SLeila Ghaffari CEED_VECTOR_ACTIVE); 49377841947SLeila Ghaffari user->op_rhs_vol = op; 49477841947SLeila Ghaffari } 49577841947SLeila Ghaffari 49677841947SLeila Ghaffari // -- CEED operator for IFunction 49777841947SLeila Ghaffari if (ceed_data->qf_ifunction_vol) { 49877841947SLeila Ghaffari CeedOperator op; 49977841947SLeila Ghaffari CeedOperatorCreate(ceed, ceed_data->qf_ifunction_vol, NULL, NULL, &op); 50077841947SLeila Ghaffari CeedOperatorSetField(op, "q", ceed_data->elem_restr_q, ceed_data->basis_q, 50177841947SLeila Ghaffari CEED_VECTOR_ACTIVE); 502a3ae0734SJed Brown CeedOperatorSetField(op, "Grad_q", ceed_data->elem_restr_q, ceed_data->basis_q, 50377841947SLeila Ghaffari CEED_VECTOR_ACTIVE); 504a61c78d6SJeremy L Thompson CeedOperatorSetField(op, "q dot", ceed_data->elem_restr_q, ceed_data->basis_q, 50577841947SLeila Ghaffari user->q_dot_ceed); 506a61c78d6SJeremy L Thompson CeedOperatorSetField(op, "qdata", ceed_data->elem_restr_qd_i, 507e6225c47SLeila Ghaffari CEED_BASIS_COLLOCATED, ceed_data->q_data); 50877841947SLeila Ghaffari CeedOperatorSetField(op, "x", ceed_data->elem_restr_x, ceed_data->basis_x, 50977841947SLeila Ghaffari ceed_data->x_coord); 51077841947SLeila Ghaffari CeedOperatorSetField(op, "v", ceed_data->elem_restr_q, ceed_data->basis_q, 51177841947SLeila Ghaffari CEED_VECTOR_ACTIVE); 512a3ae0734SJed Brown CeedOperatorSetField(op, "Grad_v", ceed_data->elem_restr_q, ceed_data->basis_q, 51377841947SLeila Ghaffari CEED_VECTOR_ACTIVE); 514a3ae0734SJed Brown CeedOperatorSetField(op, "jac_data", elem_restr_jd_i, 515a3ae0734SJed Brown CEED_BASIS_COLLOCATED, jac_data); 516a3ae0734SJed Brown 51777841947SLeila Ghaffari user->op_ifunction_vol = op; 51877841947SLeila Ghaffari } 51977841947SLeila Ghaffari 520e334ad8fSJed Brown CeedOperator op_ijacobian_vol = NULL; 521e334ad8fSJed Brown if (qf_ijacobian_vol) { 522e334ad8fSJed Brown CeedOperator op; 523e334ad8fSJed Brown CeedOperatorCreate(ceed, qf_ijacobian_vol, NULL, NULL, &op); 524e334ad8fSJed Brown CeedOperatorSetField(op, "dq", ceed_data->elem_restr_q, ceed_data->basis_q, 525e334ad8fSJed Brown CEED_VECTOR_ACTIVE); 526e334ad8fSJed Brown CeedOperatorSetField(op, "Grad_dq", ceed_data->elem_restr_q, ceed_data->basis_q, 527e334ad8fSJed Brown CEED_VECTOR_ACTIVE); 528e334ad8fSJed Brown CeedOperatorSetField(op, "qdata", ceed_data->elem_restr_qd_i, 529e334ad8fSJed Brown CEED_BASIS_COLLOCATED, ceed_data->q_data); 530e334ad8fSJed Brown CeedOperatorSetField(op, "x", ceed_data->elem_restr_x, ceed_data->basis_x, 531e334ad8fSJed Brown ceed_data->x_coord); 532e334ad8fSJed Brown CeedOperatorSetField(op, "jac_data", elem_restr_jd_i, 533e334ad8fSJed Brown CEED_BASIS_COLLOCATED, jac_data); 534e334ad8fSJed Brown CeedOperatorSetField(op, "v", ceed_data->elem_restr_q, ceed_data->basis_q, 535e334ad8fSJed Brown CEED_VECTOR_ACTIVE); 536e334ad8fSJed Brown CeedOperatorSetField(op, "Grad_v", ceed_data->elem_restr_q, ceed_data->basis_q, 537e334ad8fSJed Brown CEED_VECTOR_ACTIVE); 538e334ad8fSJed Brown op_ijacobian_vol = op; 539e334ad8fSJed Brown CeedQFunctionDestroy(&qf_ijacobian_vol); 540e334ad8fSJed Brown } 541e334ad8fSJed Brown 54277841947SLeila Ghaffari // ***************************************************************************** 54377841947SLeila Ghaffari // Set up CEED objects for the exterior domain (surface) 54477841947SLeila Ghaffari // ***************************************************************************** 54577841947SLeila Ghaffari CeedInt height = 1, 54677841947SLeila Ghaffari dim_sur = dim - height, 54777841947SLeila Ghaffari P_sur = app_ctx->degree + 1, 54877841947SLeila Ghaffari Q_sur = P_sur + app_ctx->q_extra; 549e334ad8fSJed Brown const CeedInt q_data_size_sur = problem->q_data_size_sur, 550e334ad8fSJed Brown jac_data_size_sur = problem->jac_data_size_sur; 55177841947SLeila Ghaffari 55277841947SLeila Ghaffari // ----------------------------------------------------------------------------- 55377841947SLeila Ghaffari // CEED Bases 55477841947SLeila Ghaffari // ----------------------------------------------------------------------------- 55577841947SLeila Ghaffari CeedBasisCreateTensorH1Lagrange(ceed, dim_sur, num_comp_q, P_sur, Q_sur, 55677841947SLeila Ghaffari CEED_GAUSS, &ceed_data->basis_q_sur); 55777841947SLeila Ghaffari CeedBasisCreateTensorH1Lagrange(ceed, dim_sur, num_comp_x, 2, Q_sur, CEED_GAUSS, 55877841947SLeila Ghaffari &ceed_data->basis_x_sur); 55977841947SLeila Ghaffari 56077841947SLeila Ghaffari // ----------------------------------------------------------------------------- 56177841947SLeila Ghaffari // CEED QFunctions 56277841947SLeila Ghaffari // ----------------------------------------------------------------------------- 56377841947SLeila Ghaffari // -- Create QFunction for quadrature data 56491e5af17SJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->setup_sur.qfunction, 56591e5af17SJed Brown problem->setup_sur.qfunction_loc, 56677841947SLeila Ghaffari &ceed_data->qf_setup_sur); 567841e4c73SJed Brown if (problem->setup_sur.qfunction_context) { 568841e4c73SJed Brown CeedQFunctionSetContext(ceed_data->qf_setup_sur, 569841e4c73SJed Brown problem->setup_sur.qfunction_context); 570841e4c73SJed Brown CeedQFunctionContextDestroy(&problem->setup_sur.qfunction_context); 571841e4c73SJed Brown } 57277841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_setup_sur, "dx", num_comp_x*dim_sur, 57377841947SLeila Ghaffari CEED_EVAL_GRAD); 57477841947SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_setup_sur, "weight", 1, CEED_EVAL_WEIGHT); 575a61c78d6SJeremy L Thompson CeedQFunctionAddOutput(ceed_data->qf_setup_sur, "surface qdata", 5762fe7aee7SLeila Ghaffari q_data_size_sur, CEED_EVAL_NONE); 57777841947SLeila Ghaffari 5782fe7aee7SLeila Ghaffari // -- Creat QFunction for inflow boundaries 57991e5af17SJed Brown if (problem->apply_inflow.qfunction) { 58091e5af17SJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->apply_inflow.qfunction, 58191e5af17SJed Brown problem->apply_inflow.qfunction_loc, &ceed_data->qf_apply_inflow); 582841e4c73SJed Brown CeedQFunctionSetContext(ceed_data->qf_apply_inflow, 583841e4c73SJed Brown problem->apply_inflow.qfunction_context); 584841e4c73SJed Brown CeedQFunctionContextDestroy(&problem->apply_inflow.qfunction_context); 5852fe7aee7SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_apply_inflow, "q", num_comp_q, 58677841947SLeila Ghaffari CEED_EVAL_INTERP); 5872fe7aee7SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_apply_inflow, "surface qdata", 5882fe7aee7SLeila Ghaffari q_data_size_sur, CEED_EVAL_NONE); 5892fe7aee7SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_apply_inflow, "x", num_comp_x, 59077841947SLeila Ghaffari CEED_EVAL_INTERP); 5912fe7aee7SLeila Ghaffari CeedQFunctionAddOutput(ceed_data->qf_apply_inflow, "v", num_comp_q, 5922fe7aee7SLeila Ghaffari CEED_EVAL_INTERP); 5932fe7aee7SLeila Ghaffari } 594e334ad8fSJed Brown if (problem->apply_inflow_jacobian.qfunction) { 595e334ad8fSJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->apply_inflow_jacobian.qfunction, 596e334ad8fSJed Brown problem->apply_inflow_jacobian.qfunction_loc, 597e334ad8fSJed Brown &ceed_data->qf_apply_inflow_jacobian); 598e334ad8fSJed Brown CeedQFunctionSetContext(ceed_data->qf_apply_inflow_jacobian, 599e334ad8fSJed Brown problem->apply_inflow_jacobian.qfunction_context); 600e334ad8fSJed Brown CeedQFunctionContextDestroy(&problem->apply_inflow_jacobian.qfunction_context); 601e334ad8fSJed Brown CeedQFunctionAddInput(ceed_data->qf_apply_inflow_jacobian, "dq", num_comp_q, 602e334ad8fSJed Brown CEED_EVAL_INTERP); 603e334ad8fSJed Brown CeedQFunctionAddInput(ceed_data->qf_apply_inflow_jacobian, "surface qdata", 604e334ad8fSJed Brown q_data_size_sur, CEED_EVAL_NONE); 605e334ad8fSJed Brown CeedQFunctionAddInput(ceed_data->qf_apply_inflow_jacobian, "x", num_comp_x, 606e334ad8fSJed Brown CEED_EVAL_INTERP); 607e334ad8fSJed Brown CeedQFunctionAddOutput(ceed_data->qf_apply_inflow_jacobian, "v", num_comp_q, 608e334ad8fSJed Brown CEED_EVAL_INTERP); 609e334ad8fSJed Brown } 6102fe7aee7SLeila Ghaffari 6112fe7aee7SLeila Ghaffari // -- Creat QFunction for outflow boundaries 61291e5af17SJed Brown if (problem->apply_outflow.qfunction) { 61391e5af17SJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->apply_outflow.qfunction, 61491e5af17SJed Brown problem->apply_outflow.qfunction_loc, &ceed_data->qf_apply_outflow); 615841e4c73SJed Brown CeedQFunctionSetContext(ceed_data->qf_apply_outflow, 616841e4c73SJed Brown problem->apply_outflow.qfunction_context); 617841e4c73SJed Brown CeedQFunctionContextDestroy(&problem->apply_outflow.qfunction_context); 6182fe7aee7SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_apply_outflow, "q", num_comp_q, 6192fe7aee7SLeila Ghaffari CEED_EVAL_INTERP); 6202fe7aee7SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_apply_outflow, "surface qdata", 6212fe7aee7SLeila Ghaffari q_data_size_sur, CEED_EVAL_NONE); 6222fe7aee7SLeila Ghaffari CeedQFunctionAddInput(ceed_data->qf_apply_outflow, "x", num_comp_x, 6232fe7aee7SLeila Ghaffari CEED_EVAL_INTERP); 6242fe7aee7SLeila Ghaffari CeedQFunctionAddOutput(ceed_data->qf_apply_outflow, "v", num_comp_q, 62577841947SLeila Ghaffari CEED_EVAL_INTERP); 626*39c69132SJed Brown if (jac_data_size_sur) 627e334ad8fSJed Brown CeedQFunctionAddOutput(ceed_data->qf_apply_outflow, "surface jacobian data", 628e334ad8fSJed Brown jac_data_size_sur, 629e334ad8fSJed Brown CEED_EVAL_NONE); 630e334ad8fSJed Brown } 631e334ad8fSJed Brown if (problem->apply_outflow_jacobian.qfunction) { 632e334ad8fSJed Brown CeedQFunctionCreateInterior(ceed, 1, problem->apply_outflow_jacobian.qfunction, 633e334ad8fSJed Brown problem->apply_outflow_jacobian.qfunction_loc, 634e334ad8fSJed Brown &ceed_data->qf_apply_outflow_jacobian); 635e334ad8fSJed Brown CeedQFunctionSetContext(ceed_data->qf_apply_outflow_jacobian, 636e334ad8fSJed Brown problem->apply_outflow_jacobian.qfunction_context); 637e334ad8fSJed Brown CeedQFunctionContextDestroy(&problem->apply_outflow_jacobian.qfunction_context); 638e334ad8fSJed Brown CeedQFunctionAddInput(ceed_data->qf_apply_outflow_jacobian, "dq", num_comp_q, 639e334ad8fSJed Brown CEED_EVAL_INTERP); 640e334ad8fSJed Brown CeedQFunctionAddInput(ceed_data->qf_apply_outflow_jacobian, "surface qdata", 641e334ad8fSJed Brown q_data_size_sur, CEED_EVAL_NONE); 642e334ad8fSJed Brown CeedQFunctionAddInput(ceed_data->qf_apply_outflow_jacobian, 643e334ad8fSJed Brown "surface jacobian data", 644e334ad8fSJed Brown jac_data_size_sur, CEED_EVAL_NONE); 645e334ad8fSJed Brown CeedQFunctionAddOutput(ceed_data->qf_apply_outflow_jacobian, "v", num_comp_q, 646e334ad8fSJed Brown CEED_EVAL_INTERP); 64777841947SLeila Ghaffari } 64877841947SLeila Ghaffari 64977841947SLeila Ghaffari // ***************************************************************************** 65077841947SLeila Ghaffari // CEED Operator Apply 65177841947SLeila Ghaffari // ***************************************************************************** 65277841947SLeila Ghaffari // -- Apply CEED Operator for the geometric data 65377841947SLeila Ghaffari CeedOperatorApply(ceed_data->op_setup_vol, ceed_data->x_coord, 65477841947SLeila Ghaffari ceed_data->q_data, CEED_REQUEST_IMMEDIATE); 65577841947SLeila Ghaffari 65677841947SLeila Ghaffari // -- Create and apply CEED Composite Operator for the entire domain 65777841947SLeila Ghaffari if (!user->phys->implicit) { // RHS 65877841947SLeila Ghaffari ierr = CreateOperatorForDomain(ceed, dm, bc, ceed_data, user->phys, 659e334ad8fSJed Brown user->op_rhs_vol, NULL, height, P_sur, Q_sur, 660e334ad8fSJed Brown q_data_size_sur, 0, 661e334ad8fSJed Brown &user->op_rhs, NULL); CHKERRQ(ierr); 66277841947SLeila Ghaffari } else { // IFunction 66377841947SLeila Ghaffari ierr = CreateOperatorForDomain(ceed, dm, bc, ceed_data, user->phys, 664e334ad8fSJed Brown user->op_ifunction_vol, op_ijacobian_vol, 665e334ad8fSJed Brown height, P_sur, Q_sur, 666e334ad8fSJed Brown q_data_size_sur, jac_data_size_sur, 667e334ad8fSJed Brown &user->op_ifunction, 668e334ad8fSJed Brown op_ijacobian_vol ? &user->op_ijacobian : NULL); CHKERRQ(ierr); 669e334ad8fSJed Brown if (user->op_ijacobian) { 670e334ad8fSJed Brown CeedOperatorContextGetFieldLabel(user->op_ijacobian, "ijacobian time shift", 671e334ad8fSJed Brown &user->phys->ijacobian_time_shift_label); 672e334ad8fSJed Brown } 673e334ad8fSJed Brown 67477841947SLeila Ghaffari } 675a3ae0734SJed Brown CeedElemRestrictionDestroy(&elem_restr_jd_i); 676a3ae0734SJed Brown CeedVectorDestroy(&jac_data); 67777841947SLeila Ghaffari PetscFunctionReturn(0); 67877841947SLeila Ghaffari } 679