1!----------------------------------------------------------------------- 2! 3! Header with QFunctions 4! 5 include 't500-operator-f.h' 6!----------------------------------------------------------------------- 7 program test 8 implicit none 9 include 'ceed/fortran.h' 10 11 integer ceed,err,i,j 12 integer stridesu(3) 13 integer erestrictx,erestrictu,erestrictui 14 integer bx,bu 15 integer qf_setup,qf_mass 16 integer op_setup,op_mass 17 integer qdata,x,u,v 18 integer nelem,p,q 19 parameter(nelem=15) 20 parameter(p=5) 21 parameter(q=8) 22 integer nx,nu 23 parameter(nx=nelem+1) 24 parameter(nu=nelem*(p-1)+1) 25 integer indx(nelem*2) 26 integer indu(nelem*p) 27 real*8 arrx(nx) 28 integer*8 voffset,xoffset 29 30 real*8 hv(nu) 31 32 character arg*32 33 34 external setup,mass 35 36 call getarg(1,arg) 37 call ceedinit(trim(arg)//char(0),ceed,err) 38 39 do i=0,nx-1 40 arrx(i+1)=i/(nx-1.d0) 41 enddo 42 do i=0,nelem-1 43 indx(2*i+1)=i 44 indx(2*i+2)=i+1 45 enddo 46 47 call ceedelemrestrictioncreate(ceed,nelem,2,1,1,nx,ceed_mem_host,& 48 & ceed_use_pointer,indx,erestrictx,err) 49 50 do i=0,nelem-1 51 do j=0,p-1 52 indu(p*i+j+1)=i*(p-1)+j 53 enddo 54 enddo 55 56 call ceedelemrestrictioncreate(ceed,nelem,p,1,1,nu,ceed_mem_host,& 57 & ceed_use_pointer,indu,erestrictu,err) 58 stridesu=[1,q,q] 59 call ceedelemrestrictioncreatestrided(ceed,nelem,q,1,q*nelem,stridesu,& 60 & erestrictui,err) 61 62 call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q,ceed_gauss,bx,err) 63 call ceedbasiscreatetensorh1lagrange(ceed,1,1,p,q,ceed_gauss,bu,err) 64 65 call ceedqfunctioncreateinterior(ceed,1,setup,& 66 &SOURCE_DIR& 67 &//'t500-operator.h:setup'//char(0),qf_setup,err) 68 call ceedqfunctionaddinput(qf_setup,'weight',1,ceed_eval_weight,err) 69 call ceedqfunctionaddinput(qf_setup,'dx',1,ceed_eval_grad,err) 70 call ceedqfunctionaddoutput(qf_setup,'rho',1,ceed_eval_none,err) 71 72 call ceedqfunctioncreateinterior(ceed,1,mass,& 73 &SOURCE_DIR& 74 &//'t500-operator.h:mass'//char(0),qf_mass,err) 75 call ceedqfunctionaddinput(qf_mass,'rho',1,ceed_eval_none,err) 76 call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err) 77 call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err) 78 79 call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,& 80 & ceed_qfunction_none,op_setup,err) 81 call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,& 82 & ceed_qfunction_none,op_mass,err) 83 84 call ceedvectorcreate(ceed,nx,x,err) 85 xoffset=0 86 call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 87 call ceedvectorcreate(ceed,nelem*q,qdata,err) 88 89 call ceedoperatorsetfield(op_setup,'weight',ceed_elemrestriction_none,& 90 & bx,ceed_vector_none,err) 91 call ceedoperatorsetfield(op_setup,'dx',erestrictx,bx,& 92 & ceed_vector_active,err) 93 call ceedoperatorsetfield(op_setup,'rho',erestrictui,& 94 ceed_basis_none,ceed_vector_active,err) 95 call ceedoperatorsetfield(op_mass,'rho',erestrictui,& 96 ceed_basis_none,qdata,err) 97 call ceedoperatorsetfield(op_mass,'u',erestrictu,bu,& 98 & ceed_vector_active,err) 99 call ceedoperatorsetfield(op_mass,'v',erestrictu,bu,& 100 & ceed_vector_active,err) 101 102 call ceedoperatorapply(op_setup,x,qdata,ceed_request_immediate,err) 103 104 call ceedvectorcreate(ceed,nu,u,err) 105 call ceedvectorsetvalue(u,0.d0,err) 106 call ceedvectorcreate(ceed,nu,v,err) 107 call ceedoperatorapply(op_mass,u,v,ceed_request_immediate,err) 108 109 call ceedvectorgetarrayread(v,ceed_mem_host,hv,voffset,err) 110 do i=1,nu 111 if (abs(hv(voffset+i))>1.0d-10) then 112! LCOV_EXCL_START 113 write(*,*) '[',i,'] v ',hv(voffset+i),' != 0.0' 114! LCOV_EXCL_STOP 115 endif 116 enddo 117 call ceedvectorrestorearrayread(v,hv,voffset,err) 118 119 call ceedvectordestroy(qdata,err) 120 call ceedvectordestroy(x,err) 121 call ceedvectordestroy(u,err) 122 call ceedvectordestroy(v,err) 123 call ceedoperatordestroy(op_mass,err) 124 call ceedoperatordestroy(op_setup,err) 125 call ceedqfunctiondestroy(qf_mass,err) 126 call ceedqfunctiondestroy(qf_setup,err) 127 call ceedbasisdestroy(bu,err) 128 call ceedbasisdestroy(bx,err) 129 call ceedelemrestrictiondestroy(erestrictu,err) 130 call ceedelemrestrictiondestroy(erestrictx,err) 131 call ceedelemrestrictiondestroy(erestrictui,err) 132 call ceeddestroy(ceed,err) 133 end 134!----------------------------------------------------------------------- 135