o
    ØÝ2g·  ã                   @  s¬   U d Z ddlmZ ddlmZmZ ddlmZmZm	Z	m
Z
mZmZmZmZmZ eeeeeef ZeeeeefZded< d!dd„Zd"dd„Zd#dd„Zd$dd„Zd%dd„Zd S )&zTConstant folding of expressions.

For example, 3 + 5 can be constant folded into 8.
é    )Úannotations)ÚFinalÚUnion)	ÚComplexExprÚ
ExpressionÚ	FloatExprÚIntExprÚNameExprÚOpExprÚStrExprÚ	UnaryExprÚVarr   ÚCONST_TYPESÚexprr   Ú
cur_mod_idÚstrÚreturnúConstantValue | Nonec                 C  s  t | tƒr| jS t | tƒr| jS t | tƒr| jS t | tƒr | jS t | tƒrU| jdkr,dS | jdkr3dS | j}t |t	ƒrS|j
rS|j dd¡d |krS|j}t |tƒrS|S dS t | tƒrwt| j|ƒ}t| j|ƒ}|duru|durut| j||ƒS dS t | tƒrŒt| j|ƒ}|durŒt| j|ƒS dS )	aý  Return the constant value of an expression for supported operations.

    Among other things, support int arithmetic and string
    concatenation. For example, the expression 3 + 5 has the constant
    value 8.

    Also bind simple references to final constants defined in the
    current module (cur_mod_id). Binding to references is best effort
    -- we don't bind references to other modules. Mypyc trusts these
    to be correct in compiled modules, so that it can replace a
    constant expression (or a reference to one) with the statically
    computed value. We don't want to infer constant values based on
    stubs, in particular, as these might not match the implementation
    (due to version skew, for example).

    Return None if unsuccessful.
    ÚTrueTÚFalseFÚ.é   r   N)Ú
isinstancer   Úvaluer   r   r   r	   ÚnameÚnoder   Úis_finalÚfullnameÚrsplitÚfinal_valuer   r
   Úconstant_fold_exprÚleftÚrightÚconstant_fold_binary_opÚopr   r   Úconstant_fold_unary_op)r   r   r   r   r!   r"   © r&   úJ/home/garg/my-data/venv/lib/python3.10/site-packages/mypy/constant_fold.pyr       sD   






ÿþ


÷
ür    r$   r!   ÚConstantValuer"   c                 C  s  t |tƒrt |tƒrt| ||ƒS t |tƒr t |tƒr t| ||ƒS t |tƒr0t |tƒr0t| ||ƒS t |tƒr@t |tƒr@t| ||ƒS | dkrRt |tƒrRt |tƒrR|| S | dkrdt |tƒrdt |tƒrd|| S | dkrvt |tƒrvt |tƒrv|| S | dkrŠt |ttfƒrŠt |tƒrŠ|| S | dkržt |tƒržt |ttfƒrž|| S | dkr²t |ttfƒr²t |tƒr²|| S | dkrÆt |tƒrÆt |ttfƒrÆ|| S d S )Nú+Ú*ú-)r   ÚintÚconstant_fold_binary_int_opÚfloatÚconstant_fold_binary_float_opr   Úcomplex)r$   r!   r"   r&   r&   r'   r#   O   s.       r#   r,   úint | float | Nonec                 C  s  | dkr|| S | dkr|| S | dkr|| S | dkr&|dkr$|| S d S | dkr4|dkr2|| S d S | dkrB|dkr@|| S d S | dkrJ||@ S | d	krR||B S | d
krZ||A S | dkrh|dkrf||> S d S | dkrv|dkrt||? S d S | dkr‹|dkr‹|| }t |tƒs‰J ‚|S d S )Nr)   r+   r*   ú/r   ú//ú%ú&ú|ú^z<<z>>ú**)r   r,   ©r$   r!   r"   Úretr&   r&   r'   r-   r   sL   éìï	øûr-   úint | floatúfloat | Nonec                 C  s  t |tƒrt |tƒrJ | ||fƒ‚| dkr|| S | dkr!|| S | dkr)|| S | dkr7|dkr5|| S d S | dkrE|dkrC|| S d S | dkrS|dkrQ|| S d S | dkr€|dk r`t |tƒsd|dkr€z|| }W n
 tyt   Y d S w t |tƒs~J |ƒ‚|S d S )	Nr)   r+   r*   r2   r   r3   r4   r8   )r   r,   ÚOverflowErrorr.   r9   r&   r&   r'   r/   –   s:   "ðóöÿr/   r   c                 C  sR   | dkrt |ttfƒr| S | dkrt |tƒr| S | dkr't |ttfƒr'|S d S )Nr+   ú~r)   )r   r,   r.   )r$   r   r&   r&   r'   r%   ´   s   r%   N)r   r   r   r   r   r   )r$   r   r!   r(   r"   r(   r   r   )r$   r   r!   r,   r"   r,   r   r1   )r$   r   r!   r;   r"   r;   r   r<   )r$   r   r   r(   r   r1   )Ú__doc__Ú
__future__r   Útypingr   r   Ú
mypy.nodesr   r   r   r   r	   r
   r   r   r   r,   Úboolr.   r0   r   r(   r   Ú__annotations__r    r#   r-   r/   r%   r&   r&   r&   r'   Ú<module>   s    ,

4
#
$