ó šßÈ[c@sddlZddlmZddlmZddlmZddlmZddlm Z dd l m Z dd l m Z ejejƒd efd „ƒYƒZd efd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdS(iÿÿÿÿN(t OrderedDicti(tcorei(tsix(tTable(tcparser(tzip(t set_localet FastBasiccBsqeZdZdZdZeZeZeZ eZ id„Z d„Z d„Z d„Zd„Zeed„ZRS( sF 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. t fast_basics9Basic table with custom delimiter using the fast C enginecKsk|jddƒdk r?|jddƒdkr?tdƒ‚n|jƒ}|j|ƒ|jddƒ}|dk r‚t|ƒnd|_|jddƒ|_|jddƒ|_ |j dk rÙt|j ƒ|_ nt|jd d ƒƒ|_ |jddƒ|_ |jd |j dk r.|j d nd ƒ}|jd |ƒ|_ ||_ t|_t|_dS( Nt header_startis2header_start cannot be set to None for this Readert delimitert tcomments# t#t quotechart"t data_starti(tgettNonet ValueErrortcopytupdatetpoptstrR t write_commentR RR RtkwargstTruetstrip_whitespace_lineststrip_whitespace_fields(tselftdefault_kwargst user_kwargsRR tdata_start_default((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyt__init__s&  ! %  cCs|jjƒdS(N(tenginet read_header(R((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyt _read_header9sc Csä|jdk r6t|jƒdkr6tjdƒ‚n„|jdkrWtjdƒ‚nc|jdk r—|jdkr—t|tƒ r—tjdƒ‚n#|jdkr¸tjdƒ‚nt|j ƒdkrßtjdƒ‚nÛt|j ƒdkrtjdƒ‚n´d |j kr'tjd ƒ‚n“d |j krHtjd ƒ‚nrd |j kritjdƒ‚nQd|j krŠtjdƒ‚n0d|j ks¨d|j krºtjdƒ‚n|j j dt ƒ|_tj||j|jd|j d|jd|jd|j d|jd|j|j |_|jƒ}|jƒ|dk r`|\}}}ni}i}i}tdƒ#|jj|||ƒ\}}WdQXtƒ}|r¿||ds\$          $  cCsÚ|jjƒp|jjƒ}|jr¬ddddddg}xj|D]_}tj|ƒst|ƒdks|d|ks|d|krFtd j|ƒƒ‚qFqFWn|j rÖt|ƒd krÖtd ƒ‚ndS( NR t,t|s t'Riiÿÿÿÿs8Column name {0!r} does not meet strict name requirementsis2Strict name guessing requires at least two columns( R"tget_header_namesR:R+Rt _is_numberR1Rtformattguessing(RR/tbadstname((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR7ys   cCs|j||iƒdS(s Use a fast Cython method to write table data to output, where output is a filename or file-like object. N(t_write(RR;toutput((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pytwrite‰scCsti|jd6|jd6|jd6|jd6}|j|ƒ|j|jƒtj||}|j|||ƒdS(NR Rtstrip_whitespaceR ( R RRRRRRt FastWriterRL(RR;RKRt header_outputt output_typest write_kwargstwriter((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRJs     (t__name__t __module__t__doc__t _format_namet _descriptionRt_fastR5R,RGR+R!R$R8R7RLRJ(((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRs   ;  tFastCsvcBs8eZdZdZdZeZeZd„Zd„Z RS(s  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. tfast_csvs4Comma-separated values table using the fast C enginecKs+tt|ƒjidd6dd6|dS(NRAR R (tsuperRYR!R(RR((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR!«scCs*|j||itjdfgd6ƒdS(ss Override the default write method of `FastBasic` to output masked values as empty fields. tt fill_valuesN(RJRtmasked(RR;RK((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRL®s( RSRTRURVRWRRXR,R!RL(((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRYŸs tFastTabcBs)eZdZdZdZeZd„ZRS(sl A faster version of the ordinary :class:`Tab` reader that uses the optimized C parsing engine. tfast_tabs2Tab-separated values table using the fast C enginecKs6tt|ƒjidd6|t|_t|_dS(Ns R (R[R_R!R5RR(RR((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR!¿s  (RSRTRURVRWRRXR!(((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR_¶s t FastNoHeadercBs2eZdZdZdZeZd„Zd„ZRS(sª 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{}". tfast_no_headers3Basic table with no headers using the fast C enginecKs+tt|ƒjidd6dd6|dS(NR iR(R[RaR!R(RR((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR!ÏscCs|j||iddƒdS(s€ Override the default writing behavior in `FastBasic` so that columns names are not included in output. RON(RJR(RR;RK((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRLÒs( RSRTRURVRWRRXR!RL(((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRaÅs  R4cBsDeZdZdZdZeZd„Zd„Zd„Z d„Z RS(sá 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. tfast_commented_headers8Columns name in a commented line using the fast C enginecKs5tt|ƒji|d|kr1d|_ndS(NRi(R[R4R!R(RR((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR!äs cCs¢tt|ƒj|ƒ}d|jkrž|j}|dkrVt|jdƒ|}n|jd| |jd|d|jd<|jdsž|jd=qžn|S(s Read input data (file-like object, filename, list of strings, or single string) into a Table and return the result. R.ii(R[R4R8R0R R1(RR;touttidx((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR8ës  + cCsí|jj}g}xj|jƒD]\}|jƒ}|r|d|jkr|j|dƒt|ƒ|jdkr{Pq{qqWt|ƒ|jkr¦tj dƒ‚n|jj ||jgƒd|j_|jj ƒ|jj |ƒdS(Niisnot enough commented lines( R"tsourcet splitlinestlstripR tappendR1R Rt CParserErrortsetup_tokenizerR#(Rttmptcommented_linestline((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR$þs     cCs|j||iddƒdS(sr Override the default writing behavior in `FastBasic` so that column names are commented. ROR N(RJ(RR;RK((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRLs( RSRTRURVRWRRXR!R8R$RL(((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR4Ús   tFastRdbcBs;eZdZdZdZeZd„Zd„Zd„Z RS(så 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). tfast_rdbs0Tab-separated with a type definition header linecKs=tt|ƒjidd6dd6|t|_t|_dS(Ns R iR(R[RoR!R5RR(RR((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR!#s' c Cs|jj}d}d}x‹|jƒD]q}| r`|jƒr`|jƒd|jkr`|}q%| r%|jƒr%|jƒd|jkr%|}Pq%q%Wtdƒ‚|jj|gƒd|j_|jj ƒ|jj ƒ}|jj|gƒ|jj gƒ|jj ƒt |jj ƒƒt |ƒkrAtdƒ‚nt d„|Dƒƒrotdj|ƒƒ‚ni}i}i}x{t|jj ƒ|ƒD]a\} } | djƒdkrÝd|| Css5RDB type definitions do not all match [num](N|S): {0}iÿÿÿÿtsi(R"RfRgtstripRhR RRkR R#R:t set_namesR1tanyRFRtlower( RRltline1tline2RnttypesR=R>R?RItcol_type((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyR$(sF , ,    ! %     cCs|j||idtƒdS(s Override the default writing behavior in `FastBasic` to output a line with column types after the column name line. RPN(RJR(RR;RK((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRLXs( RSRTRURVRWRRXR!R$RL(((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyRos  0(Rqt collectionsRR\RtexternRR;RRtextern.six.movesRtutilsRt add_metaclasstMetaBaseReadertobjectRRYR_RaR4Ro(((s9lib/python2.7/site-packages/astropy/io/ascii/fastbasic.pyts ?