U e{jA @sddlZddlZddlZddlmZddlZddlmZ ddl m Z ddl mZmZmZmZmZddlmZddlmZddlmZmZdd lmZd d lmZmZmZm Z d d lm!Z"d d l#m$Z$m%Z%d dl&m'Z'd dl(m)Z)d dl(m*Z+d dl,m-Z-dddddddddg Z.dd/e$dZ0ddZ1d d!Z2d"d#Z3d.d'd(Z4d)d*Z5d/d+d,Z6e 7d-ee4e 8d-ee6e 9d-ee2dS)0N)deepcopy)units)registry)Column MaskedColumnTablemeta serialize)Time)serialize_context_as)AstropyDeprecationWarningAstropyUserWarning)NOT_OVERWRITING_MSG) BinTableHDU GroupsHDUHDUListTableHDU)append) KEYWORD_NAMES_fortran_to_python_format) table_to_hdu)FITS_SIGNATURE)fitsopen)firstZXTENSIONZBITPIXZNAXISZNAXIS1ZNAXIS2ZPCOUNTZGCOUNTZTFIELDSZTHEAP(|z)[0-9]+cCstt|dk SN)rematchCOLUMN_KEYWORD_REGEXP)keywordr"V/var/www/html/bwave/python/venv/lib/python3.8/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 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) tellreadseekrlowerendswith isinstancerrrr)originfilepathfileobjargskwargspossigr"r"r#is_fits0s  r3c Cs,z$|jdd}|jdd}Wnttfk r@|YSXd}g}|jd|d|D]@}|r|d|dd|d<n||ddt|d k}q`|jd||d=|jds|jd=t|}|d d |jd <|d D]0}d D]&}||krt||dj|||qqt |}|S)zDecode a Table ``tbl`` that has astropy Columns + appropriate meta-data into the corresponding table with mixin columns (as appropriate). comments$--BEGIN-ASTROPY-SERIALIZED-COLUMNS--"--END-ASTROPY-SERIALIZED-COLUMNS--FrNFGr__serialized_columns__datatype descriptionrname) rindex ValueErrorKeyErrorrlenZget_header_from_yamlsetattrinfor Z_construct_mixins_from_columns) tbli0i1Zcontinuation_linelineslinerDcolattrr"r"r#_decode_mixinsNs.     rLFTwarncCst|trVt}t|D] \}} t| tttfr| ||<qt|dkr|dkrpt dt |dt t |}| |}||kr||} nt d|nt|dkrL|dk r>d} z| |} Wn tk rd|d} Yn2X| t|krd|d} n| |krd|} | dk r>t | d t |d t|t |} nt d nRt|tttfrn|} n:|rxd }t|||d } zt| ||||dWS| X| j}g}|jD]}d }}|jjr|jjdjn|jj}|jdk r||j|jk}d}nD|r.t|tjr.t||j}n|rLt|tjrL|jdk}|s^t |rxt!||j|j|d d}nt"||j|jd d}|j#dk rt$j%|j#d|d|_#|j&dk rt'|j&|_(|)|qt*|d d}| j+}|rddl,m-}|||}|j.D]\}}}|dkrV|dkr*d}||j/krH|j/|)|n |g|j/|<nf||j/krt|j/|t0r|j/|)|n|j/||g|j/|<n t1|s|t2krn ||j/|<qt3|}|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 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`. When set to `True` then ``mask_invalid`` is set to `False` since the masking would cause loading the full data array. 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`. unit_parse_strict : str, optional Behaviour when encountering invalid column units in the FITS header. Default is "warn", which will emit a ``UnitsWarning`` and create a :class:`~astropy.units.core.UnrecognizedUnit`. Values are the ones allowed by the ``parse_strict`` argument of :class:`~astropy.units.core.Unit`: ``raise``, ``warn`` and ``silent``. mask_invalid : bool, optional By default the code masks NaNs in float columns and empty strings in string columns. Set this parameter to `False` to avoid the performance penalty of doing this masking step. The masking is always deactivated when using ``memmap=True`` (see above). rNz^hdu= was not specified but multiple tables are present, reading in first available table (hdu=)zNo table found in hdu=zSpecified hdu=z not foundz No table found in specified hdu=z(, reading in first available table (hdu=z;) instead. This will result in an error in future versions!zNo table foundF)character_as_bytesmemmap)hduastropy_nativeunit_parse_strict mask_invalidrT)datar>maskcopy)rVr>rXfits)formatZ parse_strict)rX) fits_to_time)COMMENTZHISTORYr\r4)4r+rdict enumeraterrrrBwarningsrMrr Zindex_ofr@rAr fits_opencloseread_table_fitsrVcolumnsZdtypeZsubdtypetypenullr> issubclassnpZinexactisnan characterarrayanyrrunituZUnitZdisprrZrrheaderZfitstimer[Zcardsrlistr$REMOVE_KEYWORDSrL)inputrQrRrPrOrSrTZtablesZihduZhdu_itemtablemsgZhdiZhdulistrVrcrJZmaskedrWZcoltypecolumnthdrr[keyvaluecommentr"r"r#rb}s@                          rbc srtdd|D}tdtj|tfd}W5QRX||krJ|sJ|S||krlt|j}t|j |dd}d}|j |i|j }z||j|i|_t |}W5||_X|j|=d|jkrg|jd<|jd d |D]|td krd g}nTttd td d }fd dt|dd|ddD}|ddd|d<|jd|q|jd d|S)zEncode a Table ``tbl`` that may have mixin columns to a Table with only astropy Columns + appropriate meta-data to allow subsequent decoding. c3s$|]tfdddDVqdS)c3s$|]}tj|ddifkVqdSr)getattrrD).0rKrJr"r# dsz+_encode_mixins...r<N)rk)r{r"r|r#r}cs z!_encode_mixins..rY)Zexclude_classesF)rrXr:r4r5rr8cs g|]\}}||dqS)\r")r{rFrGrIr"r# sz"_encode_mixins..Nr7rr6)rkZitercolsr r Zrepresent_mixins_as_columnsr rrrrc setdefaultrXZget_yaml_from_tablerrBrorangezipextend) rEZ info_lostZ encode_tblZ meta_copyZser_colZ tbl_meta_copyZmeta_yaml_linesrHZidxsr"rr#_encode_mixins[s<       (rcCstt|}t|dd}t|trLtj|rL|r:t|n|sLtt ||rft ||j |j ddn ||dS)aZ 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. append : bool Whether to append the table to an existing file T)rOF)verifyN)rrr+strospathexistsremoveOSErrorrrZ fits_appendrVrnZwriteto)rqoutput overwriterZ table_hdur"r"r#write_table_fitss  rrY)NFFTrMT)FF):rrr_rXrnumpyrgZastropyrrmZ astropy.iorZ io_registryZ astropy.tablerrrrr Z astropy.timer Zastropy.utils.data_infor Zastropy.utils.exceptionsr r Zastropy.utils.miscrr~rrrrrrrtrrZ conveniencerZ hdu.hdulistrrr`utilrrpjoinr r$r3rLrbrrZregister_readerZregister_writerZregister_identifierr"r"r"r#sZ            1 _K #