File: //home/arjun/projects/env/lib/python3.10/site-packages/PIL/__pycache__/ImageStat.cpython-310.pyc
o
weT � @ s. d dl Z d dlZd dlZG dd� d�ZeZdS )� Nc @ sf e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )�StatNc C sn z|r
|� |�| _n|� � | _W n ty || _Y nw t| jt�s)d}t|��ttt| j�d ��| _d S )Nz$first argument must be image or list� ) � histogram�h�AttributeError�
isinstance�list� TypeError�range�len�bands)�self�
image_or_list�mask�msg� r �F/home/arjun/projects/env/lib/python3.10/site-packages/PIL/ImageStat.py�__init__ s
�
� z
Stat.__init__c C s8 |dd� dkr t |��t| d| �� }t| ||� |S )zCalculate missing attributeN� �_get)r �getattr�setattr)r
�id�vr r r � __getattr__+ s
zStat.__getattr__c C s@ dd� }g }t dt| j�d�D ]}|�|| j|d� �� q|S )z-Get min/max values for each band in the imagec S s: d}d}t d�D ]}| | rt||�}t||�}q||fS )N� r r )r
�min�max)r �n�x�ir r r �minmax7 s
�z Stat._getextrema.<locals>.minmaxr r N�r
r r �append)r
r! r r r r r � _getextrema4 s
zStat._getextremac
C sB g }t dt| j�d�D ]}|�t�tj| j||d � �� q |S )z(Get total number of pixels in each layerr r )r
r r r# � functools�reduce�operator�add�r
r r r r r � _getcountE s $zStat._getcountc C sR g }t dt| j�d�D ]}d}t d�D ]
}||| j|| 7 }q|�|� q |S )z#Get sum of all pixels in each layerr r � r" )r
r r � layer_sum�jr r r �_getsumM s z Stat._getsumc C sZ g }t dt| j�d�D ] }d}t d�D ]}||d t| j|| � 7 }q|�|� q |S )z+Get squared sum of all pixels in each layerr r r+ � )r
r r �floatr# )r
r r �sum2r- r r r �_getsum2X s z
Stat._getsum2c C s. g }| j D ]}|�| j| | j| � q|S )z&Get average pixel level for each layer)r r# �sum�countr) r r r �_getmeanc s
z
Stat._getmeanc C sd g }| j D ]*}d}| j| d }|d }td�D ]}|| j|| }||kr) nq|�|� q|S )z%Get median pixel level for each layerr r/ r )r r4 r
r r# )r
r r �s�half�br- r r r �
_getmediank s
� zStat._getmedianc C s4 g }| j D ]}|�t�| j| | j| �� q|S )zGet RMS for each layer)r r# �math�sqrtr1 r4 r) r r r �_getrmsz s
"z Stat._getrmsc C sD g }| j D ]}| j| }|�| j| | j| d | | � q|S )zGet variance for each layerg @)r r4 r# r1 r3 )r
r r r r r r �_getvar� s
(z Stat._getvarc C s* g }| j D ]
}|�t�| j| �� q|S )z%Get standard deviation for each layer)r r# r: r; �varr) r r r �
_getstddev� s
zStat._getstddev)N)�__name__�
__module__� __qualname__r r r$ r* r. r2 r5 r9 r< r= r? r r r r r s
r )r% r: r' r �Globalr r r r �<module> s
w