xref: /libCEED/tests/t101-vector.c (revision 0436c2ad4dd95c9ccd89f5346568014600de10a7)
1*0436c2adSjeremylt /// @file
2*0436c2adSjeremylt /// Test CeedVectorSetValue
3*0436c2adSjeremylt /// \test Test CeedVectorSetValue
4*0436c2adSjeremylt #include <ceed.h>
5*0436c2adSjeremylt 
6*0436c2adSjeremylt static int CheckValues(Ceed ceed, CeedVector x, CeedScalar value) {
7*0436c2adSjeremylt   const CeedScalar *b;
8*0436c2adSjeremylt   CeedInt n;
9*0436c2adSjeremylt   CeedVectorGetLength(x, &n);
10*0436c2adSjeremylt   CeedVectorGetArrayRead(x, CEED_MEM_HOST, &b);
11*0436c2adSjeremylt   for (CeedInt i=0; i<n; i++) {
12*0436c2adSjeremylt     if (b[i] != value)
13*0436c2adSjeremylt       // LCOV_EXCL_START
14*0436c2adSjeremylt       printf("Error reading array b[%d] = %f",i,
15*0436c2adSjeremylt              (double)b[i]);
16*0436c2adSjeremylt     // LCOV_EXCL_STOP
17*0436c2adSjeremylt   }
18*0436c2adSjeremylt   CeedVectorRestoreArrayRead(x, &b);
19*0436c2adSjeremylt   return 0;
20*0436c2adSjeremylt }
21*0436c2adSjeremylt 
22*0436c2adSjeremylt int main(int argc, char **argv) {
23*0436c2adSjeremylt   Ceed ceed;
24*0436c2adSjeremylt   CeedVector x;
25*0436c2adSjeremylt   CeedInt n;
26*0436c2adSjeremylt   CeedScalar a[10];
27*0436c2adSjeremylt   const CeedScalar *b;
28*0436c2adSjeremylt 
29*0436c2adSjeremylt   CeedInit(argv[1], &ceed);
30*0436c2adSjeremylt   n = 10;
31*0436c2adSjeremylt   CeedVectorCreate(ceed, n, &x);
32*0436c2adSjeremylt   for (CeedInt i=0; i<n; i++)
33*0436c2adSjeremylt     a[i] = 10 + i;
34*0436c2adSjeremylt   CeedVectorSetArray(x, CEED_MEM_HOST, CEED_USE_POINTER, a);
35*0436c2adSjeremylt 
36*0436c2adSjeremylt   CeedVectorGetArrayRead(x, CEED_MEM_HOST, &b);
37*0436c2adSjeremylt   for (CeedInt i=0; i<n; i++)
38*0436c2adSjeremylt     if (b[i] != 10+i)
39*0436c2adSjeremylt       // LCOV_EXCL_START
40*0436c2adSjeremylt       printf("Error reading array b[%d] = %f",i,
41*0436c2adSjeremylt              (double)b[i]);
42*0436c2adSjeremylt   // LCOV_EXCL_STOP
43*0436c2adSjeremylt   CeedVectorRestoreArrayRead(x, &b);
44*0436c2adSjeremylt 
45*0436c2adSjeremylt   CeedVectorSetValue(x, 3.0);
46*0436c2adSjeremylt   CheckValues(ceed, x, 3.0);
47*0436c2adSjeremylt   CeedVectorDestroy(&x);
48*0436c2adSjeremylt 
49*0436c2adSjeremylt   CeedVectorCreate(ceed, n, &x);
50*0436c2adSjeremylt   // Set value before setting or getting the array
51*0436c2adSjeremylt   CeedVectorSetValue(x, 5.0);
52*0436c2adSjeremylt   CheckValues(ceed, x, 5.0);
53*0436c2adSjeremylt   CeedVectorDestroy(&x);
54*0436c2adSjeremylt 
55*0436c2adSjeremylt   CeedDestroy(&ceed);
56*0436c2adSjeremylt   return 0;
57*0436c2adSjeremylt }
58