/// @file
/// Test CeedVectorGetArrayWrite to modify array
/// \test Test CeedVectorGetArrayWrite to modify array
#include <ceed.h>

int main(int argc, char **argv) {
  Ceed          ceed;
  CeedVector    x;
  const CeedInt n = 10;
  CeedScalar   *a;

  CeedInit(argv[1], &ceed);

  CeedVectorCreate(ceed, n, &x);

  CeedVectorGetArrayWrite(x, CEED_MEM_HOST, &a);
  for (CeedInt i = 0; i < n; i++) a[i] = 3 * i;
  CeedVectorRestoreArray(x, &a);

  CeedVectorGetArrayRead(x, CEED_MEM_HOST, (const CeedScalar **)&a);
  for (CeedInt i = 0; i < n; i++) {
    if (a[i] != (CeedScalar)(3 * i)) printf("Error writing array a[%" CeedInt_FMT "] = %f\n", i, a[i]);
  }
  CeedVectorRestoreArrayRead(x, (const CeedScalar **)&a);

  CeedVectorDestroy(&x);
  CeedDestroy(&ceed);
  return 0;
}