pe2set_mod.F90

Go to the documentation of this file.

MODULE pe2set_mod
CONTAINS
SUBROUTINE pe2set(KPE,KPRGPNS,KPRGPEW,KPRTRW,KPRTRV)

#ifdef DOC

!**** *PE2SET* - Convert from PE number to set numbers

!     Purpose.
!     --------
!        Convert from PE number to set numbers in both
!                  grid-point space and spectral space

!**   Interface.
!     ----------
!        *CALL* *PE2SET(KPE,KPRGPNS,KPRGPEW,KPRTRW,KPRTRV)

!        Explicit arguments :  
!        --------------------
!                  input:   KPE     - integer processor number 
!                                     in the range 1 .. NPROC
!                  output:  KPRGPNS - integer A set number in grid space
!                                     in the range 1 .. NPRGPNS
!                           KPRGPEW - integer B set number in grid space
!                                     in the range 1 .. NPRGPEW
!                           KPRTRW  - integer A set number in spectral space
!                                     in the range 1 .. NPRTRW 
!                           KPRTRV  - integer B set number in spectral space
!                                     in the range 1 .. NPRTRV 

!        Implicit arguments :  YOMMP parameters
!                              NPRGPNS,NPRGPEW,NPRTRW,NPRTRV,NPROC

!        --------------------
!     Method.
!     -------

!        PE allocation order is row oriented (e.g. NPRGPNS or NPRTRW = 4):

!                1  2  3  4 
!                5  6  7  8 
!                9 10 11 12 
!               13 14 15 16 
!                .  .  .  .

!     Externals.
!     ----------
!         NONE

!     Reference.
!     ----------
!        ECMWF Research Department documentation of the IFS

!     Author.
!     -------
!        David Dent *ECMWF*

!     Modifications.
!     --------------
!        Original : 98-08-19
!        Revision : 98-10-13 row ordering
!     ------------------------------------------------------------------
#endif

USE parkind1  ,ONLY : jpim     ,jprb

USE tpm_distr
USE eq_regions_mod
USE abort_trans_mod


IMPLICIT NONE
INTEGER(KIND=JPIM),INTENT(IN)  :: KPE
INTEGER(KIND=JPIM),INTENT(OUT) :: KPRGPNS,KPRGPEW,KPRTRW,KPRTRV

INTEGER(KIND=JPIM) :: IPE,JA
!     ------------------------------------------------------------------

!*       1.    Check input argument for validity 
!              ---------------------------------

IF(kpe <= 0.OR.kpe > nproc) THEN
  WRITE(*,'(A,2I8)') ' PE2SET INVALID ARGUMENT ',kpe,nproc
  CALL abort_trans(' PE2SET INVALID ARGUMENT ')

ELSE

!*       2.    Compute output parameters
!              -------------------------

  IF( leq_regions )THEN
    kprgpns=1
    ipe=kpe
    DO ja=1,n_regions_ns
      IF( ipe > n_regions(ja) )THEN
        ipe=ipe-n_regions(ja)
        kprgpns=kprgpns+1
        cycle
      ENDIF
      kprgpew=ipe
      EXIT
    ENDDO
  ELSE
    kprgpew=mod(kpe-1,nprgpew)+1
    kprgpns=(kpe-1)/nprgpew+1
  ENDIF
  kprtrv =mod(kpe-1,nprtrv)+1
  kprtrw =(kpe-1)/nprtrv+1

ENDIF

END SUBROUTINE pe2set
END MODULE pe2set_mod