| 1 | MODULE zpermute |
|---|
| 2 | !!===================================================================== |
|---|
| 3 | !! *** MODULE zpermute *** |
|---|
| 4 | !! Manipulate array : Library to read input files |
|---|
| 5 | !!==================================================================== |
|---|
| 6 | !! History : 3.3.1 ! 2011 11 (S. Pickles) Original code |
|---|
| 7 | !! |
|---|
| 8 | !!-------------------------------------------------------------------- |
|---|
| 9 | !! permute_z_last : return pointer to a copy of an array with indices |
|---|
| 10 | !! permuted back to z_last ordering |
|---|
| 11 | !!-------------------------------------------------------------------- |
|---|
| 12 | USE par_oce, ONLY : wp ! ocean parameters - working precision |
|---|
| 13 | USE lib_mpp, ONLY : ctl_warn, ctl_stop ! warning and fatal messages |
|---|
| 14 | |
|---|
| 15 | IMPLICIT NONE |
|---|
| 16 | PRIVATE |
|---|
| 17 | |
|---|
| 18 | PUBLIC permute_z_last |
|---|
| 19 | |
|---|
| 20 | !!---------------------------------------------------------------------- |
|---|
| 21 | !! NEMO/OPA 3.3 , NEMO Consortium (2011) |
|---|
| 22 | !! $Id$ |
|---|
| 23 | !! Software governed by the CeCILL licence (NEMOGCM/NEMO_CeCILL.txt) |
|---|
| 24 | !!---------------------------------------------------------------------- |
|---|
| 25 | |
|---|
| 26 | CONTAINS |
|---|
| 27 | |
|---|
| 28 | FUNCTION permute_z_last( pain ) RESULT ( p_permuted ) |
|---|
| 29 | |
|---|
| 30 | !!---------------------------------------------------------------------- |
|---|
| 31 | !! *** ROUTINE permute_z_last *** |
|---|
| 32 | !! |
|---|
| 33 | !! ** Purpose : |
|---|
| 34 | !! |
|---|
| 35 | !!---------------------------------------------------------------------- |
|---|
| 36 | |
|---|
| 37 | REAL(wp), INTENT(in) :: pain(:,:,:) |
|---|
| 38 | |
|---|
| 39 | REAL(wp), ALLOCATABLE, SAVE, DIMENSION(:,:,:), TARGET :: pain_reordered |
|---|
| 40 | REAL(wp), POINTER, DIMENSION(:,:,:) :: p_permuted |
|---|
| 41 | |
|---|
| 42 | INTEGER, SAVE :: jiss=-1, jjss=-1, jkss=-1 |
|---|
| 43 | INTEGER :: jisz, jjsz, jksz |
|---|
| 44 | INTEGER :: ji, jj, jk |
|---|
| 45 | INTEGER :: ifail |
|---|
| 46 | |
|---|
| 47 | #if defined key_z_first |
|---|
| 48 | jisz = SIZE(pain, DIM=2) |
|---|
| 49 | jjsz = SIZE(pain, DIM=3) |
|---|
| 50 | jksz = SIZE(pain, DIM=1) |
|---|
| 51 | #else |
|---|
| 52 | jisz = SIZE(pain, DIM=1) |
|---|
| 53 | jjsz = SIZE(pain, DIM=2) |
|---|
| 54 | jksz = SIZE(pain, DIM=3) |
|---|
| 55 | #endif |
|---|
| 56 | |
|---|
| 57 | IF ( (jiss /= jisz) .OR. (jjss /= jjsz) .OR. (jkss /= jksz)) THEN |
|---|
| 58 | IF (ALLOCATED(pain_reordered)) THEN |
|---|
| 59 | DEALLOCATE(pain_reordered) |
|---|
| 60 | END IF |
|---|
| 61 | ELSE |
|---|
| 62 | IF (.NOT. ALLOCATED(pain_reordered)) THEN |
|---|
| 63 | CALL ctl_warn("permute_z_last: unexpected need to allocate array") |
|---|
| 64 | END IF |
|---|
| 65 | END IF |
|---|
| 66 | |
|---|
| 67 | IF (.NOT. ALLOCATED(pain_reordered)) THEN |
|---|
| 68 | ALLOCATE(pain_reordered(jisz,jjsz,jksz), STAT=ifail) |
|---|
| 69 | IF (ifail /= 0) THEN |
|---|
| 70 | CALL ctl_stop("permute_z_last: failure allocating output array") |
|---|
| 71 | END IF |
|---|
| 72 | END IF |
|---|
| 73 | |
|---|
| 74 | DO jk = 1, jksz |
|---|
| 75 | DO jj = 1, jjsz |
|---|
| 76 | DO ji = 1, jisz |
|---|
| 77 | !FTRANS pain :I :I :z |
|---|
| 78 | pain_reordered(ji, jj, jk) = pain(ji, jj, jk) |
|---|
| 79 | END DO |
|---|
| 80 | END DO |
|---|
| 81 | END DO |
|---|
| 82 | |
|---|
| 83 | jiss = jisz |
|---|
| 84 | jjss = jjsz |
|---|
| 85 | jkss = jksz |
|---|
| 86 | p_permuted => pain_reordered |
|---|
| 87 | |
|---|
| 88 | END FUNCTION permute_z_last |
|---|
| 89 | |
|---|
| 90 | !!====================================================================== |
|---|
| 91 | END MODULE zpermute |
|---|
