set2pe_mod.F90

Go to the documentation of this file.

MODULE set2pe_mod
CONTAINS
SUBROUTINE set2pe(KPE,KPRGPNS,KPRGPEW,KPRTRW,KPRTRV)

#ifdef DOC

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

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

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

!        Explicit arguments :  
!        --------------------

!                  input :  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 
!                  output:  KPE     - integer processor number 
!                                     in the range 1 .. NPROC

!                  Normally, one pair of input set numbers will be set to zero
!                  SET2PE will compute KPE from the first pair if they are valid numbers.
!                  else from the other pair,

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

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

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

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

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

!     Modifications.
!     --------------
!        Original : 98-08-19
!     ------------------------------------------------------------------
#endif

USE parkind1  ,ONLY : jpim     ,jprb

USE tpm_distr
USE eq_regions_mod
USE abort_trans_mod

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

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

!*       1.    Choose from input parameters
!              ----------------------------

IF(kprgpns > 0.AND.kprgpew > 0) THEN

  IF( leq_regions )THEN
    IF( kprgpns > n_regions_ns )THEN
      WRITE(*,'(A,2I8)') ' SET2PE INVALID ARGUMENT ',kprgpns,n_regions_ns
      CALL abor1(' SET2PE INVALID ARGUMENT ')
    ENDIF
    IF( kprgpew > n_regions(kprgpns) )THEN
      WRITE(*,'(A,2I8)') ' SET2PE INVALID ARGUMENT ',kprgpew,n_regions(kprgpns)
      CALL abor1(' SET2PE INVALID ARGUMENT ')
    ENDIF
    kpe=0
    DO ja=1,kprgpns-1
      kpe=kpe+n_regions(ja)
    ENDDO
    kpe=kpe+kprgpew    
  ELSE
    IF(kprgpns <= nprgpns.AND.kprgpew <= nprgpew) THEN

!*       2.    Grid-space set values supplied
!              ------------------------------

      kpe=(kprgpns-1)*nprgpew + kprgpew
    ELSE
      WRITE(*,'(A,2I8)') ' SET2PE INVALID ARGUMENT ',kprgpns,kprgpew
      CALL abort_trans(' SET2PE INVALID ARGUMENT ')
    ENDIF
  ENDIF

ELSE

!*       3.    Spectral space set values supplied
!              ----------------------------------

  IF(kprtrw <= nprtrw.AND.kprtrv <= nprtrv) THEN
    kpe=(kprtrw-1)*nprtrv + kprtrv
  ELSE
    WRITE(*,'(A,2I8)') ' SET2PE INVALID ARGUMENT ',kprtrw,kprtrv
    CALL abort_trans(' SET2PE INVALID ARGUMENT ')
  ENDIF

ENDIF

END SUBROUTINE set2pe
END MODULE set2pe_mod