xref: /libCEED/tests/t401-qfunction-f.f90 (revision 752c3701a992135134df075f4ef18abc790b3495)
18980d4a7Sjeremylt!-----------------------------------------------------------------------
2*752c3701SJeremy L Thompson!
3*752c3701SJeremy L Thompson! Header with QFunctions
4*752c3701SJeremy L Thompson!
5*752c3701SJeremy L Thompson      include 't401-qfunction-f.h'
68980d4a7Sjeremylt!-----------------------------------------------------------------------
78980d4a7Sjeremylt      program test
81f9a83abSJed Brown      implicit none
98980d4a7Sjeremylt      include 'ceedf.h'
108980d4a7Sjeremylt
118980d4a7Sjeremylt      integer ceed,err
128980d4a7Sjeremylt      integer qdata,w,u,v
138980d4a7Sjeremylt      integer qf_setup,qf_mass
148980d4a7Sjeremylt      integer q,i
158980d4a7Sjeremylt      parameter(q=8)
168980d4a7Sjeremylt      real*8 ww(q)
178980d4a7Sjeremylt      real*8 uu(q)
188980d4a7Sjeremylt      real*8 vv(q)
198980d4a7Sjeremylt      real*8 vvv(q)
208980d4a7Sjeremylt      integer ctxsize
218980d4a7Sjeremylt      parameter(ctxsize=5)
228980d4a7Sjeremylt      real*8 ctx(5)
238980d4a7Sjeremylt      real*8 x
248980d4a7Sjeremylt      character arg*32
25c8b9fe72Sjeremylt      integer*8 uoffset,voffset,woffset
268980d4a7Sjeremylt
278980d4a7Sjeremylt      external setup,mass
288980d4a7Sjeremylt
298980d4a7Sjeremylt      ctx=(/1.d0,2.d0,3.d0,4.d0,5.d0/)
308980d4a7Sjeremylt
318980d4a7Sjeremylt      call getarg(1,arg)
328980d4a7Sjeremylt      call ceedinit(trim(arg)//char(0),ceed,err)
338980d4a7Sjeremylt
348980d4a7Sjeremylt      call ceedqfunctioncreateinterior(ceed,1,setup,&
352d50dd3dSjeremylt     &SOURCE_DIR&
362d50dd3dSjeremylt     &//'t401-qfunction.h:setup'//char(0),qf_setup,err)
3784e209c4Sjeremylt      call ceedqfunctionaddinput(qf_setup,'w', 1,ceed_eval_weight,err)
3884e209c4Sjeremylt      call ceedqfunctionaddoutput(qf_setup,'qdata',1,ceed_eval_none,err)
398980d4a7Sjeremylt
408980d4a7Sjeremylt      call ceedqfunctioncreateinterior(ceed,1,mass,&
412d50dd3dSjeremylt     &SOURCE_DIR&
422d50dd3dSjeremylt     &//'t401-qfunction.h:mass'//char(0),qf_mass,err)
4384e209c4Sjeremylt      call ceedqfunctionaddinput(qf_mass,'qdata',1,ceed_eval_none,err)
448980d4a7Sjeremylt      call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err)
458980d4a7Sjeremylt      call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err)
468980d4a7Sjeremylt
478980d4a7Sjeremylt      call ceedqfunctionsetcontext(qf_mass,ctx,ctxsize,err)
488980d4a7Sjeremylt
498980d4a7Sjeremylt      do i=0,q-1
508980d4a7Sjeremylt        x=2.0*i/(q-1)-1
518980d4a7Sjeremylt        ww(i+1)=1-x*x
528980d4a7Sjeremylt        uu(i+1)=2+3*x+5*x*x
538980d4a7Sjeremylt        vvv(i+1)=ww(i+1)*uu(i+1)
548980d4a7Sjeremylt      enddo
558980d4a7Sjeremylt
568980d4a7Sjeremylt      call ceedvectorcreate(ceed,q,w,err)
57c8b9fe72Sjeremylt      woffset=0
58c8b9fe72Sjeremylt      call ceedvectorsetarray(w,ceed_mem_host,ceed_use_pointer,ww,woffset,err)
598980d4a7Sjeremylt      call ceedvectorcreate(ceed,q,u,err)
60c8b9fe72Sjeremylt      uoffset=0
61c8b9fe72Sjeremylt      call ceedvectorsetarray(u,ceed_mem_host,ceed_use_pointer,uu,uoffset,err)
628980d4a7Sjeremylt      call ceedvectorcreate(ceed,q,v,err)
638980d4a7Sjeremylt      call ceedvectorsetvalue(v,0.d0,err)
648980d4a7Sjeremylt      call ceedvectorcreate(ceed,q,qdata,err)
658980d4a7Sjeremylt      call ceedvectorsetvalue(qdata,0.d0,err)
668980d4a7Sjeremylt
678980d4a7Sjeremylt      call ceedqfunctionapply(qf_setup,q,w,ceed_null,ceed_null,ceed_null,&
688980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
698980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
708980d4a7Sjeremylt             &qdata,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
718980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
728980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,err)
738980d4a7Sjeremylt
748980d4a7Sjeremylt      call ceedqfunctionapply(qf_mass,q,u,qdata,ceed_null,ceed_null,&
758980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
768980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
778980d4a7Sjeremylt             &v,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
788980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
798980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,err)
808980d4a7Sjeremylt
81c8b9fe72Sjeremylt      call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err)
828980d4a7Sjeremylt      do i=1,q
83c8b9fe72Sjeremylt        if (abs(vv(i+voffset)-ctx(5)*vvv(i)) > 1.0D-14) then
84a2546046Sjeremylt! LCOV_EXCL_START
85c8b9fe72Sjeremylt          write(*,*) 'v(i)=',vv(i+voffset),', 5*vv(i)=',ctx(5)*vvv(i)
86de996c55Sjeremylt! LCOV_EXCL_STOP
878980d4a7Sjeremylt        endif
888980d4a7Sjeremylt      enddo
89c8b9fe72Sjeremylt      call ceedvectorrestorearrayread(v,vv,voffset,err)
908980d4a7Sjeremylt
918980d4a7Sjeremylt      call ceedvectordestroy(u,err)
928980d4a7Sjeremylt      call ceedvectordestroy(v,err)
938980d4a7Sjeremylt      call ceedvectordestroy(w,err)
948980d4a7Sjeremylt      call ceedvectordestroy(qdata,err)
958980d4a7Sjeremylt      call ceedqfunctiondestroy(qf_setup,err)
968980d4a7Sjeremylt      call ceedqfunctiondestroy(qf_mass,err)
978980d4a7Sjeremylt      call ceeddestroy(ceed,err)
988980d4a7Sjeremylt      end
998980d4a7Sjeremylt!-----------------------------------------------------------------------
100