o 2ga@sdZddlmZddlmZddlmZddlmZddl m Z m Z ddl m Z Gdd d ZGd d d ZGd d d eZdS)zGHelpers that store information about functions and the related classes.) annotations)FuncItem)ClassIR)INVALID_FUNC_DEF) BasicBlockValue)AssignmentTargetc @seZdZdZeddddddddf d-ddZd.ddZed/ddZed/ddZ ed0ddZ e j d1d dZ ed2d"d#Z e j d3d%d#Z ed4d'd(Z e j d5d)d(Z ed6d+d,ZdS)7FuncInfoz;Contains information about functions as they are generated.NFfitemrnamestr class_name str | None namespace is_nestedboolcontains_nested is_decorated in_non_extadd_nested_funcs_to_envreturnNonec CsR||_||_||_||_d|_d|_d|_d|_||_||_ ||_ ||_ | |_ dSN) r r rns_callable_class _env_class_generator_class _curr_env_regrrrrr) selfr r rrrrrrrr M/home/garg/my-data/venv/lib/python3.10/site-packages/mypyc/irbuild/context.py__init__s  zFuncInfo.__init__cCs ddd|j|j|jfDS)N_css|]}|r|VqdSrr ).0xr r r! 8sz+FuncInfo.namespaced_name..)joinr rrrr r r!namespaced_name7s zFuncInfo.namespaced_namecCs|jjp|jjSr)r is_generator is_coroutiner(r r r!r*:szFuncInfo.is_generatorcCs|jjSr)r r+r(r r r!r+>szFuncInfo.is_coroutine ImplicitClasscC|jdusJ|jSrrr(r r r!callable_classBzFuncInfo.callable_classclscC ||_dSrr.rr1r r r!r/G rcCr-rrr(r r r! env_classKr0zFuncInfo.env_classircCr2rr5rr7r r r!r6Pr4GeneratorClasscCr-rrr(r r r!generator_classTr0zFuncInfo.generator_classcCr2rr:r3r r r!r;Yr4rcCr-rrr(r r r! curr_env_reg]r0zFuncInfo.curr_env_reg)r rr r rrrr rrrrrrrrrrrr)rr )rr)rr,)r1r,rr)rrr7rrr)rr9)r1r9rrrr)__name__ __module__ __qualname____doc__rr"r)propertyr*r+r/setterr6r;r=r r r r!r s> (        r c@steZdZdZdddZedd d Zejdd d Zedd dZejdddZedddZ e jdddZ dS)r,aContains information regarding implicitly generated classes. Implicit classes are generated for nested functions and generator functions. They are not explicitly defined in the source code. NOTE: This is both a concrete class and used as a base class. r7rrrcCs||_d|_d|_d|_dSr)r7 _self_regr _prev_env_regr8r r r!r"ls zImplicitClass.__init__rcCr-rrFr(r r r!self_regxr0zImplicitClass.self_regregcCr2rrHrrJr r r!rI}r4cCr-rr<r(r r r!r=r0zImplicitClass.curr_env_regcCr2rr<rKr r r!r=r4cCr-rrGr(r r r! prev_env_regr0zImplicitClass.prev_env_regcCr2rrLrKr r r!rMr4Nr>r?rJrrr) r@rArBrCr"rDrIrEr=rMr r r r!r,cs      r,cs^eZdZdZdfdd Zedd d Zejdd d ZedddZejdddZZ S)r9z?Contains information about implicit generator function classes.r7rrrcs6t|d|_d|_d|_d|_t|_g|_dSr) superr"_next_label_reg_next_label_targetexc_regs send_arg_regr switch_blockcontinuation_blocksr8 __class__r r!r"s  zGeneratorClass.__init__rcCr-rrPr(r r r!next_label_regr0zGeneratorClass.next_label_regrJcCr2rrXrKr r r!rYr4rcCr-rrQr(r r r!next_label_targetr0z GeneratorClass.next_label_targettargetcCr2rrZ)rr\r r r!r[r4r>r?rN)rr)r\rrr) r@rArBrCr"rDrYrEr[ __classcell__r r rVr!r9s   r9N)rC __future__r mypy.nodesrmypyc.ir.class_irrmypyc.ir.func_irr mypyc.ir.opsrrmypyc.irbuild.targetsrr r,r9r r r r!s     W1