aboutsummaryrefslogtreecommitdiffstats
path: root/astro
diff options
context:
space:
mode:
Diffstat (limited to 'astro')
-rwxr-xr-xastro/radec.py52
1 files changed, 42 insertions, 10 deletions
diff --git a/astro/radec.py b/astro/radec.py
index 4cc2b71..546223c 100755
--- a/astro/radec.py
+++ b/astro/radec.py
@@ -1,6 +1,6 @@
#!/usr/bin/env python3
#
-# Copyright (c) 2017 Aaron LI <aly@aaronly.me>
+# Copyright (c) 2017,2019 Aaron LI <aly@aaronly.me>
# MIT License
#
@@ -8,6 +8,7 @@
Convert among various (R.A., Dec.) coordinate formats.
"""
+import re
import argparse
from astropy import units as au
@@ -15,22 +16,53 @@ from astropy.coordinates import Angle
def parse_coord(c):
- if len(c) == 6:
- # h m s d m s
- ra = Angle((float(c[0]), float(c[1]), float(c[2])), unit=au.hourangle)
- dec = Angle((float(c[3]), float(c[4]), float(c[5])), unit=au.deg)
- elif len(c) == 2:
- ra = Angle(float(c[0]), unit=au.deg)
- dec = Angle(float(c[1]), unit=au.deg)
+ ra_is_deg = True
+ if len(c) == 2:
+ ra, dec = c
+ # ?h[our] ?[deg]
+ if ra.endswith('h') or ra.endswith('hour'):
+ ra = ra.rstrip('hour')
+ ra_is_deg = False
+
+ # ?d[eg] ?[deg]
+ # ?[deg] ?[deg]
+ ra = ra.rstrip('deg')
+ dec = dec.rstrip('deg')
+
+ # ?h?m?s ?d?m?s
+ # ?:?:? ?:?:?
+ ra = re.split(r'[hms:]', ra.rstrip('s'))
+ dec = re.split(r'[dms:]', dec.rstrip('s'))
+ elif len(c) == 6:
+ # h m s d m s
+ ra, dec = c[:3], c[-3:]
+ else:
+ raise ValueError('invalid coordinate: {0}'.format(c))
+
+ if len(ra) == 1:
+ ra = float(ra[0])
+ else:
+ ra = (float(ra[0]), float(ra[1]), float(ra[2]))
+ if len(dec) == 1:
+ dec = float(dec[0])
+ else:
+ dec = (float(dec[0]), float(dec[1]), float(dec[2]))
+
+ if ra_is_deg:
+ ra = Angle(ra, unit=au.deg)
else:
- raise ValueError("invalid coordinate: {0}".format(c))
+ ra = Angle(ra, unit=au.hourangle)
+ dec = Angle(dec, unit=au.deg)
return (ra, dec)
def main():
parser = argparse.ArgumentParser(
description="Convert among multiple coordinate formats")
- parser.add_argument("coord", nargs="+")
+ parser.add_argument(
+ "coord", nargs="+",
+ help="syntax: (1) deg deg; (2) hour deg; (3) h m s d m s; " +
+ "(4) ?h?m?s ?d?m?s; (5) ?:?:? ?:?:?")
args = parser.parse_args()
ra, dec = parse_coord(args.coord)