U hdC @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)Zinbounds)rtypestoreZgepload) contextbuilder signatureargstupidxvalstackZoffptrrS/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/cpython/unsafe/tuple.pycodegens     ztuple_setitem..codegen)dtype)Z 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)rZintprr Zget_value_typeZget_constant_undefZcompile_internal) r r r r rZinner_argtypesZ inner_sigZ ll_idx_typerZ inner_argsressizeZ tuple_typerrr.s    z'build_full_slice_tuple..codegen) isinstancerZIntegerLiteralrZRequireLiteralValueintZ literal_valueZUniTupleZ slice2_type)tyctxszrrrr!rbuild_full_slice_tuple$s   r'cCs8t|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 r r rrrrRsz$unpack_single_tuple..codegen)r#rZ BaseTuplerZUnsupportedError)r%rmsgrrrrrunpack_single_tupleBs   r)N) __doc__Z numba.corerrrZnumba.core.cgutilsrZnumba.core.extendingrrr'r)rrrrs