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 x,qdata 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 28 character arg*32 29 30! LCOV_EXCL_START 31 external setup,mass 32! LCOV_EXCL_STOP 33 34 call getarg(1,arg) 35 call ceedinit(trim(arg)//char(0),ceed,err) 36 37 do i=0,nelem-1 38 indx(2*i+1)=i 39 indx(2*i+2)=i+1 40 enddo 41 42 call ceedelemrestrictioncreate(ceed,nelem,2,1,1,nx,ceed_mem_host,& 43 & ceed_use_pointer,indx,erestrictx,err) 44 45 do i=0,nelem-1 46 do j=0,p-1 47 indu(p*i+j+1)=2*(i*(p-1)+j) 48 enddo 49 enddo 50 51 call ceedelemrestrictioncreate(ceed,nelem,p,2,1,2*nu,ceed_mem_host,& 52 & ceed_use_pointer,indu,erestrictu,err) 53 stridesu=[1,q,q] 54 call ceedelemrestrictioncreatestrided(ceed,nelem,q,1,q*nelem,stridesu,& 55 & erestrictui,err) 56 57 call ceedbasiscreatetensorh1lagrange(ceed,1,1,2,q,ceed_gauss,bx,err) 58 call ceedbasiscreatetensorh1lagrange(ceed,1,2,p,q,ceed_gauss,bu,err) 59 60 call ceedqfunctioncreateinterior(ceed,1,setup,& 61 &SOURCE_DIR& 62 &//'t500-operator.h:setup'//char(0),qf_setup,err) 63 call ceedqfunctionaddinput(qf_setup,'weight',1,ceed_eval_weight,err) 64 call ceedqfunctionaddinput(qf_setup,'dx',1,ceed_eval_grad,err) 65 call ceedqfunctionaddoutput(qf_setup,'rho',1,ceed_eval_none,err) 66 67 call ceedqfunctioncreateinterior(ceed,1,mass,& 68 &SOURCE_DIR& 69 &//'t500-operator.h:mass'//char(0),qf_mass,err) 70 call ceedqfunctionaddinput(qf_mass,'rho',1,ceed_eval_none,err) 71 call ceedqfunctionaddinput(qf_mass,'u',2,ceed_eval_interp,err) 72 call ceedqfunctionaddoutput(qf_mass,'v',2,ceed_eval_interp,err) 73 74 call ceedoperatorcreate(ceed,qf_setup,ceed_qfunction_none,& 75 & ceed_qfunction_none,op_setup,err) 76 call ceedoperatorcreate(ceed,qf_mass,ceed_qfunction_none,& 77 & ceed_qfunction_none,op_mass,err) 78 79 call ceedvectorcreate(ceed,nx,x,err) 80 call ceedvectorcreate(ceed,nelem*q,qdata,err) 81 82 call ceedoperatorsetfield(op_setup,'weight',ceed_elemrestriction_none,& 83 & bx,ceed_vector_none,err) 84 call ceedoperatorsetfield(op_setup,'dx',erestrictx,bx,& 85 & ceed_vector_active,err) 86 call ceedoperatorsetfield(op_setup,'rho',erestrictui,& 87 ceed_basis_none,ceed_vector_active,err) 88 call ceedoperatorsetfield(op_mass,'rho',erestrictui,& 89 ceed_basis_none,qdata,err) 90 call ceedoperatorsetfield(op_mass,'u',erestrictu,bu,& 91 & ceed_vector_active,err) 92 call ceedoperatorsetfield(op_mass,'v',erestrictu,bu,& 93 & ceed_vector_active,err) 94 95 call ceedoperatorview(op_setup,err) 96 call ceedoperatorview(op_mass,err) 97 98 call ceedvectordestroy(qdata,err) 99 call ceedoperatordestroy(op_mass,err) 100 call ceedoperatordestroy(op_setup,err) 101 call ceedqfunctiondestroy(qf_mass,err) 102 call ceedqfunctiondestroy(qf_setup,err) 103 call ceedbasisdestroy(bu,err) 104 call ceedbasisdestroy(bx,err) 105 call ceedelemrestrictiondestroy(erestrictu,err) 106 call ceedelemrestrictiondestroy(erestrictx,err) 107 call ceedelemrestrictiondestroy(erestrictui,err) 108 call ceeddestroy(ceed,err) 109 end 110!----------------------------------------------------------------------- 111