HR-eRddlZddlZddlZddlZddlZddlZddlmZddl Z ddl Z ddl m Zddl mZddlmZddlmZddlmZddlmZd d lmZmZmZd d lmZd d lm Z d d l!m"Z"m#Z#m$Z$gdZ%ej&dgdZ'iZ( dej)zej*z +d ej,z ej-Z. d"dZ/dZ0GddeZ1Gddej2Z3dZ4ee4Gdde"Z5ee4Gdde5Z6ee4Gdde5Z7ee4Gd d!e5Z8dS)#N)warn) constants)units)QuantityInfoBase)data) format_doc)AstropyUserWarning)AngleLatitude Longitude)UnknownSiteException)matrix_transpose)BaseRepresentationCartesianDifferentialCartesianRepresentation) EarthLocationBaseGeodeticRepresentationWGS84GeodeticRepresentationWGS72GeodeticRepresentationGRS80GeodeticRepresentationGeodeticLocationlonlatheightg+6 ?WGS84cV||}|tvrtd|dtd|S)Nz Ellipsoid z not among known ones ()) ELLIPSOIDS ValueError) ellipsoiddefaults 9lib/python3.11/site-packages/astropy/coordinates/earth.py_check_ellipsoidr%;s@  ""UiUU UUUVVV c,ddlm} tj|t jj}tj | d}n#tj j $ri}t|jt jr ||d||||j|d}~wt j$r ||dwxYw|rP|dg}|ddd kr||d n|}|s||d |S) Nr )NameResolveError)timeoututf8zconnection timed out)msgresultsstatusOKzunknown failure with Google APIzno results returned) name_resolver(urllibrequesturlopenrconfremote_timeoutjsonloadsreaddecodeerrorURLError isinstancereasonsocketr)formatget)urlerr_str use_googler(resp resp_dataer,s r$_get_json_resultrFCs......K~%%c493K%LLJtyy{{11&99:: < HHH ah / / H""7>>6L>#M#MNNTU U""7>>ah>#?#?@@a G >KKKw~~2H~IIJJJ K  -- 2.. ==4 ( (D 0 0""#DEE  1  Jw~~2G~HHIII NsA)A22DA$C**2Dc$eZdZdZdZdZddZdS)EarthLocationInfoz Container for meta information like name, description, format. This is required when the object is used as a mixin column within a table, but can be used as a general way to store meta information. )xyzr"cX|d}|jdi|}||_|S)Nr")pop _parent_clsr")selfmapr"outs r$_construct_from_dictz&EarthLocationInfo._construct_from_dictts9GGK(( d%%%%!  r&rNc |||d}|d|f|dz}tjt j|djdjd  fd|jD}| |}| D]\} } t|j | | |S) a Return a new EarthLocation instance which is consistent with the input ``cols`` and has ``length`` rows. This is intended for creating an empty column object whose elements can be set in-place for table operations like join or vstack. Parameters ---------- cols : list List of input columns length : int Length of the output column object metadata_conflicts : str ('warn'|'error'|'silent') How to handle metadata conflicts name : str Output column name Returns ------- col : EarthLocation (or subclass) Empty instance of this class consistent with ``cols`` )metar> descriptiondtypeshaper)rXrWF)unitcopycVi|]%}||dvr|ntd|&S)xyz)getattr).0keycolsrs r$ z.EarthLocationInfo.new_like..sI    se||$s))b31G1G   r&) merge_cols_attributesrNuQuantitynpzerosrWrY_represent_as_dict_attrsrSitemssetattrinfo) rPralengthmetadata_conflictsnameattrsrXrQrRattrvaluers ` @r$new_likezEarthLocationInfo.new_like|s 6** $d,M    ' EIIg...z H5Q 6 6 6T!W\PU        4   '',, ;;== + +KD% CHdE * * * * r&)rN)__name__ __module__ __qualname____doc__rhrSrrrMr&r$rHrHksI <222222r&rHceZdZdZdZejgdejgdzdZejejdfZ dZ e Z dZ ed*fd Zed+d Zed d dZed,dZed d dZed-dZedZejdZedZd*dZedZedZedZedZdZd.dZeedZ dZ!dZ"d Z#gd!ifd"Z$ed#Z%ed$Z&ed%Z'fd&Z(fd'Z)fd(Z*gfd)Z+xZ,S)/ra Location on the Earth. Initialization is first attempted assuming geocentric (x, y, z) coordinates are given; if that fails, another attempt is made assuming geodetic coordinates (longitude, latitude, height above a reference ellipsoid). When using the geodetic forms, Longitudes are measured increasing to the east, so west longitudes are negative. Internally, the coordinates are stored as geocentric. To ensure a specific type of coordinates is used, use the corresponding class methods (`from_geocentric` and `from_geodetic`) or initialize the arguments with names (``x``, ``y``, ``z`` for geocentric; ``lon``, ``lat``, ``height`` for geodetic). See the class methods for details. Notes ----- This class fits into the coordinates transformation framework in that it encodes a position on the `~astropy.coordinates.ITRS` frame. To get a proper `~astropy.coordinates.ITRS` object from this object, use the ``itrs`` property. rrIrJrK)namesformats)ryNc t|dkrHt|dkr5t|dtr|dS |j|i|}nZ#t jtf$rA} |j|i|}n(#t$r}td|d|dd}~wwxYwYd}~nd}~wwxYw|S)Nr rz^Coordinates could not be parsed as either geocentric or geodetic, with respective exceptions "z" and "") lenr;rrZfrom_geocentricrd UnitsError TypeError from_geodetic Exception)clsargskwargsrPexc_geocentric exc_geodetics r$__new__zEarthLocation.__new__s t99>>c&kkQ..:d1g}3U3U.7<<>> ! &3&777DD i(    (s($9&99   J#1JJ:FJJJ   s6 A++C BB= B5B00B55B==Cct|' |j}n2#t$rtddwxYwtj|}|jdkrtjd tj||d}tj||d}tj||d}n'#tj$rtjdwxYwtj |||\}}}tj |j |j }|||c|d<|d <|d <t|||dS) a Location on Earth, initialized from geocentric coordinates. Parameters ---------- x, y, z : `~astropy.units.Quantity` or array-like Cartesian coordinates. If not quantities, ``unit`` should be given. unit : unit-like or None Physical unit of the coordinate values. If ``x``, ``y``, and/or ``z`` are quantities, they will be converted to this unit. Raises ------ astropy.units.UnitsError If the units on ``x``, ``y``, and ``z`` do not match or an invalid unit is given. ValueError If the shapes of ``x``, ``y``, and ``z`` do not match. TypeError If ``x`` is not a `~astropy.units.Quantity` and no unit is given. NzMGeocentric coordinates should be Quantities unless an explicit unit is given.rlz4Geocentric coordinates should be in units of length.FrZz5Geocentric coordinate units should all be consistent.rIrJrK)rYAttributeErrorrrdUnit physical_typerrerfbroadcast_arraysemptyrX_location_dtypesuperr)rrIrJrKrYstruc __class__s r$rzEarthLocation.from_geocentricse. < v!   8  6$<`_ . If the cache already exists the function will use it even if the version in the astropy-data repository has been updated unless the ``refresh_cache=True`` option is used. If there is no cache and the online version cannot be reached, this function falls back on a built-in list, which currently only contains the Greenwich Royal Observatory as an example case. Parameters ---------- site_name : str Name of the observatory (case-insensitive). refresh_cache : bool, optional If `True`, force replacement of the cached registry with a newly downloaded version. (Default: `False`) .. versionadded:: 5.3 Returns ------- site : `~astropy.coordinates.EarthLocation` (or subclass) instance The location of the observatory. The returned class will be the same as this class. Examples -------- >>> from astropy.coordinates import EarthLocation >>> keck = EarthLocation.of_site('Keck Observatory') # doctest: +REMOTE_DATA >>> keck.geodetic # doctest: +REMOTE_DATA +FLOAT_CMP GeodeticLocation(lon=, lat=, height=) >>> keck.info # doctest: +REMOTE_DATA name = W. M. Keck Observatory dtype = (float64, float64, float64) unit = m class = EarthLocation n_bad = 0 >>> keck.info.meta # doctest: +REMOTE_DATA {'source': 'IRAF Observatory Database', 'timezone': 'US/Hawaii'} See Also -------- get_site_names : the list of sites that this function can access force_downloadzEarthLocation.get_site_names) close_namesN) _get_site_registryrsiterrr to_geodeticrkrn)r site_namerregistryelrEnewels r$of_sitezEarthLocation.of_siteBsv)))GG )$BB#   &6AM   ",  I%C%r~~'7'78E glEJOLs! A AA c|du}|s|rtd|r(tj||d}d|}n'tj|dd}d|}d|d }t ||| }|r%|d d d } | d} | d} n2|d } t | d} t | d} |rd| dd| d|d}tj|}d|} d|d }t | || } | d dt jz}nd}|| t j z| t j z|S)a Return an object of this class for a given address by querying either the OpenStreetMap Nominatim tool [1]_ (default) or the Google geocoding API [2]_, which requires a specified API key. This is intended as a quick convenience function to get easy access to locations. If you need to specify a precise location, you should use the initializer directly and pass in a longitude, latitude, and elevation. In the background, this just issues a web query to either of the APIs noted above. This is not meant to be abused! Both OpenStreetMap and Google use IP-based query limiting and will ban your IP if you send more than a few thousand queries per hour [2]_. .. warning:: If the query returns more than one location (e.g., searching on ``address='springfield'``), this function will use the **first** returned location. Parameters ---------- address : str The address to get the location for. As per the Google maps API, this can be a fully specified street address (e.g., 123 Main St., New York, NY) or a city name (e.g., Danbury, CT), or etc. get_height : bool, optional This only works when using the Google API! See the ``google_api_key`` block below. Use the retrieved location to perform a second query to the Google maps elevation API to retrieve the height of the input address [3]_. google_api_key : str, optional A Google API key with the Geocoding API and (optionally) the elevation API enabled. See [4]_ for more information. Returns ------- location : `~astropy.coordinates.EarthLocation` (or subclass) instance The location of the input address. Will be type(this class) References ---------- .. [1] https://nominatim.openstreetmap.org/ .. [2] https://developers.google.com/maps/documentation/geocoding/start .. [3] https://developers.google.com/maps/documentation/elevation/start .. [4] https://developers.google.com/maps/documentation/geocoding/get-api-key NzCurrently, `get_height` only works when using the Google geocoding API, which requires passing a Google API key with `google_api_key`. See: https://developers.google.com/maps/documentation/geocoding/get-api-key for information on obtaining an API key.)addressr`z2https://maps.googleapis.com/maps/api/geocode/json?r5)qr>z+https://nominatim.openstreetmap.org/search?z,Unable to retrieve coordinates for address 'z'; {msg})rArBrgeometrylocationrlngrz.8f,) locationsr`z4https://maps.googleapis.com/maps/api/elevation/json?z*Unable to retrieve elevation for address ' elevationrr) r!r0parse urlencoderFfloatrdmeterrdeg)rr get_heightgoogle_api_keyrBparsgeo_urlrA geo_resultlocrrele_url ele_resultrs r$ of_addresszEarthLocation.of_addresssd$4/  j <   K<))gn*U*UVVDQ4QQGG<))6*J*JKKDJDJJGUTTT%gw:VVV  $Q- +J7Ce*Ce*CCQ-CE ##CE ##C  $'!7!7!7c!7!7!7OOD<))$//DSTSSGV7VVVG)ZJ ];/!'9FFF  S15[cAEk& QQQr&c8||jS)a\ Get list of names of observatories for use with `~astropy.coordinates.EarthLocation.of_site`. .. note:: This function is meant to access the site registry from the astropy data server, which is saved in the user's local cache. If you would like a site to be added there, issue a pull request to the `astropy-data repository `_ . If the cache already exists the function will use it even if the version in the astropy-data repository has been updated unless the ``refresh_cache=True`` option is used. If there is no cache and the online version cannot be reached, this function falls back on a built-in list, which currently only contains the Greenwich Royal Observatory as an example case. Parameters ---------- refresh_cache : bool, optional If `True`, force replacement of the cached registry with a newly downloaded version. (Default: `False`) .. versionadded:: 5.3 Returns ------- names : list of str List of valid observatory names See Also -------- of_site : Gets the actual location object for one of the sites names this returns. r)rrz)rrs r$get_site_nameszEarthLocation.get_site_namessH%%]%CCIIr&c\ddlm}m}|r|rtd|r||_ny|s|jsp t |t r|||_n||_n9#t$r,|rd}t|t||_YnwxYw|jS)a Gets the site registry. The first time this either downloads or loads from the data file packaged with astropy. Subsequent calls will use the cached version unless explicitly overridden. Parameters ---------- force_download : bool or str If not False, force replacement of the cached registry with a downloaded version. If a str, that will be used as the URL to download from (if just True, the default URL will be used). force_builtin : bool If True, load from the data file bundled with astropy and set the cache to that. Returns ------- reg : astropy.coordinates.sites.SiteRegistry r )get_builtin_sitesget_downloaded_sitesz6Cannot have both force_builtin and force_download TruezCould not access the main site list. Falling back on the built-in version, which is rather limited. If you want to retry the download, use the option 'refresh_cache=True'.) sitesrrr!_site_registryr;strOSErrorrr )rr force_builtinrrr+s r$rz EarthLocation._get_site_registrys, CBBBBBBB  W^ WUVV V  =!2!2!4!4C   =S%7 ==!.#66D-A-A.-Q-Q**-A-A-C-C* = = =%S 0111):):)<)The default ellipsoid used to convert to geodetic coordinates.)rrPs r$r"zEarthLocation.ellipsoidFs r&c.t||_dSN)r%r)rPr"s r$r"zEarthLocation.ellipsoidKs*955r&c*|S)z:Convert to geodetic coordinates for the default ellipsoid.)rrs r$rzEarthLocation.geodeticOs!!!r&ct||j}||jtj}t |ddt|}tt|j tj dtj zd|j tj z|j|jzS)aYConvert to geodetic coordinates. Parameters ---------- ellipsoid : str, optional Reference ellipsoid to use. Default is the one the coordinates were initialized with. Available are: 'WGS84', 'GRS80', 'WGS72' Returns ------- lon, lat, height : `~astropy.units.Quantity` The tuple is a ``GeodeticLocation`` namedtuple and is comprised of instances of `~astropy.coordinates.Longitude`, `~astropy.coordinates.Latitude`, and `~astropy.units.Quantity`. Raises ------ ValueError if ``ellipsoid`` is not recognized as among the ones implemented. Notes ----- For the conversion to geodetic coordinates, the ERFA routine ``gc2gd`` is used. See https://github.com/liberfa/erfa rr]FrrZr) wrap_anglerZ)r%r"r _array_dtyperdrer represent_asr rr rrrrrY)rPr"r\llhs r$rzEarthLocation.to_geodeticTs4%YGGG ii)1:66%cBUCCCPP y !   cgququ5 I I I Gqu  J$) #   r&c|jdS)z5Longitude of the location, for the default ellipsoid.rrrs r$rzEarthLocation.lony}Qr&c|jdS)z4Latitude of the location, for the default ellipsoid.r rrs r$rzEarthLocation.lat~rr&c|jdS)z2Height of the location, for the default ellipsoid.rrs r$rzEarthLocation.heightrr&c*|Sz2Convert to a tuple with X, Y, and Z as quantities.) to_geocentricrs r$ geocentriczEarthLocation.geocentrics!!###r&c*|j|j|jfSrrxrs r$rzEarthLocation.to_geocentrics''r&c|r.|jdkr#|jrtj||jd}ddlm}|||j|j|j|S||j|jz |j|jz |j|jz d||S) a[ Generates an `~astropy.coordinates.ITRS` object with the location of this object at the requested ``obstime``, either geocentric, or topocentric relative to a given ``location``. Parameters ---------- obstime : `~astropy.time.Time` or None The ``obstime`` to apply to the new `~astropy.coordinates.ITRS`, or if None, the default ``obstime`` will be used. location : `~astropy.coordinates.EarthLocation` or None A possible observer's location, for a topocentric ITRS position. If not given (default), a geocentric ITRS object will be created. Returns ------- itrs : `~astropy.coordinates.ITRS` The new object in the ITRS frame, either geocentric or topocentric relative to the given ``location``. r T)subok)ITRSN)rIrJrKobstimeF)rZrr) sizerXrf broadcast_tobuiltin_framesrrIrJrK)rPrrrs r$get_itrszEarthLocation.get_itrss.  DtyA~~'-~?4dCCCD )(((((  4$&DFdfgFFF F4###!  r&zAn `~astropy.coordinates.ITRS` object for the location of this object at the default ``obstime``.)doccddlm}||\}}tj||jd<|||S)aXGCRS position with velocity at ``obstime`` as a GCRS coordinate. Parameters ---------- obstime : `~astropy.time.Time` The ``obstime`` to calculate the GCRS position/velocity at. Returns ------- gcrs : `~astropy.coordinates.GCRS` instance With velocity included. r )GCRSs)r)rrget_gcrs_posvelrfrom_cartesian differentials)rPrrrvels r$get_gcrszEarthLocation.get_gcrssZ )(((((''00S!6!Ec!J!J#tC))))r&c$t|}|t|z}t|dtzdd}t|j|j|jd}||}||} || fS)aCalculate GCRS position and velocity given transformation matrices. The reference frame z axis must point to the Celestial Intermediate Pole (as is the case for CIRS and TETE). This private method is used in intermediate_rotation_transforms, where some of the matrices are already available for the coordinate transformation. The method is faster by an order of magnitude than just adding a zero velocity to ITRS and transforming to GCRS, because it avoids calculating the velocity via finite differencing of the results of the transformation at three separate times. ).rr]Frr)rr OMEGA_EARTHrIrJrK transformcross) rPr ref_to_itrs gcrs_to_ref ref_to_gcrs itrs_to_gcrs rot_vec_gcrs itrs_cartposrs r$_get_gcrs_posvelzEarthLocation._get_gcrs_posvels*'{33 "%5k%B%BB /  + -   ,DFDFDFOOO !!,//  %%Cxr&cdddlm}m}||||||S)a Calculate the GCRS position and velocity of this object at the requested ``obstime``. Parameters ---------- obstime : `~astropy.time.Time` The ``obstime`` to calculate the GCRS position/velocity at. Returns ------- obsgeoloc : `~astropy.coordinates.CartesianRepresentation` The GCRS position of the object obsgeovel : `~astropy.coordinates.CartesianRepresentation` The GCRS velocity of the object r )cirs_to_itrs_matgcrs_to_cirs_mat)/builtin_frames.intermediate_rotation_transformsrrr)rPrrrs r$rzEarthLocation.get_gcrs_posvelsc$        $$ %%g..0@0@0I0I   r&)sunjupitermoonc ddlm t|}d|vr|d|dt jt jt jdztj zt j d}| |g}tj tj t jtj zf}|D]} |||tjdztjdzz |gQ#t"$r}t#d|d |d }~wtj$r} | xjd z c_d } ~ wwxYw fd |D fd  d dD} | t)|jdt/|| D} t1| d d dS)aReturn the gravitational redshift at this EarthLocation. Calculates the gravitational redshift, of order 3 m/s, due to the requested solar system bodies. Parameters ---------- obstime : `~astropy.time.Time` The ``obstime`` to calculate the redshift at. bodies : iterable, optional The bodies (other than the Earth) to include in the redshift calculation. List elements should be any body name `get_body_barycentric` accepts. Defaults to Jupiter, the Sun, and the Moon. Earth is always included (because the class represents an *Earth* location). masses : dict[str, `~astropy.units.Quantity`], optional The mass or gravitational parameters (G * mass) to assume for the bodies requested in ``bodies``. Can be used to override the defaults for the Sun, Jupiter, the Moon, and the Earth, or to pass in masses for other bodies. Returns ------- redshift : `~astropy.units.Quantity` Gravitational redshift in velocity units at given obstime. r )get_body_barycentricearthgXJ\D)r r r rryrzbody "z" does not have a mass.N)zD"masses" argument values must be masses or gravitational parameters.c(g|]}|SrMrM)r_rnr rs r$ z8EarthLocation.gravitational_redshift..Ts'LLLT))$88LLLr&cJg|]}|dz  S)r])norm)r_r positionss r$rz8EarthLocation.gravitational_redshift..Vs.LLLccIbM)//11LLLr&r]c:g|]\}}| tjz |z SrM)constsc)r_GMdistances r$rz8EarthLocation.gravitational_redshift..Zs6   *82xRC&(NX %   r&) solar_systemr listremoveappendrGM_sunGM_jupGrdkgGM_earthupdatertorrKeyErrorrrrrrzipsum)rPrbodiesmasses_massesGMsM_GM_equivalencybodyerrexc distances redshiftsr rs ` @@r$gravitational_redshiftz$EarthLocation.gravitational_redshifts&@ 766666f f   MM' " " " g=}H},qt3_    vD!&AD"9"9:  D  74=++ACFQS!VO>N=OPPQQQQ P P PEEEEFFCO<     MLLLLVLLL LLLLYss^LLL 0AAFFHHIII  > > > > > > >r&ra{__doc__} Parameters ---------- lon, lat : angle-like The longitude and latitude of the point(s), in angular units. The latitude should be between -90 and 90 degrees, and the longitude will be wrapped to an angle between 0 and 360 degrees. These can also be instances of `~astropy.coordinates.Angle` and either `~astropy.coordinates.Longitude` not `~astropy.coordinates.Latitude`, depending on the parameter. height : `~astropy.units.Quantity` ['length'] The height to the point(s). copy : bool, optional If `True` (default), arrays will be copied. If `False`, arrays will be references, though possibly broadcast to ensure matching shapes. c`eZdZdZeeejdZfdZ d fd Z dZ e dZ xZS) rzBase geodetic representation.rc ntjdi|d|jvr|t|j<dSdS)NrrM)r__init_subclass____dict__r r)rrrs r$rQz,BaseGeodeticRepresentation.__init_subclass__sG!!++F+++ 3< ' '),Js~ & & & ( 'r&NTc2|$t||jsdtjz}t |||||jjtjs!tj |jj ddS)Nrrz& requires height with units of length.) r;rrdrr__init__rrY is_equivalent UnitTypeErrorrs)rPrrrrZrs r$rTz#BaseGeodeticRepresentation.__init__s >*S$."A"A>!#XF c6555{--ac22 />*RRR   r&ctjtt|j|j|j|j}t|ddS)zs Converts WGS84 geodetic coordinates to 3D rectangular (geocentric) cartesian coordinates. r]Fr)erfagd2gcr^rrrrr)rPr\s r$rz'BaseGeodeticRepresentation.to_cartesiansG j D$/ * *DHdh   'sReDDDDr&ctjtt|j|d\}}}||||dS)z{ Converts 3D rectangular cartesian coordinates (assumed geocentric) to WGS84 geodetic coordinates. r]rFr)rXgc2gdr^rr)rcartrrrs r$rz)BaseGeodeticRepresentation.from_cartesiansV  : D#. ) )4<<<+D+D  S&s3V%0000r&)NNT)rsrtrurvr r rdre attr_classesrQrTrrIrrMrNs@r$rrs''$XLLL----- EEE11[11111r&rceZdZdZdZdS)rz:Representation of points in WGS84 3D geodetic coordinates.rNrsrtrurvrrMr&r$rrDDJJJr&rceZdZdZdZdS)rz:Representation of points in WGS72 3D geodetic coordinates.WGS72Nr_rMr&r$rrr`r&rceZdZdZdZdS)rz:Representation of points in GRS80 3D geodetic coordinates.GRS80Nr_rMr&r$rrr`r&r)Nr)9 collectionsr5r= urllib.errorr0 urllib.parseurllib.requestwarningsrrXnumpyrfastropyrrrrdastropy.units.quantityr astropy.utilsrastropy.utils.decoratorsrastropy.utils.exceptionsr anglesr r r errorsrmatrix_utilitiesrrepresentationrrr__all__ namedtuplerr cycledayr#rdimensionless_anglesrr%rFrHrergeodetic_base_docrrrrrMr&r$rzsu  ''''''333333//////777777..........((((((......    *;)*<>V>V>VWW  N(!'1AE9==G #Q # % %  %%%PCCCCC(CCCLZ >Z >Z >Z >Z >AJZ >Z >Z >z(  '1'1'1'1'1!3'1'1'1T  "<   "<   "<r&