B q\*s@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZmZddlmZddlmZddlZddlmZddlmZddZeejfZGd d d Zd d ZdOd dZ ddZ!ddZ"ddZ#ddZ$ddZ%ddZ&ddZ'ddZ(dd Z)d!d"Z*d#d$Z+d%d&Z,d'd(Z-d)d*Z.d+d,Z/da0d-d.Z1d/Z2d0Z3d1d2Z4d3d4Z5d5d6Z6d7d8Z7d9d:Z8d;d<Z9d=d>Z:d?d@Z;dAdBZedPdGdHZ?dIdJZ@dKdLZAdMdNZBdS)QN)contextmanagersuppress)data) LooseVersion)wraps)AstropyUserWarningcCs||k||kS)N)abrr3lib/python3.7/site-packages/astropy/io/fits/util.pysr cs<eZdZdZdZddZddZddZfd d ZZ S) NotifierMixina 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: ... 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 NcCs&|jdkrt|_||jt|<dS)z Add an object to the list of listeners to notify of changes to this object. This adds a weakref to the list of listeners that is removed from the listeners list when the listener has no other references to it. N) _listenersweakrefWeakValueDictionaryid)selflistenerrrr _add_listenerPs  zNotifierMixin._add_listenerc Cs2|jdkrdStt|jt|=WdQRXdS)z Removes the specified listener from the listeners list. This relies on object identity (i.e. the ``is`` operator). N)rrKeyErrorr)rrrrr _remove_listener]s  zNotifierMixin._remove_listenercOsf|jdkrdSd|}xH|jD]:}|}|dkr8q$t||r$t||}t|r$|||q$WdS)aA Notify all listeners of some particular state change by calling their ``_update_`` 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. Nz _update_{0})rformat valuerefshasattrgetattrcallable)rZ notificationargskwargsZ method_namermethodrrr _notifyis    zNotifierMixin._notifycs:yt}Wntk r,|j}YnXd|d<|S)zj Exclude listeners when saving the listener's state, since they may be ephemeral. Nr)super __getstate__AttributeError__dict__copy)rstate) __class__rr r!s zNotifierMixin.__getstate__) __name__ __module__ __qualname____doc__rrrrr! __classcell__rr)r&r r !s +  r cCs tt|S)z Returns the first item returned by iterating over an iterable object. Example: >>> a = [1, 2, 3] >>> first(a) 1 )nextiter)iterablerrr firsts r/ccs|dkrt}y |}Wntk r8||}YnXxLt|tddD]6}||krN|||Vxt||D] }|VqvWqNWdS)a Generator over all subclasses of a given class, in depth first order. >>> class A: 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 classes currently defined >>> [cls.__name__ for cls in itersubclasses(object)] [...'tuple', ...'type', ...] From http://code.activestate.com/recipes/576949/ Nr')key)set__subclasses__ TypeErrorsortedoperator attrgetteradditersubclasses)cls_seenZsubssubrrr r8s  r8cstfdd}|S)z This decorator registers a custom SIGINT handler to catch and ignore SIGINT until the wrapped function is completed. c st}tdko|dk}Gfddd}|}|rJttj|}z||Wd|r|dk rvttj|nttjtj|jrtXdS)NZ MainThreadcs eZdZddZfddZdS)z5ignore_sigint..wrapped..SigintHandlercSs d|_dS)NF)sigint_received)rrrr __init__sz>ignore_sigint..wrapped..SigintHandler.__init__cstdjtd|_dS)Nz/KeyboardInterrupt ignored until {} is complete!T)warningswarnrr'rr=)rZsignumframe)funcrr __call__sz>ignore_sigint..wrapped..SigintHandler.__call__N)r'r(r)r>rCr)rBrr SigintHandlersrD) threadingZ currentThreadZ activeCountZgetNamesignalSIGINTSIG_DFLr=KeyboardInterrupt)rrZ curr_threadZ single_threadrDZsigint_handlerZ old_handler)rBrr wrappeds   zignore_sigint..wrapped)r)rBrJr)rBr ignore_sigints%rKcCs(t|\}}x|D]}PqWt||S)zmReturn the items of an iterable paired with its next item. Ex: s -> (s0,s1), (s1,s2), (s2,s3), .... ) itertoolsteezip)r.r r _rrr pairwises rPcCst|tr|dSt|tjrxt|jjtjrxtj |d t|}|jj |jj dkrt| tj |jj df}|St|tjrt|jjtj std|S)Nasciiz$string operation on non-string array) isinstancestrencodenpndarray issubclassdtypetypestr_charviewitemsizeastypebytes_r3)snsrrr encode_asciis    rccCst|trNy |dStk rJtdt|jddd}|ddSXnt|tj rt |j j tj r|jdkr|j jdd }tjg|d t |}ntj|dt |}|j jd |j jkr|tj|j jf}|St|tj r t |j j tjs td |S) NrQzanon-ASCII characters are present in the FITS file header and have been replaced by "?" charactersreplace)errorsu�?rSU)rYrRz$string operation on non-string array)rSbytesdecodeUnicodeDecodeErrorr?r@rrdrVrWrXrYrZr`sizerTarrayr]r\r^r_r[r3)radtrbrrr decode_asciis*    rocs`tdrStdr*jr*tdtds8dStdr\tfddd Ds\dSd S) z Returns True if the file-like object can be read from. This is a common- sense approximation of io.IOBase.readable. readableclosedzI/O operation on closed filereadFmodec3s|]}|jkVqdS)N)rs).0c)frr Hszisreadable..zr+T)rrprq ValueErrorany)rvr)rvr isreadable8s   rzcs`tdrStdr*jr*tdtds8dStdr\tfddd Ds\dSd S) z Returns True if the file-like object can be written to. This is a common- sense approximation of io.IOBase.writable. writablerqzI/O operation on closed filewriteFrsc3s|]}|jkVqdS)N)rs)rtru)rvrr rw`sziswritable..zwa+T)rr{rqrxry)rvr)rvr iswritablePs   r}cCs<t|tjrdSt|dr$t|jSt|dr8t|jSdS)z 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. TbufferrawF)rSioFileIOrisfiler~r)rvrrr rhs     rcCst||ddS)a: A wrapper around the `open()` builtin. This exists because `open()` returns an `io.BufferedReader` by default. This is bad, because `io.BufferedReader` doesn't support random access, which we need in some cases. We must call open with buffering=0 to get a raw random-access file reader. r) buffering)open)filenamersrrr fileobj_openzs rcCsht|tr|St|tjr$t|jSt|dr4|jSt|drD|jSt|drXt|j Stt |SdS)z 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. namerr&N) rSrTgzipGzipFile fileobj_namefileobjrrrr&rZ)rvrrr rs       rcCsbt|trdSt|dr|jSt|dr.maybe_fill..)all splitlinestextwrapfill)t)rrrr maybe_fillszfill..maybe_fillc3s|]}|VqdS)Nr)rtp)rrr rwszfill..)splitjoin)textrrZ paragraphsr)rrrr rs rc Cst|rtdkrt d||D].}t |||}tj ||||d|||<qW|Sntj |||dSn6t |j|}||}tj|||d}|}|SdS) z7Create a numpy array from a file or a file-like object.Ndarwinrz10.9TFi@)rYcount)rY)rCHUNKED_FROMFILEsysplatformrZmac_verintr^rVZfromfileemptyrangeminrYrr frombufferr$) ZinfilerYrZ chunk_sizermZbegendZ read_sizerarrr _array_from_file*s*   rlicCst|rt|tjsdd}nt}tjdkrT|jtdkrT|jddkrTt|j }n"tj drlt |j }n |||Sd}| t j}x,||jkr||||||||7}qWdS) a  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. cSs ||S)N)Ztofile)r rvrrr r fsz _array_to_file..rr<irwinN)rrSrBufferedIOBase_array_to_file_likerrnbytes_OSX_WRITE_LIMITr^ startswith_WIN_WRITE_LIMITr]rVrWZflatten)arrZoutfiler|Z chunksizeidxrrr _array_to_fileTs       rcCst|dkrdS|jjrBy||jWntk r<YnXdSttdrtxtj|ddD]}|| q\Wnn|j j }t j dkr|dkst j dkr|d krx@|j D]}|| qWnx|j D]}|| qWdS) zp Write a `~numpy.ndarray` to a file-like object (which is not supported by `numpy.ndarray.tofile`). rNnditerC)orderlittle>Zbig<)rflags contiguousr|rr3rrVrZtostringrY byteorderrZflatZbyteswap)rritemrrrr rs$    rcCsDt|}|r t|tr t|}n|s6t|ts6t|}||dS)z 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)rrSrTrcror|)rvraZbinmoderrr _write_strings  rcCsR|j|kr|S|jj|jkrDt|jtjr:t|tjsD||S||SdS)z 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)rYr^rVZ issubdtypeZnumberr]r_)rmrYrrr _convert_arrays  rcCs |jdkstd|jdd>S)zg Given a numpy dtype, finds its "zero" point, which is exactly in the middle of its range. ur<)kindAssertionErrorr^)rYrrr _unsigned_zerosrcCs|jdko|jdkS)Nr)rr^)rYrrr _is_pseudo_unsignedsrcCs t|tS)N)rSall_integer_types)valrrr _is_intsrcCs.y t|}Wntk r(t|}YnX|S)zAConverts a given string to either an int or a float if necessary.)rrxfloat)rZnumrrr _str_to_nums  rc Csg}|d}t|t||d}tj|ddtdfd}t|dkd}d}d}xt|D]} y:t|||kdd} || dd}| dkrd}Wnt k rt|}YnX||kr||}| |||t||krP|}qdW|S)z 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.  r<utf8)rY r) rmaxrrVrrUriZnonzeror Exceptionappend) inputZstrlenZwordsZnblanksZnmaxrZ blank_locoffsetZxoffsetrZlocrrr _words_groups,  rcCs2|dk rtj|}tj|d\}}t||S)z 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. N)dir)ospathdirnametempfileZmkstempclose)rrvfnrrr _tmp_names   rcCsLt|tjr|S|}x2t|drF|jdk rFt|jtjr>|jS|j}qWdS)zu If the array has an mmap.mmap at base of its base chain, return the mmap object; otherwise return None. baseN)rSmmaprr)rmrrrr _get_array_mmap(s rc csy dVWntk r}zd}t|ts0|g}|dkrHtj|jj}tj|rt |}t dd|D}||krd ||}x|D] }| qWt|t|Wdd}~XYnXdS)Nrcss|] }|jVqdS)N)rl)rthdurrr rwDsz$_free_space_check..z6Not enough space on disk: requested {}, available {}. )OSErrorrSlistrrrZ_filerisdirrZget_free_space_in_dirsumrZ_closerT)ZhdulistrexcZ error_messageZ free_spaceZ hdulist_sizerrrr _free_space_check8s"      rc Cs*y tt|Sttfk r$|SXdS)z Attempts to extract an integer number from the given value. If the extraction fails, the value of the 'default' argument is returned. N)rrr3rx)valuedefaultrrr _extract_numberOs rcCstd|dS)a8 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. zio/fits/tests/data/{}Zastropy)rZget_pkg_data_filenamer)rrrr get_testdata_filepath]src Cs$|j}|jdkrtd|jdkr&dnd}d|j||j|}||tj}t }|j dkryd|j df|_ Wnt k rYnXxt d |j d |D]~}||||}tj|j d dtd }xPt d|j d dD]8} ||d | fd kM}d |d | f|<|d | fd k}qWqW|S)z Performs an in-place rstrip operation on string arrays. This is necessary since the built-in `np.char.rstrip` in Numpy does not perform an in-place calculation. ZSUz/This function can only be used on string arraysrgr<rRz {0}{1}u{2}rrrN)rY. )rYrr3rr^rr]rVrWZ getbufsizendimshaper"rZonesbool) rmrnZbpcZdt_intr bufsizejrumaskirrr _rstrip_inplacers(  r)N)N)CrrLrrr5rrrFrrrrEr?r contextlibrrZ astropy.utilsrZdistutils.versionrZnumpyrVrZastropy.utils.exceptionsrZcmprZintegerrr r/r8rKrPrcrorzr}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr sv     u %.$ !&3+  '