[c@ sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlmZddlmZddlmZddlZyddlmZWn'ek r=defd YZnXdd lmZdd lmZm Z m!Z!m"Z"m#Z#dd l$m%Z%m&Z&dd lm'Z'ddl(m)Z)ddl*m+Z+ej,re-Z-n dZ-e ej.fZ/defdYZ0dZ1ddZ3dZ4dZ5dZ6dZ7dZ8dZ9ej,r]dZ:n dZ:ej,r{dZ;n dZ;dZ<d Z=d!Z>d"Z?d#Z@ej,re jAZAd$ZBneCjAZAd%ZBd&ZDdaEd'ZFd;ZGd=ZHd,ZId-ZJd.ZKd/ZLd0ZMd1ZNd2ZOd3ZPd4ZQd5ZRd6ZSedd7ZTd8ZUd9ZVdS(>i(tdivisionN(tcontextmanageri(tdata(t LooseVersion(tStringIORcB seZRS((t__name__t __module__(((s3lib/python2.7/site-packages/astropy/io/fits/util.pyRs(tsix(t string_typest integer_typest text_typet binary_typetnext(tziptrange(twraps(tsuppress(tAstropyUserWarningcC s||k||kS(N((tatb((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt.st NotifierMixincB s8eZdZdZdZdZdZdZRS(s Mixin class that provides services by which objects can register listeners to changes on that object. All methods provided by this class are underscored, since this is intended for internal use to communicate between classes in a generic way, and is not machinery that should be exposed to users of the classes involved. Use the ``_add_listener`` method to register a listener on an instance of the notifier. This registers the listener with a weak reference, so if no other references to the listener exist it is automatically dropped from the list and does not need to be manually removed. Call the ``_notify`` method on the notifier to update all listeners upon changes. ``_notify('change_type', *args, **kwargs)`` results in calling ``listener._update_change_type(*args, **kwargs)`` on all listeners subscribed to that notifier. If a particular listener does not have the appropriate update method it is ignored. Examples -------- >>> class Widget(NotifierMixin): ... state = 1 ... def __init__(self, name): ... self.name = name ... def update_state(self): ... self.state += 1 ... self._notify('widget_state_changed', self) ... >>> class WidgetListener(object): ... def _update_widget_state_changed(self, widget): ... print('Widget {0} changed state to {1}'.format( ... widget.name, widget.state)) ... >>> widget = Widget('fred') >>> listener = WidgetListener() >>> widget._add_listener(listener) >>> widget.update_state() Widget fred changed state to 2 cC s8|jdkr!tj|_n||jt|`` method with the given ``*args`` and ``**kwargs``. The notification does not by default include the object that actually changed (``self``), but it certainly may if required. Ns _update_{0}(RRtformatt valuerefsthasattrtgetattrtcallable(Rt notificationtargstkwargst method_nameRtmethod((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_notify{s    cC sJytt|j}Wn tk r;|jj}nXd|d<|S(sj Exclude listeners when saving the listener's state, since they may be ephemeral. RN(tsuperRt __getstate__tAttributeErrort__dict__tcopyR(Rtstate((s3lib/python2.7/site-packages/astropy/io/fits/util.pyR,s   N( RRt__doc__RRRRR*R,(((s3lib/python2.7/site-packages/astropy/io/fits/util.pyR3s + cC stt|S(s Returns the first item returned by iterating over an iterable object. Example: >>> a = [1, 2, 3] >>> first(a) 1 (R titer(titerable((s3lib/python2.7/site-packages/astropy/io/fits/util.pytfirsts cc st|ts'tdj|n|dkr?t}ny|j}Wn tk rq|j|}nXxct|dtj dD]F}||kr|j ||Vxt ||D] }|VqWqqWdS(s itersubclasses(cls) Generator over all subclasses of a given class, in depth first order. >>> class A(object): pass >>> class B(A): pass >>> class C(A): pass >>> class D(B,C): pass >>> class E(D): pass >>> >>> for cls in itersubclasses(A): ... print(cls.__name__) B D E C >>> # get ALL (new-style) classes currently defined >>> [cls.__name__ for cls in itersubclasses(object)] [...'tuple', ...'type', ...] From http://code.activestate.com/recipes/576949/ s>itersubclasses must be called with new-style classes, not {!r}tkeyRN( t isinstancettypet TypeErrorR Rtsett__subclasses__tsortedtoperatort attrgettertaddtitersubclasses(tclst_seentsubstsub((s3lib/python2.7/site-packages/astropy/io/fits/util.pyR?s    "  c stfd}|S(s This decorator registers a custom SIGINT handler to catch and ignore SIGINT until the wrapped function is completed. c stj}tjdko-|jdk}dtffdY}|}|rstjtj|}nz||Wd|r|dk rtjtj|ntjtjtj|j rt qnXdS(Nit MainThreadt SigintHandlerc s eZdZfdZRS(cS s t|_dS(N(tFalsetsigint_received(R((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt__init__sc s)tjdjjtt|_dS(Ns/KeyboardInterrupt ignored until {} is complete!(twarningstwarnR RRtTrueRG(Rtsignumtframe(tfunc(s3lib/python2.7/site-packages/astropy/io/fits/util.pyt__call__s  (RRRHRO((RN(s3lib/python2.7/site-packages/astropy/io/fits/util.pyREs ( t threadingt currentThreadt activeCounttgetNametobjecttsignaltSIGINTRtSIG_DFLRGtKeyboardInterrupt(R&R't curr_threadt single_threadREtsigint_handlert old_handler(RN(s3lib/python2.7/site-packages/astropy/io/fits/util.pytwrappeds    (R(RNR]((RNs3lib/python2.7/site-packages/astropy/io/fits/util.pyt ignore_sigints%cC s4tj|\}}x|D]}PqWt||S(smReturn the items of an iterable paired with its next item. Ex: s -> (s0,s1), (s1,s2), (s2,s3), .... (t itertoolstteeR (R3RRt_((s3lib/python2.7/site-packages/astropy/io/fits/util.pytpairwises cC s|S(s In Python 2 this is a no-op. Strings are left alone. In Python 3 this will be replaced with a function that actually encodes unicode strings to ASCII bytes. ((ts((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt encode_asciiscC s|S(s In Python 2 this is a no-op. Strings are left alone. In Python 3 this will be replaced with a function that actually decodes ascii bytes to unicode. ((Rc((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt decode_ascii&sc stj r#tdr#jStdrJjrJtdntds]tStdrtfddD rtStS(s Returns True if the file-like object can be read from. This is a common- sense approximation of io.IOBase.readable. treadabletclosedsI/O operation on closed filetreadtmodec3 s|]}|jkVqdS(N(Ri(t.0tc(tf(s3lib/python2.7/site-packages/astropy/io/fits/util.pys @ssr+( RtPY2R"RfRgt ValueErrorRFtanyRK(Rl((Rls3lib/python2.7/site-packages/astropy/io/fits/util.pyt isreadable0s ,c stj r#tdr#jStdrJjrJtdntds]tStdrtfddD rtStS(s Returns True if the file-like object can be written to. This is a common- sense approximation of io.IOBase.writable. twritableRgsI/O operation on closed filetwriteRic3 s|]}|jkVqdS(N(Ri(RjRk(Rl(s3lib/python2.7/site-packages/astropy/io/fits/util.pys Xsswa+( RRmR"RqRgRnRFRoRK(Rl((Rls3lib/python2.7/site-packages/astropy/io/fits/util.pyt iswritableHs ,cC s t|tS(s Returns True if the given object represents an OS-level file (that is, ``isinstance(f, file)``). On Python 3 this also returns True if the given object is higher level wrapper on top of a FileIO object, such as a TextIOWrapper. (R6tfile(Rl((s3lib/python2.7/site-packages/astropy/io/fits/util.pytisfileas cC sRt|tjrtSt|dr2t|jSt|drNt|jStS(s Returns True if the given object represents an OS-level file (that is, ``isinstance(f, file)``). On Python 3 this also returns True if the given object is higher level wrapper on top of a FileIO object, such as a TextIOWrapper. tbuffertraw( R6tiotFileIORKR"RuRvRwRF(Rl((s3lib/python2.7/site-packages/astropy/io/fits/util.pyRuls   cC s t||S(s A wrapper around the `open()` builtin. This exists because in Python 3, `open()` returns an `io.BufferedReader` by default. This is bad, because `io.BufferedReader` doesn't support random access, which we need in some cases. In the Python 3 case (implemented in the py3compat module) we must call open with buffering=0 to get a raw random-access file reader. (topen(tfilenameRi((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt fileobj_opens cC st||ddS(s A wrapper around the `open()` builtin. This exists because in Python 3, `open()` returns an `io.BufferedReader` by default. This is bad, because `io.BufferedReader` doesn't support random access, which we need in some cases. In the Python 3 case (implemented in the py3compat module) we must call open with buffering=0 to get a raw random-access file reader. t bufferingi(Rz(R{Ri((s3lib/python2.7/site-packages/astropy/io/fits/util.pyR|s cC st|tr|St|tjr2t|jSt|drH|jSt|dr^|jSt|drzt |j St t |SdS(s Returns the 'name' of file-like object f, if it has anything that could be called its name. Otherwise f's class or type is returned. If f is a string f itself is returned. tnameR{t __class__N( R6RtgziptGzipFilet fileobj_nametfileobjR"R~R{tstrRR7(Rl((s3lib/python2.7/site-packages/astropy/io/fits/util.pyRs  cC st|trtSt|dr)|jSt|drTt|jdrT|jjSt|drt|jdr|jjStSdS(s Returns True if the given file-like object is closed or if f is a string (and assumed to be a pathname). Returns False for all other types of objects, under the assumption that they are file-like objects with no sense of a 'closed' state. RgRtfpN(R6RRKR"RgRRRF(Rl((s3lib/python2.7/site-packages/astropy/io/fits/util.pytfileobj_closeds ! ! cC st|dr-t|jdr-|j}n_t|drC|jSt|drpt|jdrp|j}nt|dr|}ndSt|S(sm Returns the 'mode' string of a file-like object if such a thing exists. Otherwise returns None. RRit fileobj_modeRN(R"RRRRt_fileobj_normalize_mode(RlR((s3lib/python2.7/site-packages/astropy/io/fits/util.pyRs ! !  cC sw|j}t|tjrH|tjkr.dS|tjkrAdSdSnd|krs|jdd}|d7}n|S(sTakes care of some corner cases in Python where the mode string is either oddly formatted or does not truly represent the file mode. trbtwbt+tN(RiR6RRtREADtWRITERtreplace(RlRi((s3lib/python2.7/site-packages/astropy/io/fits/util.pyRs   cC s\t|dr|jStdk r8t|tjr8tSt|}|rTd|kStSdS(s Returns True if the give file or file-like object has a file open in binary mode. When in doubt, returns True by default. tbinaryRN( R"RRxRR6t TextIOBaseRFRRK(RlRi((s3lib/python2.7/site-packages/astropy/io/fits/util.pytfileobj_is_binarys  c st|tr|j|St|tr~tfdtdDx|D]}dt|*siN(R6Rt translateR tdictRRR(RcRt deletecharsRk((Rs3lib/python2.7/site-packages/astropy/io/fits/util.pyR s" cC sC|r6|j}x!|D]}d|t||jd}fddjfd|DS(s Like :func:`textwrap.wrap` but preserves existing paragraphs which :func:`textwrap.wrap` does not otherwise handle well. Also handles section headers. s c s=tfd|jDr&|Stj|SdS(Nc3 s!|]}t|kVqdS(N(tlen(Rjtl(twidth(s3lib/python2.7/site-packages/astropy/io/fits/util.pys Ds(tallt splitlinesttextwraptfill(tt(R'R(s3lib/python2.7/site-packages/astropy/io/fits/util.pyt maybe_fillCs"c3 s|]}|VqdS(N((Rjtp(R(s3lib/python2.7/site-packages/astropy/io/fits/util.pys Is(tsplittjoin(ttextRR't paragraphs((R'RRs3lib/python2.7/site-packages/astropy/io/fits/util.pyR:sc C s{t|r'tdkr[tjdkrRttjdtdkrRtaq[tantr t d |j }||krt j |d|d|St j |d|}xStd||D]?}t|||}t j |d|d|||||+qW|Sqwt j |d|d|SnPt j|j |}|j|}t j|d|d|}|j}|SdS( s7Create a numpy array from a file or a file-like object.tdarwinis10.9iitdtypetcountNi@(RutCHUNKED_FROMFILERtsystplatformRtmac_verRKRFtinttitemsizetnptfromfiletemptyRtminRRht frombufferR/( tinfileRRt chunk_sizetarraytbegtendt read_sizeRc((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_array_from_fileTs*  "   * ii iicC st|r+t|tj r+d}nt}tjdkrv|jtdkrv|jddkrvt|j }n/tjj drt |j }n |||Sd}|j t jj}x5||jkr|||||!|||7}qWdS(s  Write a numpy array to a file or a file-like object. Parameters ---------- arr : `~numpy.ndarray` The Numpy array to write. outfile : file-like A file-like object such as a Python file object, an `io.BytesIO`, or anything else with a ``write`` method. The file object must support the buffer interface in its ``write``. If writing directly to an on-disk file this delegates directly to `ndarray.tofile`. Otherwise a slower Python implementation is used. cS s |j|S(N(ttofile(RRl((s3lib/python2.7/site-packages/astropy/io/fits/util.pyRsRiiitwinN(RuR6RxtBufferedIOBaset_array_to_file_likeRRtnbytest_OSX_WRITE_LIMITRt startswitht_WIN_WRITE_LIMITtviewRtndarraytflatten(tarrtoutfileRrt chunksizetidx((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_array_to_file~s  " cC s-t|dkrdS|jjrQy|j|jWntk rIqQXdSnttdrxtj|D]}|j|j qpWn|j j }t j dkr|dkst j dkr|dkrxT|j D]}|j|jj qWn'x$|j D]}|j|j q WdS(sp Write a `~numpy.ndarray` to a file-like object (which is not supported by `numpy.ndarray.tofile`). iNtnditertlittlet>tbigt<(Rtflagst contiguousRrRR8R"RRttostringRt byteorderRtflattbyteswap(RRtitemR((s3lib/python2.7/site-packages/astropy/io/fits/util.pyRs$    cC st|}|r0t|tr0t|}nS| rVt|t rVt|}n-t|trt|tjr|j}n|j |dS(s Write a string to a file, encoding to ASCII if the file is open in binary mode, or decoding if the file is open in text mode. N( RR6R RdReRRRRRr(RlRctbinmode((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt _write_strings ! cC st|j|kr|S|jj|jkrctj|jtjoRtj|tj rc|j|S|j|SdS(s Converts an array to a new dtype--if the itemsize of the new dtype is the same as the old dtype and both types are not numeric, a view is returned. Otherwise a new array must be created. N(RRRt issubdtypetnumberRtastype(RR((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_convert_arrays cC s(|jdkstd|jdd>S(sg Given a numpy dtype, finds its "zero" point, which is exactly in the middle of its range. tuii(tkindtAssertionErrorR(R((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_unsigned_zeroscC s|jdko|jdkS(NRi(RR(R((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_is_pseudo_unsignedscC s t|tS(N(R6tall_integer_types(tval((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_is_intscC s4yt|}Wntk r/t|}nX|S(sAConverts a given string to either an int or a float if necessary.(RRntfloat(Rtnum((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt _str_to_nums  c C sOg}|jd}t|t||d}tj|djddtdf}tj|dkd}d}d}xt|D]} yLtj|||kdd} || dd}| dkrd}nWnt k rt|}nX||kr||}n|j |||!t||krAPn|}qW|S(s Split a long string into parts where each part is no longer than ``strlen`` and no word is cut into two pieces. But if there is one single word which is longer than ``strlen``, then it will be split in the middle of the word. t itutf8Rii( RtmaxRRRtencodeR tnonzeroRt Exceptiontappend( tinputtstrlentwordstnblankstnmaxRt blank_loctoffsettxoffsetRtloc((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt _words_group!s,(!      cC sJ|dk r!tjj|}ntjd|\}}tj||S(s Create a temporary file name which should not already exist. Use the directory of the input file as the base name of the mkstemp() output. tdirN(Rtostpathtdirnamettempfiletmkstemptclose(RRltfn((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt _tmp_nameHs   cC sjt|tjr|S|}xGt|dre|jdk ret|jtjrY|jS|j}qWdS(su If the array has an mmap.mmap at base of its base chain, return the mmap object; otherwise return None. tbaseN(R6tmmapR"RR(RR((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_get_array_mmapUs!cc sy dVWntk r}d}t|ts<|g}n|dkrctjj|jj}ntjj |rt j |}t d|D}||krdj ||}qnx|D]}|jqWt|t|nXdS(NRcs s|]}|jVqdS(N(tsize(Rjthdu((s3lib/python2.7/site-packages/astropy/io/fits/util.pys qss6Not enough space on disk: requested {}, available {}. (tOSErrorR6tlistRRRRt_fileR~tisdirRtget_free_space_in_dirtsumR t_closeR(thdulistRtexct error_messaget free_spacet hdulist_sizeR((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_free_space_checkes"     cC s3ytt|SWnttfk r.|SXdS(s Attempts to extract an integer number from the given value. If the extraction fails, the value of the 'default' argument is returned. N(RRR8Rn(tvaluetdefault((s3lib/python2.7/site-packages/astropy/io/fits/util.pyt_extract_number|scC stjdj|dS(s8 Return a string representing the path to the file requested from the io.fits test data set. .. versionadded:: 2.0.3 Parameters ---------- filename : str The filename of the test data file. Returns ------- filepath : str The path to the requested file. sio/fits/tests/data/{}tastropy(Rtget_pkg_data_filenameR (R{((s3lib/python2.7/site-packages/astropy/io/fits/util.pytget_testdata_filepathsIIIi(Wt __future__RRR_RxRR<RRRUtstringRRRRPRIRt contextlibRtutilsRtdistutils.versionRtnumpyRRt ImportErrorRTtexternRt extern.sixRR R R R textern.six.movesR RRt utils.compatRtutils.exceptionsRRmtcmptintegerRRR4RR?R^RbRdReRpRsRuR|RRRRRt maketransRRRRRRRRRRRRRRRRRRRRR(((s3lib/python2.7/site-packages/astropy/io/fits/util.pyts                 (   u  * .           !         & 3 +     '