o 7f@stUdZddlmZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8dd l9m:Z:dd l;mm?Z?m@Z@mAZAdd lBmCZCddlDmEZEddlFmGZGmHZHddlImJZJmKZKmLZLmMZMmNZNddlOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`ddlambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrddlsmtZtmuZuddlvmwZwddlxmyZyddlzm{Z{m|Z|m}Z}m~Z~ddlmZddlmZddlmZmZmZmZmZddlmZmZddlmZddlmZmZddlmZmZmZddlmZmZmZddlmZmZmZdd lmZmZhd!Zd"ed#<Gd$d%d%e!===>=+-<><=rint_borrow_friendly_opc@ eZdZdS) IRVisitorN__name__ __module__ __qualname__rrN/home/ertert/spirit/venv/lib/python3.10/site-packages/mypyc/irbuild/builder.pyrrNc@r)UnsupportedExceptionNrrrrrrrrc@s&eZdZdtddZduddZedddvd"d#Zedwd%d#Zdddxd(d#Zdyd)d*Zdzd-d.Zd{d1d2Z d|d4d5Z d|d6d7Z d}d9d:Z d~d?d@Z ddBdCZddDdEZddFdGZddIdJZddMdNZddPdQZdddVdWZddXdYZddZd[Zdd\d]Zdd^d_ZddbdcZddddeZddfdgZ h hdddodpZddsdtZddvdwZ h hddd}d~ZdddZ dddZ!dddZ"dddZ#dddZ$dddZ%dddZ&dddZ'dddZ(dddZ)dddZ*dddZ+dddZ,dddZ-dyddZ.dyddZ/dyddZ0dddZ1dddZ2dddZ3 hddhddddĄZ4 hddddȄZ5ddd˄Z6 ̐ddd͜dddфZ7 ddddԄZ8dddׄZ9dddۄZ:dddބZ;dddZ<dddZ=dddZ>dyddZ?dddZ@dddZAdddZBdddZCdddZDdddZEdddZFdddZGdddZHdddZIdddZJdddZKddd ZLdd d ZMdd dZNdddZOdddZPdddZQdddZRddd ZSdd"d#ZTdd'd(ZUddd,d-ZVdd/d0ZWeX ) hddd6d7ZYeZfdd<d=Z[dÐdAdBZ\ddĐdDdEZ] ddŐdGdHZ^dƐdJdKZ_dǐdLdMZ`dȐdOdPZadɐdQdRZb ddʐdUdVZcddWdXZddːdZd[Zed̐d\d]Zfdd^d_Zgd͐d`daZhddbdcZidydddeZjdΐdgdhZkdϐdjdkZldϐdldmZmdϐdndoZndАdrdsZodhS( IRBuildercurrent_modulestrtypesdict[Expression, Type]graphr errorsr9mapperrfpbvrkvisitorroptionsrtsingledispatch_impls%dict[FuncDef, list[RegisterImplInfo]]returnNonec Cst|||||_|jg|_ig|_gg|_g|_g|_dg|_||_||_ ||_ ||_ g|_ g|_ t|_g|_g|_t|_||_d|_d|_|j|_|j|_|j|_|j|_|j|_|j|_| |_||_ t!t"dd|_#|j#g|_$g|_%||_&i|_'d|_(dS)NTrF))rebuilderbuilders symtables runtime_argsfunction_name_stackclass_ir_stackblock_reachable_stackrrrr ret_types functionssetfunction_namesclasses final_namescallable_class_namesrlambda_counter temp_counterfree_variables prop_settersencapsulating_funcs nested_funcskeys nested_fitemsfuncs_to_decoratorsfdefs_to_decoratorsmodule_import_groupsrrrcr<fn_infofn_infosnonlocal_controlrimports can_borrow) selfrrrrrrrrrrrr__init__sD    zIRBuilder.__init__ module_name module_pathcCs||_||_|j||dS)zuSet the name and path of the current module. This must be called before transforming any AST nodes. N)rrr set_module)rrrrrrrszIRBuilder.set_moduleFrnoderrboolrQcCdSNr)rrrrrracceptzIRBuilder.acceptr cCrrr)rrrrrrrStatement | Expression Value | Nonec Cs||j]t|trH|j}||_z||j}|||||j}Wnt y5t ||}Ynw||_|s?| |WdSz||jWn t yYYnw WddS1sfwYdS)aTransform an expression or a statement. If can_borrow is true, prefer to generate a borrowed reference. Borrowed references are faster since they don't require reference count manipulation, but they are only safe to use in specific contexts. N) r8line isinstancerrrrcoerce node_typerrMflush_keep_alives)rrrold_can_borrowresrrrrs,    $cC|jdSr)rrrrrrrzIRBuilder.flush_keep_alivesoprKcC |j|Sr)radd)rrrrrr z IRBuilder.addtargetrCcC|j|dSr)rgoto)rrrrrrzIRBuilder.gotoblockcCrr)ractivate_blockrrrrrrrzIRBuilder.activate_blockcCrr)rgoto_and_activaterrrrrrzIRBuilder.goto_and_activaterMcC |jSr)rrrrrrr  zIRBuilder.selfobjattrrintcC|j|||Sr)r py_get_attr)rrrrrrrr#rzIRBuilder.py_get_attrvaluecCrr)rload_strrrrrrr&rzIRBuilder.load_strcCs|jt|S)zLoad bytes object from a string literal. The literal characters of BytesExpr (the characters inside b'') are stored in BytesExpr.value, whose type is 'str' not 'bytes'. Thus we perform a special conversion here. )r load_bytesrrrrrrload_bytes_from_str_literal)sz%IRBuilder.load_bytes_from_str_literalcCrr)rload_intrrrrr2rzIRBuilder.load_intfloatcCrr)r load_floatrrrrr5rzIRBuilder.load_floatlregexpr_opcCrr)runary_op)rrrrrrrr8rzIRBuilder.unary_oprregcC|j||||Sr)r binary_oprrrrrrrrr;zIRBuilder.binary_opsrc target_typerTforcecCs|jj|||||jdS)Nr)rrr)rrrrrrrrr>szIRBuilder.coercecCrr)r none_objectrrrrrArzIRBuilder.none_objectcCrr)rnonerrrrrDrzIRBuilder.nonecCrr)rtruerrrrrGrzIRBuilder.truecCrr)rfalserrrrrJrzIRBuilder.falsevalues list[Value]cC|j||Sr)r new_list_oprrrrrrrMrzIRBuilder.new_list_opcCrr)r new_set_opr rrrr PrzIRBuilder.new_set_opcCrr)rtranslate_is_oprrrrr SrzIRBuilder.translate_is_opNfunction arg_values arg_kindslist[ArgKind] | None arg_namesSequence[str | None] | NonecC|j|||||Sr)rpy_call)rr r rrrrrrrVszIRBuilder.py_callrrcCs|j|||dSr)radd_bool_branch)rrrrrrrr`zIRBuilder.add_bool_branchfullnamecCrr)rload_native_type_object)rrrrrrcrz!IRBuilder.load_native_type_objectbasename result_type RType | Nonelist[str | None] | Nonec Cs|j||||||||jSr)rgen_method_callr)rrrr rrrrrrrrfs zIRBuilder.gen_method_callcCrr)r load_module)rrrrrrtrzIRBuilder.load_moduledescrargscCrr)rcall_c)rrr rrrrr!wrzIRBuilder.call_ctypelhsrhscCrr)rint_op)rr"r#r$rrrrrr%zrzIRBuilder.int_opcCrr)rcompare_taggedrr#r$rrrrrr&}rzIRBuilder.compare_taggedcCrr)rcompare_tuplesr'rrrr(rzIRBuilder.compare_tuplesvalcCrr)r builtin_len)rr)rrrrr*rzIRBuilder.builtin_lenitemscCrr)r new_tuple)rr+rrrrr,rzIRBuilder.new_tuplenon_extr;keycCs$||}|t|j||g|dSr)rr!rvdict)rr-r.r)r key_unicoderrradd_to_non_ext_dicts zIRBuilder.add_to_non_ext_dictidcCsjd|j|<tt}}||||||||t||g|}|t||t d| |dS)N) namespace) rrCcheck_if_module_loadedrr!rrrrGrAr)rr2r needs_importoutrrrr gen_imports  zIRBuilder.gen_importr5r6cCs0||}|||d|}||||dS)anGenerate code that checks if the module `id` has been loaded yet. Arguments: id: name of module to check if imported line: line number that the import occurs on needs_import: the BasicBlock that is run if the module has not been loaded yet out: the BasicBlock that is run if the module has already been loadedzis notN)rr rr)rr2rr5r6 first_load comparisonrrrr4s z IRBuilder.check_if_module_loadedmodulecCs&|tg|}|t|||g|Sr)r!r~rur)rr:rmod_dictrrr get_moduleszIRBuilder.get_modulecCs&||||||}||||S)zLook up an attribute of a module without storing it in the local namespace. For example, get_module_attr('typing', 'TypedDict', line) results in the value of 'typing.TypedDict'. Import the module if needed. )r7r<r)rr:rr module_objrrrget_module_attrs  zIRBuilder.get_module_attrget_valCallable[[], Value]c Csdtt}}|t|||tj|||t||||j|||||dS)z:If target is NULL, assign value produced by get_val to it.N) rCrrDIS_ERRORrrBrr"r)rrr?r error_block body_blockrrrassign_if_nulls   zIRBuilder.assign_if_nullindexc Cstt}}|t|jjd|ttd|td@>ttj|}| t |tdtt j }| t |||t j ||| t||||j|||||dS)Nr)rCr%rVrr r5rHrIANDrrEEQrDBOOLrrBrr"r) rrr?rErrBrCobrrrassign_if_bitmap_unsets  z IRBuilder.assign_if_bitmap_unsetcCs4t|jdst|jdr|dS|dSNrF)r\rr]add_implicit_returnadd_implicit_unreachablerrrrmaybe_add_implicit_returns  z#IRBuilder.maybe_add_implicit_returncCsN|jjd}|js%||j|jdd}|jd|||jj j dSdSrN) rblocks terminatedrrrr gen_returnrfitemr)rrretvalrrrrOs zIRBuilder.add_implicit_returncCs&|jjd}|js|tdSdSrN)rrRrSrrPrrrrrPs z"IRBuilder.add_implicit_unreachablelvalues list[Lvalue]cCs^|D]*}t|tr,t|jtr,t|jjtr,|jj|jj}t|tr,|js,| d|qdS)Nz;Only class variables defined as ClassVar can be assigned to) rrexprrrr$rr& is_classvarerror)rrWrlvaluevarrrrdisallow_class_assignmentss   z$IRBuilder.disallow_class_assignmentscCst|jdkSN)lenrrrrrnon_function_scopeszIRBuilder.non_function_scopeFuncInfo | NonecCs|rdS|jdSr_)rbrrrrrtop_level_fn_infos zIRBuilder.top_level_fn_info) type_overrider\r rvalue_reg class_name str | NonerecCst|tsJt|jtsJ|jjdurO|dur|j}n|d|j}|dus-Jd|||p6|||j}|j ||j f| t |||jdSdS)N.zFull name not set for variable)rrrr& final_valuerrrrrappendr"rrGr)rr\rfrgrercoercedrrrinit_final_statics zIRBuilder.init_final_statictyp error_namecCs>t|j|}|dus J|\}}|jj||||d|ddS)Nzvalue for final name "z " was not set)r error_msg)r2rrload_static_checked)rrrnrro split_namer:rrrrload_final_statics   zIRBuilder.load_final_static*int | str | bytes | float | complex | boolcCst|tr|r |S|St|tr|j|St|tr%|j|St|t r0|j |St|t r;|j |St|t rF|j|SJd)zQLoad value of a final name, class-level attribute, or constant folded expression.FzUnsupported literal value)rrrrrrrrrrrbytesrcomplex load_complex)rr)rrrload_literal_value#s           zIRBuilder.load_literal_valuerF)for_readryrmcCs|dkr|j}t|tr|j}t|tr|j}|dur%|js Jt|j}|s5t|tr5|j r5| d||j t krr||j dvrmt|trSt|jtsS||j}n||}|jjrg|j|||jjddS|||S||S|j tkr|}||j}t||SJ|j t|tr||j}||j} t|| St|t r|!|} |j|j"| d} t#| |j| dSt|t$rd} g} t%|j&D]\}}|'|}| (|t|t)r| dur| d||} qt*| | St|t)r|'|j"SJd|)NrFz2Cannot assign to the first argument of classmethodF)reassignrz%Two starred expressions in assignmentUnsupported lvalue: %r)+rrrrrfuncis_special_formr&ris_clsr[kindrrr"r( type_to_rtyperr is_generatoradd_var_to_env_classgenerator_class add_local_reglookuprload_globals_dictrrorrrrEris_native_attr_refrYrnr" enumerater+get_assignment_targetrkrrq)rr\rrysymbolreg_type globals_dictrrrErrstar_idxrWidxitemtargrrrr7sd                             zIRBuilder.get_assignment_targetValue | AssignmentTargetcCst|tr|St|tr|jSt|tr-||jd|jg|j|}|dur'|SJ|jjt|t rZt|j jt rQ|j jj j rQ|oC|j}|t|j |j||dS||j |j|SJd|)N __getitem__F)borrowr{)rrQrpregisterrorrrEr"rnrrRclass_ir is_ext_classrrrFrr)rrrrregrrrrreads"       zIRBuilder.readRegister | AssignmentTargetc Cst|tr|t||||j|dSt|tr-|||j|}|t|j|dSt|trgt|j t rN|||j|}|t |j |j ||dS||j }|j|}|t|j ||g|dSt|tr||jd|j|gd|}|dusJ|jjdSt|trt|jtr|jdur|jj}t|t|jksJtt|D]}|t|||} ||j|| |qdSt |jst!|jr|jdur|"|||dS|#|||dSJd)N __setitem__FzUnsupported assignment target)$rrMrrB coerce_rvaluer"rprrnobj_typerRrNrrrrboxr!ryrorrrErqrSrrrar+rangerOassignr[r_process_sequence_assignment!process_iterator_tuple_assignment) rrrfrr. boxed_reg target_reg2rtypesi item_valuerrrrsB          zIRBuilder.assignrvaluertypecCsNt|r t|jr |j}|dkrd}|dd|d|||||S)N short_intrz1Incompatible value representations in assignment z(expression has type "z", variable has type "float"))rZr^r" short_namer[r)rrrrtypenamerrrrs  zIRBuilder.coerce_rvaluerqc Cstt|jt}|jt||g|g}tt|jD]-}|j|}|j|}t |j r7|t ||g|} n |j |d|g|j |} | | qt|j|D] \} } || | |qOdS)zOProcess assignment like 'x, y = s', where s is a variable-length list or tuple.rN)rHrar+rWrr!r}rrr[r"rzrrkzipr) rrrr expected_lenrrrrErr\rrrrrs    z%IRBuilder.process_sequence_assignmentlitemritemcCsjtt}}|t|||tj|||ttjd||t||||||dS)Nnot enough values to unpack) rCrrDrArrL VALUE_ERRORrPr)rrrrrBok_blockrrr(process_iterator_tuple_assignment_helpers    z2IRBuilder.process_iterator_tuple_assignment_helpercCs|t|g|}|jdur|jnt|j}|jd|D]=}|t|g|}tt}} |t||| tj | ||t t j d||t | | ||||q|jdur|j|dd} |t|g|} || |} tt| } || | d|}tt}} |t|| |tj| ||t t j d||t | | t| D]}|t| g|}||||q||j|j| |dS|t|g|}tt}} |t|| |tj | ||t t j d||t | | dS)NrrGrztoo many values to unpack)r!rwrrar+rxrCrrDrArrLrrPrr|r*rHrrJreversedr{)rrrfriterator split_idxrrrBrpost_star_vals iter_list iter_list_len post_star_len conditionextrarrrrs\            z+IRBuilder.process_iterator_tuple_assignmentcontinue_block break_blockcCs|jt|jd||dSrN)rrkri)rrrrrrpush_loop_stack5szIRBuilder.push_loop_stackcCrr)rpoprrrrpop_loop_stack:rzIRBuilder.pop_loop_stackcCs6t|j}|jd7_|t|||jj}|S)IMoves a given Value instance into the generator class' environment class.rG)r7rrr&rr)rr"rrrrrmake_spill_target=szIRBuilder.make_spill_targetcCs||j}|||d|S)rrF)rr"r)rrrrrrspillDs zIRBuilder.spillcCs|jjr ||S|S)aE Moves a given Value instance into the environment class for generator functions. For non-generator functions, leaves the Value instance as it is. Returns an AssignmentTarget associated with the Value for generator functions and the original Value itself for non-generator functions. )rrrrrrr maybe_spillKs zIRBuilder.maybe_spillcCs<|jjr ||St|tr|St|j}|||d|S)a= Moves a given Value instance into the environment class for generator functions. For non-generator functions, allocate a temporary Register. Returns an AssignmentTarget associated with the Value for generator functions and an assignable Register for non-generator functions. rF)rrrrrMr"r)rrrrrrmaybe_spill_assignableWs   z IRBuilder.maybe_spill_assignablee int | NonecCs>t|tr|jSt|tr|jdkrt|jtr|jj SdS)Nr)rrrr%rrY)rrrrr extract_intjs   zIRBuilder.extract_intrYcCs||j|Sr)get_sequence_type_from_typerrrYrrrget_sequence_typerrzIRBuilder.get_sequence_typer,cs^t|}t|trtfdd|jDSt|tsJ||jjdkr't S |j dS)Ncsg|]}|qSr)r.0rrrr ysz9IRBuilder.get_sequence_type_from_type..z builtins.strr) r1rr0rUmake_simplified_unionr+r)r"rrbrr )rrrrrrus  z%IRBuilder.get_sequence_type_from_typelist[Instance]cCst|j|}t|trdd|jD}n|g}g}|D]2}t|tr2|j}tdd|jj D}nt|t s;J|tdd|jj D}| t ||q|S)zpFind dict type of a dict-like expression. This is useful for dict subclasses like SymbolTable. cSsg|]}t|qSr)r1rrrrrsz0IRBuilder.get_dict_base_type..cs|] }|jdkr|VqdS)ztyping.MappingNrrrrrr z/IRBuilder.get_dict_base_type..csr)z builtins.dictNrrrrrrr) r1rrr0r+r-fallbacknextr"mror)rkr )rrYrr dict_typest dict_baserrrget_dict_base_types  zIRBuilder.get_dict_base_typecsF|}t|dkr|djdSfdd|D}t|S)NrGrcg|] }|jdqS)rrr rrrrrrz/IRBuilder.get_dict_key_type..rrarr rUrrrYdict_base_typesrrrrget_dict_key_type   zIRBuilder.get_dict_key_typecsF|}t|dkr|djdSfdd|D}t|S)NrGrcr)rGrrrrrrrz1IRBuilder.get_dict_value_type..rrrrrget_dict_value_typerzIRBuilder.get_dict_value_typecCs ||}||}t||gSr)rrrS)rrYkey_type value_typerrrget_dict_item_types   zIRBuilder.get_dict_item_typecCs||jvr ttj}nt|j|}|j|jj}| d|gg|d}ddl m }t |t rAt}|jD]}|||}q7|S| d|gg|dS)z.)rr)rrrraget_call_target_fullnamerrrmatching_call_crrrrrimplrr|rrr func_to_declallcallrrr) rrYr r rcall_c_ops_candidatesr callee_nodedeclr rrrcall_refexpr_with_args s8       z IRBuilder.call_refexpr_with_argsrcs2jjfddfddjS)Nc jSr)rleftrrYrrr9 z-IRBuilder.shortcircuit_expr..crr)rrightrrrrr:r)rshortcircuit_helperrrrrrrrshortcircuit_expr5s  zIRBuilder.shortcircuit_exprargRefExpr | TupleExprlist[ClassIR] | NonecCst|trt|jtr||r|jj|j}|r|gSdSg}|jD]}t|tt fr?| |}|dur9dS| |q$dS|S)zFlatten classes in isinstance(obj, (A, (B, C))). If at least one item is not a reference to a native class, return None. N) rrrr$rr type_to_irrr+r"flatten_classesextend)rr irrr item_partrrrr$@s    zIRBuilder.flatten_classesrrFuncInfo | strcCst|tr t|d}t|j|j|j|j|_|j |j |j |j |j|j i|j g||_|j |j|j t|jrM|j tn|j t|tdS)Nr)rrrcrerrrrrrrrrrkrrrrrr`rrrhrgrrC)rrrrrenterWs     zIRBuilder.enterJtuple[list[Register], list[RuntimeArg], list[BasicBlock], RType, FuncInfo]cCsf|j}|j|j}|j}|j}|j|jd|_|jd|_|j ||j ||fSrN) rrrrrrrrrr rR)rrrret_typerrrrleavejs        zIRBuilder.leaverr:r, self_typeIterator[None]c cs:|||j||j|||jd<|durt|}|t|z>dVW|\}}}}}t ||} |j }|j }t ||j |j | } t| ||} | |j|<| j|j|<|j| dS|\}}}}}t ||} |j }|j }t ||j |j | } t| ||} | |j|<| j|j|<|j| w)a5Generate IR for a method. If the method takes arguments, you should immediately afterwards call add_argument() for each non-self argument (self is created implicitly). Args: class_ir: Add method to this class name: Short name of the method ret_type: Return type of the method fn_info: Optionally, additional information about the method self_type: If not None, override default type of the implicit 'self' argument (by default, derive type from class_ir) rFN)r*rrkrrrR add_argumentr6r-r?rr=rrr>methodsr method_declsr) rrrr,rr.arg_regsr rRsigrr&rrr enter_methodus8                 zIRBuilder.enter_methodr] str | VarrrcCs@t|tr t|}|j||dd}|jdt|j|||S)zDeclare an argument in the current function. You should use this instead of directly calling add_local() in new code. Tis_argrF)rrr& add_localrrkr@r)rr]rnrrrrrr0s zIRBuilder.add_argumentrr! SymbolTargetcCs|jd|SrNr)rrrrrrrzIRBuilder.lookupr8cCsNt|tsJt|t|j||jd}t||jd|<|r%|jj ||S)zAdd register that represents a symbol to the symbol table. Args: is_arg: is this a function argument )r8rrF) rr!rMremangle_redefinition_namerrrprrr rk)rrrnr8rrrrr9szIRBuilder.add_localrpcCs.|||||jd|}t|tsJ|S)zALike add_local, but return an assignment target instead of value.rF)r9rrrp)rrrnr8rrrrrszIRBuilder.add_local_regclscCs|jttt|ddS)zLow-level function that adds a 'self' argument. This is only useful if using enter() instead of enter_method(). Tr7)rr&r6rR)rr=rrradd_self_to_envszIRBuilder.add_self_to_envcCs||jd|<|SrNr;)rrrrrr add_targetszIRBuilder.add_target Type | NonecCrr)rrrrrrrrzIRBuilder.type_to_rtypecCs0t|trtS||jvrtS|j|}||Sr)rrrYrrar)rr mypy_typerrrrs    zIRBuilder.node_typeFuncInfo | ImplicitClassrzcCsd||jjj|j<t|j|j}|r,||||jjj }| t |j|j||jjj | ||Sr) r env_class attributesrrn curr_env_regrrrUrrrNr?)rr]rrrz attr_targetrrrrrs  zIRBuilder.add_var_to_env_classcCs0|jsJdd|jjvo|jjdddkS)NRefExpr not resolvedrirbuiltins)rrsplitrrrris_builtin_ref_exprs"zIRBuilder.is_builtin_ref_exprrcCsr||r|js Jd||jj|jS||r1t|jtr1||js1|js+J| |jS| |j |jS)zLoads a Python-level global. This takes a NameExpr and uses its name as a key to retrieve the corresponding PyObject * from the _globals dictionary in the C-generated code. rG) rJrload_module_attr_by_fullnamerrrrr$rrload_global_strrrrrr load_globals     zIRBuilder.load_globalcCs$|}||}|t||g|Sr)rrr!ru)rrr_globalsrrrrrLs zIRBuilder.load_global_strcCs|ttd|jS)Nglobals)rrJrXrrrrrr rzIRBuilder.load_globals_dictcCs(|d\}}}||}||||SNri)rrr)rrrr:_rrrrrrK s z&IRBuilder.load_module_attr_by_fullnamecCs<||j}t|to|jjo|j|jo|j|j S)zIIs expr a direct reference to a native (struct) attribute of an instance?) rrYrrRrrhas_attrr get_method)rrY obj_rtyperrrrs   zIRBuilder.is_native_attr_refcCsd|jd<dS)zMark statements in the innermost block being processed as unreachable. This should be called after a statement that unconditionally leaves the block, such as 'break' or 'return'. FrFN)rrrrrmark_block_unreachablesz IRBuilder.mark_block_unreachablercCs t|j|Sr)r8r)rrrrrr8%rzIRBuilder.catch_errorsmsgcC|j||j|dSr)rwarningrrrVrrrrrX(zIRBuilder.warningcCrWr)rr[rrYrrrr[+rZzIRBuilder.errorcCrWr)rnoterrYrrrr[.rZzIRBuilder.notefunc_irr>cCsN|j|jf}||jvr|d|dd|dS|j||j|dS)NzDuplicate definition of "rGz" not supported by mypyc)rgrrr[rrrk)rr\rrrrr add_function1s   zIRBuilder.add_function)rrrrrr rr9rrfrrkrrrrtrrrr)rrrrrr)rrrrrrQ)rr rr)rrrrrr)rr)rrKrrQ)rrCrr)rrCrr)rrM)rrQrrrrrrQ)rrrrQ)rrrrQ)rrrrQ)rrQrrrrrrQ) rrQrrQrrrrrrQ)F) rrQrrTrrrrrrQrrQ)rrrrrrQ)NN) r rQr rrrrrrrrrQ)rrQrrCrrCrr)rrrrQ)rrQrrr rrrrrrrrrrrQ)rrrrQ)rrr rrrrrQ) r"rTr#rQr$rQrrrrrrQ) r#rQr$rQrrrrrrQ)r)rQrrrrQ)r+rrrrrQ) r-r;r.rr)rQrrrr)r2rrrrr) r2rrrr5rCr6rCrr)r:rrrrrQ)r:rrrrrrrQ)rrMr?r@rrrr) rrMr?r@rErrrrr)rWrXrrrr)rr)rrcr) r\rrfrQrgrhrerrr) rrrnrTrrrorhrrQ)r)rtrrQ)rF)r\rrrryrrrm)rFF)rrrrrrrrQ)rrrfrQrrrr)rrQrrTrrrrQ)rrqrrQrrrr)rrmrrQrrrr)rrqrfrQrrrr)rrCrrCrr)r"rTrrm)rrQrrm)rrQrr)rrQrr)rrrr)rYrrrT)rr,rrT)rYrrr)rYrrr,)r:rrr)rYrrr)rnr$rr)rYrrr)rr&rrrrr rrnr,rrrr)rYrrr)rYrr rr rrrQ)rYrrrQ)r r!rr")r)rr(rr)rr+)rN) rr:rrr,rTrr(r.rrr/)r]r6rnrTrrrrM)rr!rr:)rr!rnrTr8rrrM)rr!rnrTr8rrrp)r=r:rrp)rr!rr:rr:)rnr@rrT)rrrrT) r]r!rrTrrBrzrrrm)rYrrrQ)rrrrrrQ)rrrrrrQ)rrrr)rVrrrrr)r\r>rrrr)prrrrrr rrrrrrrrrrrrrrrrrrrrr r rrrrrr!r%r&r(r*r,r1r7r4r<r>rDrMrQrOrPr^rbrdrmrsrxrrrrrrrrrrrrrrrrrrrrrrrrrr r rrrr$r*r-rr5rr0rr9rr>r?rrrrJrMrLrrKrrUr8rXr[r[r]rrrrrs  J                                         J  #   G        +  )     rrrrcsjjd}jD]=jrFjdfdd }tts%Jj}|j j s7 j|jj q  j||jj |d7}q dS) zGenerate blocks for arguments that have default values. If the passed value is an error value, then assign the default value to the argument. rrrQcsjdusJtjrjSjjs3jdjj}j |j f t j |j Sjj}j jjjj|< tjjj|jSrP) initializerrsrr is_nestedrvariablerrrkr"rrJrcallable_classr&rDrFself_regrr)r rrUrrr get_defaultFs  z%gen_arg_defaults..get_defaultrGNr^)rrU argumentsr_rrarrprr" error_overlaprDrrM)rnbrerrrdrgen_arg_defaults:s   rirrcCs |ddS)aIRemangle names produced by mypy when allow-redefinition is used and a name is used with multiple types within a single block. We only need to do this for locals, because the name is used as the name of the register; for globals, the name itself is stored in a register for the purpose of doing dict lookups. ' __redef__)replacer)rrrr<es r<refrcCs0t|jtrt|jj}t|tr|jjS|jSr)rrr#r1rr)r"r)rmrrrrrps   r)rrrr)rrrr)rmrrr)__doc__ __future__r contextlibrtypingrrrrrr typing_extensionsr mypy.buildr mypy.maptyper mypy.nodesr rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r& mypy.typesr'r(r)r*r+r,r-r.r/r0r1 mypy.utilr2 mypy.visitorr3r4 mypyc.commonr5r6r7 mypyc.crashr8 mypyc.errorsr9mypyc.ir.class_irr:r;mypyc.ir.func_irr<r=r>r?r@ mypyc.ir.opsrArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQmypyc.ir.rtypesrRrSrTrUrVrWrXrYrZr[r\r]r^r_r`rarbmypyc.irbuild.contextrcrdmypyc.irbuild.ll_builderremypyc.irbuild.mapperrfmypyc.irbuild.nonlocalcontrolrgrhrirjmypyc.irbuild.prebuildvisitorrkmypyc.irbuild.preparerlmypyc.irbuild.targetsrmrnrorprqmypyc.irbuild.utilrrrs mypyc.optionsrtmypyc.primitives.dict_opsrurvmypyc.primitives.generic_opsrwrxrymypyc.primitives.list_opsrzr{r|mypyc.primitives.misc_opsr}r~rmypyc.primitives.registryrrr__annotations__r Exceptionrr:rrir<rrrrrsb     p4   LL       - +