ó šßČ[c@`sdZddlmZmZmZddlZddlmZmZddl m Z ddl m Z ddl mZdd lmZd d lmZd d lmZd d lmZddlmZddlmZdefd„ƒYZdefd„ƒYZdejfd„ƒYZdejfd„ƒYZdej fd„ƒYZ!dej"fd„ƒYZ#dej$fd„ƒYZ%dS(sĖAn extensible ASCII table reader and writer. ipac.py: Classes to read IPAC table format :Copyright: Smithsonian Astrophysical Observatory (2011) :Author: Tom Aldcroft (aldcroft@head.cfa.harvard.edu) i(tabsolute_importtdivisiontprint_functionN(t defaultdictt OrderedDict(twrap(twarni(tsix(tzipi(tcore(t fixedwidth(tbasic(tAstropyUserWarning(tget_auto_format_functIpacFormatErrorDBMScB`seZd„ZRS(cC`sdjtt|ƒjƒdƒS(Ns {0} See {1}sHhttp://irsa.ipac.caltech.edu/applications/DDGEN/Doc/DBMSrestriction.html(tformattsupert Exceptiont__str__(tself((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyRs(t__name__t __module__R(((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyRstIpacFormatErrorcB`seZd„ZRS(cC`sdjtt|ƒjƒdƒS(Ns {0} See {1}sAhttp://irsa.ipac.caltech.edu/applications/DDGEN/Doc/ipac_tbl.html(RRRR(R((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR$s(RRR(((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR#stIpacHeaderSplittercB`sMeZdZdZdZdZdZeZ dZ dZ dZ dZ d„ZRS(s®Splitter for Ipac Headers. This splitter is similar its parent when reading, but supports a fixed width format (as required for Ipac table headers) for writing. t|ts\s*\\s\\c C`s|jp d}|jpd}|||}||}||}gt||ƒD]$\}} d| t|ƒ|^qP}||j|ƒ|S(NRt (t delimiter_padt delimiterRtlentjoin( RtvalstwidthstpadRt padded_delimt bookend_leftt bookend_righttvaltwidth((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR:s  :N(RRt__doc__tNonet process_linet process_valRRtFalsetskipinitialspacetcommentt write_commentt col_startstcol_endsR(((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR*st IpacHeadercB`s°eZdZeZdejfdejfdejfdejfdejfdejfdejffZ dZ dZ d „Z d „Zd „Zd „Zd „Zd„ZRS(sIPAC table headertintegertlongtdoubletfloattrealtchartdatetignorecc`s\|jj}xI|D]A}|jƒ}|j|ƒr|j|ƒr|j|ƒVqqWdS(s‰Generator to yield IPAC header lines, i.e. those starting and ending with delimiter character (with trailing whitespace stripped)N(tsplitterRtrstript startswithtendswithtstrip(Rtlinestdelimtline((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyt process_linesVs    c C`sHd„}|d}g|dtintRR5R<R=(R%tquote((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pytprocess_keyword_valueds     ttabletcommentstkeywordss&\\(?P \w+)\s* = (?P .+) $s\tnametvalues\ iN( RtretcompiletVERBOSER<tmatchtgroupt isinstanceRt string_typesR>tappend( RR?tmetaRGt table_metaRJt re_keywordRAtmRKR%tprev_val((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyt update_meta_s.        cC`sXxQ|jD](\}}|j|jjƒƒr |Sq Wtdj|j|jƒƒ‚dS(Ns(Unknown data type ""{}"" for column "{}"(t col_type_listR<traw_typetlowert ValueErrorRRK(Rtcolt col_type_keytcol_type((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyt get_col_typeœs  c C`s;|j|ƒ}g|j|ƒD] }|^q}t|ƒdkrRtdƒ‚n!t|ƒdkrstdƒ‚ng}d}xt|dƒD]{\}}tjd|jdƒƒ} || _|t|ƒ| _ t|ƒdkr|d|jdƒ| _ |j | ƒ| _ nt|ƒdkrF|d|jƒp=d| _nt|ƒd krÆ|d |jƒ} t| j tjƒr‡d nd } |jjj| | | jfƒn| j d}|j| ƒ|jd krź| jd8_q|jd kr| j d7_ qqWg|D]} | j^q|_||_dS(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 isEAt least one header line beginning and ending with delimiter requiredis&More than four header lines were foundiRKs -iiRt0trighttleftN(RBR:RR^t enumerateR tColumnR>tstarttendR\RbttypeR(tunitt issubclasstStrTypetdatat fill_valuesRTRKtipac_definitiontnamestcols( RR?t header_linesRt header_valsRrRhtiRKR_tnulltfillvaltx((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pytget_cols¤s: "  !"  cC`s$|jrt}nt}|j}|jr°ttƒ}x'|jD]}||jƒcd7R(Rt IpacFormatEtnamelistt countnamelistRKRxt doublenamesRXt dtypelisttunitlisttnullistR_t col_dtypetcol_unitt col_formatRvtauto_format_funct format_func((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pytstr_valsÜsh     )    *           !cC`s7x0|jƒD]"}|j|jj||ƒƒq W|S(såWrite header. The width of each column is determined in Ipac.write. Writing the header must be delayed until that time. This function is called from there, once the width information is available.(R˜RTR:R(RR?R R((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pytwrite"s N(RRR'Rtsplitter_classR tIntTypet FloatTypeRmR[t definitionR(t start_lineRBRZRbRyR˜R™(((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR1Es"       =  8 FtIpacDataSplittercB`seZdZdZeZRS(RR(RRRRtTruetbookend(((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyRŸ/stIpacDatacB`s;eZdZdZdZeZejdfgZ d„Z RS(sIPAC table data readers[|\\]iRvcC`s1x*|D]"}|j|jj||ƒƒqW|S(s. IPAC writer, modified from FixedWidth writer (RTR:R(RR?R t vals_listR((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR™<s  ( RRR'R-RžRŸRšR R‰RoR™(((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR¢5s tIpaccB`sMeZdZdZdgZeZdZeZ e Z de d„Z d„ZRS(sÉRead or write an IPAC format table. See http://irsa.ipac.caltech.edu/applications/DDGEN/Doc/ipac_tbl.html:: \\name=value \\ Comment | column1 | column2 | column3 | column4 | column5 | | double | double | int | double | char | | unit | unit | unit | unit | unit | | null | null | null | null | null | 2.0978 29.09056 73765 2.06000 B8IVpMnHg Or:: |-----ra---|----dec---|---sao---|------v---|----sptype--------| 2.09708 29.09056 73765 2.06000 B8IVpMnHg The comments and keywords defined in the header are available via the output table ``meta`` attribute:: >>> import os >>> from astropy.io import ascii >>> filename = os.path.join(ascii.__path__[0], 'tests/t/ipac.dat') >>> data = ascii.read(filename) >>> print(data.meta['comments']) ['This is an example of a valid comment'] >>> for name, keyword in data.meta['keywords'].items(): ... print(name, keyword['value']) ... intval 1 floatval 2300.0 date Wed Sp 20 09:48:36 1995 key_continue IPAC keywords can continue across lines Note that there are different conventions for characters occuring below the position of the ``|`` symbol in IPAC tables. By default, any character below a ``|`` will be ignored (since this is the current standard), but if you need to read files that assume characters below the ``|`` symbols belong to the column before or after the ``|``, you can specify ``definition='left'`` or ``definition='right'`` respectively when reading the table (the default is ``definition='ignore'``). The following examples demonstrate the different conventions: * ``definition='ignore'``:: | ra | dec | | float | float | 1.2345 6.7890 * ``definition='left'``:: | ra | dec | | float | float | 1.2345 6.7890 * ``definition='right'``:: | ra | dec | | float | float | 1.2345 6.7890 IPAC tables can specify a null value in the header that is shown in place of missing or bad data. On writing, this value defaults to ``null``. To specify a different null value, use the ``fill_values`` option to replace masked values with a string or number of your choice as described in :ref:`io_ascii_write_parameters`:: >>> from astropy.io.ascii import masked >>> fill = [(masked, 'N/A', 'ra'), (masked, -999, 'sptype')] >>> ascii.write(data, format='ipac', fill_values=fill) \ This is an example of a valid comment ... | ra| dec| sai| v2| sptype| | double| double| long| double| char| | unit| unit| unit| unit| ergs| | N/A| null| null| null| -999| N/A 29.09056 null 2.06 -999 2345678901.0 3456789012.0 456789012 4567890123.0 567890123456789012 Parameters ---------- definition : str, optional Specify the convention for characters in the data table that occur directly below the pipe (``|``) symbol in the header column definition: * 'ignore' - Any character beneath a pipe symbol is ignored (default) * 'right' - Character is associated with the column to the right * 'left' - Character is associated with the column to the left DBMS : bool, optional If true, this verifies that written tables adhere (semantically) to the `IPAC/DBMS `_ definition of IPAC tables. If 'False' it only checks for the (less strict) `IPAC `_ definition. tipacsIPAC format tableR9cC`sJtt|ƒjƒ|dkr.||j_n tdƒ‚||j_dS(NR9ReRds-definition should be one of ignore/left/right(R9ReRd(RR¤t__init__theaderRpR^R‚(RRR‚((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR¦¬s   cC`s}|jjjtjdfƒttj|jƒƒ|j _ |j j |j |j |jƒtj||j |j|jƒttj|jƒƒ}||j _ ||j_ g}d|jkr9xy|jdD]g}tt|ƒƒdkrłtdtƒnx6tt|ƒdddddƒD]}|j|ƒqWqĖWnd |jkrÄ|jd }xl|D]a}y1||d }|jd j|jƒ|ƒƒWq\tk r¼td j|ƒtƒq\Xq\Wng|jD]} | dkrĪ| ^qĪ} t| ƒrtd j| ƒtƒn|jj|jj ƒxXt|jjƒƒD]A\} } tg|j j ƒD]} t| | ƒ^q\ƒ| _!q=Wg}|jj ƒ}x!t"|ŒD]} |j| ƒq¤Wxat|jjƒƒD]J\} } |rtg|D]} t| | ƒ^qķƒ| _#qŃd| _#qŃWg|jjƒD]} t| j#| j!ƒ^q/}|j j$||ƒ|jj$|||ƒ|S(s Write ``table`` as list of strings. Parameters ---------- table : `~astropy.table.Table` Input table data Returns ------- lines : list List of strings corresponding to ASCII table RvRIiNs9Comment string > 78 characters was automatically wrapped.iPtinitial_indents\ tsubsequent_indentRJRLs \{0}={1!r}sTable metadata keyword {0} has been skipped. IPAC metadata must be in the form {{'keywords':{{'keyword': {{'value': value}} }}s‚Table metadata keyword(s) {0} were not written. IPAC metadata must be in the form {{'keywords':{{'keyword': {{'value': value}} }}i(RJRI(%RnRoRTR R‰tlistRt itervaluestcolumnsR§Rrtcheck_column_namesRqt strict_namestguessingt_apply_include_exclude_namest include_namest exclude_namesRURRˆRR RRR>t TypeErrortanyt_set_fill_valuesRftvaluestmaxR˜t headwidthRR&R™(RRHtnew_colsR?R-RAtkeydicttkeywordR%tkeyt ignored_keysRuR_Rt data_str_valst col_str_itersR ((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR™µsZ   (  #  (   "9"/ 1(RRR't _format_namet_io_registry_format_aliasesR t_io_registry_can_writet _descriptionR¢t data_classR1t header_classR+R¦R™(((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyR¤Cs`  (&R't __future__RRRRMt collectionsRRttextwrapRtwarningsRtexternRtextern.six.movesRRR R R tutils.exceptionsR t table.pprintR RRRt BaseSplitterRtFixedWidthHeaderR1tFixedWidthSplitterRŸtFixedWidthDataR¢tBasicR¤(((s4lib/python2.7/site-packages/astropy/io/ascii/ipac.pyt s& ź