xref: /libCEED/tests/t505-operator-f.f90 (revision 4092d0ee9dee1dc94927b92ec4a4f5b5b7bb02dc)
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
5      real*8 u1(1)
6      real*8 u2(1)
7      real*8 v1(1)
8      integer q,ierr
9
10      do i=1,q
11        v1(i)=u1(i)*u2(i)
12      enddo
13
14      ierr=0
15      end
16!-----------------------------------------------------------------------
17      subroutine mass(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,&
18&           u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr)
19      real*8 ctx
20      real*8 u1(1)
21      real*8 u2(1)
22      real*8 v1(1)
23      integer q,ierr
24
25      do i=1,q
26        v1(i)=u2(i)*u1(i)
27      enddo
28
29      ierr=0
30      end
31!-----------------------------------------------------------------------
32      program test
33
34      include 'ceedf.h'
35
36      integer ceed,err,i,j
37      integer imode
38      parameter(imode=ceed_noninterlaced)
39      integer erestrictx,erestrictu,erestrictxi,erestrictui
40      integer bx,bu
41      integer qf_setup,qf_mass
42      integer op_setup,op_mass
43      integer qdata,x,u,v
44      integer nelem,p,q
45      parameter(nelem=15)
46      parameter(p=5)
47      parameter(q=8)
48      integer nx,nu
49      parameter(nx=nelem+1)
50      parameter(nu=nelem*(p-1)+1)
51      integer indx(nelem*2)
52      integer indu(nelem*p)
53      real*8 arrx(nx)
54      integer*8 voffset,xoffset
55
56      real*8 hv(nu)
57      real*8 total
58
59      character arg*32
60
61      external setup,mass
62
63      call getarg(1,arg)
64      call ceedinit(trim(arg)//char(0),ceed,err)
65
66      do i=0,nx-1
67        arrx(i+1)=i/(nx-1.d0)
68      enddo
69      do i=0,nelem-1
70        indx(2*i+1)=i
71        indx(2*i+2)=i+1
72      enddo
73
74      call ceedelemrestrictioncreate(ceed,imode,nelem,2,nx,1,ceed_mem_host,&
75     & ceed_use_pointer,indx,erestrictx,err)
76      call ceedelemrestrictioncreateidentity(ceed,imode,nelem,2,2*nelem,1,&
77     & erestrictxi,err)
78
79      do i=0,nelem-1
80        do j=0,p-1
81          indu(p*i+j+1)=i*(p-1)+j
82        enddo
83      enddo
84
85      call ceedelemrestrictioncreate(ceed,imode,nelem,p,nu,1,ceed_mem_host,&
86     & ceed_use_pointer,indu,erestrictu,err)
87      call ceedelemrestrictioncreateidentity(ceed,imode,nelem,q,q*nelem,1,&
88     & erestrictui,err)
89
90      call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q,ceed_gauss,bx,err)
91      call ceedbasiscreatetensorh1lagrange(ceed,1,1,p,q,ceed_gauss,bu,err)
92
93      call ceedqfunctioncreateinterior(ceed,1,setup,&
94     &SOURCE_DIR&
95     &//'t500-operator.h:setup'//char(0),qf_setup,err)
96      call ceedqfunctionaddinput(qf_setup,'_weight',1,ceed_eval_weight,err)
97      call ceedqfunctionaddinput(qf_setup,'dx',1,ceed_eval_grad,err)
98      call ceedqfunctionaddoutput(qf_setup,'rho',1,ceed_eval_none,err)
99
100      call ceedqfunctioncreateinterior(ceed,1,mass,&
101     &SOURCE_DIR&
102     &//'t500-operator.h:mass'//char(0),qf_mass,err)
103      call ceedqfunctionaddinput(qf_mass,'rho',1,ceed_eval_none,err)
104      call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err)
105      call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err)
106
107      call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,&
108     & ceed_qfunction_none,op_setup,err)
109      call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,&
110     & ceed_qfunction_none,op_mass,err)
111
112      call ceedvectorcreate(ceed,nx,x,err)
113      xoffset=0
114      call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err)
115      call ceedvectorcreate(ceed,nelem*q,qdata,err)
116
117      call ceedoperatorsetfield(op_setup,'_weight',erestrictxi,bx,&
118     & ceed_vector_none,err)
119      call ceedoperatorsetfield(op_setup,'dx',erestrictx,bx,&
120     & ceed_vector_active,err)
121      call ceedoperatorsetfield(op_setup,'rho',erestrictui,&
122     & ceed_basis_collocated,ceed_vector_active,err)
123      call ceedoperatorsetfield(op_mass,'rho',erestrictui,&
124     & ceed_basis_collocated,qdata,err)
125      call ceedoperatorsetfield(op_mass,'u',erestrictu,bu,&
126     & ceed_vector_active,err)
127      call ceedoperatorsetfield(op_mass,'v',erestrictu,bu,&
128     & ceed_vector_active,err)
129
130      call ceedoperatorapply(op_setup,x,qdata,ceed_request_immediate,err)
131
132      call ceedvectorcreate(ceed,nu,u,err)
133      call ceedvectorsetvalue(u,1.d0,err)
134      call ceedvectorcreate(ceed,nu,v,err)
135      call ceedvectorsetvalue(v,0.d0,err)
136
137      call ceedoperatorapplyadd(op_mass,u,v,ceed_request_immediate,err)
138
139      call ceedvectorgetarrayread(v,ceed_mem_host,hv,voffset,err)
140      total=0.
141      do i=1,nu
142        total=total+hv(voffset+i)
143      enddo
144      if (abs(total-1.)>1.0d-10) then
145! LCOV_EXCL_START
146        write(*,*) 'Computed Area: ',total,' != True Area: 1.0'
147! LCOV_EXCL_STOP
148      endif
149      call ceedvectorrestorearrayread(v,hv,voffset,err)
150
151      call ceedvectorsetvalue(v,1.d0,err)
152      call ceedoperatorapplyadd(op_mass,u,v,ceed_request_immediate,err)
153
154      call ceedvectorgetarrayread(v,ceed_mem_host,hv,voffset,err)
155      total=-nu
156      do i=1,nu
157        total=total+hv(voffset+i)
158      enddo
159      if (abs(total-1.)>1.0d-10) then
160! LCOV_EXCL_START
161        write(*,*) 'Computed Area: ',total,' != True Area: 1.0'
162! LCOV_EXCL_STOP
163      endif
164      call ceedvectorrestorearrayread(v,hv,voffset,err)
165
166      call ceedvectordestroy(qdata,err)
167      call ceedvectordestroy(x,err)
168      call ceedvectordestroy(u,err)
169      call ceedvectordestroy(v,err)
170      call ceedoperatordestroy(op_mass,err)
171      call ceedoperatordestroy(op_setup,err)
172      call ceedqfunctiondestroy(qf_mass,err)
173      call ceedqfunctiondestroy(qf_setup,err)
174      call ceedbasisdestroy(bu,err)
175      call ceedbasisdestroy(bx,err)
176      call ceedelemrestrictiondestroy(erestrictu,err)
177      call ceedelemrestrictiondestroy(erestrictx,err)
178      call ceedelemrestrictiondestroy(erestrictui,err)
179      call ceedelemrestrictiondestroy(erestrictxi,err)
180      call ceeddestroy(ceed,err)
181      end
182!-----------------------------------------------------------------------
183