xref: /honee/src/setuplibceed.c (revision 002797a32e7e6bab43fe78a1f2bfae476ddb23d4)
1a515125bSLeila Ghaffari // Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at
2a515125bSLeila Ghaffari // the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights
3a515125bSLeila Ghaffari // reserved. See files LICENSE and NOTICE for details.
4a515125bSLeila Ghaffari //
5a515125bSLeila Ghaffari // This file is part of CEED, a collection of benchmarks, miniapps, software
6a515125bSLeila Ghaffari // libraries and APIs for efficient high-order finite element and spectral
7a515125bSLeila Ghaffari // element discretizations for exascale applications. For more information and
8a515125bSLeila Ghaffari // source code availability see http://github.com/ceed.
9a515125bSLeila Ghaffari //
10a515125bSLeila Ghaffari // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
11a515125bSLeila Ghaffari // a collaborative effort of two U.S. Department of Energy organizations (Office
12a515125bSLeila Ghaffari // of Science and the National Nuclear Security Administration) responsible for
13a515125bSLeila Ghaffari // the planning and preparation of a capable exascale ecosystem, including
14a515125bSLeila Ghaffari // software, applications, hardware, advanced system engineering and early
15a515125bSLeila Ghaffari // testbed platforms, in support of the nation's exascale computing imperative.
16a515125bSLeila Ghaffari 
17a515125bSLeila Ghaffari /// @file
18a515125bSLeila Ghaffari /// Setup libCEED for Navier-Stokes example using PETSc
19a515125bSLeila Ghaffari 
20a515125bSLeila Ghaffari #include "../navierstokes.h"
21a515125bSLeila Ghaffari 
22a515125bSLeila Ghaffari // Utility function - essential BC dofs are encoded in closure indices as -(i+1).
23a515125bSLeila Ghaffari PetscInt Involute(PetscInt i) {
24a515125bSLeila Ghaffari   return i >= 0 ? i : -(i+1);
25a515125bSLeila Ghaffari }
26a515125bSLeila Ghaffari 
27a515125bSLeila Ghaffari // Utility function to create local CEED restriction
286b1ccf21SJeremy L Thompson PetscErrorCode CreateRestrictionFromPlex(Ceed ceed, DM dm, CeedInt height,
296b1ccf21SJeremy L Thompson     DMLabel domain_label, CeedInt value, CeedElemRestriction *elem_restr) {
306b1ccf21SJeremy L Thompson   PetscInt num_elem, elem_size, num_dof, num_comp, *elem_restr_offsets;
31a515125bSLeila Ghaffari   PetscErrorCode ierr;
326b1ccf21SJeremy L Thompson 
33a515125bSLeila Ghaffari   PetscFunctionBeginUser;
346b1ccf21SJeremy L Thompson   ierr = DMPlexGetLocalOffsets(dm, domain_label, value, height, 0, &num_elem,
356b1ccf21SJeremy L Thompson                                &elem_size, &num_comp, &num_dof, &elem_restr_offsets);
366b1ccf21SJeremy L Thompson   CHKERRQ(ierr);
37a515125bSLeila Ghaffari 
386b1ccf21SJeremy L Thompson   CeedElemRestrictionCreate(ceed, num_elem, elem_size, num_comp,
396b1ccf21SJeremy L Thompson                             1, num_dof, CEED_MEM_HOST, CEED_COPY_VALUES,
406b1ccf21SJeremy L Thompson                             elem_restr_offsets, elem_restr);
416b1ccf21SJeremy L Thompson   ierr = PetscFree(elem_restr_offsets); CHKERRQ(ierr);
42a515125bSLeila Ghaffari 
43a515125bSLeila Ghaffari   PetscFunctionReturn(0);
44a515125bSLeila Ghaffari }
45a515125bSLeila Ghaffari 
46a515125bSLeila Ghaffari // Utility function to get Ceed Restriction for each domain
47a515125bSLeila Ghaffari PetscErrorCode GetRestrictionForDomain(Ceed ceed, DM dm, CeedInt height,
48a515125bSLeila Ghaffari                                        DMLabel domain_label, PetscInt value,
496b1ccf21SJeremy L Thompson                                        CeedInt Q, CeedInt q_data_size,
50a515125bSLeila Ghaffari                                        CeedElemRestriction *elem_restr_q,
51a515125bSLeila Ghaffari                                        CeedElemRestriction *elem_restr_x,
52a515125bSLeila Ghaffari                                        CeedElemRestriction *elem_restr_qd_i) {
53a515125bSLeila Ghaffari   DM             dm_coord;
54a515125bSLeila Ghaffari   CeedInt        dim, loc_num_elem;
55a515125bSLeila Ghaffari   CeedInt        Q_dim;
56a515125bSLeila Ghaffari   PetscErrorCode ierr;
57a515125bSLeila Ghaffari   PetscFunctionBeginUser;
58a515125bSLeila Ghaffari 
59a515125bSLeila Ghaffari   ierr = DMGetDimension(dm, &dim); CHKERRQ(ierr);
60a515125bSLeila Ghaffari   dim -= height;
61a515125bSLeila Ghaffari   Q_dim = CeedIntPow(Q, dim);
62a515125bSLeila Ghaffari   ierr = DMGetCoordinateDM(dm, &dm_coord); CHKERRQ(ierr);
63a515125bSLeila Ghaffari   ierr = DMPlexSetClosurePermutationTensor(dm_coord, PETSC_DETERMINE, NULL);
64a515125bSLeila Ghaffari   CHKERRQ(ierr);
656b1ccf21SJeremy L Thompson   ierr = CreateRestrictionFromPlex(ceed, dm, height, domain_label, value,
666b1ccf21SJeremy L Thompson                                    elem_restr_q);
676b1ccf21SJeremy L Thompson   CHKERRQ(ierr);
686b1ccf21SJeremy L Thompson   ierr = CreateRestrictionFromPlex(ceed, dm_coord, height, domain_label, value,
696b1ccf21SJeremy L Thompson                                    elem_restr_x);
706b1ccf21SJeremy L Thompson   CHKERRQ(ierr);
71a515125bSLeila Ghaffari   CeedElemRestrictionGetNumElements(*elem_restr_q, &loc_num_elem);
72a515125bSLeila Ghaffari   CeedElemRestrictionCreateStrided(ceed, loc_num_elem, Q_dim,
73a515125bSLeila Ghaffari                                    q_data_size, q_data_size*loc_num_elem*Q_dim,
74a515125bSLeila Ghaffari                                    CEED_STRIDES_BACKEND, elem_restr_qd_i);
75a515125bSLeila Ghaffari   PetscFunctionReturn(0);
76a515125bSLeila Ghaffari }
77a515125bSLeila Ghaffari 
78a515125bSLeila Ghaffari // Utility function to create CEED Composite Operator for the entire domain
79a515125bSLeila Ghaffari PetscErrorCode CreateOperatorForDomain(Ceed ceed, DM dm, SimpleBC bc,
80a515125bSLeila Ghaffari                                        CeedData ceed_data, Physics phys,
81a515125bSLeila Ghaffari                                        CeedOperator op_apply_vol, CeedInt height,
82a515125bSLeila Ghaffari                                        CeedInt P_sur, CeedInt Q_sur, CeedInt q_data_size_sur,
83a515125bSLeila Ghaffari                                        CeedOperator *op_apply) {
84*002797a3SLeila Ghaffari   //CeedInt        dim;
85a515125bSLeila Ghaffari   DMLabel        domain_label;
86a515125bSLeila Ghaffari   PetscErrorCode ierr;
87a515125bSLeila Ghaffari   PetscFunctionBeginUser;
88a515125bSLeila Ghaffari 
89a515125bSLeila Ghaffari   // Create Composite Operaters
90a515125bSLeila Ghaffari   CeedCompositeOperatorCreate(ceed, op_apply);
91a515125bSLeila Ghaffari 
92a515125bSLeila Ghaffari   // --Apply Sub-Operator for the volume
93a515125bSLeila Ghaffari   CeedCompositeOperatorAddSub(*op_apply, op_apply_vol);
94a515125bSLeila Ghaffari 
95a515125bSLeila Ghaffari   // -- Create Sub-Operator for in/outflow BCs
96139613f2SLeila Ghaffari   if (phys->has_neumann) {
97a515125bSLeila Ghaffari     // --- Setup
98a515125bSLeila Ghaffari     ierr = DMGetLabel(dm, "Face Sets", &domain_label); CHKERRQ(ierr);
99*002797a3SLeila Ghaffari     //ierr = DMGetDimension(dm, &dim); CHKERRQ(ierr);
100a515125bSLeila Ghaffari 
101a515125bSLeila Ghaffari     // --- Get number of quadrature points for the boundaries
102a515125bSLeila Ghaffari     CeedInt num_qpts_sur;
103a515125bSLeila Ghaffari     CeedBasisGetNumQuadraturePoints(ceed_data->basis_q_sur, &num_qpts_sur);
104a515125bSLeila Ghaffari 
105*002797a3SLeila Ghaffari     // --- Create Sub-Operator for inflow boundaries
106*002797a3SLeila Ghaffari     for (CeedInt i=0; i < bc->num_inflow; i++) {
107a515125bSLeila Ghaffari       CeedVector          q_data_sur;
108*002797a3SLeila Ghaffari       CeedOperator        op_setup_sur, op_apply_inflow;
109*002797a3SLeila Ghaffari       CeedElemRestriction elem_restr_x_sur, elem_restr_q_sur, elem_restr_qd_i_sur;
110a515125bSLeila Ghaffari 
111a515125bSLeila Ghaffari       // ---- CEED Restriction
112*002797a3SLeila Ghaffari       ierr = GetRestrictionForDomain(ceed, dm, height, domain_label, bc->inflows[i],
113*002797a3SLeila Ghaffari                                      Q_sur, q_data_size_sur, &elem_restr_q_sur, &elem_restr_x_sur,
114*002797a3SLeila Ghaffari                                      &elem_restr_qd_i_sur);
115a515125bSLeila Ghaffari       CHKERRQ(ierr);
116a515125bSLeila Ghaffari 
117a515125bSLeila Ghaffari       // ---- CEED Vector
118a515125bSLeila Ghaffari       PetscInt loc_num_elem_sur;
119a515125bSLeila Ghaffari       CeedElemRestrictionGetNumElements(elem_restr_q_sur, &loc_num_elem_sur);
120a515125bSLeila Ghaffari       CeedVectorCreate(ceed, q_data_size_sur*loc_num_elem_sur*num_qpts_sur,
121a515125bSLeila Ghaffari                        &q_data_sur);
122a515125bSLeila Ghaffari 
123a515125bSLeila Ghaffari       // ---- CEED Operator
124a515125bSLeila Ghaffari       // ----- CEED Operator for Setup (geometric factors)
125a515125bSLeila Ghaffari       CeedOperatorCreate(ceed, ceed_data->qf_setup_sur, NULL, NULL, &op_setup_sur);
126a515125bSLeila Ghaffari       CeedOperatorSetField(op_setup_sur, "dx", elem_restr_x_sur,
127139613f2SLeila Ghaffari                            ceed_data->basis_x_sur, CEED_VECTOR_ACTIVE);
128a515125bSLeila Ghaffari       CeedOperatorSetField(op_setup_sur, "weight", CEED_ELEMRESTRICTION_NONE,
129a515125bSLeila Ghaffari                            ceed_data->basis_x_sur, CEED_VECTOR_NONE);
1303b1e209bSJeremy L Thompson       CeedOperatorSetField(op_setup_sur, "surface qdata", elem_restr_qd_i_sur,
131a515125bSLeila Ghaffari                            CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
132a515125bSLeila Ghaffari 
133a515125bSLeila Ghaffari       // ----- CEED Operator for Physics
134*002797a3SLeila Ghaffari       CeedOperatorCreate(ceed, ceed_data->qf_apply_inflow, NULL, NULL,
135*002797a3SLeila Ghaffari                          &op_apply_inflow);
136*002797a3SLeila Ghaffari       CeedOperatorSetField(op_apply_inflow, "q", elem_restr_q_sur,
137a515125bSLeila Ghaffari                            ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE);
138*002797a3SLeila Ghaffari       CeedOperatorSetField(op_apply_inflow, "surface qdata", elem_restr_qd_i_sur,
139a515125bSLeila Ghaffari                            CEED_BASIS_COLLOCATED, q_data_sur);
140*002797a3SLeila Ghaffari       CeedOperatorSetField(op_apply_inflow, "x", elem_restr_x_sur,
141a515125bSLeila Ghaffari                            ceed_data->basis_x_sur, ceed_data->x_coord);
142*002797a3SLeila Ghaffari       CeedOperatorSetField(op_apply_inflow, "v", elem_restr_q_sur,
143a515125bSLeila Ghaffari                            ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE);
144a515125bSLeila Ghaffari 
145a515125bSLeila Ghaffari       // ----- Apply CEED operator for Setup
146a515125bSLeila Ghaffari       CeedOperatorApply(op_setup_sur, ceed_data->x_coord, q_data_sur,
147a515125bSLeila Ghaffari                         CEED_REQUEST_IMMEDIATE);
148a515125bSLeila Ghaffari 
149*002797a3SLeila Ghaffari       // ----- Apply Sub-Operator for Physics
150*002797a3SLeila Ghaffari       CeedCompositeOperatorAddSub(*op_apply, op_apply_inflow);
151a515125bSLeila Ghaffari 
152a515125bSLeila Ghaffari       // ----- Cleanup
153a515125bSLeila Ghaffari       CeedVectorDestroy(&q_data_sur);
154a515125bSLeila Ghaffari       CeedElemRestrictionDestroy(&elem_restr_q_sur);
155a515125bSLeila Ghaffari       CeedElemRestrictionDestroy(&elem_restr_x_sur);
156a515125bSLeila Ghaffari       CeedElemRestrictionDestroy(&elem_restr_qd_i_sur);
157a515125bSLeila Ghaffari       CeedOperatorDestroy(&op_setup_sur);
158*002797a3SLeila Ghaffari       CeedOperatorDestroy(&op_apply_inflow);
159a515125bSLeila Ghaffari     }
160a515125bSLeila Ghaffari 
161*002797a3SLeila Ghaffari     // --- Create Sub-Operator for outflow boundaries
162*002797a3SLeila Ghaffari     for (CeedInt i=0; i < bc->num_outflow; i++) {
163*002797a3SLeila Ghaffari       CeedVector          q_data_sur;
164*002797a3SLeila Ghaffari       CeedOperator        op_setup_sur, op_apply_outflow;
165*002797a3SLeila Ghaffari       CeedElemRestriction elem_restr_x_sur, elem_restr_q_sur, elem_restr_qd_i_sur;
166*002797a3SLeila Ghaffari 
167*002797a3SLeila Ghaffari       // ---- CEED Restriction
168*002797a3SLeila Ghaffari       ierr = GetRestrictionForDomain(ceed, dm, height, domain_label, bc->outflows[i],
169*002797a3SLeila Ghaffari                                      Q_sur, q_data_size_sur, &elem_restr_q_sur, &elem_restr_x_sur,
170*002797a3SLeila Ghaffari                                      &elem_restr_qd_i_sur);
171*002797a3SLeila Ghaffari       CHKERRQ(ierr);
172*002797a3SLeila Ghaffari 
173*002797a3SLeila Ghaffari       // ---- CEED Vector
174*002797a3SLeila Ghaffari       PetscInt loc_num_elem_sur;
175*002797a3SLeila Ghaffari       CeedElemRestrictionGetNumElements(elem_restr_q_sur, &loc_num_elem_sur);
176*002797a3SLeila Ghaffari       CeedVectorCreate(ceed, q_data_size_sur*loc_num_elem_sur*num_qpts_sur,
177*002797a3SLeila Ghaffari                        &q_data_sur);
178*002797a3SLeila Ghaffari 
179*002797a3SLeila Ghaffari       // ---- CEED Operator
180*002797a3SLeila Ghaffari       // ----- CEED Operator for Setup (geometric factors)
181*002797a3SLeila Ghaffari       CeedOperatorCreate(ceed, ceed_data->qf_setup_sur, NULL, NULL, &op_setup_sur);
182*002797a3SLeila Ghaffari       CeedOperatorSetField(op_setup_sur, "dx", elem_restr_x_sur,
183*002797a3SLeila Ghaffari                            ceed_data->basis_x_sur, CEED_VECTOR_ACTIVE);
184*002797a3SLeila Ghaffari       CeedOperatorSetField(op_setup_sur, "weight", CEED_ELEMRESTRICTION_NONE,
185*002797a3SLeila Ghaffari                            ceed_data->basis_x_sur, CEED_VECTOR_NONE);
186*002797a3SLeila Ghaffari       CeedOperatorSetField(op_setup_sur, "surface qdata", elem_restr_qd_i_sur,
187*002797a3SLeila Ghaffari                            CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
188*002797a3SLeila Ghaffari 
189*002797a3SLeila Ghaffari       // ----- CEED Operator for Physics
190*002797a3SLeila Ghaffari       CeedOperatorCreate(ceed, ceed_data->qf_apply_outflow, NULL, NULL,
191*002797a3SLeila Ghaffari                          &op_apply_outflow);
192*002797a3SLeila Ghaffari       CeedOperatorSetField(op_apply_outflow, "q", elem_restr_q_sur,
193*002797a3SLeila Ghaffari                            ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE);
194*002797a3SLeila Ghaffari       CeedOperatorSetField(op_apply_outflow, "surface qdata", elem_restr_qd_i_sur,
195*002797a3SLeila Ghaffari                            CEED_BASIS_COLLOCATED, q_data_sur);
196*002797a3SLeila Ghaffari       CeedOperatorSetField(op_apply_outflow, "x", elem_restr_x_sur,
197*002797a3SLeila Ghaffari                            ceed_data->basis_x_sur, ceed_data->x_coord);
198*002797a3SLeila Ghaffari       CeedOperatorSetField(op_apply_outflow, "v", elem_restr_q_sur,
199*002797a3SLeila Ghaffari                            ceed_data->basis_q_sur, CEED_VECTOR_ACTIVE);
200*002797a3SLeila Ghaffari 
201*002797a3SLeila Ghaffari       // ----- Apply CEED operator for Setup
202*002797a3SLeila Ghaffari       CeedOperatorApply(op_setup_sur, ceed_data->x_coord, q_data_sur,
203*002797a3SLeila Ghaffari                         CEED_REQUEST_IMMEDIATE);
204*002797a3SLeila Ghaffari 
205*002797a3SLeila Ghaffari       // ----- Apply Sub-Operator for Physics
206*002797a3SLeila Ghaffari       CeedCompositeOperatorAddSub(*op_apply, op_apply_outflow);
207*002797a3SLeila Ghaffari 
208*002797a3SLeila Ghaffari       // ----- Cleanup
209*002797a3SLeila Ghaffari       CeedVectorDestroy(&q_data_sur);
210*002797a3SLeila Ghaffari       CeedElemRestrictionDestroy(&elem_restr_q_sur);
211*002797a3SLeila Ghaffari       CeedElemRestrictionDestroy(&elem_restr_x_sur);
212*002797a3SLeila Ghaffari       CeedElemRestrictionDestroy(&elem_restr_qd_i_sur);
213*002797a3SLeila Ghaffari       CeedOperatorDestroy(&op_setup_sur);
214*002797a3SLeila Ghaffari       CeedOperatorDestroy(&op_apply_outflow);
215*002797a3SLeila Ghaffari     }
216*002797a3SLeila Ghaffari   }
217a515125bSLeila Ghaffari   PetscFunctionReturn(0);
218a515125bSLeila Ghaffari }
219a515125bSLeila Ghaffari 
220a515125bSLeila Ghaffari PetscErrorCode SetupLibceed(Ceed ceed, CeedData ceed_data, DM dm, User user,
221a515125bSLeila Ghaffari                             AppCtx app_ctx, ProblemData *problem, SimpleBC bc) {
222a515125bSLeila Ghaffari   PetscErrorCode ierr;
223a515125bSLeila Ghaffari   PetscFunctionBeginUser;
224a515125bSLeila Ghaffari 
225a515125bSLeila Ghaffari   // *****************************************************************************
226a515125bSLeila Ghaffari   // Set up CEED objects for the interior domain (volume)
227a515125bSLeila Ghaffari   // *****************************************************************************
228a515125bSLeila Ghaffari   const PetscInt num_comp_q      = 5;
229a515125bSLeila Ghaffari   const CeedInt  dim             = problem->dim,
230a515125bSLeila Ghaffari                  num_comp_x      = problem->dim,
231a515125bSLeila Ghaffari                  q_data_size_vol = problem->q_data_size_vol,
232a515125bSLeila Ghaffari                  P               = app_ctx->degree + 1,
233a515125bSLeila Ghaffari                  Q               = P + app_ctx->q_extra;
234a515125bSLeila Ghaffari 
235a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
236a515125bSLeila Ghaffari   // CEED Bases
237a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
238a515125bSLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, num_comp_q, P, Q, CEED_GAUSS,
239a515125bSLeila Ghaffari                                   &ceed_data->basis_q);
240a515125bSLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, num_comp_x, 2, Q, CEED_GAUSS,
241a515125bSLeila Ghaffari                                   &ceed_data->basis_x);
242a515125bSLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim, num_comp_x, 2, P,
243a515125bSLeila Ghaffari                                   CEED_GAUSS_LOBATTO, &ceed_data->basis_xc);
244a515125bSLeila Ghaffari 
245a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
246a515125bSLeila Ghaffari   // CEED Restrictions
247a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
248a515125bSLeila Ghaffari   // -- Create restriction
2496b1ccf21SJeremy L Thompson   ierr = GetRestrictionForDomain(ceed, dm, 0, 0, 0, Q, q_data_size_vol,
2506b1ccf21SJeremy L Thompson                                  &ceed_data->elem_restr_q, &ceed_data->elem_restr_x,
251a515125bSLeila Ghaffari                                  &ceed_data->elem_restr_qd_i); CHKERRQ(ierr);
252a515125bSLeila Ghaffari   // -- Create E vectors
253a515125bSLeila Ghaffari   CeedElemRestrictionCreateVector(ceed_data->elem_restr_q, &user->q_ceed, NULL);
254a515125bSLeila Ghaffari   CeedElemRestrictionCreateVector(ceed_data->elem_restr_q, &user->q_dot_ceed,
255a515125bSLeila Ghaffari                                   NULL);
256a515125bSLeila Ghaffari   CeedElemRestrictionCreateVector(ceed_data->elem_restr_q, &user->g_ceed, NULL);
257a515125bSLeila Ghaffari 
258a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
259a515125bSLeila Ghaffari   // CEED QFunctions
260a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
261a515125bSLeila Ghaffari   // -- Create QFunction for quadrature data
262a515125bSLeila Ghaffari   CeedQFunctionCreateInterior(ceed, 1, problem->setup_vol, problem->setup_vol_loc,
263a515125bSLeila Ghaffari                               &ceed_data->qf_setup_vol);
264a515125bSLeila Ghaffari   CeedQFunctionAddInput(ceed_data->qf_setup_vol, "dx", num_comp_x*dim,
265a515125bSLeila Ghaffari                         CEED_EVAL_GRAD);
266a515125bSLeila Ghaffari   CeedQFunctionAddInput(ceed_data->qf_setup_vol, "weight", 1, CEED_EVAL_WEIGHT);
2673b1e209bSJeremy L Thompson   CeedQFunctionAddOutput(ceed_data->qf_setup_vol, "qdata", q_data_size_vol,
268a515125bSLeila Ghaffari                          CEED_EVAL_NONE);
269a515125bSLeila Ghaffari 
270a515125bSLeila Ghaffari   // -- Create QFunction for ICs
271a515125bSLeila Ghaffari   CeedQFunctionCreateInterior(ceed, 1, problem->ics, problem->ics_loc,
272a515125bSLeila Ghaffari                               &ceed_data->qf_ics);
273a515125bSLeila Ghaffari   CeedQFunctionAddInput(ceed_data->qf_ics, "x", num_comp_x, CEED_EVAL_INTERP);
274a515125bSLeila Ghaffari   CeedQFunctionAddOutput(ceed_data->qf_ics, "q0", num_comp_q, CEED_EVAL_NONE);
275a515125bSLeila Ghaffari 
276a515125bSLeila Ghaffari   // -- Create QFunction for RHS
277a515125bSLeila Ghaffari   if (problem->apply_vol_rhs) {
278a515125bSLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->apply_vol_rhs,
279a515125bSLeila Ghaffari                                 problem->apply_vol_rhs_loc, &ceed_data->qf_rhs_vol);
280a515125bSLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_rhs_vol, "q", num_comp_q, CEED_EVAL_INTERP);
281a515125bSLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_rhs_vol, "dq", num_comp_q*dim,
282a515125bSLeila Ghaffari                           CEED_EVAL_GRAD);
2833b1e209bSJeremy L Thompson     CeedQFunctionAddInput(ceed_data->qf_rhs_vol, "qdata", q_data_size_vol,
284a515125bSLeila Ghaffari                           CEED_EVAL_NONE);
285a515125bSLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_rhs_vol, "x", num_comp_x, CEED_EVAL_INTERP);
286a515125bSLeila Ghaffari     CeedQFunctionAddOutput(ceed_data->qf_rhs_vol, "v", num_comp_q,
287a515125bSLeila Ghaffari                            CEED_EVAL_INTERP);
288a515125bSLeila Ghaffari     CeedQFunctionAddOutput(ceed_data->qf_rhs_vol, "dv", num_comp_q*dim,
289a515125bSLeila Ghaffari                            CEED_EVAL_GRAD);
290a515125bSLeila Ghaffari   }
291a515125bSLeila Ghaffari 
292a515125bSLeila Ghaffari   // -- Create QFunction for IFunction
293a515125bSLeila Ghaffari   if (problem->apply_vol_ifunction) {
294a515125bSLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->apply_vol_ifunction,
295a515125bSLeila Ghaffari                                 problem->apply_vol_ifunction_loc, &ceed_data->qf_ifunction_vol);
296a515125bSLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "q", num_comp_q,
297a515125bSLeila Ghaffari                           CEED_EVAL_INTERP);
298a515125bSLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "dq", num_comp_q*dim,
299a515125bSLeila Ghaffari                           CEED_EVAL_GRAD);
3003b1e209bSJeremy L Thompson     CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "q dot", num_comp_q,
301a515125bSLeila Ghaffari                           CEED_EVAL_INTERP);
3023b1e209bSJeremy L Thompson     CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "qdata", q_data_size_vol,
303a515125bSLeila Ghaffari                           CEED_EVAL_NONE);
304a515125bSLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_ifunction_vol, "x", num_comp_x,
305a515125bSLeila Ghaffari                           CEED_EVAL_INTERP);
306a515125bSLeila Ghaffari     CeedQFunctionAddOutput(ceed_data->qf_ifunction_vol, "v", num_comp_q,
307a515125bSLeila Ghaffari                            CEED_EVAL_INTERP);
308a515125bSLeila Ghaffari     CeedQFunctionAddOutput(ceed_data->qf_ifunction_vol, "dv", num_comp_q*dim,
309a515125bSLeila Ghaffari                            CEED_EVAL_GRAD);
310a515125bSLeila Ghaffari   }
311a515125bSLeila Ghaffari 
312a515125bSLeila Ghaffari   // ---------------------------------------------------------------------------
313a515125bSLeila Ghaffari   // Element coordinates
314a515125bSLeila Ghaffari   // ---------------------------------------------------------------------------
315a515125bSLeila Ghaffari   // -- Create CEED vector
316a515125bSLeila Ghaffari   CeedElemRestrictionCreateVector(ceed_data->elem_restr_x, &ceed_data->x_coord,
317a515125bSLeila Ghaffari                                   NULL);
318a515125bSLeila Ghaffari 
319a515125bSLeila Ghaffari   // -- Copy PETSc vector in CEED vector
320a515125bSLeila Ghaffari   Vec               X_loc;
321a515125bSLeila Ghaffari   const PetscScalar *X_loc_array;
322a515125bSLeila Ghaffari   ierr = DMGetCoordinatesLocal(dm, &X_loc); CHKERRQ(ierr);
32305a512bdSLeila Ghaffari   ierr = VecScale(X_loc, problem->dm_scale); CHKERRQ(ierr);
324a515125bSLeila Ghaffari   ierr = VecGetArrayRead(X_loc, &X_loc_array); CHKERRQ(ierr);
325a515125bSLeila Ghaffari   CeedVectorSetArray(ceed_data->x_coord, CEED_MEM_HOST, CEED_COPY_VALUES,
326a515125bSLeila Ghaffari                      (PetscScalar *)X_loc_array);
327a515125bSLeila Ghaffari   ierr = VecRestoreArrayRead(X_loc, &X_loc_array); CHKERRQ(ierr);
328a515125bSLeila Ghaffari 
329a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
330a515125bSLeila Ghaffari   // CEED vectors
331a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
332a515125bSLeila Ghaffari   // -- Create CEED vector for geometric data
333a515125bSLeila Ghaffari   CeedInt  num_qpts_vol;
334a515125bSLeila Ghaffari   PetscInt loc_num_elem_vol;
335a515125bSLeila Ghaffari   CeedBasisGetNumQuadraturePoints(ceed_data->basis_q, &num_qpts_vol);
336a515125bSLeila Ghaffari   CeedElemRestrictionGetNumElements(ceed_data->elem_restr_q, &loc_num_elem_vol);
337a515125bSLeila Ghaffari   CeedVectorCreate(ceed, q_data_size_vol*loc_num_elem_vol*num_qpts_vol,
338a515125bSLeila Ghaffari                    &ceed_data->q_data);
339a515125bSLeila Ghaffari 
340a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
341a515125bSLeila Ghaffari   // CEED Operators
342a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
343a515125bSLeila Ghaffari   // -- Create CEED operator for quadrature data
344a515125bSLeila Ghaffari   CeedOperatorCreate(ceed, ceed_data->qf_setup_vol, NULL, NULL,
345a515125bSLeila Ghaffari                      &ceed_data->op_setup_vol);
346a515125bSLeila Ghaffari   CeedOperatorSetField(ceed_data->op_setup_vol, "dx", ceed_data->elem_restr_x,
347a515125bSLeila Ghaffari                        ceed_data->basis_x, CEED_VECTOR_ACTIVE);
348a515125bSLeila Ghaffari   CeedOperatorSetField(ceed_data->op_setup_vol, "weight",
349139613f2SLeila Ghaffari                        CEED_ELEMRESTRICTION_NONE, ceed_data->basis_x, CEED_VECTOR_NONE);
3503b1e209bSJeremy L Thompson   CeedOperatorSetField(ceed_data->op_setup_vol, "qdata",
351139613f2SLeila Ghaffari                        ceed_data->elem_restr_qd_i, CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
352a515125bSLeila Ghaffari 
353a515125bSLeila Ghaffari   // -- Create CEED operator for ICs
354a515125bSLeila Ghaffari   CeedOperatorCreate(ceed, ceed_data->qf_ics, NULL, NULL, &ceed_data->op_ics);
355a515125bSLeila Ghaffari   CeedOperatorSetField(ceed_data->op_ics, "x", ceed_data->elem_restr_x,
356a515125bSLeila Ghaffari                        ceed_data->basis_xc, CEED_VECTOR_ACTIVE);
357a515125bSLeila Ghaffari   CeedOperatorSetField(ceed_data->op_ics, "q0", ceed_data->elem_restr_q,
358a515125bSLeila Ghaffari                        CEED_BASIS_COLLOCATED, CEED_VECTOR_ACTIVE);
359a515125bSLeila Ghaffari 
360a515125bSLeila Ghaffari   // Create CEED operator for RHS
361a515125bSLeila Ghaffari   if (ceed_data->qf_rhs_vol) {
362a515125bSLeila Ghaffari     CeedOperator op;
363a515125bSLeila Ghaffari     CeedOperatorCreate(ceed, ceed_data->qf_rhs_vol, NULL, NULL, &op);
364a515125bSLeila Ghaffari     CeedOperatorSetField(op, "q", ceed_data->elem_restr_q, ceed_data->basis_q,
365a515125bSLeila Ghaffari                          CEED_VECTOR_ACTIVE);
366a515125bSLeila Ghaffari     CeedOperatorSetField(op, "dq", ceed_data->elem_restr_q, ceed_data->basis_q,
367a515125bSLeila Ghaffari                          CEED_VECTOR_ACTIVE);
3683b1e209bSJeremy L Thompson     CeedOperatorSetField(op, "qdata", ceed_data->elem_restr_qd_i,
369139613f2SLeila Ghaffari                          CEED_BASIS_COLLOCATED, ceed_data->q_data);
370a515125bSLeila Ghaffari     CeedOperatorSetField(op, "x", ceed_data->elem_restr_x, ceed_data->basis_x,
371a515125bSLeila Ghaffari                          ceed_data->x_coord);
372a515125bSLeila Ghaffari     CeedOperatorSetField(op, "v", ceed_data->elem_restr_q, ceed_data->basis_q,
373a515125bSLeila Ghaffari                          CEED_VECTOR_ACTIVE);
374a515125bSLeila Ghaffari     CeedOperatorSetField(op, "dv", ceed_data->elem_restr_q, ceed_data->basis_q,
375a515125bSLeila Ghaffari                          CEED_VECTOR_ACTIVE);
376a515125bSLeila Ghaffari     user->op_rhs_vol = op;
377a515125bSLeila Ghaffari   }
378a515125bSLeila Ghaffari 
379a515125bSLeila Ghaffari   // -- CEED operator for IFunction
380a515125bSLeila Ghaffari   if (ceed_data->qf_ifunction_vol) {
381a515125bSLeila Ghaffari     CeedOperator op;
382a515125bSLeila Ghaffari     CeedOperatorCreate(ceed, ceed_data->qf_ifunction_vol, NULL, NULL, &op);
383a515125bSLeila Ghaffari     CeedOperatorSetField(op, "q", ceed_data->elem_restr_q, ceed_data->basis_q,
384a515125bSLeila Ghaffari                          CEED_VECTOR_ACTIVE);
385a515125bSLeila Ghaffari     CeedOperatorSetField(op, "dq", ceed_data->elem_restr_q, ceed_data->basis_q,
386a515125bSLeila Ghaffari                          CEED_VECTOR_ACTIVE);
3873b1e209bSJeremy L Thompson     CeedOperatorSetField(op, "q dot", ceed_data->elem_restr_q, ceed_data->basis_q,
388a515125bSLeila Ghaffari                          user->q_dot_ceed);
3893b1e209bSJeremy L Thompson     CeedOperatorSetField(op, "qdata", ceed_data->elem_restr_qd_i,
390139613f2SLeila Ghaffari                          CEED_BASIS_COLLOCATED, ceed_data->q_data);
391a515125bSLeila Ghaffari     CeedOperatorSetField(op, "x", ceed_data->elem_restr_x, ceed_data->basis_x,
392a515125bSLeila Ghaffari                          ceed_data->x_coord);
393a515125bSLeila Ghaffari     CeedOperatorSetField(op, "v", ceed_data->elem_restr_q, ceed_data->basis_q,
394a515125bSLeila Ghaffari                          CEED_VECTOR_ACTIVE);
395a515125bSLeila Ghaffari     CeedOperatorSetField(op, "dv", ceed_data->elem_restr_q, ceed_data->basis_q,
396a515125bSLeila Ghaffari                          CEED_VECTOR_ACTIVE);
397a515125bSLeila Ghaffari     user->op_ifunction_vol = op;
398a515125bSLeila Ghaffari   }
399a515125bSLeila Ghaffari 
400a515125bSLeila Ghaffari   // *****************************************************************************
401a515125bSLeila Ghaffari   // Set up CEED objects for the exterior domain (surface)
402a515125bSLeila Ghaffari   // *****************************************************************************
403a515125bSLeila Ghaffari   CeedInt height  = 1,
404a515125bSLeila Ghaffari           dim_sur = dim - height,
405a515125bSLeila Ghaffari           P_sur   = app_ctx->degree + 1,
406a515125bSLeila Ghaffari           Q_sur   = P_sur + app_ctx->q_extra;
407a515125bSLeila Ghaffari   const CeedInt q_data_size_sur = problem->q_data_size_sur;
408a515125bSLeila Ghaffari 
409a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
410a515125bSLeila Ghaffari   // CEED Bases
411a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
412a515125bSLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim_sur, num_comp_q, P_sur, Q_sur,
413a515125bSLeila Ghaffari                                   CEED_GAUSS, &ceed_data->basis_q_sur);
414a515125bSLeila Ghaffari   CeedBasisCreateTensorH1Lagrange(ceed, dim_sur, num_comp_x, 2, Q_sur, CEED_GAUSS,
415a515125bSLeila Ghaffari                                   &ceed_data->basis_x_sur);
416a515125bSLeila Ghaffari 
417a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
418a515125bSLeila Ghaffari   // CEED QFunctions
419a515125bSLeila Ghaffari   // -----------------------------------------------------------------------------
420a515125bSLeila Ghaffari   // -- Create QFunction for quadrature data
421a515125bSLeila Ghaffari   CeedQFunctionCreateInterior(ceed, 1, problem->setup_sur, problem->setup_sur_loc,
422a515125bSLeila Ghaffari                               &ceed_data->qf_setup_sur);
423a515125bSLeila Ghaffari   CeedQFunctionAddInput(ceed_data->qf_setup_sur, "dx", num_comp_x*dim_sur,
424a515125bSLeila Ghaffari                         CEED_EVAL_GRAD);
425a515125bSLeila Ghaffari   CeedQFunctionAddInput(ceed_data->qf_setup_sur, "weight", 1, CEED_EVAL_WEIGHT);
4263b1e209bSJeremy L Thompson   CeedQFunctionAddOutput(ceed_data->qf_setup_sur, "surface qdata",
427*002797a3SLeila Ghaffari                          q_data_size_sur, CEED_EVAL_NONE);
428a515125bSLeila Ghaffari 
429*002797a3SLeila Ghaffari   // -- Creat QFunction for inflow boundaries
430*002797a3SLeila Ghaffari   if (problem->apply_inflow) {
431*002797a3SLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->apply_inflow,
432*002797a3SLeila Ghaffari                                 problem->apply_inflow_loc, &ceed_data->qf_apply_inflow);
433*002797a3SLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_apply_inflow, "q", num_comp_q,
434a515125bSLeila Ghaffari                           CEED_EVAL_INTERP);
435*002797a3SLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_apply_inflow, "surface qdata",
436*002797a3SLeila Ghaffari                           q_data_size_sur, CEED_EVAL_NONE);
437*002797a3SLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_apply_inflow, "x", num_comp_x,
438a515125bSLeila Ghaffari                           CEED_EVAL_INTERP);
439*002797a3SLeila Ghaffari     CeedQFunctionAddOutput(ceed_data->qf_apply_inflow, "v", num_comp_q,
440*002797a3SLeila Ghaffari                            CEED_EVAL_INTERP);
441*002797a3SLeila Ghaffari   }
442*002797a3SLeila Ghaffari 
443*002797a3SLeila Ghaffari   // -- Creat QFunction for outflow boundaries
444*002797a3SLeila Ghaffari   if (problem->apply_outflow) {
445*002797a3SLeila Ghaffari     CeedQFunctionCreateInterior(ceed, 1, problem->apply_outflow,
446*002797a3SLeila Ghaffari                                 problem->apply_outflow_loc, &ceed_data->qf_apply_outflow);
447*002797a3SLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_apply_outflow, "q", num_comp_q,
448*002797a3SLeila Ghaffari                           CEED_EVAL_INTERP);
449*002797a3SLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_apply_outflow, "surface qdata",
450*002797a3SLeila Ghaffari                           q_data_size_sur, CEED_EVAL_NONE);
451*002797a3SLeila Ghaffari     CeedQFunctionAddInput(ceed_data->qf_apply_outflow, "x", num_comp_x,
452*002797a3SLeila Ghaffari                           CEED_EVAL_INTERP);
453*002797a3SLeila Ghaffari     CeedQFunctionAddOutput(ceed_data->qf_apply_outflow, "v", num_comp_q,
454a515125bSLeila Ghaffari                            CEED_EVAL_INTERP);
455a515125bSLeila Ghaffari   }
456a515125bSLeila Ghaffari 
457a515125bSLeila Ghaffari   // *****************************************************************************
458a515125bSLeila Ghaffari   // CEED Operator Apply
459a515125bSLeila Ghaffari   // *****************************************************************************
460a515125bSLeila Ghaffari   // -- Apply CEED Operator for the geometric data
461a515125bSLeila Ghaffari   CeedOperatorApply(ceed_data->op_setup_vol, ceed_data->x_coord,
462a515125bSLeila Ghaffari                     ceed_data->q_data, CEED_REQUEST_IMMEDIATE);
463a515125bSLeila Ghaffari 
464a515125bSLeila Ghaffari   // -- Create and apply CEED Composite Operator for the entire domain
465a515125bSLeila Ghaffari   if (!user->phys->implicit) { // RHS
466a515125bSLeila Ghaffari     ierr = CreateOperatorForDomain(ceed, dm, bc, ceed_data, user->phys,
467a515125bSLeila Ghaffari                                    user->op_rhs_vol, height, P_sur, Q_sur,
468a515125bSLeila Ghaffari                                    q_data_size_sur, &user->op_rhs); CHKERRQ(ierr);
469a515125bSLeila Ghaffari   } else { // IFunction
470a515125bSLeila Ghaffari     ierr = CreateOperatorForDomain(ceed, dm, bc, ceed_data, user->phys,
471a515125bSLeila Ghaffari                                    user->op_ifunction_vol, height, P_sur, Q_sur,
472a515125bSLeila Ghaffari                                    q_data_size_sur, &user->op_ifunction); CHKERRQ(ierr);
473a515125bSLeila Ghaffari   }
474a515125bSLeila Ghaffari 
475a515125bSLeila Ghaffari   PetscFunctionReturn(0);
476a515125bSLeila Ghaffari }
477