wip - header for basis eval at points

CEED · Feb 21, 2023 · 6716e67 · 6716e67
1 parent 05b44d1
commit 6716e67
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/include/ceed-impl.h b/include/ceed-impl.h
@@ -169,6 +169,7 @@ struct CeedElemRestriction_private {
 struct CeedBasis_private {
   Ceed ceed;
   int (*Apply)(CeedBasis, CeedInt, CeedTransposeMode, CeedEvalMode, CeedVector, CeedVector);
+  int (*ApplyAtPoints)(CeedBasis, CeedInt, CeedTransposeMode, CeedEvalMode, CeedVector, CeedVector, CeedVector);
   int (*Destroy)(CeedBasis);
   int              ref_count;
   bool             tensor_basis; /* flag for tensor basis */

diff --git a/include/ceed/ceed.h b/include/ceed/ceed.h
@@ -366,6 +366,8 @@ CEED_EXTERN int CeedBasisCreateProjection(CeedBasis basis_from, CeedBasis basis_
 CEED_EXTERN int CeedBasisReferenceCopy(CeedBasis basis, CeedBasis *basis_copy);
 CEED_EXTERN int CeedBasisView(CeedBasis basis, FILE *stream);
 CEED_EXTERN int CeedBasisApply(CeedBasis basis, CeedInt num_elem, CeedTransposeMode t_mode, CeedEvalMode eval_mode, CeedVector u, CeedVector v);
+CEED_EXTERN int CeedBasisApplyAtPoints(CeedBasis basis, CeedInt num_points, CeedTransposeMode t_mode, CeedEvalMode eval_mode, CeedVector x_ref,
+                                       CeedVector u, CeedVector v);
 CEED_EXTERN int CeedBasisGetCeed(CeedBasis basis, Ceed *ceed);
 CEED_EXTERN int CeedBasisGetDimension(CeedBasis basis, CeedInt *dim);
 CEED_EXTERN int CeedBasisGetTopology(CeedBasis basis, CeedElemTopology *topo);

diff --git a/interface/ceed-basis.c b/interface/ceed-basis.c
@@ -1085,6 +1085,29 @@ int CeedBasisApply(CeedBasis basis, CeedInt num_elem, CeedTransposeMode t_mode,
   return CEED_ERROR_SUCCESS;
 }
 
+/**
+  @brief Apply basis evaluation from nodes to arbitrary points
+
+  @param[in]  basis      CeedBasis to evaluate
+  @param[in]  num_points The number of points to apply the basis evaluation to;
+                           the backend will specify the ordering in CeedElemRestrictionCreateBlocked()
+  @param[in]  t_mode    \ref CEED_NOTRANSPOSE to evaluate from nodes to points;
+                          \ref CEED_TRANSPOSE to apply the transpose, mapping from points to nodes
+  @param[in]  eval_mode \ref CEED_EVAL_INTERP to use interpolated values,
+                          \ref CEED_EVAL_GRAD to use gradients
+  @param[in]  x_ref    CeedVector holding reference coordinates of each point
+  @param[in]  u        Input CeedVector, of length `num_points * num
+  @param[out] v        Output CeedVector, of length `num_points * num_comp` for `CEED_NOTRANSPOSE` with `CEED_EVAL_INTERP`
+
+  @return An error code: 0 - success, otherwise - failure
+
+  @ref User
+**/
+int CeedBasisApplyAtPoints(CeedBasis basis, CeedInt num_points, CeedTransposeMode t_mode, CeedEvalMode eval_mode, CeedVector x_ref, CeedVector u,
+                           CeedVector v) {
+  return CEED_ERROR_SUCCESS;
+}
+
 /**
   @brief Get Ceed associated with a CeedBasis