pPj,,dZddlmZddlmZmZmZmZmZddl m Z ddl m Z m Z mZmZddlmZmZmZmZmZmZmZmZmZddlmZmZmZmZddlm Z dd l!m"Z"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,dd Z- dd Z. dd Z/ddZ0ddZ1y)zGenerate a class that represents a nested function. The class defines __call__ for calling the function and allows access to non-local variables defined in outer scopes. ) annotations)CPYFUNCTION_NAME ENV_ATTR_NAMEIS_FREE_THREADEDPROPSET_PREFIX SELF_NAME)ClassIR)FuncDeclFuncIR FuncSignature RuntimeArg) BasicBlockBranchCallGetAttrIntegerRegisterReturnSetAttrValue) RInstancec_pointer_rprimitiveint_rprimitiveobject_rprimitive) IRBuilder)FuncInfo ImplicitClass)cpyfunction_get_annotationscpyfunction_get_codecpyfunction_get_defaultscpyfunction_get_kwdefaultscpyfunction_get_namecpyfunction_set_annotationscpyfunction_set_name method_new_opc|jjdx}}d}||jvr%|dzt|z}|dz }||jvr%|jj |t ||j dd}d|_|jjrd|_ |jdjrRt|jdj|jt<|j j t|g|_t%||j_|j(j+||j-|}|j/||jj0j2|jj&_|jj6s9|jj8r"t;|||jj<yyy) aGenerate an (incomplete) callable class representing a function. This can be a nested function or a function within a non-extension class. Also set up the 'self' variable for that class. This takes the most recently visited function and returns a ClassIR to represent that function. Each callable class contains an environment attribute which points to another ClassIR representing the environment class where some of its variables can be accessed. Note that some methods, such as '__call__', are not yet created here. Use additional functions, such as add_call_to_callable_class(), to add them. Return a newly constructed ClassIR representing the callable class for the nested function. _objr_T) is_generatedis_final_classN)fn_infonamespaced_namecallable_class_namesstraddr module_namereuse_freed_instance is_nestedhas_dictfn_infoscontains_nestedr env_class attributesrfinal_attributesmrorcallable_classclassesappendadd_self_to_envreadfitemlineself_reg in_non_ext is_coroutineadd_coroutine_propertiesname)builderrG base_namecountcallable_class_ir self_targets a/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/mypyc/irbuild/callable_class.pysetup_callable_classrN,s:"//99;.get_func_wrappers<<>\:"--d3 v"--0@.GH.4f-=$$T* >! % %  $ $s"AG(AG"G "G+ c$t|j|jz }tt t t f|jd|z|j}td|jjj|j|}t||||jj|jj}||jjj d<||jjj"d<|S)aGenerate a '__call__' method for a callable class representing a nested function. This takes the blocks and signature associated with a function definition and uses those to build the '__call__' method of a given callable class, used to represent that function. N__call__)traceback_name)lenargsnum_bitmap_argsr r rrret_typer r<irrGr2r rArBmethods method_decls)rHrwblockssigr-nargs call_fn_decl call_fn_irs rMadd_call_to_callable_classrs MC// /E  I0 13chhv6FF  CJ(>(>(A(A(F(FH[H[]`aLdFGMM$6$6w}}GYGYJ5?G%%j19EG**:6 rOc |jj}|j|jjdt |t 5|j dt }|j dt tt}}|j|j||j|d|}|j||||j||jt|j|j||jt|j!t"|j|j|g|dddy#1swYyxYw)z3Generate the '__get__' method for a callable class.__get__) self_typeinstanceownerisN)rArBrgr<rzrrirtranslate_is_opr@ none_objectadd_bool_branchr[r1rrYcall_cr%)rHr-rBrinstance_block class_block comparisons rMadd_get_to_callable_classrs3 ==  D   !!#   '' 4EFW&78 '1lJL ,, LL "G$7$7$=tT   KH{+ F7<<>*+~. 7>>-',,.',,xBX1Y[_` a /   s DE00E9c|j}|jt|jjj g|j }d}|jjr!|jjj}ny|jjr7|jjs!|jjj}n,|jjr|jj}|r=t|t||j }|j!|j||S)a,Create an instance of a callable class for a function. Calls to the function will actually call this instance. Note that fn_info refers to the function being assigned, whereas builder.fn_info refers to the function encapsulating the function being turned into a callable class. N)rAr1rr<rzctorrBr- is_generatorgenerator_class curr_env_regr4is_comprehension_scoper7rrmark_as_initializer)rHr-rAfunc_regrset_envs rMinstantiate_callable_classrs MME{{4 6 6 9 9 > >EJJOPH"L##66CC  " "7??+Q+Q55BB  ( (33 (M<L ##% G OrON)rHrrdNone)rHrrKr rer0rdr) rHrrwzlist[Register]r}zlist[BasicBlock]r~r r-rrdr )rHrr-rrdr)rHrr-rrdr)2__doc__ __future__r mypyc.commonrrrrrmypyc.ir.class_irr mypyc.ir.func_irr r r r mypyc.ir.opsrrrrrrrrrmypyc.ir.rtypesrrrrmypyc.irbuild.builderrmypyc.irbuild.contextrrmypyc.primitives.misc_opsrrr r!r"r#r$r%rNrFrrrrOrMrs #&HH   _^+9   BSJA> A>+2A>DGA> A>H         6 >*rO