snapshot...

modules/util/viirs.py

@@ -3,20 +3,21 @@ import h5py
 from util.util import get_grid_values, get_grid_values_all
 import glob
 from pathlib import Path
+import os
 
 mod_res_params = ['M07', 'M08', 'M10', 'M12', 'M13', 'M14', 'M15', 'M16']
 img_res_params = ['M07_highres', 'M08_highres', 'M10_highres', 'M12_highres', 'M13_highres', 'M14_highres', 'M15_highres', 'M16_highres']
 
 
 def run_all(directory):
-    mod_files = glob.glob(directory+'*.uwssec.nc')
-    img_files = glob.glob(directory+'*.uwssec.highres.nc')
+    mod_files = Path(directory).rglob('*.uwssec.nc')
 
     mod_tiles = []
     img_tiles = []
 
     for idx, mfile in enumerate(mod_files):
-        ifile = img_files[idx]
+        w_o_ext, ext = os.path.splitext(mfile)
+        ifile = w_o_ext+'uwssec'+ext
         run(mfile, ifile, mod_tiles, img_tiles)
 
     mod_nda = np.stack(mod_tiles)
@@ -55,17 +56,34 @@ def run(mod_res_filename, img_res_filename, mod_tiles, img_tiles):
     mod_data = get_grid_values(mod_h5f, mod_param, 0, 0, None, mod_num_lines, mod_num_pixels, range_name=None)
     img_data = get_grid_values(img_h5f, img_param, 0, 0, None, img_num_lines, img_num_pixels, range_name=None)
 
-    for j in range(mod_num_y_tiles):
-        j_m = j * mod_tile_width
+    num_cntr_tiles = 2
+    i_c = int(mod_num_pixels / num_cntr_tiles)  # center
+    j_skip = int(mod_num_y_tiles / num_cntr_tiles) * mod_tile_width
+    for k in range(num_cntr_tiles):
+        j_c = k * j_skip
+        j_m = j_c
         j_i = j_m * 2
-        for i in range(mod_num_x_tiles):
-            i_m = i * mod_tile_width
-            i_i = i_m * 2
-
-            nda = mod_data[j_m:j_m+mod_tile_width, i_m:i_m+mod_tile_width]
-            mod_tiles.append(nda)
-            nda = img_data[j_i:j_i+img_tile_width, i_i:i_i+img_tile_width]
-            img_tiles.append(nda)
+
+        i_m = i_c
+        i_i = i_m * 2
+
+        nda = mod_data[j_m:j_m + mod_tile_width, i_m:i_m + mod_tile_width]
+        mod_tiles.append(nda)
+
+        nda = img_data[j_i:j_i + img_tile_width, i_i:i_i + img_tile_width]
+        img_tiles.append(nda)
+
+    # for j in range(mod_num_y_tiles):
+    #     j_m = j * mod_tile_width
+    #     j_i = j_m * 2
+    #     for i in range(mod_num_x_tiles):
+    #         i_m = i * mod_tile_width
+    #         i_i = i_m * 2
+    #
+    #         nda = mod_data[j_m:j_m+mod_tile_width, i_m:i_m+mod_tile_width]
+    #         mod_tiles.append(nda)
+    #         nda = img_data[j_i:j_i+img_tile_width, i_i:i_i+img_tile_width]
+    #         img_tiles.append(nda)
 
     mod_h5f.close()
     img_h5f.close()