o 6��a8� @s:UdZddlmZddlZddlZz2ddlZddlm Z ddl mZmZddl mZmZmZddlmZejege fZdZded <WneyWZzeZWYdZ[ndZ[wwgZd ed<d(dd�Zd)dd�Ze�Zded<e�Zded<d*dd�Zejs�edur�d+dd �Z Gd!d"�d"e�Z!d,d%d&�Z"dSd,d'd&�Z"dS)-z6A module containing `numpy`-specific plugins for mypy.�)�annotationsN)�Type)�Plugin�AnalyzeTypeContext)�MypyFile� ImportFrom� Statement)�PRI_MEDzt.Optional[ModuleNotFoundError]�MYPY_EX�t.List[str]�__all__�return�t.Dict[str, str]cCs�dtjfdtjfdtjfdtjfdtjfdtjfdtjfdtjfd tj fd tj fg }i}|D]\}}d|�jj}d|�d �|d|��<q.|S)N� _NBitByte� _NBitShort� _NBitIntC� _NBitIntP�_NBitInt� _NBitLongLong� _NBitHalf�_NBitSingle�_NBitDouble�_NBitLongDouble�znumpy._�Bit�numpy.typing._nbit.) �np�byte�short�intc�intp�int_�longlong�half�single�double� longdouble�dtype�itemsize)�names�ret�name�typ�n�r.�:/usr/lib/python3/dist-packages/numpy/typing/mypy_plugin.py�_get_precision_dicts � r0cs.tjtjtjtjg}hd��fdd�|D�S)N>�int128�int256�float80�float96�uint128�uint256�float128�float256� complex160� complex192� complex256� complex512csg|] }|j�vr|j�qSr.)�__name__)�.0�i��extended_namesr.r/� =sz0_get_extended_precision_list..)r� ulonglongr"r&�clongdouble)�extended_typesr.r@r/�_get_extended_precision_list-srFzt.Final�_PRECISION_DICT�_EXTENDED_PRECISION_LIST�ctxrrcCs2|\}}}|j�d�d}td|��}|�|�S)z;Replace a type-alias with a concrete ``NBitBase`` subclass.�.��r)r+�splitrG� named_type)rIr,�_�apir+�name_newr.r.r/�_hookHs rQ�iterable�t.Iterable[Statement]�id�str�intcCs8t|�D]\}}t|dd�|kr|Sqtd|��)z>Identify the first ``ImportFrom`` instance the specified `id`.rTNzBFailed to identify a `ImportFrom` instance with the following id: )� enumerate�getattr� ValueError)rRrTr?�valuer.r.r/�_indexQs��r[c@s$eZdZdZddd�Zddd�Zd S)�_NumpyPluginzCA plugin for assigning platform-specific `numpy.number` precisions.�fullnamerUr �t.Optional[_HookFunc]cCs|tvrtSdS)z�Set the precision of platform-specific `numpy.number` subclasses. For example: `numpy.int_`, `numpy.longlong` and `numpy.longdouble`. N)rGrQ)�selfr]r.r.r/�get_type_analyze_hook]sz"_NumpyPlugin.get_type_analyze_hook�filer�t.List[t.Tuple[int, str, int]]cCs^t|jdfg}|jdkr-tdddd�tD�d�}d|_|j|jfD]}t|d�}|||<q!|S) z�Import platform-specific extended-precision `numpy.number` subclasses. For example: `numpy.float96`, `numpy.float128` and `numpy.complex256`. rK�numpyz numpy.typing._extended_precisionrcSsg|]}||f�qSr.r.)r>�vr.r.r/rBqsz4_NumpyPlugin.get_additional_deps..)r)T)r r]rrH�is_top_level�defs�importsr[)r_rar*rg�lstr?r.r.r/�get_additional_depsfs � z _NumpyPlugin.get_additional_depsN)r]rUr r^)rarr rb)r=� __module__�__qualname__�__doc__r`rir.r.r.r/r\Zs r\�version�t.Type[_NumpyPlugin]cCstS�zAn entry-point for mypy.)r\�rmr.r.r/�plugin|�rqcCst�ro)r rpr.r.r/rq�rr)r r)r r)rIrr r)rRrSrTrUr rV)rmrUr rn)#rl� __future__r�typing�trcr� mypy.types�mypyr�mypy.pluginrr� mypy.nodesrrr� mypy.buildr �Callable� _HookFuncr �__annotations__�ModuleNotFoundError�exrr0rFrGrHrQ� TYPE_CHECKINGr[r\rqr.r.r.r/�s6�� "