U ,d=&@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 ;/tmp/pip-unpacked-wheel-eu7e0c37/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)Z 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'basesdctoptsZbase_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..)r2)rr5r r rr1}sz_MetaTargetConfig.find_optionsN)rrrrrr1r 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$r7r?AssertionErrorr/)r Z copy_fromr r rrszTargetConfig.__init__cCsrg}g}|jD]8}|dt||}||s<||q||q|jj}|dd|dd|dS)N=(, z, [z]))r0getattris_setappend __class__rjoin)r argsZdefsr:msgclsnamer r rr>s    zTargetConfig.__repr__cCsttt|Sr)hashtuplesortedvaluesrr r r__hash__szTargetConfig.__hash__cCs"t|tr||kStSdSr)r7r?rPNotImplemented)r otherr r r__eq__s zTargetConfig.__eq__csfddjDS)z)Returns a dict of all the values csi|]}|t|qSr )rE)r9r:rr rr<sz'TargetConfig.values..)r0rr rrrPszTargetConfig.valuescCs||||jkS)zIs the option set? ) _guard_optionr$r r'r r rrFs 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)rUr$poprVr 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)rUrFrrr3rE_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)rBr r8r r r sz(TargetConfig.summary..rCrD)) _summary_argsrHrrI)r rJrLr r rsummaryszTargetConfig.summarycCs(||jkr$|dt|}t|dS)Nz is not a valid option for )r0r ValueError)r r'rKr r rrUs 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. )rOr0rFrEr rG)r rJr:optZflagvalr;r r rr_s     zTargetConfig._summary_argscCsfg}|d||jj|ddg|jD]"\}}|||t|jq4d| S)z]Returns a ``bytes`` object suitable for use as a dictionary for compression. ZnumbaTrueFalse) rGrHrextendr0r2strr rIencode)rbufr:rbr 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)rjzlib compressobjZ_BEST_COMPRESSION _ZLIB_CONFIGcompressr`rhrGflushbase64 b64encoderIdecode)r rkcomprir 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]rkrm)resubrt b64decoderjrn decompressobjrqrG decompressunconsumed_tailrsrIrv)rryrZ unescapedrawrkdcrir r rdemangle/s zTargetConfig.demangle)N)rrrrrqrr>rQrTrPrFrXrYrZr[rgr`rUr_r"bytesrjrxrr r r rr?s&    r?) metaclass)rrrnrttypesrZ numba.corerrZThreadLocalStackrrr r#r=rYr?r r r rs  ""/