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*15910d16Sjeremylt integer stridesu(3),stridesqd(3) 71*15910d16Sjeremylt integer erestrictx,erestrictu,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) 129b7ec98d8SJeremy L Thompson 13061dbc9d2Sjeremylt call ceedelemrestrictioncreate(ceed,imode,nelem,p*p,ndofs,1,& 131b7ec98d8SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictu,err) 1327509a596Sjeremylt stridesu=[1,q*q,q*q] 1337509a596Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,q*q,nqpts,& 1347509a596Sjeremylt & 1,stridesu,erestrictui,err) 135b7ec98d8SJeremy L Thompson 1367509a596Sjeremylt stridesqd=[1,q*q,q*q*d*(d+1)/2] 1377509a596Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,q*q,nqpts,& 1387509a596Sjeremylt & d*(d+1)/2,stridesqd,erestrictqi,err) 139b7ec98d8SJeremy L Thompson 140b7ec98d8SJeremy L Thompson! Bases 141b7ec98d8SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,d,p,q,ceed_gauss,bx,err) 142b7ec98d8SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,1,p,q,ceed_gauss,bu,err) 143b7ec98d8SJeremy L Thompson 144b7ec98d8SJeremy L Thompson! QFunction - setup mass 145b7ec98d8SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,setup_mass,& 146b7ec98d8SJeremy L Thompson &SOURCE_DIR& 147b7ec98d8SJeremy L Thompson &//'t532-operator.h:setup_mass'//char(0),qf_setup_mass,err) 148b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_mass,'dx',d*d,ceed_eval_grad,err) 149b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_mass,'_weight',1,ceed_eval_weight,err) 150b7ec98d8SJeremy L Thompson call ceedqfunctionaddoutput(qf_setup_mass,'qdata',1,ceed_eval_none,err) 151b7ec98d8SJeremy L Thompson 152b7ec98d8SJeremy L Thompson! Operator - setup mass 153442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_setup_mass,ceed_qfunction_none,& 154442e7f0bSjeremylt & ceed_qfunction_none,op_setup_mass,err) 155b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_mass,'dx',erestrictx,& 156a8d32208Sjeremylt & bx,ceed_vector_active,err) 157*15910d16Sjeremylt call ceedoperatorsetfield(op_setup_mass,'_weight',& 158*15910d16Sjeremylt & ceed_elemrestriction_none,bx,ceed_vector_none,err) 159b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_mass,'qdata',erestrictui,& 160a8d32208Sjeremylt & ceed_basis_collocated,ceed_vector_active,err) 161b7ec98d8SJeremy L Thompson 162b7ec98d8SJeremy L Thompson! QFunction - setup diff 163b7ec98d8SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,setup_diff,& 164b7ec98d8SJeremy L Thompson &SOURCE_DIR& 165b7ec98d8SJeremy L Thompson &//'t532-operator.h:setup_diff'//char(0),qf_setup_diff,err) 166b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_diff,'dx',d*d,ceed_eval_grad,err) 167b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_diff,'_weight',1,ceed_eval_weight,err) 168b7ec98d8SJeremy L Thompson call ceedqfunctionaddoutput(qf_setup_diff,'qdata',& 169b7ec98d8SJeremy L Thompson & d*(d+1)/2,ceed_eval_none,err) 170b7ec98d8SJeremy L Thompson 171b7ec98d8SJeremy L Thompson! Operator - setup diff 172442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_setup_diff,ceed_qfunction_none,& 173442e7f0bSjeremylt & ceed_qfunction_none,op_setup_diff,err) 174b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_diff,'dx',erestrictx,& 175a8d32208Sjeremylt & bx,ceed_vector_active,err) 176*15910d16Sjeremylt call ceedoperatorsetfield(op_setup_diff,'_weight',& 177*15910d16Sjeremylt & ceed_elemrestriction_none,bx,ceed_vector_none,err) 178b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_setup_diff,'qdata',erestrictqi,& 179a8d32208Sjeremylt & ceed_basis_collocated,ceed_vector_active,err) 180b7ec98d8SJeremy L Thompson 181b7ec98d8SJeremy L Thompson! Apply Setup Operators 182b7ec98d8SJeremy L Thompson call ceedoperatorapply(op_setup_mass,x,qdata_mass,& 183b7ec98d8SJeremy L Thompson & ceed_request_immediate,err) 184b7ec98d8SJeremy L Thompson call ceedoperatorapply(op_setup_diff,x,qdata_diff,& 185b7ec98d8SJeremy L Thompson & ceed_request_immediate,err) 186b7ec98d8SJeremy L Thompson 187b7ec98d8SJeremy L Thompson! QFunction - apply 188b7ec98d8SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,apply,& 189b7ec98d8SJeremy L Thompson &SOURCE_DIR& 190b7ec98d8SJeremy L Thompson &//'t532-operator.h:apply'//char(0),qf_apply,err) 191b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'du',d,ceed_eval_grad,err) 192b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'qdata_mass',1,ceed_eval_none,err) 193b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'qdata_diff',& 194b7ec98d8SJeremy L Thompson & d*(d+1)/2,ceed_eval_none,err) 195b7ec98d8SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'u',1,ceed_eval_interp,err) 196b7ec98d8SJeremy L Thompson call ceedqfunctionaddoutput(qf_apply,'v',1,ceed_eval_interp,err) 197b7ec98d8SJeremy L Thompson call ceedqfunctionaddoutput(qf_apply,'dv',d,ceed_eval_grad,err) 198b7ec98d8SJeremy L Thompson 199b7ec98d8SJeremy L Thompson! Operator - apply 200442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_apply,ceed_qfunction_none,& 201442e7f0bSjeremylt & ceed_qfunction_none,op_apply,err) 202b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'du',erestrictu,& 203a8d32208Sjeremylt & bu,ceed_vector_active,err) 204b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'qdata_mass',erestrictui,& 205a8d32208Sjeremylt & ceed_basis_collocated,qdata_mass,err) 206b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'qdata_diff',erestrictqi,& 207a8d32208Sjeremylt & ceed_basis_collocated,qdata_diff,err) 208b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'u',erestrictu,& 209a8d32208Sjeremylt & bu,ceed_vector_active,err) 210b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'v',erestrictu,& 211a8d32208Sjeremylt & bu,ceed_vector_active,err) 212b7ec98d8SJeremy L Thompson call ceedoperatorsetfield(op_apply,'dv',erestrictu,& 213a8d32208Sjeremylt & bu,ceed_vector_active,err) 214b7ec98d8SJeremy L Thompson 215b7ec98d8SJeremy L Thompson! Assemble Diagonal 216b7ec98d8SJeremy L Thompson call ceedoperatorassemblelineardiagonal(op_apply,a,& 217b7ec98d8SJeremy L Thompson & ceed_request_immediate,err) 218b7ec98d8SJeremy L Thompson 219b7ec98d8SJeremy L Thompson! Manually assemble diagonal 220b7ec98d8SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,u,err) 221b7ec98d8SJeremy L Thompson call ceedvectorsetvalue(u,0.d0,err) 222b7ec98d8SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,v,err) 223b7ec98d8SJeremy L Thompson do i=1,ndofs 224b7ec98d8SJeremy L Thompson call ceedvectorgetarray(u,ceed_mem_host,uu,uoffset,err) 225b7ec98d8SJeremy L Thompson uu(i+uoffset)=1.d0 226b7ec98d8SJeremy L Thompson if (i>1) then 227b7ec98d8SJeremy L Thompson uu(i-1+uoffset)=0.d0 228b7ec98d8SJeremy L Thompson endif 229b7ec98d8SJeremy L Thompson call ceedvectorrestorearray(u,uu,uoffset,err) 230b7ec98d8SJeremy L Thompson 231b7ec98d8SJeremy L Thompson call ceedoperatorapply(op_apply,u,v,ceed_request_immediate,err) 232b7ec98d8SJeremy L Thompson 233b7ec98d8SJeremy L Thompson call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err) 234b7ec98d8SJeremy L Thompson atrue(i)=vv(voffset+i) 235b7ec98d8SJeremy L Thompson call ceedvectorrestorearrayread(v,vv,voffset,err) 236b7ec98d8SJeremy L Thompson enddo 237b7ec98d8SJeremy L Thompson 238b7ec98d8SJeremy L Thompson! Check Output 239b7ec98d8SJeremy L Thompson call ceedvectorgetarrayread(a,ceed_mem_host,aa,aoffset,err) 240b7ec98d8SJeremy L Thompson do i=1,ndofs 241b7ec98d8SJeremy L Thompson if (abs(aa(aoffset+i)-atrue(i))>1.0d-14) then 242b7ec98d8SJeremy L Thompson! LCOV_EXCL_START 243b7ec98d8SJeremy L Thompson write(*,*) '[',i,'] Error in assembly: ',aa(aoffset+i),' != ',& 244b7ec98d8SJeremy L Thompson & atrue(i) 245b7ec98d8SJeremy L Thompson! LCOV_EXCL_STOP 246b7ec98d8SJeremy L Thompson endif 247b7ec98d8SJeremy L Thompson enddo 248b7ec98d8SJeremy L Thompson call ceedvectorrestorearrayread(a,aa,aoffset,err) 249b7ec98d8SJeremy L Thompson 250b7ec98d8SJeremy L Thompson! Cleanup 251b7ec98d8SJeremy L Thompson call ceedqfunctiondestroy(qf_setup_mass,err) 252b7ec98d8SJeremy L Thompson call ceedqfunctiondestroy(qf_setup_diff,err) 253b7ec98d8SJeremy L Thompson call ceedqfunctiondestroy(qf_apply,err) 254b7ec98d8SJeremy L Thompson call ceedoperatordestroy(op_setup_mass,err) 255b7ec98d8SJeremy L Thompson call ceedoperatordestroy(op_setup_diff,err) 256b7ec98d8SJeremy L Thompson call ceedoperatordestroy(op_apply,err) 257b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictu,err) 258b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictx,err) 259b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictui,err) 260b7ec98d8SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictqi,err) 261b7ec98d8SJeremy L Thompson call ceedbasisdestroy(bu,err) 262b7ec98d8SJeremy L Thompson call ceedbasisdestroy(bx,err) 263b7ec98d8SJeremy L Thompson call ceedvectordestroy(x,err) 264b7ec98d8SJeremy L Thompson call ceedvectordestroy(a,err) 265b7ec98d8SJeremy L Thompson call ceedvectordestroy(u,err) 266b7ec98d8SJeremy L Thompson call ceedvectordestroy(v,err) 267b7ec98d8SJeremy L Thompson call ceedvectordestroy(qdata_mass,err) 268b7ec98d8SJeremy L Thompson call ceedvectordestroy(qdata_diff,err) 269b7ec98d8SJeremy L Thompson call ceeddestroy(ceed,err) 270b7ec98d8SJeremy L Thompson end 271b7ec98d8SJeremy L Thompson!----------------------------------------------------------------------- 272