Update to isd_generate.py to allow for a NAIF radius override

ale/isd_generate.py

@@ -16,7 +16,7 @@
 import os
 from pathlib import Path
 import sys
-
+import json
 import ale
 
 logger = logging.getLogger(__name__)
@@ -51,6 +51,34 @@ def main():
              "and the default strategy of replacing their final suffix with "
              ".json will be used to generate the output file paths."
     )
+    parser.add_argument(
+        "--semimajor",
+        required='--semiminor' in sys.argv,
+        type=float,
+        default=None,
+        help="Optional spherical radius (km) override.  Setting "
+             " '--semimajor 3396.19' "
+             "will override both semi-major and semi-minor radius values with the same value.  "
+             "An ellipse can be defined if '--semiminor' is also sent.  "
+             "If not specified, the default radius "
+             "values (e.g.; from NAIF kernels or the ISIS Cube) will be used.  "
+             "When is needed? Beyond a specialized need, it is common "
+             "that planetary bodies are defined as a triaxial body.  "
+             "In most of these cases, the IAU WGCCRE report recommends the use of a "
+             "best-fit sphere for a derived map product.  "
+             "For current IAU spherical recommendations see: "
+             "https://doi.org/10.1007/s10569-017-9805-5 or "
+             "http://voparis-vespa-crs.obspm.fr:8080/web/  "
+             "Make sure radius values are in kilometers."
+    )
+    parser.add_argument(
+        "--semiminor",
+        type=float,
+        default=None,
+        help="Optional semi-minor radius (km) override. When using this parameter, you must also define the semi-major radius. Setting "
+             " '--semimajor 3396.19 --semiminor 3376.2' "
+             "will override the semi-major and semi-minor radii to define an ellipse.  "
+    )
     parser.add_argument(
         "-v", "--verbose",
         action="store_true",
@@ -84,9 +112,17 @@ def main():
         except KeyError:
             k = [args.kernel, ]
 
+    if args.semimajor is None:
+        radii = None
+    else:
+        if args.semiminor is None:  # set a sphere
+          radii = [args.semimajor, args.semimajor]
+        else:                       # set as ellipse
+          radii = [args.semimajor, args.semiminor]
+
     if len(args.input) == 1:
         try:
-            file_to_isd(args.input[0], args.out, kernels=k, log_level=log_level)
+            file_to_isd(args.input[0], args.out, radii, kernels=k, log_level=log_level)
         except Exception as err:
             # Seriously, this just throws a generic Exception?
             sys.exit(f"File {args.input[0]}: {err}")
@@ -96,7 +132,7 @@ def main():
         ) as executor:
             futures = {
                 executor.submit(
-                    file_to_isd, f, **{"kernels": k, "log_level": log_level}
+                    file_to_isd, f, **{"radii": radii, "kernels": k, "log_level": log_level}
                 ): f for f in args.input
             }
             for f in concurrent.futures.as_completed(futures):
@@ -113,6 +149,7 @@ def main():
 def file_to_isd(
     file: os.PathLike,
     out: os.PathLike = None,
+    radii: list = None,
     kernels: list = None,
     log_level=logging.WARNING
 ):
@@ -143,6 +180,14 @@ def file_to_isd(
     else:
         usgscsm_str = ale.loads(file)
 
+    if radii is not None:
+        usgscsm_json = json.loads(usgscsm_str)
+        usgscsm_json["radii"]["semimajor"] = radii[0]
+        usgscsm_json["radii"]["semiminor"] = radii[1]
+        logger.info(f"Overriding radius to:")
+        logger.info(usgscsm_json["radii"])
+        usgscsm_str = json.dumps(usgscsm_json, indent=2)
+
     logger.info(f"Writing: {isd_file}")
     isd_file.write_text(usgscsm_str)