HR-etddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZmZddlmZddlZddlmZddlmZddlmZeeejfZdZe ej!fZ"Gdd Z#d Z$d0d Z%d Z&ej'dd dkrddlm(Z(ndZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4da5dZ6dZ7dZ8d Z9d!Z:d"Z;d#Zd&Z?d'Z@d(ZAd)ZBd*ZCed0d+ZDd,ZEd-ZFd.ZGd/ZHdS)1N)contextmanagersuppresswraps)Version)data)AstropyUserWarningc||k||kz SN)abs 4lib/python3.11/site-packages/astropy/io/fits/util.pyrsAEa!e$c8eZdZdZdZdZdZdZfdZxZ 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 Ncr|jtj|_||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) _listenersweakrefWeakValueDictionaryidselflisteners r _add_listenerzNotifierMixin._add_listenerOs4 ? "%9;;DO(08 %%%rc|jdStt5|jt|=ddddS#1swxYwYdS)z Removes the specified listener from the listeners list. This relies on object identity (i.e. the ``is`` operator). N)rrKeyErrorrrs r_remove_listenerzNotifierMixin._remove_listener[s ? " F h   . .8 - . . . . . . . . . . . . . . . . . .sAAAc|jdSd|}|jD]F}|}|t||r't||}t |r||i|GdS)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. N_update_)r valuerefshasattrgetattrcallable)r notificationargskwargs method_namermethods r_notifyzNotifierMixin._notifyfs ? " F/// 1133 , ,H xzzHx-- , ;77F##,FD+F+++ , ,rc t}n)#t$r|j}YnwxYwd|d<|S)zj Exclude listeners when saving the listener's state, since they may be ephemeral. Nr)super __getstate__AttributeError__dict__copy)rstate __class__s rr.zNotifierMixin.__getstate__sb )GG((**EE ) ) )M&&((EEE )#l s $#A  A ) __name__ __module__ __qualname____doc__rrrr+r. __classcell__)r3s@rrr!sw))VJ 1 1 1 . . .,,,2rrc:tt|S)z Returns the first item returned by iterating over an iterable object. Examples -------- >>> a = [1, 2, 3] >>> first(a) 1 )nextiter)iterables rfirstr=s X  rc#TK|t} |}n%#t$r||}YnwxYwt|t jdD]6}||vr0|||Vt||D]}|V7dS)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/ Nr4)key)set__subclasses__ TypeErrorsortedoperator attrgetteradditersubclasses)cls_seensubssubs rrGrGs, }'!!## '''!!#&&'d 3J ? ?@@@ e   IIcNNNIII%c511   s)A  A c<tfd}|S)z This decorator registers a custom SIGINT handler to catch and ignore SIGINT until the wrapped function is completed. crtj}tjdko |jdk}Gfdd}|}|rt jtj|} |i||rY| t jtj|n)t jtjtj|jr tdSdS#|rY| t jtj|n)t jtjtj|jrtwxYw)N MainThreadc eZdZdZfdZdS)5ignore_sigint..wrapped..SigintHandlercd|_dS)NF)sigint_received)rs r__init__z>ignore_sigint..wrapped..SigintHandler.__init__s',$$$rc\tjdjdtd|_dS)Nz KeyboardInterrupt ignored until z is complete!T)warningswarnr4r rS)rsignumframefuncs r__call__z>ignore_sigint..wrapped..SigintHandler.__call__s: St}SSS&(,$$$rN)r4r5r6rTr[)rZsr SigintHandlerrQs= - - - , , , , , , ,rr\) threadingcurrent_thread active_countnamesignalSIGINTSIG_DFLrSKeyboardInterrupt)r'r( curr_thread single_threadr\sigint_handler old_handlerrZs rwrappedzignore_sigint..wrappedsh .00  " $ $ ) Nk.>,.N  , , , , , , , , , ,'  G - ~FFK , D$ !& ! ! ! ,*M&-====M&-@@@!1,++ , , ,,  ,*M&-====M&-@@@!1,++++++s 2CAD6r)rZris` r ignore_sigintrjs6  4[[$,$,$,$,[$,L Nr) )pairwisecZtj|\}}|D]}t||S)zuReturn the items of an iterable paired with its next item. Ex: s -> (s0,s1), (s1,s2), (s2,s3), .... ) itertoolsteezip)r<r r_s rrnrns9 }X&&1  A 1ayyrct|tr|dSt|tjrt |jjtjrtj |d t|}|jj |jj dz kr.| tj |jj dz f}|St|tjr3t |jjtj std|S)Nascii$string operation on non-string array) isinstancestrencodenpndarray issubclassdtypetypestr_charviewitemsizeastypebytes_rB)snss r encode_asciirs!S@xx   Arz " "@z!','H'H@ W^^Aw ' ' , ,T!WW 5 5 8  01 4 4 4BIqw'7!';<==B Arz " "@:aglBI+N+N@>??? Hrct|trm |dS#t$rJt jdt |dd}|ddcYSwxYwt|tj rt|j j tj r|jdkrW|j jdd }tjg| t|}n@tj|dt|}|j jd z |j jkr+|tj|j jf}|St|tj r3t|j j tjst-d |S) Nruzanon-ASCII characters are present in the FITS file header and have been replaced by "?" charactersreplace)errorsu�?rSUr~rvrw)rxbytesdecodeUnicodeDecodeErrorrVrWr rr{r|r}r~rrsizeryarrayrrrrrrB)rdtrs r decode_asciirs!U@ ,88G$$ $! , , , MG"    33A99Xs++ + + + , Arz " "@z!', 'J'J@ 6Q;;$$S#..B"B''',,T!WW55BB7++00a99B 8 q AG$4 4 4BGQW%5677B Arz " "@:aglBG+L+L@>??? Hs,AB?BctdrStdrjrtdtdsdStdrt fddDsdSd S) z Returns True if the file-like object can be read from. This is a common- sense approximation of io.IOBase.readable. readableclosedI/O operation on closed filereadFmodec3*K|] }|jvVdSr r.0cfs r zisreadable..Cs)%@%@aa16k%@%@%@%@%@%@rzr+T)r#rr ValueErroranyrs`r isreadabler4s q*zz||q(997888 1f  uq&#%@%@%@%@4%@%@%@"@"@u 4rctdrStdrjrtdtdsdStdrt fddDsdSd S) z Returns True if the file-like object can be written to. This is a common- sense approximation of io.IOBase.writable. writablerrwriteFrc3*K|] }|jvVdSr rrs rrziswritable..Zs)%A%Aaa16k%A%A%A%A%A%Arzwa+T)r#rrrrrs`r iswritablerKs q*zz||q(997888 1g  uq&#%A%A%A%A5%A%A%A"A"Au 4rct|tjrdSt|drt |jSt|drt |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)rxioFileIOr#isfilerrrs rrrbsd!RYt H  ah E  ae}} 5rcvt|ttfr|St|tjrt |jSt|dr|jSt|dr|j St|drt|j Stt|S)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. r`filenamer3) rxryrgzipGzipFile fileobj_namefileobjr#r`rr3rrs rrrss !c5\"" At} % %AI&&& F  v J  z K 1;477||rc&t|trdSt|dr|jSt|dr!t|jdr |jjSt|dr!t|jdr |jjSdS)a 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. TrrfpF)rx path_liker#rrrrs rfileobj_closedrs!Ytq(x I  719h#?#?y D  gadH55t{urc,t|drt|jdr|j}nYt|dr|jSt|drt|jdr|j}nt|dr|}ndSt |S)zm Returns the 'mode' string of a file-like object if such a thing exists. Otherwise returns None. rr fileobj_moderN)r#rrr_fileobj_normalize_mode)rrs rrrsq)F!;!;) N # # ~ D   gadF33 $ F  t "7 + ++rc|j}t|tjr&|tjkrdS|tjkrdSdSd|vr|dd}|dz }|S)zTakes care of some corner cases in Python where the mode string is either oddly formatted or does not truly represent the file mode. rbwbN+)rrxrrREADWRITErrrs rrrsv 6D!T]## 49  4 TZ  44 d{{||C$$   Krct|dr|jSt|tjrdSt |}|rd|vSdS)z Returns True if the give file or file-like object has a file open in binary mode. When in doubt, returns True by default. binaryFrT)r#rrxr TextIOBaserrs rfileobj_is_binaryrsXq(x!R]##u ??D d{trc|r+|}|D]}d|t|<||Sr )r1ord translate)rtable deletecharsrs rrrsI!  ! !A E#a&&MM ;;u  rc |d}fddfd|DS)z Like :func:`textwrap.wrap` but preserves existing paragraphs which :func:`textwrap.wrap` does not otherwise handle well. Also handles section headers. z ctfd|Dr|Stj|fiS)Nc3>K|]}t|kVdSr )len)rlinewidths rrz+fill..maybe_fill..s.<.maybe_fillsO <<<.s+99zz!}}999999r)splitjoin)textrr( paragraphsrs `` @rrrs_ F##J555555 ;;9999j999 9 99rct|rtLtjdkr:t tjdt dkrdandatrt d|jz }||krtj |||Stj || }td||D]4}t|||z}tj ||||z |||<5|Stj |||Stj |j|z}||}tj|||f }|}|S) z7Create a numpy array from a file or a file-like object.Ndarwinrz10.9TFi@)r~countr)rr~shape)rCHUNKED_FROMFILEsysplatformrmac_verintrr{fromfileemptyrangeminr~rr|r1) infiler~r chunk_sizerbegend read_sizers r_array_from_filers f~~   #|x''GH4D4F4Fq4I,J,JWNN--$(  #(  AWu~566Jz!!{6eDDDDe444 E:66WWCeS:%566C%'[uCRUI%V%V%VE#c'NN ;vU%@@@ @ HUOO,u4 KK " " !5AAA  rlicT |}n#t$rd}YnwxYwt|r|rd}nt}tjdkr1|jtdzkr|jdzdkrt|jz}n;tj drt|jz}n |||Sd}| tj }||jkr)|||||z|||z }||jk'dSdS) a Write a numpy array to a file or a file-like object. Parameters ---------- arr : 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. Fc,||Sr )tofile)r rs rrz _array_to_file..PsQXXa[[rrrNirwinN)seekabler/r_array_to_file_likerrnbytes_OSX_WRITE_LIMITr startswith_WIN_WRITE_LIMITrr{r|flatten)arroutfilerr chunksizeidxs r_array_to_filer:sZ ##%% g$8$((#   J*Q. . . J  " "% 4   ' '#$ 4 uS'""" C ((2:   & & ( (C    c#i'('222 y       s  &&ct|dkrdS|jjr- ||jdS#t $rYnwxYwt tdrBtj|dD])}|| *dS|j j }tj dkr|dkstj dkrK|d krE|j D];}|| big<)rflags contiguousrrrBr#r{rtobytesr~ byteorderrflatbyteswap)rritemrs rrrtsx 3xx1}} y    MM#( # # # F    D  r8.Ic--- * *D MM$,,.. ) ) ) ) * * I' MX % %)s*:*: MU " "yC'7'7 9 9 dmmoo55778888 9 9 . . dllnn---- . .s? A  A ct|}|r%t|trt|}n&|s$t|tst |}||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)rrxryrrr)rrbinmodes r _write_stringrsn ""G:a%% OO Z3// OOGGAJJJJJrc |j|kr|S|jj|jkrXtj|jtjrtj|tjs||S||S)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. )r~rr{ issubdtypenumberrr)rr~s r_convert_arrayrs|  {e   / / ek29-- 024-ry2Q2Q 0 zz%   ||E"""rcl|jdkr |jdkrdS|jdksJd|jdzdz zS)zg Given a numpy dtype, finds its "zero" point, which is exactly in the middle of its range. irNiukindrrs r _pseudo_zeror"sL  zSU^q00t :     !#a' ((rcZ|jdkr |jdkp|jdko |jdkS)NrrkrrNr rs r_is_pseudo_integerr$s8 J#  5%.A"5  c1en1rc,t|tSr )rxall_integer_types)vals r_is_intr(s c, - --rcf t|}n#t$rt|}YnwxYw|S)zAConverts a given string to either an int or a float if necessary.)rrfloat)r'nums r _str_to_numr,sE#hh Cjj Js ..cg}t|}tj|ddzd}tj|dkd}d}d} tj|||zkdd}n#t $rt|}YnwxYw|dkr||dz dz}nd}||krt ||z|}||||||krn|}|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 are any single words which are longer than ``strlen``, then they will be split in the middle of the word. utf8 S1rrTrN)rr{ frombufferrznonzero IndexErrorrappend) rrwordsslenr blank_locoffsetxoffsetlocs r _words_groupr<s6 E q66D -((4/t < < > '* Ls"'B B&%B&c|tj|}tj|\}}tj||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)ospathdirnametempfilemkstempclose)inputrfns r _tmp_namerG sF  &&   ' ' 'EArHQKKK Irct|tjr|S|}t|drM|jHt|jtjr|jS|j}t|dr |jDdSdSdSdS)zu If the array has an mmap.mmap at base of its base chain, return the mmap object; otherwise return None. baseN)rxmmapr#rI)rrIs r_get_array_mmaprKs %## D $  DI$9 di + + 9 y $  DI$9$9$9$9$9rc#K dVdS#t$r}d}t|ts|g}|)tj|jj}tj|rItj |}td|D}||krd ||}|D]}| t|t|zd}~wwxYw)Nrc3$K|] }|jV dSr )r)rhdus rrz$_free_space_check..4s$;;Csx;;;;;;rz6Not enough space on disk: requested {}, available {}. )OSErrorrxlistr?r@rA_filer`isdirrget_free_space_in_dirsumformat_closery)hdulistrAexc error_message free_space hdulist_sizerNs r_free_space_checkr\(s0  000 '4(( iG ?googm&899G 7== ! ! 3G<C%C99C>cn tt|S#ttf$r|cYSwxYw)z Attempts to extract an integer number from the given value. If the extraction fails, the value of the 'default' argument is returned. )rr,rBr)valuedefaults r_extract_numberr`BsH ;u%%&&& z "s 44c2tjd|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/astropy)rget_pkg_data_filename)rs rget_testdata_filepathrdOs "  %&FH&F&F R RRrc|j}|jdvrtd|jdkrdnd}d|j|zd|jd|}||t j}t j}|j d kr& d |j d f|_ n#t$rYnwxYwtd |j d |D]}||||z}t j |j d d t }td |j d  d D].} ||d| fdkz}d |d| f|<|d| fd k}/|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. SUz/This function can only be used on string arraysrrNrv(z,)rrkr1rNr. )r~r!rBrrrr{r| getbufsizendimrr/ronesbool) rrbpcdt_intrbufsizejrmaskrs r_rstrip_inplacerrcs B wdIJJJw#~~!!1C ;# ; ;r| ; ;c ; ;F 62:&&AmooG vzz !'"+oAGG    D  1agaj' * * " " a!g+o wqwss|4000 rAGBK<,, " "A Ac1fIO #DAc1fIdOS!V9>DD " LsB B('B(cvt|dsdS ddlm}t||S#t$rYdSwxYw)zCheck whether data is a dask array. We avoid importing dask unless it is likely it is a dask array, so that non-dask code is not slowed down. computeFr)Array)r# dask.arrayrurx ImportError)rrus r_is_dask_arrayrxsj 4 # #u'$$$$$$ $&&& uus * 88r )IrrrprJrDr?rrarrBrr]rVr contextlibrr functoolsrnumpyr{packaging.versionr astropy.utilsrastropy.utils.exceptionsr ryrPathLikercmprintegerr&rr=rGrj version_infornrrrrrrrrrrrrrrrrrrrrr"r$r(r,r<rGrKr\r`rdrrrxr rrrs    ////////%%%%%%777777 % % $$"*%mmmmmmmm`    !!!!H---`BQB7"""""""""       ! ! ! H.."6*,,,@2(:::,"""J777t'.'.'.T"###$ ) ) ) ...(((V      00002   SSS(...b'''''r