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 894b7b29bSJeremy L Thompson #ifndef CEED_REF_H 994b7b29bSJeremy L Thompson #define CEED_REF_H 103d576824SJeremy L Thompson 1149aac155SJeremy L Thompson #include <ceed.h> 12ec3da8bcSJed Brown #include <ceed/backend.h> 133d576824SJeremy L Thompson #include <stdbool.h> 143d576824SJeremy L Thompson #include <stdint.h> 1521617c04Sjeremylt 1621617c04Sjeremylt typedef struct { 178c1105f8SJeremy L Thompson CeedScalar *collo_grad_1d; 188c1105f8SJeremy L Thompson bool has_collo_interp; 1984a01de5SJeremy L Thompson } CeedBasis_Ref; 2084a01de5SJeremy L Thompson 2184a01de5SJeremy L Thompson typedef struct { 2221617c04Sjeremylt CeedScalar *array; 239c774eddSJeremy L Thompson CeedScalar *array_borrowed; 249c774eddSJeremy L Thompson CeedScalar *array_owned; 2521617c04Sjeremylt } CeedVector_Ref; 2621617c04Sjeremylt 2721617c04Sjeremylt typedef struct { 28d979a051Sjeremylt const CeedInt *offsets; 29d979a051Sjeremylt CeedInt *offsets_allocated; 3077d1c127SSebastian Grimberg const bool *orients; /* Orientation, if it exists, is true when the dof must be flipped */ 3177d1c127SSebastian Grimberg bool *orients_allocated; 320c73c039SSebastian Grimberg const CeedInt8 *curl_orients; /* Tridiagonal matrix (row-major) for a general transformation during restriction */ 330c73c039SSebastian Grimberg CeedInt8 *curl_orients_allocated; 347c1dbaffSSebastian Grimberg int (*Apply)(CeedElemRestriction, CeedInt, CeedInt, CeedInt, CeedInt, CeedInt, CeedTransposeMode, bool, bool, CeedVector, CeedVector, 357c1dbaffSSebastian Grimberg CeedRequest *); 3621617c04Sjeremylt } CeedElemRestriction_Ref; 3721617c04Sjeremylt 3821617c04Sjeremylt typedef struct { 39aedaa0e5Sjeremylt const CeedScalar **inputs; 40aedaa0e5Sjeremylt CeedScalar **outputs; 41aedaa0e5Sjeremylt } CeedQFunction_Ref; 42aedaa0e5Sjeremylt 43aedaa0e5Sjeremylt typedef struct { 44777ff853SJeremy L Thompson void *data; 459c774eddSJeremy L Thompson void *data_borrowed; 469c774eddSJeremy L Thompson void *data_owned; 47777ff853SJeremy L Thompson } CeedQFunctionContext_Ref; 48777ff853SJeremy L Thompson 49777ff853SJeremy L Thompson typedef struct { 50edb2538eSJeremy L Thompson bool is_identity_qf, is_identity_rstr_op; 514fc1f125SJeremy L Thompson CeedVector *e_vecs_full; /* Full E-vectors, inputs followed by outputs */ 524fc1f125SJeremy L Thompson uint64_t *input_states; /* State counter of inputs */ 534fc1f125SJeremy L Thompson CeedVector *e_vecs_in; /* Single element input E-vectors */ 544fc1f125SJeremy L Thompson CeedVector *e_vecs_out; /* Single element output E-vectors */ 554fc1f125SJeremy L Thompson CeedVector *q_vecs_in; /* Single element input Q-vectors */ 564fc1f125SJeremy L Thompson CeedVector *q_vecs_out; /* Single element output Q-vectors */ 574fc1f125SJeremy L Thompson CeedInt num_inputs, num_outputs; 584fc1f125SJeremy L Thompson CeedInt num_active_in, num_active_out; 59bb219a0fSJeremy L Thompson CeedVector *qf_active_in; 6021617c04Sjeremylt } CeedOperator_Ref; 6121617c04Sjeremylt 621f9221feSJeremy L Thompson CEED_INTERN int CeedVectorCreate_Ref(CeedSize n, CeedVector vec); 6321617c04Sjeremylt 64fcbe8c06SSebastian Grimberg CEED_INTERN int CeedElemRestrictionCreate_Ref(CeedMemType mem_type, CeedCopyMode copy_mode, const CeedInt *offsets, const bool *orients, 650c73c039SSebastian Grimberg const CeedInt8 *curl_orients, CeedElemRestriction r); 6621617c04Sjeremylt 672b730f8bSJeremy L Thompson CEED_INTERN int CeedBasisCreateTensorH1_Ref(CeedInt dim, CeedInt P_1d, CeedInt Q_1d, const CeedScalar *interp_1d, const CeedScalar *grad_1d, 68d1d35e2fSjeremylt const CeedScalar *q_ref_1d, const CeedScalar *q_weight_1d, CeedBasis basis); 692b730f8bSJeremy L Thompson CEED_INTERN int CeedBasisCreateH1_Ref(CeedElemTopology topo, CeedInt dim, CeedInt num_nodes, CeedInt num_qpts, const CeedScalar *interp, 702b730f8bSJeremy L Thompson const CeedScalar *grad, const CeedScalar *q_ref, const CeedScalar *q_weight, CeedBasis basis); 712b730f8bSJeremy L Thompson CEED_INTERN int CeedBasisCreateHdiv_Ref(CeedElemTopology topo, CeedInt dim, CeedInt num_nodes, CeedInt num_qpts, const CeedScalar *interp, 722b730f8bSJeremy L Thompson const CeedScalar *div, const CeedScalar *q_ref, const CeedScalar *q_weight, CeedBasis basis); 73c4e3f59bSSebastian Grimberg CEED_INTERN int CeedBasisCreateHcurl_Ref(CeedElemTopology topo, CeedInt dim, CeedInt num_nodes, CeedInt num_qpts, const CeedScalar *interp, 74c4e3f59bSSebastian Grimberg const CeedScalar *curl, const CeedScalar *q_ref, const CeedScalar *q_weight, CeedBasis basis); 7521617c04Sjeremylt 76*a71faab1SSebastian Grimberg CEED_INTERN int CeedTensorContractCreate_Ref(CeedTensorContract contract); 772f86a920SJeremy L Thompson 7821617c04Sjeremylt CEED_INTERN int CeedQFunctionCreate_Ref(CeedQFunction qf); 7921617c04Sjeremylt 80777ff853SJeremy L Thompson CEED_INTERN int CeedQFunctionContextCreate_Ref(CeedQFunctionContext ctx); 81777ff853SJeremy L Thompson 8221617c04Sjeremylt CEED_INTERN int CeedOperatorCreate_Ref(CeedOperator op); 83c04a41a7SJeremy L Thompson 8494b7b29bSJeremy L Thompson #endif // CEED_REF_H 85