U ,d+@s&ddlmZmZddlZddlZddlZddlmZmZddl m Z m Z m Z ddl mZeddgZdZd d Zd d Zd dZeejZeejZeeBZeeddgZejZejZGdddeZdZdZ dZ!ej"#dZ$ddZ%ddZ&GdddeZ'GdddeZ(Gddde j)Z*dS)) namedtuple OrderedDictN)CodeType ModuleType)errorsutils serialize) PYVERSION opcode_infoZargsizecCst|dd}|rt||S|S)z Objects that wraps function should provide a "__numba__" magic attribute that contains a name of an attribute that contains the actual python function object. Z __numba__Ngetattr)objattrr7/tmp/pip-unpacked-wheel-eu7e0c37/numba/core/bytecode.pyget_function_objects  rcCst|dt|ddS)z"Shamelessly borrowed from llpython__code__ func_codeNr )rrrrget_code_objectsrcCs0g}|D]"}tj|}|dk r||q|SN)disopmapgetappend)seqlstscrrr _as_opcodes#s   r RETURN_VALUE RAISE_VARARGSc@sPeZdZdZdZddZeddZeddZd d Z d d Z ed dZ dS) ByteCodeInstz Attributes ---------- - offset: byte offset of opcode - opcode: opcode integer value - arg: instruction arg - lineno: -1 means unknown )offsetnextopcodeopnamearglinenocCs.||_||_||_tj||_||_d|_dS)N)r#r$r%rr&r'r(selfr#r%r' nextoffsetrrr__init__Cs  zByteCodeInst.__init__cCs |jtkSr)r%JUMP_OPSr+rrris_jumpKszByteCodeInst.is_jumpcCs |jtkSr)r%TERM_OPSr/rrr is_terminatorOszByteCodeInst.is_terminatorcCsx|js ttdkrJ|jtkr,|j|jdS|jtks:t|jddSn*|jtkr`|j|jS|jtksnt|jSdS)N) r )r0AssertionErrorr r%JREL_OPSr$r'JABS_OPSr/rrrget_jump_targetSs    zByteCodeInst.get_jump_targetcCsd|j|j|jfS)Nz%s(arg=%s, lineno=%d))r&r'r(r/rrr__repr__jszByteCodeInst.__repr__cCs&|jdrdS|jdkrdSdSdS)zREffect of the block stack Returns +1 (push), 0 (none) or -1 (pop) ZSETUP_ POP_BLOCKr)rN)r& startswithr/rrr block_effectms   zByteCodeInst.block_effectN) __name__ __module__ __qualname____doc__ __slots__r-propertyr0r2r8r9r=rrrrr"4s   r"r:NOPccsd}t|}d}}||kr||}|t7}|tkr|||B}ttD]}||||d|>O}qH|t7}|tkr|dt>}qn d}|t7}d}||||fV|}qdS)zd Returns a 4-int-tuple of (bytecode offset, opcode, argument, offset of next bytecode). rN)lenCODE_LEN HAVE_ARGUMENTrangeARG_LEN EXTENDED_ARG NO_ARG_LEN)code extended_argnr#iopr'jrrr_unpack_opargss&   rSccsJdtdtfV|D]2\}}}}|tkr.|t7}|t|||tfVqdS)zpPatch the bytecode stream. - Adds a NOP bytecode at the start to avoid jump target being at the entry. rN) OPCODE_NOP _FIXED_OFFSETr7)Z bc_streamr#r%r'r,rrr_patched_opargss rVc@s8eZdZddZddZddZddZeZd d Zd S) ByteCodeItercCs ||_ttt|jj|_dSr)rMiterrVrSco_code)r+rMrrrr-szByteCodeIter.__init__cCs|Srrr/rrr__iter__szByteCodeIter.__iter__cCs t|jSr)r$rXr/rrr _fetch_opcodeszByteCodeIter._fetch_opcodecCs$|\}}}}|t||||dfS)N)r#r%r'r,)r[r"r*rrrr$s zByteCodeIter.nextcCs4d}t|D]"}t|j\}}||d|>O}q |S)NrrE)rIr$rX)r+sizebufrP_offsetbyterrrread_args  zByteCodeIter.read_argN) r>r?r@r-rZr[r$__next__r`rrrrrWs rWc@s\eZdZdZdZddZeddZddZd d Z d d Z d dZ eddZ ddZ dS)ByteCodezF The decoded bytecode of a function, and related information. )func_idco_names co_varnames co_consts co_cellvars co_freevarstablelabelscCs|j}tddt|jD}|dtt|}|||||_ |j |_ |j |_ |j |_ |j |_ |j|_||_t||_dS)Ncss|]}|tVqdSr)rU).0xrrr sz$ByteCode.__init__..r)rMsetr findlabelsrYaddrrW_compute_linenorcrdrerfrgrhrisortedrj)r+rcrMrjrirrrr-s   zByteCode.__init__cCsbt|D]"\}}|t}||kr |||_q |tj}|D]}|jdkrV|j}q@||_q@|S)zI Compute the line numbers for all bytecode instructions. r)rfindlinestartsrUr(values)clsrirMr#r(Z adj_offsetZknowninstrrrrqs    zByteCode._compute_linenocCst|jSr)rXrirtr/rrrrZszByteCode.__iter__cCs |j|Srrir+r#rrr __getitem__szByteCode.__getitem__cCs ||jkSrrwrxrrr __contains__szByteCode.__contains__cs*fdddfddjDS)Ncs|djjkrdSdSdS)Nr:> )r#rj)rPr/rr label_markersz#ByteCode.dump..label_marker c3s |]}d|f|VqdS)z %s %10s %sNr)rkrP)r}rrrmsz ByteCode.dump..)joinriitemsr/r)r}r+rdumps z ByteCode.dumpc Csi}|j}|dtj}t|tr(|j}|D]R}|jdkr0||j } | |kr0z || } Wnt k rx|| } YnX| || <q0|D]4} t| t rt t | } |||| | j| jq|S)za Compute the globals used by the function with the given bytecode table. __builtins__ LOAD_GLOBAL) __globals__rrbuiltins isinstancer__dict__rtr&r'KeyErrorrrrWupdate_compute_used_globalsrfrd) rufuncrirfrddZglobsrrvnamevaluecoZsubtablerrrrs,         zByteCode._compute_used_globalscCs||jj|j|j|jS)zv Get a {name: value} map of the globals used by this code object and any nested code objects. )rrcrrirfrdr/rrrget_used_globals$szByteCode.get_used_globalsN)r>r?r@rArBr- classmethodrqrZryrzrrrrrrrrbs   rbc@sBeZdZdZedZeddZddZ ddZ ed d Z d S) FunctionIdentityz A function's identity and metadata. Note this typically represents a function whose bytecode is being compiled, not necessarily the top-level user function (the two might be distinct, e.g. in the `@generated_jit` case). r:cCst|}t|}t|}|s,td|z |j}Wntk rP|j}YnX|}||_ ||_ | dd|_ ||_ t||_|jdkrtjn|jj|_t||_||_|j|_|j|_t|j|_t|j|_t|j }d!|j ||_"||_#|S)zD Create the FunctionIdentity of the given function. z %s does not provide its bytecode.r)Nz{}${})$rrrZ pysignaturerZByteCodeSupportErrorr@AttributeErrorr>r func_qualnamesplit func_namerMinspect getmodulemoduleZ_dynamic_modnamemodnameisgeneratorfunctionZ is_generatorpysig co_filenamefilenameco_firstlinenoZ firstlinenorF parameters arg_countlist arg_namesr$ _unique_idsformatZ unique_nameZ unique_id)rupyfuncrrMrrr+uidrrr from_function7s>        zFunctionIdentity.from_functioncCs ||jS)z:Copy the object and increment the unique counter. )rrr/rrrderiveaszFunctionIdentity.derivecCs t|jdS)4 NOTE: part of ReduceMixin protocol )r)dictrr/rrr_reduce_statesfszFunctionIdentity._reduce_statescCs ||S)r)r)rurrrr_rebuildlszFunctionIdentity._rebuildN) r>r?r@rA itertoolscountrrrrrrrrrrr-s  )r)+ collectionsrrrrrtypesrrZ numba.corerrrZnumba.core.utilsr r rUrrr frozensethasjrelr6hasjabsr7r.r1rKrHobjectr"rGrJrLr&indexrTrSrVrWrbZ ReduceMixinrrrrrs6     F f