11d102b48SJeremy L Thompson!----------------------------------------------------------------------- 2752c3701SJeremy L Thompson! 3752c3701SJeremy L Thompson! Header with QFunctions 4752c3701SJeremy L Thompson! 5752c3701SJeremy L Thompson include 't532-operator-f.h' 61d102b48SJeremy L Thompson!----------------------------------------------------------------------- 71d102b48SJeremy L Thompson program test 81f9a83abSJed Brown implicit none 9ec3da8bcSJed Brown include 'ceed/fortran.h' 101d102b48SJeremy L Thompson 111d102b48SJeremy L Thompson integer ceed,err,i,j,k 1215910d16Sjeremylt integer stridesu(3),stridesqd(3) 1315910d16Sjeremylt integer erestrictx,erestrictu,erestrictui 141d102b48SJeremy L Thompson integer erestrictqi,erestrictlini 151d102b48SJeremy L Thompson integer bx,bu 161d102b48SJeremy L Thompson integer qf_setup_mass,qf_setup_diff,qf_apply,qf_apply_lin 171d102b48SJeremy L Thompson integer op_setup_mass,op_setup_diff,op_apply,op_apply_lin 181d102b48SJeremy L Thompson integer qdata_mass,qdata_diff,x,a,u,v 191d102b48SJeremy L Thompson integer nelem,p,q,d 201d102b48SJeremy L Thompson integer row,col,offset 211d102b48SJeremy L Thompson parameter(nelem=6) 221d102b48SJeremy L Thompson parameter(p=3) 231d102b48SJeremy L Thompson parameter(q=4) 241d102b48SJeremy L Thompson parameter(d=2) 251d102b48SJeremy L Thompson integer ndofs,nqpts,nx,ny 261d102b48SJeremy L Thompson parameter(nx=3) 271d102b48SJeremy L Thompson parameter(ny=2) 281d102b48SJeremy L Thompson parameter(ndofs=(nx*2+1)*(ny*2+1)) 291d102b48SJeremy L Thompson parameter(nqpts=nelem*q*q) 301d102b48SJeremy L Thompson integer indx(nelem*p*p) 311d102b48SJeremy L Thompson real*8 arrx(d*ndofs),vv(ndofs) 321d102b48SJeremy L Thompson real*8 total 331d102b48SJeremy L Thompson integer*8 xoffset,voffset 341d102b48SJeremy L Thompson 351d102b48SJeremy L Thompson character arg*32 361d102b48SJeremy L Thompson 371d102b48SJeremy L Thompson external setup_mass,setup_diff,apply,apply_lin 381d102b48SJeremy L Thompson 391d102b48SJeremy L Thompson call getarg(1,arg) 401d102b48SJeremy L Thompson 411d102b48SJeremy L Thompson call ceedinit(trim(arg)//char(0),ceed,err) 421d102b48SJeremy L Thompson 431d102b48SJeremy L Thompson! DoF Coordinates 441d102b48SJeremy L Thompson do i=0,nx*2 451d102b48SJeremy L Thompson do j=0,ny*2 461d102b48SJeremy L Thompson arrx(i+j*(nx*2+1)+0*ndofs+1)=1.d0*i/(2*nx) 471d102b48SJeremy L Thompson arrx(i+j*(nx*2+1)+1*ndofs+1)=1.d0*j/(2*ny) 481d102b48SJeremy L Thompson enddo 491d102b48SJeremy L Thompson enddo 501d102b48SJeremy L Thompson call ceedvectorcreate(ceed,d*ndofs,x,err) 511d102b48SJeremy L Thompson xoffset=0 521d102b48SJeremy L Thompson call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 531d102b48SJeremy L Thompson 541d102b48SJeremy L Thompson! Qdata Vector 551d102b48SJeremy L Thompson call ceedvectorcreate(ceed,nqpts,qdata_mass,err) 561d102b48SJeremy L Thompson call ceedvectorcreate(ceed,nqpts*d*(d+1)/2,qdata_diff,err) 571d102b48SJeremy L Thompson 581d102b48SJeremy L Thompson! Element Setup 591d102b48SJeremy L Thompson do i=0,nelem-1 601d102b48SJeremy L Thompson col=mod(i,nx) 611d102b48SJeremy L Thompson row=i/nx 621d102b48SJeremy L Thompson offset=col*(p-1)+row*(nx*2+1)*(p-1) 631d102b48SJeremy L Thompson do j=0,p-1 641d102b48SJeremy L Thompson do k=0,p-1 651d102b48SJeremy L Thompson indx(p*(p*i+k)+j+1)=offset+k*(nx*2+1)+j 661d102b48SJeremy L Thompson enddo 671d102b48SJeremy L Thompson enddo 681d102b48SJeremy L Thompson enddo 691d102b48SJeremy L Thompson 701d102b48SJeremy L Thompson! Restrictions 71d979a051Sjeremylt call ceedelemrestrictioncreate(ceed,nelem,p*p,d,ndofs,d*ndofs,& 721d102b48SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictx,err) 731d102b48SJeremy L Thompson 74d979a051Sjeremylt call ceedelemrestrictioncreate(ceed,nelem,p*p,1,1,ndofs,& 751d102b48SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictu,err) 767509a596Sjeremylt stridesu=[1,q*q,q*q] 77d979a051Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,q*q,1,nqpts,& 78d979a051Sjeremylt & stridesu,erestrictui,err) 791d102b48SJeremy L Thompson 807509a596Sjeremylt stridesqd=[1,q*q,q*q*d*(d+1)/2] 81d979a051Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,q*q,d*(d+1)/2,& 82d979a051Sjeremylt & d*(d+1)/2*nqpts,stridesqd,erestrictqi,err) 831d102b48SJeremy L Thompson 841d102b48SJeremy L Thompson! Bases 851d102b48SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,d,p,q,ceed_gauss,bx,err) 861d102b48SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,1,p,q,ceed_gauss,bu,err) 871d102b48SJeremy L Thompson 881d102b48SJeremy L Thompson! QFunction - setup mass 891d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,setup_mass,& 901d102b48SJeremy L Thompson &SOURCE_DIR& 911d102b48SJeremy L Thompson &//'t532-operator.h:setup_mass'//char(0),qf_setup_mass,err) 921d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_mass,'dx',d*d,ceed_eval_grad,err) 93a61c78d6SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_mass,'weight',1,ceed_eval_weight,err) 941d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_setup_mass,'qdata',1,ceed_eval_none,err) 951d102b48SJeremy L Thompson 961d102b48SJeremy L Thompson! Operator - setup mass 97442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_setup_mass,ceed_qfunction_none,& 98442e7f0bSjeremylt & ceed_qfunction_none,op_setup_mass,err) 991d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup_mass,'dx',erestrictx,& 100a8d32208Sjeremylt & bx,ceed_vector_active,err) 101a61c78d6SJeremy L Thompson call ceedoperatorsetfield(op_setup_mass,'weight',& 10215910d16Sjeremylt & ceed_elemrestriction_none,bx,ceed_vector_none,err) 1031d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup_mass,'qdata',erestrictui,& 104*356036faSJeremy L Thompson ceed_basis_none,ceed_vector_active,err) 1051d102b48SJeremy L Thompson 1061d102b48SJeremy L Thompson! QFunction - setup diff 1071d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,setup_diff,& 1081d102b48SJeremy L Thompson &SOURCE_DIR& 1091d102b48SJeremy L Thompson &//'t532-operator.h:setup_diff'//char(0),qf_setup_diff,err) 1101d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_diff,'dx',d*d,ceed_eval_grad,err) 111a61c78d6SJeremy L Thompson call ceedqfunctionaddinput(qf_setup_diff,'weight',1,ceed_eval_weight,err) 1121d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_setup_diff,'qdata',& 1131d102b48SJeremy L Thompson & d*(d+1)/2,ceed_eval_none,err) 1141d102b48SJeremy L Thompson 1151d102b48SJeremy L Thompson! Operator - setup diff 116442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_setup_diff,ceed_qfunction_none,& 117442e7f0bSjeremylt & ceed_qfunction_none,op_setup_diff,err) 1181d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup_diff,'dx',erestrictx,& 119a8d32208Sjeremylt & bx,ceed_vector_active,err) 120a61c78d6SJeremy L Thompson call ceedoperatorsetfield(op_setup_diff,'weight',& 12115910d16Sjeremylt & ceed_elemrestriction_none,bx,ceed_vector_none,err) 1221d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup_diff,'qdata',erestrictqi,& 123*356036faSJeremy L Thompson ceed_basis_none,ceed_vector_active,err) 1241d102b48SJeremy L Thompson 1251d102b48SJeremy L Thompson! Apply Setup Operators 1261d102b48SJeremy L Thompson call ceedoperatorapply(op_setup_mass,x,qdata_mass,& 1271d102b48SJeremy L Thompson & ceed_request_immediate,err) 1281d102b48SJeremy L Thompson call ceedoperatorapply(op_setup_diff,x,qdata_diff,& 1291d102b48SJeremy L Thompson & ceed_request_immediate,err) 1301d102b48SJeremy L Thompson 1311d102b48SJeremy L Thompson! QFunction - apply 1321d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,apply,& 1331d102b48SJeremy L Thompson &SOURCE_DIR& 1341d102b48SJeremy L Thompson &//'t532-operator.h:apply'//char(0),qf_apply,err) 1351d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'du',d,ceed_eval_grad,err) 136a61c78d6SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'mass qdata',1,ceed_eval_none,err) 137a61c78d6SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'diff qdata',& 1381d102b48SJeremy L Thompson & d*(d+1)/2,ceed_eval_none,err) 1391d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_apply,'u',1,ceed_eval_interp,err) 1401d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_apply,'v',1,ceed_eval_interp,err) 1411d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_apply,'dv',d,ceed_eval_grad,err) 1421d102b48SJeremy L Thompson 1431d102b48SJeremy L Thompson! Operator - apply 144442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_apply,ceed_qfunction_none,& 145442e7f0bSjeremylt & ceed_qfunction_none,op_apply,err) 1461d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply,'du',erestrictu,& 147a8d32208Sjeremylt & bu,ceed_vector_active,err) 148a61c78d6SJeremy L Thompson call ceedoperatorsetfield(op_apply,'mass qdata',erestrictui,& 149*356036faSJeremy L Thompson ceed_basis_none,qdata_mass,err) 150a61c78d6SJeremy L Thompson call ceedoperatorsetfield(op_apply,'diff qdata',erestrictqi,& 151*356036faSJeremy L Thompson ceed_basis_none,qdata_diff,err) 1521d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply,'u',erestrictu,& 153a8d32208Sjeremylt & bu,ceed_vector_active,err) 1541d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply,'v',erestrictu,& 155a8d32208Sjeremylt & bu,ceed_vector_active,err) 1561d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply,'dv',erestrictu,& 157a8d32208Sjeremylt & bu,ceed_vector_active,err) 1581d102b48SJeremy L Thompson 1591d102b48SJeremy L Thompson! Apply Original Operator 1601d102b48SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,u,err) 1611d102b48SJeremy L Thompson call ceedvectorsetvalue(u,1.d0,err) 1621d102b48SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,v,err) 1631d102b48SJeremy L Thompson call ceedvectorsetvalue(v,0.d0,err) 1641d102b48SJeremy L Thompson call ceedoperatorapply(op_apply,u,v,ceed_request_immediate,err) 1651d102b48SJeremy L Thompson 1661d102b48SJeremy L Thompson! Check Output 1671d102b48SJeremy L Thompson call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err) 1681d102b48SJeremy L Thompson total=0. 1691d102b48SJeremy L Thompson do i=1,ndofs 1701d102b48SJeremy L Thompson total=total+vv(voffset+i) 1711d102b48SJeremy L Thompson enddo 1721d102b48SJeremy L Thompson if (abs(total-1.)>1.0d-10) then 1731d102b48SJeremy L Thompson! LCOV_EXCL_START 1741d102b48SJeremy L Thompson write(*,*) 'Error: True operator computed area = ',total,' != 1.0' 1751d102b48SJeremy L Thompson! LCOV_EXCL_STOP 1761d102b48SJeremy L Thompson endif 1771d102b48SJeremy L Thompson call ceedvectorrestorearrayread(v,vv,voffset,err) 1781d102b48SJeremy L Thompson 1791d102b48SJeremy L Thompson! Assemble QFunction 18080ac2e43SJeremy L Thompson call ceedoperatorlinearassembleqfunction(op_apply,a,erestrictlini,& 1811d102b48SJeremy L Thompson & ceed_request_immediate,err) 1821d102b48SJeremy L Thompson 1831d102b48SJeremy L Thompson! QFunction - apply linearized 1841d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,apply_lin,& 1851d102b48SJeremy L Thompson &SOURCE_DIR& 1861d102b48SJeremy L Thompson &//'t532-operator.h:apply_lin'//char(0),qf_apply_lin,err) 1871d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_apply_lin,'du',d,ceed_eval_grad,err) 1881d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_apply_lin,'qdata',(d+1)*(d+1),& 1891d102b48SJeremy L Thompson & ceed_eval_none,err) 1901d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_apply_lin,'u',1,ceed_eval_interp,err) 1911d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_apply_lin,'v',1,ceed_eval_interp,err) 1921d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_apply_lin,'dv',d,ceed_eval_grad,err) 1931d102b48SJeremy L Thompson 1941d102b48SJeremy L Thompson! Operator - apply linearized 195442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_apply_lin,ceed_qfunction_none,& 196442e7f0bSjeremylt & ceed_qfunction_none,op_apply_lin,err) 1971d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply_lin,'du',erestrictu,& 198a8d32208Sjeremylt & bu,ceed_vector_active,err) 1991d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply_lin,'qdata',erestrictlini,& 200*356036faSJeremy L Thompson ceed_basis_none,a,err) 2011d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply_lin,'u',erestrictu,& 202a8d32208Sjeremylt & bu,ceed_vector_active,err) 2031d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply_lin,'v',erestrictu,& 204a8d32208Sjeremylt & bu,ceed_vector_active,err) 2051d102b48SJeremy L Thompson call ceedoperatorsetfield(op_apply_lin,'dv',erestrictu,& 206a8d32208Sjeremylt & bu,ceed_vector_active,err) 2071d102b48SJeremy L Thompson 2081d102b48SJeremy L Thompson! Apply Linearized QFunction Operator 2091d102b48SJeremy L Thompson call ceedvectorsetvalue(v,0.d0,err) 2101d102b48SJeremy L Thompson call ceedoperatorapply(op_apply_lin,u,v,ceed_request_immediate,err) 2111d102b48SJeremy L Thompson 2121d102b48SJeremy L Thompson! Check Output 2131d102b48SJeremy L Thompson call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err) 2141d102b48SJeremy L Thompson total=0. 2151d102b48SJeremy L Thompson do i=1,ndofs 2161d102b48SJeremy L Thompson total=total+vv(voffset+i) 2171d102b48SJeremy L Thompson enddo 2181d102b48SJeremy L Thompson if (abs(total-1.)>1.0d-10) then 2191d102b48SJeremy L Thompson! LCOV_EXCL_START 2201d102b48SJeremy L Thompson write(*,*) 'Error: Assembled operator computed area = ',total,' != 1.0' 2211d102b48SJeremy L Thompson! LCOV_EXCL_STOP 2221d102b48SJeremy L Thompson endif 2231d102b48SJeremy L Thompson call ceedvectorrestorearrayread(v,vv,voffset,err) 2241d102b48SJeremy L Thompson 2251d102b48SJeremy L Thompson! Cleanup 2261d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_setup_mass,err) 2271d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_setup_diff,err) 2281d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_apply,err) 2291d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_apply_lin,err) 2301d102b48SJeremy L Thompson call ceedoperatordestroy(op_setup_mass,err) 2311d102b48SJeremy L Thompson call ceedoperatordestroy(op_setup_diff,err) 2321d102b48SJeremy L Thompson call ceedoperatordestroy(op_apply,err) 2331d102b48SJeremy L Thompson call ceedoperatordestroy(op_apply_lin,err) 2341d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictu,err) 2351d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictx,err) 2361d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictui,err) 2371d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictqi,err) 2381d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictlini,err) 2391d102b48SJeremy L Thompson call ceedbasisdestroy(bu,err) 2401d102b48SJeremy L Thompson call ceedbasisdestroy(bx,err) 2411d102b48SJeremy L Thompson call ceedvectordestroy(x,err) 2421d102b48SJeremy L Thompson call ceedvectordestroy(a,err) 2431d102b48SJeremy L Thompson call ceedvectordestroy(u,err) 2441d102b48SJeremy L Thompson call ceedvectordestroy(v,err) 2451d102b48SJeremy L Thompson call ceedvectordestroy(qdata_mass,err) 2461d102b48SJeremy L Thompson call ceedvectordestroy(qdata_diff,err) 2471d102b48SJeremy L Thompson call ceeddestroy(ceed,err) 2481d102b48SJeremy L Thompson end 2491d102b48SJeremy L Thompson!----------------------------------------------------------------------- 250