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 v1(q+i)=u2(q+i)*u1(i) 28 enddo 29 30 ierr=0 31 end 32!----------------------------------------------------------------------- 33 program test 34 35 include 'ceedf.h' 36 37 integer ceed,err,i,j 38 integer imode 39 parameter(imode=ceed_noninterlaced) 40 integer imodeu 41 parameter(imodeu=ceed_interlaced) 42 integer stridesu_small(3),stridesu_large(3) 43 integer erestrictx,erestrictu 44 integer erestrictui_small,erestrictui_large 45 integer bx_small,bu_small,bx_large,bu_large 46 integer qf_setup,qf_mass 47 integer op_setup_small,op_mass_small,op_setup_large,op_mass_large 48 integer qdata_small,qdata_large,x,u,v 49 integer nelem,p,q,scale 50 parameter(nelem=15) 51 parameter(p=5) 52 parameter(q=8) 53 parameter(scale=3) 54 integer nx,nu 55 parameter(nx=nelem+1) 56 parameter(nu=nelem*(p-1)+1) 57 integer indx(nelem*2) 58 integer indu(nelem*p) 59 real*8 arrx(nx) 60 integer*8 voffset,xoffset 61 62 real*8 hu(nu*2),hv(nu*2) 63 real*8 total1,total2 64 65 character arg*32 66 67 external setup,mass 68 69 call getarg(1,arg) 70 call ceedinit(trim(arg)//char(0),ceed,err) 71 72 do i=0,nx-1 73 arrx(i+1)=i/(nx-1.d0) 74 enddo 75 do i=0,nelem-1 76 indx(2*i+1)=i 77 indx(2*i+2)=i+1 78 enddo 79 80 call ceedelemrestrictioncreate(ceed,imode,nelem,2,nx,1,ceed_mem_host,& 81 & ceed_use_pointer,indx,erestrictx,err) 82 83 do i=0,nelem-1 84 do j=0,p-1 85 indu(p*i+j+1)=i*(p-1)+j 86 enddo 87 enddo 88 89 call ceedelemrestrictioncreate(ceed,imodeu,nelem,p,nu,2,ceed_mem_host,& 90 & ceed_use_pointer,indu,erestrictu,err) 91 stridesu_small=[1,q,q] 92 call ceedelemrestrictioncreatestrided(ceed,nelem,q,q*nelem,1,& 93 & stridesu_small,erestrictui_small,err) 94 stridesu_large=[1,q*scale,q*scale] 95 call ceedelemrestrictioncreatestrided(ceed,nelem,q*scale,q*nelem*scale,& 96 & 1,stridesu_large,erestrictui_large,err) 97 98 call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q,ceed_gauss,bx_small,err) 99 call ceedbasiscreatetensorh1lagrange(ceed,1,2,p,q,ceed_gauss,bu_small,err) 100 call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q*scale,& 101 & ceed_gauss,bx_large,err) 102 call ceedbasiscreatetensorh1lagrange(ceed,1,2,p,q*scale,& 103 & ceed_gauss,bu_large,err) 104 105! Common QFunctions 106 107 call ceedqfunctioncreateinterior(ceed,1,setup,& 108 &SOURCE_DIR& 109 &//'t502-operator.h:setup'//char(0),qf_setup,err) 110 call ceedqfunctionaddinput(qf_setup,'_weight',1,ceed_eval_weight,err) 111 call ceedqfunctionaddinput(qf_setup,'x',1,ceed_eval_grad,err) 112 call ceedqfunctionaddoutput(qf_setup,'rho',1,ceed_eval_none,err) 113 114 call ceedqfunctioncreateinterior(ceed,1,mass,& 115 &SOURCE_DIR& 116 &//'t502-operator.h:mass'//char(0),qf_mass,err) 117 call ceedqfunctionaddinput(qf_mass,'rho',1,ceed_eval_none,err) 118 call ceedqfunctionaddinput(qf_mass,'u',2,ceed_eval_interp,err) 119 call ceedqfunctionaddoutput(qf_mass,'v',2,ceed_eval_interp,err) 120 121 call ceedvectorcreate(ceed,nx,x,err) 122 xoffset=0 123 call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 124 125! Small operator 126 127 call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,& 128 & ceed_qfunction_none,op_setup_small,err) 129 call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,& 130 & ceed_qfunction_none,op_mass_small,err) 131 132 call ceedvectorcreate(ceed,nelem*q,qdata_small,err) 133 134 call ceedoperatorsetfield(op_setup_small,'_weight',& 135 & ceed_elemrestriction_none,bx_small,ceed_vector_none,err) 136 call ceedoperatorsetfield(op_setup_small,'x',erestrictx,& 137 & bx_small,ceed_vector_active,err) 138 call ceedoperatorsetfield(op_setup_small,'rho',erestrictui_small,& 139 & ceed_basis_collocated,ceed_vector_active,err) 140 call ceedoperatorsetfield(op_mass_small,'rho',erestrictui_small,& 141 & ceed_basis_collocated,qdata_small,err) 142 call ceedoperatorsetfield(op_mass_small,'u',erestrictu,& 143 & bu_small,ceed_vector_active,err) 144 call ceedoperatorsetfield(op_mass_small,'v',erestrictu,& 145 & bu_small,ceed_vector_active,err) 146 147! Large operator 148 149 call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,& 150 & ceed_qfunction_none,op_setup_large,err) 151 call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,& 152 & ceed_qfunction_none,op_mass_large,err) 153 154 call ceedvectorcreate(ceed,nelem*q*scale,qdata_large,err) 155 156 call ceedoperatorsetfield(op_setup_large,'_weight',& 157 & ceed_elemrestriction_none,bx_large,ceed_vector_none,err) 158 call ceedoperatorsetfield(op_setup_large,'x',erestrictx,& 159 & bx_large,ceed_vector_active,err) 160 call ceedoperatorsetfield(op_setup_large,'rho',erestrictui_large,& 161 & ceed_basis_collocated,ceed_vector_active,err) 162 call ceedoperatorsetfield(op_mass_large,'rho',erestrictui_large,& 163 & ceed_basis_collocated,qdata_large,err) 164 call ceedoperatorsetfield(op_mass_large,'u',erestrictu,& 165 & bu_large,ceed_vector_active,err) 166 call ceedoperatorsetfield(op_mass_large,'v',erestrictu,& 167 & bu_large,ceed_vector_active,err) 168 169! Setup U, V 170 171 call ceedvectorcreate(ceed,2*nu,u,err) 172 call ceedvectorgetarray(u,ceed_mem_host,hu,voffset,err) 173 do i=1,nu 174 hu(voffset+2*i-1)=1. 175 hu(voffset+2*i)=2. 176 enddo 177 call ceedvectorrestorearray(u,hu,voffset,err) 178 call ceedvectorcreate(ceed,2*nu,v,err) 179 180! Small apply 181 182 call ceedoperatorapply(op_setup_small,x,qdata_small,& 183 & ceed_request_immediate,err) 184 call ceedoperatorapply(op_mass_small,u,v,ceed_request_immediate,err) 185 186 call ceedvectorgetarrayread(v,ceed_mem_host,hv,voffset,err) 187 total1=0. 188 total2=0. 189 do i=1,nu 190 total1=total1+hv(voffset+2*i-1) 191 total2=total2+hv(voffset+2*i) 192 enddo 193 if (abs(total1-1.)>1.0d-10) then 194! LCOV_EXCL_START 195 write(*,*) 'Computed Area: ',total1,' != True Area: 1.0' 196! LCOV_EXCL_STOP 197 endif 198 if (abs(total2-2.)>1.0d-10) then 199! LCOV_EXCL_START 200 write(*,*) 'Computed Area: ',total2,' != True Area: 2.0' 201! LCOV_EXCL_STOP 202 endif 203 call ceedvectorrestorearrayread(v,hv,voffset,err) 204 205! Large apply 206 207 call ceedoperatorapply(op_setup_large,x,qdata_large,& 208 & ceed_request_immediate,err) 209 call ceedoperatorapply(op_mass_large,u,v,ceed_request_immediate,err) 210 211 call ceedvectorgetarrayread(v,ceed_mem_host,hv,voffset,err) 212 total1=0. 213 total2=0. 214 do i=1,nu 215 total1=total1+hv(voffset+2*i-1) 216 total2=total2+hv(voffset+2*i) 217 enddo 218 if (abs(total1-1.)>1.0d-10) then 219! LCOV_EXCL_START 220 write(*,*) 'Computed Area: ',total1,' != True Area: 1.0' 221! LCOV_EXCL_STOP 222 endif 223 if (abs(total2-2.)>1.0d-10) then 224! LCOV_EXCL_START 225 write(*,*) 'Computed Area: ',total2,' != True Area: 2.0' 226! LCOV_EXCL_STOP 227 endif 228 call ceedvectorrestorearrayread(v,hv,voffset,err) 229 230 call ceedvectordestroy(qdata_small,err) 231 call ceedvectordestroy(qdata_large,err) 232 call ceedvectordestroy(x,err) 233 call ceedvectordestroy(u,err) 234 call ceedvectordestroy(v,err) 235 call ceedoperatordestroy(op_mass_small,err) 236 call ceedoperatordestroy(op_setup_small,err) 237 call ceedoperatordestroy(op_mass_large,err) 238 call ceedoperatordestroy(op_setup_large,err) 239 call ceedqfunctiondestroy(qf_mass,err) 240 call ceedqfunctiondestroy(qf_setup,err) 241 call ceedbasisdestroy(bu_small,err) 242 call ceedbasisdestroy(bx_small,err) 243 call ceedbasisdestroy(bu_large,err) 244 call ceedbasisdestroy(bx_large,err) 245 call ceedelemrestrictiondestroy(erestrictu,err) 246 call ceedelemrestrictiondestroy(erestrictx,err) 247 call ceedelemrestrictiondestroy(erestrictui_small,err) 248 call ceedelemrestrictiondestroy(erestrictui_large,err) 249 call ceeddestroy(ceed,err) 250 end 251!----------------------------------------------------------------------- 252 253