Updated pickle with input output arguments

olaflow/processing/pickle.py

@@ -1,6 +1,5 @@
 import argparse
 import gzip
-import configparser
 import logging
 import pathlib
 import pickle
@@ -14,20 +13,29 @@ from .olaflow import OFModel
 
 parser = argparse.ArgumentParser(description="Post-process olaflow results")
 parser.add_argument("-v", "--verbose", action="count", default=0)
-parser.add_argument("-c", "--config", default="config.ini")
-parser.add_argument("-o", "--output", type=pathlib.Path)
+parser.add_argument(
+    "-i", "--input", type=pathlib.Path, help="Olaflow output directory", required=True
+)
+parser.add_argument(
+    "-o",
+    "--output",
+    type=pathlib.Path,
+    help="Output directory for pickled data",
+    required=True,
+)
+parser.add_argument(
+    "-z", "--compress", action="store_true", help="Enable gzip compression"
+)
 args = parser.parse_args()
 
 logging.basicConfig(level=max((10, 20 - 10 * args.verbose)))
 log = logging.getLogger("ola_post")
 
 log.info("Starting sws -> olaFlow converter")
-config = configparser.ConfigParser()
-config.read(args.config)
-out = pathlib.Path(config.get("post", "out"))
+out = args.output
 out.mkdir(parents=True, exist_ok=True)
 
-olaflow_root = args.output
+olaflow_root = args.input
 model = OFModel(olaflow_root)
 model.read_mesh()
 model.read_time()
@@ -40,5 +48,9 @@ model.read_field_all("U")
 
 model.read_post("graphUniform", "alpha.water")
 
-with gzip.open(out.joinpath("pickle.gz"), "wb") as f:
+with (
+    gzip.open(out.joinpath("pickle.gz"), "wb")
+    if args.compress
+    else out.joinpath("pickle").open("wb")
+) as f:
     pickle.dump(model, f)