o 2g@sdZddlmZddlmZmZmZmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZmZddlmZddlmZdddZdddZdddZdS)zIRBuilder AST transform helpers shared between expressions and statements. Shared code that is tightly coupled to mypy ASTs can be put here instead of making mypyc.irbuild.builder larger. ) annotations) LDEF BytesExprComparisonExpr Expression FloatExprIntExpr MemberExprNameExprOpExprStrExpr UnaryExprVar) BasicBlock)is_fixed_width_rtype is_tagged) IRBuilder)constant_fold_exprselfrertruerfalsereturnNonecCst|trC|jdvrC|jdkr)t}t||j||||t||j||dSt}t||j||||t||j||dSt|trW|jdkrWt||j ||dSt ||||}|rbdS| |}| |||dS)N)andorrnot) isinstancer oprprocess_conditionalleftactivate_blockrightr expr$maybe_process_conditional_comparisonacceptadd_bool_branch)rrrrnewresregr*Q/home/garg/my-data/venv/lib/python3.10/site-packages/mypyc/irbuild/ast_helpers.pyrs"    rboolc CsNt|tr t|jdkrdS||jd}||jd}t|s&t|r.t|s0t|s0dS|jd}|dvr;dS|jd}|jd}t||} |j || d} |j |dd} t|s`t|rt|sm| | ||j } n t|sy| | ||j } | | | ||j } |j || ||dS|j | | ||j } ||| ||dS)akTransform simple tagged integer comparisons in a conditional context. Return True if the operation is supported (and was transformed). Otherwise, do nothing and return False. Args: self: IR form Builder e: Arbitrary expression true: Branch target if comparison is true false: Branch target if comparison is false Fr)z==z!=z>=) can_borrowT)rrlenoperands node_typerr operatorsis_borrow_friendly_exprr%coerceline binary_opbuilderflush_keep_alivesr&) rrrrltypertyper left_expr right_expr borrow_leftr r"r)r*r*r+r$8sD     r$r#cCszt|ttttfr dSt|ttttfrt ||durdSt|tr/t|j t r/|j t kr/dSt|tr;||r;dSdS)zCan the result of the expression borrowed temporarily? Borrowing means keeping a reference without incrementing the reference count. TNF)rrrr rr r r r rnoderkindris_native_attr_ref)rr#r*r*r+r6gs r6N) rrrrrrrrrr) rrrrrrrrrr,)rrr#rrr,)__doc__ __future__r mypy.nodesrrrrrrr r r r r r mypyc.ir.opsrmypyc.ir.rtypesrrmypyc.irbuild.builderrmypyc.irbuild.constant_foldrrr$r6r*r*r*r+s 8     /