updated extract_h5_dataset utility to use utils

Author: jpswinski

utils/extract_h5_dataset.py

@@ -3,9 +3,8 @@
 #
 
 import sys
-import logging
 from sliderule import icesat2
-from utils import parse_command_line
+from utils import parse_command_line, initialize_client
 
 ###############################################################################
 # MAIN
@@ -14,33 +13,26 @@
 if __name__ == '__main__':
 
     # set script defaults
-    cfg = {
-        "url":          'localhost',
-        "organization": None,
-        "asset":        'atlas-local',
+    local = {
         "dataset":      '/gt2l/heights/h_ph',
-        "resource":     'ATL03_20181017222812_02950102_005_01.h5',
         "col":          0,
         "startrow":     0,
         "numrows":      -1
     }
 
-    # parse configuration parameters
-    parse_command_line(sys.argv, cfg)
-
-    # Configure Logging #
-    logging.basicConfig(level=logging.INFO)
+    # Initialize Client
+    parms, cfg = initialize_client(sys.argv)
 
-    # Configure SlideRule #
-    icesat2.init(cfg["url"], True, organization=cfg["organization"])
+    # parse configuration parameters
+    parse_command_line(sys.argv, local)
 
     # Read Dataset #
-    datasets = [ {"dataset": cfg["dataset"], "col": cfg["col"], "startrow": cfg["startrow"], "numrows": cfg["numrows"]} ]
+    datasets = [ {"dataset": local["dataset"], "col": local["col"], "startrow": local["startrow"], "numrows": local["numrows"]} ]
     rawdata = icesat2.h5p(datasets, cfg["resource"], cfg["asset"])
     print(rawdata)
 
     # Write Dataset to File #
-    filename = cfg["dataset"][cfg["dataset"].rfind("/")+1:]
+    filename = local["dataset"][local["dataset"].rfind("/")+1:]
     f = open(filename + ".bin", 'w+b')
-    f.write(bytearray(rawdata[cfg["dataset"]]))
+    f.write(bytearray(rawdata[local["dataset"]]))
     f.close()

utils/utils.py

@@ -78,7 +78,6 @@ def initialize_client(args):
         "maxi":                     1,
         "atl03_geo_fields":         [],
         "atl03_ph_fields":          [],
-        "samples":                  [],
         "profile":                  True,
         "verbose":                  True,
         "timeout":                  0,
@@ -120,10 +119,6 @@ def initialize_client(args):
     if len(cfg['atl03_ph_fields']) > 0:
         parms['atl03_ph_fields'] = cfg['atl03_ph_fields']
 
-    # Add Rasters to Sample
-    if len(cfg['samples']) > 0:
-        parms['samples'] = cfg['samples']
-
     # Add ATL08 Classification
     if len(cfg['atl08_class']) > 0:
         parms['atl08_class'] = cfg['atl08_class']