aboutsummaryrefslogtreecommitdiffstats
path: root/astro/fits/fitsimgsub.py
diff options
context:
space:
mode:
authorAaron LI <aly@aaronly.me>2017-12-05 22:55:43 +0800
committerAaron LI <aly@aaronly.me>2017-12-05 22:55:43 +0800
commitbe6297a655249cf963b7a50e65ed0881833de806 (patch)
tree5872df5bbba2b5d8d5062a0cbff251b5fd0998ba /astro/fits/fitsimgsub.py
parent09805bcdb9a948d434893e24fcbe515dc4ef5d2a (diff)
downloadatoolbox-be6297a655249cf963b7a50e65ed0881833de806.tar.bz2
Add astro/fitsimage.py: FITS image manipulation tool
Replace fitsimgadd.py and fitsimgsub.py
Diffstat (limited to 'astro/fits/fitsimgsub.py')
-rwxr-xr-xastro/fits/fitsimgsub.py55
1 files changed, 0 insertions, 55 deletions
diff --git a/astro/fits/fitsimgsub.py b/astro/fits/fitsimgsub.py
deleted file mode 100755
index cae0190..0000000
--- a/astro/fits/fitsimgsub.py
+++ /dev/null
@@ -1,55 +0,0 @@
-#!/usr/bin/env python3
-#
-# Copyright (c) Weitian LI <weitian@aaronly.me>
-# MIT license
-#
-
-"""
-Subtract a FITS image by one or more FITS images of same shape.
-"""
-
-import os
-import sys
-import argparse
-
-from astropy.io import fits
-
-
-def main():
- parser = argparse.ArgumentParser(
- description="Subtract a FITS image by >=1 images of same shape")
- parser.add_argument("-C", "--clobber", action="store_true",
- help="overwrite existing files")
- parser.add_argument("-1", "--infile1", required=True,
- help="the FITS image from which to be subtracted")
- parser.add_argument("-2", "--infile2", nargs="+",
- help="one or more FITS images to be subtracted by")
- parser.add_argument("-o", "--outfile", required=True,
- help="filename of subtracted FITS image")
- args = parser.parse_args()
-
- if os.path.exists(args.outfile):
- if args.clobber:
- os.remove(args.outfile)
- print("WARNING: removed existing output file: %s" % args.outfile)
- else:
- raise OSError("output file already exists: %s" % args.outfile)
-
- with fits.open(args.infile1) as f:
- image = f[0].data
- header = f[0].header
- print("Opened FITS image: %s" % args.infile1)
- print("Image shape: %s" % str(list(reversed(image.shape))))
-
- for fn in args.infile2:
- print("Subtracting FITS image: %s ..." % fn)
- image -= fits.open(fn)[0].data
-
- header.add_history(" ".join(sys.argv))
- hdu = fits.PrimaryHDU(data=image, header=header)
- hdu.writeto(args.outfile)
- print("Wrote subtracted FITS image to: %s" % args.outfile)
-
-
-if __name__ == "__main__":
- main()