Source code for astroplan.target

# Licensed under a 3-clause BSD style license - see LICENSE.rst
from __future__ import (absolute_import, division, print_function,
                        unicode_literals)

# Standard library
from abc import ABCMeta

# Third-party
import astropy.units as u
from astropy.coordinates import SkyCoord, ICRS, UnitSphericalRepresentation

__all__ = ["Target", "FixedTarget", "NonFixedTarget"]

# Docstring code examples include printed SkyCoords, but the format changed
# in astropy 1.3. Thus the doctest needs astropy >=1.3 and this is the
# easiest way to make it work.

__doctest_requires__ = {'FixedTarget.*': ['astropy.modeling.Hermite1D']}


[docs]class Target(object): """ Abstract base class for target objects. This is an abstract base class -- you can't instantiate examples of this class, but must work with one of its subclasses such as `~astroplan.target.FixedTarget` or `~astroplan.target.NonFixedTarget`. """ __metaclass__ = ABCMeta def __init__(self, name=None, ra=None, dec=None, marker=None): """ Defines a single observation target. Parameters ---------- name : str, optional ra : WHAT TYPE IS ra ? dec : WHAT TYPE IS dec ? marker : str, optional User-defined markers to differentiate between different types of targets (e.g., guides, high-priority, etc.). """ raise NotImplementedError() @property def ra(self): """ Right ascension. """ if isinstance(self, FixedTarget): return self.coord.ra raise NotImplementedError() @property def dec(self): """ Declination. """ if isinstance(self, FixedTarget): return self.coord.dec raise NotImplementedError()
[docs]class FixedTarget(Target): """ Coordinates and metadata for an object that is "fixed" with respect to the celestial sphere. Examples -------- Create a `~astroplan.FixedTarget` object for Sirius: >>> from astroplan import FixedTarget >>> from astropy.coordinates import SkyCoord >>> import astropy.units as u >>> sirius_coord = SkyCoord(ra=101.28715533*u.deg, dec=16.71611586*u.deg) >>> sirius = FixedTarget(coord=sirius_coord, name="Sirius") Create an equivalent `~astroplan.FixedTarget` object for Sirius by querying for the coordinates of Sirius by name: >>> from astroplan import FixedTarget >>> sirius = FixedTarget.from_name("Sirius") """ def __init__(self, coord, name=None, **kwargs): """ Parameters ---------- coord : `~astropy.coordinates.SkyCoord` Coordinate of the target name : str (optional) Name of the target, used for plotting and representing the target as a string """ if not (hasattr(coord, 'transform_to') and hasattr(coord, 'represent_as')): raise TypeError('`coord` must be a coordinate object.') self.name = name self.coord = coord
[docs] @classmethod def from_name(cls, query_name, name=None, **kwargs): """ Initialize a `FixedTarget` by querying for a name from the CDS name resolver, using the machinery in `~astropy.coordinates.SkyCoord.from_name`. This Parameters ---------- query_name : str Name of the target used to query for coordinates. name : string or `None` Name of the target to use within astroplan. If `None`, query_name is used as ``name``. Examples -------- >>> from astroplan import FixedTarget >>> sirius = FixedTarget.from_name("Sirius") >>> sirius.coord # doctest: +FLOAT_CMP <SkyCoord (ICRS): (ra, dec) in deg ( 101.28715533, -16.71611586)> """ # Allow manual override for name keyword so that the target name can # be different from the query name, otherwise assume name=queryname. if name is None: name = query_name return cls(SkyCoord.from_name(query_name), name=name, **kwargs)
def __repr__(self): """ String representation of `~astroplan.FixedTarget`. Examples -------- Show string representation of a `~astroplan.FixedTarget` for Vega: >>> from astroplan import FixedTarget >>> from astropy.coordinates import SkyCoord >>> vega_coord = SkyCoord(ra='279.23473479d', dec='38.78368896d') >>> vega = FixedTarget(coord=vega_coord, name="Vega") >>> print(vega) # doctest: +FLOAT_CMP <FixedTarget "Vega" at SkyCoord (ICRS): (ra, dec) in deg ( 279.23473479, 38.78368894)> """ class_name = self.__class__.__name__ fmt_coord = repr(self.coord).replace('\n ', '')[1:-1] return '<{} "{}" at {}>'.format(class_name, self.name, fmt_coord) @classmethod def _from_name_mock(cls, query_name, name=None): """ Mock method to replace `FixedTarget.from_name` in tests without internet connection. """ # The lowercase method will be run on names, so enter keys in lowercase: stars = { "rigel": {"ra": 78.63446707*u.deg, "dec": -8.20163837*u.deg}, "sirius": {"ra": 101.28715533*u.deg, "dec": -16.71611586*u.deg}, "vega": {"ra": 279.23473479*u.deg, "dec": 38.78368896*u.deg}, "aldebaran": {"ra": 68.98016279*u.deg, "dec": 16.50930235*u.deg}, "polaris": {"ra": 37.95456067*u.deg, "dec": 89.26410897*u.deg}, "deneb": {"ra": 310.35797975*u.deg, "dec": 45.28033881*u.deg}, "m13": {"ra": 250.423475*u.deg, "dec": 36.4613194*u.deg}, "altair": {"ra": 297.6958273*u.deg, "dec": 8.8683212*u.deg}, "hd 209458": {"ra": 330.79*u.deg, "dec": 18.88*u.deg} } if query_name.lower() in stars: return cls(coord=SkyCoord(**stars[query_name.lower()]), name=query_name) else: raise ValueError("Target named {} not in mocked FixedTarget " "method".format(query_name))
[docs]class NonFixedTarget(Target): """ Placeholder for future function. """
def get_skycoord(targets): """ Return an `~astropy.coordinates.SkyCoord` object. When performing calculations it is usually most efficient to have a single `~astropy.coordinates.SkyCoord` object, rather than a list of `FixedTarget` or `~astropy.coordinates.SkyCoord` objects. This is a convenience routine to do that. Parameters ----------- targets : list, `~astropy.coordinates.SkyCoord`, `Fixedtarget` either a single target or a list of targets Returns -------- coord : `~astropy.coordinates.SkyCoord` a single SkyCoord object, which may be non-scalar """ if not isinstance(targets, list): return getattr(targets, 'coord', targets) # get the SkyCoord object itself coords = [getattr(target, 'coord', target) for target in targets] # are all SkyCoordinate's in equivalent frames? If not, convert to ICRS convert_to_icrs = not all( [coord.frame.is_equivalent_frame(coords[0].frame) for coord in coords[1:]]) # we also need to be careful about handling mixtures of # UnitSphericalRepresentations and others targets_is_unitsphericalrep = [x.data.__class__ is UnitSphericalRepresentation for x in coords] longitudes = [] latitudes = [] distances = [] get_distances = not all(targets_is_unitsphericalrep) if convert_to_icrs: # mixture of frames for coordinate in coords: icrs_coordinate = coordinate.icrs longitudes.append(icrs_coordinate.ra) latitudes.append(icrs_coordinate.dec) if get_distances: distances.append(icrs_coordinate.distance) frame = ICRS() else: # all the same frame, get the longitude and latitude names try: # from astropy v2.0, keys are classes lon_name, lat_name = [ mapping.framename for mapping in coords[0].frame_specific_representation_info[UnitSphericalRepresentation]] except BaseException: # whereas prior to that they were strings. lon_name, lat_name = [mapping.framename for mapping in coords[0].frame_specific_representation_info['spherical']] frame = coords[0].frame for coordinate in coords: longitudes.append(getattr(coordinate, lon_name)) latitudes.append(getattr(coordinate, lat_name)) if get_distances: distances.append(coordinate.distance) # now let's deal with the fact that we may have a mixture of coords with distances and # coords with UnitSphericalRepresentations if all(targets_is_unitsphericalrep): return SkyCoord(longitudes, latitudes, frame=frame) elif not any(targets_is_unitsphericalrep): return SkyCoord(longitudes, latitudes, distances, frame=frame) else: """ We have a mixture of coords with distances and without. Since we don't know in advance the origin of the frame where further transformation will take place, it's not safe to drop the distances from those coords with them set. Instead, let's assign large distances to those objects with none. """ distances = [distance if distance != 1 else 100*u.kpc for distance in distances] return SkyCoord(longitudes, latitudes, distances, frame=frame) class SpecialObjectFlag(object): """ Flag this object as a special non-fixed target, which has a ``get_*`` method within astropy (like the Sun or Moon) """ pass class SunFlag(SpecialObjectFlag): """ Flag for a computation with the Sun """ approx_sidereal_drift = 5 * u.min class MoonFlag(SpecialObjectFlag): """ Flag for a computation with the Moon """ approx_sidereal_drift = 60 * u.min