d]28dZddlZddlZddlmZmZdZ dZ gZ iZ iZ iZ iZ dZdZd Zd Zd Zd Zd Zd(dZe de ddZdZdZdZdZdZdZdZdZdZ gdZ!dZ"dZ#gZ$dZ%ej&'ej(ej&'dkrd Z)nej*Z)d!Z+d"Z,e,d#Z-e,d$Z.d%Z/ed&Z0e1d'kr e0dSdS))a5Utilities for handling different array flavors in PyTables. Variables ========= `__docformat`__ The format of documentation strings in this module. `internal_flavor` The flavor used internally by PyTables. `all_flavors` List of all flavors available to PyTables. `alias_map` Maps old flavor names to the most similar current flavor. `description_map` Maps flavors to short descriptions of their supported objects. `identifier_map` Maps flavors to functions that can identify their objects. The function associated with a given flavor will return a true value if the object passed to it can be identified as being of that flavor. See the `flavor_of()` function for a friendlier interface to flavor identification. `converter_map` Maps (source, destination) flavor pairs to converter functions. Converter functions get an array of the source flavor and return an array of the destination flavor. See the `array_of_flavor()` and `flavor_to_flavor()` functions for friendlier interfaces to flavor conversion. N) FlavorError FlavorWarningreStructuredTextnumpyc|tvr9ddtD}td|d|dS)z5Raise a ``FlavorError`` if the `flavor` is not valid.z, c3K|]}|VdSN).0flavs -lib/python3.11/site-packages/tables/flavor.py zcheck_flavor..Xs"#A#ATD#A#A#A#A#A#Az flavor ``zH`` is unsupported or unavailable; available flavors in this system are: N) all_flavorsjoinr)flavoravailable_flavss r check_flavorrTs][  ))#A#A[#A#A#AAAkvv ()) )! rcx||f}|tvrtd|d|dt|}||S)aGet a version of the given `array` in a different flavor. The input `array` must be of the given `src_flavor`, and the returned array will be of the indicated `dst_flavor`. Both flavors may be the same, but it is not guaranteed that the returned array will be the same object as the input one in this case. If the conversion is not supported, a ``FlavorError`` is raised. zconversion from flavor ``z`` to flavor ``z/`` is unsupported or unavailable in this system) converter_mapr)array src_flavor dst_flavorconvkeyconvfuncs rarray_of_flavor2r_sY:&Gm##k'ZZ566 6W%H 8E??rc t|||S#t$r7}tj|jdd|dt |cYd}~Sd}~wwxYw)aGet a version of the given `array` in a different flavor. The input `array` must be of the given `src_flavor`, and the returned array will be of the indicated `dst_flavor` (see below for an exception to this). Both flavors may be the same, but it is not guaranteed that the returned array will be the same object as the input one in this case. If the conversion is not supported, a `FlavorWarning` is issued and the input `array` is returned as is. rz; returning an object of the ``z`` flavor insteadN)rrwarningswarnargsr)rrrfes rflavor_to_flavorr#vsyz:>>>  ZZZ12? A A A s A,A AAc.t|t|S)zGet a version of the given `array` in a different `dst_flavor`. The input `array` must be of the internal flavor, and the returned array will be of the given `dst_flavor`. See `flavor_to_flavor()` for more information. )r#internal_flavorrrs rinternal_to_flavorr's E?J ? ??rc.t||tS)zGet a version of the given `array` in the internal flavor. The input `array` must be of the given `src_flavor`, and the returned array will be of the internal flavor. If the conversion is not supported, a ``FlavorError`` is raised. )rr%)rrs rarray_as_internalr)s E: ? ??rctD]}t||r|cSt|j}ddtD}t d|d|)zIdentify the flavor of a given `array`. If the `array` can not be matched with any flavor, a ``TypeError`` is raised. z; c30K|]}t|VdSr description_map)r fls rrzflavor_of..s'JJ 3JJJJJJrzobjects of type ``zD`` are not supported in this context, sorry; supported objects are: )ridentifier_maptype__name__r TypeError)rr type_namesupported_descss r flavor_ofr5s & !% ( ( MMM U $IiiJJkJJJJJO )'0yy// C D DDrc>t|t||S)zGet a version of the given `array` in a different `dst_flavor`. The flavor of the input `array` is guessed, and the returned array will be of the given `dst_flavor`. If the conversion is not supported, a ``FlavorError`` is raised. )rr5r&s rarray_of_flavorr7s E9U#3#3Z @ @@rpythoncttt|z thz }|D]}t|dS)a&Disable all flavors except those in keep. Providing an empty keep sequence implies disabling all flavors (but the internal one). If the sequence is not specified, only optional flavors are disabled. .. important:: Once you disable a flavor, it can not be enabled again. N)setrr%_disable_flavor)keepremovers rrestrict_flavorsr?sN  D )_,= =F    rr9c\tD]#}td|z}|D] }|t|< $dS)z(Register aliases of *available* flavors.z _%s_aliasesN)reval alias_map)raliasesaliass r_register_aliasesrEsP&&}v-.. & &E%Ie   &&&rcNtD]}td|zt|<dS)z-Register descriptions of *available* flavors.z_%s_descN)rrAr-rs r_register_descriptionsrHs5<<"&zF':";";<.identitys rN_conv__to_)rrA NameErrorr)rNrrrs r_register_convertersrRs! C C % C CJ H ( E E E E EFF ( ( (++'H ( C:B z:67 C C Cs/AAcvttttdS)z!Register all *available* flavors.N)rErHrJrRr rr _register_allrT s:rcg}tD] \}}||kr||!|D] }t|= dS)z3Deregister aliases of a given `flavor` (no checks).N)rBitemsappend)r rm_aliasesan_aliasa_flavors r_deregister_aliasesr[skJ ) 1 1((8 v     h ' ' '   h    rct|=dS)z7Deregister description of a given `flavor` (no checks).Nr,rGs r_deregister_descriptionr]s rct|=dS)z?Deregister identifier function of a given `flavor` (no checks).N)r/rGs r_deregister_identifierr_#s vrcjg}tD]}||vr|||D] }t|= dS)z?Deregister converter functions of a given `flavor` (no checks).N)rrW)rrm_flavor_pairs flavor_pairs r_deregister_convertersrc)s\O$00 [  " "; / / /&'' + & &''rct|t|t|t|t|dS)z2Completely disable the given `flavor` (no checks).N)r[r]r_rcrr>rGs rr<r<4sVF###6"""6"""vr)ListTupleIntFloatStringVLStringObjectz;homogeneous list or tuple, integer, float, complex or bytesc jt|ttttt t fSr ) isinstancetuplelistintfloatcomplexbytesrMs r _is_pythonrtHs eeT3wF G GGrzNumPy array, record or scalarz1.19.0ctj5tjd tj|g|Ri|}n"#tj$rt dwxYw dddn #1swxYwY|S)Nerrorz-cannot guess the desired dtype from the input)rcatch_warnings simplefilternprVisibleDeprecationWarning ValueError)rr!kwargss rtoarrayr}Qs  $ & & E E  !' * * * E888888/ E E E CEEE E E E E E E E E E E E E E E E E s'A/AA/AA//A36A3cNt|tjtjfSr )rmryndarraygenericrMs r _is_numpyr_s ebj"*5 6 66rcBfd}j|_j|_|S)zDecorate `convfunc` to return a *contiguous* NumPy array. Note: When arrays are 0-strided, the copy is avoided. This allows to use `array` to still carry info about the dtype and shape. c|}t|dr8|jjs,t|jdkr|}|S)Nflagsr)hasattrr contiguoussumstridescopy)rnparrrs r conv_to_numpyz(_numpy_contiguous..conv_to_numpyjsV E7 # # !K* !EM""a''JJLLE r)r1__doc__)rrs` r_numpy_contiguousrcs:&.M$,M rctj|}|jjdkr& |dS#t $rYnwxYw|SNUS)ryasarraydtypekindastypeUnicodeEncodeErrorrrs r_conv_numpy_to_numpyrvsa Ju  E {3 <<$$ $!    D  Ls; AAct|}|jjdkr& |dS#t$rYnwxYw|Sr)r}rrrrrs r_conv_python_to_numpyrs[ ENNE {3 <<$$ $!    D  Ls6 AAcn|jdkr|}n|}|S)Nr )shapetolistitemrMs r_conv_numpy_to_pythonrs2 {b   Lrc6ddl}|dS)zRun ``doctest`` on this module.rN)doctesttestmod)rs r_testrs#NNN OOr__main__)r8)2rrrry exceptionsrr __docformat__r%rrBr-r/rrrr#r'r)r5r7r?rWrErHrJrRrTr[r]r_rcr<_python_aliases _python_descrt_numpy_aliases _numpy_desclib NumpyVersion __version__r}rrrrrrrr1r rrrs""H22222222# 9-  0  ?D )))., @ @ @ @ @ @DDD$ A A A     * 7 8&&&<<< 999CCC*       ''' 3 HHH- 6r~&&"&*=*=h*G*GGG    hG777&         z EGGGGGr