B îq\IYã @s"ddlZddlZddlmZmZddlZddlmZm Z ddl m Z ddl mZddlmZddlmZmZdd lmZdd lmZdd lmZe d ¡Ze d ¡Zdddddddddg ZdZdZd  d !e¡¡Z"dd„Z#dddd œZ$d!d"„Z%d#d$„Z&d%d&„Z'd'd(„Z(d)d*„Z)d+d,„Z*d-d.„Z+dS)/éN)Ú defaultdictÚ OrderedDicté)ÚHeaderÚCard)Úunits)Ú EarthLocation)ÚColumn)ÚTimeÚ TimeDelta)ÚBARYCENTRIC_SCALES)ÚFITS_DEPRECATED_SCALES)ÚAstropyUserWarningz(?P[A-Z]+)[-]+z(?P[A-Z]+)\s*ÚsÚdÚaZcyÚminÚhZyrZtaZBa)ÚTIMESYSÚMJDREFÚJDREFÚDATEREFÚTREFPOSZTREFDIRZTIMEUNITZTIMEOFFSzOBSGEO-XzOBSGEO-YzOBSGEO-ZzOBSGEO-LzOBSGEO-BzOBSGEO-HÚDATEzDATE-OBSzDATE-AVGzDATE-BEGzDATE-ENDzMJD-OBSzMJD-AVGzMJD-BEGzMJD-END)ÚTCTYPÚTCUNIÚTRPOSz ({0})[0-9]+ú|cCst t|¡dk S)z” Check if the FITS header keyword is a time column-specific keyword. Parameters ---------- keyword : str FITS keyword. N)ÚreÚmatchÚCOLUMN_TIME_KEYWORD_REGEXP)Úkeyword©r"ú7lib/python3.7/site-packages/astropy/io/fits/fitstime.pyÚis_time_column_keyword3s r$)ÚUTCzDefault time scale)gzTime columns are jd = jd1 + jd2)Ú TOPOCENTERzTime reference position)rrrcsxt ˆdˆd ¡¡ˆd<ˆdtjkrˆddkrRt dt¡dˆd<dˆd<ˆddkr|t dt¡d ˆd<d ˆd<q˜td   ˆd¡ƒ‚nd ˆd<‡fd d „dDƒ}t |ƒdkrÎt j |dt jiŽˆd<nf|rät d  |¡t¡‡fdd „dDƒ}t |ƒdkrt j|Žˆd<n |r,t d  |¡t¡d ˆd<x>dD](\}}|ˆkr:ˆ||dœˆd<Pq:Wdddœˆd<d S)zö Given the global time reference frame information, verify that each global time coordinate attribute will be given a valid value. Parameters ---------- global_info : dict Global time reference frame information. rÚscaleÚgpszãGlobal time scale (TIMESYS) has a FITS recognized time scale value "GPS". In Astropy, "GPS" is a time from epoch format which runs synchronously with TAI; GPS is approximately 19 s ahead of TAI. Hence, this format will be used.ÚtaiÚformatZlocala+Global time scale (TIMESYS) has a FITS recognized time scale value "LOCAL". However, the standard states that "LOCAL" should be tied to one of the existing scales because it is intrinsically unreliable and/or ill-defined. Astropy will thus use the default global time scale "UTC" instead of "LOCAL".ÚutcNzÀGlobal time scale (TIMESYS) should have a FITS recognized time scale value (got {!r}). The FITS standard states that the use of local time scales should be restricted to alternate coordinates.csg|]}|ˆkrˆ|‘qSr"r")Ú.0Úattr)Ú global_infor"r#ú {sz'_verify_global_info..)zOBSGEO-XzOBSGEO-YzOBSGEO-ZéÚunitÚlocationzaThe geocentric observatory location {} is not completely specified (X, Y, Z) and will be ignored.csg|]}|ˆkrˆ|‘qSr"r")r,r-)r.r"r#r/Œs)zOBSGEO-LzOBSGEO-BzOBSGEO-HzeThe geodetic observatory location {} is not completely specified (lon, lat, alt) and will be ignored.))rÚmjd)rÚjd)rÚfits)Úvalr*Úref_timerr3)r ÚgetÚlowerr ÚSCALESÚwarningsÚwarnrÚAssertionErrorr*ÚlenrZfrom_geocentricÚuÚmZ from_geodetic)r.Zobs_geoÚkeyZformat_r")r.r#Ú_verify_global_infoEsR      rBcCsö| dd¡}| dd¡}| dd¡}|dk r&t |dd…¡rVt |dd…¡rVdS| ¡tjkrz| ¡|d<d|d<nª|t ¡krœt||d<d|d<nˆ|d kr¾|d|d<|d|d<nf|d krêt   d   |¡t ¡d |d<d |d<n:|dkr t   d  |¡t ¡|d|d<|d|d<ndSn<|dk r:|t ksD|dk r^|d|d<|d|d<ndS|dk r¨|dkrž|d|d<|ddkr¦t   dt ¡nd|d<n>|ddkrÞ|d|d<|ddkræt   dt ¡nd|d<|d|d<dS)a  Given the column-specific time reference frame information, verify that each column-specific time coordinate attribute has a valid value. Return True if the coordinate column is time, or else return False. Parameters ---------- global_info : dict Global time reference frame information. column_info : dict Column-specific time reference frame override information. rNrréFr'r*ÚTIMEZGPSzÞTable column "{}" has a FITS recognized time scale value "GPS". In Astropy, "GPS" is a time from epoch format which runs synchronously with TAI; GPS runs ahead of TAI approximately by 19 s. Hence, this format will be used.r)r(ZLOCALaTable column "{}" has a FITS recognized time scale value "LOCAL". However, the standard states that "LOCAL" should be tied to one of the existing scales because it is intrinsically unreliable and/or ill-defined. Astropy will thus use the global time scale (TIMESYS) as the default.r&r2zÑTime column reference position "TRPOSn" value is "TOPOCENTER". However, the observatory position is not properly specified. The FITS standard does not support this and hence reference position will be ignored.rz´Time column reference position "TRPOSn" is not specified. The default value for it is "TOPOCENTER", but due to unspecified observatory position, reference position will be ignored.r7T)r8Ú TCTYP_RE_TYPErÚ TCTYP_RE_ALGOr9r r:r Úkeysr;r<r*rÚFITS_TIME_UNIT)Ú column_infor.r'r1r2r"r"r#Ú_verify_column_info¨sf    $                rJcCsx|jj ¡dkrt|jjtkrt|d|d|dddœ}|ddkrp|d |d <|d dkrpt d  |jj¡t¡|SdS) a/ Check if a column without corresponding time column keywords in the FITS header represents time or not. If yes, return the time column information needed for its conversion to Time. This is only applicable to the special-case where a column has the name 'TIME' and a time unit. rDr'r*r7N)r'r*r7r2rr&r2z\Time column "{}" reference position will be ignored due to unspecified observatory position.) ÚinfoÚnameÚupperr1rHr;r<r*r)Úcolr.rIr"r"r#Ú_get_info_if_time_columns    rOc CsàxÚ| ¡D]Î\}}| d¡rŽ||jkrØ|dkr2dn|d}y2d|krVt| d¡dƒnd}t|d||d}Wntk r€YnX||j|<q | d ¡r ||jkr yt|d |dd }Wntk rÌYnX||j|<q Wd S) a Convert the table metadata for time informational keywords to astropy Time. Parameters ---------- table : `~astropy.table.Table` The table whose time metadata is to be converted. global_info : dict Global time reference frame information. rr+r'Ú.éÿÿÿÿrr5)r*r'Ú precisionzMJD-r3)r*r'N)ÚitemsÚ startswithÚmetar>Úsplitr Ú ValueError)Útabler.rAÚvaluer'rRr"r"r#Ú_convert_global_time4s&      rZc Cs”yP|jjjdkrLtt|jjjdd…ƒddƒ}t|d|d||dd S|d d krjt|d |dd S|d ddkrì|d d dkrì|jddkrÌ|jdkrÌt|d|d|d|d d |ddSt||d|d d |ddSt|d d|d|d d |dd}|jddkrD|jdkrDt |d|dƒ}nt |ƒ}||St k rŽ}zt   d  ||jj¡t¡|Sd}~XYnXdS)a Convert time columns to astropy Time columns. Parameters ---------- col : `~astropy.table.Column` The time coordinate column to be converted to Time. column_info : dict Column-specific time reference frame override information. )ÚSÚUéNérr5r'r2)r*r'rRr2r*r()r*r2r7r6)r4r3rQr).r).r)r'r*r2z`The exception "{}" was encountered while trying to convert the time column "{}" to Astropy Time.)rKZdtypeZkindÚmaxÚintÚstrr ÚshapeÚndimr Ú Exceptionr;r<r*rLr)rNrIrRr7Z delta_timeÚerrr"r"r#Ú_convert_time_columnZs<             rfc CsPdddœ}ttƒ}|jdd}xˆ|jD]~\}}}|tkrL|||<| |¡q&t|ƒr„t d|¡  ¡\}} ||t | ƒ|<| |¡q&|dkr&t d|¡r&||||<q&Wt |ƒt ||ƒ|rþx>|  ¡D]2\} } t| |ƒrÈ|j| d } t|| | ƒ|| <qÈWxLt|jƒD]>\} } | d |kr t|| |ƒ} | r t|| | ƒ|| <q W|S) as Read FITS binary table time columns as `~astropy.time.Time`. This method reads the metadata associated with time coordinates, as stored in a FITS binary table header, converts time columns into `~astropy.time.Time` columns and reads global reference times as `~astropy.time.Time` instances. Parameters ---------- hdr : `~astropy.io.fits.header.Header` FITS Header table : `~astropy.table.Table` The table whose time columns are to be read as Time Returns ------- hdr : `~astropy.io.fits.header.Header` Modified FITS Header (time metadata removed) r%r&)rrT)Ústripz([A-Z]+)([0-9]+))zOBSGEO-XzOBSGEO-YzOBSGEO-Zz TTYPE[0-9]+r)rrÚcopyZcardsÚ TIME_KEYWORDSÚremover$rrÚgroupsr`rBrZrSrJÚcolnamesrfÚ enumeraterO) ÚhdrrXr.Z time_columnsZhcopyrArYÚcommentÚbaseÚidxrIZcolnamer"r"r#Ú fits_to_time—s:      rrc sú|jdd}tdd„t ¡Dƒƒ}ttƒ|jd<|jd}|j t ¡}d‰x¤|D]š}|j j ddkr€|  |j j t|jƒ¡qRt |j|jg¡}t |d |j¡}|  |j j t|d d ¡|j |j j ¡d }|j ¡||j j d <d ||j j d<t|dƒdkr&ˆdk rît d |j j ¡t¡qRd||j j d<|jtkr`t d |j j |j ¡¡t¡ˆdkrÒ|j ‰ˆj!d kr¸xRdD]0}|j"ttˆ|ƒ #t$j%¡ƒd | ¡¡dq‚Wn| &‡fdd„dDƒ¡qRˆ|j krRt'd ˆ|j ¡ƒ‚qRW||fS)as Replace Time columns in a Table with non-mixin columns containing each element as a vector of two doubles (jd1, jd2) and return a FITS header with appropriate time coordinate keywords. jd = jd1 + jd2 represents time in the Julian Date format with high-precision. Parameters ---------- table : `~astropy.table.Table` The table whose Time columns are to be replaced. Returns ------- table : `~astropy.table.Table` The table with replaced Time columns hdr : `~astropy.io.fits.header.Header` Header containing global time reference frame FITS keywords F)Z copy_datacSs&g|]\}}t||d|dd‘qS)rr)r!rYro)r)r,rAr6r"r"r#r/ûsz time_to_fits..Z__coordinate_columns__Nr5Zformatted_valuerr)r1rZ coord_typeZ coord_unitr2z‘Time Column "{}" has no specified location, but global Time Position is present, which will be the default for this column in FITS specification.r&Z time_ref_poszREarth Location "TOPOCENTER" for Time Column "{}" is incompatabile with scale "{}".)ÚxÚyÚzz OBSGEO-{})rLcs0g|](}td | ¡¡tˆ|ƒ tj¡d‘qS)z OBSGEO-{})r!rY)rr*rMÚgetattrÚto_valuer?r@)r,Údim)r2r"r#r/7sz†Multiple Time Columns with different geocentric observatory locations ({}, {}) encountered.This is not supported by the FITS standard.)(rhrÚGLOBAL_TIME_INFOrSrrrUÚcolumnsÚ isinstancer rKZserialize_methodZreplace_columnrLr rYÚnpZarrayZjd1Zjd2ZrollaxisrcrlÚindexr'rMrvr;r<r*rr r2ÚsizeZ add_columnrwr?r@ÚextendrW) rXZnewtablernZ coord_metaZ time_colsrNZjd12Únrxr")r2r#Ú time_to_fitsâsR          r),rr;Ú collectionsrrZnumpyr|ÚrrZastropyrr?Zastropy.coordinatesrZ astropy.tabler Z astropy.timer r Zastropy.time.corer Zastropy.time.formatsr Zastropy.utils.exceptionsrÚcompilerErFrHriZCOLUMN_TIME_KEYWORDSr*Újoinr r$ryrBrJrOrZrfrrrr"r"r"r#Ús:           cn&=K