Changeset 2286 in lmdz_wrf for trunk/tools

Timestamp:

Jan 24, 2019, 3:02:17 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `multi_index_mat2DI': Subroutine to provide the indices of the different locations of a value inside a 2D integer matrix
• `multi_index_mat3DI': Subroutine to provide the indices of the different locations of a value inside a 3D integer matrix
• `multi_index_mat4DI': Subroutine to provide the indices of the different locations of a value inside a 4D integer matrix

File:

• trunk/tools/module_scientific.f90 (modified) (2 diffs)

trunk/tools/module_scientific.f90

@@ -28 +28 @@
 ! look_clockwise_borders: Subroutine to look clock-wise for a next point within a collection of borders 
 !   (limits of a region)
+! multi_index_mat2DI: Subroutine to provide the indices of the different locations of a value inside a 2D integer matrix
+! multi_index_mat3DI: Subroutine to provide the indices of the different locations of a value inside a 3D integer matrix
+! multi_index_mat4DI: Subroutine to provide the indices of the different locations of a value inside a 4D integer matrix
 ! multi_index_mat2DRK: Subroutine to provide the indices of the different locations of a value inside a 2D RK matrix
 ! multi_index_mat3DRK: Subroutine to provide the indices of the different locations of a value inside a 3D RK matrix
@@ -5211 +5214 @@
   END SUBROUTINE spaceweightstats
 
+  SUBROUTINE multi_index_mat2DI(d1, d2, d12, mat, value, Nindices, indices)
+  ! Subroutine to provide the indices of the different locations of a value inside a 2D integer matrix
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: d1, d2, d12
+    INTEGER, DIMENSION(d1,d2), INTENT(in)                :: mat
+    INTEGER,INTENT(in)                                   :: value
+    INTEGER, INTENT(out)                                 :: Nindices
+    INTEGER, DIMENSION(2,d12), INTENT(out)               :: indices
+
+! Local
+    INTEGER                                              :: i,j
+    INTEGER                                              :: Ncounts1D, icount1D
+    INTEGER, DIMENSION(d2)                               :: diffmat1D
+
+    !!!!!!! Variables
+    ! d1, d2: shape of the 2D matrix
+    ! mat: 2D matrix
+    ! value: value to be looking for
+    ! Nindices: number of times value found within matrix
+    ! indices: indices of the found values
+
+    fname = 'multi_index_mat2DI'
+
+    Nindices = 0
+    indices = 0
+    DO i=1, d1
+      diffmat1D = mat(i,:) - value
+      IF (ANY(diffmat1D == 0)) THEN
+        Ncounts1D = COUNT(diffmat1D == 0)
+        icount1D = 0
+        DO j=1, d2
+          IF (diffmat1D(j) == 0) THEN
+            Nindices = Nindices + 1
+            indices(1,Nindices) = i
+            indices(2,Nindices) = j
+            icount1D = icount1D + 1
+            IF (icount1D == Ncounts1D) EXIT
+          END IF
+        END DO
+      END IF
+    END DO
+
+  END SUBROUTINE multi_index_mat2DI
+
+  SUBROUTINE multi_index_mat3DI(d1, d2, d3, d123, mat, value, Nindices, indices)
+  ! Subroutine to provide the indices of the different locations of a value inside a 3D integer matrix
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: d1, d2, d3, d123
+    INTEGER, DIMENSION(d1,d2,d3), INTENT(in)             :: mat
+    INTEGER, INTENT(in)                                  :: value
+    INTEGER, INTENT(out)                                 :: Nindices
+    INTEGER, DIMENSION(3,d123), INTENT(out)              :: indices
+
+! Local
+    INTEGER                                              :: i,j,k
+    INTEGER                                              :: Ncounts1D, icount1D
+    INTEGER                                              :: Ncounts2D, icount2D
+    INTEGER, DIMENSION(d2,d3)                            :: diffmat2D
+    INTEGER, DIMENSION(d3)                               :: diffmat1D
+
+    !!!!!!! Variables
+    ! d1, d2, d3: shape of the 3D matrix
+    ! mat: 3D matrix
+    ! value: value to be looking for
+    ! Nindices: number of times value found within matrix
+    ! indices: indices of the found values
+
+    fname = 'multi_index_mat3DI'
+
+    Nindices = 0
+    indices = 0
+    DO i=1, d1
+      diffmat2D = mat(i,:,:) - value
+      IF (ANY(diffmat2D == 0)) THEN
+        Ncounts2D = COUNT(diffmat2D == 0)
+        icount2D = 0
+        DO j=1, d2
+          diffmat1D = mat(i,j,:) - value
+          IF (ANY(diffmat1D == 0)) THEN
+            Ncounts1D = COUNT(diffmat1D == 0)
+            icount1D = 0
+            DO k=1, d3
+              IF (diffmat1D(k) == 0) THEN
+                Nindices = Nindices + 1
+                indices(1,Nindices) = i
+                indices(2,Nindices) = j
+                indices(3,Nindices) = k
+                icount1D = icount1D + 1
+                IF (icount1D == Ncounts1D) EXIT
+              END IF
+            END DO
+            icount2D = icount2D + icount1D
+            IF (icount2D == Ncounts2D) EXIT
+          END IF
+        END DO
+      END IF
+    END DO
+
+  END SUBROUTINE multi_index_mat3DI
+
+  SUBROUTINE multi_index_mat4DI(d1, d2, d3, d4, d1234, mat, value, Nindices, indices)
+  ! Subroutine to provide the indices of the different locations of a value inside a 4D integer matrix
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: d1, d2, d3, d4, d1234
+    INTEGER, DIMENSION(d1,d2,d3,d4), INTENT(in)          :: mat
+    INTEGER, INTENT(in)                                  :: value
+    INTEGER, INTENT(out)                                 :: Nindices
+    INTEGER, DIMENSION(4,d1234), INTENT(out)             :: indices
+
+! Local
+    INTEGER                                              :: i,j,k,l
+    INTEGER                                              :: Ncounts1D, icount1D
+    INTEGER                                              :: Ncounts2D, icount2D
+    INTEGER                                              :: Ncounts3D, icount3D
+    INTEGER, DIMENSION(d2,d3,d4)                         :: diffmat3D
+    INTEGER, DIMENSION(d3,d4)                            :: diffmat2D
+    INTEGER, DIMENSION(d4)                               :: diffmat1D
+
+    !!!!!!! Variables
+    ! d1, d2, d3, d4: shape of the 4D matrix
+    ! mat: 4D matrix
+    ! value: value to be looking for
+    ! Nindices: number of times value found within matrix
+    ! indices: indices of the found values
+
+    fname = 'multi_index_mat4DI'
+
+    Nindices = 0
+    indices = 0
+    DO i=1, d1
+      diffmat3D = mat(i,:,:,:) - value
+      IF (ANY(diffmat3D == 0)) THEN
+        Ncounts3D = COUNT(diffmat3D == 0)
+        icount3D = 0
+        DO j=1, d2
+          diffmat2D = mat(i,j,:,:) - value
+          IF (ANY(diffmat2D == 0)) THEN
+            Ncounts2D = COUNT(diffmat2D == 0)
+            icount2D = 0
+            DO k=1, d3
+              diffmat1D = mat(i,j,k,:) - value
+              IF (ANY(diffmat1D == 0)) THEN
+                Ncounts1D = COUNT(diffmat1D == 0)
+                icount1D = 0
+                DO l=1, d4
+                  IF (diffmat1D(l) == 0) THEN
+                    Nindices = Nindices + 1
+                    indices(1,Nindices) = i
+                    indices(2,Nindices) = j
+                    indices(3,Nindices) = k
+                    indices(4,Nindices) = l
+                    icount1D = icount1D + 1
+                    IF (icount1D == Ncounts1D) EXIT
+                  END IF
+                END DO
+              icount2D = icount2D + icount1D
+              IF (icount2D == Ncounts2D) EXIT
+              END IF
+            END DO
+            icount3D = icount3D + icount1D
+            IF (icount3D == Ncounts3D) EXIT
+          END IF
+        END DO
+      END IF
+    END DO
+
+  END SUBROUTINE multi_index_mat4DI
 
   SUBROUTINE multi_index_mat2DRK(d1, d2, d12, mat, value, Nindices, indices)