a ߙfb5@sdZddlZddlmZddlZddlZddlmZ ddl m Z ddl m Z ddlmZmZdd lmZdd lmZmZdd lmZmZdd lmZgd ZddZddZddZiZdddZ ddZ!ddZ"ddZ#dS)z This module contains convenience functions for coordinate-related functionality. This is generally just wrapping around the object-oriented coordinates framework, but it is useful for some users who are used to more functional interfaces. N)Sequence)units)c)ascii) isiterabledataSkyCoord)GCRSPrecessedGeocentric)SphericalRepresentationCartesianRepresentation)get_jd12)cartesian_to_sphericalspherical_to_cartesianget_sunget_constellationconcatenate_representations concatenatecCsbt|ds|tj}t|ds(|tj}t|ds<|tj}t|||}|t}|j|j|jfS)a Converts 3D rectangular cartesian coordinates to spherical polar coordinates. Note that the resulting angles are latitude/longitude or elevation/azimuthal form. I.e., the origin is along the equator rather than at the north pole. .. note:: This function simply wraps functionality provided by the `~astropy.coordinates.CartesianRepresentation` and `~astropy.coordinates.SphericalRepresentation` classes. In general, for both performance and readability, we suggest using these classes directly. But for situations where a quick one-off conversion makes sense, this function is provided. Parameters ---------- x : scalar, array-like, or `~astropy.units.Quantity` The first Cartesian coordinate. y : scalar, array-like, or `~astropy.units.Quantity` The second Cartesian coordinate. z : scalar, array-like, or `~astropy.units.Quantity` The third Cartesian coordinate. Returns ------- r : `~astropy.units.Quantity` The radial coordinate (in the same units as the inputs). lat : `~astropy.units.Quantity` ['angle'] The latitude in radians lon : `~astropy.units.Quantity` ['angle'] The longitude in radians unit) hasattrudimensionless_unscaledr represent_asr distancelatlon)xyzcartsphr#8lib/python3.9/site-packages/astropy/coordinates/funcs.pyrs#        rcCsdt|ds|tj}t|ds(|tj}t|ds<|tj}t|||d}|t}|j|j|j fS)ai Converts spherical polar coordinates to rectangular cartesian coordinates. Note that the input angles should be in latitude/longitude or elevation/azimuthal form. I.e., the origin is along the equator rather than at the north pole. .. note:: This is a low-level function used internally in `astropy.coordinates`. It is provided for users if they really want to use it, but it is recommended that you use the `astropy.coordinates` coordinate systems. Parameters ---------- r : scalar, array-like, or `~astropy.units.Quantity` The radial coordinate (in the same units as the inputs). lat : scalar, array-like, or `~astropy.units.Quantity` ['angle'] The latitude (in radians if array or scalar) lon : scalar, array-like, or `~astropy.units.Quantity` ['angle'] The longitude (in radians if array or scalar) Returns ------- x : float or array The first cartesian coordinate. y : float or array The second cartesian coordinate. z : float or array The third cartesian coordinate. r)rrr) rrrZradianr rrrrr )rrrr"r!r#r#r$rNs"       rc Cstjt|d\}}|d}|d}|ttjtj}t tj |ddd}dtj |dddd}t || |j d | ||}t| |d tj| |d tj| |d tjd }t|t|ddS)a6 Determines the location of the sun at a given time (or times, if the input is an array `~astropy.time.Time` object), in geocentric coordinates. Parameters ---------- time : `~astropy.time.Time` The time(s) at which to compute the location of the sun. Returns ------- newsc : `~astropy.coordinates.SkyCoord` The location of the sun as a `~astropy.coordinates.SkyCoord` in the `~astropy.coordinates.GCRS` frame. Notes ----- The algorithm for determining the sun/earth relative position is based on the simplified version of VSOP2000 that is part of ERFA. Compared to JPL's ephemeris, it should be good to about 4 km (in the Sun-Earth vector) from 1900-2100 C.E., 8 km for the 1800-2200 span, and perhaps 250 km over the 1000-3000. Ztdbpv)Zaxisrg?)r).r).r).r()rrr )Zobstimeframe)erfaZepv00rrto_valuerZaudnpZsqrtsumZabZreshapeshaperZAUr r ) timeZearth_pv_helioZ earth_pv_baryZearth_pZearth_vZdsunZ invlorentzZ properdirZcartrepr#r#r$r}srFiaucs|dkrtdtstd}tj|gdd}tjddd}td d |d Dt fd d |d D}|td<|td<ntd}td}|j }t .t dtj|tdd}Wdn1s0Y|r|jj} |jj} n|jj} |jj} tjt| td } | dk} t|D]X\} }|d| k| |dk@| |dk@}| | | |@<| dk} t| dkr2qq2td|| |r|d | }n|| }|r|dS|SdS)a Determines the constellation(s) a given coordinate object contains. Parameters ---------- coord : coordinate-like The object to determine the constellation of. short_name : bool If True, the returned names are the IAU-sanctioned abbreviated names. Otherwise, full names for the constellations are used. constellation_list : str The set of constellations to use. Currently only ``'iau'`` is supported, meaning the 88 "modern" constellations endorsed by the IAU. Returns ------- constellation : str or string array If ``coords`` contains a scalar coordinate, returns the name of the constellation. If it is an array coordinate object, it returns an array of names. Notes ----- To determine which constellation a point on the sky is in, this precesses to B1875, and then uses the Delporte boundaries of the 88 modern constellations, as tabulated by `Roman 1987 `_. r3z8only 'iau' us currently supported for constellation_listz#data/constellation_data_roman87.dat)ralraudeclname)nameszdata/constellation_names.datZUTF8)encodingcSs.g|]&}|ds|dd|ddfqS)#N) startswith).0lr#r#r$ s z%get_constellation.. csg|] }|qSr#r#)r>ZnmZcnames_short_to_longr#r$r@r7ctable cnames_longignoreZB1875)ZequinoxN)Zdtyper)r4r5r6rz-Could not find constellation for coordinates ) ValueError_constellation_datarZget_pkg_data_contentsrreaddictsplitr/Zarrayisscalarwarningscatch_warnings simplefilterr,Z ErfaWarningZ transform_tor ZraZravelZhourZdecZdegZoneslenint enumerater0)coordZ short_nameZconstellation_listZcdatarDZcnamesrErLZ constel_coordZrahZdecdZ constellidxZnotidedirowZmskr8r#rBr$rsL   . $  rcsXg}|D]Jt|djfdd|D}ttj|}|>}||q|S)z Helper function for the concatenate function below. Gets and concatenates all of the individual components for an iterable of representations or differentials. rcsg|]}t|qSr#)getattrr-)r>rr7Zunit0r#r$r@rCz+_concatenate_components..)rVrr/rZ atleast_1dappend)Z reps_difsr8valuesZ data_valsZ concat_valsr#rWr$_concatenate_componentss rZcst|ttjfstdt|dtfdd|DrBtdt|j }|}tdd|D}|rtdd|Drt d|rt|dj d tfd d|Drtd td d |Dj }|}| d |i}|S)a Combine multiple representation objects into a single instance by concatenating the data in each component. Currently, all of the input representations have to be the same type. This properly handles differential or velocity data, but all input objects must have the same differential object type as well. Parameters ---------- reps : sequence of `~astropy.coordinates.BaseRepresentation` The objects to concatenate Returns ------- rep : `~astropy.coordinates.BaseRepresentation` subclass instance A single representation object with its data set to the concatenation of all the elements of the input sequence of representations. z;Input must be a list or iterable of representation objects.rc3s|]}t|kVqdS)N)typer>r%)rep_typer#r$ 5rCz.concatenate_representations..z2Input representations must all have the same type.css|]}d|jvVqdSsN differentialsr>Zrepr#r#r$r^>rCcss|]}d|jvVqdSr_rarcr#r#r$r^?rCzZInput representations must either all contain differentials, or not contain differentials.r`c3s*|]"}d|jvp t|jdkVqdSr_)rbr[r\)dif_typer#r$r^Fs z?All input representations must have the same differential type.cSsg|]}|jdqS)r`rar\r#r#r$r@LrCz/concatenate_representations..) isinstancerr/Zndarray TypeErrorr[anyrZZ attr_classeskeysrGrbZwith_differentials)ZrepsrYZnew_repZhas_diffZnew_difr#)rdr]r$rs0  rcCs~t|ddst|stddd|D}|ddD]&}||ds6td ||dq6ttd d|D|djd S) a| Combine multiple coordinate objects into a single `~astropy.coordinates.SkyCoord`. "Coordinate objects" here mean frame objects with data, `~astropy.coordinates.SkyCoord`, or representation objects. Currently, they must all be in the same frame, but in a future version this may be relaxed to allow inhomogeneous sequences of objects. Parameters ---------- coords : sequence of coordinate-like The objects to concatenate Returns ------- cskycoord : SkyCoord A single sky coordinate with its data set to the concatenation of all the elements in ``coords`` rLFz,The argument to concatenate must be iterablecSsg|]}t|ddqS)F)copyr )r>rSr#r#r$r@lrCzconcatenate..rNrz0All inputs must have equivalent frames: {} != {}cSsg|] }|jqSr#)r)r>rr#r#r$r@vrCr*) rVrrfZis_equivalent_framerGformatr rr+)ZcoordsZscsZscr#r#r$rTsr)Fr3)$__doc__rMcollections.abcrZnumpyr/r,ZastropyrrZastropy.constantsrZ astropy.iorZ astropy.utilsrrZsky_coordinater Zbuiltin_framesr r Zrepresentationr rZbuiltin_frames.utilsr__all__rrrrHrrZrrr#r#r#r$s*      0// Y;