/// @file
/// Test setting QFunctionContext fields from Operator
/// \test Test setting QFunctionContext fields from Operator
#include <ceed.h>
#include <stddef.h>
#include "t500-operator.h"

typedef struct {
  int count;
  double other;
} TestContext1;

typedef struct {
  double time;
  double other;
} TestContext2;

int main(int argc, char **argv) {
  Ceed ceed;
  CeedQFunctionContext qf_ctx_sub_1, qf_ctx_sub_2;
  CeedQFunction qf_sub_1, qf_sub_2;
  CeedOperator op_sub_1, op_sub_2, op_composite;
  TestContext1 ctx_data_1 = {
    .count = 42,
    .other = -3.0,
  };
  TestContext2 ctx_data_2 = {
    .time = 1.0,
    .other = -3.0,
  };

  CeedInit(argv[1], &ceed);

  // First sub-operator
  CeedQFunctionContextCreate(ceed, &qf_ctx_sub_1);
  CeedQFunctionContextSetData(qf_ctx_sub_1, CEED_MEM_HOST, CEED_USE_POINTER,
                              sizeof(TestContext1), &ctx_data_1);
  CeedQFunctionContextRegisterInt32(qf_ctx_sub_1, "count", offsetof(TestContext1,
                                    count), "some sort of counter");
  CeedQFunctionContextRegisterDouble(qf_ctx_sub_1, "other", offsetof(TestContext1,
                                     other), "some other value");

  CeedQFunctionCreateInterior(ceed, 1, setup, setup_loc, &qf_sub_1);
  CeedQFunctionSetContext(qf_sub_1, qf_ctx_sub_1);

  CeedOperatorCreate(ceed, qf_sub_1, CEED_QFUNCTION_NONE, CEED_QFUNCTION_NONE,
                     &op_sub_1);

  // Check setting field in operator
  CeedOperatorContextSetInt32(op_sub_1, "count", 43);
  if (ctx_data_1.count != 43)
    // LCOV_EXCL_START
    printf("Incorrect context data for count: %d != 43", ctx_data_1.count);
  // LCOV_EXCL_STOP

  // Second sub-operator
  CeedQFunctionContextCreate(ceed, &qf_ctx_sub_2);
  CeedQFunctionContextSetData(qf_ctx_sub_2, CEED_MEM_HOST, CEED_USE_POINTER,
                              sizeof(TestContext2), &ctx_data_2);
  CeedQFunctionContextRegisterDouble(qf_ctx_sub_2, "time", offsetof(TestContext2,
                                     time), "current time");
  CeedQFunctionContextRegisterDouble(qf_ctx_sub_2, "other", offsetof(TestContext2,
                                     other), "some other value");

  CeedQFunctionCreateInterior(ceed, 1, mass, mass_loc, &qf_sub_2);
  CeedQFunctionSetContext(qf_sub_2, qf_ctx_sub_2);

  CeedOperatorCreate(ceed, qf_sub_2, CEED_QFUNCTION_NONE, CEED_QFUNCTION_NONE,
                     &op_sub_2);

  // Composite operator
  CeedCompositeOperatorCreate(ceed, &op_composite);
  CeedCompositeOperatorAddSub(op_composite, op_sub_1);
  CeedCompositeOperatorAddSub(op_composite, op_sub_2);

  // Check setting field in context of single sub-operator for composite operator
  CeedOperatorContextSetDouble(op_composite, "time", 2.0);
  if (ctx_data_2.time != 2.0)
    // LCOV_EXCL_START
    printf("Incorrect context data for time: %f != 2.0", ctx_data_2.time);
  // LCOV_EXCL_STOP

  // Check setting field in context of multiple sub-operators for composite operator
  CeedOperatorContextSetDouble(op_composite, "other", 9000.);
  if (ctx_data_1.other != 9000.0)
    // LCOV_EXCL_START
    printf("Incorrect context data for other: %f != 2.0", ctx_data_1.other);
  // LCOV_EXCL_STOP
  if (ctx_data_2.other != 9000.0)
    // LCOV_EXCL_START
    printf("Incorrect context data for other: %f != 2.0", ctx_data_2.other);
  // LCOV_EXCL_STOP

  CeedQFunctionContextDestroy(&qf_ctx_sub_1);
  CeedQFunctionContextDestroy(&qf_ctx_sub_2);
  CeedQFunctionDestroy(&qf_sub_1);
  CeedQFunctionDestroy(&qf_sub_2);
  CeedOperatorDestroy(&op_sub_1);
  CeedOperatorDestroy(&op_sub_2);
  CeedOperatorDestroy(&op_composite);
  CeedDestroy(&ceed);
  return 0;
}