U |eC @sXdZddlmZmZmZddlmZddlmZeddZ eddZ ed d Z d S) z This file provides internal compiler utilities that support certain special operations with tuple and workarounds for limitations enforced in userland. )typestypingerrors) alloca_once) intrinsiccCsdd}||||j}||fS)aReturn a copy of the tuple with item at *idx* replaced with *val*. Operation: ``out = tup[:idx] + (val,) + tup[idx + 1:] **Warning** - No boundchecking. - The dtype of the tuple cannot be changed. *val* is always cast to the existing dtype of the tuple. c SsR|\}}}t||j}||||j||d|gdd}|||||S)NrT)inbounds)rtypestoregepload) contextbuilder signatureargstupidxvalstackZoffptrrW/var/www/website-v5/atlas_env/lib/python3.8/site-packages/numba/cpython/unsafe/tuple.pycodegens     ztuple_setitem..codegen)dtype) typingctxrrrrsigrrr tuple_setitem s  rcsNt|tjst|t|jtjtjd|}fdd}||fS)z"Creates a sz-tuple of full slices.)rcountc sXdd}tjg}tjf|}|tj}|}||g} ||||| } | S)NcSs(|}t|D]}t||tdd}q |S)N)rangerslice)length empty_tupleoutirrrimpl/s z5build_full_slice_tuple..codegen..impl)rintprrget_value_typeget_constant_undefcompile_internal) r r rrr"Zinner_argtypesZ inner_sigZ ll_idx_typer inner_argsressizeZ tuple_typerrr.s    z'build_full_slice_tuple..codegen) isinstancerIntegerLiteralrRequireLiteralValueint literal_valueUniTuple slice2_type)tyctxszrrrr)rbuild_full_slice_tuple$s   r4cCs8t|tjs d|}t|||}dd}||fS)aRThis exists to handle the situation y = (*x,), the interpreter injects a call to it in the case of a single value unpack. It's not possible at interpreting time to differentiate between an unpack on a variable sized container e.g. list and a fixed one, e.g. tuple. This function handles the situation should it arise. zBOnly tuples are supported when unpacking a single item, got type: cSs|dS)Nrr)r r rrrrrrRsz$unpack_single_tuple..codegen)r+r BaseTuplerUnsupportedError)r2rmsgrrrrrunpack_single_tupleBs   r8N) __doc__ numba.corerrrnumba.core.cgutilsrnumba.core.extendingrrr4r8rrrrs