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