! D02PF_A1T1W_F Example Program Text
! Mark 30.2 Release. NAG Copyright 2024.
Module d02pf_a1t1w_fe_mod
! D02PF_A1T1W_F Example Program Module:
! Parameters and User-defined Routines
! .. Use Statements ..
Use iso_c_binding, Only: c_ptr
Use nagad_library, Only: nagad_a1t1w_w_rtype, Operator (*), &
Assignment (=), Operator (-)
! .. Implicit None Statement ..
Implicit None
! .. Accessibility Statements ..
Private
Public :: f
! .. Parameters ..
Integer, Parameter, Public :: liwsav = 130, n = 2, nin = 5, &
nout = 6
Integer, Parameter, Public :: lrwsav = 350 + 32*n
Contains
Subroutine f(ad_handle,t,n,y,yp,iuser,ruser)
! .. Implicit None Statement ..
Implicit None
! .. Scalar Arguments ..
Type (c_ptr), Intent (Inout) :: ad_handle
Type (nagad_a1t1w_w_rtype), Intent (In) :: t
Integer, Intent (In) :: n
! .. Array Arguments ..
Type (nagad_a1t1w_w_rtype), Intent (Inout) :: ruser(*)
Type (nagad_a1t1w_w_rtype), Intent (In) :: y(n)
Type (nagad_a1t1w_w_rtype), Intent (Out) :: yp(n)
Integer, Intent (Inout) :: iuser(*)
! .. Executable Statements ..
yp(1) = ruser(1)*y(2)
yp(2) = -ruser(2)*y(1)
Return
End Subroutine f
End Module d02pf_a1t1w_fe_mod
Program d02pf_a1t1w_fe
! D02PF_A1T1W_F Example Main Program
! .. Use Statements ..
Use d02pf_a1t1w_fe_mod, Only: f, liwsav, lrwsav, n, nin, nout
Use iso_c_binding, Only: c_ptr
Use nagad_library, Only: d02pf_a1t1w_f, d02pq_a1t1w_f, d02pt_a1t1w_f, &
nagad_a1t1w_get_derivative, &
nagad_a1t1w_inc_derivative, &
nagad_a1t1w_ir_interpret_adjoint, &
nagad_a1t1w_ir_register_variable, &
nagad_a1t1w_ir_remove, nagad_a1t1w_w_rtype, &
nagad_t1w_w_rtype, x10aa_a1t1w_f, &
x10ab_a1t1w_f, x10za_a1t1w_f, Operator (>=), &
Assignment (=)
Use nag_library, Only: nag_wp
! .. Implicit None Statement ..
Implicit None
! .. Local Scalars ..
Type (c_ptr) :: ad_handle
Type (nagad_a1t1w_w_rtype) :: hnext, hs, te, tnow, tol, ts, waste
Type (nagad_t1w_w_rtype) :: t_t
Real (Kind=nag_wp) :: dr, hstart, t, tend, tolr, tstart
Integer :: fevals, i, ifail, method, stepcost, &
stepsok
! .. Local Arrays ..
Type (nagad_a1t1w_w_rtype) :: ruser(2), th(n), y(n), ynow(n), &
ypnow(n)
Type (nagad_a1t1w_w_rtype), Allocatable :: rwsav(:)
Real (Kind=nag_wp) :: thresh(n), yinit(n), yr(n)
Integer :: iuser(1)
Integer, Allocatable :: iwsav(:)
! .. Executable Statements ..
Write (nout,*) 'D02PF_A1T1W_F Example Program Results'
Allocate (iwsav(liwsav),rwsav(lrwsav))
! Set initial conditions and input
! Skip heading in data file
Read (nin,*)
Read (nin,*) method
Read (nin,*) tstart, tend
Read (nin,*) yinit(1:n)
Read (nin,*) hstart
Read (nin,*) thresh(1:n)
! Set control for output
tolr = 1.0E-4_nag_wp
ts = tstart
te = tend
y(1:n) = yinit(1:n)
tol = tolr
th(1:n) = thresh(1:n)
hs = hstart
! Create AD tape
Call x10za_a1t1w_f
! Create AD configuration data object
ifail = 0
Call x10aa_a1t1w_f(ad_handle,ifail)
ruser(1) = 1.0_nag_wp
ruser(2) = 1.0_nag_wp
! Register variables to differentiate w.r.t.
ruser(1:2)%value%tangent = 1.0_nag_wp
Call nagad_a1t1w_ir_register_variable(ruser)
! ifail: behaviour on error exit
! =0 for hard exit, =1 for quiet-soft, =-1 for noisy-soft
ifail = 0
Call d02pq_a1t1w_f(ad_handle,n,ts,te,y,tol,th,method,hs,iwsav,rwsav, &
ifail)
Write (nout,99999) tolr
Write (nout,99998)
Write (nout,99997) tstart, yinit(1:n)
loop: Do
ifail = 0
Call d02pf_a1t1w_f(ad_handle,f,n,tnow,ynow,ypnow,iuser,ruser,iwsav, &
rwsav,ifail)
t = tnow%value%value
yr(1:n) = ynow(1:n)%value%value
Write (nout,99997) t, yr(1:n)
If (tnow>=te) Then
Exit loop
End If
End Do loop
ifail = 0
Call d02pt_a1t1w_f(ad_handle,fevals,stepcost,waste,stepsok,hnext,iwsav, &
rwsav,ifail)
Write (nout,99996) fevals
99999 Format (/,' Calculation with TOL = ',1P,E8.1)
99998 Format (/,' t y1 y2',/)
99997 Format (1X,F6.3,2(3X,F7.3))
99996 Format (/,' Cost of the integration in evaluations of F is',I6)
t_t = 1.0_nag_wp
Do i = 1, n
Call nagad_a1t1w_inc_derivative(ynow(i),t_t)
End Do
Call nagad_a1t1w_ir_interpret_adjoint(ifail)
Write (nout,*)
Write (nout,*) &
' Derivatives calculated: Second order, adjoints of tangents'
Write (nout,*) ' Computational mode : algorithmic'
! Get derivatives
dr = 0.0_nag_wp
Do i = 1, n
t_t = nagad_a1t1w_get_derivative(ruser(i))
dr = dr + t_t%tangent
End Do
Write (nout,*) ' Sum of Hessian terms for y w.r.t system parameters :'
Write (nout,*)
Write (nout,99995) ' Sum{i,j,k} d^2y_k/druser(i)druser(j) =', dr
99995 Format (1X,A,1X,E12.5)
! Remove computational data object and tape
ifail = 0
Call x10ab_a1t1w_f(ad_handle,ifail)
Call nagad_a1t1w_ir_remove
End Program d02pf_a1t1w_fe