File: //lib/python3/dist-packages/PIL/__pycache__/ImageStat.cpython-38.pyc
U
g�
^= � @ 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 sl z |r|� |�| _n
|� � | _W n tk
r: || _Y nX t| jt�sPtd��ttt| j�d ��| _d S )Nz$first argument must be image or list� ) � histogram�h�AttributeError�
isinstance�list� TypeError�range�len�bands)�selfZ
image_or_list�mask� r �//usr/lib/python3/dist-packages/PIL/ImageStat.py�__init__ s z
Stat.__init__c C s8 |dd� dkrt |��t| d| �� }t| ||� |S )zCalculate missing attributeN� Z_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 �minmax6 s
z Stat._getextrema.<locals>.minmaxr r N�r
r r �append)r
r r r r r r � _getextrema3 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 � _getcountD s $zStat._getcountc C sR g }t dt| j�d�D ]6}d}t d�D ]}||| j|| 7 }q&|�|� q|S )z#Get sum of all pixels in each layerr r � r )r
r r ZlayerSum�jr r r �_getsumL 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 �_getsum2W 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 �_getmeanb s
z
Stat._getmeanc C sd g }| j D ]T}d}| j| d }|d }td�D ]"}|| j|| }||kr0 qTq0|�|� q
|S )z%Get median pixel level for each layerr r+ r )r r0 r
r r )r
r r �sZhalf�br) r r r �
_getmedianj s
zStat._getmedianc C s4 g }| j D ]$}|�t�| j| | j| �� q
|S )zGet RMS for each layer)r r �math�sqrtr- r0 r& r r r �_getrmsy s
"z Stat._getrmsc C sD g }| j D ]4}| j| }|�| j| | j| d | | � q
|S )zGet variance for each layerg @)r r0 r r- r/ )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 r5 r6 �varr& r r r �
_getstddev� s
zStat._getstddev)N)�__name__�
__module__� __qualname__r r r! r' r* r. r1 r4 r7 r8 r: r r r r r s
r )r" r5 r$ r ZGlobalr r r r �<module> s v