o 2g @sUdZddlmZddlmZmZddlmZmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZddlmZddlmZeeeeeefZeeeeefZde d <dddZ!dddZ"dS)adConstant folding of IR values. For example, 3 + 5 can be constant folded into 8. This is mostly like mypy.constant_fold, but we can bind some additional NameExpr and MemberExpr references here, since we have more knowledge about which definitions can be trusted -- we constant fold only references to other compiled modules in the same compilation unit. ) annotations)FinalUnion)constant_fold_binary_opconstant_fold_unary_op) BytesExpr ComplexExpr Expression FloatExprIntExpr MemberExprNameExprOpExprStrExpr UnaryExprVar) IRBuilder)bytes_from_strr CONST_TYPESbuilderrexprr returnConstantValue | Nonec CsLt|tr|jSt|tr|jSt|tr|jSt|tr"t|jSt|tr*|jSt|trF|j }t|t rD|j rD|j }t|t rD|SdSt|trh||}|durf|\}}}|j rf|j }t|t rf|SdSt|trt||j}t||j} |dur| durt|j|| SdSt|trt||j} | durt| tst|j| SdS)zeReturn the constant value of an expression for supported operations. Return None otherwise. N) isinstancer valuer rrrrr noderis_final final_valuerr get_final_refrconstant_fold_exprleftright constant_fold_binary_op_extendedoprrbytesr) rrrrfinalfn final_varnativer r!rr)S/home/garg/my-data/venv/lib/python3.10/site-packages/mypyc/irbuild/constant_fold.pyr%sJ               rr#strr ConstantValuer!cCst|tst|tst|||S|dkr"t|tr"t|tr"||S|dkr4t|tr4t|tr4||S|dkrFt|trFt|trF||SdS)zLike mypy's constant_fold_binary_op(), but includes bytes support. mypy cannot use constant folded bytes easily so it's simpler to only support them in mypyc. +*N)rr$rint)r#r r!r)r)r*r"Ns r"N)rrrr rr)r#r+r r,r!r,rr)#__doc__ __future__rtypingrrmypy.constant_foldrr mypy.nodesrrr r r r r rrrrmypyc.irbuild.builderrmypyc.irbuild.utilrr/floatcomplexr+r$r,r__annotations__rr"r)r)r)r*s 4  )