HR-e|+dZddlZddlmZGddejZGddejZGd d ejZ Gd d eZ Gd deZ Gdde Z GddeZ Gdde ZGddejZGddeZGddeZGddeZGdde ZGdd ejZGd!d"eZGd#d$eZGd%d&e ZGd'd(eZGd)d*eZGd+d,eZdS)-a+An extensible ASCII table reader and writer. basic.py: Basic table read / write functionality for simple character delimited files with various options for column header definition. :Copyright: Smithsonian Astrophysical Observatory (2011) :Author: Tom Aldcroft (aldcroft@head.cfa.harvard.edu) N)coreceZdZdZdZdZdZdS) BasicHeaderz Basic table Header Reader. Set a few defaults for common ascii table formats (start at line 0, comments begin with ``#`` and possibly white space) r\s*## N__name__ __module__ __qualname____doc__ start_linecomment write_comment6lib/python3.11/site-packages/astropy/io/ascii/basic.pyrr)JGMMMrrceZdZdZdZdZdZdS) BasicDataz Basic table Data Reader. Set a few defaults for common ascii table formats (start at line 1, comments begin with ``#`` and possibly white space) rrrNr rrrrrrrrc(eZdZdZdZdZdgZeZe Z dS)BasicaCharacter-delimited table with a single header line at the top. Lines beginning with a comment character (default='#') as the first non-whitespace character are comments. Example table:: # Column definition is the first uncommented line # Default delimiter is the space character. apples oranges pears # Data starts after the header column definition, blank lines ignored 1 2 3 4 5 6 basicz"Basic table with custom delimitersasciiN) r r r r _format_name _description_io_registry_format_aliasesr header_classr data_classrrrrr,s6 L7L#*)LJJJrrceZdZdZdZdS)NoHeaderHeaderz Reader for table header without a header. Set the start of header line number to `None`, which tells the basic reader there is no header line. Nr r r r rrrrr!r!EsJJJrr!ceZdZdZdZdS) NoHeaderDataz{ Reader for table data without a header. Data starts at first uncommented line since there is no header line. rNr"rrrr$r$Ps JJJrr$c"eZdZdZdZdZeZeZ dS)NoHeadera6Character-delimited table with no header line. When reading, columns are autonamed using header.auto_format which defaults to "col%d". Otherwise this reader the same as the :class:`Basic` class from which it is derived. Example:: # Table data 1 2 "hello there" 3 4 world no_headerzBasic table with no headersN) r r r r rrr!rr$rrrrr&r&Zs.  L0L!LJJJrr&ceZdZdZdZdZdS)CommentedHeaderHeaderz Header class for which the column definition line starts with the comment character. See the :class:`CommentedHeader` class for an example. c#Ktj|j}|D]7}||}|r||dV8dS)z Return only lines that start with the comment regexp. For these lines strip out the matching characters. N)recompilermatchend)selflines re_commentliner-s r process_linesz#CommentedHeaderHeader.process_linesssk Z --  * *D$$T**E *599;;==)))) * *rcz||j|j|jzdS)N)appendrsplitterjoincolnames)r/r0s rwritezCommentedHeaderHeader.write~s4 T'$-*<* character. This matches the format produced by ``np.savetxt()``, with ``delimiter=','``, and ``header=''``. Example:: # col1 col2 col3 # Comment line 1 2 3 4 5 6 commented_headerz Column names in a commented linecPt|}d|jvrz|jj}|dkrt |jd|z}|jdd||jd|dzdz|jd<|jds|jd=|S)z Read input data (file-like object, filename, list of strings, or single string) into a Table and return the result. commentsrNr)superreadmetaheaderrlen)r/tableoutidx __class__s rr@zCommentedHeader.reads ggll5!!  ! !+(CQww#(:.//#5$TcT*SXj-A#'))-LL HZ 8J' )HZ( rcp|j||j||dS)zL Write comment lines after, rather than before, the header. N)rBr9write_comments)r/r0rAs r write_headerzCommentedHeader.write_headers8 %    ""5$/////r) r r r r rrr)rr$rr@rJ __classcell__)rGs@rr;r;sg&&L5L(LJ*0000000rr;ceZdZdZdZdZdS)TabHeaderSplitterz0Split lines on tab and do not remove whitespace. c |dzS)N r)r/r2s r process_linezTabHeaderSplitter.process_lines d{rN)r r r r delimiterrQrrrrMrMs.::IrrMceZdZdZdZdZdS)TabDataSplitterzT Don't strip data value whitespace since that is significant in TSV tables. NF)r r r r process_valskipinitialspacerrrrTrTs'KrrTceZdZdZeZdS) TabHeaderz@ Reader for header of tables with tab separated header. N)r r r r rMsplitter_classrrrrXrXs'NNNrrXceZdZdZeZdS)TabDataz< Reader for data of tables with tab separated data. N)r r r r rTrYrrrr[r[s%NNNrr[c"eZdZdZdZdZeZeZ dS)TabaTab-separated table. Unlike the :class:`Basic` reader, whitespace is not stripped from the beginning and end of either lines or individual column values. Example:: col1 col2 col3 # Comment line 1 2 5 tabz%Basic table with tab-separated valuesN) r r r r rrrXrr[rrrrr]r]s.  L:LLJJJrr]ceZdZdZdZdS) CsvSplitterz@ Split on comma for CSV (comma-separated-value) tables. ,N)r r r r rRrrrr`r`sIIIrr`ceZdZdZeZdZdZdS) CsvHeaderzK Header that uses the :class:`astropy.io.ascii.basic.CsvSplitter`. N)r r r r r`rYrrrrrrcrcs)!NGMMMrrcc2eZdZdZeZejdfgZdZ dZ dS)CsvDatazI Data that uses the :class:`astropy.io.ascii.basic.CsvSplitter`. N) r r r r r`rYrmasked fill_valuesrrrrrreres8!NK$%KGMMMrrec6eZdZdZdZdgZdZdZdZe Z e Z dZ dS)CsvaCSV (comma-separated-values) table. This file format may contain rows with fewer entries than the number of columns, a situation that occurs in output from some spreadsheet editors. The missing entries are marked as masked in the output table. Masked values (indicated by an empty '' field value when reading) are written out in the same way with an empty ('') field. This is different from the typical default for `astropy.io.ascii` in which missing values are indicated by ``--``. Since the `CSV format `_ does not formally support comments, any comments defined for the table via ``tbl.meta['comments']`` are ignored by default. If you would still like to write those comments then include a keyword ``comment='#'`` to the ``write()`` call. Example:: num,ra,dec,radius,mag 1,32.23222,10.1211 2,38.12321,-88.1321,2.2,17.0 csvTz.csvzComma-separated-valuesc~t||kr)||t|z dgz|S)aQ Adjust row if it is too short. If a data row is shorter than the header, add empty values to make it the right length. Note that this will *not* be called if the row already matches the header. Parameters ---------- str_vals : list A list of value strings from the current row of the table. ncols : int The expected number of entries from the table header. Returns ------- str_vals : list List of strings to be parsed into data entries in the output table. rf)rCextend)r/str_valsncolss rinconsistent_handlerzCsv.inconsistent_handler/s>( x==5 OOUS]]2rd: ; ; ;rN)r r r r rr_io_registry_can_write_io_registry_suffixrrcrrerrprrrrjrj sW2L#('! +LLJrrjcBeZdZdZejejdZdZdZ dZ dS) RdbHeaderz Header for RDB tables. )nsc|jdS)N)raw_type)r/cols rget_type_map_keyzRdbHeader.get_type_map_keyPs|Brcn||}dttd||D}t |dkrt d|\|_}t |jt |krtjdtd|Drtjd|| t|j |D]&\}}||_ | ||_'dS)av Initialize the header Column objects from the table ``lines``. This is a specialized get_cols for the RDB type: Line 0: RDB col names Line 1: RDB col definitions Line 2+: RDB data rows Parameters ---------- lines : list List of table lines Returns ------- None cg|]\}}|Srr).0_hls r z&RdbHeader.get_cols..gsWWW51bBWWWrzRDB header requires 2 lineszDRDB header mismatch between number of column names and column types.c3XK|]%}tjd|tj V&dS)z \d*(N|S)$N)r+r- IGNORECASE)r~xs r z%RdbHeader.get_cols..qs5OO28L!R];;;OOOOOOrz3RDB types definitions do not all match [num](N|S): N)r3zipranger6rC ValueErrornamesrInconsistentTableErrorany_set_cols_from_namescolsry get_col_typetype)r/r0 header_linesheader_vals_list raw_typesrzrys rget_colszRdbHeader.get_colsSsC&))%00 WWCa$-- :U:U,V,VWWW  A % %:;; ; 0 I tz??c)nn , ,-V  OOYOOO O O -QiQQ  !!### I66 . .MC#CL((--CHH . .rc4||j|jg}|jD].}|jjjdvrdnd}||/||j|dS)N)SUrN)r5r6r7r8rinfodtypekind)r/r0 rdb_typesrzrdb_types rr9zRdbHeader.write{s T]'' 66777 9 ' 'C!hn1Z??ssSH   X & & & & T]'' 2233333rN) r r r r rNumTypeStrType col_type_mapr{rr9rrrrtrtIs_DL99L   &.&.&.P44444rrtceZdZdZdZdS)RdbDataz= Data reader for RDB data. Starts reading at line 2. rNr"rrrrrsJJJrrc,eZdZdZdZdgZdZdZeZ e Z dS)Rdba4Tab-separated file with an extra line after the column definition line that specifies either numeric (N) or string (S) data. See: https://www.drdobbs.com/rdb-a-unix-command-line-database/199101326 Example:: col1 col2 col3 N S N 1 2 5 rdbz.rdbz0Tab-separated with a type definition header lineN) r r r r rrrrrrtrrrrrrrrs<  L#(' ELLJJJrr)r r+rfr BaseHeaderrBaseDatar BaseReaderrr!r$r&r)r;DefaultSplitterrMrTrXr[r]r`rcrerjrtrrrrrrs      $/           DO2[9u&MMMMMKMMM*4040404040e404040n,'''''' '''%%%%%i%%%%($& i:::::%:::z:4:4:4:4:4 :4:4:4zg#r