File: //usr/lib/python3/dist-packages/django/contrib/gis/gdal/raster/__pycache__/band.cpython-310.pyc
o
3�a7 � @ s� d dl mZmZmZmZ d dlmZ d dlmZ d dl
m Z d dl m
Z
d dlmZ ddlmZmZmZmZ G d d
� d
e �ZG d d � d e�Zd
S )� )�byref�c_double�c_int�c_void_p)�
GDALException)�raster)�GDALRasterBase)�numpy)� force_str� )�GDAL_COLOR_TYPES�GDAL_INTEGER_TYPES�GDAL_PIXEL_TYPES�GDAL_TO_CTYPESc @ s� e Zd ZdZdd� Zdd� Zedd� �Zedd � �Zed
d � �Z ed d
� �Z
dZ d#dd�Z edd� �Z
edd� �Zedd� �Zedd� �Zedd� �Zejdd� �Zd$d d �Zd$d d �Zd%d!d"�Zd S )&�GDALBandzQ
Wrap a GDAL raster band, needs to be obtained from a GDALRaster object.
c C s || _ t�|j|�| _d S �N)�source�capi�get_ds_raster_band�_ptr)�selfr �index� r �E/usr/lib/python3/dist-packages/django/contrib/gis/gdal/raster/band.py�__init__ s zGDALBand.__init__c C s | j �� d| _dS )z�
Call the flush method on the Band's parent raster and force a refresh
of the statistics attribute when requested the next time.
TN)r �_flush�_stats_refresh�r r r r r s
zGDALBand._flushc C s t t�| j��S )z<
Return the description string of the band.
)r
r �get_band_descriptionr r r r r � description s zGDALBand.descriptionc C � t �| j�S )z7
Width (X axis) in pixels of the band.
)r �get_band_xsizer r r r r �width% � zGDALBand.widthc C r )z8
Height (Y axis) in pixels of the band.
)r �get_band_ysizer r r r r �height, r# zGDALBand.heightc C s | j | j S )zA
Return the total number of pixels in this band.
)r" r% r r r r � pixel_count3 r# zGDALBand.pixel_countFc C s� t � t � t � t � f\}}}}| jt|�t|�t|�t|�t|�t� t� g}|s*| jr.tj}n
d} |�dt| �� tj }z||� |j
|j
|j
|j
f}
W n t yV d}
Y nw d| _|
S )a�
Compute statistics on the pixel values of this band.
The return value is a tuple with the following structure:
(minimum, maximum, mean, standard deviation).
If approximate=True, the statistics may be computed based on overviews
or a subset of image tiles.
If refresh=True, the statistics will be computed from the data directly,
and the cache will be updated where applicable.
For empty bands (where all pixel values are nodata), all statistics
values are returned as None.
For raster formats using Persistent Auxiliary Metadata (PAM) services,
the statistics might be cached in an auxiliary file.
T� )NNNNF) r r r r r r r �compute_band_statistics�insert�get_band_statistics�valuer ) r �refresh� approximate�smin�smax�smean�sstd�
stats_args�func�force�resultr r r �
statistics< s" �
�zGDALBand.statisticsc C � | � � d S )z?
Return the minimum pixel value for this band.
r �r6 r r r r �minj r# z GDALBand.minc C r7 )z?
Return the maximum pixel value for this band.
r r8 r r r r �maxq r# z GDALBand.maxc C r7 )zC
Return the mean of all pixel values of this band.
r' r8 r r r r �meanx r# z
GDALBand.meanc C r7 )zQ
Return the standard deviation of all pixel values of this band.
� r8 r r r r �std r# z GDALBand.stdc C s8 t � }t�| j|�}|sd}|S | �� tv rt|�}|S )zQ
Return the nodata value for this band, or None if it isn't set.
N)r r �get_band_nodata_valuer �datatyper
�int)r �
nodata_existsr+ r r r � nodata_value� s �zGDALBand.nodata_valuec C sT |du rt js td��t �| j� nt|ttf�s td��t �| j|� | �� dS )z5
Set the nodata value for this band.
Nz-GDAL >= 2.1 required to delete nodata values.z%Nodata value must be numeric or None.) r �delete_band_nodata_value�
ValueErrorr �
isinstancer@ �float�set_band_nodata_valuer )r r+ r r r rB � s c C � t �| j�}|r t| }|S )z?
Return the GDAL Pixel Datatype for this band.
)r �get_band_datatyper r )r � as_string�dtyper r r r? � s zGDALBand.datatypec C rH )z3Return the GDAL color interpretation for this band.)r �get_band_color_interpr r )r rJ �colorr r r � color_interp� s zGDALBand.color_interpNc C sd |pd}|p| j |d | j|d f}|p|}tdd� |D ��r%td��|d | j ks3|d | jkr7td��t| �� |d |d }|du rOd}|� }nd}t|ttf�s`t rft|t j
�rf|� |�}n||� }t �
| j||d |d |d |d t|�|d |d | �� dd� |du r�|r�t|�S t r�t j|t �|�d ��tt|���S t|�S | �� dS )
ae
Read or writes pixel values for this band. Blocks of data can
be accessed by specifying the width, height and offset of the
desired block. The same specification can be used to update
parts of a raster by providing an array of values.
Allowed input data types are bytes, memoryview, list, tuple, and array.
)r r r r c s s � | ]}|d kV qdS )r Nr )�.0�xr r r � <genexpr>� s � z GDALBand.data.<locals>.<genexpr>z Offset too big for this raster.zSize is larger than raster.N)rK )r" r% �anyrD r r? rE �bytes�
memoryviewr �ndarray�from_buffer_copyr �band_ior r �
frombufferrK �reshape�tuple�reversed�listr ) r �data�offset�size�shape�
as_memoryview� ctypes_array� access_flag�
data_arrayr r r r] � s< �
�� z
GDALBand.data)FF)F)NNNNF)�__name__�
__module__� __qualname__�__doc__r r �propertyr r" r% r& r r6 r9 r: r; r= rB �setterr? rN r] r r r r r s:
.
r c s4 e Zd Z� fdd�Zdd� Zdd� Zdd� Z� ZS ) �BandListc s || _ t� �� d S r )r �superr )r r �� __class__r r r � s zBandList.__init__c c s, � t dt| �d �D ] }t| j|�V q
d S )Nr )�range�lenr r )r �idxr r r �__iter__� s ��zBandList.__iter__c C s t �| jj�S r )r �get_ds_raster_countr r r r r r �__len__� s zBandList.__len__c C s. z t | j|d �W S ty td| ��w )Nr zUnable to get band index %d)r r r )r r r r r � __getitem__� s
�zBandList.__getitem__)re rf rg r rr rt ru �
__classcell__r r rm r rk � s
rk N)�ctypesr r r r � django.contrib.gis.gdal.errorr �"django.contrib.gis.gdal.prototypesr r �#django.contrib.gis.gdal.raster.baser � django.contrib.gis.shortcutsr �django.utils.encodingr
�constr r
r r r r\ rk r r r r �<module> s _