Setup module 'acispy' and clean 'scripts' directory

Setup a module 'acispy' to hold some generic Python modules for
better/easier reuse, and clean up the 'scripts' directory, which will be
used to hold the CLI tools.

1 files changed, 30 insertions, 0 deletions

diff --git a/acispy/ds9.py b/acispy/ds9.py
new file mode 100644
index 0000000..e8c8e88
--- /dev/null
+++ b/acispy/ds9.py
@@ -0,0 +1,30 @@
+# Copyright (c) 2017 Weitian LI <liweitianux@live.com>
+# MIT license
+
+"""
+Wrapper function to view FITS files using DS9.
+"""
+
+import subprocess
+
+
+def ds9_view(filename, regfile=None, regformat="ciao", regsystem="physical",
+             cmap="he", binfactor=2, scale="linear", smooth=None):
+    """
+    Wrapper function to view FITS files using DS9.
+    """
+    cmd = [
+        "ds9", filename,
+        "-cmap", cmap,
+        "-bin", "factor", str(binfactor),
+        "-scale", scale,
+    ]
+    if regfile:
+        cmd += [
+            "-regions", "format", regformat,
+            "-regions", "system", regsystem,
+            "-regions", regfile,
+        ]
+    if smooth:
+        cmd += ["-smooth", "yes", "-smooth", "radius", str(smooth)]
+    subprocess.check_call(cmd)