xref: /libCEED/tests/t400-qfunction-f.f90 (revision 9cf88b28e874a91fbe464b60f41b3973b01eeda4)
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(1)
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)=u1(i)*u2(i)
26      enddo
27
28      ierr=0
29      end
30!-----------------------------------------------------------------------
31      program test
32
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      real*8 x
45      character arg*32
46      integer*8 uoffset,voffset,woffset
47
48      external setup,mass
49
50      call getarg(1,arg)
51      call ceedinit(trim(arg)//char(0),ceed,err)
52
53      call ceedqfunctioncreateinterior(ceed,1,setup,&
54     &SOURCE_DIR&
55     &//'t400-qfunction.h:setup'//char(0),qf_setup,err)
56      call ceedqfunctionaddinput(qf_setup,'w', 1,ceed_eval_weight,err)
57      call ceedqfunctionaddoutput(qf_setup,'qdata',1,ceed_eval_none,err)
58
59      call ceedqfunctioncreateinterior(ceed,1,mass,&
60     &SOURCE_DIR&
61     &//'t400-qfunction.h:mass'//char(0),qf_mass,err)
62      call ceedqfunctionaddinput(qf_mass,'qdata',1,ceed_eval_none,err)
63      call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err)
64      call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err)
65
66      do i=0,q-1
67        x=2.0*i/(q-1)-1
68        ww(i+1)=1-x*x
69        uu(i+1)=2+3*x+5*x*x
70        vvv(i+1)=ww(i+1)*uu(i+1)
71      enddo
72
73      call ceedvectorcreate(ceed,q,w,err)
74      woffset=0
75      call ceedvectorsetarray(w,ceed_mem_host,ceed_use_pointer,ww,woffset,err)
76      call ceedvectorcreate(ceed,q,u,err)
77      uoffset=0
78      call ceedvectorsetarray(u,ceed_mem_host,ceed_use_pointer,uu,uoffset,err)
79      call ceedvectorcreate(ceed,q,v,err)
80      call ceedvectorsetvalue(v,0.d0,err)
81      call ceedvectorcreate(ceed,q,qdata,err)
82      call ceedvectorsetvalue(qdata,0.d0,err)
83
84      call ceedqfunctionapply(qf_setup,q,w,ceed_null,ceed_null,ceed_null,&
85             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
86             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
87             &qdata,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
88             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
89             &ceed_null,ceed_null,ceed_null,ceed_null,err)
90
91      call ceedqfunctionapply(qf_mass,q,u,qdata,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             &v,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 ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err)
99      do i=1,q
100        if (abs(vv(i+voffset)-vvv(i)) > 1.0D-14) then
101! LCOV_EXCL_START
102          write(*,*) 'v(i)=',vv(i+voffset),', vv(i)=',vvv(i)
103! LCOV_EXCL_STOP
104        endif
105      enddo
106      call ceedvectorrestorearrayread(v,vv,voffset,err)
107
108      call ceedvectordestroy(u,err)
109      call ceedvectordestroy(v,err)
110      call ceedvectordestroy(w,err)
111      call ceedvectordestroy(qdata,err)
112      call ceedqfunctiondestroy(qf_setup,err)
113      call ceedqfunctiondestroy(qf_mass,err)
114      call ceeddestroy(ceed,err)
115      end
116!-----------------------------------------------------------------------
117