1!----------------------------------------------------------------------- 2 subroutine setup(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,& 3& u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr) 4 real*8 ctx 5 real*8 u1(1) 6 real*8 u2(1) 7 real*8 v1(1) 8 integer q,ierr 9 10 do i=1,q 11 v1(i)=u1(i)*u2(i) 12 enddo 13 14 ierr=0 15 end 16!----------------------------------------------------------------------- 17 subroutine mass(ctx,q,u1,u2,u3,u4,u5,u6,u7,u8,u9,u10,u11,u12,u13,u14,& 18& u15,u16,v1,v2,v3,v4,v5,v6,v7,v8,v9,v10,v11,v12,v13,v14,v15,v16,ierr) 19 real*8 ctx 20 real*8 u1(1) 21 real*8 u2(1) 22 real*8 v1(1) 23 integer q,ierr 24 25 do i=1,q 26 v1(i)=u2(i)*u1(i) 27 enddo 28 29 ierr=0 30 end 31!----------------------------------------------------------------------- 32 program test 33 34 include 'ceedf.h' 35 36 integer ceed,err,i,j 37 integer imode 38 parameter(imode=ceed_noninterlaced) 39 integer stridesu(3) 40 integer erestrictx,erestrictu,erestrictui 41 integer bx,bu 42 integer qf_setup,qf_mass 43 integer op_setup,op_mass 44 integer qdata,x,u,v 45 integer nelem,p,q 46 parameter(nelem=15) 47 parameter(p=5) 48 parameter(q=8) 49 integer nx,nu 50 parameter(nx=nelem+1) 51 parameter(nu=nelem*(p-1)+1) 52 integer indx(nelem*2) 53 integer indu(nelem*p) 54 real*8 arrx(nx) 55 integer*8 voffset,xoffset 56 57 real*8 hv(nu) 58 real*8 total 59 60 character arg*32 61 62 external setup,mass 63 64 call getarg(1,arg) 65 call ceedinit(trim(arg)//char(0),ceed,err) 66 67 do i=0,nx-1 68 arrx(i+1)=i/(nx-1.d0) 69 enddo 70 do i=0,nelem-1 71 indx(2*i+1)=i 72 indx(2*i+2)=i+1 73 enddo 74 75 call ceedelemrestrictioncreate(ceed,imode,nelem,2,nx,1,ceed_mem_host,& 76 & ceed_use_pointer,indx,erestrictx,err) 77 78 do i=0,nelem-1 79 do j=0,p-1 80 indu(p*i+j+1)=i*(p-1)+j 81 enddo 82 enddo 83 84 call ceedelemrestrictioncreate(ceed,imode,nelem,p,nu,1,ceed_mem_host,& 85 & ceed_use_pointer,indu,erestrictu,err) 86 stridesu=[1,q,q] 87 call ceedelemrestrictioncreatestrided(ceed,nelem,q,q*nelem,1,stridesu,& 88 & erestrictui,err) 89 90 call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q,ceed_gauss,bx,err) 91 call ceedbasiscreatetensorh1lagrange(ceed,1,1,p,q,ceed_gauss,bu,err) 92 93 call ceedqfunctioncreateinterior(ceed,1,setup,& 94 &SOURCE_DIR& 95 &//'t500-operator.h:setup'//char(0),qf_setup,err) 96 call ceedqfunctionaddinput(qf_setup,'_weight',1,ceed_eval_weight,err) 97 call ceedqfunctionaddinput(qf_setup,'dx',1,ceed_eval_grad,err) 98 call ceedqfunctionaddoutput(qf_setup,'rho',1,ceed_eval_none,err) 99 100 call ceedqfunctioncreateinterior(ceed,1,mass,& 101 &SOURCE_DIR& 102 &//'t500-operator.h:mass'//char(0),qf_mass,err) 103 call ceedqfunctionaddinput(qf_mass,'rho',1,ceed_eval_none,err) 104 call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err) 105 call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err) 106 107 call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,& 108 & ceed_qfunction_none,op_setup,err) 109 call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,& 110 & ceed_qfunction_none,op_mass,err) 111 112 call ceedvectorcreate(ceed,nx,x,err) 113 xoffset=0 114 call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 115 call ceedvectorcreate(ceed,nelem*q,qdata,err) 116 117 call ceedoperatorsetfield(op_setup,'_weight',ceed_elemrestriction_none,& 118 & bx,ceed_vector_none,err) 119 call ceedoperatorsetfield(op_setup,'dx',erestrictx,bx,& 120 & ceed_vector_active,err) 121 call ceedoperatorsetfield(op_setup,'rho',erestrictui,& 122 & ceed_basis_collocated,ceed_vector_active,err) 123 call ceedoperatorsetfield(op_mass,'rho',erestrictui,& 124 & ceed_basis_collocated,qdata,err) 125 call ceedoperatorsetfield(op_mass,'u',erestrictu,bu,& 126 & ceed_vector_active,err) 127 call ceedoperatorsetfield(op_mass,'v',erestrictu,bu,& 128 & ceed_vector_active,err) 129 130 call ceedoperatorapply(op_setup,x,qdata,ceed_request_immediate,err) 131 132 call ceedvectorcreate(ceed,nu,u,err) 133 call ceedvectorsetvalue(u,1.d0,err) 134 call ceedvectorcreate(ceed,nu,v,err) 135 call ceedoperatorapply(op_mass,u,v,ceed_request_immediate,err) 136 137 call ceedvectorgetarrayread(v,ceed_mem_host,hv,voffset,err) 138 total=0. 139 do i=1,nu 140 total=total+hv(voffset+i) 141 enddo 142 if (abs(total-1.)>1.0d-10) then 143! LCOV_EXCL_START 144 write(*,*) 'Computed Area: ',total,' != True Area: 1.0' 145! LCOV_EXCL_STOP 146 endif 147 call ceedvectorrestorearrayread(v,hv,voffset,err) 148 149 call ceedvectordestroy(qdata,err) 150 call ceedvectordestroy(x,err) 151 call ceedvectordestroy(u,err) 152 call ceedvectordestroy(v,err) 153 call ceedoperatordestroy(op_mass,err) 154 call ceedoperatordestroy(op_setup,err) 155 call ceedqfunctiondestroy(qf_mass,err) 156 call ceedqfunctiondestroy(qf_setup,err) 157 call ceedbasisdestroy(bu,err) 158 call ceedbasisdestroy(bx,err) 159 call ceedelemrestrictiondestroy(erestrictu,err) 160 call ceedelemrestrictiondestroy(erestrictx,err) 161 call ceedelemrestrictiondestroy(erestrictui,err) 162 call ceeddestroy(ceed,err) 163 end 164!----------------------------------------------------------------------- 165