xref: /libCEED/tests/t412-qfunction.c (revision 49aac155e7a09736f56fb3abac0f57dab29f7cbf)

/// @file
/// Test creation, evaluation, and destruction of identity QFunction with size>1
/// \test Test creation, evaluation, and destruction of identity QFunction with size>1
#include <ceed.h>
#include <math.h>
#include <stdio.h>

int main(int argc, char **argv) {
  Ceed          ceed;
  CeedVector    in[16], out[16];
  CeedVector    u, v;
  CeedQFunction qf;
  CeedInt       q = 8, size = 3;
  CeedScalar    u_array[q * size];

  CeedInit(argv[1], &ceed);

  CeedVectorCreate(ceed, q * size, &u);
  for (CeedInt i = 0; i < q * size; i++) u_array[i] = i * i;
  CeedVectorSetArray(u, CEED_MEM_HOST, CEED_USE_POINTER, u_array);
  CeedVectorCreate(ceed, q * size, &v);
  CeedVectorSetValue(v, 0);

  CeedQFunctionCreateIdentity(ceed, size, CEED_EVAL_INTERP, CEED_EVAL_NONE, &qf);
  {
    in[0]  = u;
    out[0] = v;
    CeedQFunctionApply(qf, q, in, out);
  }

  // Verify results
  {
    const CeedScalar *v_array;

    CeedVectorGetArrayRead(v, CEED_MEM_HOST, &v_array);
    for (CeedInt i = 0; i < q * size; i++) {
      if (fabs(v_array[i] - u_array[i]) > 1e-12) printf("[%" CeedInt_FMT "] v %f != u %f\n", i, v_array[i], u_array[i]);
    }
    CeedVectorRestoreArrayRead(v, &v_array);
  }

  CeedVectorDestroy(&u);
  CeedVectorDestroy(&v);
  CeedQFunctionDestroy(&qf);
  CeedDestroy(&ceed);
  return 0;
}