B q\: @szddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z mZmZmZddlmZddlmZddlmZdd lmZmZd d lmZmZmZmZd d lm Z m!Z!m"Z"d d l#m$Z$d dl%m&Z'd dl(m)Z)d dl*m+Z+m,Z,dZ-dddddddddg Z.dd/e dZ0ddZ1d d!Z2d"d#Z3d-d&d'Z4d(d)Z5d.d*d+Z6e7d,e e4e8d,e e6e9d,e e2dS)/N) OrderedDict)registry)units)Table serializemetaColumn MaskedColumn)has_info_class)Time)AstropyUserWarning) MixinInfoserialize_context_as)HDUListTableHDU BinTableHDU GroupsHDU) KEYWORD_NAMESASCII_DEFAULT_WIDTHS_fortran_to_python_format) table_to_hdu)fitsopen)first) VerifyError VerifyWarningsSIMPLE = TZXTENSIONZBITPIXZNAXISZNAXIS1ZNAXIS2ZPCOUNTZGCOUNTZTFIELDSZTHEAP(|z)[0-9]+cCstt|dk S)N)rematchCOLUMN_KEYWORD_REGEXP)keywordr"6lib/python3.7/site-packages/astropy/io/fits/connect.pyis_column_keyword%sr$cOsj|dk r,|}|d}|||tkS|dk rH|drfdSnt|dttt t frbdSdSdS)a$ Determine whether `origin` is a FITS file. Parameters ---------- origin : str or readable file-like object Path or file object containing a potential FITS file. Returns ------- is_fits : bool Returns `True` if the given file is a FITS file. N)z.fitsz.fits.gzz.fitz.fit.gzz.ftsz.fts.gzTrF) tellreadseekFITS_SIGNATURElowerendswith isinstancerrrr)originfilepathZfileobjargskwargsposZsigr"r"r#is_fits)s  r2c Cs4y$|jdd}|jdd}Wnttfk r<|SXd}g}xZ|jd|d|D]@}|r|d|dd|d<n||ddt|d k}q^W|jd||d=|jds|jd=t|}|d d |jd <x@|d D]4}x.d D]&}||krt||dj|||qWqWt |}|S)zDecode a Table ``tbl`` that has astropy Columns + appropriate meta-data into the corresponding table with mixin columns (as appropriate). commentsz$--BEGIN-ASTROPY-SERIALIZED-COLUMNS--z"--END-ASTROPY-SERIALIZED-COLUMNS--FrNFGr__serialized_columns__Zdatatype) descriptionrname) rindex ValueErrorKeyErrorappendlenZget_header_from_yamlsetattrinforZ_construct_mixins_from_columns) tbli0i1Zcontinuation_linelinesliner@colattrr"r"r#_decode_mixinsFs.   " rHFTcCst|trt}x,t|D] \}}t|tttfr|||<qWt|dkr|dkrpt d t |t t |}| |}||kr||}qtd |n"t|dkr|t |}ntdn>t|tttfr|}n(t|||d} zt| ||dS| Xtdd |jD} |j} g} x| jD]} | rpt| | j| jd d }| jdk r|| jd |j|j| jk<nt| | j| jd d }| jdk rtj| jd dd|_| jdk rt| j|_ | |q$Wt!| | d d}|j"}|rddl#m$}|||}x|j%D]\}}}|dkrT|dkr(d}||j&krF|j&| |n |g|j&|<nf||j&krt|j&|t'r|j&| |n|j&||g|j&|<n t(|s|t)krn ||j&|<qWt*|}|S)a| Read a Table object from an FITS file If the ``astropy_native`` argument is ``True``, then input FITS columns which are representations of an astropy core object will be converted to that class and stored in the ``Table`` as "mixin columns". Currently this is limited to FITS columns which adhere to the FITS Time standard, in which case they will be converted to a `~astropy.time.Time` column in the output table. Parameters ---------- input : str or file-like object or compatible `astropy.io.fits` HDU object If a string, the filename to read the table from. If a file object, or a compatible HDU object, the object to extract the table from. The following `astropy.io.fits` HDU objects can be used as input: - :class:`~astropy.io.fits.hdu.table.TableHDU` - :class:`~astropy.io.fits.hdu.table.BinTableHDU` - :class:`~astropy.io.fits.hdu.table.GroupsHDU` - :class:`~astropy.io.fits.hdu.hdulist.HDUList` hdu : int or str, optional The HDU to read the table from. astropy_native : bool, optional Read in FITS columns as native astropy objects where possible instead of standard Table Column objects. Default is False. memmap : bool, optional Whether to use memory mapping, which accesses data on disk as needed. If you are only accessing part of the data, this is often more efficient. If you want to access all the values in the table, and you are able to fit the table in memory, you may be better off leaving memory mapping off. However, if your table would not fit in memory, you should set this to `True`. character_as_bytes : bool, optional If `True`, string columns are stored as Numpy byte arrays (dtype ``S``) and are converted on-the-fly to unicode strings when accessing individual elements. If you need to use Numpy unicode arrays (dtype ``U``) internally, you should set this to `False`, but note that this will use more memory. If set to `False`, string columns will not be memory-mapped even if ``memmap`` is `True`. rNzbhdu= was not specified but multiple tables are present, reading in first available table (hdu={0})zNo table found in hdu={0}zNo table found)character_as_bytesmemmap)hduastropy_nativecss|]}|jdk VqdS)N)null).0rFr"r"r# sz"read_table_fits..F)datar9copyTfitsZsilent)formatZ parse_strict)maskedrQ) fits_to_time)COMMENTZHISTORYrVr3)+r,rr enumeraterrrr>warningswarnrSrr Zindex_ofr; fits_openread_table_fitscloseanycolumnsrPr r9rMZset_fill_valuemaskrZunituZUnitZdisprr=rheaderZfitstimerUZcardsrlistr$REMOVE_KEYWORDSrH)inputrKrLrJrIZtablesZihduZhdu_itemtableZhdulistrTrPr^rFcolumntZhdrrUkeyvaluecommentr"r"r#r[ts|+                    r[c stdd|D}y ddl}Wnltk rxV|D]6}t|tr<|jtjt fkr..)r8rN)r])rNr")rFr#rO"sz!_encode_mixins..rNzBcannot write type {} column '{}' to FITS without PyYAML installed.ztable contains column(s) with defined 'format', 'description', or 'meta' info attributes. These will be dropped unless you install PyYAML.rR)Zexclude_classesr7r3z$--BEGIN-ASTROPY-SERIALIZED-COLUMNS--r5cs g|]\}}||dqS)\r")rNrBrC)rEr"r# jsz"_encode_mixins..r4rz"--END-ASTROPY-SERIALIZED-COLUMNS--)r]Zitercolsyaml ImportErrorr r __class__r`ZQuantityr TypeErrorrS__name__r@r9rXrYr rrZ_represent_mixins_as_columnsr setdefaultrQZget_yaml_from_tabler=r>rbrangezipextend) rAZ info_lostrorFZ encode_tblZser_colZ tbl_meta_copyZmeta_yaml_linesrDZidxsr")rEr#_encode_mixinssN          (rxcCsVt|}t|dd}t|trHtj|rH|r:t|ntd || |dS)a Write a Table object to a FITS file Parameters ---------- input : Table The table to write out. output : str The filename to write the table to. overwrite : bool Whether to overwrite any existing file without warning. T)rIzFile exists: {0}N) rxrr,strospathexistsremoveOSErrorrSZwriteto)rdoutputZ overwriteZ table_hdur"r"r#write_table_fitsss  rrR)NFFT)F):rzrrX collectionsrZ astropy.iorZ io_registryZastropyrr`Z astropy.tablerrrrr Zastropy.table.tabler Z astropy.timer Zastropy.utils.exceptionsr Zastropy.utils.data_infor rrlrrrrrfrrrZ conveniencerZ hdu.hdulistrrZutilrZverifyrrr)rcjoinr r$r2rHr[rxrZregister_readerZregister_writerZregister_identifierr"r"r"r#s>           . &Y