1b7ec98d8SJeremy L Thompson!----------------------------------------------------------------------- 2b7ec98d8SJeremy L Thompson subroutine setup_mass(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,& 3b7ec98d8SJeremy L Thompson& u14,u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,& 4b7ec98d8SJeremy L Thompson& v16,ierr) 5b7ec98d8SJeremy L Thompson real*8 ctx 6b7ec98d8SJeremy L Thompson real*8 u1(1) 7b7ec98d8SJeremy L Thompson real*8 u2(1) 8b7ec98d8SJeremy L Thompson real*8 v1(1) 9b7ec98d8SJeremy L Thompson integer q,ierr 10b7ec98d8SJeremy L Thompson 11b7ec98d8SJeremy L Thompson do i=1,q 12b7ec98d8SJeremy L Thompson v1(i)=u2(i)*(u1(i+q*0)*u1(i+q*3)-u1(i+q*1)*u1(i+q*2)) 13b7ec98d8SJeremy L Thompson enddo 14b7ec98d8SJeremy L Thompson 15b7ec98d8SJeremy L Thompson ierr=0 16b7ec98d8SJeremy L Thompson end 17b7ec98d8SJeremy L Thompson!----------------------------------------------------------------------- 18b7ec98d8SJeremy L Thompson subroutine setup_diff(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,& 19b7ec98d8SJeremy L Thompson& u14,u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,& 20b7ec98d8SJeremy L Thompson& v16,ierr) 21b7ec98d8SJeremy L Thompson real*8 ctx 22b7ec98d8SJeremy L Thompson real*8 u1(1) 23b7ec98d8SJeremy L Thompson real*8 u2(1) 24b7ec98d8SJeremy L Thompson real*8 v1(1) 25b7ec98d8SJeremy L Thompson real*8 w 26b7ec98d8SJeremy L Thompson integer q,ierr 27b7ec98d8SJeremy L Thompson 28b7ec98d8SJeremy L Thompson do i=1,q 29b7ec98d8SJeremy L Thompson w=u2(i)/(u1(i+q*0)*u1(i+q*3)-u1(i+q*1)*u1(i+q*2)) 30b7ec98d8SJeremy L Thompson v1(i+q*0)=w*(u1(i+q*2)*u1(i+q*2)+u1(i+q*3)*u1(i+q*3)) 31b7ec98d8SJeremy L Thompson v1(i+q*1)=w*(u1(i+q*0)*u1(i+q*0)+u1(i+q*1)*u1(i+q*1)) 32b7ec98d8SJeremy L Thompson v1(i+q*2)=-w*(u1(i+q*0)*u1(i+q*2)+u1(i+q*2)*u1(i+q*3)) 33b7ec98d8SJeremy L Thompson enddo 34b7ec98d8SJeremy L Thompson 35b7ec98d8SJeremy L Thompson ierr=0 36b7ec98d8SJeremy L Thompson end 37b7ec98d8SJeremy L Thompson!----------------------------------------------------------------------- 38b7ec98d8SJeremy L Thompson subroutine apply(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,& 39b7ec98d8SJeremy L Thompson& u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr) 40b7ec98d8SJeremy L Thompson real*8 ctx 41b7ec98d8SJeremy L Thompson real*8 u1(1) 42b7ec98d8SJeremy L Thompson real*8 u2(1) 43b7ec98d8SJeremy L Thompson real*8 u3(1) 44b7ec98d8SJeremy L Thompson real*8 u4(1) 45b7ec98d8SJeremy L Thompson real*8 v1(1) 46b7ec98d8SJeremy L Thompson real*8 v2(1) 47b7ec98d8SJeremy L Thompson real*8 du0,du1 48b7ec98d8SJeremy L Thompson integer q,ierr 49b7ec98d8SJeremy L Thompson 50b7ec98d8SJeremy L Thompson do i=1,q 51b7ec98d8SJeremy L Thompson! mass 52b7ec98d8SJeremy L Thompson v1(i) = u2(i)*u4(i) 53b7ec98d8SJeremy L Thompson! diff 54b7ec98d8SJeremy L Thompson du0=u1(i+q*0) 55b7ec98d8SJeremy L Thompson du1=u1(i+q*1) 56b7ec98d8SJeremy L Thompson v2(i+q*0)=u3(i+q*0)*du0+u3(i+q*2)*du1 57b7ec98d8SJeremy L Thompson v2(i+q*1)=u3(i+q*2)*du0+u3(i+q*1)*du1 58b7ec98d8SJeremy L Thompson enddo 59b7ec98d8SJeremy L Thompson 60b7ec98d8SJeremy L Thompson ierr=0 61b7ec98d8SJeremy L Thompson end 62b7ec98d8SJeremy L Thompson!----------------------------------------------------------------------- 63b7ec98d8SJeremy L Thompson program test 64b7ec98d8SJeremy L Thompson 65b7ec98d8SJeremy L Thompson include 'ceedf.h' 66b7ec98d8SJeremy L Thompson 67b7ec98d8SJeremy L Thompson integer ceed,err,i,j,k 6861dbc9d2Sjeremylt integer imode 6961dbc9d2Sjeremylt parameter(imode=ceed_noninterlaced) 70*7509a596Sjeremylt integer stridesx(3),stridesu(3),stridesqd(3) 71b7ec98d8SJeremy L Thompson integer erestrictx,erestrictu,erestrictxi,erestrictui,erestrictqi 72b7ec98d8SJeremy L Thompson integer bx,bu 73b7ec98d8SJeremy L Thompson integer qf_setup_mass,qf_setup_diff,qf_apply 74b7ec98d8SJeremy L Thompson integer op_setup_mass,op_setup_diff,op_apply 75b7ec98d8SJeremy L Thompson integer qdata_mass,qdata_diff,x,a,u,v 76b7ec98d8SJeremy L Thompson integer nelem,p,q,d 77b7ec98d8SJeremy L Thompson integer row,col,offset 78b7ec98d8SJeremy L Thompson parameter(nelem=6) 79b7ec98d8SJeremy L Thompson parameter(p=3) 80b7ec98d8SJeremy L Thompson parameter(q=4) 81b7ec98d8SJeremy L Thompson parameter(d=2) 82b7ec98d8SJeremy L Thompson integer ndofs,nqpts,nx,ny 83b7ec98d8SJeremy L Thompson parameter(nx=3) 84b7ec98d8SJeremy L Thompson parameter(ny=2) 85b7ec98d8SJeremy L Thompson parameter(ndofs=(nx*2+1)*(ny*2+1)) 86b7ec98d8SJeremy L Thompson parameter(nqpts=nelem*q*q) 87b7ec98d8SJeremy L Thompson integer indx(nelem*p*p) 88b7ec98d8SJeremy L Thompson real*8 arrx(d*ndofs),aa(nqpts),uu(ndofs),vv(ndofs),atrue(ndofs) 89b7ec98d8SJeremy L Thompson integer*8 xoffset,aoffset,uoffset,voffset 90b7ec98d8SJeremy L Thompson 91b7ec98d8SJeremy L Thompson character arg*32 92b7ec98d8SJeremy L Thompson 93b7ec98d8SJeremy L Thompson external setup_mass,setup_diff,apply,apply_lin 94b7ec98d8SJeremy L Thompson 95b7ec98d8SJeremy L Thompson call getarg(1,arg) 96b7ec98d8SJeremy L Thompson 97b7ec98d8SJeremy L Thompson call ceedinit(trim(arg)//char(0),ceed,err) 98b7ec98d8SJeremy L Thompson 99b7ec98d8SJeremy L Thompson! DoF Coordinates 100b7ec98d8SJeremy L Thompson do i=0,nx*2 101b7ec98d8SJeremy L Thompson do j=0,ny*2 102b7ec98d8SJeremy L Thompson arrx(i+j*(nx*2+1)+0*ndofs+1)=1.d0*i/(2*nx) 103b7ec98d8SJeremy L Thompson arrx(i+j*(nx*2+1)+1*ndofs+1)=1.d0*j/(2*ny) 104b7ec98d8SJeremy L Thompson enddo 105b7ec98d8SJeremy L Thompson enddo 106b7ec98d8SJeremy L Thompson call ceedvectorcreate(ceed,d*ndofs,x,err) 107b7ec98d8SJeremy L Thompson xoffset=0 108b7ec98d8SJeremy L Thompson call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 109b7ec98d8SJeremy L Thompson 110b7ec98d8SJeremy L Thompson! Qdata Vector 111b7ec98d8SJeremy L Thompson call ceedvectorcreate(ceed,nqpts,qdata_mass,err) 112b7ec98d8SJeremy L Thompson call ceedvectorcreate(ceed,nqpts*d*(d+1)/2,qdata_diff,err) 113b7ec98d8SJeremy L Thompson 114b7ec98d8SJeremy L Thompson! Element Setup 115b7ec98d8SJeremy L Thompson do i=0,nelem-1 116b7ec98d8SJeremy L Thompson col=mod(i,nx) 117b7ec98d8SJeremy L Thompson row=i/nx 118b7ec98d8SJeremy L Thompson offset=col*(p-1)+row*(nx*2+1)*(p-1) 119b7ec98d8SJeremy L Thompson do j=0,p-1 120b7ec98d8SJeremy L Thompson do k=0,p-1 121b7ec98d8SJeremy L Thompson indx(p*(p*i+k)+j+1)=offset+k*(nx*2+1)+j 122b7ec98d8SJeremy L Thompson enddo 123b7ec98d8SJeremy L Thompson enddo 124b7ec98d8SJeremy L Thompson enddo 125b7ec98d8SJeremy L Thompson 126b7ec98d8SJeremy L Thompson! Restrictions 12761dbc9d2Sjeremylt call ceedelemrestrictioncreate(ceed,imode,nelem,p*p,ndofs,d,& 128b7ec98d8SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictx,err) 129*7509a596Sjeremylt stridesx=[1,p*p,p*p*d] 130*7509a596Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,p*p,& 131*7509a596Sjeremylt & nelem*p*p,d,stridesx,erestrictxi,err) 132b7ec98d8SJeremy L Thompson 13361dbc9d2Sjeremylt call ceedelemrestrictioncreate(ceed,imode,nelem,p*p,ndofs,1,& 134b7ec98d8SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictu,err) 135*7509a596Sjeremylt stridesu=[1,q*q,q*q] 136*7509a596Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,q*q,nqpts,& 137*7509a596Sjeremylt & 1,stridesu,erestrictui,err) 138b7ec98d8SJeremy L Thompson 139*7509a596Sjeremylt stridesqd=[1,q*q,q*q*d*(d+1)/2] 140*7509a596Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,q*q,nqpts,& 141*7509a596Sjeremylt & d*(d+1)/2,stridesqd,erestrictqi,err) 142b7ec98d8SJeremy L Thompson 143b7ec98d8SJeremy L Thompson! Bases 144b7ec98d8SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,d,p,q,ceed_gauss,bx,err) 145b7ec98d8SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,1,p,q,ceed_gauss,bu,err) 146b7ec98d8SJeremy L Thompson 147b7ec98d8SJeremy L Thompson! QFunction - setup mass 148b7ec98d8SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,setup_mass,& 149b7ec98d8SJeremy L Thompson &SOURCE_DIR& 150b7ec98d8SJeremy L Thompson &//'t532-operator.h:setup_mass'//char(0),qf_setup_mass,err) 151b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_mass,'dx',d*d,ceed_eval_grad,err) 152b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_mass,'_weight',1,ceed_eval_weight,err) 153b7ec98d8SJeremy L Thompson call ceedqfunctionaddoutput(qf_setup_mass,'qdata',1,ceed_eval_none,err) 154b7ec98d8SJeremy L Thompson 155b7ec98d8SJeremy L Thompson! Operator - setup mass 156442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_setup_mass,ceed_qfunction_none,& 157442e7f0bSjeremylt & ceed_qfunction_none,op_setup_mass,err) 158b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_mass,'dx',erestrictx,& 159a8d32208Sjeremylt & bx,ceed_vector_active,err) 160b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_mass,'_weight',erestrictxi,& 161a8d32208Sjeremylt & bx,ceed_vector_none,err) 162b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_mass,'qdata',erestrictui,& 163a8d32208Sjeremylt & ceed_basis_collocated,ceed_vector_active,err) 164b7ec98d8SJeremy L Thompson 165b7ec98d8SJeremy L Thompson! QFunction - setup diff 166b7ec98d8SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,setup_diff,& 167b7ec98d8SJeremy L Thompson &SOURCE_DIR& 168b7ec98d8SJeremy L Thompson &//'t532-operator.h:setup_diff'//char(0),qf_setup_diff,err) 169b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_diff,'dx',d*d,ceed_eval_grad,err) 170b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_diff,'_weight',1,ceed_eval_weight,err) 171b7ec98d8SJeremy L Thompson call ceedqfunctionaddoutput(qf_setup_diff,'qdata',& 172b7ec98d8SJeremy L Thompson & d*(d+1)/2,ceed_eval_none,err) 173b7ec98d8SJeremy L Thompson 174b7ec98d8SJeremy L Thompson! Operator - setup diff 175442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_setup_diff,ceed_qfunction_none,& 176442e7f0bSjeremylt & ceed_qfunction_none,op_setup_diff,err) 177b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_diff,'dx',erestrictx,& 178a8d32208Sjeremylt & bx,ceed_vector_active,err) 179b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_diff,'_weight',erestrictxi,& 180a8d32208Sjeremylt & bx,ceed_vector_none,err) 181b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_diff,'qdata',erestrictqi,& 182a8d32208Sjeremylt & ceed_basis_collocated,ceed_vector_active,err) 183b7ec98d8SJeremy L Thompson 184b7ec98d8SJeremy L Thompson! Apply Setup Operators 185b7ec98d8SJeremy L Thompson call ceedoperatorapply(op_setup_mass,x,qdata_mass,& 186b7ec98d8SJeremy L Thompson & ceed_request_immediate,err) 187b7ec98d8SJeremy L Thompson call ceedoperatorapply(op_setup_diff,x,qdata_diff,& 188b7ec98d8SJeremy L Thompson & ceed_request_immediate,err) 189b7ec98d8SJeremy L Thompson 190b7ec98d8SJeremy L Thompson! QFunction - apply 191b7ec98d8SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,apply,& 192b7ec98d8SJeremy L Thompson &SOURCE_DIR& 193b7ec98d8SJeremy L Thompson &//'t532-operator.h:apply'//char(0),qf_apply,err) 194b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'du',d,ceed_eval_grad,err) 195b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'qdata_mass',1,ceed_eval_none,err) 196b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'qdata_diff',& 197b7ec98d8SJeremy L Thompson & d*(d+1)/2,ceed_eval_none,err) 198b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'u',1,ceed_eval_interp,err) 199b7ec98d8SJeremy L Thompson call ceedqfunctionaddoutput(qf_apply,'v',1,ceed_eval_interp,err) 200b7ec98d8SJeremy L Thompson call ceedqfunctionaddoutput(qf_apply,'dv',d,ceed_eval_grad,err) 201b7ec98d8SJeremy L Thompson 202b7ec98d8SJeremy L Thompson! Operator - apply 203442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_apply,ceed_qfunction_none,& 204442e7f0bSjeremylt & ceed_qfunction_none,op_apply,err) 205b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'du',erestrictu,& 206a8d32208Sjeremylt & bu,ceed_vector_active,err) 207b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'qdata_mass',erestrictui,& 208a8d32208Sjeremylt & ceed_basis_collocated,qdata_mass,err) 209b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'qdata_diff',erestrictqi,& 210a8d32208Sjeremylt & ceed_basis_collocated,qdata_diff,err) 211b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'u',erestrictu,& 212a8d32208Sjeremylt & bu,ceed_vector_active,err) 213b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'v',erestrictu,& 214a8d32208Sjeremylt & bu,ceed_vector_active,err) 215b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'dv',erestrictu,& 216a8d32208Sjeremylt & bu,ceed_vector_active,err) 217b7ec98d8SJeremy L Thompson 218b7ec98d8SJeremy L Thompson! Assemble Diagonal 219b7ec98d8SJeremy L Thompson call ceedoperatorassemblelineardiagonal(op_apply,a,& 220b7ec98d8SJeremy L Thompson & ceed_request_immediate,err) 221b7ec98d8SJeremy L Thompson 222b7ec98d8SJeremy L Thompson! Manually assemble diagonal 223b7ec98d8SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,u,err) 224b7ec98d8SJeremy L Thompson call ceedvectorsetvalue(u,0.d0,err) 225b7ec98d8SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,v,err) 226b7ec98d8SJeremy L Thompson do i=1,ndofs 227b7ec98d8SJeremy L Thompson call ceedvectorgetarray(u,ceed_mem_host,uu,uoffset,err) 228b7ec98d8SJeremy L Thompson uu(i+uoffset)=1.d0 229b7ec98d8SJeremy L Thompson if (i>1) then 230b7ec98d8SJeremy L Thompson uu(i-1+uoffset)=0.d0 231b7ec98d8SJeremy L Thompson endif 232b7ec98d8SJeremy L Thompson call ceedvectorrestorearray(u,uu,uoffset,err) 233b7ec98d8SJeremy L Thompson 234b7ec98d8SJeremy L Thompson call ceedoperatorapply(op_apply,u,v,ceed_request_immediate,err) 235b7ec98d8SJeremy L Thompson 236b7ec98d8SJeremy L Thompson call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err) 237b7ec98d8SJeremy L Thompson atrue(i)=vv(voffset+i) 238b7ec98d8SJeremy L Thompson call ceedvectorrestorearrayread(v,vv,voffset,err) 239b7ec98d8SJeremy L Thompson enddo 240b7ec98d8SJeremy L Thompson 241b7ec98d8SJeremy L Thompson! Check Output 242b7ec98d8SJeremy L Thompson call ceedvectorgetarrayread(a,ceed_mem_host,aa,aoffset,err) 243b7ec98d8SJeremy L Thompson do i=1,ndofs 244b7ec98d8SJeremy L Thompson if (abs(aa(aoffset+i)-atrue(i))>1.0d-14) then 245b7ec98d8SJeremy L Thompson! LCOV_EXCL_START 246b7ec98d8SJeremy L Thompson write(*,*) '[',i,'] Error in assembly: ',aa(aoffset+i),' != ',& 247b7ec98d8SJeremy L Thompson & atrue(i) 248b7ec98d8SJeremy L Thompson! LCOV_EXCL_STOP 249b7ec98d8SJeremy L Thompson endif 250b7ec98d8SJeremy L Thompson enddo 251b7ec98d8SJeremy L Thompson call ceedvectorrestorearrayread(a,aa,aoffset,err) 252b7ec98d8SJeremy L Thompson 253b7ec98d8SJeremy L Thompson! Cleanup 254b7ec98d8SJeremy L Thompson call ceedqfunctiondestroy(qf_setup_mass,err) 255b7ec98d8SJeremy L Thompson call ceedqfunctiondestroy(qf_setup_diff,err) 256b7ec98d8SJeremy L Thompson call ceedqfunctiondestroy(qf_apply,err) 257b7ec98d8SJeremy L Thompson call ceedoperatordestroy(op_setup_mass,err) 258b7ec98d8SJeremy L Thompson call ceedoperatordestroy(op_setup_diff,err) 259b7ec98d8SJeremy L Thompson call ceedoperatordestroy(op_apply,err) 260b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictu,err) 261b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictx,err) 262b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictxi,err) 263b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictui,err) 264b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictqi,err) 265b7ec98d8SJeremy L Thompson call ceedbasisdestroy(bu,err) 266b7ec98d8SJeremy L Thompson call ceedbasisdestroy(bx,err) 267b7ec98d8SJeremy L Thompson call ceedvectordestroy(x,err) 268b7ec98d8SJeremy L Thompson call ceedvectordestroy(a,err) 269b7ec98d8SJeremy L Thompson call ceedvectordestroy(u,err) 270b7ec98d8SJeremy L Thompson call ceedvectordestroy(v,err) 271b7ec98d8SJeremy L Thompson call ceedvectordestroy(qdata_mass,err) 272b7ec98d8SJeremy L Thompson call ceedvectordestroy(qdata_diff,err) 273b7ec98d8SJeremy L Thompson call ceeddestroy(ceed,err) 274b7ec98d8SJeremy L Thompson end 275b7ec98d8SJeremy L Thompson!----------------------------------------------------------------------- 276