From 971d1cbc30d5c23860d562b91945d6e3e3e935f6 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 11 May 2022 12:45:57 -0500 Subject: [PATCH] snapshot... --- modules/util/viirs.py | 44 ++++++++++++++++++++++++++++++------------- 1 file changed, 31 insertions(+), 13 deletions(-) diff --git a/modules/util/viirs.py b/modules/util/viirs.py index ba43e9f5..7d130db4 100644 --- a/modules/util/viirs.py +++ b/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() -- GitLab