snapshot...

modules/util/util.py

@@ -1411,4 +1411,41 @@ def resample_2d_linear(x, y, z, x_new, y_new):
 
 def resample_2d_linear_one(x, y, z, x_new, y_new):
     f = interp2d(x, y, z)
-    return f(x_new, y_new)
+    return f(x_new, y_new)
\ No newline at end of file
+
+
+def get_training_parameters(day_night='DAY', l1b_andor_l2='both', satellite='GOES16', use_dnb=False):
+    if day_night == 'DAY':
+        train_params_l2 = ['cld_height_acha', 'cld_geo_thick', 'cld_temp_acha', 'cld_press_acha', 'supercooled_cloud_fraction',
+                           'cld_emiss_acha', 'conv_cloud_fraction', 'cld_reff_dcomp', 'cld_opd_dcomp', 'iwc_dcomp', 'lwc_dcomp']
+
+        if satellite == 'GOES16':
+            train_params_l1b = ['temp_10_4um_nom', 'temp_11_0um_nom', 'temp_12_0um_nom', 'temp_13_3um_nom', 'temp_3_75um_nom',
+                                'temp_6_2um_nom', 'temp_6_7um_nom', 'temp_7_3um_nom', 'temp_9_7um_nom',
+                                'refl_0_47um_nom', 'refl_0_65um_nom', 'refl_0_86um_nom', 'refl_1_38um_nom', 'refl_1_60um_nom']
+                                # 'refl_2_10um_nom']
+        elif satellite == 'H08':
+            train_params_l1b = ['temp_10_4um_nom', 'temp_12_0um_nom', 'temp_8_5um_nom', 'temp_3_75um_nom', 'refl_2_10um_nom',
+                                'refl_1_60um_nom', 'refl_0_86um_nom', 'refl_0_47um_nom']
+    else:
+        train_params_l2 = ['cld_height_acha', 'cld_geo_thick', 'cld_temp_acha', 'cld_press_acha', 'supercooled_cloud_fraction',
+                           'cld_emiss_acha', 'conv_cloud_fraction', 'cld_reff_acha', 'cld_opd_acha']
+
+        if use_dnb is True:
+            train_params_l2 = ['cld_height_acha', 'cld_geo_thick', 'cld_temp_acha', 'cld_press_acha', 'supercooled_cloud_fraction',
+                               'cld_emiss_acha', 'conv_cloud_fraction', 'cld_reff_dcomp', 'cld_opd_dcomp', 'iwc_dcomp', 'lwc_dcomp']
+
+        if satellite == 'GOES16':
+            train_params_l1b = ['temp_10_4um_nom', 'temp_11_0um_nom', 'temp_12_0um_nom', 'temp_13_3um_nom', 'temp_3_75um_nom',
+                                'temp_6_2um_nom', 'temp_6_7um_nom', 'temp_7_3um_nom', 'temp_9_7um_nom']
+        elif satellite == 'H08':
+            train_params_l1b = ['temp_10_4um_nom', 'temp_12_0um_nom', 'temp_8_5um_nom', 'temp_3_75um_nom']
+
+    if l1b_andor_l2 == 'both':
+        train_params = train_params_l1b + train_params_l2
+    elif l1b_andor_l2 == 'l1b':
+        train_params = train_params_l1b
+    elif l1b_andor_l2 == 'l2':
+        train_params = train_params_l2
+
+    return train_params, train_params_l1b, train_params_l2