a ߙfb%@s^dZddlZddlZddlmZddlmZddl m Z dgZ dd giZ Gd ddej ZdS) z` This module contains the classes and utility functions for distance and cartesian coordinates. N)units)AstropyWarning)AngleDistance*Zscipyc sfeZdZdZejZdZdfdd Ze dd Z dd d Z e d d Z e ddZe ddZZS)ra A one-dimensional distance. This can be initialized in one of five ways: * A distance ``value`` (array or float) and a ``unit`` * A `~astropy.units.Quantity` object * A redshift and (optionally) a cosmology. * Providing a distance modulus * Providing a parallax Parameters ---------- value : scalar or `~astropy.units.Quantity` ['length'] The value of this distance. unit : `~astropy.units.UnitBase` ['length'] The units for this distance, *if* ``value`` is not a `~astropy.units.Quantity`. Must have dimensions of distance. z : float A redshift for this distance. It will be converted to a distance by computing the luminosity distance for this redshift given the cosmology specified by ``cosmology``. Must be given as a keyword argument. cosmology : `~astropy.cosmology.Cosmology` or None A cosmology that will be used to compute the distance from ``z``. If `None`, the current cosmology will be used (see `astropy.cosmology` for details). distmod : float or `~astropy.units.Quantity` The distance modulus for this distance. Note that if ``unit`` is not provided, a guess will be made at the unit between AU, pc, kpc, and Mpc. parallax : `~astropy.units.Quantity` or `~astropy.coordinates.Angle` The parallax in angular units. dtype : `~numpy.dtype`, optional See `~astropy.units.Quantity`. copy : bool, optional See `~astropy.units.Quantity`. order : {'C', 'F', 'A'}, optional See `~astropy.units.Quantity`. subok : bool, optional See `~astropy.units.Quantity`. ndmin : int, optional See `~astropy.units.Quantity`. allow_negative : bool, optional Whether to allow negative distances (which are possible is some cosmologies). Default: ``False``. Raises ------ `~astropy.units.UnitsError` If the ``unit`` is not a distance. ValueError If value specified is less than 0 and ``allow_negative=False``. If ``z`` is provided with a ``unit`` or ``cosmology`` is provided when ``z`` is *not* given, or ``value`` is given as well as ``z``. If none of ``value``, ``z``, ``distmod``, or ``parallax`` were given. Examples -------- >>> from astropy import units as u >>> from astropy.cosmology import WMAP5, WMAP7 >>> d1 = Distance(10, u.Mpc) >>> d2 = Distance(40, unit=u.au) >>> d3 = Distance(value=5, unit=u.kpc) >>> d4 = Distance(z=0.23) >>> d5 = Distance(z=0.23, cosmology=WMAP5) >>> d6 = Distance(distmod=24.47) >>> d7 = Distance(Distance(10 * u.Mpc)) >>> d8 = Distance(parallax=21.34*u.mas) TNFrc  stdd||||fD} | dkr,tdn| dkrgz#Distance.__new__..rzQnone of `value`, `z`, `distmod`, or `parallax` were given to Distance constructorrzZmore than one of `value`, `z`, `distmod`, or `parallax` were given to Distance constructordefault_cosmologyFzHA `cosmology` was given but `z` was not provided in Distance constructor)Z equivalencieszNegative parallaxes are converted to NaN distances even when `allow_negative=True`, because negative parallaxes cannot be transformed into distances. See discussion in this paper: https://arxiv.org/abs/1507.02105a Some parallaxes are negative, which are notinterpretable as distances. See the discussion in this paper: https://arxiv.org/abs/1507.02105 . If you want parallaxes to pass through, with negative parallaxes instead becoming NaN, use the `allow_negative=True` argument.)dtypecopyordersubokndminignore)ZinvalidzXDistance must be >= 0. Use the argument 'allow_negative=True' to allow negative values.)sum ValueErrorastropy.cosmologyrgetluminosity_distance_distmod_to_pcnplog10valueZmeanuZMpcZkpcZAUpcto_valueparallaxanywarningswarnrsuper__new__Zerrstate)clsr!Zunitz cosmologydistmodr%rrrrrZallow_negativeZ n_not_nonerZ meanlogvalZdistanceZ any_negative __class__rr r*cs\       0 zDistance.__new__cCs|S)zShort for ``self.compute_z()``) compute_zselfrrr r,sz Distance.zcKsHddlm}|dur(ddlm}|}|dd||j|fi|S)a The redshift for this distance assuming its physical distance is a luminosity distance. Parameters ---------- cosmology : `~astropy.cosmology.Cosmology` or None The cosmology to assume for this calculation, or `None` to use the current cosmology (see `astropy.cosmology` for details). **atzkw keyword arguments for :func:`~astropy.cosmology.z_at_value` Returns ------- z : `~astropy.units.Quantity` The redshift of this distance given the provided ``cosmology``. Warnings -------- This method can be slow for large arrays. The redshift is determined using :func:`astropy.cosmology.z_at_value`, which handles vector inputs (e.g. an array of distances) by element-wise calling of :func:`scipy.optimize.minimize_scalar`. For faster results consider using an interpolation table; :func:`astropy.cosmology.z_at_value` provides details. See Also -------- :func:`astropy.cosmology.z_at_value` : Find the redshift corresponding to a :meth:`astropy.cosmology.FLRW.luminosity_distance`. r) z_at_valueNrZztolg|=)rr4rr setdefaultr)r3r-Zatzkwr4rrrr r1s   zDistance.compute_zcCs,dt|tjd}tj|tjddS)z3The distance modulus as a `~astropy.units.Quantity`@Fr)rr r$r"r#Quantitymag)r3valrrr r.szDistance.distmodcCs,t|tj}|d|jddtjddS)N r6Fr7)r"r8r9r!r#)r+Zdmrrr rszDistance._distmod_to_pccCst|tjtS)zs