aboutsummaryrefslogtreecommitdiffstats
path: root/astro
diff options
context:
space:
mode:
authorAaron LI <aly@aaronly.me>2018-11-16 00:29:11 +0800
committerAaron LI <aly@aaronly.me>2018-11-16 00:39:48 +0800
commitb372f84351579326ba71c95c309a90e7c6f47ce5 (patch)
tree478d66baa589d2aff57adee12af3ab26ec748206 /astro
parent4c377ad4fbe98f85fcaedfc332da9c44fcecbc23 (diff)
downloadatoolbox-b372f84351579326ba71c95c309a90e7c6f47ce5.tar.bz2
astro/fitsimage.py: Add "shift" sub-command
Diffstat (limited to 'astro')
-rwxr-xr-xastro/fitsimage.py38
1 files changed, 38 insertions, 0 deletions
diff --git a/astro/fitsimage.py b/astro/fitsimage.py
index af49f7e..620373b 100755
--- a/astro/fitsimage.py
+++ b/astro/fitsimage.py
@@ -170,6 +170,17 @@ class FITSImage:
raise ValueError("invalid rotate to: %s" % to)
return self.image
+ def shift(self, x, y=None):
+ y = y or x
+ ny, nx = self.image.shape
+ image = np.zeros((ny, nx))
+ image[y:, x:] = self.image[:ny-y, :nx-x]
+ image[:y, :x] = self.image[ny-y:, nx-x:]
+ image[:y, x:] = self.image[ny-y:, :nx-x]
+ image[y:, :x] = self.image[:ny-y, nx-x:]
+ self.image = image
+ return self.image
+
def write(self, outfile, clobber=False):
self.header.add_history(" ".join(sys.argv))
hdu = fits.PrimaryHDU(data=self.data, header=self.header)
@@ -364,6 +375,17 @@ def cmd_rotate(args):
print("Saved rotated FITS image to: %s" % args.outfile)
+def cmd_shift(args):
+ """
+ Sub-command: "shift", shift the image and padding accordingly.
+ """
+ fimage = FITSImage(args.infile)
+ print("Shift image by (%d,%d) ..." % (args.x, args.y or args.x))
+ fimage.shift(x=args.x, y=args.y)
+ fimage.write(args.outfile, clobber=args.clobber)
+ print("Saved shifted FITS image to: %s" % args.outfile)
+
+
def main():
parser = argparse.ArgumentParser(
description="FITS image manipulation tool")
@@ -514,6 +536,22 @@ def main():
help="rotate 180 degree")
parser_rot.set_defaults(func=cmd_rotate)
+ # sub-command: "shift"
+ parser_sft = subparsers.add_parser(
+ "sft", aliases=["shift"],
+ help="shift the image and pad accordingly")
+ parser_sft.add_argument("-C", "--clobber", action="store_true",
+ help="overwrite existing output file")
+ parser_sft.add_argument("-i", "--infile", required=True,
+ help="input FITS image")
+ parser_sft.add_argument("-o", "--outfile", required=True,
+ help="output shifted and padded FITS image")
+ parser_sft.add_argument("-x", type=int, required=True,
+ help="numer of horizontal pixels")
+ parser_sft.add_argument("-y", type=int,
+ help="numer of vertical pixels")
+ parser_sft.set_defaults(func=cmd_shift)
+
args = parser.parse_args()
args.func(args)