diff options
author | Aaron LI <aly@aaronly.me> | 2017-11-22 14:19:43 +0800 |
---|---|---|
committer | Aaron LI <aly@aaronly.me> | 2017-11-22 14:19:43 +0800 |
commit | b0a0dd6b6157de8bfcbce7d23c6519579010f515 (patch) | |
tree | a3f73383d3eead64ac790bbbc9f62a400dc8c0fb /astro | |
parent | 91fa275d07f811b463efc55847377634d706705a (diff) | |
download | atoolbox-b0a0dd6b6157de8bfcbce7d23c6519579010f515.tar.bz2 |
astro/calc_psd.py: Add argument --center
Diffstat (limited to 'astro')
-rwxr-xr-x | astro/calc_psd.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/astro/calc_psd.py b/astro/calc_psd.py index f8f9f6b..6f0c2f6 100755 --- a/astro/calc_psd.py +++ b/astro/calc_psd.py @@ -57,6 +57,8 @@ class PSD: def __init__(self, image, pixel=(1.0, "pixel"), step=1.1, meanstd=False, bunit=None): self.image = np.array(image, dtype=float) + if image.ndim != 2: + raise ValueError("input image is not 2D!") self.shape = self.image.shape if self.shape[0] != self.shape[1]: raise ValueError("input image is not square!") @@ -338,6 +340,9 @@ def main(): description="Calculate radial power spectral density") parser.add_argument("-C", "--clobber", dest="clobber", action="store_true", help="overwrite the output files if already exist") + parser.add_argument("-c", "--center", dest="center", type=int, + help="crop the central box region of specified " + + "size before calculating the power spectrum") parser.add_argument("-s", "--step", dest="step", type=float, default=1.1, help="step ratio (>1; default: 1.1) between 2 " + "consecutive radial frequency points, " + @@ -393,6 +398,16 @@ def main(): else: raise ValueError("image has different unit: %s" % bunit2) + if args.center: + csize = args.center + if csize >= min(image.shape): + raise ValueError("--center %d exceeds image size" % csize) + rows, cols = image.shape + rc, cc = rows//2, cols//2 + cs1, cs2 = csize//2, (csize+1)//2 + image = image[(rc-cs1):(rc+cs2), (cc-cs1):(cc+cs2)] + print("Cropped image with central box of size: %dx%d" % image.shape) + psd = PSD(image=image, pixel=pixel, step=args.step, meanstd=args.meanstd, bunit=bunit) psd.calc_psd() |