ó šßÈ[c@`s†dZddlmZmZmZddlmZmZddlm Z ddl m Z m Z ddlm Z d e j fd „ƒYZd e fd „ƒYZd e jfd„ƒYZde jfd„ƒYZde jfd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd „ƒYZd!S("sßAn extensible ASCII table reader and writer. fixedwidth.py: Read or write a table with fixed width columns. :Copyright: Smithsonian Astrophysical Observatory (2011) :Author: Tom Aldcroft (aldcroft@head.cfa.harvard.edu) i(tabsolute_importtdivisiontprint_functioni(tziptrangei(tcore(tInconsistentTableErrortDefaultSplitter(tbasictFixedWidthSplittercB`s2eZdZdZeZdZd„Zd„ZRS(s™ Split line based on fixed start and end positions for each ``col`` in ``self.cols``. This class requires that the Header class will have defined ``col.start`` and ``col.end`` for each column. The reference to the ``header.cols`` gets put in the splitter object by the base Reader.read() function just in time for splitting data lines by a ``data`` object. Note that the ``start`` and ``end`` positions are defined in the pythonic style so line[start:end] is the desired substring for a column. This splitter class does not have a hook for ``process_lines`` since that is generally not useful for fixed-width input. tt|cc`spxi|D]a}g|jD]}||j|j!^q}|jrcg|D]}|j|ƒ^qFVq|VqWdS(N(tcolststarttendt process_val(tselftlinestlinetxtvals((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyt__call__(s  ) $c C`s§|jp d}|jpd}|||}|jrL||}||}n d}d}gt||ƒD]$\}} d| t|ƒ|^qh}||j|ƒ|S(NR t (t delimiter_padt delimitertbookendRtlentjoin( RRtwidthstpadRt padded_delimt bookend_leftt bookend_righttvaltwidth((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyR0s   :( t__name__t __module__t__doc__RtFalseRRRR(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyR s  tFixedWidthHeaderSplittercB`seZdZdZRS(s$Splitter class that splits on ``|``.R (R#R$R%R(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyR'>stFixedWidthHeadercB`sJeZdZeZdZedƒZd„Z d„Z d„Z d„Z RS( s* Fixed width table header reader. s`~!#$%^&*-_+=\|":t'cC`sFx?t|j|ƒƒD]\}}||krPqqWtdƒ‚|S(NsNo header line found in table(t enumeratet process_linesR(RRtindextiR((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pytget_lineMs "  cC`s[tj|j|j|ƒƒ}tj|j|j|ƒƒ}|dkrä|dk rctdƒ‚n|jj|ƒ}|sŠtdƒ‚n|j |dƒ\}}}gt dt |ƒdƒD]}|j j |ƒ^qÀ|_n/|dk rà|j||ƒ} t t| ƒt|jjdgƒƒdkr?tdƒ‚n|jjt|jjdgƒƒ} t| ƒj| ƒstdj | ƒƒ‚n|j | ƒ\}|_} g| D]"} | dk rÎ| dnd^q²|_n|j||ƒ} |j | ƒ\}}}||_|jƒx7t|jƒD]&\}} ||| _||| _q-WdS( s: Initialize the header Column objects from the table ``lines``. Based on the previously set Header attributes find or create the column names. Sets ``self.cols`` with the list of Columns. Parameters ---------- lines : list List of table lines s:Cannot set position_line without also setting header_starts7No data lines found so cannot autogenerate column namesiiRs]Position line should only contain delimiters and one other character, e.g. "--- ------- ---".s/Characters in position line must be part of {0}N(Rt_get_line_indext start_lineR+t position_linetNonet ValueErrortdataRtget_fixedwidth_paramsRRt auto_formattformattnamesR.tsettsplitterRtset_of_position_line_characterstuniontissubsett col_startstcol_endst_set_cols_from_namesR*R R R(RRR0R1t data_linesRtstartstendsR-RtcharsetR?Rtcol((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pytget_colsUs8   ; .$5   c C`sp|jdk r¾|jdk r¾t|jƒ}g|jD]"}|dk rS|dnd^q7}t|ƒt|ƒkr†tdƒ‚ngt||ƒD]\}}|||!jƒ^q–}n¥|jdkrË|jdkrË|j|j j ƒ}dg}g}xW|D]O}|rF|j |dt|ƒƒ|j |ddƒq|dcd7}|||fS(s¾ Split ``line`` on the delimiter and determine column values and column start and end positions. This might include null columns with zero length (e.g. for ``header row = "| col1 || col2 | col3 |"`` or ``header2_row = "----- ------- -----"``). The null columns are stripped out. Returns the values between delimiters and the corresponding start and end positions. Parameters ---------- line : str Input line Returns ------- vals : list List of values. starts : list List of starting indices. ends : list List of ending indices. is=Fixed width col_starts and col_ends must have the same lengthiiÿÿÿÿs Error parsing fixed width headerN( R>R2R?tlistRR3RtstriptsplitR:RtappendR( RRRBRRCR RRR!((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyR5 s428   %0 5cC`sdS(N((RR((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pytwriteÞsNs`~!#$%^&*-_+=\|":'( R#R$R%R'tsplitter_classR2R1R9R;R.RFR5RK(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyR(Cs   K >tFixedWidthDatacB`seZdZeZd„ZRS(s! Base table data reader. c C`sßg}|jƒ}x!t|ŒD]}|j|ƒqWxt|jƒD]n\}}tg|D]}t||ƒ^q\ƒ|_|jj dk rFt|jt|j j ƒƒ|_qFqFWg|jD]}|j^qÂ}|jj dk r$|j|j jg|jD]}|j j ^q|ƒƒn|jjdk r®|jj}t|ƒdkrltdj|ƒƒ‚ng|jD]}||j^qv}|j|j j||ƒƒnx*|D]"}|j|j j||ƒƒqµW|S(Nis-Position_char="{}" must be a single character(tstr_valsRRJR*R tmaxRR"theaderR0R2tinfotnameR:RR1t position_charR3R7( RRt vals_listt col_str_itersRR-RERtchar((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRKës, ,(.   #  (R#R$R%R RLRK(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRMäst FixedWidthcB`s;eZdZdZdZeZeZddde d„Z RS(sr Read or write a fixed width table with a single header line that defines column names and positions. Examples:: # Bar delimiter in header and data | Col1 | Col2 | Col3 | | 1.2 | hello there | 3 | | 2.4 | many words | 7 | # Bar delimiter in header only Col1 | Col2 | Col3 1.2 hello there 3 2.4 many words 7 # No delimiter with column positions specified as input Col1 Col2Col3 1.2hello there 3 2.4many words 7 See the :ref:`fixed_width_gallery` for specific usage examples. t fixed_widths Fixed widthRcC`sMtt|ƒjƒ||jj_||jj_||j_||j_ dS(N( tsuperRWt__init__R4R:RRRPR>R?(RR>R?RR((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRZ*s  N( R#R$R%t _format_namet _descriptionR(t header_classRMt data_classR2tTrueRZ(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRW s tFixedWidthNoHeaderHeadercB`seZdZdZRS(s7Header reader for fixed with tables with no header lineN(R#R$R%R2R0(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyR`2stFixedWidthNoHeaderDatacB`seZdZdZRS(s6Data reader for fixed width tables with no header linei(R#R$R%R0(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRa7stFixedWidthNoHeadercB`s;eZdZdZdZeZeZddde d„Z RS(sH Read or write a fixed width table which has no header line. Column names are either input (``names`` keyword) or auto-generated. Column positions are determined either by input (``col_starts`` and ``col_stops`` keywords) or by splitting the first data line. In the latter case a ``delimiter`` is required to split the data line. Examples:: # Bar delimiter in header and data | 1.2 | hello there | 3 | | 2.4 | many words | 7 | # Compact table having no delimiter and column positions specified as input 1.2hello there3 2.4many words 7 This class is just a convenience wrapper around the ``FixedWidth`` reader but with ``header.start_line = None`` and ``data.start_line = 0``. See the :ref:`fixed_width_gallery` for specific usage examples. tfixed_width_no_headersFixed width with no headerRcC`s)tt|ƒj||d|d|ƒdS(NRR(RYRbRZ(RR>R?RR((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRZ[sN( R#R$R%R[R\R`R]RaR^R2R_RZ(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRb<s tFixedWidthTwoLineHeadercB`seZdZeZRS(sàHeader reader for fixed width tables splitting on whitespace. For fixed width tables with several header lines, there is typically a white-space delimited format line, so splitting on white space is needed. (R#R$R%RRL(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRd`stFixedWidthTwoLineDataSplittercB`seZdZdZRS(s5Splitter for fixed width tables splitting on ``' '``.R(R#R$R%R(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRejstFixedWidthTwoLineDatacB`seZdZeZRS(s8Data reader for fixed with tables with two header lines.(R#R$R%ReRL(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRfostFixedWidthTwoLinecB`s;eZdZdZdZeZeZddde d„Z RS(sÞ Read or write a fixed width table which has two header lines. The first header line defines the column names and the second implicitly defines the column positions. Examples:: # Typical case with column extent defined by ---- under column names. col1 col2 <== header_start = 0 ----- ------------ <== position_line = 1, position_char = "-" 1 bee flies <== data_start = 2 2 fish swims # Pretty-printed table +------+------------+ | Col1 | Col2 | +------+------------+ | 1.2 | "hello" | | 2.4 | there world| +------+------------+ See the :ref:`fixed_width_gallery` for specific usage examples. tfixed_width_two_lines#Fixed width with second header lineit-cC`sKtt|ƒjd|d|ƒ||j_||j_|d|j_dS(NRRi(RYRgRZRPR1RSR4R0(RR1RSRR((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRZ’s  N( R#R$R%R[R\RfR^RdR]R2R&RZ(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyRgts N(R%t __future__RRRtextern.six.movesRRR RRRRt BaseSplitterR R't BasicHeaderR(t BasicDataRMtBasicRWR`RaRbRdReRfRg(((s:lib/python2.7/site-packages/astropy/io/ascii/fixedwidth.pyt s"*¡&($