xref: /libCEED/tests/t504-operator-f.f90 (revision ccedf6b0eedde5bdb26f1d628a64390ea2c01243)
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! LCOV_EXCL_START
5      real*8 ctx
6      real*8 u1(1)
7      real*8 u2(1)
8      real*8 v1(1)
9      integer q,ierr
10
11      do i=1,q
12        v1(i)=u1(i)*u2(i)
13      enddo
14
15      ierr=0
16      end
17! LCOV_EXCL_STOP
18!-----------------------------------------------------------------------
19      subroutine mass(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,&
20&           u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr)
21! LCOV_EXCL_START
22      real*8 ctx
23      real*8 u1(1)
24      real*8 u2(1)
25      real*8 v1(1)
26      integer q,ierr
27
28      do i=1,q
29        v1(i)=u2(i)*u1(i)
30      enddo
31
32      ierr=0
33      end
34! LCOV_EXCL_STOP
35!-----------------------------------------------------------------------
36      program test
37
38      include 'ceedf.h'
39
40      integer ceed,err,i,j
41      integer imode
42      parameter(imode=ceed_noninterlaced)
43      integer stridesx(3),stridesu(3)
44      integer erestrictx,erestrictu,erestrictxi,erestrictui
45      integer bx,bu
46      integer qf_setup,qf_mass
47      integer op_setup,op_mass
48      integer qdata
49      integer nelem,p,q
50      parameter(nelem=15)
51      parameter(p=5)
52      parameter(q=8)
53      integer nx,nu
54      parameter(nx=nelem+1)
55      parameter(nu=nelem*(p-1)+1)
56      integer indx(nelem*2)
57      integer indu(nelem*p)
58
59      character arg*32
60
61! LCOV_EXCL_START
62      external setup,mass
63! LCOV_EXCL_STOP
64
65      call getarg(1,arg)
66      call ceedinit(trim(arg)//char(0),ceed,err)
67
68      do i=0,nelem-1
69        indx(2*i+1)=i
70        indx(2*i+2)=i+1
71      enddo
72
73      call ceedelemrestrictioncreate(ceed,imode,nelem,2,nx,1,ceed_mem_host,&
74     & ceed_use_pointer,indx,erestrictx,err)
75      stridesx=[1,2,2]
76      call ceedelemrestrictioncreatestrided(ceed,nelem,2,2*nelem,1,stridesx,&
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      stridesu=[1,q,q]
88      call ceedelemrestrictioncreatestrided(ceed,nelem,q,q*nelem,1,stridesu,&
89     & erestrictui,err)
90
91      call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q,ceed_gauss,bx,err)
92      call ceedbasiscreatetensorh1lagrange(ceed,1,1,p,q,ceed_gauss,bu,err)
93
94      call ceedqfunctioncreateinterior(ceed,1,setup,&
95     &SOURCE_DIR&
96     &//'t500-operator.h:setup'//char(0),qf_setup,err)
97      call ceedqfunctionaddinput(qf_setup,'_weight',1,ceed_eval_weight,err)
98      call ceedqfunctionaddinput(qf_setup,'dx',1,ceed_eval_grad,err)
99      call ceedqfunctionaddoutput(qf_setup,'rho',1,ceed_eval_none,err)
100
101      call ceedqfunctioncreateinterior(ceed,1,mass,&
102     &SOURCE_DIR&
103     &//'t500-operator.h:mass'//char(0),qf_mass,err)
104      call ceedqfunctionaddinput(qf_mass,'rho',1,ceed_eval_none,err)
105      call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err)
106      call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err)
107
108      call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,&
109     & ceed_qfunction_none,op_setup,err)
110      call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,&
111     & ceed_qfunction_none,op_mass,err)
112
113      call ceedvectorcreate(ceed,nx,x,err)
114      call ceedvectorcreate(ceed,nelem*q,qdata,err)
115
116      call ceedoperatorsetfield(op_setup,'_weight',erestrictxi,bx,&
117     & ceed_vector_none,err)
118      call ceedoperatorsetfield(op_setup,'dx',erestrictx,bx,&
119     & ceed_vector_active,err)
120      call ceedoperatorsetfield(op_setup,'rho',erestrictui,&
121     & ceed_basis_collocated,ceed_vector_active,err)
122      call ceedoperatorsetfield(op_mass,'rho',erestrictui,&
123     & ceed_basis_collocated,qdata,err)
124      call ceedoperatorsetfield(op_mass,'u',erestrictu,bu,&
125     & ceed_vector_active,err)
126      call ceedoperatorsetfield(op_mass,'v',erestrictu,bu,&
127     & ceed_vector_active,err)
128
129      call ceedoperatorview(op_setup,err)
130      call ceedoperatorview(op_mass,err)
131
132      call ceedvectordestroy(qdata,err)
133      call ceedoperatordestroy(op_mass,err)
134      call ceedoperatordestroy(op_setup,err)
135      call ceedqfunctiondestroy(qf_mass,err)
136      call ceedqfunctiondestroy(qf_setup,err)
137      call ceedbasisdestroy(bu,err)
138      call ceedbasisdestroy(bx,err)
139      call ceedelemrestrictiondestroy(erestrictu,err)
140      call ceedelemrestrictiondestroy(erestrictx,err)
141      call ceedelemrestrictiondestroy(erestrictui,err)
142      call ceedelemrestrictiondestroy(erestrictxi,err)
143      call ceeddestroy(ceed,err)
144      end
145!-----------------------------------------------------------------------
146