Merge pull request #594 from duytnguyendtn/radquantity

Support Astropy Quantity as radius arg for Registry SkyCoord Spatial constraint

Author: bsipocz

CHANGES.rst

@@ -46,6 +46,8 @@ Enhancements and Fixes
 - Tables returned by RegistryResource.get_tables() now have a utype
   attribute [#576]
 
+- Registry Spatial constraint now supports Astropy Quantities for the radius argument [#594]
+
 Deprecations and Removals
 -------------------------
 

docs/registry/index.rst

@@ -158,13 +158,30 @@ And to look for tap resources *in* a specific cone, you would do
 
   >>> from astropy.coordinates import SkyCoord
   >>> registry.search(registry.Freetext("Wolf-Rayet"),
-  ...                 registry.Spatial((SkyCoord("23d +3d"), 3), intersect="enclosed")) # doctest: +IGNORE_OUTPUT
-  <DALResultsTable length=1>
-             ivoid             ...
-                               ...
-             object            ...
-  ---------------------------- ...
-  ivo://cds.vizier/j/aj/166/68 ...
+  ...                 registry.Spatial((SkyCoord("23d +3d"), 3), intersect="enclosed"))
+  <DALResultsTable length=2>
+               ivoid              ...
+                                  ...
+               object             ...
+  ------------------------------- ...
+  ivo://cds.vizier/j/a+a/688/a104 ...
+     ivo://cds.vizier/j/aj/166/68 ...
+
+Astropy Quantities are also supported for the radius angle of a SkyCoord-defined circular region:
+
+.. doctest-remote-data::
+
+  >>> from astropy.coordinates import SkyCoord
+  >>> from astropy import units as u
+  >>> registry.search(registry.Freetext("Wolf-Rayet"),
+  ...                 registry.Spatial((SkyCoord("23d +3d"), 180*u.Unit('arcmin')), intersect="enclosed"))
+  <DALResultsTable length=2>
+               ivoid              ...
+                                  ...
+               object             ...
+  ------------------------------- ...
+  ivo://cds.vizier/j/a+a/688/a104 ...
+     ivo://cds.vizier/j/aj/166/68 ...
 
 Where ``intersect`` can take the following values:
   * 'covers' is the default and returns resources that cover the geometry provided,

pyvo/registry/rtcons.py

@@ -673,6 +673,10 @@ class Spatial(SubqueriedConstraint):
     are interpreted in degrees)::
 
         >>> resources = registry.Spatial((SkyCoord("23d +3d"), 3))
+
+    Or you can provide the radius angle as an Astropy Quantity:
+
+        >>> resources = registry.Spatial((SkyCoord("23d +3d"), 1*u.rad))
     """
     _keyword = "spatial"
     _subquery_table = "rr.stc_spatial"
@@ -689,7 +693,7 @@ def __init__(self, geom_spec, order=6, intersect="covers"):
             as a DALI point, a 3-sequence as a DALI circle, a 2n sequence
             as a DALI polygon.  Additionally, strings are interpreted
             as ASCII MOCs, SkyCoords as points, and a pair of a
-            SkyCoord and a float as a circle.  Other types (proper
+            SkyCoord and a float or Quantity as a circle.  Other types (proper
             geometries or MOCPy objects) might be supported in the
             future.
         order : int, optional
@@ -719,9 +723,16 @@ def tomoc(s):
 
         elif len(geom_spec) == 2:
             if isinstance(geom_spec[0], SkyCoord):
+                # If radius given is astropy quantity, then convert to degrees
+                if isinstance(geom_spec[1], u.Quantity):
+                    if geom_spec[1].unit.physical_type != 'angle':
+                        raise ValueError("Radius quantity is not of type angle.")
+                    radius = geom_spec[1].to(u.deg).value
+                else:
+                    radius = geom_spec[1]
                 geom = tomoc(format_function_call("CIRCLE",
                                                   [geom_spec[0].ra.value, geom_spec[0].dec.value,
-                                                   geom_spec[1]]))
+                                                   radius]))
             else:
                 geom = tomoc(format_function_call("POINT", geom_spec))
 

pyvo/registry/tests/test_rtcons.py

@@ -303,6 +303,15 @@ def test_SkyCoord_Circle(self):
         assert cons.get_search_condition(FAKE_GAVO) == _make_subquery(
             "rr.stc_spatial", "1 = CONTAINS(MOC(6, CIRCLE(3.0, -30.0, 3)), coverage)")
 
+    def test_SkyCoord_Circle_RadiusQuantity(self):
+        for radius in [3*u.deg, 180*u.Unit('arcmin'), 10800*u.Unit('arcsec')]:
+            cons = registry.Spatial((SkyCoord(3 * u.deg, -30 * u.deg), radius))
+            assert cons.get_search_condition(FAKE_GAVO) == _make_subquery(
+                "rr.stc_spatial", "1 = CONTAINS(MOC(6, CIRCLE(3.0, -30.0, 3.0)), coverage)")
+
+        with pytest.raises(ValueError, match="is not of type angle."):
+            cons = registry.Spatial((SkyCoord(3 * u.deg, -30 * u.deg), (1 * u.m)))
+
     def test_enclosed(self):
         cons = registry.Spatial("0/1-3", intersect="enclosed")
         assert cons.get_search_condition(FAKE_GAVO) == _make_subquery(