[c@`sdZddlmZmZmZddlZddlZddlZddlm Z ddl Z ddl m Z ddlZddlZddlmZddlmZdd lmZdd lmZmZd d lmZd dddddgZededdfZide df6de df6de!df6de!df6Z"edZ#dZ$dZ%dZ&ddZ(de)fd YZ*de*fd!YZ+de+fd"YZ,de,fd#YZ-dS($sThis module contains functions and methods that relate to the DataInfo class which provides a container for informational attributes as well as summary info methods. A DataInfo object is attached to the Quantity, SkyCoord, and Time classes in astropy. Here it allows those classes to be used in Tables and uniformly carry table column attributes such as name, format, dtype, meta, and description. i(tabsolute_importtdivisiontprint_functionN(tdeepcopy(tpartial(t OrderedDict(tcontextmanageri(tsix(tzipt cStringIOi(tmetadatatdata_info_factorytdtype_info_nametBaseColumnInfotDataInfot MixinInfotParentDtypeInfotcategorytmessages3All-NaN|Mean of empty slice|Degrees of freedom <= 0tstrtStunicodetUtbytescc`s$tj}|t_dV|t_dS(skSet context for serialization. This will allow downstream code to understand the context in which a column is being serialized. Objects like Time or SkyCoord will have different default serialization representations depending on context. Parameters ---------- context : str Context name, e.g. 'fits', 'hdf5', 'ecsv', 'yaml' N(R t_serialize_context(tcontextt old_context((s6lib/python2.7/site-packages/astropy/utils/data_info.pytserialize_context_as0s  cC`smtj|}|jdkr`tjd|jjd}ttj |jf}||}n |j }|S(sReturn a human-oriented string name of the ``dtype`` arg. This can be use by astropy methods that present type information about a data object. The output is mostly equivalent to ``dtype.name`` which takes the form [B] where is like ``int`` or ``bool`` and [B] is an optional number of bits which gets included only for numeric types. For bytes, string and unicode types, the output is shown below, where is the number of characters. This representation corresponds to the Python type that matches the dtype:: Numpy S U Python 2 str unicode Python 3 bytes str Parameters ---------- dtype : str, np.dtype, type Input dtype as an object that can be converted via np.dtype() Returns ------- dtype_info_name : str String name of ``dtype`` RRs(\d+)i(RR( tnptdtypetkindtretsearchRtgrouptSTRING_TYPE_NAMESRtPY2tname(Rtlengtht type_nametout((s6lib/python2.7/site-packages/astropy/utils/data_info.pyR Cs  c`sfd}|S(s8 Factory to create a function that can be used as an ``option`` for outputting data object summary information. Examples -------- >>> from astropy.utils.data_info import data_info_factory >>> from astropy.table import Column >>> c = Column([4., 3., 2., 1.]) >>> mystats = data_info_factory(names=['min', 'median', 'max'], ... funcs=[np.min, np.median, np.max]) >>> c.info(option=mystats) min = 1.0 median = 2.5 max = 4.0 n_bad = 0 length = 4 Parameters ---------- names : list List of information attribute names funcs : list List of functions that compute the corresponding information attribute Returns ------- func : function Function that can be used as a data info option c`sg}xtD]w\}}y7t|tjrLt||}n ||}Wntk ry|jdqX|jt|qWtt|S(Ns--( Rt isinstanceRt string_typestgetattrt ExceptiontappendRR(tdattoutsR$tfuncR'(tfuncstnames(s6lib/python2.7/site-packages/astropy/utils/data_info.pyR/s ((R1R0R/((R0R1s6lib/python2.7/site-packages/astropy/utils/data_info.pyR iscC`si}xr|D]j}t||d}|dk r tjrjt|}t|tjrjt|}qjn|||s(tdictt attr_namest_attrs(tselftbound((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt__init__scC`si|dkr|}||_nG|jjd}|dkr\|jdt}|jdRN(R2t _parent_clst__dict__tgett __class__tTruet_parent(RMtinstancet owner_clsR>((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt__get__s     cC`s|dkrtdnt|tr|jdt}|jds)info must be set with a DataInfo instance(R2t ValueErrorR(RRSRTRQRKtattrs_from_parentt_attrs_no_copyRR*RLt TypeError(RMRVtvalueR>R6((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt__set__s #cC`s|jS(N(RL(RM((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt __getstate__scC`s ||_dS(N(RL(RMtstate((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt __setstate__scC`s|jdr%tt|j|S||jkrDt|j|Sy|j|}Wn'tk r~tt|j|nX|dkrt |r|}n|dkr|dkrt j d}n|S(Nt_t parent_tableRtO( t startswithtsuperRt__getattribute__RZR*RURLtKeyErrortcallableR2RR(RMR6R]((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt __getattr__s  cC`s#t|j|d}||jkrKt|t rKt|j||dSt|tr|jdkrxt dn|j||dS|j drt t |j ||dS||jkrt dj|jn|dkr|dkrdn tj|}n||j|>> from astropy.table import Column >>> c = Column([1, 2], unit='m', dtype='int32') >>> c.info() dtype = int32 unit = m class = Column n_bad = 0 length = 2 >>> c.info(['attributes', 'stats']) dtype = int32 unit = m class = Column mean = 1.5 std = 0.5 min = 1 max = 2 n_bad = 0 length = 2 Parameters ---------- option : str, function, list of (str or function) Info option, defaults to 'attributes'. out : file-like object, None Output destination, defaults to sys.stdout. If None then the OrderedDict with information attributes is returned Returns ------- info : OrderedDict if out==None else None R;R$t info_summary_s-option={0} is not an allowed information typetignoreNtmaskitn_badR%s {0} = {1}(!tsyststdoutRURR>R$R2R(tlistttupleRR)thasattrR*RYRFtwarningstcatch_warningstIGNORE_WARNINGStfilterwarningstupdateRt count_nonzeroR{tisinftisnanR+tlenR\titemstwritetostlinesep( RMtoptionR'R-R>R$toptionst ignore_kwargsR|tkeyR7((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt__call__EsD6       $    )      cC`sE|jdkr"tt|jSt}|jd||jS(NR'(RUR2RfRt__repr__tStringIORtgetvalue(RMR'((s6lib/python2.7/site-packages/astropy/utils/data_info.pyRs  (RR:RERFRGR9(N("R=t __module__t__doc__t_statstsetRZRKR[t_info_summary_attrsRsR2RUtFalseRORXR^R_RaRjRoRtRvt staticmethodR R6RR@tinfo_summary_attributeststatR*Rtinfo_summary_statsRR(((s6lib/python2.7/site-packages/astropy/utils/data_info.pyRs6          ( )bcB`sqeZdZejjddgZedgZdZ e dZ dZ dZ dZedZRS( sg Base info class for anything that can be a column in an astropy Table. There are at least two classes that inherit from this: ColumnInfo: for native astropy Column / MaskedColumn objects MixinInfo: for mixin column objects Note that this class is defined here so that mixins can use it without importing the table package. RctindicescC`s/tt|jd||r+i|_ndS(NRN(RfR ROt _format_funcs(RMRN((s6lib/python2.7/site-packages/astropy/utils/data_info.pyROscc`sl|j}|jdkr+ddlm}n |jj}|j}x%||dttiD] }|VqYWdS(sG This is a mixin-safe version of Column.iter_str_vals. i(t FORMATTERiN(RURcR2t table.columnRt formattert_pformat_col_iterR(RMtcolRRtstr_val((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt iter_str_valss   c C`s|js dSt|tr@|j|}tt|}nCt|tjrz|jjdkrztj |d}n |g}tj |}|j dkrt|t |}nxGt ||D]6\}}x'|jD]}|j||j|qWqWdS(s Adjust info indices after column modification. Parameters ---------- index : slice, int, list, or ndarray Element(s) of column to modify. This parameter can be a single row number, a list of row numbers, an ndarray of row numbers, a boolean ndarray (a mask), or a column slice. value : int, list, or ndarray New value(s) to insert col_len : int Length of the column Ntbii(RR(tsliceRtrangeRtndarrayRRtwheret atleast_1dtsizeRRtreplaceR$( RMtindexR]tcol_lentttkeysRR7t col_index((s6lib/python2.7/site-packages/astropy/utils/data_info.pytadjust_indicess $ c C`sFddlm}t|dts2g|j_|St|trjg|jD]}||^qK|j_n|jrBt|tj r|j j dkrtj |d}nd}t |d|k}g|j_xj|jD]\}|st||r|j||} nt|} | j||jjj| qWn|S(sX Given a sliced object, modify its indices to correctly represent the slice. Parameters ---------- col_slice : Column or mixin Sliced object item : slice, list, or ndarray Slice used to create col_slice col_len : int Length of original object i(t SortedArrayt _copy_indicesRig333333?(ttable.sorted_arrayRR*RTR>RR(RRRRRRRt get_sliceRt replace_rowsR,( RMt col_slicetitemRRtxt thresholdtsmallRt new_index((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt slice_indicess& ) $   c `sddlm}fd}fd}||d}x6|dD]*}tj|||d|d|}qIWtj||d R2(RIR6(R(s6lib/python2.7/site-packages/astropy/utils/data_info.pys Os ((R(R5(Rs6lib/python2.7/site-packages/astropy/utils/data_info.pytgetattrsNsiitmetadata_conflictsRRcs`s|]}|jdVqdS(iN(R:(RIR((s6lib/python2.7/site-packages/astropy/utils/data_info.pys [sscolumns have different shapesR:(ttable.np_utilsRR tmerget common_dtypeRRtpop( tcolsRR$R5RRRR'Rt uniq_shapes((R5R$s6lib/python2.7/site-packages/astropy/utils/data_info.pytmerge_cols_attributes,s N(R=RRRRKtunionRR[R2RRRORRRRR(((s6lib/python2.7/site-packages/astropy/utils/data_info.pyR s    & *cB`seZdZRS(cC`sp|dkrS|jdk rSddlm}||}|jjj|j|ntt|j ||dS(NR$i(tfix_column_name( RcR2RRtcolumnst_rename_columnR$RfRRo(RMR6R]Rtnew_name((s6lib/python2.7/site-packages/astropy/utils/data_info.pyRoes  (R=RRo(((s6lib/python2.7/site-packages/astropy/utils/data_info.pyRcscB`seZdZedgZRS(s&Mixin that gets info.dtype from parentR(R=RRRRZ(((s6lib/python2.7/site-packages/astropy/utils/data_info.pyRqs(.Rt __future__RRRRR}RptcopyRtnumpyRt functoolsRRRt collectionsRt contextlibRtexternRtextern.six.movesRR RR;R t__all__RJtRuntimeWarningRRRTR"RR R R8R2R@tobjectRR RR(((s6lib/python2.7/site-packages/astropy/utils/data_info.pyt s<          & 0