11d102b48SJeremy L Thompson!----------------------------------------------------------------------- 2752c3701SJeremy L Thompson! 3752c3701SJeremy L Thompson! Header with QFunctions 4752c3701SJeremy L Thompson! 5752c3701SJeremy L Thompson include 't531-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,qf_diff,qf_diff_lin 171d102b48SJeremy L Thompson integer op_setup,op_diff,op_diff_lin 18*68ebe796SJeremy L Thompson integer qdata,x,a,u,v,v_lin 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) 31*68ebe796SJeremy L Thompson real*8 arrx(d*ndofs),uu(ndofs),vv(ndofs),vvlin(ndofs) 32*68ebe796SJeremy L Thompson integer*8 xoffset,uoffset,voffset,vlinoffset 331d102b48SJeremy L Thompson 341d102b48SJeremy L Thompson character arg*32 351d102b48SJeremy L Thompson 361d102b48SJeremy L Thompson external setup,diff,diff_lin 371d102b48SJeremy L Thompson 381d102b48SJeremy L Thompson call getarg(1,arg) 391d102b48SJeremy L Thompson 401d102b48SJeremy L Thompson call ceedinit(trim(arg)//char(0),ceed,err) 411d102b48SJeremy L Thompson 421d102b48SJeremy L Thompson! DoF Coordinates 431d102b48SJeremy L Thompson do i=0,nx*2 441d102b48SJeremy L Thompson do j=0,ny*2 45*68ebe796SJeremy L Thompson arrx(i+j*(nx*2+1)+0*ndofs+1)=1.d0*i/(2*nx)+j*0.5 46*68ebe796SJeremy L Thompson arrx(i+j*(nx*2+1)+1*ndofs+1)=1.d0*j/(2*ny)+i*0.5 471d102b48SJeremy L Thompson enddo 481d102b48SJeremy L Thompson enddo 491d102b48SJeremy L Thompson call ceedvectorcreate(ceed,d*ndofs,x,err) 501d102b48SJeremy L Thompson xoffset=0 511d102b48SJeremy L Thompson call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 521d102b48SJeremy L Thompson 53*68ebe796SJeremy L Thompson! Input, output arrays 54*68ebe796SJeremy L Thompson do i=0,nx*2 55*68ebe796SJeremy L Thompson do j=0,ny*2 56*68ebe796SJeremy L Thompson uu(i+j*(nx*2+1)+1)=i*nx+j*ny 57*68ebe796SJeremy L Thompson enddo 58*68ebe796SJeremy L Thompson enddo 59*68ebe796SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,u,err) 60*68ebe796SJeremy L Thompson uoffset=0 61*68ebe796SJeremy L Thompson call ceedvectorsetarray(u,ceed_mem_host,ceed_use_pointer,uu,uoffset,err) 62*68ebe796SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,v,err) 63*68ebe796SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,v_lin,err) 64*68ebe796SJeremy L Thompson 651d102b48SJeremy L Thompson! Qdata Vector 661d102b48SJeremy L Thompson call ceedvectorcreate(ceed,nqpts*d*(d+1)/2,qdata,err) 671d102b48SJeremy L Thompson 681d102b48SJeremy L Thompson! Element Setup 691d102b48SJeremy L Thompson do i=0,nelem-1 701d102b48SJeremy L Thompson col=mod(i,nx) 711d102b48SJeremy L Thompson row=i/nx 721d102b48SJeremy L Thompson offset=col*(p-1)+row*(nx*2+1)*(p-1) 731d102b48SJeremy L Thompson do j=0,p-1 741d102b48SJeremy L Thompson do k=0,p-1 751d102b48SJeremy L Thompson indx(p*(p*i+k)+j+1)=offset+k*(nx*2+1)+j 761d102b48SJeremy L Thompson enddo 771d102b48SJeremy L Thompson enddo 781d102b48SJeremy L Thompson enddo 791d102b48SJeremy L Thompson 801d102b48SJeremy L Thompson! Restrictions 81d979a051Sjeremylt call ceedelemrestrictioncreate(ceed,nelem,p*p,d,ndofs,d*ndofs,& 821d102b48SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictx,err) 831d102b48SJeremy L Thompson 84d979a051Sjeremylt call ceedelemrestrictioncreate(ceed,nelem,p*p,1,1,ndofs,& 851d102b48SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictu,err) 867509a596Sjeremylt stridesu=[1,q*q,q*q] 87d979a051Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,q*q,1,nqpts,& 88d979a051Sjeremylt & stridesu,erestrictui,err) 891d102b48SJeremy L Thompson 907509a596Sjeremylt stridesqd=[1,q*q,q*q*d*(d+1)/2] 91d979a051Sjeremylt call ceedelemrestrictioncreatestrided(ceed,nelem,q*q,d*(d+1)/2,& 92d979a051Sjeremylt & d*(d+1)/2*nqpts,stridesqd,erestrictqi,err) 931d102b48SJeremy L Thompson 941d102b48SJeremy L Thompson! Bases 951d102b48SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,d,p,q,ceed_gauss,& 961d102b48SJeremy L Thompson & bx,err) 971d102b48SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,1,p,q,ceed_gauss,& 981d102b48SJeremy L Thompson & bu,err) 991d102b48SJeremy L Thompson 1001d102b48SJeremy L Thompson! QFunction - setup 1011d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,setup,& 1021d102b48SJeremy L Thompson &SOURCE_DIR& 1031d102b48SJeremy L Thompson &//'t531-operator.h:setup'//char(0),qf_setup,err) 1041d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_setup,'dx',d*d,ceed_eval_grad,err) 105a61c78d6SJeremy L Thompson call ceedqfunctionaddinput(qf_setup,'weight',1,ceed_eval_weight,err) 1061d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_setup,'qdata',d*(d+1)/2,ceed_eval_none,err) 1071d102b48SJeremy L Thompson 1081d102b48SJeremy L Thompson! Operator - setup 109442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,& 110442e7f0bSjeremylt & ceed_qfunction_none,op_setup,err) 1111d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup,'dx',erestrictx,& 112a8d32208Sjeremylt & bx,ceed_vector_active,err) 113a61c78d6SJeremy L Thompson call ceedoperatorsetfield(op_setup,'weight',ceed_elemrestriction_none,& 114a8d32208Sjeremylt & bx,ceed_vector_none,err) 1151d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup,'qdata',erestrictqi,& 116356036faSJeremy L Thompson ceed_basis_none,ceed_vector_active,err) 1171d102b48SJeremy L Thompson 1181d102b48SJeremy L Thompson! Apply Setup Operator 1191d102b48SJeremy L Thompson call ceedoperatorapply(op_setup,x,qdata,ceed_request_immediate,err) 1201d102b48SJeremy L Thompson 1211d102b48SJeremy L Thompson! QFunction - apply 1221d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,diff,& 1231d102b48SJeremy L Thompson &SOURCE_DIR& 1241d102b48SJeremy L Thompson &//'t531-operator.h:diff'//char(0),qf_diff,err) 1251d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_diff,'du',d,ceed_eval_grad,err) 1261d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_diff,'qdata',d*(d+1)/2,ceed_eval_none,err) 1271d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_diff,'dv',d,ceed_eval_grad,err) 1281d102b48SJeremy L Thompson 1291d102b48SJeremy L Thompson! Operator - apply 130442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_diff,ceed_qfunction_none,& 131442e7f0bSjeremylt & ceed_qfunction_none,op_diff,err) 1321d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff,'du',erestrictu,& 133a8d32208Sjeremylt & bu,ceed_vector_active,err) 1341d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff,'qdata',erestrictqi,& 135356036faSJeremy L Thompson ceed_basis_none,qdata,err) 1361d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff,'dv',erestrictu,& 137a8d32208Sjeremylt & bu,ceed_vector_active,err) 1381d102b48SJeremy L Thompson 1391d102b48SJeremy L Thompson! Apply original Poisson Operator 1401d102b48SJeremy L Thompson call ceedoperatorapply(op_diff,u,v,ceed_request_immediate,err) 1411d102b48SJeremy L Thompson 1421d102b48SJeremy L Thompson! Assemble QFunction 14380ac2e43SJeremy L Thompson call ceedoperatorlinearassembleqfunction(op_diff,a,erestrictlini,& 1441d102b48SJeremy L Thompson & ceed_request_immediate,err) 1451d102b48SJeremy L Thompson 1461d102b48SJeremy L Thompson! QFunction - apply linearized 1471d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,diff_lin,& 1481d102b48SJeremy L Thompson &SOURCE_DIR& 1491d102b48SJeremy L Thompson &//'t531-operator.h:diff_lin'//char(0),qf_diff_lin,err) 1501d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_diff_lin,'du',d,ceed_eval_grad,err) 1511d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_diff_lin,'qdata',d*d,ceed_eval_none,err) 1521d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_diff_lin,'dv',d,ceed_eval_grad,err) 1531d102b48SJeremy L Thompson 1541d102b48SJeremy L Thompson! Operator - apply linearized 155442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_diff_lin,ceed_qfunction_none,& 156442e7f0bSjeremylt & ceed_qfunction_none,op_diff_lin,err) 1571d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff_lin,'du',erestrictu,& 158a8d32208Sjeremylt & bu,ceed_vector_active,err) 1591d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff_lin,'qdata',erestrictlini,& 160356036faSJeremy L Thompson ceed_basis_none,a,err) 1611d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff_lin,'dv',erestrictu,& 162a8d32208Sjeremylt & bu,ceed_vector_active,err) 1631d102b48SJeremy L Thompson 1641d102b48SJeremy L Thompson! Apply linearized Poisson Operator 165*68ebe796SJeremy L Thompson call ceedoperatorapply(op_diff_lin,u,v_lin,ceed_request_immediate,err) 1661d102b48SJeremy L Thompson 1671d102b48SJeremy L Thompson! Check Output 1681d102b48SJeremy L Thompson call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err) 169*68ebe796SJeremy L Thompson call ceedvectorgetarrayread(v_lin,ceed_mem_host,vvlin,vlinoffset,err) 1701d102b48SJeremy L Thompson do i=1,ndofs 171*68ebe796SJeremy L Thompson if (abs(vv(voffset+i)-vvlin(vlinoffset+i))>1.0d-14) then 1721d102b48SJeremy L Thompson! LCOV_EXCL_START 1731d102b48SJeremy L Thompson write(*,*) 'Error: Linearized operator computed v[i] = ',vv(voffset+i),& 174*68ebe796SJeremy L Thompson & ' != ',vvlin(vlinoffset+i) 1751d102b48SJeremy L Thompson! LCOV_EXCL_STOP 1761d102b48SJeremy L Thompson endif 1771d102b48SJeremy L Thompson enddo 1781d102b48SJeremy L Thompson call ceedvectorrestorearrayread(v,vv,voffset,err) 179*68ebe796SJeremy L Thompson call ceedvectorrestorearrayread(v_lin,vvlin,vlinoffset,err) 1801d102b48SJeremy L Thompson 1811d102b48SJeremy L Thompson! Cleanup 1821d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_setup,err) 1831d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_diff,err) 1841d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_diff_lin,err) 1851d102b48SJeremy L Thompson call ceedoperatordestroy(op_setup,err) 1861d102b48SJeremy L Thompson call ceedoperatordestroy(op_diff,err) 1871d102b48SJeremy L Thompson call ceedoperatordestroy(op_diff_lin,err) 1881d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictu,err) 1891d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictx,err) 1901d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictui,err) 1911d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictqi,err) 1921d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictlini,err) 1931d102b48SJeremy L Thompson call ceedbasisdestroy(bu,err) 1941d102b48SJeremy L Thompson call ceedbasisdestroy(bx,err) 1951d102b48SJeremy L Thompson call ceedvectordestroy(x,err) 1961d102b48SJeremy L Thompson call ceedvectordestroy(a,err) 1971d102b48SJeremy L Thompson call ceedvectordestroy(u,err) 1981d102b48SJeremy L Thompson call ceedvectordestroy(v,err) 199*68ebe796SJeremy L Thompson call ceedvectordestroy(v_lin,err) 2001d102b48SJeremy L Thompson call ceedvectordestroy(qdata,err) 2011d102b48SJeremy L Thompson call ceeddestroy(ceed,err) 2021d102b48SJeremy L Thompson end 2031d102b48SJeremy L Thompson!----------------------------------------------------------------------- 204