U |e=&@sdZddlZddlZddlZddlmZddlmZGdddZGdddej d d Z Gd d d Z Gd dde Z GdddZeZGddde dZdS)z[ This module contains utils for manipulating target configurations such as compiler flags. N)MappingProxyType)utilsc@s@eZdZdZdZddZeddZeddZed d Z d S) Optionz.An option to be used in ``TargetConfig``. _type_default_doccCs||_||_||_dS)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)selftypedefaultdocr T/var/www/website-v5/atlas_env/lib/python3.8/site-packages/numba/core/targetconfig.py__init__s zOption.__init__cCs|jSN)rr r r rr "sz Option.typecCs|jSr)rrr r rr &szOption.defaultcCs|jSr)rrr r rr *sz Option.docN) __name__ __module__ __qualname____doc__ __slots__rpropertyr r r r r r rr s  rc@s eZdZdS) _FlagsStackN)rrrr r r rr/srflags) stack_namec@s<eZdZdZeddZddZddZdd Zd d Z d S) 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. cCs|}|r|}nd}|S)z8Get the TOS or return None if no config is set. N)top)clsr rr r r top_or_none9s  zConfigStack.top_or_nonecCs t|_dSr)r_stkrr r rrEszConfigStack.__init__cCs |jSr)rrrr r rrHszConfigStack.topcCs t|jSr)lenrrr r r__len__KszConfigStack.__len__cCs |j|S)zgReturns a contextmanager that performs ``push(flags)`` on enter and ``pop()`` on exit. )renter)r rr r rr"NszConfigStack.enterN) rrrr classmethodrrrr!r"r r r rr3s rc@s eZdZdZddZddZdS)_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. cCsji}t|D]}||jq |||t||_dd}|jD]\}}t|||||qJdS)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. cs8fdd}fdd}fdd}t|||jS)Ncs|jjSr)_valuesgetr rnameoptionr rgetterosz=_MetaTargetConfig.__init__..make_prop..gettercs||j<dSr)r r%)r valr'r rsetterrsz=_MetaTargetConfig.__init__..make_prop..settercs |j=dSr)r%r)r(r rdelterusz=_MetaTargetConfig.__init__..make_prop..delter)rr )r(r)r*r,r-r r'r make_propns z-_MetaTargetConfig.__init__..make_propN)reversedupdateoptions find_optionsritemssetattr)rr(basesdctoptsbase_clsr.r)r r rr\s   z_MetaTargetConfig.__init__cCsdd|DS)z[Returns a new dict with all the items that are a mapping to an ``Option``. cSs i|]\}}t|tr||qSr ) isinstancer.0kvr r r s z2_MetaTargetConfig.find_options..)r3)rr6r r rr2}sz_MetaTargetConfig.find_optionsN)rrrrrr2r r r rr$Us!r$c@seZdZddZdS) _NotSetTypecCsdS)Nzr rr r r__repr__sz_NotSetType.__repr__N)rrrr@r r r rr?sr?c@seZdZdZddiZd%ddZddZd d Zd d Zd dZ ddZ ddZ e fddZ ddZedddZddZddZeedddZedd d!Zeeed"d#d$ZdS)& 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 wbitsiNcCs.i|_|dk r*t|tst|j|jdS)z Parameters ---------- copy_from : TargetConfig or None if None, creates an empty ``TargetConfig``. Otherwise, creates a copy. N)r%r9rAAssertionErrorr0)r Z copy_fromr r rrszTargetConfig.__init__cCsrg}g}|jD]8}|dt||}||s<||q||q|jj}|dd|dd|dS)N=(, z, [z]))r1getattris_setappend __class__rjoin)r argsdefsr<msgclsnamer r rr@s    zTargetConfig.__repr__cCsttt|Sr)hashtuplesortedvaluesrr r r__hash__szTargetConfig.__hash__cCs"t|tr||kStSdSr)r9rArSNotImplemented)r otherr r r__eq__s zTargetConfig.__eq__csfddjDS)z)Returns a dict of all the values csi|]}|t|qSr )rG)r;r<rr rr>sz'TargetConfig.values..)r1rr rrrSszTargetConfig.valuescCs||||jkS)zIs the option set? ) _guard_optionr%r r(r r rrHs zTargetConfig.is_setcCs|||j|ddS)zRemove the option by name if it is defined. After this, the value for the option will be set to its default value. N)rXr%poprYr r rdiscards zTargetConfig.discardcCsR||||sNt}|r:|}t||t||n|tk rNt|||dS)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)rXrHrrr4rG_NotSet)r r(r Zcstkrr r rinherit_if_not_sets  zTargetConfig.inherit_if_not_setcCs t||S)zClone this instance. )r rr r rcopyszTargetConfig.copy)returncCs0dd|D}|jj}|dd|dS)zReturns a ``str`` that summarizes this instance. In contrast to ``__repr__``, only options that are explicitly set will be shown. cSsg|]\}}|d|qS)rDr r:r r r sz(TargetConfig.summary..rErF)) _summary_argsrJrrK)r rLrOr r rsummaryszTargetConfig.summarycCs(||jkr$|dt|}t|dS)Nz is not a valid option for )r1r ValueError)r r(rNr r rrXs zTargetConfig._guard_optioncCsRg}t|jD]>}|j|}||rt||}|j|kr||f}||q|S)zreturns a sorted sequence of 2-tuple containing the ``(flag_name, flag_value)`` for flag that are set with a non-default value. )rRr1rHrGr rI)r rLr<optZflagvalr=r r rrbs     zTargetConfig._summary_argscCsfg}|d||jj|ddg|jD]"\}}|||t|jq4d| S)z]Returns a ``bytes`` object suitable for use as a dictionary for compression. numbaTrueFalse) rIrJrextendr1r3strr rKencode)rbufr<rer r r_make_compression_dictionarys  z)TargetConfig._make_compression_dictionarycCsX|}tjf|tjd|j}||g}|| t d | S)z6Return a string suitable for symbol mangling. )zdictlevel)rnzlib compressobjZ_BEST_COMPRESSION _ZLIB_CONFIGcompressrcrlrIflushbase64 b64encoderKdecode)r rocomprmr r rget_mangle_string"szTargetConfig.get_mangle_string)mangledr_cCszdd}td||}t|}|}tjfd|i|j}g}|r^|| ||j }qB|| d | S)zCReturns the demangled result from ``.get_mangle_string()`` cSs ttd|ddddS)N0xr)chrintgroup)xr r rrepl4sz#TargetConfig.demangle..replz_[a-zA-Z0-9][a-zA-Z0-9]rorq)resubrx b64decodernrr decompressobjrurI decompressunconsumed_tailrwrKrz)rr}rZ unescapedrawrodcrmr r rdemangle/s zTargetConfig.demangle)N)rrrrrurr@rTrWrSrHr[r\r]r^rkrcrXrbr#bytesrnr|rr r r rrAs&    rA) metaclass)rrrrrxtypesr numba.corerrThreadLocalStackrrr r$r?r\rAr r r rs  ""/