Changeset 3760 for LMDZ6/branches/Optimisation_LMDZ/libf/phylmd/cv3a_compress.f90

Timestamp:

Jul 15, 2020, 10:14:34 PM (4 years ago)

Author:

adurocher

Message:

Use blocks to allocate compressed arrays to the right size

File:

• LMDZ6/branches/Optimisation_LMDZ/libf/phylmd/cv3a_compress.f90 (modified) (2 diffs)

LMDZ6/branches/Optimisation_LMDZ/libf/phylmd/cv3a_compress.f90

@@ -34 +34 @@
   
   public :: array_list, add_array_i1, add_array_r1, add_array_r2, add_array_r3, &
-            cv3a_compress, cv3a_uncompress, compress_data_t
+            get_compress_size, cv3a_compress, cv3a_uncompress, compress_data_t
   
   contains
@@ -93 +93 @@
     arrays%arrays_r(1,c)%buf => array_in
     arrays%arrays_r(2,c)%buf => array_out
-  end subroutine 
+  end subroutine
+
+  function get_compress_size(len, mask) result(ncum)
+    use mod_phys_lmdz_omp_data, only : omp_rank
+    integer, INTENT (IN) :: len ! lenght of the contiguous dimension of arrays to compress
+    logical, INTENT (IN) :: mask(len) ! Mask of convective cells
+    integer :: ncum ! Number of cells in compressed arrays (first dimension)
+    
+    select case (compress_mode)
+      case (COMPRESS_MODE_COMPRESS)
+          ncum = count(mask)
+      case (COMPRESS_MODE_COPY)
+          ncum = len
+      case default
+        call abort_physic("get_compress_size", "Unknown compress mode", 1)
+    end select
+  end function
     
   ! Compress arrays from 'arrays' according to 'mask'.