HR-e?ddlZddlZddlmZddlmZddlmZddlm Z m Z Gdde j Z Gd d e Z Gd d e ZGdde ZGdde ZGdde ZdS)N) OrderedDict)Table) _set_locale)corecparserc\eZdZdZdZdZdZdZdZdZ ifdZ dZ dZ d Z d Zd Z dd Zd S) FastBasicaF This class is intended to handle the same format addressed by the ordinary :class:`Basic` writer, but it acts as a wrapper for underlying C code and is therefore much faster. Unlike the other ASCII readers and writers, this class is not very extensible and is restricted by optimization requirements. fast_basicz9Basic table with custom delimiter using the fast C engineTFc .|dd%|ddtdtj|}|tj||dd}|t |nd|_|dd|_|dd|_ |j t |j |_ t |d d |_ |dd|_ |d |j |j d znd }|d ||_ ||_ d |_d |_dS)N header_startrz2header_start cannot be set to None for this Reader delimiter commentz# # quotechar" data_startrT)get ValueErrorcopydeepcopyupdatepopstrr write_commentrrr rkwargsstrip_whitespace_linesstrip_whitespace_fields)selfdefault_kwargs user_kwargsrrdata_start_defaults :lib/python3.11/site-packages/astropy/io/ascii/fastbasic.py__init__zFastBasic.__init__sp   ~q 1 1 =22:QRR R ~.. dmK00111JJ{C00 +4+@Yd#ZZ 488zz)S11 < #t|,,DLVZZ S99::"JJ~q99)__ 43D3P$+a//VW  !**\3EFF &*#'+$$$c8|jdS)N)engine read_header)r s r$ _read_headerzFastBasic._read_header@s !!!!!r&c J|j,t|jdkrtjd|jtjd|j4|jdkr)t |tstjd|jdkrtjdt|jdkrtjdt|j dkrtjd d |j vrtjd d |j vrtjd d|j vrtjdd|j vrtjdd|j vs d|j vrtjd|j dd|_ |j dd}t |tsi}| dd| dd|_||j d<t!j||j|jf|j|j|j|j |j|jd|j |_|}|||\}}}ni}i}i}t1d5|j|||\}}dddn #1swxYwY|||} |jr|jj| jd<| S)z Read input data (file-like object, filename, list of strings, or single string) into a Table and return the result. Nrz-The C reader does not support a comment regexz1The C reader does not allow data_start to be NonerzYThe C reader does not allow header_start to be negative except for commented-header filesz5The C reader does not allow data_start to be negativez,The C reader only supports 1-char delimitersz5The C reader only supports a length-1 quote character convertersz(>%&UVV V _ $%C    )!A%%t%899&%= _q %G  A % %%&TUU U  A % %%G T[ ( (%N 4; & &%B DK ' '%C 4; & &%B  + +/@DK/O/O%&RSS S KOONEBBkoomT:: +t,, K$'''#.??3H%#P#P %0 M"o   '  (  n*Ln 0   k    ++--   &-< *GY GIJ    N N![--gy*MMND( N N N N N N N N N N N N N N NoodH--  # V=A[=UCH9 : s; K''K+.K+ct}|r||d<tj|j}t |||S)z:Actually make the output table give the data and comments.rJnamesrC)rr_deduplicate_namesr( get_namesr)r rIrJrCrNs r$rAzFastBasic.make_tablesP}}  ('D ' (=(=(?(?@@TT2222r&c|jp|j}|jrWgd}|D]P}t j|s't |dks|d|vs |d|vrtd|dQ|jr%t |dkrtd|dSdS)N)r,| 'rrz Column name z' does not meet strict name requirementsrz9Table format guessing requires at least two columns, got ) r(get_header_namesrPr2r _is_numberr9rguessing)r rNbadsnames r$r?zFastBasic.check_headers ,,..I$+2G2G2I2I   222D  OD))4yyA~~Aw$Bx4''$VtVVV( = SZZ1__SESS   __r&c4|||idS)z Use a fast Cython method to write table data to output, where output is a filename or file-like object. N_writer rDoutputs r$writezFastBasic.writes E62&&&&&r&ctj|d|j|j|j|jd}||||jtj |fi|}| |||dS)Nr)max_ndim)rrstrip_whitespacer) r_check_multidim_tablerrrrrrr FastWriterra)r rDr`r! header_output output_types write_kwargswriters r$r^zFastBasic._writes "515555 $ <)   N+++DK(((#E::\:: V]L99999r&N)TF)__name__ __module__ __qualname____doc__ _format_name _description_fastr6rYr2r%r*r@rAr?rar^r&r$r r s LNL EOHL&(!,!,!,!,F""" ZZZx333*'''OT::::::r&r ) metaclassc8eZdZdZdZdZdZdZfdZdZ xZ S)FastCsva  A faster version of the ordinary :class:`Csv` writer that uses the optimized C parsing engine. Note that this reader will append empty field values to the end of any row with not enough columns, while :class:`FastBasic` simply raises an error. fast_csvz4Comma-separated values table using the fast C engineTc Btjdddfi|dS)NrR)rrsuperr%r r __class__s r$r%zFastCsv.__init__s0st<<GGGGGGGr&cR|||dtjdfgidS)zs Override the default write method of `FastBasic` to output masked values as empty fields. fill_valuesN)r^rmaskedr_s r$raz FastCsv.writes/ E6MT["4E3F#GHHHHHr&) rkrlrmrnrorprqr6r%ra __classcell__r{s@r$rurussLIL EOHHHHHIIIIIIIr&ruc.eZdZdZdZdZdZfdZxZS)FastTabzl A faster version of the ordinary :class:`Tab` reader that uses the optimized C parsing engine. fast_tabz2Tab-separated values table using the fast C engineTc \tjddifi|d|_d|_dS)NrrTFryr%rrrzs r$r%zFastTab.__init__s=+t,77777&+#',$$$r&) rkrlrmrnrorprqr%rrs@r$rrsS LGL E---------r&rc4eZdZdZdZdZdZfdZdZxZ S) FastNoHeaderz This class uses the fast C engine to read tables with no header line. If the names parameter is unspecified, the columns will be autonamed with "col{}". fast_no_headerz3Basic table with no headers using the fast C engineTc Btjdddfi|dS)Nr)r rrxrzs r$r%zFastNoHeader.__init__ s0$a@@KKFKKKKKr&c8|||iddS)z Override the default writing behavior in `FastBasic` so that columns names are not included in output. Nrgr]r_s r$razFastNoHeader.writes$ E62T :::::r&) rkrlrmrnrorprqr%rarrs@r$rrsg $LHL ELLLLL;;;;;;;r&rc@eZdZdZdZdZdZfdZdZdZ dZ xZ S) r<z A faster version of the :class:`CommentedHeader` reader, which looks for column names in a commented line. ``header_start`` denotes the index of the header line among all commented lines and is 0 by default. fast_commented_headerz8Columns name in a commented line using the fast C engineTc Vtjifi|d|vr d|_dSdS)Nrr)ryr%rrzs r$r%zFastCommentedHeader.__init__"sB&&v&&& v % %DOOO & %r&c&t}|rE|j}|dkrt||z}|d|||dzdz|d<|ds|d=tj|j}t|||S)z Actually make the output table give the data and comments. This is slightly different from the base FastBasic method in the way comments are handled. rNrrJrM)rr r9rrOr(rPr)r rIrJrCidxrNs r$rAzFastCommentedHeader.make_table)s }}  %#CQww(mmc)'~q0CCD  # %$' (=(=(?(?@@TT2222r&c4|jj}g}|D]c}|}|rK|d|jkr:||ddt ||jdzkrndt ||jkrtj d|j ||jgd|j_|j |j |dS)Nrrznot enough commented lines) r(source splitlineslstriprappendr9r r CParserErrorsetup_tokenizerr))r tmpcommented_lineslines r$r*z FastCommentedHeader._read_header;sk NN$$  D;;==D Q4<//&&tABBx000''4+|s;|r'|d|jkr|}n|t d|j|gd|j_|j dd|j |j kr|j g|j|g|j dd|j |j dt t krtjdt|j t kr& fd |j D t!d Drtjd i}i}i}t#|j D]B\}} | d d krd||<d||<d||<3d||<d||<d||<C|j||||fS)Nr~rzRDB header requires 2 linesF) deduplicate filter_namesT)rzCRDB header mismatch between number of column names and column typescFg|]}|Srr)index).0n col_namestypess r$ z(FastRdb._read_header..s*PPP1U9??1--.PPPr&c3XK|]%}tjd|tj V&dS)z \d*(N|S)$N)rematch IGNORECASE)rxs r$ z'FastRdb._read_header..s5KK28L!R];;;KKKKKKr&z2RDB type definitions do not all match [num](N|S): rVsr)r(rrstriprrrrr r)rWrP set_namesr9rInconsistentTableErroranyziplower) r rline1line2rrFrGrHr[col_typerrs @@r$r*zFastRdb._read_headerfsk NN$$ < $,.N.N tzz||  a0@DL0P0P:;; ; ##UG,,,#$   EFFF ,,.. DK))++ + + K ! !" % % % ##UG,,, DuEEEK))++  E222 y>>SZZ ' '-U  t{$$&& ' '3u:: 5 5PPPPP 8M8M8O8OPPPE KKUKKK K K -LULL   !$+"7"7"9"95AA % %ND(|!!##s** ! "# $#$ 4  ! "# $#$ 4   ##C(((J//r&c8|||iddS)z Override the default writing behavior in `FastBasic` to output a line with column types after the column name line. T)rhNr]r_s r$raz FastRdb.writes$ E62D 99999r&) rkrlrmrnrorprqr%r*rarrs@r$rrVsr LEL E----- >0>0>0@:::::::r&r)rr collectionsr astropy.tablerastropy.utils.miscrr~rrMetaBaseReaderr rurrr<rrrr&r$rs ######******I:I:I:I:I:$-I:I:I:I:XIIIIIiIII0 - - - - -i - - - ;;;;;9;;;,<@<@<@<@<@)<@<@<@~U:U:U:U:U:iU:U:U:U:U:r&