xref: /libCEED/tests/t401-qfunction-f.f90 (revision 5f67fade47e323fa44018f277580acfe24400ad4)
1!-----------------------------------------------------------------------
2      subroutine setup(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,&
3&           u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr)
4      real*8 ctx(1)
5      real*8 u1(8)
6      real*8 v1(8)
7      integer q,ierr
8
9      do i=1,q
10        v1(i)=u1(i)
11      enddo
12
13      ierr=0
14      end
15!-----------------------------------------------------------------------
16      subroutine mass(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,&
17&           u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr)
18      real*8 ctx(5)
19      real*8 u1(8)
20      real*8 u2(8)
21      real*8 v1(8)
22      integer q,ierr
23
24      do i=1,q
25        v1(i)=ctx(5)*u1(i)*u2(i)
26      enddo
27
28      ierr=0
29      end
30!-----------------------------------------------------------------------
31      program test
32      implicit none
33      include 'ceedf.h'
34
35      integer ceed,err
36      integer qdata,w,u,v
37      integer qf_setup,qf_mass
38      integer q,i
39      parameter(q=8)
40      real*8 ww(q)
41      real*8 uu(q)
42      real*8 vv(q)
43      real*8 vvv(q)
44      integer ctxsize
45      parameter(ctxsize=5)
46      real*8 ctx(5)
47      real*8 x
48      character arg*32
49      integer*8 uoffset,voffset,woffset
50
51      external setup,mass
52
53      ctx=(/1.d0,2.d0,3.d0,4.d0,5.d0/)
54
55      call getarg(1,arg)
56      call ceedinit(trim(arg)//char(0),ceed,err)
57
58      call ceedqfunctioncreateinterior(ceed,1,setup,&
59     &SOURCE_DIR&
60     &//'t401-qfunction.h:setup'//char(0),qf_setup,err)
61      call ceedqfunctionaddinput(qf_setup,'w', 1,ceed_eval_weight,err)
62      call ceedqfunctionaddoutput(qf_setup,'qdata',1,ceed_eval_none,err)
63
64      call ceedqfunctioncreateinterior(ceed,1,mass,&
65     &SOURCE_DIR&
66     &//'t401-qfunction.h:mass'//char(0),qf_mass,err)
67      call ceedqfunctionaddinput(qf_mass,'qdata',1,ceed_eval_none,err)
68      call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err)
69      call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err)
70
71      call ceedqfunctionsetcontext(qf_mass,ctx,ctxsize,err)
72
73      do i=0,q-1
74        x=2.0*i/(q-1)-1
75        ww(i+1)=1-x*x
76        uu(i+1)=2+3*x+5*x*x
77        vvv(i+1)=ww(i+1)*uu(i+1)
78      enddo
79
80      call ceedvectorcreate(ceed,q,w,err)
81      woffset=0
82      call ceedvectorsetarray(w,ceed_mem_host,ceed_use_pointer,ww,woffset,err)
83      call ceedvectorcreate(ceed,q,u,err)
84      uoffset=0
85      call ceedvectorsetarray(u,ceed_mem_host,ceed_use_pointer,uu,uoffset,err)
86      call ceedvectorcreate(ceed,q,v,err)
87      call ceedvectorsetvalue(v,0.d0,err)
88      call ceedvectorcreate(ceed,q,qdata,err)
89      call ceedvectorsetvalue(qdata,0.d0,err)
90
91      call ceedqfunctionapply(qf_setup,q,w,ceed_null,ceed_null,ceed_null,&
92             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
93             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
94             &qdata,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
95             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
96             &ceed_null,ceed_null,ceed_null,ceed_null,err)
97
98      call ceedqfunctionapply(qf_mass,q,u,qdata,ceed_null,ceed_null,&
99             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
100             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
101             &v,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
102             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
103             &ceed_null,ceed_null,ceed_null,ceed_null,err)
104
105      call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err)
106      do i=1,q
107        if (abs(vv(i+voffset)-ctx(5)*vvv(i)) > 1.0D-14) then
108! LCOV_EXCL_START
109          write(*,*) 'v(i)=',vv(i+voffset),', 5*vv(i)=',ctx(5)*vvv(i)
110! LCOV_EXCL_STOP
111        endif
112      enddo
113      call ceedvectorrestorearrayread(v,vv,voffset,err)
114
115      call ceedvectordestroy(u,err)
116      call ceedvectordestroy(v,err)
117      call ceedvectordestroy(w,err)
118      call ceedvectordestroy(qdata,err)
119      call ceedqfunctiondestroy(qf_setup,err)
120      call ceedqfunctiondestroy(qf_mass,err)
121      call ceeddestroy(ceed,err)
122      end
123!-----------------------------------------------------------------------
124