11d102b48SJeremy L Thompson!----------------------------------------------------------------------- 21d102b48SJeremy L Thompson subroutine setup(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,& 31d102b48SJeremy L Thompson& u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr) 41d102b48SJeremy L Thompson real*8 ctx 51d102b48SJeremy L Thompson real*8 u1(1) 61d102b48SJeremy L Thompson real*8 u2(1) 71d102b48SJeremy L Thompson real*8 v1(1) 81d102b48SJeremy L Thompson real*8 w 91d102b48SJeremy L Thompson integer q,ierr 101d102b48SJeremy L Thompson 111d102b48SJeremy L Thompson do i=1,q 121d102b48SJeremy L Thompson w=u2(i)/(u1(i+q*0)*u1(i+q*3)-u1(i+q*1)*u1(i+q*2)) 131d102b48SJeremy L Thompson v1(i+q*0)=w*(u1(i+q*2)*u1(i+q*2)+u1(i+q*3)*u1(i+q*3)) 141d102b48SJeremy L Thompson v1(i+q*1)=-w*(u1(i+q*0)*u1(i+q*2)+u1(i+q*2)*u1(i+q*3)) 151d102b48SJeremy L Thompson v1(i+q*2)=w*(u1(i+q*0)*u1(i+q*0)+u1(i+q*1)*u1(i+q*1)) 161d102b48SJeremy L Thompson enddo 171d102b48SJeremy L Thompson 181d102b48SJeremy L Thompson ierr=0 191d102b48SJeremy L Thompson end 201d102b48SJeremy L Thompson!----------------------------------------------------------------------- 211d102b48SJeremy L Thompson subroutine diff(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,& 221d102b48SJeremy L Thompson& u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr) 231d102b48SJeremy L Thompson real*8 ctx 241d102b48SJeremy L Thompson real*8 u1(1) 251d102b48SJeremy L Thompson real*8 u2(1) 261d102b48SJeremy L Thompson real*8 v1(1) 271d102b48SJeremy L Thompson real*8 du0,du1 281d102b48SJeremy L Thompson integer q,ierr 291d102b48SJeremy L Thompson 301d102b48SJeremy L Thompson do i=1,q 311d102b48SJeremy L Thompson du0=u1(i+q*0) 321d102b48SJeremy L Thompson du1=u1(i+q*1) 331d102b48SJeremy L Thompson v1(i+q*0)=u2(i+q*0)*du0+u2(i+q*1)*du1 341d102b48SJeremy L Thompson v1(i+q*1)=u2(i+q*1)*du0+u2(i+q*2)*du1 351d102b48SJeremy L Thompson enddo 361d102b48SJeremy L Thompson 371d102b48SJeremy L Thompson ierr=0 381d102b48SJeremy L Thompson end 391d102b48SJeremy L Thompson!----------------------------------------------------------------------- 401d102b48SJeremy L Thompson subroutine diff_lin(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,& 411d102b48SJeremy L Thompson& u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr) 421d102b48SJeremy L Thompson real*8 ctx 431d102b48SJeremy L Thompson real*8 u1(1) 441d102b48SJeremy L Thompson real*8 u2(1) 451d102b48SJeremy L Thompson real*8 v1(1) 461d102b48SJeremy L Thompson real*8 du0,du1 471d102b48SJeremy L Thompson integer q,ierr 481d102b48SJeremy L Thompson 491d102b48SJeremy L Thompson do i=1,q 501d102b48SJeremy L Thompson du0=u1(i+q*0) 511d102b48SJeremy L Thompson du1=u1(i+q*1) 521d102b48SJeremy L Thompson v1(i+q*0)=u2(i+q*0)*du0+u2(i+q*1)*du1 531d102b48SJeremy L Thompson v1(i+q*1)=u2(i+q*2)*du0+u2(i+q*3)*du1 541d102b48SJeremy L Thompson enddo 551d102b48SJeremy L Thompson 561d102b48SJeremy L Thompson ierr=0 571d102b48SJeremy L Thompson end 581d102b48SJeremy L Thompson!----------------------------------------------------------------------- 591d102b48SJeremy L Thompson program test 601d102b48SJeremy L Thompson 611d102b48SJeremy L Thompson include 'ceedf.h' 621d102b48SJeremy L Thompson 631d102b48SJeremy L Thompson integer ceed,err,i,j,k 641d102b48SJeremy L Thompson integer erestrictx,erestrictu,erestrictxi,erestrictui 651d102b48SJeremy L Thompson integer erestrictqi,erestrictlini 661d102b48SJeremy L Thompson integer bx,bu 671d102b48SJeremy L Thompson integer qf_setup,qf_diff,qf_diff_lin 681d102b48SJeremy L Thompson integer op_setup,op_diff,op_diff_lin 691d102b48SJeremy L Thompson integer qdata,x,a,u,v 701d102b48SJeremy L Thompson integer nelem,p,q,d 711d102b48SJeremy L Thompson integer row,col,offset 721d102b48SJeremy L Thompson parameter(nelem=6) 731d102b48SJeremy L Thompson parameter(p=3) 741d102b48SJeremy L Thompson parameter(q=4) 751d102b48SJeremy L Thompson parameter(d=2) 761d102b48SJeremy L Thompson integer ndofs,nqpts,nx,ny 771d102b48SJeremy L Thompson parameter(nx=3) 781d102b48SJeremy L Thompson parameter(ny=2) 791d102b48SJeremy L Thompson parameter(ndofs=(nx*2+1)*(ny*2+1)) 801d102b48SJeremy L Thompson parameter(nqpts=nelem*q*q) 811d102b48SJeremy L Thompson integer indx(nelem*p*p) 821d102b48SJeremy L Thompson real*8 arrx(d*ndofs),vv(ndofs) 831d102b48SJeremy L Thompson integer*8 xoffset,voffset 841d102b48SJeremy L Thompson 851d102b48SJeremy L Thompson character arg*32 861d102b48SJeremy L Thompson 871d102b48SJeremy L Thompson external setup,diff,diff_lin 881d102b48SJeremy L Thompson 891d102b48SJeremy L Thompson call getarg(1,arg) 901d102b48SJeremy L Thompson 911d102b48SJeremy L Thompson call ceedinit(trim(arg)//char(0),ceed,err) 921d102b48SJeremy L Thompson 931d102b48SJeremy L Thompson! DoF Coordinates 941d102b48SJeremy L Thompson do i=0,nx*2 951d102b48SJeremy L Thompson do j=0,ny*2 961d102b48SJeremy L Thompson arrx(i+j*(nx*2+1)+0*ndofs+1)=1.d0*i/(2*nx) 971d102b48SJeremy L Thompson arrx(i+j*(nx*2+1)+1*ndofs+1)=1.d0*j/(2*ny) 981d102b48SJeremy L Thompson enddo 991d102b48SJeremy L Thompson enddo 1001d102b48SJeremy L Thompson call ceedvectorcreate(ceed,d*ndofs,x,err) 1011d102b48SJeremy L Thompson xoffset=0 1021d102b48SJeremy L Thompson call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 1031d102b48SJeremy L Thompson 1041d102b48SJeremy L Thompson! Qdata Vector 1051d102b48SJeremy L Thompson call ceedvectorcreate(ceed,nqpts*d*(d+1)/2,qdata,err) 1061d102b48SJeremy L Thompson 1071d102b48SJeremy L Thompson! Element Setup 1081d102b48SJeremy L Thompson do i=0,nelem-1 1091d102b48SJeremy L Thompson col=mod(i,nx) 1101d102b48SJeremy L Thompson row=i/nx 1111d102b48SJeremy L Thompson offset=col*(p-1)+row*(nx*2+1)*(p-1) 1121d102b48SJeremy L Thompson do j=0,p-1 1131d102b48SJeremy L Thompson do k=0,p-1 1141d102b48SJeremy L Thompson indx(p*(p*i+k)+j+1)=offset+k*(nx*2+1)+j 1151d102b48SJeremy L Thompson enddo 1161d102b48SJeremy L Thompson enddo 1171d102b48SJeremy L Thompson enddo 1181d102b48SJeremy L Thompson 1191d102b48SJeremy L Thompson! Restrictions 1201d102b48SJeremy L Thompson call ceedelemrestrictioncreate(ceed,nelem,p*p,ndofs,d,& 1211d102b48SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictx,err) 1221d102b48SJeremy L Thompson call ceedelemrestrictioncreateidentity(ceed,nelem,p*p,& 1231d102b48SJeremy L Thompson & nelem*p*p,d,erestrictxi,err) 1241d102b48SJeremy L Thompson 1251d102b48SJeremy L Thompson call ceedelemrestrictioncreate(ceed,nelem,p*p,ndofs,1,& 1261d102b48SJeremy L Thompson & ceed_mem_host,ceed_use_pointer,indx,erestrictu,err) 1271d102b48SJeremy L Thompson call ceedelemrestrictioncreateidentity(ceed,nelem,q*q,nqpts,& 1281d102b48SJeremy L Thompson & 1,erestrictui,err) 1291d102b48SJeremy L Thompson 1301d102b48SJeremy L Thompson call ceedelemrestrictioncreateidentity(ceed,nelem,q*q,nqpts,& 1311d102b48SJeremy L Thompson & d*(d+1)/2,erestrictqi,err) 1321d102b48SJeremy L Thompson 1331d102b48SJeremy L Thompson! Bases 1341d102b48SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,d,p,q,ceed_gauss,& 1351d102b48SJeremy L Thompson & bx,err) 1361d102b48SJeremy L Thompson call ceedbasiscreatetensorh1lagrange(ceed,d,1,p,q,ceed_gauss,& 1371d102b48SJeremy L Thompson & bu,err) 1381d102b48SJeremy L Thompson 1391d102b48SJeremy L Thompson! QFunction - setup 1401d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,setup,& 1411d102b48SJeremy L Thompson &SOURCE_DIR& 1421d102b48SJeremy L Thompson &//'t531-operator.h:setup'//char(0),qf_setup,err) 1431d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_setup,'dx',d*d,ceed_eval_grad,err) 1441d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_setup,'_weight',1,ceed_eval_weight,err) 1451d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_setup,'qdata',d*(d+1)/2,ceed_eval_none,err) 1461d102b48SJeremy L Thompson 1471d102b48SJeremy L Thompson! Operator - setup 148*442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,& 149*442e7f0bSjeremylt & ceed_qfunction_none,op_setup,err) 1501d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup,'dx',erestrictx,& 1511d102b48SJeremy L Thompson & ceed_notranspose,bx,ceed_vector_active,err) 1521d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup,'_weight',erestrictxi,& 1531d102b48SJeremy L Thompson & ceed_notranspose,bx,ceed_vector_none,err) 1541d102b48SJeremy L Thompson call ceedoperatorsetfield(op_setup,'qdata',erestrictqi,& 1551d102b48SJeremy L Thompson & ceed_notranspose,ceed_basis_collocated,ceed_vector_active,err) 1561d102b48SJeremy L Thompson 1571d102b48SJeremy L Thompson! Apply Setup Operator 1581d102b48SJeremy L Thompson call ceedoperatorapply(op_setup,x,qdata,ceed_request_immediate,err) 1591d102b48SJeremy L Thompson 1601d102b48SJeremy L Thompson! QFunction - apply 1611d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,diff,& 1621d102b48SJeremy L Thompson &SOURCE_DIR& 1631d102b48SJeremy L Thompson &//'t531-operator.h:diff'//char(0),qf_diff,err) 1641d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_diff,'du',d,ceed_eval_grad,err) 1651d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_diff,'qdata',d*(d+1)/2,ceed_eval_none,err) 1661d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_diff,'dv',d,ceed_eval_grad,err) 1671d102b48SJeremy L Thompson 1681d102b48SJeremy L Thompson! Operator - apply 169*442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_diff,ceed_qfunction_none,& 170*442e7f0bSjeremylt & ceed_qfunction_none,op_diff,err) 1711d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff,'du',erestrictu,& 1721d102b48SJeremy L Thompson & ceed_notranspose,bu,ceed_vector_active,err) 1731d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff,'qdata',erestrictqi,& 1741d102b48SJeremy L Thompson & ceed_notranspose,ceed_basis_collocated,qdata,err) 1751d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff,'dv',erestrictu,& 1761d102b48SJeremy L Thompson & ceed_notranspose,bu,ceed_vector_active,err) 1771d102b48SJeremy L Thompson 1781d102b48SJeremy L Thompson! Apply original Poisson Operator 1791d102b48SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,u,err) 1801d102b48SJeremy L Thompson call ceedvectorsetvalue(u,1.d0,err) 1811d102b48SJeremy L Thompson call ceedvectorcreate(ceed,ndofs,v,err) 1821d102b48SJeremy L Thompson call ceedvectorsetvalue(v,0.d0,err) 1831d102b48SJeremy L Thompson call ceedoperatorapply(op_diff,u,v,ceed_request_immediate,err) 1841d102b48SJeremy L Thompson 1851d102b48SJeremy L Thompson! Check Output 1861d102b48SJeremy L Thompson call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err) 1871d102b48SJeremy L Thompson do i=1,ndofs 1881d102b48SJeremy L Thompson if (abs(vv(voffset+i))>1.0d-14) then 1891d102b48SJeremy L Thompson! LCOV_EXCL_START 1901d102b48SJeremy L Thompson write(*,*) 'Error: Operator computed v[i] = ',vv(voffset+i),' != 0.0' 1911d102b48SJeremy L Thompson! LCOV_EXCL_STOP 1921d102b48SJeremy L Thompson endif 1931d102b48SJeremy L Thompson enddo 1941d102b48SJeremy L Thompson call ceedvectorrestorearrayread(v,vv,voffset,err) 1951d102b48SJeremy L Thompson 1961d102b48SJeremy L Thompson! Assemble QFunction 1971d102b48SJeremy L Thompson call ceedoperatorassemblelinearqfunction(op_diff,a,erestrictlini,& 1981d102b48SJeremy L Thompson & ceed_request_immediate,err) 1991d102b48SJeremy L Thompson 2001d102b48SJeremy L Thompson! QFunction - apply linearized 2011d102b48SJeremy L Thompson call ceedqfunctioncreateinterior(ceed,1,diff_lin,& 2021d102b48SJeremy L Thompson &SOURCE_DIR& 2031d102b48SJeremy L Thompson &//'t531-operator.h:diff_lin'//char(0),qf_diff_lin,err) 2041d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_diff_lin,'du',d,ceed_eval_grad,err) 2051d102b48SJeremy L Thompson call ceedqfunctionaddinput(qf_diff_lin,'qdata',d*d,ceed_eval_none,err) 2061d102b48SJeremy L Thompson call ceedqfunctionaddoutput(qf_diff_lin,'dv',d,ceed_eval_grad,err) 2071d102b48SJeremy L Thompson 2081d102b48SJeremy L Thompson! Operator - apply linearized 209*442e7f0bSjeremylt call ceedoperatorcreate(ceed,qf_diff_lin,ceed_qfunction_none,& 210*442e7f0bSjeremylt & ceed_qfunction_none,op_diff_lin,err) 2111d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff_lin,'du',erestrictu,& 2121d102b48SJeremy L Thompson & ceed_notranspose,bu,ceed_vector_active,err) 2131d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff_lin,'qdata',erestrictlini,& 2141d102b48SJeremy L Thompson & ceed_notranspose,ceed_basis_collocated,a,err) 2151d102b48SJeremy L Thompson call ceedoperatorsetfield(op_diff_lin,'dv',erestrictu,& 2161d102b48SJeremy L Thompson & ceed_notranspose,bu,ceed_vector_active,err) 2171d102b48SJeremy L Thompson 2181d102b48SJeremy L Thompson! Apply linearized Poisson Operator 2191d102b48SJeremy L Thompson call ceedvectorsetvalue(v,0.d0,err) 2201d102b48SJeremy L Thompson call ceedoperatorapply(op_diff_lin,u,v,ceed_request_immediate,err) 2211d102b48SJeremy L Thompson 2221d102b48SJeremy L Thompson! Check Output 2231d102b48SJeremy L Thompson call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err) 2241d102b48SJeremy L Thompson do i=1,ndofs 2251d102b48SJeremy L Thompson if (abs(vv(voffset+i))>1.0d-14) then 2261d102b48SJeremy L Thompson! LCOV_EXCL_START 2271d102b48SJeremy L Thompson write(*,*) 'Error: Linearized operator computed v[i] = ',vv(voffset+i),& 2281d102b48SJeremy L Thompson & ' != 0.0' 2291d102b48SJeremy L Thompson! LCOV_EXCL_STOP 2301d102b48SJeremy L Thompson endif 2311d102b48SJeremy L Thompson enddo 2321d102b48SJeremy L Thompson call ceedvectorrestorearrayread(v,vv,voffset,err) 2331d102b48SJeremy L Thompson 2341d102b48SJeremy L Thompson! Cleanup 2351d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_setup,err) 2361d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_diff,err) 2371d102b48SJeremy L Thompson call ceedqfunctiondestroy(qf_diff_lin,err) 2381d102b48SJeremy L Thompson call ceedoperatordestroy(op_setup,err) 2391d102b48SJeremy L Thompson call ceedoperatordestroy(op_diff,err) 2401d102b48SJeremy L Thompson call ceedoperatordestroy(op_diff_lin,err) 2411d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictu,err) 2421d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictx,err) 2431d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictxi,err) 2441d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictui,err) 2451d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictqi,err) 2461d102b48SJeremy L Thompson call ceedelemrestrictiondestroy(erestrictlini,err) 2471d102b48SJeremy L Thompson call ceedbasisdestroy(bu,err) 2481d102b48SJeremy L Thompson call ceedbasisdestroy(bx,err) 2491d102b48SJeremy L Thompson call ceedvectordestroy(x,err) 2501d102b48SJeremy L Thompson call ceedvectordestroy(a,err) 2511d102b48SJeremy L Thompson call ceedvectordestroy(u,err) 2521d102b48SJeremy L Thompson call ceedvectordestroy(v,err) 2531d102b48SJeremy L Thompson call ceedvectordestroy(qdata,err) 2541d102b48SJeremy L Thompson call ceeddestroy(ceed,err) 2551d102b48SJeremy L Thompson end 2561d102b48SJeremy L Thompson!----------------------------------------------------------------------- 257