xref: /libCEED/backends/ref/ceed-ref-operator.c (revision bcbe1c99be4be6e5341a66695d3b47170158a1bf)
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.
321617c04Sjeremylt //
43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
521617c04Sjeremylt //
63d8e8822SJeremy L Thompson // This file is part of CEED:  http://github.com/ceed
721617c04Sjeremylt 
849aac155SJeremy L Thompson #include <ceed.h>
9ec3da8bcSJed Brown #include <ceed/backend.h>
103d576824SJeremy L Thompson #include <stdbool.h>
113d576824SJeremy L Thompson #include <stddef.h>
123d576824SJeremy L Thompson #include <stdint.h>
132b730f8bSJeremy L Thompson 
1421617c04Sjeremylt #include "ceed-ref.h"
1521617c04Sjeremylt 
16f10650afSjeremylt //------------------------------------------------------------------------------
17f10650afSjeremylt // Setup Input/Output Fields
18f10650afSjeremylt //------------------------------------------------------------------------------
192b730f8bSJeremy L Thompson static int CeedOperatorSetupFields_Ref(CeedQFunction qf, CeedOperator op, bool is_input, CeedVector *e_vecs_full, CeedVector *e_vecs,
202b730f8bSJeremy L Thompson                                        CeedVector *q_vecs, CeedInt start_e, CeedInt num_fields, CeedInt Q) {
21aedaa0e5Sjeremylt   Ceed                ceed;
22ad70ee2cSJeremy L Thompson   CeedSize            e_size, q_size;
23ad70ee2cSJeremy L Thompson   CeedInt             num_comp, size, P;
24d1d35e2fSjeremylt   CeedQFunctionField *qf_fields;
25ad70ee2cSJeremy L Thompson   CeedOperatorField  *op_fields;
26ad70ee2cSJeremy L Thompson 
27e910d748SJeremy L Thompson   {
28e910d748SJeremy L Thompson     Ceed ceed_parent;
29e910d748SJeremy L Thompson 
30ad70ee2cSJeremy L Thompson     CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
31e910d748SJeremy L Thompson     CeedCallBackend(CeedGetParent(ceed, &ceed_parent));
32e910d748SJeremy L Thompson     if (ceed_parent) ceed = ceed_parent;
33e910d748SJeremy L Thompson   }
344fc1f125SJeremy L Thompson   if (is_input) {
352b730f8bSJeremy L Thompson     CeedCallBackend(CeedOperatorGetFields(op, NULL, &op_fields, NULL, NULL));
362b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL));
374fc1f125SJeremy L Thompson   } else {
382b730f8bSJeremy L Thompson     CeedCallBackend(CeedOperatorGetFields(op, NULL, NULL, NULL, &op_fields));
392b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields));
40fe2413ffSjeremylt   }
4121617c04Sjeremylt 
42885ac19cSjeremylt   // Loop over fields
43d1d35e2fSjeremylt   for (CeedInt i = 0; i < num_fields; i++) {
44d1d35e2fSjeremylt     CeedEvalMode        eval_mode;
45edb2538eSJeremy L Thompson     CeedElemRestriction elem_rstr;
46ad70ee2cSJeremy L Thompson     CeedBasis           basis;
47d1d35e2fSjeremylt 
48ad70ee2cSJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
49d1d35e2fSjeremylt     if (eval_mode != CEED_EVAL_WEIGHT) {
50edb2538eSJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_fields[i], &elem_rstr));
51edb2538eSJeremy L Thompson       CeedCallBackend(CeedElemRestrictionCreateVector(elem_rstr, NULL, &e_vecs_full[i + start_e]));
52135a076eSjeremylt     }
53135a076eSjeremylt 
54d1d35e2fSjeremylt     switch (eval_mode) {
55885ac19cSjeremylt       case CEED_EVAL_NONE:
562b730f8bSJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size));
57d2643443SJeremy L Thompson         q_size = (CeedSize)Q * size;
582b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
59aedaa0e5Sjeremylt         break;
60aedaa0e5Sjeremylt       case CEED_EVAL_INTERP:
61885ac19cSjeremylt       case CEED_EVAL_GRAD:
62a915a514Srezgarshakeri       case CEED_EVAL_DIV:
63c4e3f59bSSebastian Grimberg       case CEED_EVAL_CURL:
642b730f8bSJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
652b730f8bSJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size));
662b730f8bSJeremy L Thompson         CeedCallBackend(CeedBasisGetNumNodes(basis, &P));
672b730f8bSJeremy L Thompson         CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp));
68d2643443SJeremy L Thompson         e_size = (CeedSize)P * num_comp;
692b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, e_size, &e_vecs[i]));
70d2643443SJeremy L Thompson         q_size = (CeedSize)Q * size;
712b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
72885ac19cSjeremylt         break;
73885ac19cSjeremylt       case CEED_EVAL_WEIGHT:  // Only on input fields
742b730f8bSJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
75d2643443SJeremy L Thompson         q_size = (CeedSize)Q;
762b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
772b730f8bSJeremy L Thompson         CeedCallBackend(CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, CEED_EVAL_WEIGHT, CEED_VECTOR_NONE, q_vecs[i]));
78885ac19cSjeremylt         break;
7921617c04Sjeremylt     }
80885ac19cSjeremylt   }
81e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
8221617c04Sjeremylt }
8321617c04Sjeremylt 
84f10650afSjeremylt //------------------------------------------------------------------------------
85f10650afSjeremylt // Setup Operator
86f10650afSjeremylt //------------------------------------------------------------------------------/*
87885ac19cSjeremylt static int CeedOperatorSetup_Ref(CeedOperator op) {
888c1105f8SJeremy L Thompson   bool                is_setup_done;
89ad70ee2cSJeremy L Thompson   Ceed                ceed;
90ad70ee2cSJeremy L Thompson   CeedInt             Q, num_input_fields, num_output_fields;
91ad70ee2cSJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
92ad70ee2cSJeremy L Thompson   CeedQFunction       qf;
93ad70ee2cSJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
94ad70ee2cSJeremy L Thompson   CeedOperator_Ref   *impl;
95ad70ee2cSJeremy L Thompson 
962b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorIsSetupDone(op, &is_setup_done));
978c1105f8SJeremy L Thompson   if (is_setup_done) return CEED_ERROR_SUCCESS;
98ad70ee2cSJeremy L Thompson 
992b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
1002b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
1012b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
1022b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
1032b730f8bSJeremy L Thompson   CeedCallBackend(CeedQFunctionIsIdentity(qf, &impl->is_identity_qf));
1042b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
1052b730f8bSJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
106885ac19cSjeremylt 
107885ac19cSjeremylt   // Allocate
1082b730f8bSJeremy L Thompson   CeedCallBackend(CeedCalloc(num_input_fields + num_output_fields, &impl->e_vecs_full));
109885ac19cSjeremylt 
1102b730f8bSJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->input_states));
1112b730f8bSJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->e_vecs_in));
1122b730f8bSJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->e_vecs_out));
1132b730f8bSJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_in));
1142b730f8bSJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_out));
115885ac19cSjeremylt 
1164fc1f125SJeremy L Thompson   impl->num_inputs  = num_input_fields;
1174fc1f125SJeremy L Thompson   impl->num_outputs = num_output_fields;
118885ac19cSjeremylt 
119d1d35e2fSjeremylt   // Set up infield and outfield e_vecs and q_vecs
120885ac19cSjeremylt   // Infields
1212b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorSetupFields_Ref(qf, op, true, impl->e_vecs_full, impl->e_vecs_in, impl->q_vecs_in, 0, num_input_fields, Q));
122885ac19cSjeremylt   // Outfields
1232b730f8bSJeremy L Thompson   CeedCallBackend(
1242b730f8bSJeremy L Thompson       CeedOperatorSetupFields_Ref(qf, op, false, impl->e_vecs_full, impl->e_vecs_out, impl->q_vecs_out, num_input_fields, num_output_fields, Q));
125885ac19cSjeremylt 
12616911fdaSjeremylt   // Identity QFunctions
1270b454692Sjeremylt   if (impl->is_identity_qf) {
128d1d35e2fSjeremylt     CeedEvalMode        in_mode, out_mode;
129d1d35e2fSjeremylt     CeedQFunctionField *in_fields, *out_fields;
130ad70ee2cSJeremy L Thompson 
1312b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &in_fields, NULL, &out_fields));
1322b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(in_fields[0], &in_mode));
1332b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(out_fields[0], &out_mode));
134d1d35e2fSjeremylt 
1350b454692Sjeremylt     if (in_mode == CEED_EVAL_NONE && out_mode == CEED_EVAL_NONE) {
136edb2538eSJeremy L Thompson       impl->is_identity_rstr_op = true;
1370b454692Sjeremylt     } else {
138db002c03SJeremy L Thompson       CeedCallBackend(CeedVectorReferenceCopy(impl->q_vecs_in[0], &impl->q_vecs_out[0]));
13916911fdaSjeremylt     }
14016911fdaSjeremylt   }
14116911fdaSjeremylt 
1422b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorSetSetupDone(op));
143e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
144885ac19cSjeremylt }
145885ac19cSjeremylt 
146f10650afSjeremylt //------------------------------------------------------------------------------
147f10650afSjeremylt // Setup Operator Inputs
148f10650afSjeremylt //------------------------------------------------------------------------------
1492b730f8bSJeremy L Thompson static inline int CeedOperatorSetupInputs_Ref(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields,
1502b730f8bSJeremy L Thompson                                               CeedVector in_vec, const bool skip_active, CeedScalar *e_data_full[2 * CEED_FIELD_MAX],
151a0162de9SJeremy L Thompson                                               CeedOperator_Ref *impl, CeedRequest *request) {
152ad70ee2cSJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
153ad70ee2cSJeremy L Thompson     uint64_t            state;
154d1d35e2fSjeremylt     CeedEvalMode        eval_mode;
155d1bcdac9Sjeremylt     CeedVector          vec;
156edb2538eSJeremy L Thompson     CeedElemRestriction elem_rstr;
157885ac19cSjeremylt 
158d1bcdac9Sjeremylt     // Get input vector
1592b730f8bSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
1601d102b48SJeremy L Thompson     if (vec == CEED_VECTOR_ACTIVE) {
1612b730f8bSJeremy L Thompson       if (skip_active) continue;
1622b730f8bSJeremy L Thompson       else vec = in_vec;
1631d102b48SJeremy L Thompson     }
1641d102b48SJeremy L Thompson 
1652b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
1661d102b48SJeremy L Thompson     // Restrict and Evec
167d1d35e2fSjeremylt     if (eval_mode == CEED_EVAL_WEIGHT) {  // Skip
1681d102b48SJeremy L Thompson     } else {
169668048e2SJed Brown       // Restrict
1702b730f8bSJeremy L Thompson       CeedCallBackend(CeedVectorGetState(vec, &state));
1718d713cf6Sjeremylt       // Skip restriction if input is unchanged
1724fc1f125SJeremy L Thompson       if (state != impl->input_states[i] || vec == in_vec) {
173edb2538eSJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr));
174edb2538eSJeremy L Thompson         CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_NOTRANSPOSE, vec, impl->e_vecs_full[i], request));
1754fc1f125SJeremy L Thompson         impl->input_states[i] = state;
1768d713cf6Sjeremylt       }
177668048e2SJed Brown       // Get evec
1782b730f8bSJeremy L Thompson       CeedCallBackend(CeedVectorGetArrayRead(impl->e_vecs_full[i], CEED_MEM_HOST, (const CeedScalar **)&e_data_full[i]));
179885ac19cSjeremylt     }
180885ac19cSjeremylt   }
181e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
182885ac19cSjeremylt }
183885ac19cSjeremylt 
184f10650afSjeremylt //------------------------------------------------------------------------------
185f10650afSjeremylt // Input Basis Action
186f10650afSjeremylt //------------------------------------------------------------------------------
1872b730f8bSJeremy L Thompson static inline int CeedOperatorInputBasis_Ref(CeedInt e, CeedInt Q, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields,
1882b730f8bSJeremy L Thompson                                              CeedInt num_input_fields, const bool skip_active, CeedScalar *e_data_full[2 * CEED_FIELD_MAX],
1892b730f8bSJeremy L Thompson                                              CeedOperator_Ref *impl) {
190ad70ee2cSJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
191a915a514Srezgarshakeri     CeedInt             elem_size, size, num_comp;
192d1d35e2fSjeremylt     CeedEvalMode        eval_mode;
193edb2538eSJeremy L Thompson     CeedElemRestriction elem_rstr;
1941d102b48SJeremy L Thompson     CeedBasis           basis;
1951d102b48SJeremy L Thompson 
1961d102b48SJeremy L Thompson     // Skip active input
197d1d35e2fSjeremylt     if (skip_active) {
1981d102b48SJeremy L Thompson       CeedVector vec;
199ad70ee2cSJeremy L Thompson 
2002b730f8bSJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
2012b730f8bSJeremy L Thompson       if (vec == CEED_VECTOR_ACTIVE) continue;
2021d102b48SJeremy L Thompson     }
203d1d35e2fSjeremylt     // Get elem_size, eval_mode, size
204edb2538eSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr));
205edb2538eSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetElementSize(elem_rstr, &elem_size));
2062b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
2072b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetSize(qf_input_fields[i], &size));
208885ac19cSjeremylt     // Basis action
209d1d35e2fSjeremylt     switch (eval_mode) {
210885ac19cSjeremylt       case CEED_EVAL_NONE:
2112b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorSetArray(impl->q_vecs_in[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data_full[i][e * Q * size]));
212885ac19cSjeremylt         break;
213885ac19cSjeremylt       case CEED_EVAL_INTERP:
214885ac19cSjeremylt       case CEED_EVAL_GRAD:
215a915a514Srezgarshakeri       case CEED_EVAL_DIV:
216c4e3f59bSSebastian Grimberg       case CEED_EVAL_CURL:
217a915a514Srezgarshakeri         CeedCallBackend(CeedOperatorFieldGetBasis(op_input_fields[i], &basis));
218a915a514Srezgarshakeri         CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp));
219a915a514Srezgarshakeri         CeedCallBackend(CeedVectorSetArray(impl->e_vecs_in[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data_full[i][e * elem_size * num_comp]));
220c4e3f59bSSebastian Grimberg         CeedCallBackend(CeedBasisApply(basis, 1, CEED_NOTRANSPOSE, eval_mode, impl->e_vecs_in[i], impl->q_vecs_in[i]));
221a915a514Srezgarshakeri         break;
222885ac19cSjeremylt       case CEED_EVAL_WEIGHT:
223885ac19cSjeremylt         break;  // No action
224885ac19cSjeremylt     }
225885ac19cSjeremylt   }
226e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
227885ac19cSjeremylt }
228885ac19cSjeremylt 
229f10650afSjeremylt //------------------------------------------------------------------------------
230f10650afSjeremylt // Output Basis Action
231f10650afSjeremylt //------------------------------------------------------------------------------
2322b730f8bSJeremy L Thompson static inline int CeedOperatorOutputBasis_Ref(CeedInt e, CeedInt Q, CeedQFunctionField *qf_output_fields, CeedOperatorField *op_output_fields,
233d1d35e2fSjeremylt                                               CeedInt num_input_fields, CeedInt num_output_fields, CeedOperator op,
2344fc1f125SJeremy L Thompson                                               CeedScalar *e_data_full[2 * CEED_FIELD_MAX], CeedOperator_Ref *impl) {
235ad70ee2cSJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
236a915a514Srezgarshakeri     CeedInt             elem_size, num_comp;
237d1d35e2fSjeremylt     CeedEvalMode        eval_mode;
238edb2538eSJeremy L Thompson     CeedElemRestriction elem_rstr;
2391d102b48SJeremy L Thompson     CeedBasis           basis;
2401d102b48SJeremy L Thompson 
241a915a514Srezgarshakeri     // Get elem_size, eval_mode
242edb2538eSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &elem_rstr));
243edb2538eSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetElementSize(elem_rstr, &elem_size));
2442b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
245885ac19cSjeremylt     // Basis action
246d1d35e2fSjeremylt     switch (eval_mode) {
247885ac19cSjeremylt       case CEED_EVAL_NONE:
248885ac19cSjeremylt         break;  // No action
249885ac19cSjeremylt       case CEED_EVAL_INTERP:
250885ac19cSjeremylt       case CEED_EVAL_GRAD:
251a915a514Srezgarshakeri       case CEED_EVAL_DIV:
252c4e3f59bSSebastian Grimberg       case CEED_EVAL_CURL:
253a915a514Srezgarshakeri         CeedCallBackend(CeedOperatorFieldGetBasis(op_output_fields[i], &basis));
254a915a514Srezgarshakeri         CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp));
255a915a514Srezgarshakeri         CeedCallBackend(
256a915a514Srezgarshakeri             CeedVectorSetArray(impl->e_vecs_out[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data_full[i + num_input_fields][e * elem_size * num_comp]));
257c4e3f59bSSebastian Grimberg         CeedCallBackend(CeedBasisApply(basis, 1, CEED_TRANSPOSE, eval_mode, impl->q_vecs_out[i], impl->e_vecs_out[i]));
258a915a514Srezgarshakeri         break;
259c042f62fSJeremy L Thompson       // LCOV_EXCL_START
260bbfacfcdSjeremylt       case CEED_EVAL_WEIGHT: {
2614ce2993fSjeremylt         Ceed ceed;
262ad70ee2cSJeremy L Thompson 
2632b730f8bSJeremy L Thompson         CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
2646574a04fSJeremy L Thompson         return CeedError(ceed, CEED_ERROR_BACKEND, "CEED_EVAL_WEIGHT cannot be an output evaluation mode");
2651d102b48SJeremy L Thompson         // LCOV_EXCL_STOP
266885ac19cSjeremylt       }
267885ac19cSjeremylt     }
268885ac19cSjeremylt   }
269e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2701d102b48SJeremy L Thompson }
2711d102b48SJeremy L Thompson 
272f10650afSjeremylt //------------------------------------------------------------------------------
273f10650afSjeremylt // Restore Input Vectors
274f10650afSjeremylt //------------------------------------------------------------------------------
2752b730f8bSJeremy L Thompson static inline int CeedOperatorRestoreInputs_Ref(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields, CeedOperatorField *op_input_fields,
2762b730f8bSJeremy L Thompson                                                 const bool skip_active, CeedScalar *e_data_full[2 * CEED_FIELD_MAX], CeedOperator_Ref *impl) {
277ad70ee2cSJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
278d1d35e2fSjeremylt     CeedEvalMode eval_mode;
2791d102b48SJeremy L Thompson 
2801d102b48SJeremy L Thompson     // Skip active inputs
281d1d35e2fSjeremylt     if (skip_active) {
2821d102b48SJeremy L Thompson       CeedVector vec;
283ad70ee2cSJeremy L Thompson 
2842b730f8bSJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
2852b730f8bSJeremy L Thompson       if (vec == CEED_VECTOR_ACTIVE) continue;
2861d102b48SJeremy L Thompson     }
2871d102b48SJeremy L Thompson     // Restore input
2882b730f8bSJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
289d1d35e2fSjeremylt     if (eval_mode == CEED_EVAL_WEIGHT) {  // Skip
2901d102b48SJeremy L Thompson     } else {
2912b730f8bSJeremy L Thompson       CeedCallBackend(CeedVectorRestoreArrayRead(impl->e_vecs_full[i], (const CeedScalar **)&e_data_full[i]));
2921d102b48SJeremy L Thompson     }
2931d102b48SJeremy L Thompson   }
294e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
2951d102b48SJeremy L Thompson }
2961d102b48SJeremy L Thompson 
297f10650afSjeremylt //------------------------------------------------------------------------------
298f10650afSjeremylt // Operator Apply
299f10650afSjeremylt //------------------------------------------------------------------------------
3002b730f8bSJeremy L Thompson static int CeedOperatorApplyAdd_Ref(CeedOperator op, CeedVector in_vec, CeedVector out_vec, CeedRequest *request) {
301d1d35e2fSjeremylt   CeedInt             Q, num_elem, num_input_fields, num_output_fields, size;
302ad70ee2cSJeremy L Thompson   CeedEvalMode        eval_mode;
303ad70ee2cSJeremy L Thompson   CeedScalar         *e_data_full[2 * CEED_FIELD_MAX] = {NULL};
304ad70ee2cSJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
305ad70ee2cSJeremy L Thompson   CeedQFunction       qf;
306ad70ee2cSJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
307ad70ee2cSJeremy L Thompson   CeedOperator_Ref   *impl;
308ad70ee2cSJeremy L Thompson 
309ad70ee2cSJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
310ad70ee2cSJeremy L Thompson   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
3112b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
3122b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem));
3132b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
3142b730f8bSJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
3151d102b48SJeremy L Thompson 
3161d102b48SJeremy L Thompson   // Setup
3172b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorSetup_Ref(op));
3181d102b48SJeremy L Thompson 
3190b454692Sjeremylt   // Restriction only operator
320edb2538eSJeremy L Thompson   if (impl->is_identity_rstr_op) {
321edb2538eSJeremy L Thompson     CeedElemRestriction elem_rstr;
322ad70ee2cSJeremy L Thompson 
323edb2538eSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[0], &elem_rstr));
324edb2538eSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_NOTRANSPOSE, in_vec, impl->e_vecs_full[0], request));
325edb2538eSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[0], &elem_rstr));
326edb2538eSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_TRANSPOSE, impl->e_vecs_full[0], out_vec, request));
3270b454692Sjeremylt     return CEED_ERROR_SUCCESS;
3280b454692Sjeremylt   }
3290b454692Sjeremylt 
3301d102b48SJeremy L Thompson   // Input Evecs and Restriction
3312b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorSetupInputs_Ref(num_input_fields, qf_input_fields, op_input_fields, in_vec, false, e_data_full, impl, request));
3321d102b48SJeremy L Thompson 
3331d102b48SJeremy L Thompson   // Output Evecs
334d1d35e2fSjeremylt   for (CeedInt i = 0; i < num_output_fields; i++) {
3352b730f8bSJeremy L Thompson     CeedCallBackend(CeedVectorGetArrayWrite(impl->e_vecs_full[i + impl->num_inputs], CEED_MEM_HOST, &e_data_full[i + num_input_fields]));
3361d102b48SJeremy L Thompson   }
3371d102b48SJeremy L Thompson 
3381d102b48SJeremy L Thompson   // Loop through elements
339d1d35e2fSjeremylt   for (CeedInt e = 0; e < num_elem; e++) {
3401d102b48SJeremy L Thompson     // Output pointers
341d1d35e2fSjeremylt     for (CeedInt i = 0; i < num_output_fields; i++) {
3422b730f8bSJeremy L Thompson       CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
343d1d35e2fSjeremylt       if (eval_mode == CEED_EVAL_NONE) {
3442b730f8bSJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[i], &size));
3452b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorSetArray(impl->q_vecs_out[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data_full[i + num_input_fields][e * Q * size]));
3461d102b48SJeremy L Thompson       }
3471d102b48SJeremy L Thompson     }
3481d102b48SJeremy L Thompson 
34916911fdaSjeremylt     // Input basis apply
3502b730f8bSJeremy L Thompson     CeedCallBackend(CeedOperatorInputBasis_Ref(e, Q, qf_input_fields, op_input_fields, num_input_fields, false, e_data_full, impl));
35116911fdaSjeremylt 
3521d102b48SJeremy L Thompson     // Q function
3530b454692Sjeremylt     if (!impl->is_identity_qf) {
3542b730f8bSJeremy L Thompson       CeedCallBackend(CeedQFunctionApply(qf, Q, impl->q_vecs_in, impl->q_vecs_out));
35516911fdaSjeremylt     }
3561d102b48SJeremy L Thompson 
3571d102b48SJeremy L Thompson     // Output basis apply
3582b730f8bSJeremy L Thompson     CeedCallBackend(
3592b730f8bSJeremy L Thompson         CeedOperatorOutputBasis_Ref(e, Q, qf_output_fields, op_output_fields, num_input_fields, num_output_fields, op, e_data_full, impl));
3601d102b48SJeremy L Thompson   }
361885ac19cSjeremylt 
362885ac19cSjeremylt   // Output restriction
363d1d35e2fSjeremylt   for (CeedInt i = 0; i < num_output_fields; i++) {
364ad70ee2cSJeremy L Thompson     CeedVector          vec;
365edb2538eSJeremy L Thompson     CeedElemRestriction elem_rstr;
366ad70ee2cSJeremy L Thompson 
367d1d35e2fSjeremylt     // Restore Evec
3682b730f8bSJeremy L Thompson     CeedCallBackend(CeedVectorRestoreArray(impl->e_vecs_full[i + impl->num_inputs], &e_data_full[i + num_input_fields]));
369d1bcdac9Sjeremylt     // Get output vector
3702b730f8bSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec));
371668048e2SJed Brown     // Active
3722b730f8bSJeremy L Thompson     if (vec == CEED_VECTOR_ACTIVE) vec = out_vec;
3737ca8db16Sjeremylt     // Restrict
374edb2538eSJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &elem_rstr));
375edb2538eSJeremy L Thompson     CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_TRANSPOSE, impl->e_vecs_full[i + impl->num_inputs], vec, request));
376885ac19cSjeremylt   }
377885ac19cSjeremylt 
3787ca8db16Sjeremylt   // Restore input arrays
3792b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorRestoreInputs_Ref(num_input_fields, qf_input_fields, op_input_fields, false, e_data_full, impl));
380e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
38121617c04Sjeremylt }
38221617c04Sjeremylt 
383f10650afSjeremylt //------------------------------------------------------------------------------
38470a7ffb3SJeremy L Thompson // Core code for assembling linear QFunction
385f10650afSjeremylt //------------------------------------------------------------------------------
3862b730f8bSJeremy L Thompson static inline int CeedOperatorLinearAssembleQFunctionCore_Ref(CeedOperator op, bool build_objects, CeedVector *assembled, CeedElemRestriction *rstr,
38770a7ffb3SJeremy L Thompson                                                               CeedRequest *request) {
388e910d748SJeremy L Thompson   Ceed                ceed, ceed_parent;
389e984cf9aSJeremy L Thompson   CeedSize            q_size;
390ad70ee2cSJeremy L Thompson   CeedInt             num_active_in, num_active_out, Q, num_elem, num_input_fields, num_output_fields, size;
391ad70ee2cSJeremy L Thompson   CeedScalar         *assembled_array, *e_data_full[2 * CEED_FIELD_MAX] = {NULL};
392ad70ee2cSJeremy L Thompson   CeedVector         *active_in;
393ad70ee2cSJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
394ad70ee2cSJeremy L Thompson   CeedQFunction       qf;
395ad70ee2cSJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
396ad70ee2cSJeremy L Thompson   CeedOperator_Ref   *impl;
397ad70ee2cSJeremy L Thompson 
3982b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
399e910d748SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFallbackParentCeed(op, &ceed_parent));
400e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
401e984cf9aSJeremy L Thompson   active_in     = impl->qf_active_in;
402e984cf9aSJeremy L Thompson   num_active_in = impl->num_active_in, num_active_out = impl->num_active_out;
403e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
404e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
405e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem));
406e984cf9aSJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
407e984cf9aSJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
4081d102b48SJeremy L Thompson 
4091d102b48SJeremy L Thompson   // Setup
4102b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorSetup_Ref(op));
4111d102b48SJeremy L Thompson 
412506b1a0cSSebastian Grimberg   // Check for restriction only operator
413506b1a0cSSebastian Grimberg   CeedCheck(!impl->is_identity_rstr_op, ceed, CEED_ERROR_BACKEND, "Assembling restriction only operators is not supported");
41416911fdaSjeremylt 
4151d102b48SJeremy L Thompson   // Input Evecs and Restriction
4162b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorSetupInputs_Ref(num_input_fields, qf_input_fields, op_input_fields, NULL, true, e_data_full, impl, request));
4171d102b48SJeremy L Thompson 
4181d102b48SJeremy L Thompson   // Count number of active input fields
419bb219a0fSJeremy L Thompson   if (!num_active_in) {
420d1d35e2fSjeremylt     for (CeedInt i = 0; i < num_input_fields; i++) {
421ad70ee2cSJeremy L Thompson       CeedScalar *q_vec_array;
422ad70ee2cSJeremy L Thompson       CeedVector  vec;
423ad70ee2cSJeremy L Thompson 
4241d102b48SJeremy L Thompson       // Get input vector
4252b730f8bSJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
4261d102b48SJeremy L Thompson       // Check if active input
4271d102b48SJeremy L Thompson       if (vec == CEED_VECTOR_ACTIVE) {
4282b730f8bSJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_input_fields[i], &size));
4292b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorSetValue(impl->q_vecs_in[i], 0.0));
430ad70ee2cSJeremy L Thompson         CeedCallBackend(CeedVectorGetArray(impl->q_vecs_in[i], CEED_MEM_HOST, &q_vec_array));
4312b730f8bSJeremy L Thompson         CeedCallBackend(CeedRealloc(num_active_in + size, &active_in));
4321d102b48SJeremy L Thompson         for (CeedInt field = 0; field < size; field++) {
433d2643443SJeremy L Thompson           q_size = (CeedSize)Q;
434e910d748SJeremy L Thompson           CeedCallBackend(CeedVectorCreate(ceed_parent, q_size, &active_in[num_active_in + field]));
435ad70ee2cSJeremy L Thompson           CeedCallBackend(CeedVectorSetArray(active_in[num_active_in + field], CEED_MEM_HOST, CEED_USE_POINTER, &q_vec_array[field * Q]));
4361d102b48SJeremy L Thompson         }
437d1d35e2fSjeremylt         num_active_in += size;
438ad70ee2cSJeremy L Thompson         CeedCallBackend(CeedVectorRestoreArray(impl->q_vecs_in[i], &q_vec_array));
4391d102b48SJeremy L Thompson       }
4401d102b48SJeremy L Thompson     }
4414fc1f125SJeremy L Thompson     impl->num_active_in = num_active_in;
442bb219a0fSJeremy L Thompson     impl->qf_active_in  = active_in;
443bb219a0fSJeremy L Thompson   }
4441d102b48SJeremy L Thompson 
4451d102b48SJeremy L Thompson   // Count number of active output fields
446bb219a0fSJeremy L Thompson   if (!num_active_out) {
447d1d35e2fSjeremylt     for (CeedInt i = 0; i < num_output_fields; i++) {
448ad70ee2cSJeremy L Thompson       CeedVector vec;
449ad70ee2cSJeremy L Thompson 
4501d102b48SJeremy L Thompson       // Get output vector
4512b730f8bSJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec));
4521d102b48SJeremy L Thompson       // Check if active output
4531d102b48SJeremy L Thompson       if (vec == CEED_VECTOR_ACTIVE) {
4542b730f8bSJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[i], &size));
455d1d35e2fSjeremylt         num_active_out += size;
4561d102b48SJeremy L Thompson       }
4571d102b48SJeremy L Thompson     }
4584fc1f125SJeremy L Thompson     impl->num_active_out = num_active_out;
459bb219a0fSJeremy L Thompson   }
4601d102b48SJeremy L Thompson 
4611d102b48SJeremy L Thompson   // Check sizes
4626574a04fSJeremy L Thompson   CeedCheck(num_active_in > 0 && num_active_out > 0, ceed, CEED_ERROR_BACKEND, "Cannot assemble QFunction without active inputs and outputs");
4631d102b48SJeremy L Thompson 
46470a7ffb3SJeremy L Thompson   // Build objects if needed
46570a7ffb3SJeremy L Thompson   if (build_objects) {
466ad70ee2cSJeremy L Thompson     const CeedSize l_size     = (CeedSize)num_elem * Q * num_active_in * num_active_out;
467d1d35e2fSjeremylt     CeedInt        strides[3] = {1, Q, num_active_in * num_active_out * Q}; /* *NOPAD* */
468ad70ee2cSJeremy L Thompson 
469ad70ee2cSJeremy L Thompson     // Create output restriction
470e910d748SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionCreateStrided(ceed_parent, num_elem, Q, num_active_in * num_active_out,
4712b730f8bSJeremy L Thompson                                                      num_active_in * num_active_out * num_elem * Q, strides, rstr));
4721d102b48SJeremy L Thompson     // Create assembled vector
473e910d748SJeremy L Thompson     CeedCallBackend(CeedVectorCreate(ceed_parent, l_size, assembled));
47470a7ffb3SJeremy L Thompson   }
47570a7ffb3SJeremy L Thompson   // Clear output vector
4762b730f8bSJeremy L Thompson   CeedCallBackend(CeedVectorSetValue(*assembled, 0.0));
477ad70ee2cSJeremy L Thompson   CeedCallBackend(CeedVectorGetArray(*assembled, CEED_MEM_HOST, &assembled_array));
4781d102b48SJeremy L Thompson 
4791d102b48SJeremy L Thompson   // Loop through elements
480d1d35e2fSjeremylt   for (CeedInt e = 0; e < num_elem; e++) {
4811d102b48SJeremy L Thompson     // Input basis apply
4822b730f8bSJeremy L Thompson     CeedCallBackend(CeedOperatorInputBasis_Ref(e, Q, qf_input_fields, op_input_fields, num_input_fields, true, e_data_full, impl));
4831d102b48SJeremy L Thompson 
4841d102b48SJeremy L Thompson     // Assemble QFunction
485d1d35e2fSjeremylt     for (CeedInt in = 0; in < num_active_in; in++) {
4861d102b48SJeremy L Thompson       // Set Inputs
4872b730f8bSJeremy L Thompson       CeedCallBackend(CeedVectorSetValue(active_in[in], 1.0));
488d1d35e2fSjeremylt       if (num_active_in > 1) {
4892b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorSetValue(active_in[(in + num_active_in - 1) % num_active_in], 0.0));
49042ea3801Sjeremylt       }
491506b1a0cSSebastian Grimberg       if (!impl->is_identity_qf) {
4921d102b48SJeremy L Thompson         // Set Outputs
493d1d35e2fSjeremylt         for (CeedInt out = 0; out < num_output_fields; out++) {
494ad70ee2cSJeremy L Thompson           CeedVector vec;
495ad70ee2cSJeremy L Thompson 
4961d102b48SJeremy L Thompson           // Get output vector
4972b730f8bSJeremy L Thompson           CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[out], &vec));
4981d102b48SJeremy L Thompson           // Check if active output
4991d102b48SJeremy L Thompson           if (vec == CEED_VECTOR_ACTIVE) {
500ad70ee2cSJeremy L Thompson             CeedCallBackend(CeedVectorSetArray(impl->q_vecs_out[out], CEED_MEM_HOST, CEED_USE_POINTER, assembled_array));
5012b730f8bSJeremy L Thompson             CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[out], &size));
502ad70ee2cSJeremy L Thompson             assembled_array += size * Q;  // Advance the pointer by the size of the output
5031d102b48SJeremy L Thompson           }
5041d102b48SJeremy L Thompson         }
5051d102b48SJeremy L Thompson         // Apply QFunction
5062b730f8bSJeremy L Thompson         CeedCallBackend(CeedQFunctionApply(qf, Q, impl->q_vecs_in, impl->q_vecs_out));
507506b1a0cSSebastian Grimberg       } else {
508506b1a0cSSebastian Grimberg         const CeedScalar *q_vec_array;
509506b1a0cSSebastian Grimberg 
510506b1a0cSSebastian Grimberg         // Copy Identity Outputs
511506b1a0cSSebastian Grimberg         CeedCallBackend(CeedQFunctionFieldGetSize(qf_output_fields[0], &size));
512506b1a0cSSebastian Grimberg         CeedCallBackend(CeedVectorGetArrayRead(impl->q_vecs_out[0], CEED_MEM_HOST, &q_vec_array));
513506b1a0cSSebastian Grimberg         for (CeedInt i = 0; i < size * Q; i++) assembled_array[i] = q_vec_array[i];
514506b1a0cSSebastian Grimberg         CeedCallBackend(CeedVectorRestoreArrayRead(impl->q_vecs_out[0], &q_vec_array));
515506b1a0cSSebastian Grimberg         assembled_array += size * Q;
516506b1a0cSSebastian Grimberg       }
5171d102b48SJeremy L Thompson     }
5181d102b48SJeremy L Thompson   }
5191d102b48SJeremy L Thompson 
5201d102b48SJeremy L Thompson   // Un-set output Qvecs to prevent accidental overwrite of Assembled
521506b1a0cSSebastian Grimberg   if (!impl->is_identity_qf) {
522d1d35e2fSjeremylt     for (CeedInt out = 0; out < num_output_fields; out++) {
523ad70ee2cSJeremy L Thompson       CeedVector vec;
524ad70ee2cSJeremy L Thompson 
5251d102b48SJeremy L Thompson       // Get output vector
5262b730f8bSJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[out], &vec));
5271d102b48SJeremy L Thompson       // Check if active output
528056ea4bdSJeremy L Thompson       if (vec == CEED_VECTOR_ACTIVE && num_elem > 0) {
5292b730f8bSJeremy L Thompson         CeedCallBackend(CeedVectorTakeArray(impl->q_vecs_out[out], CEED_MEM_HOST, NULL));
5301d102b48SJeremy L Thompson       }
5311d102b48SJeremy L Thompson     }
532506b1a0cSSebastian Grimberg   }
5331d102b48SJeremy L Thompson 
5341d102b48SJeremy L Thompson   // Restore input arrays
5352b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorRestoreInputs_Ref(num_input_fields, qf_input_fields, op_input_fields, true, e_data_full, impl));
5361d102b48SJeremy L Thompson 
5371d102b48SJeremy L Thompson   // Restore output
538ad70ee2cSJeremy L Thompson   CeedCallBackend(CeedVectorRestoreArray(*assembled, &assembled_array));
539e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
5401d102b48SJeremy L Thompson }
5411d102b48SJeremy L Thompson 
542f10650afSjeremylt //------------------------------------------------------------------------------
54370a7ffb3SJeremy L Thompson // Assemble Linear QFunction
54470a7ffb3SJeremy L Thompson //------------------------------------------------------------------------------
5452b730f8bSJeremy L Thompson static int CeedOperatorLinearAssembleQFunction_Ref(CeedOperator op, CeedVector *assembled, CeedElemRestriction *rstr, CeedRequest *request) {
5462b730f8bSJeremy L Thompson   return CeedOperatorLinearAssembleQFunctionCore_Ref(op, true, assembled, rstr, request);
54770a7ffb3SJeremy L Thompson }
54870a7ffb3SJeremy L Thompson 
54970a7ffb3SJeremy L Thompson //------------------------------------------------------------------------------
55070a7ffb3SJeremy L Thompson // Update Assembled Linear QFunction
55170a7ffb3SJeremy L Thompson //------------------------------------------------------------------------------
5522b730f8bSJeremy L Thompson static int CeedOperatorLinearAssembleQFunctionUpdate_Ref(CeedOperator op, CeedVector assembled, CeedElemRestriction rstr, CeedRequest *request) {
5532b730f8bSJeremy L Thompson   return CeedOperatorLinearAssembleQFunctionCore_Ref(op, false, &assembled, &rstr, request);
55470a7ffb3SJeremy L Thompson }
55570a7ffb3SJeremy L Thompson 
55670a7ffb3SJeremy L Thompson //------------------------------------------------------------------------------
55748acf710SJeremy L Thompson // Setup Input/Output Fields
55848acf710SJeremy L Thompson //------------------------------------------------------------------------------
55948acf710SJeremy L Thompson static int CeedOperatorSetupFieldsAtPoints_Ref(CeedQFunction qf, CeedOperator op, bool is_input, CeedVector *e_vecs_full, CeedVector *e_vecs,
56048acf710SJeremy L Thompson                                                CeedVector *q_vecs, CeedInt start_e, CeedInt num_fields, CeedInt Q) {
56148acf710SJeremy L Thompson   Ceed                ceed;
56248acf710SJeremy L Thompson   CeedSize            e_size, q_size;
56348acf710SJeremy L Thompson   CeedInt             max_num_points, num_comp, size, P;
56448acf710SJeremy L Thompson   CeedQFunctionField *qf_fields;
56548acf710SJeremy L Thompson   CeedOperatorField  *op_fields;
56648acf710SJeremy L Thompson 
567e910d748SJeremy L Thompson   {
568e910d748SJeremy L Thompson     Ceed ceed_parent;
569e910d748SJeremy L Thompson 
57048acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
571e910d748SJeremy L Thompson     CeedCallBackend(CeedGetParent(ceed, &ceed_parent));
572e910d748SJeremy L Thompson     if (ceed_parent) ceed = ceed_parent;
573e910d748SJeremy L Thompson   }
57448acf710SJeremy L Thompson   if (is_input) {
57548acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorGetFields(op, NULL, &op_fields, NULL, NULL));
57648acf710SJeremy L Thompson     CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_fields, NULL, NULL));
57748acf710SJeremy L Thompson   } else {
57848acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorGetFields(op, NULL, NULL, NULL, &op_fields));
57948acf710SJeremy L Thompson     CeedCallBackend(CeedQFunctionGetFields(qf, NULL, NULL, NULL, &qf_fields));
58048acf710SJeremy L Thompson   }
58148acf710SJeremy L Thompson 
58248acf710SJeremy L Thompson   // Get max number of points
58348acf710SJeremy L Thompson   {
58448acf710SJeremy L Thompson     CeedInt             dim;
58548acf710SJeremy L Thompson     CeedElemRestriction rstr_points = NULL;
58648acf710SJeremy L Thompson     CeedOperator_Ref   *impl;
58748acf710SJeremy L Thompson 
58848acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorAtPointsGetPoints(op, &rstr_points, NULL));
58948acf710SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetMaxPointsInElement(rstr_points, &max_num_points));
59048acf710SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetNumComponents(rstr_points, &dim));
59148acf710SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionDestroy(&rstr_points));
59248acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorGetData(op, &impl));
59348acf710SJeremy L Thompson     if (is_input) CeedCallBackend(CeedVectorCreate(ceed, dim * max_num_points, &impl->point_coords_elem));
59448acf710SJeremy L Thompson   }
59548acf710SJeremy L Thompson 
59648acf710SJeremy L Thompson   // Loop over fields
59748acf710SJeremy L Thompson   for (CeedInt i = 0; i < num_fields; i++) {
59848acf710SJeremy L Thompson     CeedEvalMode eval_mode;
59948acf710SJeremy L Thompson     CeedBasis    basis;
60048acf710SJeremy L Thompson 
60148acf710SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_fields[i], &eval_mode));
60248acf710SJeremy L Thompson     if (eval_mode != CEED_EVAL_WEIGHT) {
60348acf710SJeremy L Thompson       CeedRestrictionType rstr_type;
60448acf710SJeremy L Thompson       CeedElemRestriction elem_rstr;
60548acf710SJeremy L Thompson 
60648acf710SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_fields[i], &elem_rstr));
60748acf710SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionGetNumComponents(elem_rstr, &num_comp));
60848acf710SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionGetType(elem_rstr, &rstr_type));
60948acf710SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionCreateVector(elem_rstr, NULL, &e_vecs_full[i + start_e]));
61048acf710SJeremy L Thompson     }
61148acf710SJeremy L Thompson 
61248acf710SJeremy L Thompson     switch (eval_mode) {
61338e83183SJeremy L Thompson       case CEED_EVAL_NONE: {
61438e83183SJeremy L Thompson         CeedVector vec;
61538e83183SJeremy L Thompson 
61648acf710SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size));
61748acf710SJeremy L Thompson         e_size = (CeedSize)max_num_points * size;
61848acf710SJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, e_size, &e_vecs[i]));
61938e83183SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetVector(op_fields[i], &vec));
62038e83183SJeremy L Thompson         if (vec == CEED_VECTOR_ACTIVE || !is_input) {
62138e83183SJeremy L Thompson           CeedCallBackend(CeedVectorReferenceCopy(e_vecs[i], &q_vecs[i]));
62238e83183SJeremy L Thompson         } else {
62348acf710SJeremy L Thompson           q_size = (CeedSize)max_num_points * size;
62448acf710SJeremy L Thompson           CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
62538e83183SJeremy L Thompson         }
62648acf710SJeremy L Thompson         break;
62738e83183SJeremy L Thompson       }
62848acf710SJeremy L Thompson       case CEED_EVAL_INTERP:
62948acf710SJeremy L Thompson       case CEED_EVAL_GRAD:
63048acf710SJeremy L Thompson       case CEED_EVAL_DIV:
63148acf710SJeremy L Thompson       case CEED_EVAL_CURL:
63248acf710SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
63348acf710SJeremy L Thompson         CeedCallBackend(CeedQFunctionFieldGetSize(qf_fields[i], &size));
63448acf710SJeremy L Thompson         CeedCallBackend(CeedBasisGetNumNodes(basis, &P));
63548acf710SJeremy L Thompson         CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp));
63648acf710SJeremy L Thompson         e_size = (CeedSize)P * num_comp;
63748acf710SJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, e_size, &e_vecs[i]));
63848acf710SJeremy L Thompson         q_size = (CeedSize)max_num_points * size;
63948acf710SJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
64048acf710SJeremy L Thompson         break;
64148acf710SJeremy L Thompson       case CEED_EVAL_WEIGHT:  // Only on input fields
64248acf710SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_fields[i], &basis));
64348acf710SJeremy L Thompson         q_size = (CeedSize)max_num_points;
64448acf710SJeremy L Thompson         CeedCallBackend(CeedVectorCreate(ceed, q_size, &q_vecs[i]));
64548acf710SJeremy L Thompson         CeedCallBackend(
64648acf710SJeremy L Thompson             CeedBasisApplyAtPoints(basis, max_num_points, CEED_NOTRANSPOSE, CEED_EVAL_WEIGHT, CEED_VECTOR_NONE, CEED_VECTOR_NONE, q_vecs[i]));
64748acf710SJeremy L Thompson         break;
64848acf710SJeremy L Thompson     }
64948acf710SJeremy L Thompson     if (is_input && e_vecs[i]) {
65048acf710SJeremy L Thompson       CeedCallBackend(CeedVectorSetArray(e_vecs[i], CEED_MEM_HOST, CEED_COPY_VALUES, NULL));
65148acf710SJeremy L Thompson     }
65248acf710SJeremy L Thompson   }
65348acf710SJeremy L Thompson   return CEED_ERROR_SUCCESS;
65448acf710SJeremy L Thompson }
65548acf710SJeremy L Thompson 
65648acf710SJeremy L Thompson //------------------------------------------------------------------------------
65748acf710SJeremy L Thompson // Setup Operator
65848acf710SJeremy L Thompson //------------------------------------------------------------------------------
65948acf710SJeremy L Thompson static int CeedOperatorSetupAtPoints_Ref(CeedOperator op) {
66048acf710SJeremy L Thompson   bool                is_setup_done;
66148acf710SJeremy L Thompson   Ceed                ceed;
66248acf710SJeremy L Thompson   CeedInt             Q, num_input_fields, num_output_fields;
66348acf710SJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
66448acf710SJeremy L Thompson   CeedQFunction       qf;
66548acf710SJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
66648acf710SJeremy L Thompson   CeedOperator_Ref   *impl;
66748acf710SJeremy L Thompson 
66848acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorIsSetupDone(op, &is_setup_done));
66948acf710SJeremy L Thompson   if (is_setup_done) return CEED_ERROR_SUCCESS;
67048acf710SJeremy L Thompson 
67148acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
67248acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
67348acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
67448acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumQuadraturePoints(op, &Q));
67548acf710SJeremy L Thompson   CeedCallBackend(CeedQFunctionIsIdentity(qf, &impl->is_identity_qf));
67648acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
67748acf710SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
67848acf710SJeremy L Thompson 
67948acf710SJeremy L Thompson   // Allocate
68048acf710SJeremy L Thompson   CeedCallBackend(CeedCalloc(num_input_fields + num_output_fields, &impl->e_vecs_full));
68148acf710SJeremy L Thompson 
68248acf710SJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->input_states));
68348acf710SJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->e_vecs_in));
68448acf710SJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->e_vecs_out));
68548acf710SJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_in));
68648acf710SJeremy L Thompson   CeedCallBackend(CeedCalloc(CEED_FIELD_MAX, &impl->q_vecs_out));
68748acf710SJeremy L Thompson 
68848acf710SJeremy L Thompson   impl->num_inputs  = num_input_fields;
68948acf710SJeremy L Thompson   impl->num_outputs = num_output_fields;
69048acf710SJeremy L Thompson 
69148acf710SJeremy L Thompson   // Set up infield and outfield pointer arrays
69248acf710SJeremy L Thompson   // Infields
69348acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupFieldsAtPoints_Ref(qf, op, true, impl->e_vecs_full, impl->e_vecs_in, impl->q_vecs_in, 0, num_input_fields, Q));
69448acf710SJeremy L Thompson   // Outfields
69548acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupFieldsAtPoints_Ref(qf, op, false, impl->e_vecs_full, impl->e_vecs_out, impl->q_vecs_out, num_input_fields,
69648acf710SJeremy L Thompson                                                       num_output_fields, Q));
69748acf710SJeremy L Thompson 
69848acf710SJeremy L Thompson   // Identity QFunctions
69948acf710SJeremy L Thompson   if (impl->is_identity_qf) {
70048acf710SJeremy L Thompson     CeedCallBackend(CeedVectorReferenceCopy(impl->q_vecs_in[0], &impl->q_vecs_out[0]));
70138e83183SJeremy L Thompson     CeedCallBackend(CeedVectorReferenceCopy(impl->q_vecs_in[0], &impl->e_vecs_out[0]));
70248acf710SJeremy L Thompson   }
70348acf710SJeremy L Thompson 
70448acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorSetSetupDone(op));
70548acf710SJeremy L Thompson   return CEED_ERROR_SUCCESS;
70648acf710SJeremy L Thompson }
70748acf710SJeremy L Thompson 
70848acf710SJeremy L Thompson //------------------------------------------------------------------------------
70948acf710SJeremy L Thompson // Setup Input Fields
71048acf710SJeremy L Thompson //------------------------------------------------------------------------------
71148acf710SJeremy L Thompson static inline int CeedOperatorSetupInputsAtPoints_Ref(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields,
71248acf710SJeremy L Thompson                                                       CeedOperatorField *op_input_fields, CeedVector in_vec, CeedScalar *e_data[2 * CEED_FIELD_MAX],
71348acf710SJeremy L Thompson                                                       CeedOperator_Ref *impl, CeedRequest *request) {
71448acf710SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
71548acf710SJeremy L Thompson     uint64_t            state;
71648acf710SJeremy L Thompson     CeedEvalMode        eval_mode;
71748acf710SJeremy L Thompson     CeedVector          vec;
71848acf710SJeremy L Thompson     CeedElemRestriction elem_rstr;
71948acf710SJeremy L Thompson 
72048acf710SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
72148acf710SJeremy L Thompson     if (eval_mode == CEED_EVAL_WEIGHT) {  // Skip
72248acf710SJeremy L Thompson     } else {
72348acf710SJeremy L Thompson       // Get input vector
72448acf710SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
72548acf710SJeremy L Thompson       CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr));
72648acf710SJeremy L Thompson       if (vec != CEED_VECTOR_ACTIVE) {
72748acf710SJeremy L Thompson         // Restrict
72848acf710SJeremy L Thompson         CeedCallBackend(CeedVectorGetState(vec, &state));
72948acf710SJeremy L Thompson         if (state != impl->input_states[i]) {
73048acf710SJeremy L Thompson           CeedCallBackend(CeedElemRestrictionApply(elem_rstr, CEED_NOTRANSPOSE, vec, impl->e_vecs_full[i], request));
73148acf710SJeremy L Thompson           impl->input_states[i] = state;
73248acf710SJeremy L Thompson         }
73348acf710SJeremy L Thompson         // Get evec
73448acf710SJeremy L Thompson         CeedCallBackend(CeedVectorGetArrayRead(impl->e_vecs_full[i], CEED_MEM_HOST, (const CeedScalar **)&e_data[i]));
73548acf710SJeremy L Thompson       }
73648acf710SJeremy L Thompson     }
73748acf710SJeremy L Thompson   }
73848acf710SJeremy L Thompson   return CEED_ERROR_SUCCESS;
73948acf710SJeremy L Thompson }
74048acf710SJeremy L Thompson 
74148acf710SJeremy L Thompson //------------------------------------------------------------------------------
74248acf710SJeremy L Thompson // Input Basis Action
74348acf710SJeremy L Thompson //------------------------------------------------------------------------------
74448acf710SJeremy L Thompson static inline int CeedOperatorInputBasisAtPoints_Ref(CeedInt e, CeedInt num_points_offset, CeedInt num_points, CeedQFunctionField *qf_input_fields,
74548acf710SJeremy L Thompson                                                      CeedOperatorField *op_input_fields, CeedInt num_input_fields, CeedVector in_vec,
74648acf710SJeremy L Thompson                                                      CeedVector point_coords_elem, bool skip_active, CeedScalar *e_data[2 * CEED_FIELD_MAX],
74748acf710SJeremy L Thompson                                                      CeedOperator_Ref *impl, CeedRequest *request) {
74848acf710SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
74948acf710SJeremy L Thompson     bool                is_active_input = false;
75048acf710SJeremy L Thompson     CeedInt             elem_size, size, num_comp;
75148acf710SJeremy L Thompson     CeedRestrictionType rstr_type;
75248acf710SJeremy L Thompson     CeedEvalMode        eval_mode;
75348acf710SJeremy L Thompson     CeedVector          vec;
75448acf710SJeremy L Thompson     CeedElemRestriction elem_rstr;
75548acf710SJeremy L Thompson     CeedBasis           basis;
75648acf710SJeremy L Thompson 
75748acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
75848acf710SJeremy L Thompson     // Skip active input
75948acf710SJeremy L Thompson     is_active_input = vec == CEED_VECTOR_ACTIVE;
76048acf710SJeremy L Thompson     if (skip_active && is_active_input) continue;
76148acf710SJeremy L Thompson 
76248acf710SJeremy L Thompson     // Get elem_size, eval_mode, size
76348acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_input_fields[i], &elem_rstr));
76448acf710SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetType(elem_rstr, &rstr_type));
76548acf710SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
76648acf710SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetSize(qf_input_fields[i], &size));
76748acf710SJeremy L Thompson     // Restrict block active input
76848acf710SJeremy L Thompson     if (is_active_input) {
76948acf710SJeremy L Thompson       if (rstr_type == CEED_RESTRICTION_POINTS) {
77048acf710SJeremy L Thompson         CeedCallBackend(CeedElemRestrictionApplyAtPointsInElement(elem_rstr, e, CEED_NOTRANSPOSE, in_vec, impl->e_vecs_in[i], request));
77148acf710SJeremy L Thompson       } else {
77248acf710SJeremy L Thompson         CeedCallBackend(CeedElemRestrictionApplyBlock(elem_rstr, e, CEED_NOTRANSPOSE, in_vec, impl->e_vecs_in[i], request));
77348acf710SJeremy L Thompson       }
77448acf710SJeremy L Thompson     }
77548acf710SJeremy L Thompson     // Basis action
77648acf710SJeremy L Thompson     switch (eval_mode) {
77748acf710SJeremy L Thompson       case CEED_EVAL_NONE:
77848acf710SJeremy L Thompson         if (!is_active_input) {
77948acf710SJeremy L Thompson           CeedCallBackend(CeedVectorSetArray(impl->q_vecs_in[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data[i][num_points_offset * size]));
78048acf710SJeremy L Thompson         }
78148acf710SJeremy L Thompson         break;
78248acf710SJeremy L Thompson       // Note - these basis eval modes require FEM fields
78348acf710SJeremy L Thompson       case CEED_EVAL_INTERP:
78448acf710SJeremy L Thompson       case CEED_EVAL_GRAD:
78548acf710SJeremy L Thompson       case CEED_EVAL_DIV:
78648acf710SJeremy L Thompson       case CEED_EVAL_CURL:
78748acf710SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_input_fields[i], &basis));
78848acf710SJeremy L Thompson         if (!is_active_input) {
78948acf710SJeremy L Thompson           CeedCallBackend(CeedBasisGetNumComponents(basis, &num_comp));
79048acf710SJeremy L Thompson           CeedCallBackend(CeedElemRestrictionGetElementSize(elem_rstr, &elem_size));
79148acf710SJeremy L Thompson           CeedCallBackend(CeedVectorSetArray(impl->e_vecs_in[i], CEED_MEM_HOST, CEED_USE_POINTER, &e_data[i][e * elem_size * num_comp]));
79248acf710SJeremy L Thompson         }
79348acf710SJeremy L Thompson         CeedCallBackend(
79448acf710SJeremy L Thompson             CeedBasisApplyAtPoints(basis, num_points, CEED_NOTRANSPOSE, eval_mode, point_coords_elem, impl->e_vecs_in[i], impl->q_vecs_in[i]));
79548acf710SJeremy L Thompson         break;
79648acf710SJeremy L Thompson       case CEED_EVAL_WEIGHT:
79748acf710SJeremy L Thompson         break;  // No action
79848acf710SJeremy L Thompson     }
79948acf710SJeremy L Thompson   }
80048acf710SJeremy L Thompson   return CEED_ERROR_SUCCESS;
80148acf710SJeremy L Thompson }
80248acf710SJeremy L Thompson 
80348acf710SJeremy L Thompson //------------------------------------------------------------------------------
80448acf710SJeremy L Thompson // Output Basis Action
80548acf710SJeremy L Thompson //------------------------------------------------------------------------------
80648acf710SJeremy L Thompson static inline int CeedOperatorOutputBasisAtPoints_Ref(CeedInt e, CeedInt num_points_offset, CeedInt num_points, CeedQFunctionField *qf_output_fields,
80748acf710SJeremy L Thompson                                                       CeedOperatorField *op_output_fields, CeedInt num_input_fields, CeedInt num_output_fields,
80848acf710SJeremy L Thompson                                                       CeedOperator op, CeedVector out_vec, CeedVector point_coords_elem, CeedOperator_Ref *impl,
80948acf710SJeremy L Thompson                                                       CeedRequest *request) {
81048acf710SJeremy L Thompson   for (CeedInt i = 0; i < num_output_fields; i++) {
81148acf710SJeremy L Thompson     CeedRestrictionType rstr_type;
81248acf710SJeremy L Thompson     CeedEvalMode        eval_mode;
81348acf710SJeremy L Thompson     CeedVector          vec;
81448acf710SJeremy L Thompson     CeedElemRestriction elem_rstr;
81548acf710SJeremy L Thompson     CeedBasis           basis;
81648acf710SJeremy L Thompson 
81748acf710SJeremy L Thompson     // Get elem_size, eval_mode, size
81848acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetElemRestriction(op_output_fields[i], &elem_rstr));
81948acf710SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_output_fields[i], &eval_mode));
82048acf710SJeremy L Thompson     // Basis action
82148acf710SJeremy L Thompson     switch (eval_mode) {
82248acf710SJeremy L Thompson       case CEED_EVAL_NONE:
82348acf710SJeremy L Thompson         break;  // No action
82448acf710SJeremy L Thompson       case CEED_EVAL_INTERP:
82548acf710SJeremy L Thompson       case CEED_EVAL_GRAD:
82648acf710SJeremy L Thompson       case CEED_EVAL_DIV:
82748acf710SJeremy L Thompson       case CEED_EVAL_CURL:
82848acf710SJeremy L Thompson         CeedCallBackend(CeedOperatorFieldGetBasis(op_output_fields[i], &basis));
82948acf710SJeremy L Thompson         CeedCallBackend(
83048acf710SJeremy L Thompson             CeedBasisApplyAtPoints(basis, num_points, CEED_TRANSPOSE, eval_mode, point_coords_elem, impl->q_vecs_out[i], impl->e_vecs_out[i]));
83148acf710SJeremy L Thompson         break;
83248acf710SJeremy L Thompson       // LCOV_EXCL_START
83348acf710SJeremy L Thompson       case CEED_EVAL_WEIGHT: {
83448acf710SJeremy L Thompson         Ceed ceed;
835*bcbe1c99SJeremy L Thompson 
83648acf710SJeremy L Thompson         CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
83748acf710SJeremy L Thompson         return CeedError(ceed, CEED_ERROR_BACKEND, "CEED_EVAL_WEIGHT cannot be an output evaluation mode");
83848acf710SJeremy L Thompson         // LCOV_EXCL_STOP
83948acf710SJeremy L Thompson       }
84048acf710SJeremy L Thompson     }
84148acf710SJeremy L Thompson     // Restrict output block
84248acf710SJeremy L Thompson     // Get output vector
84348acf710SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetType(elem_rstr, &rstr_type));
84448acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_output_fields[i], &vec));
84548acf710SJeremy L Thompson     if (vec == CEED_VECTOR_ACTIVE) vec = out_vec;
84648acf710SJeremy L Thompson     // Restrict
84748acf710SJeremy L Thompson     if (rstr_type == CEED_RESTRICTION_POINTS) {
84848acf710SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionApplyAtPointsInElement(elem_rstr, e, CEED_TRANSPOSE, impl->e_vecs_out[i], vec, request));
84948acf710SJeremy L Thompson     } else {
85048acf710SJeremy L Thompson       CeedCallBackend(CeedElemRestrictionApplyBlock(elem_rstr, e, CEED_TRANSPOSE, impl->e_vecs_out[i], vec, request));
85148acf710SJeremy L Thompson     }
85248acf710SJeremy L Thompson   }
85348acf710SJeremy L Thompson   return CEED_ERROR_SUCCESS;
85448acf710SJeremy L Thompson }
85548acf710SJeremy L Thompson 
85648acf710SJeremy L Thompson //------------------------------------------------------------------------------
85748acf710SJeremy L Thompson // Restore Input Vectors
85848acf710SJeremy L Thompson //------------------------------------------------------------------------------
85948acf710SJeremy L Thompson static inline int CeedOperatorRestoreInputsAtPoints_Ref(CeedInt num_input_fields, CeedQFunctionField *qf_input_fields,
86048acf710SJeremy L Thompson                                                         CeedOperatorField *op_input_fields, CeedScalar *e_data[2 * CEED_FIELD_MAX],
86148acf710SJeremy L Thompson                                                         CeedOperator_Ref *impl) {
86248acf710SJeremy L Thompson   for (CeedInt i = 0; i < num_input_fields; i++) {
86348acf710SJeremy L Thompson     CeedEvalMode eval_mode;
86448acf710SJeremy L Thompson     CeedVector   vec;
86548acf710SJeremy L Thompson 
86648acf710SJeremy L Thompson     CeedCallBackend(CeedQFunctionFieldGetEvalMode(qf_input_fields[i], &eval_mode));
86748acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorFieldGetVector(op_input_fields[i], &vec));
86848acf710SJeremy L Thompson     if (eval_mode != CEED_EVAL_WEIGHT && vec != CEED_VECTOR_ACTIVE) {
86948acf710SJeremy L Thompson       CeedCallBackend(CeedVectorRestoreArrayRead(impl->e_vecs_full[i], (const CeedScalar **)&e_data[i]));
87048acf710SJeremy L Thompson     }
87148acf710SJeremy L Thompson   }
87248acf710SJeremy L Thompson   return CEED_ERROR_SUCCESS;
87348acf710SJeremy L Thompson }
87448acf710SJeremy L Thompson 
87548acf710SJeremy L Thompson //------------------------------------------------------------------------------
87648acf710SJeremy L Thompson // Operator Apply
87748acf710SJeremy L Thompson //------------------------------------------------------------------------------
87848acf710SJeremy L Thompson static int CeedOperatorApplyAddAtPoints_Ref(CeedOperator op, CeedVector in_vec, CeedVector out_vec, CeedRequest *request) {
87948acf710SJeremy L Thompson   Ceed                ceed;
88048acf710SJeremy L Thompson   CeedInt             num_points_offset          = 0, num_input_fields, num_output_fields, num_elem;
88148acf710SJeremy L Thompson   CeedScalar         *e_data[2 * CEED_FIELD_MAX] = {0};
88248acf710SJeremy L Thompson   CeedVector          point_coords               = NULL;
88348acf710SJeremy L Thompson   CeedElemRestriction rstr_points                = NULL;
88448acf710SJeremy L Thompson   CeedQFunctionField *qf_input_fields, *qf_output_fields;
88548acf710SJeremy L Thompson   CeedQFunction       qf;
88648acf710SJeremy L Thompson   CeedOperatorField  *op_input_fields, *op_output_fields;
88748acf710SJeremy L Thompson   CeedOperator_Ref   *impl;
88848acf710SJeremy L Thompson 
88948acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
89048acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
89148acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetNumElements(op, &num_elem));
89248acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetQFunction(op, &qf));
89348acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetFields(op, &num_input_fields, &op_input_fields, &num_output_fields, &op_output_fields));
89448acf710SJeremy L Thompson   CeedCallBackend(CeedQFunctionGetFields(qf, NULL, &qf_input_fields, NULL, &qf_output_fields));
89548acf710SJeremy L Thompson 
89648acf710SJeremy L Thompson   // Setup
89748acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupAtPoints_Ref(op));
89848acf710SJeremy L Thompson 
89948acf710SJeremy L Thompson   // Point coordinates
90048acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorAtPointsGetPoints(op, &rstr_points, &point_coords));
90148acf710SJeremy L Thompson 
90248acf710SJeremy L Thompson   // Input Evecs and Restriction
90348acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorSetupInputsAtPoints_Ref(num_input_fields, qf_input_fields, op_input_fields, in_vec, e_data, impl, request));
90448acf710SJeremy L Thompson 
90548acf710SJeremy L Thompson   // Loop through elements
90648acf710SJeremy L Thompson   for (CeedInt e = 0; e < num_elem; e++) {
90748acf710SJeremy L Thompson     CeedInt num_points;
90848acf710SJeremy L Thompson 
90948acf710SJeremy L Thompson     // Setup points for element
91048acf710SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionApplyAtPointsInElement(rstr_points, e, CEED_NOTRANSPOSE, point_coords, impl->point_coords_elem, request));
91148acf710SJeremy L Thompson     CeedCallBackend(CeedElemRestrictionGetNumPointsInElement(rstr_points, e, &num_points));
91248acf710SJeremy L Thompson 
91348acf710SJeremy L Thompson     // Input basis apply
91448acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorInputBasisAtPoints_Ref(e, num_points_offset, num_points, qf_input_fields, op_input_fields, num_input_fields, in_vec,
91548acf710SJeremy L Thompson                                                        impl->point_coords_elem, false, e_data, impl, request));
91648acf710SJeremy L Thompson 
91748acf710SJeremy L Thompson     // Q function
91848acf710SJeremy L Thompson     if (!impl->is_identity_qf) {
91948acf710SJeremy L Thompson       CeedCallBackend(CeedQFunctionApply(qf, num_points, impl->q_vecs_in, impl->q_vecs_out));
92048acf710SJeremy L Thompson     }
92148acf710SJeremy L Thompson 
92248acf710SJeremy L Thompson     // Output basis apply and restriction
92348acf710SJeremy L Thompson     CeedCallBackend(CeedOperatorOutputBasisAtPoints_Ref(e, num_points_offset, num_points, qf_output_fields, op_output_fields, num_input_fields,
92448acf710SJeremy L Thompson                                                         num_output_fields, op, out_vec, impl->point_coords_elem, impl, request));
92548acf710SJeremy L Thompson 
92648acf710SJeremy L Thompson     num_points_offset += num_points;
92748acf710SJeremy L Thompson   }
92848acf710SJeremy L Thompson 
92948acf710SJeremy L Thompson   // Restore input arrays
93048acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorRestoreInputsAtPoints_Ref(num_input_fields, qf_input_fields, op_input_fields, e_data, impl));
93148acf710SJeremy L Thompson 
93248acf710SJeremy L Thompson   // Cleanup point coordinates
93348acf710SJeremy L Thompson   CeedCallBackend(CeedVectorDestroy(&point_coords));
93448acf710SJeremy L Thompson   CeedCallBackend(CeedElemRestrictionDestroy(&rstr_points));
93548acf710SJeremy L Thompson   return CEED_ERROR_SUCCESS;
93648acf710SJeremy L Thompson }
93748acf710SJeremy L Thompson 
93848acf710SJeremy L Thompson //------------------------------------------------------------------------------
939f10650afSjeremylt // Operator Destroy
940f10650afSjeremylt //------------------------------------------------------------------------------
941f10650afSjeremylt static int CeedOperatorDestroy_Ref(CeedOperator op) {
942f10650afSjeremylt   CeedOperator_Ref *impl;
943f10650afSjeremylt 
944ad70ee2cSJeremy L Thompson   CeedCallBackend(CeedOperatorGetData(op, &impl));
9454fc1f125SJeremy L Thompson   for (CeedInt i = 0; i < impl->num_inputs + impl->num_outputs; i++) {
9462b730f8bSJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->e_vecs_full[i]));
947f10650afSjeremylt   }
9482b730f8bSJeremy L Thompson   CeedCallBackend(CeedFree(&impl->e_vecs_full));
9492b730f8bSJeremy L Thompson   CeedCallBackend(CeedFree(&impl->input_states));
950f10650afSjeremylt 
9514fc1f125SJeremy L Thompson   for (CeedInt i = 0; i < impl->num_inputs; i++) {
9522b730f8bSJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->e_vecs_in[i]));
9532b730f8bSJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->q_vecs_in[i]));
954f10650afSjeremylt   }
9552b730f8bSJeremy L Thompson   CeedCallBackend(CeedFree(&impl->e_vecs_in));
9562b730f8bSJeremy L Thompson   CeedCallBackend(CeedFree(&impl->q_vecs_in));
957f10650afSjeremylt 
9584fc1f125SJeremy L Thompson   for (CeedInt i = 0; i < impl->num_outputs; i++) {
9592b730f8bSJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->e_vecs_out[i]));
9602b730f8bSJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->q_vecs_out[i]));
961f10650afSjeremylt   }
9622b730f8bSJeremy L Thompson   CeedCallBackend(CeedFree(&impl->e_vecs_out));
9632b730f8bSJeremy L Thompson   CeedCallBackend(CeedFree(&impl->q_vecs_out));
96448acf710SJeremy L Thompson   CeedCallBackend(CeedVectorDestroy(&impl->point_coords_elem));
965f10650afSjeremylt 
966bb219a0fSJeremy L Thompson   // QFunction assembly
9674fc1f125SJeremy L Thompson   for (CeedInt i = 0; i < impl->num_active_in; i++) {
9682b730f8bSJeremy L Thompson     CeedCallBackend(CeedVectorDestroy(&impl->qf_active_in[i]));
969bb219a0fSJeremy L Thompson   }
9702b730f8bSJeremy L Thompson   CeedCallBackend(CeedFree(&impl->qf_active_in));
971bb219a0fSJeremy L Thompson 
9722b730f8bSJeremy L Thompson   CeedCallBackend(CeedFree(&impl));
973e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
974f10650afSjeremylt }
975f10650afSjeremylt 
976f10650afSjeremylt //------------------------------------------------------------------------------
977713f43c3Sjeremylt // Operator Create
978f10650afSjeremylt //------------------------------------------------------------------------------
97921617c04Sjeremylt int CeedOperatorCreate_Ref(CeedOperator op) {
980fe2413ffSjeremylt   Ceed              ceed;
9814ce2993fSjeremylt   CeedOperator_Ref *impl;
98221617c04Sjeremylt 
983ad70ee2cSJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
9842b730f8bSJeremy L Thompson   CeedCallBackend(CeedCalloc(1, &impl));
9852b730f8bSJeremy L Thompson   CeedCallBackend(CeedOperatorSetData(op, impl));
9862b730f8bSJeremy L Thompson   CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "LinearAssembleQFunction", CeedOperatorLinearAssembleQFunction_Ref));
9872b730f8bSJeremy L Thompson   CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "LinearAssembleQFunctionUpdate", CeedOperatorLinearAssembleQFunctionUpdate_Ref));
9882b730f8bSJeremy L Thompson   CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "ApplyAdd", CeedOperatorApplyAdd_Ref));
9892b730f8bSJeremy L Thompson   CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "Destroy", CeedOperatorDestroy_Ref));
990e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
99121617c04Sjeremylt }
9922a86cc9dSSebastian Grimberg 
9932a86cc9dSSebastian Grimberg //------------------------------------------------------------------------------
99448acf710SJeremy L Thompson // Operator Create At Points
99548acf710SJeremy L Thompson //------------------------------------------------------------------------------
99648acf710SJeremy L Thompson int CeedOperatorCreateAtPoints_Ref(CeedOperator op) {
99748acf710SJeremy L Thompson   Ceed              ceed;
99848acf710SJeremy L Thompson   CeedOperator_Ref *impl;
99948acf710SJeremy L Thompson 
100048acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorGetCeed(op, &ceed));
100148acf710SJeremy L Thompson   CeedCallBackend(CeedCalloc(1, &impl));
100248acf710SJeremy L Thompson   CeedCallBackend(CeedOperatorSetData(op, impl));
100348acf710SJeremy L Thompson   CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "ApplyAdd", CeedOperatorApplyAddAtPoints_Ref));
100448acf710SJeremy L Thompson   CeedCallBackend(CeedSetBackendFunction(ceed, "Operator", op, "Destroy", CeedOperatorDestroy_Ref));
100548acf710SJeremy L Thompson   return CEED_ERROR_SUCCESS;
100648acf710SJeremy L Thompson }
100748acf710SJeremy L Thompson 
100848acf710SJeremy L Thompson //------------------------------------------------------------------------------
1009