U e{j?@sddlZddlZddlmZddlmZddlmZddlm Z m Z Gddde j d Z Gd d d e Z Gd d d e ZGddde ZGddde ZGddde ZdS)N) OrderedDict)Table) _set_locale)corecparserc@sfeZdZdZdZdZdZdZdZdZ ifddZ dd Z d d Z d d Z ddZddZdddZdS) 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. Z fast_basicz9Basic table with custom delimiter using the fast C engineTFcKs|dddk r(|dddkr(tdt|}|t||dd}|dk r^t|nd|_|dd|_|dd|_ |j dk rt|j |_ t|d d |_ |dd|_ |d |j dk r|j d nd }|d ||_ ||_ d |_d |_dS)N header_startrz2header_start cannot be set to None for this Reader delimiter commentz# # quotechar" data_startrT)get ValueErrorcopydeepcopyupdatepopstrr write_commentr rr rkwargsstrip_whitespace_linesstrip_whitespace_fields)selfdefault_kwargsZ user_kwargsrr Zdata_start_defaultrY/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/io/ascii/fastbasic.py__init__s.    zFastBasic.__init__cCs|jdS)N)engine read_header)rrrr _read_header@szFastBasic._read_headerc Cs6|jdk r&t|jdkr&tdn|jdkrr&rZ _is_numberr,rguessing)rr<Zbadsnamerrrr0s&    zFastBasic.check_headercCs|||idS)z Use a fast Cython method to write table data to output, where output is a filename or file-like object. N_writerr4outputrrrwriteszFastBasic.writecCsZtj|dd|j|j|j|jd}||||jtj |f|}| |||dS)Nr)Zmax_ndim)r rZstrip_whitespacer ) rZ_check_multidim_tabler rrrrrrZ FastWriterrK)rr4rJr header_output output_typesZ write_kwargswriterrrrrHs  zFastBasic._writeN)TF)__name__ __module__ __qualname____doc__ _format_name _description_fastr*rEr&r r#r1r2r0rKrHrrrrr s  #\ r) metaclasscs8eZdZdZdZdZdZdZfddZddZ Z 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. Zfast_csvz4Comma-separated values table using the fast C engineTc stjdddf|dS)Nr?)r r superr rr __class__rrr szFastCsv.__init__cCs|||dtjdfgidS)zs Override the default write method of `FastBasic` to output masked values as empty fields. Z fill_valuesN)rHrZmaskedrIrrrrKsz FastCsv.write) rOrPrQrRrSrTrUr*r rK __classcell__rrr[rrWs rWcs,eZdZdZdZdZdZfddZZS)FastTabzl A faster version of the ordinary :class:`Tab` reader that uses the optimized C parsing engine. Zfast_tabz2Tab-separated values table using the fast C engineTc s$tjddif|d|_d|_dS)Nr rAFrYr rrrZr[rrr szFastTab.__init__) rOrPrQrRrSrTrUr r^rrr[rr_s r_cs4eZdZdZdZdZdZfddZddZZ 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{}". Zfast_no_headerz3Basic table with no headers using the fast C engineTc stjdddf|dS)Nr)r rrXrZr[rrr szFastNoHeader.__init__cCs|j||idddS)z Override the default writing behavior in `FastBasic` so that columns names are not included in output. NrLrGrIrrrrKszFastNoHeader.write) rOrPrQrRrSrTrUr rKr^rrr[rras  racsDeZdZdZdZdZdZfddZddZd d Z d d Z Z 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. Zfast_commented_headerz8Columns name in a commented line using the fast C engineTc s"tjif|d|krd|_dS)Nrr)rYr rrZr[rrr "szFastCommentedHeader.__init__cCspt}|rR|j}|dkr$t||}|d|||dd|d<|dsR|d=t|j}t|||dS)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. rNrr9r;)rr r,rr=r!r>r)rr8r9r3idxr<rrrr2)s  zFastCommentedHeader.make_tablecCs|jj}g}|D]F}|}|r|d|jkr||ddt||jdkrq\qt||jkrtt d|j ||jgd|j_|j |j |dS)Nrrznot enough commented lines) r!source splitlineslstripr appendr,r rZ CParserErrorsetup_tokenizerr")rtmpZcommented_lineslinerrrr#;s   z FastCommentedHeader._read_headercCs|j||idddS)zr Override the default writing behavior in `FastBasic` so that column names are commented. r rbNrGrIrrrrKNszFastCommentedHeader.write) rOrPrQrRrSrTrUr r2r#rKr^rrr[rr.s r.cs<eZdZdZdZdZdZfddZddZd d Z Z S) FastRdbz A faster version of the :class:`Rdb` reader. This format is similar to tab-delimited, but it also contains a header line after the column name line denoting the type of each column (N for numeric, S for string). Zfast_rdbz0Tab-separated with a type definition header lineTc s&tjdddf|d|_d|_dS)NrA)r rFr`rZr[rrr aszFastRdb.__init__c s|jj}d}d}|D]N}|s@|r@|d|jkr@|}q|s|r|d|jkr|}qpqtd|j|gd|j_|jj ddd|j |j kr|j g|j|g|jj ddd|j |jj ddt t kr tdt |j t kr>fd d |j Dtd d Drbtd i}i}i}t|j D]N\}} | ddkrd||<d||<d||<nd||<d||<d||<q~|j||||fS)Nr]rzRDB header requires 2 linesF) deduplicateZ filter_namesT)rmzCRDB header mismatch between number of column names and column typescsg|]}|qSr)index).0nZ col_namestypesrr sz(FastRdb._read_header..css |]}td|tj VqdS)z \d*(N|S)$N)rematch IGNORECASE)roxrrr sz'FastRdb._read_header..z2RDB type definitions do not all match [num](N|S): rCsr)r!rdrestriprfr rrhr r"rDr>Z set_namesr,rZInconsistentTableErroranyziplower) rriline1line2rjr5r6r7rFZcol_typerrqrr#fsV       zFastRdb._read_headercCs|j||idddS)z Override the default writing behavior in `FastBasic` to output a line with column types after the column name line. T)rMNrGrIrrrrKsz FastRdb.write) rOrPrQrRrSrTrUr r#rKr^rrr[rrkVs @rk)rrt collectionsrZ astropy.tablerZastropy.utils.miscrr]rrZMetaBaseReaderrrWr_rar.rkrrrrs   M?