o 8ήc@sTddlmZddlmZmZddlmZddlmZedddgZ Gdd d Z d S) ) namedtuple)typesir) mk_unique_var) 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)selflowererscopelocrJ/tmp/pip-target-vg8gfxp4/lib/python/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)r r)).0xrrr _sz=ParforLoweringBuilder.make_tuple_variable..r&)rrTuple from_typesr,rExpr build_tuple)rvarlistr)rvartystuptyrrrmake_tuple_variablePs  z)ParforLoweringBuilder.make_tuple_variable pf_assigncCsD|j}t|jt||}||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 ) rrVarr rr r)Assignr lower_instrr'r(r)rvarr,rrrr,es   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 ) rrrJr rKr setdefaultr)r rLrMrrrassign_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%)rrCcallrrr r#)r callable_noder1r2rQrrrrQs 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%) rr rSetItemrrnoner)r rr"rL)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%) r rrCgetitemrrr)r rr")rrUrVr(rXrZrrrrZs zParforLoweringBuilder.getitemN)r8)r=)rI)r* __module__ __qualname____doc__rpropertyrrr r#r7rrJr<rHr,rPrCrQrSrYrZrrrrr s&     r N) collectionsr numba.corerrnumba.core.ir_utilsrnumba.core.typingrr.r rrrrs