Kg'=HUdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z mZmZddlmZddlmZddlZddlm Zdd lmZerdd lmZd d ej8d dZej<Zde d<ejBjEejFdk\rdZ$nQejBjEejFdkrdZ$n& ejJdgjMddZ$d-dZ(d.dZ)d/dZ*e$f d0dZ+d1dZ, d2 d3dZ-d4dZ.d5d6dZ/d7dZ0iZ1de d<d5d8d Z2d9d!Z3d:d;d"Z4ejjfdd+Z=e>d,k(re=yy#e'$rdZ$YwxYw)?zUtility functions.) annotationsN) perf_counter)AnyLiteralTextIO TYPE_CHECKING)Path)Callable)array_of_flavor)Atombiglittle irrelevant)><=| bool | Nonecopy_if_neededz2.0.0z1.28.0F)copyc|dvry|S)z2Fix the byteorder depending on the PyTables types.)stringboolint8uint8objectr)ptype byteorders ,lib/python3.12/site-packages/tables/utils.pycorrect_byteorderr"2s ==ct|turyt|drZt|dr|jdk7ry |j t |t rtjdtdyt |tjryt |tjr,|jdk(r|jjd d k(ryy#t$rYywxYw) z/Check if an object can work as an index or not.T __index__shaperFzKusing a boolean instead of an integer will result in an error in the future) stacklevelr i)typeinthasattrr&r% isinstancerwarningswarnDeprecationWarning TypeErrornpintegerndarraydtypestrindexs r!is_idxr9:s E{c  $ 5' "u{{b'8  OO %& *&   E2:: & 5"**% [[B  KKOOA # %   sclt|d}|jdk(r$tj||j|}|S|jdt |jj }tj |d|jfg}||ddd|j|j}|S)AConvert a generic object into a NumPy object compliant with atom.numpyr)r5rN)r5f0)r r&r2arrayr5lenemptyview)arratomrnparrnewshapenparr2s r!convert_to_np_atomrLjs C )E zzRdjjt< L ;;7#djj&6&6"7!78(B +;*<=t Q DJJ' Lr#c|jdvrdnt}t|||}t|jj }|dvr#|t j k7r|j}|S)r?)time64T)rr)r*rrL byteordersr5r sysbyteswap)objrHrrIr s r!convert_to_np_atom2rSs_99 *4D sD$ /E5;;001I%%)s}}*D  Lr#cft|j}|dk(rtj|tjst d|d|j std|dtj|tjstd|dy|dk(rtj|tjr t|dytj|jtjst d|jd|jjstd|jd tj|jtjstd |jd y|d k(r>tj|tjr t|dyt|dy|dk(r@t|dtj|tjstd|d ytd |)aCheck for file access in the specified `mode`. `mode` is one of the modes supported by `File` objects. If the file indicated by `filename` can be accessed using that `mode`, the function ends successfully. Else, an ``IOError`` is raised explaining the reason of the failure. All this paraphernalia is used to avoid the lengthy and scaring HDF5 messages produced when there are problems opening a file. No changes are ever made to the file system. rz``z`` does not existz`` is not a regular filezfile ``z `` exists but it can not be readwzr+z`` is not a directoryz directory ``z#`` exists but it can not be writtenazinvalid mode: N)r resolveosaccessF_OKFileNotFoundErroris_fileIsADirectoryErrorR_OKPermissionErrorcheck_file_accessparentis_dirNotADirectoryErrorW_OK ValueError)filenamemodepaths r!raras > ! ! #D s{yyrww'#b.?$@A A||~#b.F$GH Hyyrww'!$?@ (  99T277 # dD )99T[["''2'"T[[M9J(KLL;;%%'( %:;99T[["''2%"4;;-03  99T277 # dD ) dC ( $$yyrww'!$BC ( >$233r#cZjfd}t|ddjS)aCreate a *lazy attribute* from the result of `fget`. This function is intended to be used as a *method decorator*. It returns a *property* which caches the result of calling the `fget` instance method. The docstring of `fget` is used for the property itself. For instance: >>> class MyClass(object): ... @lazyattr ... def attribute(self): ... 'Attribute description.' ... print('creating value') ... return 10 ... >>> type(MyClass.attribute) >>> MyClass.attribute.__doc__ 'Attribute description.' >>> obj = MyClass() >>> obj.__dict__ {} >>> obj.attribute creating value 10 >>> obj.__dict__ {'attribute': 10} >>> obj.attribute 10 >>> del obj.attribute Traceback (most recent call last): ... AttributeError: ... .. warning:: Please note that this decorator *changes the type of the decorated object* from an instance method into a property. cL|j}|vr|S|x|<}|SN)__dict__)selfmydictvaluefgetnames r!newfgetzlazyattr..newfgets4 6>$< #Dz)t u r#N)__name__property__doc__)rqrsrrs` @r!lazyattrrws+P ==D GT4 66r#cVtdjjD]}|jdrt |j d}2|jdrt |j d}`|jdrt |j d}|jdrt |j d}|jdrt |j d}|jdst |j d} t d |zt d d d d d t dd dd d t dd d d d t} t d| |z d| S)z2Show the used memory (only works for Linux 2.6.x).z/proc/self/statuszVmSize:r zVmRSS:zVmData:zVmStk:zVmExe:zVmLib:z Memory usage: ******* %s *******zVmSize: z>7z kB VmRSS: z kBzVmData: z kB VmStk: zVmExe: z kB VmLib: zWallClock time: z.3f)r read_text splitlines startswithr+splitprintclock) explaintrefencodinglinevmsizevmrssvmdatavmstkvmexevmlibtnows r! show_statsr si()335@@B ??9 %a)F __X & Q(E __Y 'a)F __X & Q(E __X & Q(E __X & Q(EC ,w 67 HVBK|E":S 9: HVBK|E":S 9: HU2Jl5*C 89 7D TD[- ./ Kr#c| }|dkrtj|ntj|}tjtjd| z}d|z}t j ||z|z }|S)zQuantize data to improve compression. Data is quantized using around(scale*data)/scale, where scale is 2**bits, and bits is determined from the least_significant_digit. For example, if least_significant_digit=1, bits will be 4. r r')mathfloorceillog2r2around)dataleast_significant_digitexpbitsscaledatouts r!quantizer#sk # "C 1W$**S/$))C.C 99TYYrC4x( )D tGE YYut| $u ,F Mr#z&dict[str, list[weakref.ReferenceType]]tracked_classesc|S|jj}|tvr gt|<t|jt j |yy)zLog instance creation.N) __class__rtrappendweakrefref)instancerrs r!log_instance_creationr:sK |!!**  &$&OD !$$W[[%:; r#cN|dk(rtt}|S|jS)z=Return the list of tracked classes matching the input string.*)sortedrr|)scs r!string_to_classesrCs$Cx ? #wwyr#cft|}|Dcgc]}|tt|fc}Scc}w)z$Return the list of logged instances.)rrDr)classes classnamescns r!fetch_logged_instancesrLs3"7+J5? @ZrR_R() *Z @@ @s.c rt|D])}|j|dtt|d+y)z-Write to file the number of logged instances.z:  N)rwriterDr)rfile classnames r!count_logged_instancesrRs5&w/  i[3y'A#B"C2FG0r#ct|D]L}|jd|dt|D])}|}| |jdt|z+Ny)z+Write to file the list of loggen instances.r: Nz %s )rrrrepr)rrrrrRs r!list_logged_instancesrXsV&w/  R {#&'"9-C%C :S 12.0r#c t|D]~}|jd|dt|D][}|}| |jd|z|jj D]\}}|jd|dd|d ]y)zDump the logged instances.rrNz %s: z z>20z : )rrrrmitems)rrrrrRkeyrps r!dump_logged_instancesrbs&w/  R {#&'"9-C%C ;,-"%,,"4"4"6JCJJ#c#eWB?@#7 .0r#c0eZdZdZdfd Zdfd ZxZS) CacheDictz8A dictionary that prevents itself from growing too much.c2||_t| |yrl) maxentriessuper__init__)rnrrs r!rzCacheDict.__init__ts$ r#ct||jkDr1|jdz }t|d|D]}t||t|||y)Nr)rDrlistr __delitem__ __setitem__)rnrrpentries_to_removekrs r!rzCacheDict.__setitem__xsU t9t & $" 4 $Z 2!23#A&4 C'r#rr+returnNone)rr6rprrr)rt __module__ __qualname__rvrr __classcell__)rs@r!rrqsB((r#rcReZdZdZd dZd dZd dZd dZd dZddZ ddd Z dd Z y) NailedDictz=A dictionary which ignores its items when it has nails on it.c.||_i|_d|_yNr)r_cache _nailcount)rnrs r!rzNailedDict.__init__s$ r#c8|jjy)zClear teh dictionsry.N)rclearrns r!rzNailedDict.clears r#c.|xjdz c_y)zIncrease the nail count.r Nrrs r!nailzNailedDict.nail 1r#c.|xjdzc_y)zDecrease the nail count.r Nrrs r!unnailzNailedDict.unnailrr#c>|jdkDry||jvS)NrF)rrrnrs r! __contains__zNailedDict.__contains__s ??Q dkk!!r#cT|jdkDr t||j|Sr)rKeyErrorrrs r! __getitem__zNailedDict.__getitem__s' ??Q 3- {{3r#Nc\|jdkDr|S|jj||S)z'Return the value for the specified key.r)rrget)rnrdefaults r!rzNailedDict.gets( ??Q N{{sG,,r#c|jdkDry|j}t||jkDr/t |jdzd}t |d|D]}||=|||<y)Nrrr )rrrDrmaxr)rnrrpcacherrs r!rzNailedDict.__setitem__sg ??Q    u: ' #DOOr$91 = %[!3"34!H5c r#rrr)rrrr)rrrrrl)rrrz Any | Nonerr)rrrprrr) rtrrrvrrrrrrrrrr#r!rrs-G"  - r#rchttdrmdtjvr,tjd}t |t r6|dkDr1|St tj ddjSdtjvr#t tjd}|dkDr|Sy)z'Detect the number of cores on a system.sysconfSC_NPROCESSORS_ONLNrzsysctl -n hw.ncpur NUMBER_OF_PROCESSORS) r,rY sysconf_namesrr-r+popen2readenviron)ncpuss r!detect_number_of_coresrsr9 B$4$4 4JJ45E%%%!) ryy!45a8==?@ @+BJJ567 19L r#c,ddl}|jy)zRun ``doctest`` on this module.rN)doctesttestmod)rs r!_testrs OOr#__main__)rr6r r6rr6)r8rrr)r8zint | float | np.ndarrayrr+)rG npt.ArrayLikerHr rrr np.ndarray)rRrrHr rr)rU)rgr6rhzLiteral['r', 'w', 'a', 'r+']rr)rqzCallable[[Any], Any]rrurl)rr6rfloatrr)rrrr+)rrrrz str | Nonerr)rr6rz list[str])r)rr6rzlist[tuple[str, int]])rr6rrrr)rr+r)?rv __future__rrYrPrrr.timerr~typingrrrrpathlibr collections.abcr r@r2 numpy.typingnptflavorr rHr r rOint64SizeType__annotations__lib NumpyVersion __version__rrC __array__r1r"r9r=rLrSrarwrrrrrrstdoutrrrdictrrrrrtrr#r!r s" &66$#       8866r~~&'1NVV(83N! T* B0 9G "*5< "9<:4:45:4 :4z17h4(;=7<<A 9< H 8;zz38;zz A(("77t & z GQ s %FF! F!