HR-e!O2dZddlZddlZddlZddlmZddlZddlm Z ddl m Z ddl m Z mZddZdd Zd Zdd Zdd Zd ZddZddZGddejZGdde jZGdde jZGdde jZdS)z This package contains functions for reading and writing QDP tables that are not meant to be used directly, but instead are available as readers/writers in `astropy.table`. See :ref:`astropy:table_io` for more details. N)Iterable)Table)AstropyUserWarning)basiccorecd}d}|}|d}d|d}d|d|d|d }d |d |d |d }tj|tj}|}|sdS||} | t d|| D]9\} } | | dkr(dt| |cS| cSdS)aInterpret a QDP file line. Parameters ---------- line : str a single line of the file Returns ------- type : str Line type: "comment", "command", or "data" Examples -------- >>> _line_type("READ SERR 3") 'command' >>> _line_type(" \n !some gibberish") 'comment' >>> _line_type(" ") 'comment' >>> _line_type(" 21345.45") 'data,1' >>> _line_type(" 21345.45 1.53e-3 1e-3 .04 NO nan") 'data,6' >>> _line_type(" 21345.45,1.53e-3,1e-3,.04,NO,nan", delimiter=',') 'data,6' >>> _line_type(" 21345.45 ! a comment to disturb") 'data,1' >>> _line_type("NO NO NO NO NO") 'new' >>> _line_type("NO,NO,NO,NO,NO", delimiter=',') 'new' >>> _line_type("N O N NOON OON O") Traceback (most recent call last): ... ValueError: Unrecognized QDP line... >>> _line_type(" some non-comment gibberish") Traceback (most recent call last): ... ValueError: Unrecognized QDP line... z'[+-]?(\d+(\.\d*)?|\.\d+)([eE][+-]?\d+)?zREAD [TS]ERR(\s+[0-9]+)+Nz\s+zNO(zNO)+(z|NO|[-+]?nan)(z|NO|[-+]?nan))*)z^\s*((?Pz )|(?Pz )|(?Pz)?\s*(\!(?P.*))?\s*$commentzUnrecognized QDP line: datadata,)sep) recompile IGNORECASEstripmatch ValueError groupdictitemslensplit) line delimiter _decimal_re _command_rer_new_re_data_re_type_re _line_type_rertype_vals 4lib/python3.11/site-packages/astropy/io/ascii/qdp.py _line_typer$sBT=K-K CSGRKRRsRR[RRRHvKvv7vvxvvvHJx77M ::<>> line0 = "! A comment" >>> line1 = "543 12 456.0" >>> lines = [line0, line1] >>> types, ncol = _get_type_from_list_of_lines(lines) >>> types[0] 'comment' >>> types[1] 'data,3' >>> ncol 3 >>> lines.append("23") >>> _get_type_from_list_of_lines(lines) Traceback (most recent call last): ... ValueError: Inconsistent number of columns c2g|]}t|S)r)r$).0rrs r# z0_get_type_from_list_of_lines..zs& E E EtZ 2 2 2 E E Er%Nr zInconsistent number of columns) startswithintr)linesrtypes current_ncolr!ncols ` r#_get_type_from_list_of_linesr2XsD F E E Eu E E EELCC   G $ $ CuQRRy>>D## %% !ABBB , r%c>d|vr|d}nt|trFt|5}d|D}dddn #1swxYwYn't|t r|}nt d|S)N c6g|]}|S)r)r)rs r#r*z(_get_lines_from_file..s ???dTZZ\\???r%zinvalid value of qdb_file)r isinstancestropen readlinesrr)qdp_filer.fobjs r#_get_lines_from_filer=s xt$$ Hc " "6 (^^ @t??dnn.>.>???E @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ Hh ' '64555 LsA**A.1A.cdt|D}|gx}}n@tj|}|dg}|dg}|Gt |t |dzz}|t |z|krt dd}t|D]d}|dz|z } ||d krd | } | || dz } | ||<| |vr| d ||dz<|dz }D| |vr| d ||dz<| d ||dz<|dz }det jd|DrJ|S)aGive list of column names from the READ SERR and TERR commands. Parameters ---------- err_specs : dict ``{'serr': [n0, n1, ...], 'terr': [n2, n3, ...]}`` Error specifications for symmetric and two-sided errors ncols : int Number of data columns Other Parameters ---------------- names : list of str Name of data columns (defaults to ['col1', 'col2', ...]), _not_ including error columns. Returns ------- colnames : list List containing the column names. Error columns will have the name of the main column plus ``_err`` for symmetric errors, and ``_perr`` and ``_nerr`` for positive and negative errors respectively Examples -------- >>> col_in = ['MJD', 'Rate'] >>> cols = _interpret_err_lines(None, 2, names=col_in) >>> cols[0] 'MJD' >>> err_specs = {'terr': [1], 'serr': [2]} >>> ncols = 5 >>> cols = _interpret_err_lines(err_specs, ncols, names=col_in) >>> cols[0] 'MJD' >>> cols[2] 'MJD_nerr' >>> cols[4] 'Rate_err' >>> _interpret_err_lines(err_specs, 6, names=col_in) Traceback (most recent call last): ... ValueError: Inconsistent number of input colnames cg|]}dSr6)r)is r#r*z(_interpret_err_lines..s)))q)))r%Nserrterrz%Inconsistent number of input colnamesrrrAcol_err_perr_nerrcg|]}|dk Sr@r6r)cs r#r*z(_interpret_err_lines..s1111qBw111r%)rangecopydeepcopypoprrnpany) err_specsncolsnamescolnames serr_cols terr_colsall_error_colsshiftrBcol_num colname_roots r#_interpret_err_linesr]sX*)E%LL)))H "" IIM),, MM&"-- MM&"--  Y#i..1*<< CJJ &% / /DEE E E 5\\a%%- A;"   &W   1-L%' i  !-333HQUO QJE  i  !-444HQUO!-444HQUO QJE  v1111122222 Or%ct|}t||\}}g}i}d}d} d} d} d} t||D]\} }| d} |dkr | | dzz } <|dkr3| dkr| } d} |ikrt jdt| | dzz } u|d rz|ikr| dkr|  dD]k}| }t|d kr.,s4Q4Q4QSVV4Q4Q4Qr%rEr)rUNOnew)rUrowsinitial_commentscomments)r=r2ziprlstripwarningswarnrr,rrlowerr]upperappendrQmamaskedr-rfloatrmeta)r;input_colnamesrr.contentsr1 table_listrSrV comment_textrf command_lines current_rowsrdatatypecliner`valuesv new_tables r#_get_tables_from_qdp_filer~s* ! * *E1%9MMMNHdJIHLMLeX..>>hzz||""3'' y D4K 'L  y ""#/ ! B O& TD[ (M    v & & B=B#6#6*002288>>RRE#kkmm1133G7||a'' 4Q4QWQRR[4Q4Q4QIgaj..0011/ 4~VVV#! FZZ ** 0 07799$$MM"%,////0 c!ff----%000 eAhh/////0    ' ' '  u  '!|DDD 5E5K5K5M5M5S5S66 12.:-?-?-A-A-G-G-M-M z*! !!),,,#   |<<< -=-C-C-E-E-K-KD-Q-Q )*%1%7%7%9%9%?%?%E%E z")$$$ s"G>>,H-,H-cd}g}g}t|D]\}}|dr|||z |dz }8|dra|||z t||dzks||dzdst d|dz }|dr-||dz dst d||fS) a-Get which column names are error columns. Examples -------- >>> colnames = ['a', 'a_err', 'b', 'b_perr', 'b_nerr'] >>> serr, terr = _understand_err_col(colnames) >>> np.allclose(serr, [1]) True >>> np.allclose(terr, [2]) True >>> serr, terr = _understand_err_col(['a', 'a_nerr']) Traceback (most recent call last): ... ValueError: Missing positive error... >>> serr, terr = _understand_err_col(['a', 'a_perr']) Traceback (most recent call last): ... ValueError: Missing negative error... rrGrrHrIzMissing negative errorrEzMissing positive error) enumerateendswithrnrr)rVrZrCrDrBrFs r#_understand_err_colrYs$( E D DH%% 7 73 <<   7 KKE " " " QJEE \\' " " 7 KKE " " "8}}A%%Xa!e_-E-Eg-N-N% !9::: QJEE \\' " " 78AE?+C+CG+L+L 7566 6 :r%cr|tjdtd}t|||}||S)aRead a table from a QDP file. Parameters ---------- qdp_file : str Input QDP file name Other Parameters ---------------- names : list of str Name of data columns (defaults to ['col1', 'col2', ...]), _not_ including error columns. table_id : int, default 0 Number of the table to be read from the QDP file. This is useful when multiple tables present in the file. By default, the first is read. delimiter : str Any delimiter accepted by the `sep` argument of str.split() Returns ------- tables : list of `~astropy.table.Table` List containing all the tables present inside the QDP file Nz9table_id not specified. Reading the first available tabler)rsr)rjrkrr~)r;rUtable_idrtabless r#_read_table_qdprsS4 G     &)F ( r%c~ddl}|}d|jvr`|jdgkrO|jdD]A}|}|dsd|z}t ||B|t |j\}}n,|dg}|dg}|gkr3d d |D}t d |||gkr3d d |D}t d ||d |jvr`|jd gkrO|jd D]A}|}|dsd|z}t ||B|j} t dd | z||D]s} g} | D]H} tj | st| } nd} | | It d | |t|}||9t#|d5}t ||dddn #1swxYwY|dS)aWrite a table to a QDP file. Parameters ---------- table : :class:`~astropy.table.Table` Input table to be written filename : str Output QDP file name Other Parameters ---------------- err_specs : dict Dictionary of the format {'serr': [1], 'terr': [2, 3]}, specifying which columns have symmetric and two-sided errors (see QDP format specification) rNrfr_)filerCrD c,g|]}t|Sr6r8r)r"s r#r*z$_write_table_qdp..===Cs3xx===r%z READ SERR c,g|]}t|Sr6rrs r#r*z$_write_table_qdp..rr%z READ TERR rgrcwr4)ioStringIOrrrr,printrrVrPjoinrQro is_maskedr8rngetvaluecloser9r)tablefilenamerSrr<rrWrX col_stringrVrowr{r"rep full_strings r#_write_table_qdprs("III ;;==DUZ''EJ7I,Jb,P,PJ12 # #D::<BB 99MM&"-- MM&"-- BXX==9===>>  ':''d3333BXX==9===>>  ':''d3333UZEJz$:b$@$@Jz* # #D::< Extra space > | > v >! MJD Err (pos) Err(neg) Rate Error >53000.123456 2.378e-05 -2.378472e-05 NO 0.212439 These readers and writer classes will strive to understand which of the comments belong to all the tables, and which ones to each single table. General comments will be stored in the ``initial_comments`` meta of each table. The comments of each table will be stored in the ``comments`` meta. Example:: t = Table.read(example_qdp, format='ascii.qdp', table_id=1, names=['a', 'b', 'c', 'd']) reads the second table (``table_id=1``) in file ``example.qdp`` containing the table above. There are four column names but seven data columns, why? Because the ``READ SERR`` and ``READ TERR`` commands say that there are three error columns. ``t.meta['initial_comments']`` will contain the initial two comment lines in the file, while ``t.meta['comments']`` will contain ``Table 1 comment`` The table can be written to another file, preserving the same information, as:: t.write(test_file, err_specs={'terr': [1], 'serr': [3]}) Note how the ``terr`` and ``serr`` commands are passed to the writer. qdpTz.qdpzQuick and Dandy PlotterNct||_||_||_||_dSN)super__init__rrUrSr)selfrrUrSr __class__s r#rz QDP.__init__os9    "r%c|j|d|_t|j|j|j|jS)Nr4)newline)rrUr)inputter get_linesr.rrrUr)rrs r#readzQDP.readvsI],,UD,AA  J]*n     r%c\||t||j}|S)N)rS)_check_multidim_tablerrS)rrr.s r#writez QDP.writes/ ""5))) $.AAA r%)NNNN)rrrr _format_name_io_registry_can_write_io_registry_suffix _descriptionr header_classr data_classrrr __classcell__)rs@r#rr sYYvL! ,LLJ   r%rr)NN)NNN)rrNrrjcollections.abcrnumpyrQ astropy.tablerastropy.utils.exceptionsrrArrr$r2r=r]r~rrrDefaultSplitterrCommentedHeaderHeaderr BasicDatarBasicrr6r%r#rs  $$$$$$777777AAAAH,,,,^   UUUUpiiiiX%%%P%%%%PB#B#B#B#J$&+eowwwww%+wwwwwr%