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