o ۾i@sHddlmZddlmZmZddlmZedddgZGdddZd S) ) namedtuple)typesir) signature BoundFuncfuncsigc@seZdZdZddZeddZeddZedd Zed d Z ifd d Z d!de j fddZ d"de j fddZd#de j fddZde j fddZifde jfddZde jfddZde jfddZd S)$ParforLoweringBuilderz@Helper class for building Numba-IR and lowering for Parfor. cCs||_||_||_dSN)_lowerer_scope_loc)selflowererscopelocrW/home/ubuntu/.local/lib/python3.10/site-packages/numba/parfors/parfor_lowering_utils.py__init__ s zParforLoweringBuilder.__init__cC|jjSr )r contextrrrr_contextzParforLoweringBuilder._contextcCrr )rtyping_contextrrrr _typingctxrz ParforLoweringBuilder._typingctxcC |jjjSr )r fndesctypemaprrrr_typemap zParforLoweringBuilder._typemapcCrr )r r calltypesrrrr _calltypesr z ParforLoweringBuilder._calltypesc CsR|j}|jd}|j}|j|||}|jtj|||d||d} t| |dS)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_varrrrbind_global_function"s  z*ParforLoweringBuilder.bind_global_functionpf_constreturncCs|jtj||jd||dS)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(rrrmake_const_variable;sz)ParforLoweringBuilder.make_const_variablepf_tuplecs>j}fdd|D}tj|}jtj||||dS)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 csg|]}j|jqSr)rr().0xrrr ^sz=ParforLoweringBuilder.make_tuple_variable..r%)r rTuple from_typesr+rExpr build_tuple)rvarlistr(rvartystuptyrrrmake_tuple_variableOs  z)ParforLoweringBuilder.make_tuple_variable pf_assigncCs>|j}|j||}||j|j<t|||}|j||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 ) r r redefinerr(rAssignr lower_instrr&r'r(rvarr+rrrr+ds   zParforLoweringBuilder.assigncCsD|j}t|j||}t|||}|j|j||j ||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 ) r rVarr rJr setdefaultr(r rKrLrrrassign_inplace{s  z$ParforLoweringBuilder.assign_inplacecCs(tjj|j|||jd}|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$)rrBcallrr rr")r callable_noder0r1rQrrrrQs zParforLoweringBuilder.callcCsZ|j}|j}tj||||d}ttj||j||j||j|jj j |<|j ||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$) r rrSetItemrrnoner(r rr!rK)robjindexvalrtmsetitemrrrrYs zParforLoweringBuilder.setitemcCs@|j}tjj|||jd}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$) rrrBgetitemr rr(r rr!)rrUrVr'rXrZrrrrZs zParforLoweringBuilder.getitemN)r7)r<)rH)r) __module__ __qualname____doc__rpropertyrrrr"r6rrNr;rGr+rPrBrQrSrYrZrrrrr s&     r N) collectionsr numba.corerrnumba.core.typingrr-r rrrrs