U ?h@sHddlmZddlmZmZddlmZedddgZGdddZd S) ) namedtuple)typesir) signatureZ BoundFuncfuncsigc@seZdZdZddZeddZeddZedd Zed d Z ifd d Z d!e j dddZ d"e j dddZd#e j dddZe j dddZife jdddZe jdddZe jdddZd S)$ParforLoweringBuilderz@Helper class for building Numba-IR and lowering for Parfor. cCs||_||_||_dSN)_lowerer_scope_loc)selfZlowererscopelocrU/var/www/html/venv/lib/python3.8/site-packages/numba/parfors/parfor_lowering_utils.py__init__ szParforLoweringBuilder.__init__cCs|jjSr )r contextr rrr_contextszParforLoweringBuilder._contextcCs|jjSr )rZtyping_contextrrrr _typingctxsz ParforLoweringBuilder._typingctxcCs |jjjSr )r fndesctypemaprrrr_typemapszParforLoweringBuilder._typemapcCs |jjjSr )r r calltypesrrrr _calltypessz 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 Z_funcrrhstypname)rr)r __name__rZresolve_function_typeassignrGlobal _CallableNode) r ZfobjZftypeargskwsrvarnameZgvnameZfunc_sigZfunc_varrrrbind_global_function"s z*ParforLoweringBuilder.bind_global_functionpf_const)returncCs|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 rr)r"rZConstr )r cvalrr rrrmake_const_variable;s z)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 rTupleZ from_typesr"rExprZ build_tuple)r Zvarlistr rZvartysZtuptyrrrmake_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 Zredefinerr rAssignr lower_instr rrr 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 r5r setdefaultr r r6r7rrrassign_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)rr2callrr rr)r Z callable_noder%r&r<rrrr<s 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 rrr6)r objindexvalrtmsetitemrrrrCs 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) rrr2getitemr rr r rr)r r?r@rrBrDrrrrDszParforLoweringBuilder.getitemN)r))r-)r4)r! __module__ __qualname____doc__rpropertyrrrrr(rr9r,r3r"r;r2r<r=rCrDrrrrr s$     rN) collectionsrZ numba.corerrZnumba.core.typingrr$rrrrrs