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 erestrictxi,erestrictui,erestrictqi 40 integer bx,bu 41 integer qf_setup_mass,qf_apply 42 integer op_setup_mass,op_apply,op_inv 43 integer qdata_mass,x,u,v 44 integer nelem,p,q,d 45 parameter(nelem=1) 46 parameter(p=4) 47 parameter(q=5) 48 parameter(d=2) 49 integer ndofs,nqpts 50 parameter(ndofs=p*p) 51 parameter(nqpts=nelem*q*q) 52 real*8 arrx(d*nelem*2*2),uu(ndofs) 53 integer*8 xoffset,uoffset 54 55 character arg*32 56 57 external setup_mass,apply 58 59 call getarg(1,arg) 60 61 call ceedinit(trim(arg)//char(0),ceed,err) 62 63! DoF Coordinates 64 do i=0,1 65 do j=0,1 66 arrx(i+1+j*2+0*4)=i 67 arrx(i+1+j*2+1*4)=j 68 enddo 69 enddo 70 call ceedvectorcreate(ceed,d*nelem*2*2,x,err) 71 xoffset=0 72 call ceedvectorsetarray(x,ceed_mem_host,ceed_use_pointer,arrx,xoffset,err) 73 74! Qdata Vector 75 call ceedvectorcreate(ceed,nqpts,qdata_mass,err) 76 77! Restrictions 78 call ceedelemrestrictioncreateidentity(ceed,nelem,2*2,nelem*2*2,d,& 79 & erestrictxi,err) 80 81 call ceedelemrestrictioncreateidentity(ceed,nelem,p*p,ndofs,1,& 82 & erestrictui,err) 83 84 call ceedelemrestrictioncreateidentity(ceed,nelem,q*q,nqpts,1,& 85 & erestrictqi,err) 86 87! Bases 88 call ceedbasiscreatetensorh1lagrange(ceed,d,d,2,q,ceed_gauss,bx,err) 89 call ceedbasiscreatetensorh1lagrange(ceed,d,1,p,q,ceed_gauss,bu,err) 90 91! QFunction - setup mass 92 call ceedqfunctioncreateinterior(ceed,1,setup_mass,& 93 &SOURCE_DIR& 94 &//'t540-operator.h:setup_mass'//char(0),qf_setup_mass,err) 95 call ceedqfunctionaddinput(qf_setup_mass,'dx',d*d,ceed_eval_grad,err) 96 call ceedqfunctionaddinput(qf_setup_mass,'_weight',1,ceed_eval_weight,err) 97 call ceedqfunctionaddoutput(qf_setup_mass,'qdata',1,ceed_eval_none,err) 98 99! Operator - setup mass 100 call ceedoperatorcreate(ceed,qf_setup_mass,ceed_qfunction_none,& 101 & ceed_qfunction_none,op_setup_mass,err) 102 call ceedoperatorsetfield(op_setup_mass,'dx',erestrictxi,& 103 & ceed_notranspose,bx,ceed_vector_active,err) 104 call ceedoperatorsetfield(op_setup_mass,'_weight',erestrictxi,& 105 & ceed_notranspose,bx,ceed_vector_none,err) 106 call ceedoperatorsetfield(op_setup_mass,'qdata',erestrictqi,& 107 & ceed_notranspose,ceed_basis_collocated,ceed_vector_active,err) 108 109! Apply Setup Operators 110 call ceedoperatorapply(op_setup_mass,x,qdata_mass,& 111 & ceed_request_immediate,err) 112 113! QFunction - apply 114 call ceedqfunctioncreateinterior(ceed,1,apply,& 115 &SOURCE_DIR& 116 &//'t540-operator.h:apply'//char(0),qf_apply,err) 117 call ceedqfunctionaddinput(qf_apply,'u',1,ceed_eval_interp,err) 118 call ceedqfunctionaddinput(qf_apply,'qdata_mass',1,ceed_eval_none,err) 119 call ceedqfunctionaddoutput(qf_apply,'v',1,ceed_eval_interp,err) 120 121! Operator - apply 122 call ceedoperatorcreate(ceed,qf_apply,ceed_qfunction_none,& 123 & ceed_qfunction_none,op_apply,err) 124 call ceedoperatorsetfield(op_apply,'u',erestrictui,& 125 & ceed_notranspose,bu,ceed_vector_active,err) 126 call ceedoperatorsetfield(op_apply,'qdata_mass',erestrictqi,& 127 & ceed_notranspose,ceed_basis_collocated,qdata_mass,err) 128 call ceedoperatorsetfield(op_apply,'v',erestrictui,& 129 & ceed_notranspose,bu,ceed_vector_active,err) 130 131! Apply original operator 132 call ceedvectorcreate(ceed,ndofs,u,err) 133 call ceedvectorsetvalue(u,1.d0,err) 134 call ceedvectorcreate(ceed,ndofs,v,err) 135 call ceedvectorsetvalue(v,0.d0,err) 136 call ceedoperatorapply(op_apply,u,v,ceed_request_immediate,err) 137 138! Create FDM element inverse 139 call ceedoperatorcreatefdmelementinverse(op_apply,op_inv,& 140 & ceed_request_immediate,err) 141 142! Apply FDM element inverse 143 call ceedoperatorapply(op_inv,v,u,ceed_request_immediate,err) 144 145! Check Output 146 call ceedvectorgetarrayread(u,ceed_mem_host,uu,uoffset,err) 147 do i=1,ndofs 148 if (abs(uu(uoffset+i)-1.0)>1.0d-14) then 149! LCOV_EXCL_START 150 write(*,*) '[',i,'] Error in inverse: ',uu(uoffset+i),' != 1.0' 151! LCOV_EXCL_STOP 152 endif 153 enddo 154 call ceedvectorrestorearrayread(u,uu,uoffset,err) 155 156! Cleanup 157 call ceedqfunctiondestroy(qf_setup_mass,err) 158 call ceedqfunctiondestroy(qf_apply,err) 159 call ceedoperatordestroy(op_setup_mass,err) 160 call ceedoperatordestroy(op_apply,err) 161 call ceedoperatordestroy(op_inv,err) 162 call ceedelemrestrictiondestroy(erestrictxi,err) 163 call ceedelemrestrictiondestroy(erestrictui,err) 164 call ceedelemrestrictiondestroy(erestrictqi,err) 165 call ceedbasisdestroy(bu,err) 166 call ceedbasisdestroy(bx,err) 167 call ceedvectordestroy(x,err) 168 call ceedvectordestroy(u,err) 169 call ceedvectordestroy(v,err) 170 call ceedvectordestroy(qdata_mass,err) 171 call ceeddestroy(ceed,err) 172 end 173!----------------------------------------------------------------------- 174