o TdC @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 signatureargstupidxvalstackoffptrrm/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/cpython/unsafe/tuple.pycodegens     ztuple_setitem..codegen)dtype) typingctxrrrrsigrrr tuple_setitem s  rcsNt|tjs t|t|jtjtjd|}fdd}||fS)z"Creates a sz-tuple of full slices.)rcountc sZdd}tjg}tjg|R}|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#inner_argtypes inner_sig ll_idx_typer inner_argsressize tuple_typerrr.s    z'build_full_slice_tuple..codegen) isinstancerIntegerLiteralrRequireLiteralValueint literal_valueUniTuple slice2_type)tyctxszrrrr-rbuild_full_slice_tuple$s   r9cCs8t|tjsd|}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)r0r BaseTuplerUnsupportedError)r7rmsgrrrrrunpack_single_tupleBs  r=N) __doc__ numba.corerrrnumba.core.cgutilsrnumba.core.extendingrrr9r=rrrrs