Add module ds9.py for easier image viewing

1 files changed, 27 insertions, 0 deletions

diff --git a/scripts/ds9.py b/scripts/ds9.py
new file mode 100644
index 0000000..af25a42
--- /dev/null
+++ b/scripts/ds9.py
@@ -0,0 +1,27 @@
+# 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,
+        "-regions", "format", regformat,
+        "-regions", "system", regsystem,
+        "-regions", regfile,
+        "-cmap", cmap,
+        "-bin", "factor", binfactor,
+        "-scale", scale,
+    ]
+    if smooth:
+        cmd += ["-smooth", "yes", "-smooth", "radius", smooth]
+    subprocess.check_call(cmd)