/// @file
/// Test creation, evaluation, and destruction for QFunction
/// \test Test creation, evaluation, and destruction for QFunction
#include "t401-qfunction.h"

#include <ceed.h>
#include <math.h>

int main(int argc, char **argv) {
  Ceed                 ceed;
  CeedVector           in[16], out[16];
  CeedVector           Q_data, W, U, V;
  CeedQFunction        qf_setup, qf_mass;
  CeedQFunctionContext ctx;
  CeedInt              Q = 8;
  const CeedScalar    *vv;
  CeedScalar           w[Q], u[Q], v[Q], ctx_data[5] = {1, 2, 3, 4, 5};

  CeedInit(argv[1], &ceed);

  CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_setup);
  CeedQFunctionAddInput(qf_setup, "w", 1, CEED_EVAL_WEIGHT);
  CeedQFunctionAddOutput(qf_setup, "qdata", 1, CEED_EVAL_NONE);

  CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_mass);
  CeedQFunctionAddInput(qf_mass, "qdata", 1, CEED_EVAL_NONE);
  CeedQFunctionAddInput(qf_mass, "u", 1, CEED_EVAL_INTERP);
  CeedQFunctionAddOutput(qf_mass, "v", 1, CEED_EVAL_INTERP);

  CeedQFunctionContextCreate(ceed, &ctx);
  CeedQFunctionContextSetData(ctx, CEED_MEM_HOST, CEED_USE_POINTER, sizeof(ctx_data), &ctx_data);
  CeedQFunctionSetContext(qf_mass, ctx);

  for (CeedInt i = 0; i < Q; i++) {
    CeedScalar x = 2. * i / (Q - 1) - 1;
    w[i]         = 1 - x * x;
    u[i]         = 2 + 3 * x + 5 * x * x;
    v[i]         = w[i] * u[i];
  }

  CeedVectorCreate(ceed, Q, &W);
  CeedVectorSetArray(W, CEED_MEM_HOST, CEED_USE_POINTER, w);
  CeedVectorCreate(ceed, Q, &U);
  CeedVectorSetArray(U, CEED_MEM_HOST, CEED_USE_POINTER, u);
  CeedVectorCreate(ceed, Q, &V);
  CeedVectorSetValue(V, 0);
  CeedVectorCreate(ceed, Q, &Q_data);
  CeedVectorSetValue(Q_data, 0);

  {
    in[0]  = W;
    out[0] = Q_data;
    CeedQFunctionApply(qf_setup, Q, in, out);
  }
  {
    in[0]  = W;
    in[1]  = U;
    out[0] = V;
    CeedQFunctionApply(qf_mass, Q, in, out);
  }

  CeedVectorGetArrayRead(V, CEED_MEM_HOST, &vv);
  for (CeedInt i = 0; i < Q; i++) {
    if (fabs(ctx_data[4] * v[i] - vv[i]) > 100. * CEED_EPSILON) printf("[%" CeedInt_FMT "] v %f != vv %f\n", i, v[i], vv[i]);
  }
  CeedVectorRestoreArrayRead(V, &vv);

  CeedVectorDestroy(&W);
  CeedVectorDestroy(&U);
  CeedVectorDestroy(&V);
  CeedVectorDestroy(&Q_data);
  CeedQFunctionDestroy(&qf_setup);
  CeedQFunctionDestroy(&qf_mass);
  CeedQFunctionContextDestroy(&ctx);
  CeedDestroy(&ceed);
  return 0;
}