a ߙfb\v@sddlZddlmZddlZddlZddlmZmZddl m Z ddl m Z mZmZmZddlmZddlmZmZmZed Zed ejZd d Zd ZddZddZddZddZ ddZ!ddZ"dS)N)Sequence)UnitIrreducibleUnit)units)BaseCoordinateFrameframe_transform_graph _get_repr_cls _get_diff_cls)ICRS)BaseRepresentationSphericalRepresentationUnitSphericalRepresentationz(\+|\-)zJ # J prefix ([0-9]{6,7}\.?[0-9]{0,2}) # RA as HHMMSS.ss or DDDMMSS.ss, optional decimal digits ([\+\-][0-9]{6}\.?[0-9]{0,2})\s*$ # Dec as DDMMSS.ss, optional decimal digits cCsjt|tr:t}||vr.td|t|t|}n,t |rTt |t rT|}ntd|j j |S)zk Get a frame class from the input `frame`, which could be a frame name string, or frame class. z?Coordinate frame name "{}" is not a known coordinate frame ({})z@Coordinate frame must be a frame name or frame class, not a '{}') isinstancestrrZ get_names ValueErrorformatsortedZ lookup_nameinspectZisclass issubclassr __class____name__)frame frame_names frame_clsrIlib/python3.9/site-packages/astropy/coordinates/sky_coordinate_parsers.py_get_frame_class s  rzCoordinate attribute '{0}'={1!r} conflicts with keyword argument '{0}'={2!r}. This usually means an attribute was set on one of the input objects and also in the keyword arguments to {3}c Csddlm}d}i}|dd}|durt||r|jD]R}||vr~tt||||kr~tt |t||||dq:t||||<q:|j }t|t r| D]4}||vrtd |q||st||||<q|j}|d|j|d|j|durt|}|D]R}t|ttjfrPt|dkrPt|d krP|d }d}} t|t rj|}nt||r||j }|dur.|j} |d } | dur|d| | D]x}||vr||st||t||krtd |t||||n$||vr||st||||<q| dur|durH| }n|| urtd | j|jq|durxt}t|t std |j|jD]}||vr||||<qd|vrt|d|d<|dd} | durt| |d<||fS)an Determines the coordinate frame from input SkyCoord args and kwargs. This function extracts (removes) all frame attributes from the kwargs and determines the frame class either using the kwargs, or using the first element in the args (if a single frame object is passed in, for example). This function allows a frame to be specified as a string like 'icrs' or a frame class like ICRS, or an instance ICRS(), as long as the instance frame attributes don't conflict with kwargs passed in (which could require a three-way merge with the coordinate data possibly specified via the args). rSkyCoordNrrzCannot specify frame attribute '{}' directly as an argument to SkyCoord because a frame instance was passed in. Either pass a frame class, or modify the frame attributes of the input frame instance.representation_typedifferential_typerszFrame attribute '{}' has conflicting values between the input coordinate data and either keyword arguments or the frame specification (frame=...): {} =/= {}zfCannot override frame='{}' of input coordinate with new frame='{}'. Instead, transform the coordinate.z!Frame class has unexpected type: )sky_coordinaterpopr_extra_frameattr_namesnpanygetattrr_conflict_err_msgrrrget_frame_attr_namesZis_frame_attr_defaultr setdefaultr r!rrndarraylenget_representation_clsrr rframe_attributesr r ) argskwargsrrZframe_cls_kwargsrattrargZcoord_frame_objZcoord_frame_clsZ frame_diffr!rrr_get_frame_without_data>s                                 r4cCs(i}i}d}t|}|D]}|tjvr||||<qt||}|t||||rd} |jt krt| } | d} | d} d| t|vrd | | } t d d d d |D| |rt|dkrt|d|||\} }d t|dd d vr|dj}nxt|dkr|i} i}|j}|j}t||||D]*\}}}}|jj|}|||d||<qNnt d t||D]4\}}||vrt t ||||d|||<q| D]H\}}||vrt|||krt t ||||d|||<q|||fS)aS Extract coordinate data from the args and kwargs passed to SkyCoord. By this point, we assume that all of the frame attributes have been extracted from kwargs (see _get_frame_without_data()), so all that are left are (1) extra SkyCoord attributes, and (2) the coordinate data, specified in any of the valid ways. NrrZpm_z By default, most frame classes expect the longitudinal proper motion to include the cos(latitude) term, named `pm_{}_cos{}`. Did you mean to pass in this component?z%Unrecognized keyword argument(s) {}{}z, css|]}d|dVqdS)'Nr).0keyrrr sz)_parse_coordinate_data..info__dict__runitz;Must supply no more than three positional arguments, got {}r)listkeysrr/r$#_get_representation_component_unitsupdate_get_representation_attrsr r "get_representation_component_namesrrjoinr-_parse_coordinate_argr(r:representation_component_namesvalueszip attr_classesitemsr)r&r')rr0r1Zvalid_skycoord_kwargsZvalid_componentsr:Z attr_namesr2rZ pm_messagerZlon_nameZlat_nameZ_skycoord_kwargsZ _componentsframe_attr_namesrepr_attr_namesr3frame_attr_namerepr_attr_namer>Z attr_classZ coord_valuevaluerrr_parse_coordinate_datas~                 rQc Csd|vrgd}n|d}t|tr^dd|dD}t|dkrv|d|d|dg}nt|ttfrv|||g}zDdd|D}|d d td t|Dt|d krt Wn.t y}zt d |WYd }~n d }~00|S)zc Get the unit from kwargs for the *representation* components (not the differentials). r>)NNNcSsg|] }|qSr)stripr7xrrr Tz7_get_representation_component_units..,rrcSsg|]}|rt|ndqSN)rrSrrrrU\rVcss|] }dVqdSrXrrSrrrr9]rVz6_get_representation_component_units..r<zSUnit keyword must have one to three unit values as tuple or comma-separated string.N) r$rrsplitr-rrextendranger Exception)r0r1rerrrrrrAHs&        rAc) sPddlmd}i}i}t|j}t|j}t|jj} t|} t |t r`d}|g}t |t fr|j s~t d|j|jg} g} |D]4} t |jtr| dkr| | q| t| q| D]&}||}||=||=||=| |=q|jjrd|jjvr|jjd}|j|j|djd}|dD]p\}}|dkrxt||sxd |vrxqH| t|||d ||||| |j|qHtjD]B}t||d }t |p||v}|r|d ur|||<qnt |tr|jrd|jvr|d}||j|fd d |D} |dD]T\}}| tjd||d ||||| jdj|q`n||jfd d |D} nt |tjr|j j!dvr|j"dkr|j#ddkr|$} nt |t%tjfrg}d|jvoHd|jv}t tft&fdd|Drfdd |D}|dd D]*}|'|dst d(||dqt |djt}|dj)jD]}t|dj)|||<q|dj*D]}t|d|||<qg} t+||D]~\}} |r>| dkr>q"g}|D],}t||} ||j,rj| -dn| qFt.|}!t |!t/j0r| j1|!_2| |!q"q|D]l}"t |"t r |"3}#t|#dkrd4|#d dd4|#dd f}"n|rt5|"}"n|#}"||"qzt6t7dd|D}$Wn0t8yb}%zt d|%WYd }%~%n d }%~%00t|$dkrt d(|$|$d}$|$| krt d(|$|j| dd t+|D} |rdd | D} nt d z4t+|| | |D] \}&}'}}(|'||(dd!||&<qWn8t8yF}%zt d"(||&|%WYd }%~%n d }%~%00||fS)#am Single unnamed arg supplied. This must be: - Coordinate frame with data - Representation - SkyCoord - List or tuple of: - String which splits into two values - Iterable with two values - SkyCoord, frame, or representation objects. Returns a dict mapping coordinate attribute names to values (or lists of values) rrFTz6Cannot initialize from a frame without coordinate dataZdistancer"Z d_distancer>Ncsg|]}t|qSrr(r7rOdatarrrUrVz)_parse_coordinate_arg..csg|]}t|qSrr^r_r`rrrUrVifr<ZraZdecc3s|]}t|VqdSrX)rr7coord) coord_typesrrr9rVz(_parse_coordinate_arg..csg|]}|fiqSrrrd)r init_kwargsrrrUrVrz5List of inputs don't have equivalent frames: {} != {} css|]}t|VqdSrX)r-rSrrrr9 rVz>One or more elements of input sequence does not have a length.zCInput coordinate values must have same number of elements, found {}zFInput coordinates have {} values but representation {} only accepts {}cSsg|] }t|qSr)r?rSrrrrU"rVcSsg|] }|dqS)rrrSrrrrU%rVz,Cannot parse coordinates from first argument)r>copyz6Cannot parse first argument data "{}" for attribute {})9r#rr?rGr@rHr rJr-rrrZhas_datarraZ represent_asrappendr(indexZ differentialsr.rDrKhasattrZget_namerr/r*r r&r,ZdtypekindndimshapeZ transposerr'Zis_equivalent_framerrr%rIZisscalarZreshapeZ concatenateuZQuantityr>Z_unitrYrE _parse_ra_decrsetr\))ZcoordsrrrgZ is_scalarZ componentsZskycoord_kwargsrLrMrepr_attr_classesZ n_attr_namesrHZrepr_attr_name_to_droprOZ nametodropZnameidxZorig_velZvelZfrnameZreprnamer2rPZ use_valueZdiffsZvalsZis_ra_dec_representationZscsZscZallunitsphreprZfattrnmZdata_attr_nameZ data_valsZdata_valZ concat_valsrecoord1Zn_coordsr]rNrepr_attr_classr>r)rrfrargrrFgs"                              &   rFc Cs|j}|jj}i}t|||D]\}}}||d} | dur&z|| |d||<Wq&tjy} z2d|d|j d|d} t| | WYd} ~ q&d} ~ 00q&|j } | dur| d D]2\} }| j|}|| d} | dur|| || <q|S)a` Find instances of the "representation attributes" for specifying data for this frame. Pop them off of kwargs, run through the appropriate class constructor (to validate and apply unit), and put into the output valid_kwargs. "Representation attributes" are the frame-specific aliases for the underlying data values in the representation, e.g. "ra" for "lon" for many equatorial spherical representations, or "w" for "x" in the cartesian representation of Galactic. This also gets any *differential* kwargs, because they go into the same frame initializer later on. Nr=zUnit 'z' (z) could not be applied to 'z'. This can occur when passing units for some coordinate components when other components are specified as Quantity objects. Either pass a list of units for all components (and unit-less coordinate data), or pass Quantities for all components.r") rGr@r rJrHrIr$rqZUnitConversionErrorZ physical_typer!rDrK)rrr1rLrtZ valid_kwargsrNrvr>rPr]Z error_messager!Z frame_nameZ repr_nameZdiff_attr_classrrrrC7s(   $  rCcCst|tr|}ntdt|dkrPd|ddd|ddf}nt|dkrt|}|dd|ddf}nPt|dkrt|}|r| }t|dd dd kr8|dddd|ddd d|dd d|dddd|ddd d|dd df}np|dddd|ddd d|dd d|dddd|ddd d|dd df}n$t|}|dd|ddf}n|}|S) aj Parse RA and Dec values from a coordinate string. Currently the following formats are supported: * space separated 6-value format * space separated <6-value format, this requires a plus or minus sign separation between RA and Dec * sign separated format * JHHMMSS.ss+DDMMSS.ss format, with up to two optional decimal digits * JDDDMMSS.ss+DDMMSS.ss format, with up to two optional decimal digits Parameters ---------- coord_str : str Coordinate string to parse. Returns ------- coord : str or list of str Parsed coordinate values. zcoord_str must be a single strrhriNr<rcrr.) rrrY TypeErrorr-rE PLUS_MINUS_REJ_PREFIXED_RA_DEC_REmatchgroups)Z coord_strrureZmatch_jrrrrres.   (   6666 rr)#recollections.abcrrZnumpyr&Z astropy.unitsrrZastropyrrqZ baseframerrr r Zbuiltin_framesr Zrepresentationr r rcompiler|VERBOSEr}rr)r4rQrArFrCrrrrrrs.    uQ.