File: //lib/python3/dist-packages/django/contrib/gis/gdal/raster/__pycache__/source.cpython-310.pyc
o
3�a-F � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z m
Z
m Z m Z d dl
mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZmZmZm Z d d l m Z m Z d d
l m!Z! d d l"m#Z#m$Z$ d d l%m&Z& G d
d� de'�Z(G dd� de�Z)dS )� N)� addressof�byref�c_buffer�c_char_p�c_double�c_int�c_void_p� string_at)�Driver)�
GDALException)�raster��BandList)�GDALRasterBase)�GDAL_RESAMPLE_ALGORITHMS�VSI_DELETE_BUFFER_ON_READ�VSI_FILESYSTEM_BASE_PATH�VSI_TAKE_BUFFER_OWNERSHIP)�SpatialReference� SRSException)�
json_regex)� force_bytes� force_str)�cached_propertyc s\ e Zd Zdddd�Z� fdd�Zedd� �Zejd d� �Zed
d � �Zejd d � �Z� Z S )
�TransformPoint)r � )� � )� � )�origin�scale�skewc sH |j | j| d }|j | j| d }t� �||g� || _|| _d S )Nr r )� geotransform�indices�super�__init__�_raster�_prop)�selfr �prop�x�y�� __class__� �G/usr/lib/python3/dist-packages/django/contrib/gis/gdal/raster/source.pyr& s
zTransformPoint.__init__c C � | d S �Nr r/ �r) r/ r/ r0 r+ & � zTransformPoint.xc C �( | j j}||| j| j d < || j _d S r2 �r' r# r$ r( �r) �value�gtfr/ r/ r0 r+ * � c C r1 �Nr r/ r3 r/ r/ r0 r, 0 r4 zTransformPoint.yc C r5 r; r6 r7 r/ r/ r0 r, 4 r: )
�__name__�
__module__� __qualname__r$ r& �propertyr+ �setterr, �
__classcell__r/ r/ r- r0 r s �
r c sH e Zd ZdZejZd7dd�Z� fdd�Zdd� Z d d
� Z
d d � Z e d
d� �Z
edd� �Ze dd� �Zedd� �Ze dd� �Ze dd� �Ze dd� �Zejdd� �Ze d d � �Zejd d � �Ze d d � �Zejd!d � �Ze d"d#� �Ze d$d%� �Ze d&d'� �Ze d(d)� �Ze d*d+� �Z d8d.d/�Z d9d1d2�Z , -d:d3d4�Z e d5d6� �Z � Z!S );�
GDALRasterz0
Wrap a raster GDAL Data Source object.
Fc C s~ |rdnd| _ t�� t|t�rt�|�rt�|�}t|t�rR|� t
�s0t j �
|�s0td| ��z
t�t|�| j �| _W d S tyQ } ztd�||���d }~ww t|t�r�d| _ t�|�}t|�| _t j �t
tt�� ��}t� t|�t | j�|t � z
t�t|�| j �| _W d S ty� t� t|�� td��w t|t ��r�d| _ t|�!dd��}|j"dkr�d|vr�td �|j"���d
|vs�d |vr�td ��d
|vr�td��g }|�!di ��#� D ]\}} d�|| �}
|�$|
�%� �&� � q�|�$d � t't(|� |� }t�)|jt|�!dd��|d
|d |�!dt(|�!dg ���|�!dd�t |��| _t*|�!dg ��D ]G\} } | j+| }
d| v �r]| d |
_,|
j,d u�r]d| v�sTd| v �sTd| v �r]|
j-|
j,fdd� |
j-| �!d�| �!d�| �!d�| �!d �d � �q,|�!d
�| _.d |v �r�|d \| j/_0| j/_1d |v �r�|d \| j2_0| j2_1d |v �r�|d \| j3_0| j3_1d S d S t|t4��r�|| _d S td!�t5|����)"Nr r z(Unable to read raster source input "%s".z+Could not open the datasource at "{}" ({}).z1Failed creating VSI raster from the input buffer.�driver�MEM�namez5Specify name for creation of raster with driver "{}".�width�heightz;Specify width and height attributes for JSON or dict input.�sridz$Specify srid for JSON or dict input.�
papsz_optionsz{}={}� � nr_of_bands�bands�datatype� � nodata_value�data�size�shape)r r )rP rR �offset)rP rQ rR rS r r! r" z%Invalid data source input type: "{}".)6�_writer
�ensure_registered�
isinstance�strr �match�json�loads�
startswithr �os�path�existsr �capi�open_dsr �_ptr�format�bytes�sys� getsizeofr � _ds_input�join�uuid�uuid4� create_vsi_file_from_mem_bufferr r �unlink_vsi_file�dict�getrE �items�append�upper�encoder �len� create_ds� enumeraterL rO rP �srsr r+ r, r! r" r �type)r) �ds_input�write�errrQ �vsi_pathrC rI �key�val�option�i�
band_input�bandr/ r/ r0 r& A s�
�
����
� �
�
�
�
zGDALRaster.__init__c s$ | j r t�t| j�� t� �� d S �N)� is_vsi_basedr_ rk r rE r% �__del__r3 r- r/ r0 r� � s zGDALRaster.__del__c C s | j S r� )rE r3 r/ r/ r0 �__str__� s zGDALRaster.__str__c C s dt t| j�� S )zJ
Short-hand representation because WKB may be very large.
z<Raster object at %s>)�hexr ra r3 r/ r/ r0 �__repr__� s zGDALRaster.__repr__c C s | j std��t�| j� dS )a
Flush all data from memory into the source file if it exists.
The data that needs flushing are geotransforms, coordinate systems,
nodata_values and pixel values. This function will be called
automatically wherever it is needed.
z9Raster needs to be opened in write mode to change values.N)rT r r_ �flush_dsra r3 r/ r/ r0 �_flush� s zGDALRaster._flushc C s4 | j sd S t� }t�t| j�t|�t�}t||j �S r� )
r� r r_ � get_mem_buffer_from_vsi_filer rE r r r r8 )r) �
out_length�datr/ r/ r0 �
vsi_buffer� s � zGDALRaster.vsi_bufferc C s | j o| j�t�S r� )ra rE r[ r r3 r/ r/ r0 r� � s zGDALRaster.is_vsi_basedc C s t t�| j��S )zi
Return the name of this raster. Corresponds to filename
for file-based rasters.
)r r_ �get_ds_descriptionra r3 r/ r/ r0 rE � s zGDALRaster.namec C s t �| j�}t|�S )z>
Return the GDAL Driver used for this raster.
)r_ �
get_ds_driverra r
)r) � ds_driverr/ r/ r0 rC � s zGDALRaster.driverc C � t �| j�S )z+
Width (X axis) in pixels.
)r_ � get_ds_xsizera r3 r/ r/ r0 rF � � zGDALRaster.widthc C r� )z,
Height (Y axis) in pixels.
)r_ � get_ds_ysizera r3 r/ r/ r0 rG r� zGDALRaster.heightc C s: zt �| j�}|s W dS t|dd�W S ty Y dS w )zF
Return the SpatialReference used in this GDALRaster.
N�wkt)�srs_type)r_ �get_ds_projection_refra r r )r) r� r/ r/ r0 ru s �zGDALRaster.srsc C sP t |t�r|}nt |ttf�rt|�}ntd��t�| j|j� � � | �
� dS )z�
Set the spatial reference used in this GDALRaster. The input can be
a SpatialReference or any parameter accepted by the SpatialReference
constructor.
z/Could not create a SpatialReference from input.N) rV r �intrW �
ValueErrorr_ �set_ds_projection_refra r� rq r� )r) r8 ru r/ r/ r0 ru s
c C s | j jS )zA
Shortcut to access the srid of this GDALRaster.
)ru rH r3 r/ r/ r0 rH ( s zGDALRaster.sridc C s
|| _ dS )zE
Shortcut to set this GDALRaster's srs from an srid.
N)ru )r) r8 r/ r/ r0 rH / �
c C s$ t d � }t�| jt|�� t|�S )z�
Return the geotransform of the data source.
Return the default geotransform if it does not exist or has not been
set previously. The default is [0.0, 1.0, 0.0, 0.0, 0.0, -1.0].
rN )r r_ �get_ds_geotransformra r �list)r) r9 r/ r/ r0 r# 6 s
zGDALRaster.geotransformc C sP t |�dkstdd� |D ��std��td |� }t�| jt|�� | �� dS )z)Set the geotransform for the data source.rN c s s � | ]
}t |ttf�V qd S r� )rV r� �float)�.0r+ r/ r/ r0 � <genexpr>E s � z*GDALRaster.geotransform.<locals>.<genexpr>z.Geotransform must consist of 6 numeric values.N) rr �allr� r r_ �set_ds_geotransformra r r� )r) �valuesr/ r/ r0 r# B s
c C �
t | d�S )z3
Coordinates of the raster origin.
r �r r3 r/ r/ r0 r L r� zGDALRaster.originc C r� )z@
Pixel scale in units of the raster projection.
r! r� r3 r/ r/ r0 r! S r� zGDALRaster.scalec C r� )z7
Skew of pixels (rotation parameters).
r" r� r3 r/ r/ r0 r" Z r� zGDALRaster.skewc C sp | j j| jj| j }| j j| jj| j }t|| j j�}t|| j j�}t|| j j�}t|| j j�}||||fS )zJ
Return the extent as a 4-tuple (xmin, ymin, xmax, ymax).
)r r+ r! rF r, rG �min�max)r) �xval�yval�xmin�xmax�ymin�ymaxr/ r/ r0 �extenta s zGDALRaster.extentc C s t | �S r� r
r3 r/ r/ r0 rL q r4 zGDALRaster.bands�NearestNeighbour� c C s |� d| j� |� d| j� |� d| jj� |� d| j� |� d| j� |� d| j� |� d| jj � d|vrB| j d | jj |d< d
|vrO| j
d � � |d
< d d
� | j
D �|d< t |dd�}t
| }t�| j| jj�� |j|jj�� |d|t� t� t� �
|�� |S )a
Return a warped GDALRaster with the given input characteristics.
The input is expected to be a dictionary containing the parameters
of the target raster. Allowed values are width, height, SRID, origin,
scale, skew, datatype, driver, and name (filename).
By default, the warp functions keeps all parameters equal to the values
of the original source raster. For the name of the target raster, the
name of the source raster will be used and appended with
_copy. + source_driver_name.
In addition, the resampling algorithm can be specified with the "resampling"
input parameter. The default is NearestNeighbor. For a list of all options
consult the GDAL_RESAMPLE_ALGORITHMS constant.
rF rG rH r r! r" rC rE �_copy.rM r c S s g | ]}d |j i�qS �rO r� )r� �bndr/ r/ r0 �
<listcomp>� s z#GDALRaster.warp.<locals>.<listcomp>rL T�rx r� )�
setdefaultrF rG ru rH r r! r" rC rE rL rM rB r r_ �reproject_imagera r� rq r r� )r) rw �
resampling� max_error�target� algorithmr/ r/ r0 �warpu s, �zGDALRaster.warpNc
C sr |r|}n| j jdkr| jd | j j }n tj�ttt�� ��}t t
� | j j t
|�| j t� t� t� t� �| jd�S )z"Return a clone of this GDALRaster.rD r� r� )rC rE r\ r] rg r rW rh ri rB r_ �copy_dsra r r r r rT )r) rE �
clone_namer/ r/ r0 �clone� s" � �zGDALRaster.clonec
C s� t | }t|t�r |}nt|ttf�rt|�}ntd��|j| jkr/|r*|| jjkr/| � |�S t
� | j | j
j�� |j�� ||t� �}t|�}|j|j|j|jj|jjg|jj|jjg|jj|jjgd�} |rj|| d< |rp|| d< | j| ||d�S )zk
Return a copy of this raster reprojected into the given spatial
reference system.
zETransform only accepts SpatialReference, string, and integer objects.)rH rF rG r r! r" rC rE )r� r� )r rV r r� rW � TypeErrorrH rC rE r� r_ �auto_create_warped_vrtra ru r� rq r rB rF rG r r+ r, r! r" r� )
r) ru rC rE r� r� r� �
target_srsr� rP r/ r/ r0 � transform� s6
�
��
zGDALRaster.transformc C s t jstd��t �| jd��� S )z�
Return information about this raster in a string format equivalent
to the output of the gdalinfo command line utility.
u5 GDAL ≥ 2.1 is required for using the info property.N)r_ � get_ds_infor� �ptr�decoder3 r/ r/ r0 �info� s zGDALRaster.info)F)r� r� r� )NNr� r� )"r< r= r> �__doc__r_ �close_ds�
destructorr& r� r� r� r� r? r� r r� rE rC rF rG ru r@ rH r# r r! r" r� rL r� r� r� r� rA r/ r/ r- r0 rB ; sb
~
7
�0rB )*rY r\ rd rh �ctypesr r r r r r r r � django.contrib.gis.gdal.driverr
� django.contrib.gis.gdal.errorr �"django.contrib.gis.gdal.prototypesr r_ �#django.contrib.gis.gdal.raster.bandr �#django.contrib.gis.gdal.raster.baser �$django.contrib.gis.gdal.raster.constr r r r �django.contrib.gis.gdal.srsr r �django.contrib.gis.geometryr �django.utils.encodingr r �django.utils.functionalr r� r rB r/ r/ r/ r0 �<module> s" ( #