o
�J�hm � @ sB d dl mZ d dlZd dlmZ ddlmZ G dd� d�ZeZdS )� )� annotationsN)�cached_property� )�Imagec @ s� e Zd Z d d dd �Zed!d d ��Zed"dd��Zed#dd��Zed#dd��Zed#dd��Z ed"dd��Z
ed#dd��Z ed#dd ��Z ed#d d ��Z
dS )$�StatN�
image_or_list�Image.Image | list[int]�mask�Image.Image | None�return�Nonec C sT t |tj�r
|�|�| _nt |t�r|| _nd}t|��ttt| j�d ��| _dS )a
Calculate statistics for the given image. If a mask is included,
only the regions covered by that mask are included in the
statistics. You can also pass in a previously calculated histogram.
:param image: A PIL image, or a precalculated histogram.
.. note::
For a PIL image, calculations rely on the
:py:meth:`~PIL.Image.Image.histogram` method. The pixel counts are
grouped into 256 bins, even if the image has more than 8 bits per
channel. So ``I`` and ``F`` mode images have a maximum ``mean``,
``median`` and ``rms`` of 255, and cannot have an ``extrema`` maximum
of more than 255.
:param mask: An optional mask.
z$first argument must be image or list� N) �
isinstancer � histogram�h�list� TypeError�range�len�bands)�selfr r �msg� r �KC:\pinokio\api\whisper-webui.git\app\env\lib\site-packages\PIL\ImageStat.py�__init__ s
z
Stat.__init__�list[tuple[int, int]]c s, d dd�� � �fdd�t d t�j�d
�D �S )
au
Min/max values for each band in the image.
.. note::
This relies on the :py:meth:`~PIL.Image.Image.histogram` method, and
simply returns the low and high bins used. This is correct for
images with 8 bits per channel, but fails for other modes such as
``I`` or ``F``. Instead, use :py:meth:`~PIL.Image.Image.getextrema` to
return per-band extrema for the image. This is more correct and
efficient because, for non-8-bit modes, the histogram method uses
:py:meth:`~PIL.Image.Image.getextrema` to determine the bins used.
r � list[int]r �tuple[int, int]c S sV d\}}t d�D ]
}| | r|} nqt ddd�D ]
}| | r&|} ||fS q||fS )N)�� r r
r �����)r )r Zres_minZres_max�ir r r �minmaxM s ��z Stat.extrema.<locals>.minmaxc s g | ] }� �j |d � ��qS �N)r ��.0r �r! r r r �
<listcomp>Y s z Stat.extrema.<locals>.<listcomp>r r
N)r r r r �r r r �r r r% r �extrema> s
" z Stat.extremar c s � fdd�t dt� j�d�D �S )z2Total number of pixels for each band in the image.c s" g | ]
}t � j||d � ��qS )r
)�sumr r# r( r r r& ^ s " z Stat.count.<locals>.<listcomp>r r
r' r( r r( r �count[ s z
Stat.count� list[float]c C sR g }t dt| j�d�D ]}d}t d�D ]
}||| j|| 7 }q|�|� q |S )z-Sum of all pixels for each band in the image.r r
� )r r r �append)r �vr Z layer_sum�jr r r r* ` s zStat.sumc C sZ g }t dt| j�d�D ] }d}t d�D ]}||d t| j|| � 7 }q|�|� q |S )z5Squared sum of all pixels for each band in the image.r r
r- � )r r r �floatr. )r r/ r �sum2r0 r r r r3 l s z Stat.sum2c � � fdd�� j D �S )zAAverage (arithmetic mean) pixel level for each band in the image.c s g | ] }� j | � j| �qS r )r* r+ r# r( r r r& { s z Stat.mean.<locals>.<listcomp>�r r( r r( r �meanx � z Stat.meanc C sd g }| j D ]*}d}| j| d }|d }td�D ]}|| j|| }||kr) nq|�|� q|S )z.Median pixel level for each band in the image.r r1 r
)r r+ r r r. )r r/ r �s�half�br0 r r r �median} s
� z Stat.medianc r4 )z2RMS (root-mean-square) for each band in the image.c s& g | ]}t �� j| � j| ��qS r )�math�sqrtr3 r+ r# r( r r r& � s & z Stat.rms.<locals>.<listcomp>r5 r( r r( r �rms� r7 zStat.rmsc r4 )z$Variance for each band in the image.c s8 g | ]}� j | � j| d � j| � j| �qS )g @)r3 r* r+ r# r( r r r& � s *��z Stat.var.<locals>.<listcomp>r5 r( r r( r �var� s
�zStat.varc r4 )z.Standard deviation for each band in the image.c s g | ]
}t �� j| ��qS r )r< r= r? r# r( r r r& � s z Stat.stddev.<locals>.<listcomp>r5 r( r r( r �stddev� r7 z Stat.stddevr" )r r r r
r r )r r )r r )r r, )�__name__�
__module__� __qualname__r r r) r+ r* r3 r6 r; r> r? r@ r r r r r s* � r ) �
__future__r r< � functoolsr � r r �Globalr r r r �<module> s