You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
584 lines
16 KiB
Fortran
584 lines
16 KiB
Fortran
module psb_z_base_vect_mod
|
|
|
|
use psb_const_mod
|
|
use psb_error_mod
|
|
|
|
type psb_z_base_vect_type
|
|
complex(psb_dpk_), allocatable :: v(:)
|
|
contains
|
|
procedure, pass(x) :: get_nrows => z_base_get_nrows
|
|
procedure, pass(x) :: dot_v => z_base_dot_v
|
|
procedure, pass(x) :: dot_a => z_base_dot_a
|
|
generic, public :: dot => dot_v, dot_a
|
|
procedure, pass(y) :: axpby_v => z_base_axpby_v
|
|
procedure, pass(y) :: axpby_a => z_base_axpby_a
|
|
generic, public :: axpby => axpby_v, axpby_a
|
|
procedure, pass(y) :: mlt_v => z_base_mlt_v
|
|
procedure, pass(y) :: mlt_a => z_base_mlt_a
|
|
procedure, pass(z) :: mlt_a_2 => z_base_mlt_a_2
|
|
procedure, pass(z) :: mlt_v_2 => z_base_mlt_v_2
|
|
procedure, pass(z) :: mlt_va => z_base_mlt_va
|
|
procedure, pass(z) :: mlt_av => z_base_mlt_av
|
|
generic, public :: mlt => mlt_v, mlt_a, mlt_a_2, mlt_v_2, mlt_av, mlt_va
|
|
procedure, pass(x) :: scal => z_base_scal
|
|
procedure, pass(x) :: nrm2 => z_base_nrm2
|
|
procedure, pass(x) :: amax => z_base_amax
|
|
procedure, pass(x) :: asum => z_base_asum
|
|
procedure, pass(x) :: all => z_base_all
|
|
procedure, pass(x) :: zero => z_base_zero
|
|
procedure, pass(x) :: asb => z_base_asb
|
|
procedure, pass(x) :: sync => z_base_sync
|
|
procedure, pass(x) :: gthab => z_base_gthab
|
|
procedure, pass(x) :: gthzv => z_base_gthzv
|
|
generic, public :: gth => gthab, gthzv
|
|
procedure, pass(y) :: sctb => z_base_sctb
|
|
generic, public :: sct => sctb
|
|
procedure, pass(x) :: free => z_base_free
|
|
procedure, pass(x) :: ins => z_base_ins
|
|
procedure, pass(x) :: bld_x => z_base_bld_x
|
|
procedure, pass(x) :: bld_n => z_base_bld_n
|
|
generic, public :: bld => bld_x, bld_n
|
|
procedure, pass(x) :: getCopy => z_base_getCopy
|
|
procedure, pass(x) :: cpy_vect => z_base_cpy_vect
|
|
generic, public :: assignment(=) => cpy_vect, set_scal
|
|
procedure, pass(x) :: set_scal => z_base_set_scal
|
|
procedure, pass(x) :: set_vect => z_base_set_vect
|
|
generic, public :: set => set_vect, set_scal
|
|
end type psb_z_base_vect_type
|
|
|
|
public :: psb_z_base_vect
|
|
private :: constructor, size_const
|
|
interface psb_z_base_vect
|
|
module procedure constructor, size_const
|
|
end interface psb_z_base_vect
|
|
|
|
contains
|
|
|
|
subroutine z_base_bld_x(x,this)
|
|
use psb_realloc_mod
|
|
complex(psb_dpk_), intent(in) :: this(:)
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
integer :: info
|
|
|
|
call psb_realloc(size(this),x%v,info)
|
|
if (info /= 0) then
|
|
call psb_errpush(psb_err_alloc_dealloc_,'base_vect_bld')
|
|
return
|
|
end if
|
|
x%v(:) = this(:)
|
|
|
|
end subroutine z_base_bld_x
|
|
|
|
|
|
subroutine z_base_bld_n(x,n)
|
|
integer, intent(in) :: n
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
integer :: info
|
|
|
|
call x%asb(n,info)
|
|
|
|
end subroutine z_base_bld_n
|
|
|
|
function z_base_getCopy(x) result(res)
|
|
class(psb_z_base_vect_type), intent(in) :: x
|
|
complex(psb_dpk_), allocatable :: res(:)
|
|
integer :: info
|
|
|
|
allocate(res(x%get_nrows()),stat=info)
|
|
if (info /= 0) then
|
|
call psb_errpush(psb_err_alloc_dealloc_,'base_getCopy')
|
|
return
|
|
end if
|
|
res(:) = x%v(:)
|
|
end function z_base_getCopy
|
|
|
|
subroutine z_base_cpy_vect(res,x)
|
|
complex(psb_dpk_), allocatable, intent(out) :: res(:)
|
|
class(psb_z_base_vect_type), intent(in) :: x
|
|
integer :: info
|
|
|
|
res = x%v
|
|
|
|
end subroutine z_base_cpy_vect
|
|
|
|
subroutine z_base_set_scal(x,val)
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
complex(psb_dpk_), intent(in) :: val
|
|
|
|
integer :: info
|
|
x%v = val
|
|
|
|
end subroutine z_base_set_scal
|
|
|
|
subroutine z_base_set_vect(x,val)
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
complex(psb_dpk_), intent(in) :: val(:)
|
|
integer :: nr
|
|
integer :: info
|
|
if (allocated(x%v)) then
|
|
nr = min(size(x%v),size(val))
|
|
x%v(1:nr) = val(1:nr)
|
|
else
|
|
x%v = val
|
|
end if
|
|
end subroutine z_base_set_vect
|
|
|
|
|
|
|
|
function constructor(x) result(this)
|
|
complex(psb_dpk_) :: x(:)
|
|
type(psb_z_base_vect_type) :: this
|
|
integer :: info
|
|
|
|
this%v = x
|
|
call this%asb(size(x),info)
|
|
end function constructor
|
|
|
|
|
|
function size_const(n) result(this)
|
|
integer, intent(in) :: n
|
|
type(psb_z_base_vect_type) :: this
|
|
integer :: info
|
|
|
|
call this%asb(n,info)
|
|
|
|
end function size_const
|
|
|
|
|
|
function z_base_get_nrows(x) result(res)
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(in) :: x
|
|
integer :: res
|
|
res = -1
|
|
if (allocated(x%v)) res = size(x%v)
|
|
end function z_base_get_nrows
|
|
|
|
function z_base_dot_v(n,x,y) result(res)
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x, y
|
|
integer, intent(in) :: n
|
|
complex(psb_dpk_) :: res
|
|
complex(psb_dpk_), external :: zdotc
|
|
|
|
res = zzero
|
|
!
|
|
! Note: this is the base implementation.
|
|
! When we get here, we are sure that X is of
|
|
! TYPE psb_z_base_vect
|
|
!
|
|
select type(yy => y)
|
|
type is (psb_z_base_vect_type)
|
|
res = zdotc(n,x%v,1,y%v,1)
|
|
class default
|
|
res = y%dot(n,x%v)
|
|
end select
|
|
|
|
end function z_base_dot_v
|
|
|
|
function z_base_dot_a(n,x,y) result(res)
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
complex(psb_dpk_), intent(in) :: y(:)
|
|
integer, intent(in) :: n
|
|
complex(psb_dpk_) :: res
|
|
complex(psb_dpk_), external :: zdotc
|
|
|
|
res = zdotc(n,y,1,x%v,1)
|
|
|
|
end function z_base_dot_a
|
|
|
|
subroutine z_base_axpby_v(m,alpha, x, beta, y, info)
|
|
use psi_serial_mod
|
|
implicit none
|
|
integer, intent(in) :: m
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
class(psb_z_base_vect_type), intent(inout) :: y
|
|
complex(psb_dpk_), intent (in) :: alpha, beta
|
|
integer, intent(out) :: info
|
|
|
|
select type(xx => x)
|
|
type is (psb_z_base_vect_type)
|
|
call psb_geaxpby(m,alpha,x%v,beta,y%v,info)
|
|
class default
|
|
call y%axpby(m,alpha,x%v,beta,info)
|
|
end select
|
|
|
|
end subroutine z_base_axpby_v
|
|
|
|
subroutine z_base_axpby_a(m,alpha, x, beta, y, info)
|
|
use psi_serial_mod
|
|
implicit none
|
|
integer, intent(in) :: m
|
|
complex(psb_dpk_), intent(in) :: x(:)
|
|
class(psb_z_base_vect_type), intent(inout) :: y
|
|
complex(psb_dpk_), intent (in) :: alpha, beta
|
|
integer, intent(out) :: info
|
|
|
|
call psb_geaxpby(m,alpha,x,beta,y%v,info)
|
|
|
|
end subroutine z_base_axpby_a
|
|
|
|
|
|
subroutine z_base_mlt_v(x, y, info)
|
|
use psi_serial_mod
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
class(psb_z_base_vect_type), intent(inout) :: y
|
|
integer, intent(out) :: info
|
|
integer :: i, n
|
|
|
|
info = 0
|
|
select type(xx => x)
|
|
type is (psb_z_base_vect_type)
|
|
n = min(size(y%v), size(xx%v))
|
|
do i=1, n
|
|
y%v(i) = y%v(i)*xx%v(i)
|
|
end do
|
|
class default
|
|
call y%mlt(x%v,info)
|
|
end select
|
|
|
|
end subroutine z_base_mlt_v
|
|
|
|
subroutine z_base_mlt_a(x, y, info)
|
|
use psi_serial_mod
|
|
implicit none
|
|
complex(psb_dpk_), intent(in) :: x(:)
|
|
class(psb_z_base_vect_type), intent(inout) :: y
|
|
integer, intent(out) :: info
|
|
integer :: i, n
|
|
|
|
info = 0
|
|
n = min(size(y%v), size(x))
|
|
do i=1, n
|
|
y%v(i) = y%v(i)*x(i)
|
|
end do
|
|
|
|
end subroutine z_base_mlt_a
|
|
|
|
|
|
subroutine z_base_mlt_a_2(alpha,x,y,beta,z,info)
|
|
use psi_serial_mod
|
|
implicit none
|
|
complex(psb_dpk_), intent(in) :: alpha,beta
|
|
complex(psb_dpk_), intent(in) :: y(:)
|
|
complex(psb_dpk_), intent(in) :: x(:)
|
|
class(psb_z_base_vect_type), intent(inout) :: z
|
|
integer, intent(out) :: info
|
|
integer :: i, n
|
|
|
|
info = 0
|
|
n = min(size(z%v), size(x), size(y))
|
|
!!$ write(0,*) 'Mlt_a_2: ',n
|
|
if (alpha == zzero) then
|
|
if (beta == zone) then
|
|
return
|
|
else
|
|
do i=1, n
|
|
z%v(i) = beta*z%v(i)
|
|
end do
|
|
end if
|
|
else
|
|
if (alpha == zone) then
|
|
if (beta == zzero) then
|
|
do i=1, n
|
|
z%v(i) = y(i)*x(i)
|
|
end do
|
|
else if (beta == zone) then
|
|
do i=1, n
|
|
z%v(i) = z%v(i) + y(i)*x(i)
|
|
end do
|
|
else
|
|
do i=1, n
|
|
z%v(i) = beta*z%v(i) + y(i)*x(i)
|
|
end do
|
|
end if
|
|
else if (alpha == -zone) then
|
|
if (beta == zzero) then
|
|
do i=1, n
|
|
z%v(i) = -y(i)*x(i)
|
|
end do
|
|
else if (beta == zone) then
|
|
do i=1, n
|
|
z%v(i) = z%v(i) - y(i)*x(i)
|
|
end do
|
|
else
|
|
do i=1, n
|
|
z%v(i) = beta*z%v(i) - y(i)*x(i)
|
|
end do
|
|
end if
|
|
else
|
|
if (beta == zzero) then
|
|
do i=1, n
|
|
z%v(i) = alpha*y(i)*x(i)
|
|
end do
|
|
else if (beta == zone) then
|
|
do i=1, n
|
|
z%v(i) = z%v(i) + alpha*y(i)*x(i)
|
|
end do
|
|
else
|
|
do i=1, n
|
|
z%v(i) = beta*z%v(i) + alpha*y(i)*x(i)
|
|
end do
|
|
end if
|
|
end if
|
|
end if
|
|
end subroutine z_base_mlt_a_2
|
|
|
|
subroutine z_base_mlt_v_2(alpha,x,y,beta,z,info,conjgx,conjgy)
|
|
use psi_serial_mod
|
|
use psb_string_mod
|
|
implicit none
|
|
complex(psb_dpk_), intent(in) :: alpha,beta
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
class(psb_z_base_vect_type), intent(inout) :: y
|
|
class(psb_z_base_vect_type), intent(inout) :: z
|
|
integer, intent(out) :: info
|
|
character(len=1), intent(in), optional :: conjgx, conjgy
|
|
integer :: i, n
|
|
|
|
info = 0
|
|
if (present(conjgx)) then
|
|
if (psb_toupper(conjgx)=='C') x%v=conjg(x%v)
|
|
end if
|
|
if (present(conjgy)) then
|
|
if (psb_toupper(conjgy)=='C') y%v=conjg(y%v)
|
|
end if
|
|
call z%mlt(alpha,x%v,y%v,beta,info)
|
|
if (present(conjgx)) then
|
|
if (psb_toupper(conjgx)=='C') x%v=conjg(x%v)
|
|
end if
|
|
if (present(conjgy)) then
|
|
if (psb_toupper(conjgy)=='C') y%v=conjg(y%v)
|
|
end if
|
|
|
|
end subroutine z_base_mlt_v_2
|
|
|
|
subroutine z_base_mlt_av(alpha,x,y,beta,z,info)
|
|
use psi_serial_mod
|
|
implicit none
|
|
complex(psb_dpk_), intent(in) :: alpha,beta
|
|
complex(psb_dpk_), intent(in) :: x(:)
|
|
class(psb_z_base_vect_type), intent(inout) :: y
|
|
class(psb_z_base_vect_type), intent(inout) :: z
|
|
integer, intent(out) :: info
|
|
integer :: i, n
|
|
|
|
info = 0
|
|
|
|
call z%mlt(alpha,x,y%v,beta,info)
|
|
|
|
end subroutine z_base_mlt_av
|
|
|
|
subroutine z_base_mlt_va(alpha,x,y,beta,z,info)
|
|
use psi_serial_mod
|
|
implicit none
|
|
complex(psb_dpk_), intent(in) :: alpha,beta
|
|
complex(psb_dpk_), intent(in) :: y(:)
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
class(psb_z_base_vect_type), intent(inout) :: z
|
|
integer, intent(out) :: info
|
|
integer :: i, n
|
|
|
|
info = 0
|
|
|
|
call z%mlt(alpha,y,x,beta,info)
|
|
|
|
end subroutine z_base_mlt_va
|
|
|
|
subroutine z_base_scal(alpha, x)
|
|
use psi_serial_mod
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
complex(psb_dpk_), intent (in) :: alpha
|
|
|
|
if (allocated(x%v)) x%v = alpha*x%v
|
|
|
|
end subroutine z_base_scal
|
|
|
|
|
|
function z_base_nrm2(n,x) result(res)
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
integer, intent(in) :: n
|
|
real(psb_dpk_) :: res
|
|
real(psb_dpk_), external :: dznrm2
|
|
|
|
res = dznrm2(n,x%v,1)
|
|
|
|
end function z_base_nrm2
|
|
|
|
function z_base_amax(n,x) result(res)
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
integer, intent(in) :: n
|
|
real(psb_dpk_) :: res
|
|
|
|
res = maxval(abs(x%v(1:n)))
|
|
|
|
end function z_base_amax
|
|
|
|
function z_base_asum(n,x) result(res)
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
integer, intent(in) :: n
|
|
real(psb_dpk_) :: res
|
|
|
|
res = sum(abs(x%v(1:n)))
|
|
|
|
end function z_base_asum
|
|
|
|
subroutine z_base_all(n, x, info)
|
|
use psi_serial_mod
|
|
use psb_realloc_mod
|
|
implicit none
|
|
integer, intent(in) :: n
|
|
class(psb_z_base_vect_type), intent(out) :: x
|
|
integer, intent(out) :: info
|
|
|
|
call psb_realloc(n,x%v,info)
|
|
|
|
end subroutine z_base_all
|
|
|
|
subroutine z_base_zero(x)
|
|
use psi_serial_mod
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
|
|
if (allocated(x%v)) x%v=zzero
|
|
|
|
end subroutine z_base_zero
|
|
|
|
subroutine z_base_asb(n, x, info)
|
|
use psi_serial_mod
|
|
use psb_realloc_mod
|
|
implicit none
|
|
integer, intent(in) :: n
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
integer, intent(out) :: info
|
|
|
|
if (x%get_nrows() < n) &
|
|
& call psb_realloc(n,x%v,info)
|
|
if (info /= 0) &
|
|
& call psb_errpush(psb_err_alloc_dealloc_,'vect_asb')
|
|
|
|
end subroutine z_base_asb
|
|
|
|
subroutine z_base_sync(x)
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
|
|
!
|
|
! The base version does nothing, it's just
|
|
! a placeholder.
|
|
!
|
|
|
|
end subroutine z_base_sync
|
|
|
|
subroutine z_base_gthab(n,idx,alpha,x,beta,y)
|
|
use psi_serial_mod
|
|
integer :: n, idx(:)
|
|
complex(psb_dpk_) :: alpha, beta, y(:)
|
|
class(psb_z_base_vect_type) :: x
|
|
|
|
call x%sync()
|
|
call psi_gth(n,idx,alpha,x%v,beta,y)
|
|
|
|
end subroutine z_base_gthab
|
|
|
|
subroutine z_base_gthzv(n,idx,x,y)
|
|
use psi_serial_mod
|
|
integer :: n, idx(:)
|
|
complex(psb_dpk_) :: y(:)
|
|
class(psb_z_base_vect_type) :: x
|
|
|
|
call x%sync()
|
|
call psi_gth(n,idx,x%v,y)
|
|
|
|
end subroutine z_base_gthzv
|
|
|
|
subroutine z_base_sctb(n,idx,x,beta,y)
|
|
use psi_serial_mod
|
|
integer :: n, idx(:)
|
|
complex(psb_dpk_) :: beta, x(:)
|
|
class(psb_z_base_vect_type) :: y
|
|
|
|
call y%sync()
|
|
call psi_sct(n,idx,x,beta,y%v)
|
|
|
|
end subroutine z_base_sctb
|
|
|
|
subroutine z_base_free(x, info)
|
|
use psi_serial_mod
|
|
use psb_realloc_mod
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
integer, intent(out) :: info
|
|
|
|
info = 0
|
|
if (allocated(x%v)) deallocate(x%v, stat=info)
|
|
if (info /= 0) call &
|
|
& psb_errpush(psb_err_alloc_dealloc_,'vect_free')
|
|
|
|
end subroutine z_base_free
|
|
|
|
subroutine z_base_ins(n,irl,val,dupl,x,info)
|
|
use psi_serial_mod
|
|
implicit none
|
|
class(psb_z_base_vect_type), intent(inout) :: x
|
|
integer, intent(in) :: n, dupl
|
|
integer, intent(in) :: irl(:)
|
|
complex(psb_dpk_), intent(in) :: val(:)
|
|
integer, intent(out) :: info
|
|
|
|
integer :: i
|
|
|
|
info = 0
|
|
if (psb_errstatus_fatal()) return
|
|
|
|
if (.not.allocated(x%v)) then
|
|
info = psb_err_invalid_vect_state_
|
|
else if (n > min(size(irl),size(val))) then
|
|
info = psb_err_invalid_input_
|
|
|
|
else
|
|
select case(dupl)
|
|
case(psb_dupl_ovwrt_)
|
|
do i = 1, n
|
|
!loop over all val's rows
|
|
|
|
! row actual block row
|
|
if (irl(i) > 0) then
|
|
! this row belongs to me
|
|
! copy i-th row of block val in x
|
|
x%v(irl(i)) = val(i)
|
|
end if
|
|
enddo
|
|
|
|
case(psb_dupl_add_)
|
|
|
|
do i = 1, n
|
|
!loop over all val's rows
|
|
|
|
if (irl(i) > 0) then
|
|
! this row belongs to me
|
|
! copy i-th row of block val in x
|
|
x%v(irl(i)) = x%v(irl(i)) + val(i)
|
|
end if
|
|
enddo
|
|
|
|
case default
|
|
info = 321
|
|
!!$ call psb_errpush(info,name)
|
|
!!$ goto 9999
|
|
end select
|
|
end if
|
|
if (info /= 0) then
|
|
call psb_errpush(info,'base_vect_ins')
|
|
return
|
|
end if
|
|
|
|
end subroutine z_base_ins
|
|
|
|
end module psb_z_base_vect_mod
|