xref: /libCEED/tests/t201-elemrestriction-f.f90 (revision a8d322087fa8f150327cdc2bf14a171452b711ec)
18980d4a7Sjeremylt!-----------------------------------------------------------------------
28980d4a7Sjeremylt      program test
38980d4a7Sjeremylt
48980d4a7Sjeremylt      include 'ceedf.h'
58980d4a7Sjeremylt
68980d4a7Sjeremylt      integer ceed,err
78980d4a7Sjeremylt      integer x,y
88980d4a7Sjeremylt      integer r
98980d4a7Sjeremylt
108980d4a7Sjeremylt      integer ne
118980d4a7Sjeremylt      parameter(ne=3)
12*a8d32208Sjeremylt      integer lmode
13*a8d32208Sjeremylt      parameter(lmode=ceed_notranspose)
148980d4a7Sjeremylt
158980d4a7Sjeremylt      real*8 a(2*ne)
168980d4a7Sjeremylt      real*8 yy(2*ne)
178980d4a7Sjeremylt      real*8 diff
18c8b9fe72Sjeremylt      integer*8 aoffset,yoffset
198980d4a7Sjeremylt
208980d4a7Sjeremylt      character arg*32
218980d4a7Sjeremylt
228980d4a7Sjeremylt      call getarg(1,arg)
238980d4a7Sjeremylt      call ceedinit(trim(arg)//char(0),ceed,err)
248980d4a7Sjeremylt
258980d4a7Sjeremylt      call ceedvectorcreate(ceed,2*ne,x,err)
268980d4a7Sjeremylt
278980d4a7Sjeremylt      do i=1,2*ne
288980d4a7Sjeremylt        a(i)=10+i-1
298980d4a7Sjeremylt      enddo
308980d4a7Sjeremylt
31c8b9fe72Sjeremylt      aoffset=0
32c8b9fe72Sjeremylt      call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,a,aoffset,err)
338980d4a7Sjeremylt
34*a8d32208Sjeremylt      call ceedelemrestrictioncreateidentity(ceed,lmode,ne,2,2*ne,1,r,err)
358980d4a7Sjeremylt
368980d4a7Sjeremylt      call ceedvectorcreate(ceed,2*ne,y,err);
378980d4a7Sjeremylt      call ceedvectorsetvalue(y,0.d0,err);
38*a8d32208Sjeremylt      call ceedelemrestrictionapply(r,ceed_notranspose,x,y,&
398980d4a7Sjeremylt     & ceed_request_immediate,err)
408980d4a7Sjeremylt
418980d4a7Sjeremylt      call ceedvectorgetarrayread(y,ceed_mem_host,yy,yoffset,err)
428980d4a7Sjeremylt      do i=1,ne*2
438980d4a7Sjeremylt        diff=10+i-1-yy(yoffset+i)
448980d4a7Sjeremylt        if (abs(diff) > 1.0D-15) then
45a2546046Sjeremylt! LCOV_EXCL_START
468980d4a7Sjeremylt          write(*,*) 'Error in restricted array y(',i,')=',yy(yoffset+i)
47de996c55Sjeremylt! LCOV_EXCL_STOP
488980d4a7Sjeremylt        endif
498980d4a7Sjeremylt      enddo
508980d4a7Sjeremylt      call ceedvectorrestorearrayread(y,yy,yoffset,err)
518980d4a7Sjeremylt
528980d4a7Sjeremylt      call ceedvectordestroy(x,err)
538980d4a7Sjeremylt      call ceedvectordestroy(y,err)
548980d4a7Sjeremylt      call ceedelemrestrictiondestroy(r,err)
558980d4a7Sjeremylt      call ceeddestroy(ceed,err)
568980d4a7Sjeremylt
578980d4a7Sjeremylt      end
588980d4a7Sjeremylt!-----------------------------------------------------------------------
59