From 25dcd411ef730e51e659a4cb1ca7beb55517398b Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 3 Nov 2021 14:39:11 -0500
Subject: [PATCH] get clavrx parameters from a saved pickle file

---
 modules/aeolus/datasource.py | 36 ++++++++++++++++++++----------------
 1 file changed, 20 insertions(+), 16 deletions(-)

diff --git a/modules/aeolus/datasource.py b/modules/aeolus/datasource.py
index 2c873379..8ef11bda 100644
--- a/modules/aeolus/datasource.py
+++ b/modules/aeolus/datasource.py
@@ -3,28 +3,32 @@ from datetime import timezone
 import glob
 import numpy as np
 import re
+import pickle
 from netCDF4 import Dataset
 from pathlib import Path
 
 from util.util import GenericException
 from util.lon_lat_grid import LonLatGrid
 from util.geos_nav import GEOSNavigation
-from util.util import homedir
-
-
-def get_parameters_clavrx(filename=homedir+'data/clavrx/clavrx_OR_ABI-L1b-RadF-M6C01_G16_s20192930000343.level2.nc'):
-    rg = Dataset(filename, 'r')
-    var_s = rg.variables
-    var_names = list(var_s.keys())
-    var_names_2d = []
-
-    for str in var_names:
-        v = var_s[str]
-        if len(v.shape) == 2:
-            if not (str.find('latitude') != -1 or str.find('longitude') != -1):
-                var_names_2d.append(str)
-
-    rg.close()
+from util.setup import ancillary_path
+
+
+# def get_parameters_clavrx(filename=homedir+'data/clavrx/clavrx_OR_ABI-L1b-RadF-M6C01_G16_s20192930000343.level2.nc'):
+def get_parameters_clavrx(filename=ancillary_path+'clavrx_parameters.pkl'):
+    # rg = Dataset(filename, 'r')
+    # var_s = rg.variables
+    # var_names = list(var_s.keys())
+    # var_names_2d = []
+    #
+    # for str in var_names:
+    #     v = var_s[str]
+    #     if len(v.shape) == 2:
+    #         if not (str.find('latitude') != -1 or str.find('longitude') != -1):
+    #             var_names_2d.append(str)
+    #
+    # rg.close()
+
+    var_names_2d = pickle.load(open(filename, 'rb'))
 
     return var_names_2d
 
-- 
GitLab