xref: /libCEED/tests/t104-vector.c (revision 49aac155e7a09736f56fb3abac0f57dab29f7cbf)
10436c2adSjeremylt /// @file
20436c2adSjeremylt /// Test CeedVectorGetArray to modify array
30436c2adSjeremylt /// \test Test CeedVectorGetArray to modify array
40436c2adSjeremylt #include <ceed.h>
5*49aac155SJeremy L Thompson #include <stdio.h>
60436c2adSjeremylt 
main(int argc,char ** argv)70436c2adSjeremylt int main(int argc, char **argv) {
80436c2adSjeremylt   Ceed          ceed;
90436c2adSjeremylt   CeedVector    x;
104fee36f0SJeremy L Thompson   const CeedInt len = 10;
114fee36f0SJeremy L Thompson   CeedScalar    array[len];
120436c2adSjeremylt 
130436c2adSjeremylt   CeedInit(argv[1], &ceed);
140436c2adSjeremylt 
154fee36f0SJeremy L Thompson   CeedVectorCreate(ceed, len, &x);
164fee36f0SJeremy L Thompson   for (CeedInt i = 0; i < len; i++) array[i] = 0;
174fee36f0SJeremy L Thompson   CeedVectorSetArray(x, CEED_MEM_HOST, CEED_USE_POINTER, array);
180436c2adSjeremylt 
194fee36f0SJeremy L Thompson   {
204fee36f0SJeremy L Thompson     CeedScalar *writable_array;
210436c2adSjeremylt 
224fee36f0SJeremy L Thompson     CeedVectorGetArray(x, CEED_MEM_HOST, &writable_array);
234fee36f0SJeremy L Thompson     writable_array[3] = -3.14;
244fee36f0SJeremy L Thompson     CeedVectorRestoreArray(x, &writable_array);
254fee36f0SJeremy L Thompson   }
264fee36f0SJeremy L Thompson 
274fee36f0SJeremy L Thompson   if (array[3] != (CeedScalar)(-3.14)) printf("Error writing array[3] = %f\n", (CeedScalar)array[3]);
280436c2adSjeremylt 
290436c2adSjeremylt   CeedVectorDestroy(&x);
300436c2adSjeremylt   CeedDestroy(&ceed);
310436c2adSjeremylt   return 0;
320436c2adSjeremylt }
33