module huti_cgs
use huti_aux
implicit none
#include "huti_fdefs.h"
#define X xvec
#define B rhsvec
#define RTLD work(:,1)
#define RTLD_ind 1
#define P work(:,2)
#define P_ind 2
#define Q work(:,3)
#define Q_ind 3
#define U work(:,4)
#define U_ind 4
#define T1V work(:,5)
#define T1V_ind 5
#define T2V work(:,6)
#define T2V_ind 6
#define R work(:,7)
#define R_ind 7
contains
subroutine huti_scgssolv ( ndim, wrkdim, xvec, rhsvec, ipar,&
dpar, work, matvecsubr, pcondlsubr, pcondrsubr, &
dotprodfun, normfun, stopcfun )
use huti_interfaces
implicit none
procedure( mv_iface_s ), pointer :: matvecsubr
procedure( pc_iface_s ), pointer :: pcondlsubr
procedure( pc_iface_s ), pointer :: pcondrsubr
procedure( dotp_iface_s ), pointer :: dotprodfun
procedure( norm_iface_s ), pointer :: normfun
procedure( stopc_iface_s ), pointer :: stopcfun
integer :: ndim, wrkdim
real, dimension(ndim) :: xvec, rhsvec
integer, dimension(HUTI_IPAR_DFLTSIZE) :: ipar
double precision, dimension(HUTI_DPAR_DFLTSIZE) :: dpar
real, dimension(ndim,wrkdim) :: work
real :: rho, oldrho, alpha, beta
integer :: iter_count
real :: residual, rhsnorm, precrhsnorm
iter_count = 1
HUTI_EXTOP_MATTYPE = HUTI_MAT_NOTTRPSED
if ( HUTI_STOPC .eq. HUTI_TRESID_SCALED_BYB .or. &
HUTI_STOPC .eq. HUTI_PRESID_SCALED_BYB ) then
rhsnorm = normfun( HUTI_NDIM, B, 1 )
end if
if ( HUTI_STOPC .eq. HUTI_PRESID_SCALED_BYPRECB ) then
call pcondlsubr( P, B, ipar )
precrhsnorm = normfun( HUTI_NDIM, P, 1 )
end if
if ( HUTI_INITIALX .eq. HUTI_RANDOMX ) then
call huti_srandvec ( X, ipar )
else if ( HUTI_INITIALX .ne. HUTI_USERSUPPLIEDX ) then
X = 1
end if
call matvecsubr( X, R, ipar )
R = B - R
RTLD = R
300 continue
rho = dotprodfun( HUTI_NDIM, RTLD, 1, R, 1 )
if ( rho .eq. 0 ) then
HUTI_INFO = HUTI_CGS_RHO
go to 1000
end if
if ( iter_count .eq. 1 ) then
U = R
P = U
else
beta = rho / oldrho
U = R + beta * Q
P = U + beta * Q + beta * beta * P
end if
call pcondlsubr( T2V, P, ipar )
call pcondrsubr( T1V, T2V, ipar )
call matvecsubr( T1V, T2V, ipar )
alpha = rho / dotprodfun( HUTI_NDIM, RTLD, 1, T2V, 1 )
Q = U - alpha * T2V
T2V = U + Q
call pcondlsubr( U, T2V, ipar )
call pcondrsubr( T1V, U, ipar )
X = X + alpha * T1V
call matvecsubr( T1V, T2V, ipar )
R = R - alpha * T2V
select case (HUTI_STOPC)
case (HUTI_TRUERESIDUAL)
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 )
case (HUTI_TRESID_SCALED_BYB)
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 ) / rhsnorm
case (HUTI_PSEUDORESIDUAL)
residual = normfun( HUTI_NDIM, R, 1 )
case (HUTI_PRESID_SCALED_BYB)
residual = normfun( HUTI_NDIM, R, 1 ) / rhsnorm
case (HUTI_PRESID_SCALED_BYPRECB)
residual = normfun( HUTI_NDIM, R, 1 ) / precrhsnorm
case (HUTI_XDIFF_NORM)
T1V = alpha * T1V
residual = normfun( HUTI_NDIM, T1V, 1 )
case (HUTI_USUPPLIED_STOPC)
residual = stopcfun( X, B, R, ipar, dpar )
case default
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 )
end select
if ( HUTI_DBUGLVL .ne. HUTI_NO_DEBUG ) then
if ( mod(iter_count, HUTI_DBUGLVL) .eq. 0 ) then
write (*, '(I8, E11.4)') iter_count, residual
end if
end if
if ( residual .lt. HUTI_TOLERANCE ) then
HUTI_INFO = HUTI_CONVERGENCE
go to 1000
end if
IF( residual /= residual .OR. residual > HUTI_MAXTOLERANCE ) THEN
HUTI_INFO = HUTI_DIVERGENCE
GOTO 1000
END IF
oldrho = rho
iter_count = iter_count + 1
if ( iter_count .gt. HUTI_MAXIT ) then
HUTI_INFO = HUTI_MAXITER
go to 1000
end if
go to 300
1000 continue
if ( HUTI_DBUGLVL .ne. HUTI_NO_DEBUG ) then
write (*, '(I8, E11.4)') iter_count, residual
end if
HUTI_ITERS = iter_count
return
end subroutine huti_scgssolv
subroutine huti_dcgssolv ( ndim, wrkdim, xvec, rhsvec, ipar,&
dpar, work, matvecsubr, pcondlsubr, pcondrsubr, &
dotprodfun, normfun, stopcfun )
use huti_interfaces
implicit none
procedure( mv_iface_d ), pointer :: matvecsubr
procedure( pc_iface_d ), pointer :: pcondlsubr
procedure( pc_iface_d ), pointer :: pcondrsubr
procedure( dotp_iface_d ), pointer :: dotprodfun
procedure( norm_iface_d ), pointer :: normfun
procedure( stopc_iface_d ), pointer :: stopcfun
integer :: ndim, wrkdim
double precision, dimension(ndim) :: xvec, rhsvec
integer, dimension(HUTI_IPAR_DFLTSIZE) :: ipar
double precision, dimension(HUTI_DPAR_DFLTSIZE) :: dpar
double precision, dimension(ndim,wrkdim) :: work
double precision :: rho, oldrho, alpha, beta
integer :: iter_count
double precision :: residual, rhsnorm, precrhsnorm
iter_count = 1
HUTI_EXTOP_MATTYPE = HUTI_MAT_NOTTRPSED
if ( HUTI_STOPC .eq. HUTI_TRESID_SCALED_BYB .or. &
HUTI_STOPC .eq. HUTI_PRESID_SCALED_BYB ) then
rhsnorm = normfun( HUTI_NDIM, B, 1 )
end if
if ( HUTI_STOPC .eq. HUTI_PRESID_SCALED_BYPRECB ) then
call pcondlsubr( P, B, ipar )
precrhsnorm = normfun( HUTI_NDIM, P, 1 )
end if
if ( HUTI_INITIALX .eq. HUTI_RANDOMX ) then
call huti_drandvec ( X, ipar )
else if ( HUTI_INITIALX .ne. HUTI_USERSUPPLIEDX ) then
X = 1
end if
call matvecsubr( X, R, ipar )
R = B - R
RTLD = R
300 continue
rho = dotprodfun( HUTI_NDIM, RTLD, 1, R, 1 )
if ( rho .eq. 0 ) then
HUTI_INFO = HUTI_CGS_RHO
go to 1000
end if
if ( iter_count .eq. 1 ) then
U = R
P = U
else
beta = rho / oldrho
U = R + beta * Q
P = U + beta * Q + beta * beta * P
end if
call pcondlsubr( T2V, P, ipar )
call pcondrsubr( T1V, T2V, ipar )
call matvecsubr( T1V, T2V, ipar )
alpha = rho / dotprodfun( HUTI_NDIM, RTLD, 1, T2V, 1 )
Q = U - alpha * T2V
T2V = U + Q
call pcondlsubr( U, T2V, ipar )
call pcondrsubr( T1V, U, ipar )
X = X + alpha * T1V
call matvecsubr( T1V, T2V, ipar )
R = R - alpha * T2V
select case (HUTI_STOPC)
case (HUTI_TRUERESIDUAL)
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 )
case (HUTI_TRESID_SCALED_BYB)
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 ) / rhsnorm
case (HUTI_PSEUDORESIDUAL)
residual = normfun( HUTI_NDIM, R, 1 )
case (HUTI_PRESID_SCALED_BYB)
residual = normfun( HUTI_NDIM, R, 1 ) / rhsnorm
case (HUTI_PRESID_SCALED_BYPRECB)
residual = normfun( HUTI_NDIM, R, 1 ) / precrhsnorm
case (HUTI_XDIFF_NORM)
T1V = alpha * T1V
residual = normfun( HUTI_NDIM, T1V, 1 )
case (HUTI_USUPPLIED_STOPC)
residual = stopcfun( X, B, R, ipar, dpar )
case default
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 )
end select
if ( HUTI_DBUGLVL .ne. HUTI_NO_DEBUG ) then
if ( mod(iter_count, HUTI_DBUGLVL) .eq. 0 ) then
write (*, '(I8, E11.4)') iter_count, residual
end if
end if
if ( residual .lt. HUTI_TOLERANCE ) then
HUTI_INFO = HUTI_CONVERGENCE
go to 1000
end if
IF( residual /= residual .OR. residual > HUTI_MAXTOLERANCE ) THEN
HUTI_INFO = HUTI_DIVERGENCE
GOTO 1000
END IF
oldrho = rho
iter_count = iter_count + 1
if ( iter_count .gt. HUTI_MAXIT ) then
HUTI_INFO = HUTI_MAXITER
go to 1000
end if
go to 300
1000 continue
if ( HUTI_DBUGLVL .ne. HUTI_NO_DEBUG ) then
write (*, '(I8, E11.4)') iter_count, residual
end if
HUTI_ITERS = iter_count
return
end subroutine huti_dcgssolv
subroutine huti_ccgssolv ( ndim, wrkdim, xvec, rhsvec, ipar,&
dpar, work, matvecsubr, pcondlsubr, pcondrsubr, &
dotprodfun, normfun, stopcfun )
use huti_interfaces
implicit none
procedure( mv_iface_c ), pointer :: matvecsubr
procedure( pc_iface_c ), pointer :: pcondlsubr
procedure( pc_iface_c ), pointer :: pcondrsubr
procedure( dotp_iface_c ), pointer :: dotprodfun
procedure( norm_iface_c ), pointer :: normfun
procedure( stopc_iface_c ), pointer :: stopcfun
integer :: ndim, wrkdim
complex, dimension(ndim) :: xvec, rhsvec
integer, dimension(HUTI_IPAR_DFLTSIZE) :: ipar
double precision, dimension(HUTI_DPAR_DFLTSIZE) :: dpar
complex, dimension(ndim,wrkdim) :: work
complex :: rho, oldrho, alpha, beta
integer :: iter_count
real :: residual, rhsnorm, precrhsnorm
iter_count = 1
HUTI_EXTOP_MATTYPE = HUTI_MAT_NOTTRPSED
if ( HUTI_STOPC .eq. HUTI_TRESID_SCALED_BYB .or. &
HUTI_STOPC .eq. HUTI_PRESID_SCALED_BYB ) then
rhsnorm = normfun( HUTI_NDIM, B, 1 )
end if
if ( HUTI_STOPC .eq. HUTI_PRESID_SCALED_BYPRECB ) then
call pcondlsubr( P, B, ipar )
precrhsnorm = normfun( HUTI_NDIM, P, 1 )
end if
if ( HUTI_INITIALX .eq. HUTI_RANDOMX ) then
call huti_crandvec ( X, ipar )
else if ( HUTI_INITIALX .ne. HUTI_USERSUPPLIEDX ) then
X = 1
end if
call matvecsubr( X, R, ipar )
R = B - R
RTLD = R
300 continue
rho = dotprodfun( HUTI_NDIM, RTLD, 1, R, 1 )
if ( rho .eq. 0 ) then
HUTI_INFO = HUTI_CGS_RHO
go to 1000
end if
if ( iter_count .eq. 1 ) then
U = R
P = U
else
beta = rho / oldrho
U = R + beta * Q
P = U + beta * Q + beta * beta * P
end if
call pcondlsubr( T2V, P, ipar )
call pcondrsubr( T1V, T2V, ipar )
call matvecsubr( T1V, T2V, ipar )
alpha = rho / dotprodfun( HUTI_NDIM, RTLD, 1, T2V, 1 )
Q = U - alpha * T2V
T2V = U + Q
call pcondlsubr( U, T2V, ipar )
call pcondrsubr( T1V, U, ipar )
X = X + alpha * T1V
call matvecsubr( T1V, T2V, ipar )
R = R - alpha * T2V
select case (HUTI_STOPC)
case (HUTI_TRUERESIDUAL)
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 )
case (HUTI_TRESID_SCALED_BYB)
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 ) / rhsnorm
case (HUTI_PSEUDORESIDUAL)
residual = normfun( HUTI_NDIM, R, 1 )
case (HUTI_PRESID_SCALED_BYB)
residual = normfun( HUTI_NDIM, R, 1 ) / rhsnorm
case (HUTI_PRESID_SCALED_BYPRECB)
residual = normfun( HUTI_NDIM, R, 1 ) / precrhsnorm
case (HUTI_XDIFF_NORM)
T1V = alpha * T1V
residual = normfun( HUTI_NDIM, T1V, 1 )
case (HUTI_USUPPLIED_STOPC)
residual = stopcfun( X, B, R, ipar, dpar )
case default
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 )
end select
if ( HUTI_DBUGLVL .ne. HUTI_NO_DEBUG ) then
if ( mod(iter_count, HUTI_DBUGLVL) .eq. 0 ) then
write (*, '(I8, E11.4)') iter_count, residual
end if
end if
if ( residual .lt. HUTI_TOLERANCE ) then
HUTI_INFO = HUTI_CONVERGENCE
go to 1000
end if
IF( residual /= residual .OR. residual > HUTI_MAXTOLERANCE ) THEN
HUTI_INFO = HUTI_DIVERGENCE
GOTO 1000
END IF
oldrho = rho
iter_count = iter_count + 1
if ( iter_count .gt. HUTI_MAXIT ) then
HUTI_INFO = HUTI_MAXITER
go to 1000
end if
go to 300
1000 continue
if ( HUTI_DBUGLVL .ne. HUTI_NO_DEBUG ) then
write (*, '(I8, E11.4)') iter_count, residual
end if
HUTI_ITERS = iter_count
return
end subroutine huti_ccgssolv
subroutine huti_zcgssolv ( ndim, wrkdim, xvec, rhsvec, ipar,&
dpar, work, matvecsubr, pcondlsubr, pcondrsubr, &
dotprodfun, normfun, stopcfun )
use huti_interfaces
implicit none
procedure( mv_iface_z ), pointer :: matvecsubr
procedure( pc_iface_z ), pointer :: pcondlsubr
procedure( pc_iface_z ), pointer :: pcondrsubr
procedure( dotp_iface_z ), pointer :: dotprodfun
procedure( norm_iface_z ), pointer :: normfun
procedure( stopc_iface_z ), pointer :: stopcfun
integer :: ndim, wrkdim
double complex, dimension(ndim) :: xvec, rhsvec
integer, dimension(HUTI_IPAR_DFLTSIZE) :: ipar
double precision, dimension(HUTI_DPAR_DFLTSIZE) :: dpar
double complex, dimension(ndim,wrkdim) :: work
double complex :: rho, oldrho, alpha, beta
integer :: iter_count
double precision :: residual, rhsnorm, precrhsnorm
iter_count = 1
HUTI_EXTOP_MATTYPE = HUTI_MAT_NOTTRPSED
if ( HUTI_STOPC .eq. HUTI_TRESID_SCALED_BYB .or. &
HUTI_STOPC .eq. HUTI_PRESID_SCALED_BYB ) then
rhsnorm = normfun( HUTI_NDIM, B, 1 )
end if
if ( HUTI_STOPC .eq. HUTI_PRESID_SCALED_BYPRECB ) then
call pcondlsubr( P, B, ipar )
precrhsnorm = normfun( HUTI_NDIM, P, 1 )
end if
if ( HUTI_INITIALX .eq. HUTI_RANDOMX ) then
call huti_zrandvec ( X, ipar )
else if ( HUTI_INITIALX .ne. HUTI_USERSUPPLIEDX ) then
X = 1
end if
call matvecsubr( X, R, ipar )
R = B - R
RTLD = R
300 continue
rho = dotprodfun( HUTI_NDIM, RTLD, 1, R, 1 )
if ( rho .eq. 0 ) then
HUTI_INFO = HUTI_CGS_RHO
go to 1000
end if
if ( iter_count .eq. 1 ) then
U = R
P = U
else
beta = rho / oldrho
U = R + beta * Q
P = U + beta * Q + beta * beta * P
end if
call pcondlsubr( T2V, P, ipar )
call pcondrsubr( T1V, T2V, ipar )
call matvecsubr( T1V, T2V, ipar )
alpha = rho / dotprodfun( HUTI_NDIM, RTLD, 1, T2V, 1 )
Q = U - alpha * T2V
T2V = U + Q
call pcondlsubr( U, T2V, ipar )
call pcondrsubr( T1V, U, ipar )
X = X + alpha * T1V
call matvecsubr( T1V, T2V, ipar )
R = R - alpha * T2V
select case (HUTI_STOPC)
case (HUTI_TRUERESIDUAL)
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 )
case (HUTI_TRESID_SCALED_BYB)
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 ) / rhsnorm
case (HUTI_PSEUDORESIDUAL)
residual = normfun( HUTI_NDIM, R, 1 )
case (HUTI_PRESID_SCALED_BYB)
residual = normfun( HUTI_NDIM, R, 1 ) / rhsnorm
case (HUTI_PRESID_SCALED_BYPRECB)
residual = normfun( HUTI_NDIM, R, 1 ) / precrhsnorm
case (HUTI_XDIFF_NORM)
T1V = alpha * T1V
residual = normfun( HUTI_NDIM, T1V, 1 )
case (HUTI_USUPPLIED_STOPC)
residual = stopcfun( X, B, R, ipar, dpar )
case default
call matvecsubr( X, T1V, ipar )
T1V = T1V - B
residual = normfun( HUTI_NDIM, T1V, 1 )
end select
if ( HUTI_DBUGLVL .ne. HUTI_NO_DEBUG ) then
if ( mod(iter_count, HUTI_DBUGLVL) .eq. 0 ) then
write (*, '(I8, E11.4)') iter_count, residual
end if
end if
if ( residual .lt. HUTI_TOLERANCE ) then
HUTI_INFO = HUTI_CONVERGENCE
go to 1000
end if
IF( residual /= residual .OR. residual > HUTI_MAXTOLERANCE ) THEN
HUTI_INFO = HUTI_DIVERGENCE
GOTO 1000
END IF
oldrho = rho
iter_count = iter_count + 1
if ( iter_count .gt. HUTI_MAXIT ) then
HUTI_INFO = HUTI_MAXITER
go to 1000
end if
go to 300
1000 continue
if ( HUTI_DBUGLVL .ne. HUTI_NO_DEBUG ) then
write (*, '(I8, E11.4)') iter_count, residual
end if
HUTI_ITERS = iter_count
return
end subroutine huti_zcgssolv
end module huti_cgs