a
    d                     @   s   d dl mZmZ d dlmZmZmZ d dlZg dZe	dZ
edddZG d	d dZG d
d deZG dd deeedZG dd deeedZG dd deeedZG dd deeedZdS )    )EnumEnumMeta)TypeVarUnioncastN)piResample
BorderTypeSamplePaddingg-DT!	@TConstantBase)boundc                   @   s*   e Zd Zeeeeef edddZdS )r   )valuereturnc                 C   sF   t |tu r| |  S t |tu r,| |S t || u r<|S t d S N)typestrupperint	TypeError)clsr    r   Y/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/constants.pyget   s    zConstantBase.getN)	__name__
__module____qualname__classmethodr   r   r   r   r   r   r   r   r   r      s   c                   @   s.   e Zd Zeeeef edddZdd Z	dS )EnumMetaFlags)otherr   c                    sd   t  tu r,tt  t fdd| D S t  tu rNt fdd| D S t fdd| D S )Nc                 3   s   | ]}|j   kV  qd S r   )namer   .0valr   r   r   	<genexpr>       z-EnumMetaFlags.__contains__.<locals>.<genexpr>c                 3   s   | ]}|j  kV  qd S r   )r   r!   r$   r   r   r%      r&   c                 3   s   | ]}| kV  qd S r   r   r!   r$   r   r   r%      r&   )r   r   r   anyr   )selfr   r   r$   r   __contains__   s    
zEnumMetaFlags.__contains__c                    s   d  fdd D S )Nz | c                 3   s    | ]} j  d |j V  qdS ).N)r   r    r!   r(   r   r   r%   "   r&   z)EnumMetaFlags.__repr__.<locals>.<genexpr>)joinr+   r   r+   r   __repr__!   s    zEnumMetaFlags.__repr__N)
r   r   r   r   r   r   r   boolr)   r-   r   r   r   r   r      s   r   c                   @   s   e Zd ZdZdZdZdS )r   r         N)r   r   r   ZNEARESTZBILINEARZBICUBICr   r   r   r   r   %   s   r   )	metaclassc                   @   s   e Zd ZdZdZdZdZdS )r	   r   r/   r0      N)r   r   r   ZCONSTANTZREFLECTZ	REPLICATEZCIRCULARr   r   r   r   r	   +   s   r	   c                   @   s   e Zd ZdZdZdZdS )r
   r   r/   r0   N)r   r   r   ZZEROSZBORDERZ
REFLECTIONr   r   r   r   r
   2   s   r
   c                   @   s(   e Zd ZdZdZdZdZdZdZdZ	dS )	DataKeyr   r/   r0   r2            N)
r   r   r   ZINPUTMASKZBBOXZ	BBOX_XYXYZ	BBOX_XYWHZ	KEYPOINTSZCLASSr   r   r   r   r3   9   s   r3   )enumr   r   typingr   r   r   Ztorch__all__Ztensorr   r   r   r   r   r	   r
   r3   r   r   r   r   <module>   s   
