FEAT: Post processor improvements! Including:

- NetCDF/hdf5 python packages are optional. matplotlib is still mandatory though
- Look in pwd & pwd/UA/output (lets you postproc from run/)
- Add arg to set path to output files

Author: abukowski21

srcPython/postAether.py

@@ -8,8 +8,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import matplotlib.cm as cm
-from netCDF4 import Dataset
-from h5py import File
 import argparse
 import os
 import json
@@ -36,6 +34,9 @@ def parse_args():
     parser.add_argument('-oned', \
                         help='strip 1d files of ghostcells and store in one file', \
                         action="store_true")
+    parser.add_argument('-dir', default=None, type=str,
+                        help="Directory to find Aether files in. Will look in current"
+                        " directory & $PWD/UA/output/")
 
     args = parser.parse_args()
 
@@ -544,6 +545,13 @@ def get_base_files():
             IsFound, item = if_unique(ensembleFiles, fileInfo['ensembleFile'])
             if (IsFound):
                 filesInfo[i]['ensembleMembers'] = ensembleCounter[item]
+                
+    if len(filesInfo) == 0:
+        try:
+            os.chdir("UA/output")
+            get_base_files()
+        except:
+            print("No input files found!!")
 
     return filesInfo
 
@@ -948,18 +956,33 @@ def write_and_plot_data(dataToWrite,
 # main code
 #----------------------------------------------------------------------------
 
-if __name__ == '__main__':  # main code block
+def main(args):
 
-    args = parse_args()
     isVerbose = args.v
+    
+    if args.dir:
+        if isVerbose:
+            print("changing directory to: ", args.dir)
+        os.chdir(args.dir)
 
     filesInfo = get_base_files()
+    
+    if len(filesInfo) == 0:
+        return
 
     iVar = 3
     iAlt = args.alt
 
     output_netcdf = False if args.hdf5 else True
-
+    
+    if filesInfo[0]['isNetCDF']:
+        try:
+            from netCDF4 import Dataset
+            from h5py import File
+        except InputError:
+            raise InputError(
+              "Attempting to postprocess NetCDF files, but NetCDF is not installed for Python")    
+    
     for iFile, fileInfo in enumerate(filesInfo):
         coreFile = fileInfo['coreFile']
         isNetCDF = fileInfo['isNetCDF']
@@ -1004,4 +1027,12 @@ def write_and_plot_data(dataToWrite,
                     if (isVerbose):
                         print('    ', command)
                     os.system(command)
-    
+
+# call main:
+if __name__ == '__main__':  
+
+    args = parse_args()
+
+    # This allows code to cleanly exit on error
+    main(args)
+