xref: /libCEED/tests/t503-operator-f.f90 (revision d9b786505a4dfcb66b2fcd9e3b61dd507168515d)
1361afa9aSjeremylt!-----------------------------------------------------------------------
2752c3701SJeremy L Thompson!
3752c3701SJeremy L Thompson! Header with QFunctions
4752c3701SJeremy L Thompson!
5752c3701SJeremy L Thompson      include 't500-operator-f.h'
6361afa9aSjeremylt!-----------------------------------------------------------------------
7361afa9aSjeremylt      program test
81f9a83abSJed Brown      implicit none
9ec3da8bcSJed Brown      include 'ceed/fortran.h'
10361afa9aSjeremylt
11361afa9aSjeremylt      integer ceed,err,i,j
1215910d16Sjeremylt      integer stridesu(3)
1315910d16Sjeremylt      integer erestrictx,erestrictu,erestrictui
14361afa9aSjeremylt      integer bx,bu
15361afa9aSjeremylt      integer qf_setup,qf_mass
16361afa9aSjeremylt      integer op_setup,op_mass
17361afa9aSjeremylt      integer qdata,x,u,v
18361afa9aSjeremylt      integer nelem,p,q
19361afa9aSjeremylt      parameter(nelem=15)
20361afa9aSjeremylt      parameter(p=5)
21361afa9aSjeremylt      parameter(q=8)
22361afa9aSjeremylt      integer nx,nu
23361afa9aSjeremylt      parameter(nx=nelem+1)
24361afa9aSjeremylt      parameter(nu=nelem*(p-1)+1)
25361afa9aSjeremylt      integer indx(nelem*2)
26361afa9aSjeremylt      integer indu(nelem*p)
27361afa9aSjeremylt      real*8 arrx(nx)
28361afa9aSjeremylt      integer*8 voffset,xoffset
29361afa9aSjeremylt
30361afa9aSjeremylt      real*8 hv(nu)
31361afa9aSjeremylt      real*8 total
32361afa9aSjeremylt
33361afa9aSjeremylt      character arg*32
34361afa9aSjeremylt
35361afa9aSjeremylt      external setup,mass
36361afa9aSjeremylt
37361afa9aSjeremylt      call getarg(1,arg)
38361afa9aSjeremylt      call ceedinit(trim(arg)//char(0),ceed,err)
39361afa9aSjeremylt
40361afa9aSjeremylt      call ceedvectorcreate(ceed,nu,u,err)
41361afa9aSjeremylt      call ceedvectorsetvalue(u,1.d0,err)
42361afa9aSjeremylt
43361afa9aSjeremylt      call ceedvectorcreate(ceed,nu,v,err)
44361afa9aSjeremylt
45361afa9aSjeremylt      do i=0,nx-1
46361afa9aSjeremylt        arrx(i+1)=i/(nx-1.d0)
47361afa9aSjeremylt      enddo
48361afa9aSjeremylt      call ceedvectorcreate(ceed,nx,x,err)
49361afa9aSjeremylt      xoffset=0
50361afa9aSjeremylt      call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err)
51361afa9aSjeremylt
52361afa9aSjeremylt      call ceedvectorcreate(ceed,nelem*q,qdata,err)
53361afa9aSjeremylt
54361afa9aSjeremylt      do i=0,nelem-1
55361afa9aSjeremylt        indx(2*i+1)=i
56361afa9aSjeremylt        indx(2*i+2)=i+1
57361afa9aSjeremylt      enddo
58d979a051Sjeremylt      call ceedelemrestrictioncreate(ceed,nelem,2,1,1,nx,ceed_mem_host,&
59361afa9aSjeremylt     & ceed_use_pointer,indx,erestrictx,err)
60361afa9aSjeremylt
61361afa9aSjeremylt      do i=0,nelem-1
62361afa9aSjeremylt        do j=0,p-1
63361afa9aSjeremylt          indu(p*i+j+1)=i*(p-1)+j
64361afa9aSjeremylt        enddo
65361afa9aSjeremylt      enddo
66361afa9aSjeremylt
67d979a051Sjeremylt      call ceedelemrestrictioncreate(ceed,nelem,p,1,1,nu,ceed_mem_host,&
68361afa9aSjeremylt     & ceed_use_pointer,indu,erestrictu,err)
697509a596Sjeremylt      stridesu=[1,q,q]
70d979a051Sjeremylt      call ceedelemrestrictioncreatestrided(ceed,nelem,q,1,q*nelem,stridesu,&
71361afa9aSjeremylt     & erestrictui,err)
72361afa9aSjeremylt
73361afa9aSjeremylt      call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q,ceed_gauss,bx,err)
74361afa9aSjeremylt      call ceedbasiscreatetensorh1lagrange(ceed,1,1,p,q,ceed_gauss,bu,err)
75361afa9aSjeremylt
76361afa9aSjeremylt      call ceedqfunctioncreateinterior(ceed,1,setup,&
77361afa9aSjeremylt     &SOURCE_DIR&
78a05f9790Sjeremylt     &//'t500-operator.h:setup'//char(0),qf_setup,err)
79a61c78d6SJeremy L Thompson      call ceedqfunctionaddinput(qf_setup,'weight',1,ceed_eval_weight,err)
80361afa9aSjeremylt      call ceedqfunctionaddinput(qf_setup,'dx',1,ceed_eval_grad,err)
81361afa9aSjeremylt      call ceedqfunctionaddoutput(qf_setup,'rho',1,ceed_eval_none,err)
82361afa9aSjeremylt
83361afa9aSjeremylt      call ceedqfunctioncreateinterior(ceed,1,mass,&
84361afa9aSjeremylt     &SOURCE_DIR&
85a05f9790Sjeremylt     &//'t500-operator.h:mass'//char(0),qf_mass,err)
86361afa9aSjeremylt      call ceedqfunctionaddinput(qf_mass,'rho',1,ceed_eval_none,err)
87361afa9aSjeremylt      call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err)
88361afa9aSjeremylt      call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err)
89361afa9aSjeremylt
90442e7f0bSjeremylt      call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,&
91442e7f0bSjeremylt     & ceed_qfunction_none,op_setup,err)
92442e7f0bSjeremylt      call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,&
93442e7f0bSjeremylt     & ceed_qfunction_none,op_mass,err)
94361afa9aSjeremylt
95a61c78d6SJeremy L Thompson      call ceedoperatorsetfield(op_setup,'weight',ceed_elemrestriction_none,&
9615910d16Sjeremylt     & bx,ceed_vector_none,err)
97a8d32208Sjeremylt      call ceedoperatorsetfield(op_setup,'dx',erestrictx,bx,x,err)
98361afa9aSjeremylt      call ceedoperatorsetfield(op_setup,'rho',erestrictui,&
99*356036faSJeremy L Thompson     ceed_basis_none,qdata,err)
100361afa9aSjeremylt      call ceedoperatorsetfield(op_mass,'rho',erestrictui,&
101*356036faSJeremy L Thompson     ceed_basis_none,qdata,err)
102a8d32208Sjeremylt      call ceedoperatorsetfield(op_mass,'u',erestrictu,bu,u,err)
103a8d32208Sjeremylt      call ceedoperatorsetfield(op_mass,'v',erestrictu,bu,v,err)
104361afa9aSjeremylt
105e97ff134Sjeremylt      call ceedoperatorapply(op_setup,ceed_vector_none,ceed_vector_none,&
106361afa9aSjeremylt     & ceed_request_immediate,err)
107e97ff134Sjeremylt      call ceedoperatorapply(op_mass,ceed_vector_none,ceed_vector_none,&
108361afa9aSjeremylt     & ceed_request_immediate,err)
109361afa9aSjeremylt
110361afa9aSjeremylt      call ceedvectorgetarrayread(v,ceed_mem_host,hv,voffset,err)
111361afa9aSjeremylt      total=0.
112361afa9aSjeremylt      do i=1,nu
113361afa9aSjeremylt        total=total+hv(voffset+i)
114361afa9aSjeremylt      enddo
115361afa9aSjeremylt      if (abs(total-1.)>1.0d-10) then
116361afa9aSjeremylt! LCOV_EXCL_START
117361afa9aSjeremylt        write(*,*) 'Computed Area: ',total,' != True Area: 1.0'
118361afa9aSjeremylt! LCOV_EXCL_STOP
119361afa9aSjeremylt      endif
120361afa9aSjeremylt      call ceedvectorrestorearrayread(v,hv,voffset,err)
121361afa9aSjeremylt
122361afa9aSjeremylt      call ceedvectordestroy(qdata,err)
123361afa9aSjeremylt      call ceedvectordestroy(x,err)
124361afa9aSjeremylt      call ceedvectordestroy(u,err)
125361afa9aSjeremylt      call ceedvectordestroy(v,err)
126361afa9aSjeremylt      call ceedoperatordestroy(op_mass,err)
127361afa9aSjeremylt      call ceedoperatordestroy(op_setup,err)
128361afa9aSjeremylt      call ceedqfunctiondestroy(qf_mass,err)
129361afa9aSjeremylt      call ceedqfunctiondestroy(qf_setup,err)
130361afa9aSjeremylt      call ceedbasisdestroy(bu,err)
131361afa9aSjeremylt      call ceedbasisdestroy(bx,err)
132361afa9aSjeremylt      call ceedelemrestrictiondestroy(erestrictu,err)
133361afa9aSjeremylt      call ceedelemrestrictiondestroy(erestrictx,err)
134361afa9aSjeremylt      call ceedelemrestrictiondestroy(erestrictui,err)
135361afa9aSjeremylt      call ceeddestroy(ceed,err)
136361afa9aSjeremylt      end
137361afa9aSjeremylt!-----------------------------------------------------------------------
138