xref: /libCEED/tests/t400-qfunction-f.f90 (revision d9b786505a4dfcb66b2fcd9e3b61dd507168515d)
18980d4a7Sjeremylt!-----------------------------------------------------------------------
2752c3701SJeremy L Thompson!
3752c3701SJeremy L Thompson! Header with QFunctions
4752c3701SJeremy L Thompson!
5752c3701SJeremy L Thompson      include 't400-qfunction-f.h'
68980d4a7Sjeremylt!-----------------------------------------------------------------------
78980d4a7Sjeremylt      program test
81f9a83abSJed Brown      implicit none
9*ec3da8bcSJed Brown      include 'ceed/fortran.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      real*8 x
218980d4a7Sjeremylt      character arg*32
22c8b9fe72Sjeremylt      integer*8 uoffset,voffset,woffset
238980d4a7Sjeremylt
248980d4a7Sjeremylt      external setup,mass
258980d4a7Sjeremylt
268980d4a7Sjeremylt      call getarg(1,arg)
278980d4a7Sjeremylt      call ceedinit(trim(arg)//char(0),ceed,err)
288980d4a7Sjeremylt
298980d4a7Sjeremylt      call ceedqfunctioncreateinterior(ceed,1,setup,&
302d50dd3dSjeremylt     &SOURCE_DIR&
312d50dd3dSjeremylt     &//'t400-qfunction.h:setup'//char(0),qf_setup,err)
3284e209c4Sjeremylt      call ceedqfunctionaddinput(qf_setup,'w', 1,ceed_eval_weight,err)
3384e209c4Sjeremylt      call ceedqfunctionaddoutput(qf_setup,'qdata',1,ceed_eval_none,err)
348980d4a7Sjeremylt
358980d4a7Sjeremylt      call ceedqfunctioncreateinterior(ceed,1,mass,&
362d50dd3dSjeremylt     &SOURCE_DIR&
372d50dd3dSjeremylt     &//'t400-qfunction.h:mass'//char(0),qf_mass,err)
3884e209c4Sjeremylt      call ceedqfunctionaddinput(qf_mass,'qdata',1,ceed_eval_none,err)
398980d4a7Sjeremylt      call ceedqfunctionaddinput(qf_mass,'u',1,ceed_eval_interp,err)
408980d4a7Sjeremylt      call ceedqfunctionaddoutput(qf_mass,'v',1,ceed_eval_interp,err)
418980d4a7Sjeremylt
428980d4a7Sjeremylt      do i=0,q-1
438980d4a7Sjeremylt        x=2.0*i/(q-1)-1
448980d4a7Sjeremylt        ww(i+1)=1-x*x
458980d4a7Sjeremylt        uu(i+1)=2+3*x+5*x*x
468980d4a7Sjeremylt        vvv(i+1)=ww(i+1)*uu(i+1)
478980d4a7Sjeremylt      enddo
488980d4a7Sjeremylt
498980d4a7Sjeremylt      call ceedvectorcreate(ceed,q,w,err)
50c8b9fe72Sjeremylt      woffset=0
51c8b9fe72Sjeremylt      call ceedvectorsetarray(w,ceed_mem_host,ceed_use_pointer,ww,woffset,err)
528980d4a7Sjeremylt      call ceedvectorcreate(ceed,q,u,err)
53c8b9fe72Sjeremylt      uoffset=0
54c8b9fe72Sjeremylt      call ceedvectorsetarray(u,ceed_mem_host,ceed_use_pointer,uu,uoffset,err)
558980d4a7Sjeremylt      call ceedvectorcreate(ceed,q,v,err)
568980d4a7Sjeremylt      call ceedvectorsetvalue(v,0.d0,err)
578980d4a7Sjeremylt      call ceedvectorcreate(ceed,q,qdata,err)
588980d4a7Sjeremylt      call ceedvectorsetvalue(qdata,0.d0,err)
598980d4a7Sjeremylt
608980d4a7Sjeremylt      call ceedqfunctionapply(qf_setup,q,w,ceed_null,ceed_null,ceed_null,&
618980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
628980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
638980d4a7Sjeremylt             &qdata,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
648980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,ceed_null,&
658980d4a7Sjeremylt             &ceed_null,ceed_null,ceed_null,ceed_null,err)
668980d4a7Sjeremylt
678980d4a7Sjeremylt      call ceedqfunctionapply(qf_mass,q,u,qdata,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             &v,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
74c8b9fe72Sjeremylt      call ceedvectorgetarrayread(v,ceed_mem_host,vv,voffset,err)
758980d4a7Sjeremylt      do i=1,q
76c8b9fe72Sjeremylt        if (abs(vv(i+voffset)-vvv(i)) > 1.0D-14) then
77a2546046Sjeremylt! LCOV_EXCL_START
78c8b9fe72Sjeremylt          write(*,*) 'v(i)=',vv(i+voffset),', vv(i)=',vvv(i)
79de996c55Sjeremylt! LCOV_EXCL_STOP
808980d4a7Sjeremylt        endif
818980d4a7Sjeremylt      enddo
82c8b9fe72Sjeremylt      call ceedvectorrestorearrayread(v,vv,voffset,err)
838980d4a7Sjeremylt
848980d4a7Sjeremylt      call ceedvectordestroy(u,err)
858980d4a7Sjeremylt      call ceedvectordestroy(v,err)
868980d4a7Sjeremylt      call ceedvectordestroy(w,err)
878980d4a7Sjeremylt      call ceedvectordestroy(qdata,err)
888980d4a7Sjeremylt      call ceedqfunctiondestroy(qf_setup,err)
898980d4a7Sjeremylt      call ceedqfunctiondestroy(qf_mass,err)
908980d4a7Sjeremylt      call ceeddestroy(ceed,err)
918980d4a7Sjeremylt      end
928980d4a7Sjeremylt!-----------------------------------------------------------------------
93