!!$ !!$ Parallel Sparse BLAS v2.0 !!$ (C) Copyright 2006 Salvatore Filippone University of Rome Tor Vergata !!$ Alfredo Buttari University of Rome Tor Vergata !!$ !!$ Redistribution and use in source and binary forms, with or without !!$ modification, are permitted provided that the following conditions !!$ are met: !!$ 1. Redistributions of source code must retain the above copyright !!$ notice, this list of conditions and the following disclaimer. !!$ 2. Redistributions in binary form must reproduce the above copyright !!$ notice, this list of conditions, and the following disclaimer in the !!$ documentation and/or other materials provided with the distribution. !!$ 3. The name of the PSBLAS group or the names of its contributors may !!$ not be used to endorse or promote products derived from this !!$ software without specific written permission. !!$ !!$ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS !!$ ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED !!$ TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR !!$ PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE PSBLAS GROUP OR ITS CONTRIBUTORS !!$ BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR !!$ CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF !!$ SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS !!$ INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN !!$ CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) !!$ ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE !!$ POSSIBILITY OF SUCH DAMAGE. !!$ !!$ module psb_sort_mod integer, parameter :: psb_sort_up_=1, psb_sort_down_=-1 integer, parameter :: psb_lsort_up_=2, psb_lsort_down_=-2 integer, parameter :: psb_asort_up_=3, psb_asort_down_=-3 integer, parameter :: psb_alsort_up_=4, psb_alsort_down_=-4 integer, parameter :: psb_sort_ovw_idx_=0, psb_sort_keep_idx_=1 integer, parameter :: psb_heap_resize=200 type psb_int_heap integer :: last, dir integer, allocatable :: keys(:) end type psb_int_heap interface psb_msort module procedure imsort end interface interface psb_msort_unique module procedure imsort_u end interface interface psb_qsort module procedure iqsort, dqsort, zqsort end interface interface psb_init_heap module procedure psb_init_int_heap end interface interface psb_dump_heap module procedure psb_dump_int_heap end interface interface psb_howmany_heap module procedure psb_howmany_int_heap end interface interface psb_insert_heap module procedure psb_insert_int_heap end interface interface psb_heap_get_first module procedure psb_int_heap_get_first end interface contains subroutine imsort(x,ix,dir,flag) use psb_error_mod implicit none integer, intent(inout) :: x(:) integer, optional, intent(in) :: dir, flag integer, optional, intent(inout) :: ix(:) integer :: dir_, flag_, n, err_act character(len=20) :: name name='psb_msort' call psb_erractionsave(err_act) if (present(dir)) then dir_ = dir else dir_= psb_sort_up_ end if select case(dir_) case( psb_sort_up_, psb_sort_down_) ! OK keep going case default call psb_errpush(30,name,i_err=(/3,dir_,0,0,0/)) goto 9999 end select n = size(x) if (present(ix)) then if (size(ix) < n) then call psb_errpush(35,name,i_err=(/2,size(ix),0,0,0/)) goto 9999 end if if (present(flag)) then flag_ = flag else flag_ = psb_sort_ovw_idx_ end if select case(flag_) case( psb_sort_ovw_idx_, psb_sort_keep_idx_) ! OK keep going case default call psb_errpush(30,name,i_err=(/4,flag_,0,0,0/)) goto 9999 end select call imsrx(n,x,ix,dir_,flag_) else call imsr(n,x,dir_) end if 9999 continue if (err_act.eq.psb_act_abort_) then call psb_error() return end if end subroutine imsort subroutine imsort_u(x,nout,dir) use psb_error_mod implicit none integer, intent(inout) :: x(:) integer, intent(out) :: nout integer, optional, intent(in) :: dir integer :: dir_, flag_, n, err_act character(len=20) :: name name='psb_msort_u' call psb_erractionsave(err_act) if (present(dir)) then dir_ = dir else dir_= psb_sort_up_ end if select case(dir_) case( psb_sort_up_, psb_sort_down_) ! OK keep going case default call psb_errpush(30,name,i_err=(/3,dir_,0,0,0/)) goto 9999 end select n = size(x) call imsru(n,x,dir_,nout) 9999 continue if (err_act.eq.psb_act_abort_) then call psb_error() return end if end subroutine imsort_u subroutine iqsort(x,ix,dir,flag) use psb_error_mod implicit none integer, intent(inout) :: x(:) integer, optional, intent(in) :: dir, flag integer, optional, intent(inout) :: ix(:) integer :: dir_, flag_, n, err_act character(len=20) :: name name='psb_qsort' call psb_erractionsave(err_act) if (present(flag)) then flag_ = flag else flag_ = psb_sort_ovw_idx_ end if select case(flag_) case( psb_sort_ovw_idx_, psb_sort_keep_idx_) ! OK keep going case default call psb_errpush(30,name,i_err=(/4,flag_,0,0,0/)) goto 9999 end select if (present(dir)) then dir_ = dir else dir_= psb_sort_up_ end if n = size(x) select case(dir_) case( psb_sort_up_, psb_sort_down_) if (present(ix)) then if (size(ix) < n) then call psb_errpush(35,name,i_err=(/2,size(ix),0,0,0/)) goto 9999 end if call isrx(n,x,ix,dir_,flag_) else call isr(n,x,dir_) end if case( psb_asort_up_, psb_asort_down_) ! OK keep going if (present(ix)) then if (size(ix) < n) then call psb_errpush(35,name,i_err=(/2,size(ix),0,0,0/)) goto 9999 end if call iasrx(n,x,ix,dir_,flag_) else call iasr(n,x,dir_) end if case default call psb_errpush(30,name,i_err=(/3,dir_,0,0,0/)) goto 9999 end select 9999 continue if (err_act.eq.psb_act_abort_) then call psb_error() return end if end subroutine iqsort subroutine dqsort(x,ix,dir,flag) use psb_error_mod implicit none real(kind(1.d0)), intent(inout) :: x(:) integer, optional, intent(in) :: dir, flag integer, optional, intent(inout) :: ix(:) integer :: dir_, flag_, n, err_act character(len=20) :: name name='psb_qsort' call psb_erractionsave(err_act) if (present(flag)) then flag_ = flag else flag_ = psb_sort_ovw_idx_ end if select case(flag_) case( psb_sort_ovw_idx_, psb_sort_keep_idx_) ! OK keep going case default call psb_errpush(30,name,i_err=(/4,flag_,0,0,0/)) goto 9999 end select if (present(dir)) then dir_ = dir else dir_= psb_sort_up_ end if n = size(x) select case(dir_) case( psb_sort_up_, psb_sort_down_) if (present(ix)) then if (size(ix) < n) then call psb_errpush(35,name,i_err=(/2,size(ix),0,0,0/)) goto 9999 end if call dsrx(n,x,ix,dir_,flag_) else call dsr(n,x,dir_) end if case( psb_asort_up_, psb_asort_down_) ! OK keep going if (present(ix)) then if (size(ix) < n) then call psb_errpush(35,name,i_err=(/2,size(ix),0,0,0/)) goto 9999 end if call dasrx(n,x,ix,dir_,flag_) else call dasr(n,x,dir_) end if case default call psb_errpush(30,name,i_err=(/3,dir_,0,0,0/)) goto 9999 end select 9999 continue if (err_act.eq.psb_act_abort_) then call psb_error() return end if end subroutine dqsort subroutine zqsort(x,ix,dir,flag) use psb_error_mod implicit none complex(kind(1.d0)), intent(inout) :: x(:) integer, optional, intent(in) :: dir, flag integer, optional, intent(inout) :: ix(:) integer :: dir_, flag_, n, err_act character(len=20) :: name name='psb_qsort' call psb_erractionsave(err_act) if (present(flag)) then flag_ = flag else flag_ = psb_sort_ovw_idx_ end if select case(flag_) case( psb_sort_ovw_idx_, psb_sort_keep_idx_) ! OK keep going case default call psb_errpush(30,name,i_err=(/4,flag_,0,0,0/)) goto 9999 end select if (present(dir)) then dir_ = dir else dir_= psb_lsort_up_ end if n = size(x) select case(dir_) case( psb_lsort_up_, psb_lsort_down_) if (present(ix)) then if (size(ix) < n) then call psb_errpush(35,name,i_err=(/2,size(ix),0,0,0/)) goto 9999 end if call zlsrx(n,x,ix,dir_,flag_) else call zlsr(n,x,dir_) end if case( psb_alsort_up_, psb_alsort_down_) ! OK keep going if (present(ix)) then if (size(ix) < n) then call psb_errpush(35,name,i_err=(/2,size(ix),0,0,0/)) goto 9999 end if call zalsrx(n,x,ix,dir_,flag_) else call zalsr(n,x,dir_) end if case( psb_asort_up_, psb_asort_down_) ! OK keep going if (present(ix)) then if (size(ix) < n) then call psb_errpush(35,name,i_err=(/2,size(ix),0,0,0/)) goto 9999 end if call zasrx(n,x,ix,dir_,flag_) else call zasr(n,x,dir_) end if case default call psb_errpush(30,name,i_err=(/3,dir_,0,0,0/)) goto 9999 end select 9999 continue if (err_act.eq.psb_act_abort_) then call psb_error() return end if end subroutine zqsort function psb_howmany_int_heap(heap) implicit none type(psb_int_heap), intent(in) :: heap integer :: psb_howmany_int_heap psb_howmany_int_heap = heap%last end function psb_howmany_int_heap subroutine psb_init_int_heap(heap,info,dir) use psb_realloc_mod implicit none type(psb_int_heap), intent(inout) :: heap integer, intent(out) :: info integer, intent(in), optional :: dir info = 0 heap%last=0 if (present(dir)) then heap%dir = dir else heap%dir = psb_sort_up_ endif select case(heap%dir) case (psb_sort_up_,psb_sort_down_) ! ok, do nothing case default write(0,*) 'Invalid direction, defaulting to psb_sort_up_' heap%dir = psb_sort_up_ end select call psb_ensure_size(psb_heap_resize,heap%keys,info) return end subroutine psb_init_int_heap subroutine psb_dump_int_heap(iout,heap,info) implicit none type(psb_int_heap), intent(in) :: heap integer, intent(out) :: info integer, intent(in) :: iout info = 0 if (iout < 0) then write(0,*) 'Invalid file ' info =-1 return end if write(iout,*) 'Heap direction ',heap%dir write(iout,*) 'Heap size ',heap%last if ((heap%last > 0).and.((.not.allocated(heap%keys)).or.& & (size(heap%keys) 0) then write(iout,*) heap%keys(1:heap%last) end if end if end subroutine psb_dump_int_heap subroutine psb_insert_int_heap(key,heap,info) use psb_realloc_mod implicit none integer, intent(in) :: key type(psb_int_heap), intent(inout) :: heap integer, intent(out) :: info integer :: i, i2, itemp info = 0 if (heap%last < 0) then write(0,*) 'Invalid last in heap ',heap%last info = heap%last return endif select case(heap%dir) case (psb_sort_up_) heap%last = heap%last + 1 if (heap%last > size(heap%keys)) then call psb_realloc(heap%last+psb_heap_resize,heap%keys,info) if (info /= 0) then write(0,*) 'Memory allocation failure' return end if end if i = heap%last heap%keys(i) = key do if (i<=1) exit i2 = i/2 if (heap%keys(i) < heap%keys(i2)) then itemp = heap%keys(i) heap%keys(i) = heap%keys(i2) heap%keys(i2) = itemp i = i2 else exit end if end do case (psb_sort_down_) heap%last = heap%last + 1 if (heap%last > size(heap%keys)) then call psb_realloc(heap%last+psb_heap_resize,heap%keys,info) if (info /= 0) then write(0,*) 'Memory allocation failure' return end if end if i = heap%last heap%keys(i) = key do if (i<=1) exit i2 = i/2 if (heap%keys(i) > heap%keys(i2)) then itemp = heap%keys(i) heap%keys(i) = heap%keys(i2) heap%keys(i2) = itemp i = i2 else exit end if end do case default write(0,*) 'Invalid direction in heap ',heap%dir end select return end subroutine psb_insert_int_heap subroutine psb_int_heap_get_first(key,heap,info) implicit none type(psb_int_heap), intent(inout) :: heap integer, intent(out) :: key,info integer :: i, i2, itemp,last,j info = 0 if (heap%last <= 0) then key = 0 info = -1 return endif key = heap%keys(1) heap%keys(1) = heap%keys(heap%last) heap%last = heap%last - 1 last = heap%last select case(heap%dir) case (psb_sort_up_) i = 1 do if (i > (last/2)) exit if ( (heap%keys(2*i) < heap%keys(2*i+1)) .or.& & (2*i == last)) then j = 2*i else j = 2*i + 1 end if if (heap%keys(i) > heap%keys(j)) then itemp = heap%keys(i) heap%keys(i) = heap%keys(j) heap%keys(j) = itemp i = j else exit end if end do case (psb_sort_down_) i = 1 do if (i > (last/2)) exit if ( (heap%keys(2*i) > heap%keys(2*i+1)) .or.& & (2*i == last)) then j = 2*i else j = 2*i + 1 end if if (heap%keys(i) < heap%keys(j)) then itemp = heap%keys(i) heap%keys(i) = heap%keys(j) heap%keys(j) = itemp i = j else exit end if end do case default write(0,*) 'Invalid direction in heap ',heap%dir end select return end subroutine psb_int_heap_get_first end module psb_sort_mod