mirror of
https://github.com/sfilippone/psblas3.git
synced 2026-10-16 10:52:08 +00:00
base/comm/psb_cgather.f90 base/comm/psb_chalo.f90 base/comm/psb_covrl.f90 base/comm/psb_cscatter.F90 base/comm/psb_cspgather.F90 base/comm/psb_dgather.f90 base/comm/psb_dhalo.f90 base/comm/psb_dovrl.f90 base/comm/psb_dscatter.F90 base/comm/psb_dspgather.F90 base/comm/psb_igather.f90 base/comm/psb_ihalo.f90 base/comm/psb_iovrl.f90 base/comm/psb_iscatter.F90 base/comm/psb_sgather.f90 base/comm/psb_shalo.f90 base/comm/psb_sovrl.f90 base/comm/psb_sscatter.F90 base/comm/psb_sspgather.F90 base/comm/psb_zgather.f90 base/comm/psb_zhalo.f90 base/comm/psb_zovrl.f90 base/comm/psb_zscatter.F90 base/comm/psb_zspgather.F90 base/internals/psb_indx_map_fnd_owner.F90 base/internals/psi_bld_tmphalo.f90 base/internals/psi_bld_tmpovrl.f90 base/internals/psi_compute_size.f90 base/internals/psi_crea_bnd_elem.f90 base/internals/psi_crea_index.f90 base/internals/psi_crea_ovr_elem.f90 base/internals/psi_cswapdata.F90 base/internals/psi_cswaptran.F90 base/internals/psi_desc_impl.f90 base/internals/psi_desc_index.F90 base/internals/psi_dl_check.f90 base/internals/psi_dswapdata.F90 base/internals/psi_dswaptran.F90 base/internals/psi_exist_ovr_elem.f base/internals/psi_extrct_dl.F90 base/internals/psi_fnd_owner.F90 base/internals/psi_idx_cnv.f90 base/internals/psi_idx_ins_cnv.f90 base/internals/psi_iswapdata.F90 base/internals/psi_iswaptran.F90 base/internals/psi_ldsc_pre_halo.f90 base/internals/psi_list_search.f base/internals/psi_ovrl_restr.f90 base/internals/psi_ovrl_save.f90 base/internals/psi_ovrl_upd.f90 base/internals/psi_sort_dl.f90 base/internals/psi_sswapdata.F90 base/internals/psi_sswaptran.F90 base/internals/psi_zswapdata.F90 base/internals/psi_zswaptran.F90 base/internals/srtlist.f base/modules/Makefile base/modules/error.f90 base/modules/parts.f90 base/modules/psb_base_linmap_mod.f90 base/modules/psb_base_mat_mod.f90 base/modules/psb_base_tools_mod.f90 base/modules/psb_c_base_mat_mod.f90 base/modules/psb_c_base_vect_mod.f90 base/modules/psb_c_comm_mod.f90 base/modules/psb_c_csc_mat_mod.f90 base/modules/psb_c_csr_mat_mod.f90 base/modules/psb_c_linmap_mod.f90 base/modules/psb_c_mat_mod.f90 base/modules/psb_c_psblas_mod.f90 base/modules/psb_c_tools_mod.f90 base/modules/psb_c_vect_mod.f90 base/modules/psb_check_mod.f90 base/modules/psb_const_mod.F90 base/modules/psb_d_base_mat_mod.f90 base/modules/psb_d_base_vect_mod.f90 base/modules/psb_d_comm_mod.f90 base/modules/psb_d_csc_mat_mod.f90 base/modules/psb_d_csr_mat_mod.f90 base/modules/psb_d_linmap_mod.f90 base/modules/psb_d_mat_mod.f90 base/modules/psb_d_psblas_mod.f90 base/modules/psb_d_tools_mod.f90 base/modules/psb_d_vect_mod.f90 base/modules/psb_desc_const_mod.f90 base/modules/psb_desc_type.f90 base/modules/psb_error_impl.F90 base/modules/psb_error_mod.F90 base/modules/psb_gen_block_map_mod.f90 base/modules/psb_glist_map_mod.f90 base/modules/psb_hash_map_mod.f90 base/modules/psb_hash_mod.f90 base/modules/psb_i_comm_mod.f90 base/modules/psb_indx_map_mod.f90 base/modules/psb_ip_reord_mod.f90 base/modules/psb_list_map_mod.f90 base/modules/psb_realloc_mod.F90 base/modules/psb_repl_map_mod.f90 base/modules/psb_s_base_mat_mod.f90 base/modules/psb_s_base_vect_mod.f90 base/modules/psb_s_comm_mod.f90 base/modules/psb_s_csc_mat_mod.f90 base/modules/psb_s_csr_mat_mod.f90 base/modules/psb_s_linmap_mod.f90 base/modules/psb_s_mat_mod.f90 base/modules/psb_s_psblas_mod.f90 base/modules/psb_s_tools_mod.f90 base/modules/psb_s_vect_mod.f90 base/modules/psb_serial_mod.f90 base/modules/psb_sort_mod.f90 base/modules/psb_string_mod.f90 base/modules/psb_z_base_mat_mod.f90 base/modules/psb_z_base_vect_mod.f90 base/modules/psb_z_comm_mod.f90 base/modules/psb_z_csc_mat_mod.f90 base/modules/psb_z_csr_mat_mod.f90 base/modules/psb_z_linmap_mod.f90 base/modules/psb_z_mat_mod.f90 base/modules/psb_z_psblas_mod.f90 base/modules/psb_z_tools_mod.f90 base/modules/psb_z_vect_mod.f90 base/modules/psi_bcast_mod.F90 base/modules/psi_c_mod.f90 base/modules/psi_comm_buffers_mod.F90 base/modules/psi_d_mod.f90 base/modules/psi_i_mod.f90 base/modules/psi_p2p_mod.F90 base/modules/psi_penv_mod.F90 base/modules/psi_reduce_mod.F90 base/modules/psi_s_mod.f90 base/modules/psi_serial_mod.f90 base/modules/psi_z_mod.f90 base/psblas/psb_camax.f90 base/psblas/psb_casum.f90 base/psblas/psb_caxpby.f90 base/psblas/psb_cdot.f90 base/psblas/psb_cnrm2.f90 base/psblas/psb_cnrmi.f90 base/psblas/psb_cspmm.f90 base/psblas/psb_cspsm.f90 base/psblas/psb_damax.f90 base/psblas/psb_dasum.f90 base/psblas/psb_daxpby.f90 base/psblas/psb_ddot.f90 base/psblas/psb_dnrm2.f90 base/psblas/psb_dnrmi.f90 base/psblas/psb_dspmm.f90 base/psblas/psb_dspnrm1.f90 base/psblas/psb_dspsm.f90 base/psblas/psb_samax.f90 base/psblas/psb_sasum.f90 base/psblas/psb_saxpby.f90 base/psblas/psb_sdot.f90 base/psblas/psb_snrm2.f90 base/psblas/psb_snrmi.f90 base/psblas/psb_sspmm.f90 base/psblas/psb_sspsm.f90 base/psblas/psb_sxdot.f90 base/psblas/psb_zamax.f90 base/psblas/psb_zasum.f90 base/psblas/psb_zaxpby.f90 base/psblas/psb_zdot.f90 base/psblas/psb_znrm2.f90 base/psblas/psb_znrmi.f90 base/psblas/psb_zspmm.f90 base/psblas/psb_zspsm.f90 base/serial/aux/calsr.f90 base/serial/aux/calsrx.f90 base/serial/aux/camsort_dw.f90 base/serial/aux/camsort_up.f90 base/serial/aux/camsr.f90 base/serial/aux/camsrx.f90 base/serial/aux/casr.f90 base/serial/aux/casrx.f90 base/serial/aux/clsr.f90 base/serial/aux/clsrx.f90 base/serial/aux/dasr.f90 base/serial/aux/dasrx.f90 base/serial/aux/dmsort_dw.f90 base/serial/aux/dmsort_up.f90 base/serial/aux/dmsr.f90 base/serial/aux/dmsrx.f90 base/serial/aux/dsr.f90 base/serial/aux/dsrx.f90 base/serial/aux/iasr.f90 base/serial/aux/iasrx.f90 base/serial/aux/ibsrch.f base/serial/aux/imsr.f90 base/serial/aux/imsru.f90 base/serial/aux/imsrx.f90 base/serial/aux/isaperm.f base/serial/aux/isr.f90 base/serial/aux/isrx.f90 base/serial/aux/issrch.f base/serial/aux/msort_dw.f90 base/serial/aux/msort_up.f90 base/serial/aux/sasr.f90 base/serial/aux/sasrx.f90 base/serial/aux/smsort_dw.f90 base/serial/aux/smsort_up.f90 base/serial/aux/smsr.f90 base/serial/aux/smsrx.f90 base/serial/aux/ssr.f90 base/serial/aux/ssrx.f90 base/serial/aux/zalsr.f90 base/serial/aux/zalsrx.f90 base/serial/aux/zamsort_dw.f90 base/serial/aux/zamsort_up.f90 base/serial/aux/zamsr.f90 base/serial/aux/zamsrx.f90 base/serial/aux/zasr.f90 base/serial/aux/zasrx.f90 base/serial/aux/zlsr.f90 base/serial/aux/zlsrx.f90 base/serial/f77/caxpby.f base/serial/f77/daxpby.f base/serial/f77/saxpby.f base/serial/f77/smmp.f base/serial/f77/zaxpby.f base/serial/impl/psb_base_mat_impl.f90 base/serial/impl/psb_c_base_mat_impl.f90 base/serial/impl/psb_c_coo_impl.f90 base/serial/impl/psb_c_csc_impl.f90 base/serial/impl/psb_c_csr_impl.f90 base/serial/impl/psb_c_mat_impl.F90 base/serial/impl/psb_d_base_mat_impl.f90 base/serial/impl/psb_d_coo_impl.f90 base/serial/impl/psb_d_csc_impl.f90 base/serial/impl/psb_d_csr_impl.f90 base/serial/impl/psb_d_mat_impl.F90 base/serial/impl/psb_s_base_mat_impl.f90 base/serial/impl/psb_s_coo_impl.f90 base/serial/impl/psb_s_csc_impl.f90 base/serial/impl/psb_s_csr_impl.f90 base/serial/impl/psb_s_mat_impl.F90 base/serial/impl/psb_z_base_mat_impl.f90 base/serial/impl/psb_z_coo_impl.f90 base/serial/impl/psb_z_csc_impl.f90 base/serial/impl/psb_z_csr_impl.f90 base/serial/impl/psb_z_mat_impl.F90 base/serial/psb_aspxpby.f90 base/serial/psb_cgelp.f90 base/serial/psb_cgeprt.f90 base/serial/psb_cnumbmm.f90 base/serial/psb_crwextd.f90 base/serial/psb_csymbmm.f90 base/serial/psb_dgelp.f90 base/serial/psb_dgeprt.f90 base/serial/psb_dnumbmm.f90 base/serial/psb_drwextd.f90 base/serial/psb_dsymbmm.f90 base/serial/psb_sgelp.f90 base/serial/psb_sgeprt.f90 base/serial/psb_snumbmm.f90 base/serial/psb_sort_impl.f90 base/serial/psb_spdot_srtd.f90 base/serial/psb_spge_dot.f90 base/serial/psb_srwextd.f90 base/serial/psb_ssymbmm.f90 base/serial/psb_zgelp.f90 base/serial/psb_zgeprt.f90 base/serial/psb_znumbmm.f90 base/serial/psb_zrwextd.f90 base/serial/psb_zsymbmm.f90 base/serial/psi_serial_impl.f90 base/tools/psb_c_map.f90 base/tools/psb_callc.f90 base/tools/psb_casb.f90 base/tools/psb_ccdbldext.F90 base/tools/psb_cd_inloc.f90 base/tools/psb_cd_lstext.f90 base/tools/psb_cd_reinit.f90 base/tools/psb_cd_set_bld.f90 base/tools/psb_cd_switch_ovl_indxmap.f90 base/tools/psb_cdall.f90 base/tools/psb_cdals.f90 base/tools/psb_cdalv.f90 base/tools/psb_cdcpy.F90 base/tools/psb_cdins.f90 base/tools/psb_cdprt.f90 base/tools/psb_cdren.f90 base/tools/psb_cdrep.f90 base/tools/psb_cfree.f90 base/tools/psb_cins.f90 base/tools/psb_cspalloc.f90 base/tools/psb_cspasb.f90 base/tools/psb_cspfree.f90 base/tools/psb_csphalo.F90 base/tools/psb_cspins.f90 base/tools/psb_csprn.f90 base/tools/psb_d_map.f90 base/tools/psb_dallc.f90 base/tools/psb_dasb.f90 base/tools/psb_dcdbldext.F90 base/tools/psb_dfree.f90 base/tools/psb_dins.f90 base/tools/psb_dspalloc.f90 base/tools/psb_dspasb.f90 base/tools/psb_dspfree.f90 base/tools/psb_dsphalo.F90 base/tools/psb_dspins.f90 base/tools/psb_dsprn.f90 base/tools/psb_get_overlap.f90 base/tools/psb_glob_to_loc.f90 base/tools/psb_ialloc.f90 base/tools/psb_iasb.f90 base/tools/psb_icdasb.F90 base/tools/psb_ifree.f90 base/tools/psb_iins.f90 base/tools/psb_loc_to_glob.f90 base/tools/psb_s_map.f90 base/tools/psb_sallc.f90 base/tools/psb_sasb.f90 base/tools/psb_scdbldext.F90 base/tools/psb_sfree.f90 base/tools/psb_sins.f90 base/tools/psb_sspalloc.f90 base/tools/psb_sspasb.f90 base/tools/psb_sspfree.f90 base/tools/psb_ssphalo.F90 base/tools/psb_sspins.f90 base/tools/psb_ssprn.f90 base/tools/psb_z_map.f90 base/tools/psb_zallc.f90 base/tools/psb_zasb.f90 base/tools/psb_zcdbldext.F90 base/tools/psb_zfree.f90 base/tools/psb_zins.f90 base/tools/psb_zspalloc.f90 base/tools/psb_zspasb.f90 base/tools/psb_zspfree.f90 base/tools/psb_zsphalo.F90 base/tools/psb_zspins.f90 base/tools/psb_zsprn.f90 krylov/psb_base_inner_krylov_mod.f90 krylov/psb_c_inner_krylov_mod.f90 krylov/psb_cbicg.f90 krylov/psb_ccg.f90 krylov/psb_ccgs.f90 krylov/psb_ccgstab.f90 krylov/psb_ccgstabl.f90 krylov/psb_ckrylov.f90 krylov/psb_crgmres.f90 krylov/psb_d_inner_krylov_mod.f90 krylov/psb_dbicg.f90 krylov/psb_dcg.F90 krylov/psb_dcgs.f90 krylov/psb_dcgstab.F90 krylov/psb_dcgstabl.f90 krylov/psb_dkrylov.f90 krylov/psb_drgmres.f90 krylov/psb_krylov_mod.f90 krylov/psb_s_inner_krylov_mod.f90 krylov/psb_sbicg.f90 krylov/psb_scg.F90 krylov/psb_scgs.f90 krylov/psb_scgstab.F90 krylov/psb_scgstabl.f90 krylov/psb_skrylov.f90 krylov/psb_srgmres.f90 krylov/psb_z_inner_krylov_mod.f90 krylov/psb_zbicg.f90 krylov/psb_zcg.F90 krylov/psb_zcgs.f90 krylov/psb_zcgstab.f90 krylov/psb_zcgstabl.f90 krylov/psb_zkrylov.f90 krylov/psb_zrgmres.f90 opt/psb_d_ell_impl.f90 opt/psb_d_ell_mat_mod.f90 prec/Makefile prec/impl/psb_c_bjacprec_impl.f90 prec/impl/psb_c_diagprec_impl.f90 prec/impl/psb_c_nullprec_impl.f90 prec/impl/psb_c_prec_type_impl.f90 prec/impl/psb_cilu_fct.f90 prec/impl/psb_cprecbld.f90 prec/impl/psb_cprecinit.f90 prec/impl/psb_cprecset.f90 prec/impl/psb_d_bjacprec_impl.f90 prec/impl/psb_d_diagprec_impl.f90 prec/impl/psb_d_nullprec_impl.f90 prec/impl/psb_d_prec_type_impl.f90 prec/impl/psb_dilu_fct.f90 prec/impl/psb_dprecbld.f90 prec/impl/psb_dprecinit.f90 prec/impl/psb_dprecset.f90 prec/impl/psb_s_bjacprec_impl.f90 prec/impl/psb_s_diagprec_impl.f90 prec/impl/psb_s_nullprec_impl.f90 prec/impl/psb_s_prec_type_impl.f90 prec/impl/psb_silu_fct.f90 prec/impl/psb_sprecbld.f90 prec/impl/psb_sprecinit.f90 prec/impl/psb_sprecset.f90 prec/impl/psb_z_bjacprec_impl.f90 prec/impl/psb_z_diagprec_impl.f90 prec/impl/psb_z_nullprec_impl.f90 prec/impl/psb_z_prec_type_impl.f90 prec/impl/psb_zilu_fct.f90 prec/impl/psb_zprecbld.f90 prec/impl/psb_zprecinit.f90 prec/impl/psb_zprecset.f90 prec/psb_c_base_prec_mod.f90 prec/psb_c_bjacprec.f90 prec/psb_c_diagprec.f90 prec/psb_c_nullprec.f90 prec/psb_c_prec_mod.f90 prec/psb_c_prec_type.f90 prec/psb_d_base_prec_mod.f90 prec/psb_d_bjacprec.f90 prec/psb_d_diagprec.f90 prec/psb_d_nullprec.f90 prec/psb_d_prec_mod.f90 prec/psb_d_prec_type.f90 prec/psb_prec_const_mod.f90 prec/psb_s_base_prec_mod.f90 prec/psb_s_bjacprec.f90 prec/psb_s_diagprec.f90 prec/psb_s_nullprec.f90 prec/psb_s_prec_mod.f90 prec/psb_s_prec_type.f90 prec/psb_z_base_prec_mod.f90 prec/psb_z_bjacprec.f90 prec/psb_z_diagprec.f90 prec/psb_z_nullprec.f90 prec/psb_z_prec_mod.f90 prec/psb_z_prec_type.f90 test/fileread/cf_sample.f90 test/fileread/df_sample.f90 test/fileread/getp.f90 test/fileread/sf_sample.f90 test/fileread/zf_sample.f90 test/kernel/d_file_spmv.f90 test/kernel/pdgenspmv.f90 test/kernel/s_file_spmv.f90 test/newfmt/ppde.F90 test/newfmt/spde.f90 test/pargen/ppde.f90 test/pargen/spde.f90 test/serial/d_coo_matgen.f90 test/serial/d_matgen.F90 test/serial/psb_d_cxx_impl.f90 test/serial/psb_d_cxx_mat_mod.f90 test/serial/psb_d_cyy_impl.f90 test/serial/psb_d_cyy_mat_mod.f90 test/torture/psb_c_mvsv_tester.f90 test/torture/psb_d_mvsv_tester.f90 test/torture/psb_s_mvsv_tester.f90 test/torture/psb_z_mvsv_tester.f90 test/torture/psbtf.f90 test/util/dhb2mm.f90 test/util/dmm2hb.f90 test/util/zhb2mm.f90 test/util/zmm2hb.f90 util/psb_blockpart_mod.f90 util/psb_c_hbio_impl.f90 util/psb_c_mat_dist_impl.f90 util/psb_c_mmio_impl.f90 util/psb_c_renum_impl.F90 util/psb_d_hbio_impl.f90 util/psb_d_mat_dist_impl.f90 util/psb_d_mmio_impl.f90 util/psb_d_renum_impl.F90 util/psb_gps_mod.f90 util/psb_hbio_mod.f90 util/psb_mat_dist_impl.f90 util/psb_mat_dist_mod.f90 util/psb_metispart_mod.F90 util/psb_mmio_mod.f90 util/psb_renum_mod.f90 util/psb_s_hbio_impl.f90 util/psb_s_mat_dist_impl.f90 util/psb_s_mmio_impl.f90 util/psb_s_renum_impl.F90 util/psb_z_hbio_impl.f90 util/psb_z_mat_dist_impl.f90 util/psb_z_mmio_impl.f90 util/psb_z_renum_impl.F90 Introduced use of psb_ipk_. Modified hash_mod: should now work even with psb_ipk_= 8 bytes. Still need to fix the parallel environment for long-integers.
4630 lines
105 KiB
Fortran
4630 lines
105 KiB
Fortran
!!$
|
|
!!$ Parallel Sparse BLAS version 3.0
|
|
!!$ (C) Copyright 2006, 2007, 2008, 2009, 2010
|
|
!!$ Salvatore Filippone University of Rome Tor Vergata
|
|
!!$ Alfredo Buttari CNRS-IRIT, Toulouse
|
|
!!$
|
|
!!$ 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.
|
|
!!$
|
|
!!$
|
|
!
|
|
! The merge-sort and quicksort routines are implemented in the
|
|
! serial/aux directory
|
|
! References:
|
|
! D. Knuth
|
|
! The Art of Computer Programming, vol. 3
|
|
! Addison-Wesley
|
|
!
|
|
! Aho, Hopcroft, Ullman
|
|
! Data Structures and Algorithms
|
|
! Addison-Wesley
|
|
!
|
|
|
|
logical function psb_isaperm(n,eip)
|
|
use psb_sort_mod, psb_protect_name => psb_isaperm
|
|
implicit none
|
|
|
|
integer(psb_ipk_), intent(in) :: n
|
|
integer(psb_ipk_), intent(in) :: eip(n)
|
|
integer(psb_ipk_), allocatable :: ip(:)
|
|
integer(psb_ipk_) :: i,j,m, info
|
|
|
|
|
|
psb_isaperm = .true.
|
|
if (n <= 0) return
|
|
allocate(ip(n), stat=info)
|
|
if (info /= psb_success_) return
|
|
!
|
|
! sanity check first
|
|
!
|
|
do i=1, n
|
|
ip(i) = eip(i)
|
|
if ((ip(i) < 1).or.(ip(i) > n)) then
|
|
write(psb_err_unit,*) 'Out of bounds in isaperm' ,ip(i), n
|
|
psb_isaperm = .false.
|
|
return
|
|
endif
|
|
enddo
|
|
|
|
!
|
|
! now work through the cycles, by marking each successive item as negative.
|
|
! no cycle should intersect with any other, hence the >= 1 check.
|
|
!
|
|
do m = 1, n
|
|
i = ip(m)
|
|
if (i < 0) then
|
|
ip(m) = -i
|
|
else if (i /= m) then
|
|
j = ip(i)
|
|
ip(i) = -j
|
|
i = j
|
|
do while ((j >= 1).and.(j /= m))
|
|
j = ip(i)
|
|
ip(i) = -j
|
|
i = j
|
|
enddo
|
|
ip(m) = iabs(ip(m))
|
|
if (j /= m) then
|
|
psb_isaperm = .false.
|
|
goto 9999
|
|
endif
|
|
end if
|
|
enddo
|
|
9999 continue
|
|
|
|
return
|
|
end function psb_isaperm
|
|
|
|
function psb_iblsrch(key,n,v) result(ipos)
|
|
use psb_sort_mod, psb_protect_name => psb_iblsrch
|
|
implicit none
|
|
integer(psb_ipk_) :: ipos, key, n
|
|
integer(psb_ipk_) :: v(n)
|
|
|
|
integer(psb_ipk_) :: lb, ub, m
|
|
|
|
if (n < 5) then
|
|
! don't bother with binary search for very
|
|
! small vectors
|
|
ipos = 0
|
|
do
|
|
if (ipos == n) return
|
|
if (key < v(ipos+1)) return
|
|
ipos = ipos + 1
|
|
end do
|
|
else
|
|
lb = 1
|
|
ub = n
|
|
ipos = -1
|
|
|
|
do while (lb <= ub)
|
|
m = (lb+ub)/2
|
|
if (key==v(m)) then
|
|
ipos = m
|
|
return
|
|
else if (key < v(m)) then
|
|
ub = m-1
|
|
else
|
|
lb = m + 1
|
|
end if
|
|
enddo
|
|
if (v(ub) > key) then
|
|
!!$ write(0,*) 'Check: ',ub,v(ub),key
|
|
ub = ub - 1
|
|
end if
|
|
ipos = ub
|
|
endif
|
|
return
|
|
end function psb_iblsrch
|
|
|
|
function psb_ibsrch(key,n,v) result(ipos)
|
|
use psb_sort_mod, psb_protect_name => psb_ibsrch
|
|
implicit none
|
|
integer(psb_ipk_) :: ipos, key, n
|
|
integer(psb_ipk_) :: v(n)
|
|
|
|
integer(psb_ipk_) :: lb, ub, m
|
|
|
|
lb = 1
|
|
ub = n
|
|
ipos = -1
|
|
|
|
do while (lb.le.ub)
|
|
m = (lb+ub)/2
|
|
if (key.eq.v(m)) then
|
|
ipos = m
|
|
lb = ub + 1
|
|
else if (key < v(m)) then
|
|
ub = m-1
|
|
else
|
|
lb = m + 1
|
|
end if
|
|
enddo
|
|
return
|
|
end function psb_ibsrch
|
|
|
|
function psb_issrch(key,n,v) result(ipos)
|
|
use psb_sort_mod, psb_protect_name => psb_issrch
|
|
implicit none
|
|
integer(psb_ipk_) :: ipos, key, n
|
|
integer(psb_ipk_) :: v(n)
|
|
|
|
integer(psb_ipk_) :: i
|
|
|
|
ipos = -1
|
|
do i=1,n
|
|
if (key.eq.v(i)) then
|
|
ipos = i
|
|
return
|
|
end if
|
|
enddo
|
|
|
|
return
|
|
end function psb_issrch
|
|
|
|
|
|
subroutine imsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => imsort
|
|
use psb_error_mod
|
|
implicit none
|
|
integer(psb_ipk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_value_invalid_i_,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 == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine imsort
|
|
|
|
|
|
subroutine smsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => smsort
|
|
use psb_error_mod
|
|
implicit none
|
|
real(psb_spk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_value_invalid_i_,name,i_err=(/4,flag_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
call smsrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call smsr(n,x,dir_)
|
|
end if
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine smsort
|
|
|
|
subroutine dmsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => dmsort
|
|
use psb_error_mod
|
|
implicit none
|
|
real(psb_dpk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_value_invalid_i_,name,i_err=(/4,flag_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
call dmsrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call dmsr(n,x,dir_)
|
|
end if
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine dmsort
|
|
|
|
subroutine camsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => camsort
|
|
use psb_error_mod
|
|
implicit none
|
|
complex(psb_spk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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_asort_up_
|
|
end if
|
|
select case(dir_)
|
|
case( psb_asort_up_, psb_asort_down_)
|
|
! OK keep going
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_value_invalid_i_,name,i_err=(/4,flag_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
call camsrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call camsr(n,x,dir_)
|
|
end if
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine camsort
|
|
|
|
subroutine zamsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => zamsort
|
|
use psb_error_mod
|
|
implicit none
|
|
complex(psb_dpk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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_asort_up_
|
|
end if
|
|
select case(dir_)
|
|
case( psb_asort_up_, psb_asort_down_)
|
|
! OK keep going
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_value_invalid_i_,name,i_err=(/4,flag_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
call zamsrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call zamsr(n,x,dir_)
|
|
end if
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine zamsort
|
|
|
|
|
|
subroutine imsort_u(x,nout,dir)
|
|
use psb_sort_mod, psb_protect_name => imsort_u
|
|
use psb_error_mod
|
|
implicit none
|
|
integer(psb_ipk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), intent(out) :: nout
|
|
integer(psb_ipk_), optional, intent(in) :: dir
|
|
|
|
integer(psb_ipk_) :: dir_, 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(psb_err_input_value_invalid_i_,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 == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine imsort_u
|
|
|
|
|
|
subroutine iqsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => iqsort
|
|
use psb_error_mod
|
|
implicit none
|
|
integer(psb_ipk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine iqsort
|
|
|
|
|
|
subroutine sqsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => sqsort
|
|
use psb_error_mod
|
|
implicit none
|
|
real(psb_spk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
|
|
call ssrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call ssr(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(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
|
|
call sasrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call sasr(n,x,dir_)
|
|
end if
|
|
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine sqsort
|
|
|
|
subroutine dqsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => dqsort
|
|
use psb_error_mod
|
|
implicit none
|
|
real(psb_dpk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine dqsort
|
|
|
|
|
|
subroutine cqsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => cqsort
|
|
use psb_error_mod
|
|
implicit none
|
|
complex(psb_spk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
|
|
call clsrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call clsr(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(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
|
|
call calsrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call calsr(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(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
|
|
call casrx(n,x,ix,dir_,flag_)
|
|
else
|
|
call casr(n,x,dir_)
|
|
end if
|
|
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine cqsort
|
|
|
|
|
|
subroutine zqsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => zqsort
|
|
use psb_error_mod
|
|
implicit none
|
|
complex(psb_dpk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: 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(psb_err_input_value_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_asize_invalid_i_,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(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine zqsort
|
|
|
|
|
|
|
|
|
|
subroutine ihsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => ihsort
|
|
use psb_error_mod
|
|
implicit none
|
|
integer(psb_ipk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: dir_, flag_, n, i, l, err_act,info
|
|
integer(psb_ipk_) :: key
|
|
integer(psb_ipk_) :: index
|
|
|
|
character(len=20) :: name
|
|
|
|
name='psb_hsort'
|
|
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(psb_err_input_value_invalid_i_,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
|
|
|
|
select case(dir_)
|
|
case(psb_sort_up_,psb_sort_down_,psb_asort_up_,psb_asort_down_)
|
|
! OK
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
n = size(x)
|
|
|
|
!
|
|
! Dirty trick to sort with heaps: if we want
|
|
! to sort in place upwards, first we set up a heap so that
|
|
! we can easily get the LARGEST element, then we take it out
|
|
! and put it in the last entry, and so on.
|
|
! So, we invert dir_!
|
|
!
|
|
dir_ = -dir_
|
|
|
|
if (present(ix)) then
|
|
if (size(ix) < n) then
|
|
call psb_errpush(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
if (flag_ == psb_sort_ovw_idx_) then
|
|
do i=1, n
|
|
ix(i) = i
|
|
end do
|
|
end if
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
index = ix(i)
|
|
call psi_insert_int_idx_heap(key,index,l,x,ix,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! '
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_int_idx_heap_get_first(key,index,l,x,ix,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
ix(i) = index
|
|
end do
|
|
else if (.not.present(ix)) then
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
call psi_insert_int_heap(key,l,x,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! ',l,i
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_int_heap_get_first(key,l,x,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
end do
|
|
end if
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine ihsort
|
|
|
|
|
|
subroutine shsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => shsort
|
|
use psb_error_mod
|
|
implicit none
|
|
real(psb_spk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: dir_, flag_, n, i, l, err_act,info
|
|
real(psb_spk_) :: key
|
|
integer(psb_ipk_) :: index
|
|
|
|
character(len=20) :: name
|
|
|
|
name='psb_hsort'
|
|
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(psb_err_input_value_invalid_i_,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
|
|
|
|
select case(dir_)
|
|
case(psb_sort_up_,psb_sort_down_,psb_asort_up_,psb_asort_down_)
|
|
! OK
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
n = size(x)
|
|
|
|
!
|
|
! Dirty trick to sort with heaps: if we want
|
|
! to sort in place upwards, first we set up a heap so that
|
|
! we can easily get the LARGEST element, then we take it out
|
|
! and put it in the last entry, and so on.
|
|
! So, we invert dir_!
|
|
!
|
|
dir_ = -dir_
|
|
|
|
if (present(ix)) then
|
|
if (size(ix) < n) then
|
|
call psb_errpush(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
if (flag_ == psb_sort_ovw_idx_) then
|
|
do i=1, n
|
|
ix(i) = i
|
|
end do
|
|
end if
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
index = ix(i)
|
|
call psi_insert_real_idx_heap(key,index,l,x,ix,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! '
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_real_idx_heap_get_first(key,index,l,x,ix,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
ix(i) = index
|
|
end do
|
|
else if (.not.present(ix)) then
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
call psi_insert_real_heap(key,l,x,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! ',l,i
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_real_heap_get_first(key,l,x,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
end do
|
|
end if
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine shsort
|
|
|
|
|
|
subroutine dhsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => dhsort
|
|
use psb_error_mod
|
|
implicit none
|
|
real(psb_dpk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: dir_, flag_, n, i, l, err_act,info
|
|
real(psb_dpk_) :: key
|
|
integer(psb_ipk_) :: index
|
|
|
|
character(len=20) :: name
|
|
|
|
name='psb_hsort'
|
|
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(psb_err_input_value_invalid_i_,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
|
|
|
|
select case(dir_)
|
|
case(psb_sort_up_,psb_sort_down_,psb_asort_up_,psb_asort_down_)
|
|
! OK
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
n = size(x)
|
|
|
|
!
|
|
! Dirty trick to sort with heaps: if we want
|
|
! to sort in place upwards, first we set up a heap so that
|
|
! we can easily get the LARGEST element, then we take it out
|
|
! and put it in the last entry, and so on.
|
|
! So, we invert dir_!
|
|
!
|
|
dir_ = -dir_
|
|
|
|
if (present(ix)) then
|
|
if (size(ix) < n) then
|
|
call psb_errpush(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
if (flag_ == psb_sort_ovw_idx_) then
|
|
do i=1, n
|
|
ix(i) = i
|
|
end do
|
|
end if
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
index = ix(i)
|
|
call psi_insert_double_idx_heap(key,index,l,x,ix,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! '
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_double_idx_heap_get_first(key,index,l,x,ix,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
ix(i) = index
|
|
end do
|
|
else if (.not.present(ix)) then
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
call psi_insert_double_heap(key,l,x,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! ',l,i
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_double_heap_get_first(key,l,x,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
end do
|
|
end if
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine dhsort
|
|
|
|
|
|
subroutine chsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => chsort
|
|
use psb_error_mod
|
|
implicit none
|
|
complex(psb_spk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: dir_, flag_, n, i, l, err_act,info
|
|
complex(psb_spk_) :: key
|
|
integer(psb_ipk_) :: index
|
|
|
|
character(len=20) :: name
|
|
|
|
name='psb_hsort'
|
|
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(psb_err_input_value_invalid_i_,name,i_err=(/4,flag_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
if (present(dir)) then
|
|
dir_ = dir
|
|
else
|
|
dir_= psb_asort_up_
|
|
end if
|
|
|
|
select case(dir_)
|
|
case(psb_asort_up_,psb_asort_down_)
|
|
! OK
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
n = size(x)
|
|
|
|
!
|
|
! Dirty trick to sort with heaps: if we want
|
|
! to sort in place upwards, first we set up a heap so that
|
|
! we can easily get the LARGEST element, then we take it out
|
|
! and put it in the last entry, and so on.
|
|
! So, we invert dir_!
|
|
!
|
|
dir_ = -dir_
|
|
|
|
if (present(ix)) then
|
|
if (size(ix) < n) then
|
|
call psb_errpush(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
if (flag_ == psb_sort_ovw_idx_) then
|
|
do i=1, n
|
|
ix(i) = i
|
|
end do
|
|
end if
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
index = ix(i)
|
|
call psi_insert_scomplex_idx_heap(key,index,l,x,ix,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! '
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_scomplex_idx_heap_get_first(key,index,l,x,ix,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
ix(i) = index
|
|
end do
|
|
else if (.not.present(ix)) then
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
call psi_insert_scomplex_heap(key,l,x,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! ',l,i
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_scomplex_heap_get_first(key,l,x,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
end do
|
|
end if
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine chsort
|
|
|
|
|
|
subroutine zhsort(x,ix,dir,flag)
|
|
use psb_sort_mod, psb_protect_name => zhsort
|
|
use psb_error_mod
|
|
implicit none
|
|
complex(psb_dpk_), intent(inout) :: x(:)
|
|
integer(psb_ipk_), optional, intent(in) :: dir, flag
|
|
integer(psb_ipk_), optional, intent(inout) :: ix(:)
|
|
|
|
integer(psb_ipk_) :: dir_, flag_, n, i, l, err_act,info
|
|
complex(psb_dpk_) :: key
|
|
integer(psb_ipk_) :: index
|
|
|
|
character(len=20) :: name
|
|
|
|
name='psb_hsort'
|
|
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(psb_err_input_value_invalid_i_,name,i_err=(/4,flag_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
if (present(dir)) then
|
|
dir_ = dir
|
|
else
|
|
dir_= psb_asort_up_
|
|
end if
|
|
|
|
select case(dir_)
|
|
case(psb_asort_up_,psb_asort_down_)
|
|
! OK
|
|
case default
|
|
call psb_errpush(psb_err_input_value_invalid_i_,name,i_err=(/3,dir_,0,0,0/))
|
|
goto 9999
|
|
end select
|
|
|
|
n = size(x)
|
|
|
|
!
|
|
! Dirty trick to sort with heaps: if we want
|
|
! to sort in place upwards, first we set up a heap so that
|
|
! we can easily get the LARGEST element, then we take it out
|
|
! and put it in the last entry, and so on.
|
|
! So, we invert dir_!
|
|
!
|
|
dir_ = -dir_
|
|
|
|
if (present(ix)) then
|
|
if (size(ix) < n) then
|
|
call psb_errpush(psb_err_input_asize_invalid_i_,name,i_err=(/2,size(ix),0,0,0/))
|
|
goto 9999
|
|
end if
|
|
if (flag_ == psb_sort_ovw_idx_) then
|
|
do i=1, n
|
|
ix(i) = i
|
|
end do
|
|
end if
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
index = ix(i)
|
|
call psi_insert_dcomplex_idx_heap(key,index,l,x,ix,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! '
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_dcomplex_idx_heap_get_first(key,index,l,x,ix,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
ix(i) = index
|
|
end do
|
|
else if (.not.present(ix)) then
|
|
l = 0
|
|
do i=1, n
|
|
key = x(i)
|
|
call psi_insert_dcomplex_heap(key,l,x,dir_,info)
|
|
if (l /= i) then
|
|
write(psb_err_unit,*) 'Mismatch while heapifying ! ',l,i
|
|
end if
|
|
end do
|
|
do i=n, 2, -1
|
|
call psi_dcomplex_heap_get_first(key,l,x,dir_,info)
|
|
if (l /= i-1) then
|
|
write(psb_err_unit,*) 'Mismatch while pulling out of heap ',l,i
|
|
end if
|
|
x(i) = key
|
|
end do
|
|
end if
|
|
|
|
|
|
9999 continue
|
|
if (err_act == psb_act_abort_) then
|
|
call psb_error()
|
|
return
|
|
end if
|
|
end subroutine zhsort
|
|
|
|
|
|
function psb_howmany_int_heap(heap)
|
|
use psb_sort_mod, psb_protect_name => psb_howmany_int_heap
|
|
implicit none
|
|
type(psb_int_heap), intent(in) :: heap
|
|
integer(psb_ipk_) :: 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_sort_mod, psb_protect_name => psb_init_int_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
type(psb_int_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in), optional :: dir
|
|
|
|
info = psb_success_
|
|
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_,psb_asort_up_,psb_asort_down_)
|
|
! ok, do nothing
|
|
case default
|
|
write(psb_err_unit,*) '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)
|
|
use psb_sort_mod, psb_protect_name => psb_dump_int_heap
|
|
implicit none
|
|
type(psb_int_heap), intent(in) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in) :: iout
|
|
|
|
info = psb_success_
|
|
if (iout < 0) then
|
|
write(psb_err_unit,*) '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)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else
|
|
if (heap%last > 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_sort_mod, psb_protect_name => psb_insert_int_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
|
|
integer(psb_ipk_), intent(in) :: key
|
|
type(psb_int_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
info = psb_success_
|
|
if (heap%last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',heap%last
|
|
info = heap%last
|
|
return
|
|
endif
|
|
|
|
heap%last = heap%last
|
|
call psb_ensure_size(heap%last+1,heap%keys,info,addsz=psb_heap_resize)
|
|
if (info /= psb_success_) then
|
|
write(psb_err_unit,*) 'Memory allocation failure in heap_insert'
|
|
info = -5
|
|
return
|
|
end if
|
|
call psi_insert_int_heap(key,heap%last,heap%keys,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_insert_int_heap
|
|
|
|
|
|
subroutine psb_int_heap_get_first(key,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_int_heap_get_first
|
|
implicit none
|
|
|
|
type(psb_int_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: key,info
|
|
|
|
info = psb_success_
|
|
|
|
call psi_int_heap_get_first(key,heap%last,heap%keys,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_int_heap_get_first
|
|
|
|
|
|
function psb_howmany_real_idx_heap(heap)
|
|
use psb_sort_mod, psb_protect_name => psb_howmany_real_idx_heap
|
|
implicit none
|
|
type(psb_real_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_) :: psb_howmany_real_idx_heap
|
|
psb_howmany_real_idx_heap = heap%last
|
|
end function psb_howmany_real_idx_heap
|
|
|
|
subroutine psb_init_real_idx_heap(heap,info,dir)
|
|
use psb_sort_mod, psb_protect_name => psb_init_real_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
type(psb_real_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in), optional :: dir
|
|
|
|
info = psb_success_
|
|
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_,psb_asort_up_,psb_asort_down_)
|
|
! ok, do nothing
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction, defaulting to psb_sort_up_'
|
|
heap%dir = psb_sort_up_
|
|
end select
|
|
|
|
call psb_ensure_size(psb_heap_resize,heap%keys,info)
|
|
call psb_ensure_size(psb_heap_resize,heap%idxs,info)
|
|
return
|
|
end subroutine psb_init_real_idx_heap
|
|
|
|
subroutine psb_dump_real_idx_heap(iout,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_dump_real_idx_heap
|
|
implicit none
|
|
type(psb_real_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in) :: iout
|
|
|
|
info = psb_success_
|
|
if (iout < 0) then
|
|
write(psb_err_unit,*) '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)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else if ((heap%last > 0).and.((.not.allocated(heap%idxs)).or.&
|
|
& (size(heap%idxs)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else
|
|
if (heap%last > 0) then
|
|
write(iout,*) heap%keys(1:heap%last)
|
|
write(iout,*) heap%idxs(1:heap%last)
|
|
end if
|
|
end if
|
|
end subroutine psb_dump_real_idx_heap
|
|
|
|
subroutine psb_insert_real_idx_heap(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_insert_real_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
|
|
real(psb_spk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index
|
|
type(psb_real_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
info = psb_success_
|
|
if (heap%last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',heap%last
|
|
info = heap%last
|
|
return
|
|
endif
|
|
|
|
call psb_ensure_size(heap%last+1,heap%keys,info,addsz=psb_heap_resize)
|
|
if (info == psb_success_) &
|
|
& call psb_ensure_size(heap%last+1,heap%idxs,info,addsz=psb_heap_resize)
|
|
if (info /= psb_success_) then
|
|
write(psb_err_unit,*) 'Memory allocation failure in heap_insert'
|
|
info = -5
|
|
return
|
|
end if
|
|
|
|
call psi_insert_real_idx_heap(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_insert_real_idx_heap
|
|
|
|
subroutine psb_real_idx_heap_get_first(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_real_idx_heap_get_first
|
|
implicit none
|
|
|
|
type(psb_real_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
real(psb_spk_), intent(out) :: key
|
|
|
|
info = psb_success_
|
|
|
|
call psi_real_idx_heap_get_first(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_real_idx_heap_get_first
|
|
|
|
|
|
function psb_howmany_double_idx_heap(heap)
|
|
use psb_sort_mod, psb_protect_name => psb_howmany_double_idx_heap
|
|
implicit none
|
|
type(psb_double_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_) :: psb_howmany_double_idx_heap
|
|
psb_howmany_double_idx_heap = heap%last
|
|
end function psb_howmany_double_idx_heap
|
|
|
|
subroutine psb_init_double_idx_heap(heap,info,dir)
|
|
use psb_sort_mod, psb_protect_name => psb_init_double_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
type(psb_double_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in), optional :: dir
|
|
|
|
info = psb_success_
|
|
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_,psb_asort_up_,psb_asort_down_)
|
|
! ok, do nothing
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction, defaulting to psb_sort_up_'
|
|
heap%dir = psb_sort_up_
|
|
end select
|
|
|
|
call psb_ensure_size(psb_heap_resize,heap%keys,info)
|
|
call psb_ensure_size(psb_heap_resize,heap%idxs,info)
|
|
return
|
|
end subroutine psb_init_double_idx_heap
|
|
|
|
subroutine psb_dump_double_idx_heap(iout,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_dump_double_idx_heap
|
|
implicit none
|
|
type(psb_double_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in) :: iout
|
|
|
|
info = psb_success_
|
|
if (iout < 0) then
|
|
write(psb_err_unit,*) '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)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else if ((heap%last > 0).and.((.not.allocated(heap%idxs)).or.&
|
|
& (size(heap%idxs)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else
|
|
if (heap%last > 0) then
|
|
write(iout,*) heap%keys(1:heap%last)
|
|
write(iout,*) heap%idxs(1:heap%last)
|
|
end if
|
|
end if
|
|
end subroutine psb_dump_double_idx_heap
|
|
|
|
subroutine psb_insert_double_idx_heap(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_insert_double_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
|
|
real(psb_dpk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index
|
|
type(psb_double_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
info = psb_success_
|
|
if (heap%last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',heap%last
|
|
info = heap%last
|
|
return
|
|
endif
|
|
|
|
call psb_ensure_size(heap%last+1,heap%keys,info,addsz=psb_heap_resize)
|
|
if (info == psb_success_) &
|
|
& call psb_ensure_size(heap%last+1,heap%idxs,info,addsz=psb_heap_resize)
|
|
if (info /= psb_success_) then
|
|
write(psb_err_unit,*) 'Memory allocation failure in heap_insert'
|
|
info = -5
|
|
return
|
|
end if
|
|
|
|
call psi_insert_double_idx_heap(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_insert_double_idx_heap
|
|
|
|
subroutine psb_double_idx_heap_get_first(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_double_idx_heap_get_first
|
|
implicit none
|
|
|
|
type(psb_double_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
real(psb_dpk_), intent(out) :: key
|
|
|
|
info = psb_success_
|
|
|
|
call psi_double_idx_heap_get_first(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_double_idx_heap_get_first
|
|
|
|
function psb_howmany_int_idx_heap(heap)
|
|
use psb_sort_mod, psb_protect_name => psb_howmany_int_idx_heap
|
|
implicit none
|
|
type(psb_int_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_) :: psb_howmany_int_idx_heap
|
|
psb_howmany_int_idx_heap = heap%last
|
|
end function psb_howmany_int_idx_heap
|
|
|
|
subroutine psb_init_int_idx_heap(heap,info,dir)
|
|
use psb_sort_mod, psb_protect_name => psb_init_int_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
type(psb_int_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in), optional :: dir
|
|
|
|
info = psb_success_
|
|
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_,psb_asort_up_,psb_asort_down_)
|
|
! ok, do nothing
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction, defaulting to psb_sort_up_'
|
|
heap%dir = psb_sort_up_
|
|
end select
|
|
|
|
call psb_ensure_size(psb_heap_resize,heap%keys,info)
|
|
call psb_ensure_size(psb_heap_resize,heap%idxs,info)
|
|
return
|
|
end subroutine psb_init_int_idx_heap
|
|
|
|
subroutine psb_dump_int_idx_heap(iout,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_dump_int_idx_heap
|
|
implicit none
|
|
type(psb_int_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in) :: iout
|
|
|
|
info = psb_success_
|
|
if (iout < 0) then
|
|
write(psb_err_unit,*) '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)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else if ((heap%last > 0).and.((.not.allocated(heap%idxs)).or.&
|
|
& (size(heap%idxs)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else
|
|
if (heap%last > 0) then
|
|
write(iout,*) heap%keys(1:heap%last)
|
|
write(iout,*) heap%idxs(1:heap%last)
|
|
end if
|
|
end if
|
|
end subroutine psb_dump_int_idx_heap
|
|
|
|
subroutine psb_insert_int_idx_heap(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_insert_int_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
|
|
integer(psb_ipk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index
|
|
type(psb_int_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
info = psb_success_
|
|
if (heap%last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',heap%last
|
|
info = heap%last
|
|
return
|
|
endif
|
|
|
|
call psb_ensure_size(heap%last+1,heap%keys,info,addsz=psb_heap_resize)
|
|
if (info == psb_success_) &
|
|
& call psb_ensure_size(heap%last+1,heap%idxs,info,addsz=psb_heap_resize)
|
|
if (info /= psb_success_) then
|
|
write(psb_err_unit,*) 'Memory allocation failure in heap_insert'
|
|
info = -5
|
|
return
|
|
end if
|
|
|
|
call psi_insert_int_idx_heap(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_insert_int_idx_heap
|
|
|
|
subroutine psb_int_idx_heap_get_first(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_int_idx_heap_get_first
|
|
implicit none
|
|
|
|
type(psb_int_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
integer(psb_ipk_), intent(out) :: key
|
|
|
|
info = psb_success_
|
|
|
|
call psi_int_idx_heap_get_first(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_int_idx_heap_get_first
|
|
|
|
|
|
|
|
function psb_howmany_scomplex_idx_heap(heap)
|
|
use psb_sort_mod, psb_protect_name => psb_howmany_scomplex_idx_heap
|
|
implicit none
|
|
type(psb_scomplex_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_) :: psb_howmany_scomplex_idx_heap
|
|
psb_howmany_scomplex_idx_heap = heap%last
|
|
end function psb_howmany_scomplex_idx_heap
|
|
|
|
subroutine psb_init_scomplex_idx_heap(heap,info,dir)
|
|
use psb_sort_mod, psb_protect_name => psb_init_scomplex_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
type(psb_scomplex_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in), optional :: dir
|
|
|
|
info = psb_success_
|
|
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_,psb_asort_up_,psb_asort_down_)
|
|
case (psb_asort_up_,psb_asort_down_)
|
|
! ok, do nothing
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction, defaulting to psb_sort_up_'
|
|
heap%dir = psb_asort_up_
|
|
end select
|
|
|
|
call psb_ensure_size(psb_heap_resize,heap%keys,info)
|
|
call psb_ensure_size(psb_heap_resize,heap%idxs,info)
|
|
return
|
|
end subroutine psb_init_scomplex_idx_heap
|
|
|
|
subroutine psb_dump_scomplex_idx_heap(iout,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_dump_scomplex_idx_heap
|
|
implicit none
|
|
type(psb_scomplex_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in) :: iout
|
|
|
|
info = psb_success_
|
|
if (iout < 0) then
|
|
write(psb_err_unit,*) '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)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else if ((heap%last > 0).and.((.not.allocated(heap%idxs)).or.&
|
|
& (size(heap%idxs)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else
|
|
if (heap%last > 0) then
|
|
write(iout,*) heap%keys(1:heap%last)
|
|
write(iout,*) heap%idxs(1:heap%last)
|
|
end if
|
|
end if
|
|
end subroutine psb_dump_scomplex_idx_heap
|
|
|
|
subroutine psb_insert_scomplex_idx_heap(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_insert_scomplex_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
|
|
complex(psb_spk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index
|
|
type(psb_scomplex_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
info = psb_success_
|
|
if (heap%last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',heap%last
|
|
info = heap%last
|
|
return
|
|
endif
|
|
|
|
call psb_ensure_size(heap%last+1,heap%keys,info,addsz=psb_heap_resize)
|
|
if (info == psb_success_) &
|
|
& call psb_ensure_size(heap%last+1,heap%idxs,info,addsz=psb_heap_resize)
|
|
if (info /= psb_success_) then
|
|
write(psb_err_unit,*) 'Memory allocation failure in heap_insert'
|
|
info = -5
|
|
return
|
|
end if
|
|
call psi_insert_scomplex_idx_heap(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_insert_scomplex_idx_heap
|
|
|
|
subroutine psb_scomplex_idx_heap_get_first(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_scomplex_idx_heap_get_first
|
|
implicit none
|
|
|
|
type(psb_scomplex_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
complex(psb_spk_), intent(out) :: key
|
|
|
|
|
|
info = psb_success_
|
|
|
|
call psi_scomplex_idx_heap_get_first(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_scomplex_idx_heap_get_first
|
|
|
|
|
|
|
|
function psb_howmany_dcomplex_idx_heap(heap)
|
|
use psb_sort_mod, psb_protect_name => psb_howmany_dcomplex_idx_heap
|
|
implicit none
|
|
type(psb_dcomplex_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_) :: psb_howmany_dcomplex_idx_heap
|
|
psb_howmany_dcomplex_idx_heap = heap%last
|
|
end function psb_howmany_dcomplex_idx_heap
|
|
|
|
subroutine psb_init_dcomplex_idx_heap(heap,info,dir)
|
|
use psb_sort_mod, psb_protect_name => psb_init_dcomplex_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
type(psb_dcomplex_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in), optional :: dir
|
|
|
|
info = psb_success_
|
|
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_,psb_asort_up_,psb_asort_down_)
|
|
case (psb_asort_up_,psb_asort_down_)
|
|
! ok, do nothing
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction, defaulting to psb_sort_up_'
|
|
heap%dir = psb_asort_up_
|
|
end select
|
|
|
|
call psb_ensure_size(psb_heap_resize,heap%keys,info)
|
|
call psb_ensure_size(psb_heap_resize,heap%idxs,info)
|
|
return
|
|
end subroutine psb_init_dcomplex_idx_heap
|
|
|
|
subroutine psb_dump_dcomplex_idx_heap(iout,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_dump_dcomplex_idx_heap
|
|
implicit none
|
|
type(psb_dcomplex_idx_heap), intent(in) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_), intent(in) :: iout
|
|
|
|
info = psb_success_
|
|
if (iout < 0) then
|
|
write(psb_err_unit,*) '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)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else if ((heap%last > 0).and.((.not.allocated(heap%idxs)).or.&
|
|
& (size(heap%idxs)<heap%last))) then
|
|
write(iout,*) 'Inconsistent size/allocation status!!'
|
|
else
|
|
if (heap%last > 0) then
|
|
write(iout,*) heap%keys(1:heap%last)
|
|
write(iout,*) heap%idxs(1:heap%last)
|
|
end if
|
|
end if
|
|
end subroutine psb_dump_dcomplex_idx_heap
|
|
|
|
subroutine psb_insert_dcomplex_idx_heap(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_insert_dcomplex_idx_heap
|
|
use psb_realloc_mod
|
|
implicit none
|
|
|
|
complex(psb_dpk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index
|
|
type(psb_dcomplex_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
info = psb_success_
|
|
if (heap%last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',heap%last
|
|
info = heap%last
|
|
return
|
|
endif
|
|
|
|
call psb_ensure_size(heap%last+1,heap%keys,info,addsz=psb_heap_resize)
|
|
if (info == psb_success_) &
|
|
& call psb_ensure_size(heap%last+1,heap%idxs,info,addsz=psb_heap_resize)
|
|
if (info /= psb_success_) then
|
|
write(psb_err_unit,*) 'Memory allocation failure in heap_insert'
|
|
info = -5
|
|
return
|
|
end if
|
|
call psi_insert_dcomplex_idx_heap(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_insert_dcomplex_idx_heap
|
|
|
|
subroutine psb_dcomplex_idx_heap_get_first(key,index,heap,info)
|
|
use psb_sort_mod, psb_protect_name => psb_dcomplex_idx_heap_get_first
|
|
implicit none
|
|
|
|
type(psb_dcomplex_idx_heap), intent(inout) :: heap
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
complex(psb_dpk_), intent(out) :: key
|
|
|
|
|
|
info = psb_success_
|
|
|
|
call psi_dcomplex_idx_heap_get_first(key,index,&
|
|
& heap%last,heap%keys,heap%idxs,heap%dir,info)
|
|
|
|
return
|
|
end subroutine psb_dcomplex_idx_heap_get_first
|
|
|
|
|
|
|
|
!
|
|
! These are packaged so that they can be used to implement
|
|
! a heapsort, should the need arise
|
|
!
|
|
|
|
|
|
subroutine psi_insert_int_heap(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_int_heap
|
|
implicit none
|
|
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! dir: sorting direction
|
|
|
|
integer(psb_ipk_), intent(in) :: key,dir
|
|
integer(psb_ipk_), intent(inout) :: heap(:),last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2
|
|
integer(psb_ipk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
i = last
|
|
heap(i) = key
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) < heap(i2)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) > heap(i2)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_int_heap
|
|
|
|
|
|
subroutine psi_int_heap_get_first(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_int_heap_get_first
|
|
implicit none
|
|
|
|
integer(psb_ipk_), intent(inout) :: key,last
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
integer(psb_ipk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
integer(psb_ipk_) :: i, j
|
|
integer(psb_ipk_) :: temp
|
|
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
heap(1) = heap(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) > heap(j)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) < heap(j)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_int_heap_get_first
|
|
|
|
|
|
|
|
subroutine psi_insert_real_heap(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_real_heap
|
|
implicit none
|
|
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! dir: sorting direction
|
|
|
|
real(psb_spk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
real(psb_spk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(inout) :: last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2
|
|
real(psb_spk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
i = last
|
|
heap(i) = key
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) < heap(i2)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) > heap(i2)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_real_heap
|
|
|
|
|
|
subroutine psi_real_heap_get_first(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_real_heap_get_first
|
|
implicit none
|
|
|
|
real(psb_spk_), intent(inout) :: key
|
|
integer(psb_ipk_), intent(inout) :: last
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
real(psb_spk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
integer(psb_ipk_) :: i, j
|
|
real(psb_spk_) :: temp
|
|
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
heap(1) = heap(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) > heap(j)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) < heap(j)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_real_heap_get_first
|
|
|
|
|
|
subroutine psi_insert_double_heap(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_double_heap
|
|
implicit none
|
|
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! dir: sorting direction
|
|
|
|
real(psb_dpk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
real(psb_dpk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(inout) :: last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2
|
|
real(psb_dpk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
i = last
|
|
heap(i) = key
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) < heap(i2)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) > heap(i2)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_double_heap
|
|
|
|
|
|
subroutine psi_double_heap_get_first(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_double_heap_get_first
|
|
implicit none
|
|
|
|
real(psb_dpk_), intent(inout) :: key
|
|
integer(psb_ipk_), intent(inout) :: last
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
real(psb_dpk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
integer(psb_ipk_) :: i, j
|
|
real(psb_dpk_) :: temp
|
|
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
heap(1) = heap(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) > heap(j)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) < heap(j)) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_double_heap_get_first
|
|
|
|
|
|
|
|
|
|
subroutine psi_insert_scomplex_heap(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_scomplex_heap
|
|
implicit none
|
|
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! dir: sorting direction
|
|
|
|
complex(psb_spk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
complex(psb_spk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(inout) :: last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2
|
|
complex(psb_spk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
i = last
|
|
heap(i) = key
|
|
|
|
select case(dir)
|
|
!!$ case (psb_sort_up_)
|
|
!!$
|
|
!!$ do
|
|
!!$ if (i<=1) exit
|
|
!!$ i2 = i/2
|
|
!!$ if (heap(i) < heap(i2)) then
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(i2)
|
|
!!$ heap(i2) = temp
|
|
!!$ i = i2
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
!!$
|
|
!!$
|
|
!!$ case (psb_sort_down_)
|
|
!!$
|
|
!!$ do
|
|
!!$ if (i<=1) exit
|
|
!!$ i2 = i/2
|
|
!!$ if (heap(i) > heap(i2)) then
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(i2)
|
|
!!$ heap(i2) = temp
|
|
!!$ i = i2
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_scomplex_heap
|
|
|
|
|
|
subroutine psi_scomplex_heap_get_first(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_scomplex_heap_get_first
|
|
implicit none
|
|
|
|
complex(psb_spk_), intent(inout) :: key
|
|
integer(psb_ipk_), intent(inout) :: last
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
complex(psb_spk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
integer(psb_ipk_) :: i, j
|
|
complex(psb_spk_) :: temp
|
|
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
heap(1) = heap(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
!!$ case (psb_sort_up_)
|
|
!!$
|
|
!!$ i = 1
|
|
!!$ do
|
|
!!$ if (i > (last/2)) exit
|
|
!!$ if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
!!$ & (2*i == last)) then
|
|
!!$ j = 2*i
|
|
!!$ else
|
|
!!$ j = 2*i + 1
|
|
!!$ end if
|
|
!!$
|
|
!!$ if (heap(i) > heap(j)) then
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(j)
|
|
!!$ heap(j) = temp
|
|
!!$ i = j
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
!!$
|
|
!!$
|
|
!!$ case (psb_sort_down_)
|
|
!!$
|
|
!!$ i = 1
|
|
!!$ do
|
|
!!$ if (i > (last/2)) exit
|
|
!!$ if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
!!$ & (2*i == last)) then
|
|
!!$ j = 2*i
|
|
!!$ else
|
|
!!$ j = 2*i + 1
|
|
!!$ end if
|
|
!!$
|
|
!!$ if (heap(i) < heap(j)) then
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(j)
|
|
!!$ heap(j) = temp
|
|
!!$ i = j
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_scomplex_heap_get_first
|
|
|
|
|
|
subroutine psi_insert_dcomplex_heap(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_dcomplex_heap
|
|
implicit none
|
|
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! dir: sorting direction
|
|
|
|
complex(psb_dpk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
complex(psb_dpk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(inout) :: last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2
|
|
complex(psb_dpk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
i = last
|
|
heap(i) = key
|
|
|
|
select case(dir)
|
|
!!$ case (psb_sort_up_)
|
|
!!$
|
|
!!$ do
|
|
!!$ if (i<=1) exit
|
|
!!$ i2 = i/2
|
|
!!$ if (heap(i) < heap(i2)) then
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(i2)
|
|
!!$ heap(i2) = temp
|
|
!!$ i = i2
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
!!$
|
|
!!$
|
|
!!$ case (psb_sort_down_)
|
|
!!$
|
|
!!$ do
|
|
!!$ if (i<=1) exit
|
|
!!$ i2 = i/2
|
|
!!$ if (heap(i) > heap(i2)) then
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(i2)
|
|
!!$ heap(i2) = temp
|
|
!!$ i = i2
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_dcomplex_heap
|
|
|
|
|
|
subroutine psi_dcomplex_heap_get_first(key,last,heap,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_dcomplex_heap_get_first
|
|
implicit none
|
|
|
|
complex(psb_dpk_), intent(inout) :: key
|
|
integer(psb_ipk_), intent(inout) :: last
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
complex(psb_dpk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: info
|
|
|
|
integer(psb_ipk_) :: i, j
|
|
complex(psb_dpk_) :: temp
|
|
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
heap(1) = heap(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
!!$ case (psb_sort_up_)
|
|
!!$
|
|
!!$ i = 1
|
|
!!$ do
|
|
!!$ if (i > (last/2)) exit
|
|
!!$ if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
!!$ & (2*i == last)) then
|
|
!!$ j = 2*i
|
|
!!$ else
|
|
!!$ j = 2*i + 1
|
|
!!$ end if
|
|
!!$
|
|
!!$ if (heap(i) > heap(j)) then
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(j)
|
|
!!$ heap(j) = temp
|
|
!!$ i = j
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
!!$
|
|
!!$
|
|
!!$ case (psb_sort_down_)
|
|
!!$
|
|
!!$ i = 1
|
|
!!$ do
|
|
!!$ if (i > (last/2)) exit
|
|
!!$ if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
!!$ & (2*i == last)) then
|
|
!!$ j = 2*i
|
|
!!$ else
|
|
!!$ j = 2*i + 1
|
|
!!$ end if
|
|
!!$
|
|
!!$ if (heap(i) < heap(j)) then
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(j)
|
|
!!$ heap(j) = temp
|
|
!!$ i = j
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_dcomplex_heap_get_first
|
|
|
|
|
|
|
|
|
|
subroutine psi_insert_int_idx_heap(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_int_idx_heap
|
|
|
|
implicit none
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! index: the new index
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! idxs: the indices
|
|
! dir: sorting direction
|
|
|
|
integer(psb_ipk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index,dir
|
|
integer(psb_ipk_), intent(inout) :: heap(:),last
|
|
integer(psb_ipk_), intent(inout) :: idxs(:)
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2, itemp
|
|
integer(psb_ipk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
|
|
i = last
|
|
heap(i) = key
|
|
idxs(i) = index
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) < heap(i2)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) > heap(i2)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_int_idx_heap
|
|
|
|
subroutine psi_int_idx_heap_get_first(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_int_idx_heap_get_first
|
|
implicit none
|
|
|
|
integer(psb_ipk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
integer(psb_ipk_), intent(inout) :: last,idxs(:)
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
integer(psb_ipk_), intent(out) :: key
|
|
|
|
integer(psb_ipk_) :: i, j,itemp
|
|
integer(psb_ipk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
index = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
index = idxs(1)
|
|
heap(1) = heap(last)
|
|
idxs(1) = idxs(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) > heap(j)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) < heap(j)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_int_idx_heap_get_first
|
|
|
|
subroutine psi_insert_real_idx_heap(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_real_idx_heap
|
|
|
|
implicit none
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! index: the new index
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! idxs: the indices
|
|
! dir: sorting direction
|
|
|
|
real(psb_spk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index,dir
|
|
real(psb_spk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(inout) :: idxs(:),last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2, itemp
|
|
real(psb_spk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
|
|
i = last
|
|
heap(i) = key
|
|
idxs(i) = index
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) < heap(i2)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) > heap(i2)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_real_idx_heap
|
|
|
|
subroutine psi_real_idx_heap_get_first(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_real_idx_heap_get_first
|
|
implicit none
|
|
|
|
real(psb_spk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
integer(psb_ipk_), intent(inout) :: last,idxs(:)
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
real(psb_spk_), intent(out) :: key
|
|
|
|
integer(psb_ipk_) :: i, j,itemp
|
|
real(psb_spk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
index = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
index = idxs(1)
|
|
heap(1) = heap(last)
|
|
idxs(1) = idxs(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) > heap(j)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) < heap(j)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_real_idx_heap_get_first
|
|
|
|
|
|
subroutine psi_insert_double_idx_heap(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_double_idx_heap
|
|
|
|
implicit none
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! index: the new index
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! idxs: the indices
|
|
! dir: sorting direction
|
|
|
|
real(psb_dpk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index,dir
|
|
real(psb_dpk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(inout) :: idxs(:),last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2, itemp
|
|
real(psb_dpk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
|
|
i = last
|
|
heap(i) = key
|
|
idxs(i) = index
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) < heap(i2)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (heap(i) > heap(i2)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_double_idx_heap
|
|
|
|
subroutine psi_double_idx_heap_get_first(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_double_idx_heap_get_first
|
|
implicit none
|
|
|
|
real(psb_dpk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
integer(psb_ipk_), intent(inout) :: last,idxs(:)
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
real(psb_dpk_), intent(out) :: key
|
|
|
|
integer(psb_ipk_) :: i, j,itemp
|
|
real(psb_dpk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
index = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
index = idxs(1)
|
|
heap(1) = heap(last)
|
|
idxs(1) = idxs(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
case (psb_sort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) > heap(j)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_sort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (heap(i) < heap(j)) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_double_idx_heap_get_first
|
|
|
|
|
|
subroutine psi_insert_scomplex_idx_heap(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_scomplex_idx_heap
|
|
|
|
implicit none
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! index: the new index
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! idxs: the indices
|
|
! dir: sorting direction
|
|
|
|
complex(psb_spk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index,dir
|
|
complex(psb_spk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(inout) :: idxs(:),last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2, itemp
|
|
complex(psb_spk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
|
|
i = last
|
|
heap(i) = key
|
|
idxs(i) = index
|
|
|
|
select case(dir)
|
|
!!$ case (psb_sort_up_)
|
|
!!$
|
|
!!$ do
|
|
!!$ if (i<=1) exit
|
|
!!$ i2 = i/2
|
|
!!$ if (heap(i) < heap(i2)) then
|
|
!!$ itemp = idxs(i)
|
|
!!$ idxs(i) = idxs(i2)
|
|
!!$ idxs(i2) = itemp
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(i2)
|
|
!!$ heap(i2) = temp
|
|
!!$ i = i2
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
!!$
|
|
!!$
|
|
!!$ case (psb_sort_down_)
|
|
!!$
|
|
!!$ do
|
|
!!$ if (i<=1) exit
|
|
!!$ i2 = i/2
|
|
!!$ if (heap(i) > heap(i2)) then
|
|
!!$ itemp = idxs(i)
|
|
!!$ idxs(i) = idxs(i2)
|
|
!!$ idxs(i2) = itemp
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(i2)
|
|
!!$ heap(i2) = temp
|
|
!!$ i = i2
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_scomplex_idx_heap
|
|
|
|
subroutine psi_scomplex_idx_heap_get_first(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_scomplex_idx_heap_get_first
|
|
implicit none
|
|
|
|
complex(psb_spk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
integer(psb_ipk_), intent(inout) :: last,idxs(:)
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
complex(psb_spk_), intent(out) :: key
|
|
|
|
integer(psb_ipk_) :: i, j, itemp
|
|
complex(psb_spk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
index = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
index = idxs(1)
|
|
heap(1) = heap(last)
|
|
idxs(1) = idxs(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
!!$ case (psb_sort_up_)
|
|
!!$
|
|
!!$ i = 1
|
|
!!$ do
|
|
!!$ if (i > (last/2)) exit
|
|
!!$ if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
!!$ & (2*i == last)) then
|
|
!!$ j = 2*i
|
|
!!$ else
|
|
!!$ j = 2*i + 1
|
|
!!$ end if
|
|
!!$
|
|
!!$ if (heap(i) > heap(j)) then
|
|
!!$ itemp = idxs(i)
|
|
!!$ idxs(i) = idxs(j)
|
|
!!$ idxs(j) = itemp
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(j)
|
|
!!$ heap(j) = temp
|
|
!!$ i = j
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
!!$
|
|
!!$
|
|
!!$ case (psb_sort_down_)
|
|
!!$
|
|
!!$ i = 1
|
|
!!$ do
|
|
!!$ if (i > (last/2)) exit
|
|
!!$ if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
!!$ & (2*i == last)) then
|
|
!!$ j = 2*i
|
|
!!$ else
|
|
!!$ j = 2*i + 1
|
|
!!$ end if
|
|
!!$
|
|
!!$ if (heap(i) < heap(j)) then
|
|
!!$ itemp = idxs(i)
|
|
!!$ idxs(i) = idxs(j)
|
|
!!$ idxs(j) = itemp
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(j)
|
|
!!$ heap(j) = temp
|
|
!!$ i = j
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_scomplex_idx_heap_get_first
|
|
|
|
|
|
subroutine psi_insert_dcomplex_idx_heap(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_insert_dcomplex_idx_heap
|
|
|
|
implicit none
|
|
!
|
|
! Input:
|
|
! key: the new value
|
|
! index: the new index
|
|
! last: pointer to the last occupied element in heap
|
|
! heap: the heap
|
|
! idxs: the indices
|
|
! dir: sorting direction
|
|
|
|
complex(psb_dpk_), intent(in) :: key
|
|
integer(psb_ipk_), intent(in) :: index,dir
|
|
complex(psb_dpk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(inout) :: idxs(:),last
|
|
integer(psb_ipk_), intent(out) :: info
|
|
integer(psb_ipk_) :: i, i2, itemp
|
|
complex(psb_dpk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last < 0) then
|
|
write(psb_err_unit,*) 'Invalid last in heap ',last
|
|
info = last
|
|
return
|
|
endif
|
|
|
|
last = last + 1
|
|
if (last > size(heap)) then
|
|
write(psb_err_unit,*) 'out of bounds '
|
|
info = -1
|
|
return
|
|
end if
|
|
|
|
i = last
|
|
heap(i) = key
|
|
idxs(i) = index
|
|
|
|
select case(dir)
|
|
!!$ case (psb_sort_up_)
|
|
!!$
|
|
!!$ do
|
|
!!$ if (i<=1) exit
|
|
!!$ i2 = i/2
|
|
!!$ if (heap(i) < heap(i2)) then
|
|
!!$ itemp = idxs(i)
|
|
!!$ idxs(i) = idxs(i2)
|
|
!!$ idxs(i2) = itemp
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(i2)
|
|
!!$ heap(i2) = temp
|
|
!!$ i = i2
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
!!$
|
|
!!$
|
|
!!$ case (psb_sort_down_)
|
|
!!$
|
|
!!$ do
|
|
!!$ if (i<=1) exit
|
|
!!$ i2 = i/2
|
|
!!$ if (heap(i) > heap(i2)) then
|
|
!!$ itemp = idxs(i)
|
|
!!$ idxs(i) = idxs(i2)
|
|
!!$ idxs(i2) = itemp
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(i2)
|
|
!!$ heap(i2) = temp
|
|
!!$ i = i2
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) < abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
do
|
|
if (i<=1) exit
|
|
i2 = i/2
|
|
if (abs(heap(i)) > abs(heap(i2))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(i2)
|
|
idxs(i2) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(i2)
|
|
heap(i2) = temp
|
|
i = i2
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_insert_dcomplex_idx_heap
|
|
|
|
subroutine psi_dcomplex_idx_heap_get_first(key,index,last,heap,idxs,dir,info)
|
|
use psb_sort_mod, psb_protect_name => psi_dcomplex_idx_heap_get_first
|
|
implicit none
|
|
|
|
complex(psb_dpk_), intent(inout) :: heap(:)
|
|
integer(psb_ipk_), intent(out) :: index,info
|
|
integer(psb_ipk_), intent(inout) :: last,idxs(:)
|
|
integer(psb_ipk_), intent(in) :: dir
|
|
complex(psb_dpk_), intent(out) :: key
|
|
|
|
integer(psb_ipk_) :: i, j, itemp
|
|
complex(psb_dpk_) :: temp
|
|
|
|
info = psb_success_
|
|
if (last <= 0) then
|
|
key = 0
|
|
index = 0
|
|
info = -1
|
|
return
|
|
endif
|
|
|
|
key = heap(1)
|
|
index = idxs(1)
|
|
heap(1) = heap(last)
|
|
idxs(1) = idxs(last)
|
|
last = last - 1
|
|
|
|
select case(dir)
|
|
!!$ case (psb_sort_up_)
|
|
!!$
|
|
!!$ i = 1
|
|
!!$ do
|
|
!!$ if (i > (last/2)) exit
|
|
!!$ if ( (heap(2*i) < heap(2*i+1)) .or.&
|
|
!!$ & (2*i == last)) then
|
|
!!$ j = 2*i
|
|
!!$ else
|
|
!!$ j = 2*i + 1
|
|
!!$ end if
|
|
!!$
|
|
!!$ if (heap(i) > heap(j)) then
|
|
!!$ itemp = idxs(i)
|
|
!!$ idxs(i) = idxs(j)
|
|
!!$ idxs(j) = itemp
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(j)
|
|
!!$ heap(j) = temp
|
|
!!$ i = j
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
!!$
|
|
!!$
|
|
!!$ case (psb_sort_down_)
|
|
!!$
|
|
!!$ i = 1
|
|
!!$ do
|
|
!!$ if (i > (last/2)) exit
|
|
!!$ if ( (heap(2*i) > heap(2*i+1)) .or.&
|
|
!!$ & (2*i == last)) then
|
|
!!$ j = 2*i
|
|
!!$ else
|
|
!!$ j = 2*i + 1
|
|
!!$ end if
|
|
!!$
|
|
!!$ if (heap(i) < heap(j)) then
|
|
!!$ itemp = idxs(i)
|
|
!!$ idxs(i) = idxs(j)
|
|
!!$ idxs(j) = itemp
|
|
!!$ temp = heap(i)
|
|
!!$ heap(i) = heap(j)
|
|
!!$ heap(j) = temp
|
|
!!$ i = j
|
|
!!$ else
|
|
!!$ exit
|
|
!!$ end if
|
|
!!$ end do
|
|
|
|
case (psb_asort_up_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) < abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) > abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
|
|
case (psb_asort_down_)
|
|
|
|
i = 1
|
|
do
|
|
if (i > (last/2)) exit
|
|
if ( (abs(heap(2*i)) > abs(heap(2*i+1))) .or.&
|
|
& (2*i == last)) then
|
|
j = 2*i
|
|
else
|
|
j = 2*i + 1
|
|
end if
|
|
|
|
if (abs(heap(i)) < abs(heap(j))) then
|
|
itemp = idxs(i)
|
|
idxs(i) = idxs(j)
|
|
idxs(j) = itemp
|
|
temp = heap(i)
|
|
heap(i) = heap(j)
|
|
heap(j) = temp
|
|
i = j
|
|
else
|
|
exit
|
|
end if
|
|
end do
|
|
|
|
case default
|
|
write(psb_err_unit,*) 'Invalid direction in heap ',dir
|
|
end select
|
|
|
|
return
|
|
end subroutine psi_dcomplex_idx_heap_get_first
|
|
|
|
|
|
|