aboutsummaryrefslogtreecommitdiffstats
path: root/astro
diff options
context:
space:
mode:
authorAaron LI <aly@aaronly.me>2018-07-23 22:22:51 +0800
committerAaron LI <aly@aaronly.me>2018-07-23 22:22:51 +0800
commit4e2441d44eff1b693acaad1564d1233a8cb2ef66 (patch)
treedc28dee8c2d2765a692003c58b51aaf2b611fdeb /astro
parent00dcca590cfe574651bf4fd8724fc93e08bfcff6 (diff)
downloadatoolbox-4e2441d44eff1b693acaad1564d1233a8cb2ef66.tar.bz2
astro/fitsimage.py: Add 'div' (divide) sub-command
Diffstat (limited to 'astro')
-rwxr-xr-xastro/fits/fitsimage.py44
1 files changed, 44 insertions, 0 deletions
diff --git a/astro/fits/fitsimage.py b/astro/fits/fitsimage.py
index 5f867d3..f49280a 100755
--- a/astro/fits/fitsimage.py
+++ b/astro/fits/fitsimage.py
@@ -261,6 +261,30 @@ def cmd_mul(args):
print("Saved FITS image to: %s" % args.outfile)
+def cmd_div(args):
+ """
+ Sub-command: "div", divide the image by a number or other image(s)
+ """
+ fimage = FITSImage(args.infile)
+ image = fimage.image
+ if args.number:
+ print("Divide by number: %g" % args.number)
+ image /= args.number
+ else:
+ for fn in args.files:
+ print("Divide by another image from: %s" % fn)
+ fimage2 = FITSImage(fn)
+ with np.errstate(divide="warn"):
+ image /= fimage2.image
+
+ if args.fill_value:
+ print("Filling invalid data with: %s" % args.fill_value)
+ image[~np.isfinite(image)] = float(args.fill_value)
+ fimage.image = image
+ fimage.write(args.outfile, clobber=args.clobber)
+ print("Saved FITS image to: %s" % args.outfile)
+
+
def cmd_zoom(args):
"""
Sub-command: "zoom", zoom the image to a new size with FoV coverage
@@ -356,6 +380,26 @@ def main():
help="FITS image(s) to be multiplied by")
parser_mul.set_defaults(func=cmd_mul)
+ # sub-command: "div"
+ parser_div = subparsers.add_parser(
+ "div", aliases=["divide"],
+ help="divide the image by a number or other image(s)")
+ parser_div.add_argument("-C", "--clobber", dest="clobber",
+ action="store_true",
+ help="overwrite existing output file")
+ parser_div.add_argument("-F", "--fill-value", dest="fill_value",
+ help="value to fill the invalid elements")
+ parser_div.add_argument("-i", "--infile", dest="infile", required=True,
+ help="input FITS image")
+ parser_div.add_argument("-o", "--outfile", dest="outfile", required=True,
+ help="output FITS image")
+ exgrp_div = parser_div.add_mutually_exclusive_group(required=True)
+ exgrp_div.add_argument("-n", "--number", dest="number", type=float,
+ help="number to be divided by")
+ exgrp_div.add_argument("-f", "--files", dest="files", nargs="+",
+ help="FITS image(s) to be divided by")
+ parser_div.set_defaults(func=cmd_div)
+
# sub-command: "zoom"
parser_zoom = subparsers.add_parser(
"zoom", aliases=["rescale"],