HR-eaAddlZddlZddlZddlmZddlZddlmZ 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-gdZ.dd/e$zdzZ0dZ1dZ2dZ3 d dZ4dZ5d!dZ6e j7dee4e j8dee6e j9dee2dS)"N)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)first) XTENSIONBITPIXNAXISNAXIS1NAXIS2PCOUNTGCOUNTTFIELDSTHEAP(|z)[0-9]+c:tjt|duSN)rematchCOLUMN_KEYWORD_REGEXP)keywords 7lib/python3.11/site-packages/astropy/io/fits/connect.pyis_column_keywordr.,s 8)7 3 34 ??cF|I|}|d}|||tkS|'|dSt |dttttfS)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.gzr) tellreadseekrlowerendswith isinstancerrrr)originfilepathfileobjargskwargspossigs r-is_fitsr?0sllnnll2 Sn$$  ~~(( G    d1g; J K KKr/c |jdd}|jdd}n#ttf$r|cYSwxYwd}g}|jd|dz|D]K}|r|d|ddz|d<n||ddt |d k}L|jd||dz=|jds|jd=tj|}|d d |jd <|d D]3}d D].}||vr(t||dj|||/4tj |}|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 descriptionr name) r index ValueErrorKeyErrorrlenget_header_from_yamlsetattrinfor _construct_mixins_from_columns) tbli0i1continuation_linelineslinerRcolattrs r-_decode_mixinsr\Js  Xj ! ' '(N O O Xj ! ' '(L M M  !   E$R!Vb[1,,  $b D"I-E"II LLcrc # # #IIO R"q&[) 8J ! HZ  $U + +D*.f6N)OCH %&J@@+ @ @Ds{{CK(-tSY??? @  23 7 7C JsAAAAFTwarncf t|trt}t|D],\}} t| tt t fr| ||<-t|dkrq|:tj dt|dtt|}| |}||vr ||} nJtd|t|dkr|d} | |} | t|krd|d} n | |vrd|} n#t$r d|d} YnwxYw| -tj | d t|d t|t|} ntd t|tt t fr|} nX|rd }t!||| } t#| ||||| S#| wxYw| j}g}|jD]r}d x}}|jjr|jjdjn |jj}|j||j|jk}d}nc|r:t5|t6jr t7j||j}n'|r%t5|t6jr |jdk}|st7j |r$tC||j|j|d }n"tE||j|jd }|j#!tIj%|j#d||_#|j&tO|j&|_(|)|ttU|d }| j+}|rddl,m-}|||}|j.D]\}}}|dvr>|dkrd}||j/vr!|j/|)|<|g|j/|<H||j/vrYt|j/|t`r!|j/|)||j/||g|j/|<tc|s |tdvr||j/|<tg|}|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_invalidrTr/)datarKmaskcopy)rfrKrhfits)format parse_strict)rh) fits_to_time)COMMENTHISTORYrmrA)4r7rdict enumeraterrrrOwarningsr]rrindex_ofrMrNr fits_openread_table_fitscloserfcolumnsdtypesubdtypetypenullrK issubclassnpinexactisnan characterarrayanyrrunituUnitdisprrjrrheaderfitstimerlcardsr listr.REMOVE_KEYWORDSr\)inputrbrcrar`rdretablesihduhdu_itemtablemsghdihdulistrfrvrZmaskedrgcoltypecolumnthdrrlkeyvaluecomments r-rtrtyse~%!!F'.. ( (ND((X{I$FGG ('t v;;??{ 4#(==444'  Fmm..%%Cf}}s  !?#!?!?@@@ [[A  G..--Cc%jj((>s>>>F**FFF  ;;;:3:::CCC;?MBB %f BBB2  5==)EE-.. . EHk9= > >  !!LE6HQWXXX "-"3)  MMOOOOGMMOOOO :DG|$$03 0BV#)$Q',,  8 >SX-DFF  $j"*== $8DN++DD  $j",?? $9#D  LRVD\\ L!#(^#(EFFchchUKKKF 8 &6GFK 8 5ch??FMv gE"""A ,C#******l3"""y  UG ( ( (i af}}s ""5))))$gs AF]]!&+t,, 3s ""5)))) vc{E2s s # # so'='=  AF3KK qA Hs=D66E E HHc @ td|D}td5tj|t f}dddn #1swxYwY||ur|s|S||ur+t |j}t|j |d}d}|j |i|j } ||j|i|_tj |}||_n #||_wxYw|j|=d|jvr g|jd<|jd d |D] t d krd g}nmtt!d t d zd } fd t#|dd|ddD}|ddd|d<|jd||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. c3NK|]tfddDV dS)c3LK|]}tj|ddifvVdSr()getattrrR).0r[rZs r- z+_encode_mixins..._sN   CHdD ) )$ ;      r/rIN)r)rrZs @r-rz!_encode_mixins..^si      /     r/ri)exclude_classesNF)r rhrGrArBrrEc0g|]\}}||dzS)\)rrUrVrYs r- z"_encode_mixins..s*OOOFBT"R%[4'OOOr/rDrrC)ritercolsr r represent_mixins_as_columnsr rr rrv setdefaultrhget_yaml_from_tablerrOrrangezipextend) rT info_lost encode_tbl meta_copyser_col tbl_meta_copymeta_yaml_linesrXidxsrYs @r-_encode_mixinsrVs <<>> I f % %YY:3QUPWXXX YYYYYYYYYYYYYYY S SSX&& 3;YUCCC 'GOw+++O((**M("JOG$<= 2:>>' - '''' ((&(  #OJ&&'MNNN22 t99>>DEEaTR4455DOOOOc$ss)T!""X6N6NOOOEb #2#E"I #**51111OJ&&'KLLL sA%%A),A)(D Dct|}t|d}t|trYtj|r:|rt j|n#|s!ttj ||rt||j |j ddS||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. append : bool Whether to append the table to an existing file T)r`F)verifyN)rrr7strospathexistsremoveOSErrorrrj fits_appendrfrwriteto)routput overwriter table_hdus r-write_table_fitsrs 5 ! !EUt<<27>> #9>  > If     >-4V<<== = "FINI,rsx ......FFFFFFFFFFFFFF888888RRRRRRRR222222777777777777######<<<<<<<<%%%%%%''''''......   chh}555 A@@@LLL4,,,b  Z Z Z Z zHHHV""""D FE?;;; FE+;<<< w77777r/