1!----------------------------------------------------------------------- 2! 3! Header with QFunctions 4! 5 include 't502-operator-f.h' 6!----------------------------------------------------------------------- 7 program test 8 9 include 'ceedf.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 hu(nu*2),hv(nu*2) 31 real*8 total1,total2 32 33 character arg*32 34 35 external setup,mass 36 37 call getarg(1,arg) 38 call ceedinit(trim(arg)//char(0),ceed,err) 39 40 do i=0,nx-1 41 arrx(i+1)=i/(nx-1.d0) 42 enddo 43 do i=0,nelem-1 44 indx(2*i+1)=i 45 indx(2*i+2)=i+1 46 enddo 47 48 call ceedelemrestrictioncreate(ceed,nelem,2,1,1,nx,ceed_mem_host,& 49 & ceed_use_pointer,indx,erestrictx,err) 50 51 do i=0,nelem-1 52 do j=0,p-1 53 indu(p*i+j+1)=2*(i*(p-1)+j) 54 enddo 55 enddo 56 57 call ceedelemrestrictioncreate(ceed,nelem,p,2,1,2*nu,ceed_mem_host,& 58 & ceed_use_pointer,indu,erestrictu,err) 59 stridesu=[1,q,q] 60 call ceedelemrestrictioncreatestrided(ceed,nelem,q,1,q*nelem,stridesu,& 61 & erestrictui,err) 62 63 call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q,ceed_gauss,bx,err) 64 call ceedbasiscreatetensorh1lagrange(ceed,1,2,p,q,ceed_gauss,bu,err) 65 66 call ceedqfunctioncreateinterior(ceed,1,setup,& 67 &SOURCE_DIR& 68 &//'t502-operator.h:setup'//char(0),qf_setup,err) 69 call ceedqfunctionaddinput(qf_setup,'_weight',1,ceed_eval_weight,err) 70 call ceedqfunctionaddinput(qf_setup,'dx',1,ceed_eval_grad,err) 71 call ceedqfunctionaddoutput(qf_setup,'rho',1,ceed_eval_none,err) 72 73 call ceedqfunctioncreateinterior(ceed,1,mass,& 74 &SOURCE_DIR& 75 &//'t502-operator.h:mass'//char(0),qf_mass,err) 76 call ceedqfunctionaddinput(qf_mass,'rho',1,ceed_eval_none,err) 77 call ceedqfunctionaddinput(qf_mass,'u',2,ceed_eval_interp,err) 78 call ceedqfunctionaddoutput(qf_mass,'v',2,ceed_eval_interp,err) 79 80 call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,& 81 & ceed_qfunction_none,op_setup,err) 82 call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,& 83 & ceed_qfunction_none,op_mass,err) 84 85 call ceedvectorcreate(ceed,nx,x,err) 86 xoffset=0 87 call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 88 call ceedvectorcreate(ceed,nelem*q,qdata,err) 89 90 call ceedoperatorsetfield(op_setup,'_weight',ceed_elemrestriction_none,& 91 & bx,ceed_vector_none,err) 92 call ceedoperatorsetfield(op_setup,'dx',erestrictx,bx,& 93 & ceed_vector_active,err) 94 call ceedoperatorsetfield(op_setup,'rho',erestrictui,& 95 & ceed_basis_collocated,ceed_vector_active,err) 96 call ceedoperatorsetfield(op_mass,'rho',erestrictui,& 97 & ceed_basis_collocated,qdata,err) 98 call ceedoperatorsetfield(op_mass,'u',erestrictu,bu,& 99 & ceed_vector_active,err) 100 call ceedoperatorsetfield(op_mass,'v',erestrictu,bu,& 101 & ceed_vector_active,err) 102 103 call ceedoperatorapply(op_setup,x,qdata,ceed_request_immediate,err) 104 105 call ceedvectorcreate(ceed,2*nu,u,err) 106 call ceedvectorgetarray(u,ceed_mem_host,hu,voffset,err) 107 do i=1,nu 108 hu(voffset+2*i-1)=1. 109 hu(voffset+2*i)=2. 110 enddo 111 call ceedvectorrestorearray(u,hu,voffset,err) 112 call ceedvectorcreate(ceed,2*nu,v,err) 113 call ceedoperatorapply(op_mass,u,v,ceed_request_immediate,err) 114 115 call ceedvectorgetarrayread(v,ceed_mem_host,hv,voffset,err) 116 total1=0. 117 total2=0. 118 do i=1,nu 119 total1=total1+hv(voffset+2*i-1) 120 total2=total2+hv(voffset+2*i) 121 enddo 122 if (abs(total1-1.)>1.0d-10) then 123! LCOV_EXCL_START 124 write(*,*) 'Computed Area: ',total1,' != True Area: 1.0' 125! LCOV_EXCL_STOP 126 endif 127 if (abs(total2-2.)>1.0d-10) then 128! LCOV_EXCL_START 129 write(*,*) 'Computed Area: ',total2,' != True Area: 2.0' 130! LCOV_EXCL_STOP 131 endif 132 call ceedvectorrestorearrayread(v,hv,voffset,err) 133 134 call ceedvectordestroy(qdata,err) 135 call ceedvectordestroy(x,err) 136 call ceedvectordestroy(u,err) 137 call ceedvectordestroy(v,err) 138 call ceedoperatordestroy(op_mass,err) 139 call ceedoperatordestroy(op_setup,err) 140 call ceedqfunctiondestroy(qf_mass,err) 141 call ceedqfunctiondestroy(qf_setup,err) 142 call ceedbasisdestroy(bu,err) 143 call ceedbasisdestroy(bx,err) 144 call ceedelemrestrictiondestroy(erestrictu,err) 145 call ceedelemrestrictiondestroy(erestrictx,err) 146 call ceedelemrestrictiondestroy(erestrictui,err) 147 call ceeddestroy(ceed,err) 148 end 149!----------------------------------------------------------------------- 150