DŽgVddlmZddlmZmZddlmZedddgZGddZy ) ) namedtuple)typesir) signature BoundFuncfuncsigcJeZdZdZdZedZedZedZedZ ifdZ dde jfd Z dde jfd Zdde jfd Zde jfd Zifde j"fd Zde j&fdZde j"fdZy)ParforLoweringBuilderz@Helper class for building Numba-IR and lowering for Parfor. c.||_||_||_yN)_lowerer_scope_loc)selflowererscopelocs k/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/numba/parfors/parfor_lowering_utils.py__init__zParforLoweringBuilder.__init__ s   c.|jjSr )rcontextrs r_contextzParforLoweringBuilder._contexts}}$$$rc.|jjSr )rtyping_contextrs r _typingctxz ParforLoweringBuilder._typingctxs}}+++rcB|jjjSr )rfndesctypemaprs r_typemapzParforLoweringBuilder._typemaps}}##+++rcB|jjjSr )rr calltypesrs r _calltypesz ParforLoweringBuilder._calltypess}}##---rc|j}|jd}|j}|jj|||}|j t j |||||} t| |S)a9Binds a global function to a variable. Parameters ---------- fobj : object The function to be bound. ftype : types.Type args : Sequence[types.Type] kws : Mapping[str, types.Type] Returns ------- callable: _CallableNode _funcrrhstypname)rr )r__name__rresolve_function_typeassignrGlobal _CallableNode) rfobjftypeargskwsrvarnamegvnamefunc_sigfunc_vars rbind_global_functionz*ParforLoweringBuilder.bind_global_function"svii]]O5)MM???88cJ;; &$C0e' (99rreturnch|jtj||j||S)a$Makes a constant variable Parameters ---------- cval : object The constant value typ : types.Type type of the value name : str variable name to store to Returns ------- res : ir.Var r(r))r/rConstr)rcvalr+r,s rmake_const_variablez)ParforLoweringBuilder.make_const_variable;s0 {{499-3T  rc|j}|Dcgc]}|j|j}}tjj |}|j tjj||||Scc}w)aMakes a tuple variable Parameters ---------- varlist : Sequence[ir.Var] Variables containing the values to be stored. name : str variable name to store to Returns ------- res : ir.Var r)) rr"r,rTuple from_typesr/rExpr build_tuple)rvarlistr,rxvartystuptys rmake_tuple_variablez)ParforLoweringBuilder.make_tuple_variableOstii189A$--'99 &&v.{{##GS1u4  :s Bc|j}|jj||}||j|j<t j |||}|jj||S)a!Assign a value to a new variable Parameters ---------- rhs : object The value typ : types.Type type of the value name : str variable name to store to Returns ------- res : ir.Var ) rrredefiner"r,rAssignr lower_instrr*r+r,rvarr/s rr/zParforLoweringBuilder.assignds^ iikk""4-"% chh3S)   ( rc|j}tj|j||}tj|||}|j j |j||jj||S)a@Assign a value to a new variable or inplace if it already exist Parameters ---------- rhs : object The value typ : types.Type type of the value name : str variable name to store to Returns ------- res : ir.Var ) rrVarrrLr" setdefaultr,rrMrNs rassign_inplacez$ParforLoweringBuilder.assign_inplace{sd iiffT[[$,3S)   3/   ( rctjj|j|||j}|j |j |<|S)a=Call a bound callable Parameters ---------- callable_node : _CallableNode The callee args : Sequence[ir.Var] kws : Mapping[str, ir.Var] Returns ------- res : ir.Expr The expression node for the return value of the call r()rrCcallrrr r%)r callable_noder4r5rUs rrUzParforLoweringBuilder.callsAww||M..ctyy|I - 1 1 rcf|j}|j}tj||||}t t j ||j||j||j|jjj|<|jj||S)aMakes a setitem call Parameters ---------- obj : ir.Var the object being indexed index : ir.Var the index val : ir.Var the value to be stored Returns ------- res : ir.SetItem r() rr"rSetItemrrnoner,rr r$rM)robjindexvalrtmsetitems rr^zParforLoweringBuilder.setitems ii ]]**S%#62; JJ388 bnbl3  &&w/   )rc|j}tjj|||j}t |||j ||j |jjj|<|S)a*Makes a getitem call Parameters ---------- obj : ir.Var the object being indexed index : ir.Var the index val : ir.Var the ty Returns ------- res : ir.Expr the retrieved value r() r"rrCgetitemrrr,rr r$)rrZr[r+r]r`s rr`zParforLoweringBuilder.getitemsf"]]''//#u$))/<2; CHHr%**~3  &&w/rN)pf_const)pf_tuple) pf_assign)r- __module__ __qualname____doc__rpropertyrrr"r%r:rrQr?rIr/rSrCrUrXr^r`rrr r s %%,,,,..;=:2  ( rvv *BFF..-/277&"**2"''rr N) collectionsr numba.corerrnumba.core.typingrr1r rhrrrls," ';8 JJr