xKg=&dZddlZddlZddlZddlmZddlmZGddZGddejd Z Gd d Z Gd de Z GddZeZGdde Zy)z[ This module contains utils for manipulating target configurations such as compiler flags. N)MappingProxyType)utilscJeZdZdZdZdZedZedZedZ y)Optionz.An option to be used in ``TargetConfig``. _type_default_docc.||_||_||_y)a- Parameters ---------- type : Type of the option value. It can be a callable. The setter always calls ``self._type(value)``. default : The default value for the option. doc : str Docstring for the option. Nr)selftypedefaultdocs [/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/core/targetconfig.py__init__zOption.__init__s   c|jSN)rr s rr z Option.type"s zzrc|jSr)r rs rrzOption.default&s }}rc|jSr)r rs rrz Option.doc*s yyrN) __name__ __module__ __qualname____doc__ __slots__rpropertyr rrrrrr sM+I rrc eZdZy) _FlagsStackN)rrrrrrr r /srr flags) stack_namec8eZdZdZedZdZdZdZdZ y) ConfigStackzA stack for tracking target configurations in the compiler. It stores the stack in a thread-local class attribute. All instances in the same thread will see the same stack. c@|}|r|j}|Sd}|S)z8Get the TOS or return None if no config is set. N)top)clsr r!s r top_or_nonezConfigStack.top_or_none9s,u HHJE E rc"t|_yr)r _stkrs rrzConfigStack.__init__Es M rc6|jjSr)r*r&rs rr&zConfigStack.topHsyy}}rc,t|jSr)lenr*rs r__len__zConfigStack.__len__Ks499~rc8|jj|S)zgReturns a contextmanager that performs ``push(flags)`` on enter and ``pop()`` on exit. )r*enter)r r!s rr0zConfigStack.enterNsyyu%%rN) rrrr classmethodr(rr&r.r0rrrr$r$3s/   "&rr$ceZdZdZdZdZy)_MetaTargetConfigzMetaclass for ``TargetConfig``. When a subclass of ``TargetConfig`` is created, all ``Option`` defined as class members will be parsed and corresponding getters, setters, and delters will be inserted. c 0i}t|D]}|j|j|j|j|t ||_d}|jj D]\}}t |||||y)zInvoked when subclass is created. Insert properties for each ``Option`` that are class members. All the options will be grouped inside the ``.options`` class attribute. cXfd}fd}fd}t|||jS)NcP|jjjSr)_valuesgetr)r nameoptions rgetterz=_MetaTargetConfig.__init__..make_prop..getteros||''fnn==rcBj||j<yr)r r7)r valr9r:s rsetterz=_MetaTargetConfig.__init__..make_prop..setterrs%+[[%5 T"rc |j=yr)r7r r9s rdelterz=_MetaTargetConfig.__init__..make_prop..delterusLL&r)rr)r9r:r;r>rAs`` r make_propz-_MetaTargetConfig.__init__..make_propns) > 6 'FFFFJJ? ?rN)reversedupdateoptions find_optionsritemssetattr)r'r9basesdctoptsbase_clsrBr:s rrz_MetaTargetConfig.__init__\s!H KK(( )( C$$S)*&t,  @ KK--/LD& Cyv6 70rcx|jDcic]\}}t|ts||c}}Scc}}w)z[Returns a new dict with all the items that are a mapping to an ``Option``. )rG isinstancer)r'rJkvs rrFz_MetaTargetConfig.find_options}s3"%FA 1f0E1FFFs66N)rrrrrrFrrrr3r3Us 8BGrr3ceZdZdZy) _NotSetTypecy)Nzrrs r__repr__z_NotSetType.__repr__srN)rrrrTrrrrRrRsrrRceZdZdZddiZddZdZdZdZd Z d Z d Z e fd Z d ZdefdZdZdZedefdZdefdZededefdZy) TargetConfigaBase class for ``TargetConfig``. Subclass should fill class members with ``Option``. For example: >>> class MyTargetConfig(TargetConfig): >>> a_bool_option = Option(type=bool, default=False, doc="a bool") >>> an_int_option = Option(type=int, default=0, doc="an int") The metaclass will insert properties for each ``Option``. For example: >>> tc = MyTargetConfig() >>> tc.a_bool_option = True # invokes the setter >>> print(tc.an_int_option) # print the default wbitsiNci|_|8t|tsJ|jj|jyy)z Parameters ---------- copy_from : TargetConfig or None if None, creates an empty ``TargetConfig``. Otherwise, creates a copy. N)r7rNrVrD)r copy_froms rrzTargetConfig.__init__s>  i6 66 LL   1 1 2 !rc4g}g}|jD]G}|dt||}|j|s|j|7|j|I|jj }|ddj |ddj |dS)N=(, z, [z]))rEgetattris_setappend __class__rjoin)r argsdefsrOmsgclsnames rrTzTargetConfig.__repr__sACqq)*+C;;q> C  C  ..))!DIIdO,C $/@CCrcXttt|jSr)hashtuplesortedvaluesrs r__hash__zTargetConfig.__hash__sE&/011rcpt|tr!|j|jk(StSr)rNrVrkNotImplemented)r others r__eq__zTargetConfig.__eq__s) e\ *;;=ELLN2 2! !rcV|jDcic]}|t||c}Scc}w)z)Returns a dict of all the values )rEr^)r rOs rrkzTargetConfig.valuess,.2\\:\74##\:::s&c@|j|||jvS)zIs the option set? ) _guard_optionr7r@s rr_zTargetConfig.is_sets! 4 t||##rc^|j||jj|dy)zRemove the option by name if it is defined. After this, the value for the option will be set to its default value. N)rsr7popr@s rdiscardzTargetConfig.discards& 4  t$rc|j||j|sKt}|r(|j}t ||t ||y|t urt |||yyy)aLInherit flag from ``ConfigStack``. Parameters ---------- name : str Option name. default : optional When given, it overrides the default value. It is only used when the flag is not defined locally and there is no entry in the ``ConfigStack``. N)rsr_r$r&rHr^_NotSet)r r9rcstkr&s rinherit_if_not_setzTargetConfig.inherit_if_not_setsd 4 {{4 =DhhjdGC$67'dG,( !rc$t||S)zClone this instance. )r rs rcopyzTargetConfig.copystDz$rreturnc|jDcgc] \}}|d|}}}|jj}|ddj|dScc}}w)zReturns a ``str`` that summarizes this instance. In contrast to ``__repr__``, only options that are explicitly set will be shown. r[r\r])) _summary_argsrarrb)r rOrPrcrfs rsummaryzTargetConfig.summarysa (,'9'9';<';tq!1#Qqc ';<..))!DIIdO,A..=sAcV||jvr|dt|}t|y)Nz is not a valid option for )rEr ValueError)r r9res rrszTargetConfig._guard_options2 t|| #H7T |DCS/ ! $rcg}t|jD]T}|j|}|j|s$t||}|j|k7s@||f}|j |V|S)zreturns a sorted sequence of 2-tuple containing the ``(flag_name, flag_value)`` for flag that are set with a non-default value. )rjrEr_r^rr`)r rcrOoptflagvalrPs rrzTargetConfig._summary_argssg  %A,,q/C{{1~!$*;;')G AKKN & rcg}|jd|j|jj|jddg|jj D]:\}}|j||jt |j<dj|jS)z]Returns a ``bytes`` object suitable for use as a dictionary for compression. numbaTrueFalse) r`rarextendrErGstrrrbencode)r'bufrOrs r_make_compression_dictionaryz)TargetConfig._make_compression_dictionarys  7 3==))* FG$%kk'')FAs JJqM JJs3;;' (*wws|""$$rc|j}tjd|tjd|j}|j |j jg}|j|jtjdj|jS)z6Return a string suitable for symbol mangling. )zdictlevelrr)rzlib compressobjZ_BEST_COMPRESSION _ZLIB_CONFIGcompressrrr`flushbase64 b64encoderbdecode)r rcomprs rget_mangle_stringzTargetConfig.get_mangle_string"s1135e43J3J5"&"3"35}}T\\^22456 4::<  .5577rmangledcd}tjd||}tj|}|j }t j dd|i|j}g}|r/|j|j||j}|r/|j|jdj|jS)zCReturns the demangled result from ``.get_mangle_string()`` cVttd|jdddzdS)N0xr)chrintgroup)xs rreplz#TargetConfig.demangle..repl4s's4!''!*QR.0"56 6rz_[a-zA-Z0-9][a-zA-Z0-9]rrr)resubr b64decoderr decompressobjrr` decompressunconsumed_tailrrbr)r'rr unescapedrawrdcrs rdemanglezTargetConfig.demangle/s  7FF5tWE y)002    @e @s/?/? @ JJr}}S) *$$C 288:xx}##%%rr)rrrrrrrTrlrprkr_rvrxrzr|rrrsrr1bytesrrrrrrrVrVs "S>L 3 D2" ; $ %07-, //" %U%%" 83 8&s&s&&rrV) metaclass)rrrrtypesr numba.corerrThreadLocalStackr r$r r3rRrxrVrrrrsv "D %((W &&D,G,G^ -u&.u&r