File: //lib/python3/dist-packages/django/contrib/gis/gdal/__pycache__/layer.cpython-310.pyc
o
�aP! � @ s� d dl mZmZ d dlmZ d dlmZmZ d dlm Z m
Z
d dl m Z d dl
mZ d dlmZ d dlmZ d d lmZmZmZ d d
lmZ d d l m Z m Z G d d
� d
e�Z dS )� )�byref�c_double)�GDALBase)�Envelope� OGREnvelope)�
GDALException� SRSException)�Feature)�
OGRFieldTypes)� OGRGeometry)� OGRGeomType)�ds�geom�srs)�SpatialReference)� force_bytes� force_strc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
dd� �Z e
dd� �Z e
d/dd��Z
e
dd� �Ze
dd� �Ze
dd� �Ze
dd � �Ze
d d � �Ze
d d � �Ze
d!d"� �Zd#d$� Zd%d&� Ze
ee�Zd'd(� Zd0d*d+�Zd,d-� Zd.S )1�LayerzSA class that wraps an OGR Layer, needs to be instantiated from a DataSource object.c C s6 |st d��|| _|| _t�| j�| _| �d�| _dS )a7
Initialize on an OGR C pointer to the Layer and the `DataSource` object
that owns this layer. The `DataSource` object is required so that a
reference to it is kept with this Layer. This prevents garbage
collection of the `DataSource` while this Layer is still active.
z*Cannot create Layer, invalid pointer givens
RandomReadN) r �ptr�_ds�capi�get_layer_defn�_ptr�_ldefn�test_capability� _random_read)�self� layer_ptrr
� r �?/usr/lib/python3/dist-packages/django/contrib/gis/gdal/layer.py�__init__ s zLayer.__init__c sb t |t�r|dk r
td��� �|�S t |t�r-|�� j�\}}}� fdd�t|||�D �S td��)z'Get the Feature at the specified index.r z/Negative indices are not allowed on OGR Layers.c s g | ]}� � |��qS r )�
_make_feature)�.0�fid�r r r �
<listcomp>3 � z%Layer.__getitem__.<locals>.<listcomp>z>Integers and slices may only be used when indexing OGR Layers.) �
isinstance�int�
IndexErrorr! �slice�indices�num_feat�range� TypeError)r �index�start�stop�strider r$ r � __getitem__' s
zLayer.__getitem__c c s6 � t �| j� t| j�D ] }tt �| j�| �V q dS )z'Iterate over each Feature in the Layer.N)r �
reset_readingr r- r, r �get_next_feature)r �ir r r �__iter__7 s
� �zLayer.__iter__c C � | j S )z%The length is the number of features.)r, r$ r r r �__len__> � z
Layer.__len__c C r8 )z The string name of the layer.)�namer$ r r r �__str__B r: z
Layer.__str__c C sX | j rz tt�| j|�| �W S ty Y nw | D ] }|j|kr%| S qtd| ��)a.
Helper routine for __getitem__ that constructs a Feature from the given
Feature ID. If the OGR Layer does not support random-access reading,
then each feature of the layer will be incremented through until the
a Feature is found matching the given feature ID.
zInvalid feature id: %s.)r r r � get_featurer r r# r) )r �feat_id�featr r r r! F s �
� zLayer._make_featurec C s" t � }t�| jt|�d� t|�S )z.Return the extent (an Envelope) of this layer.� )r r �
get_extentr r r )r �envr r r �extent] s z Layer.extentc C s t �| j�}t|| jjdd�S )z1Return the name of this layer in the Data Source.T�� strings_only)r � get_fd_namer r r �encoding)r r; r r r r; d s z
Layer.namer@ c C s t �| j|�S )z+Return the number of features in the Layer.)r �get_feature_countr )r �forcer r r r, j s zLayer.num_featc C s t �| j�S )z)Return the number of fields in the Layer.)r �get_field_countr r$ r r r �
num_fieldso s zLayer.num_fieldsc C s t t�| j��S )z4Return the geometry type (OGRGeomType) of the Layer.)r r �get_fd_geom_typer r$ r r r � geom_typet s zLayer.geom_typec C s2 zt �| j�}tt�|��W S ty Y dS w )z0Return the Spatial Reference used in this Layer.N)r �
get_layer_srsr r �srs_api� clone_srsr )r r r r r r y s �z Layer.srsc � � fdd�t � j�D �S )zt
Return a list of string names corresponding to each of the Fields
available in this Layer.
c s. g | ]}t t�t�� j|��� jjd d��qS )TrD )r r �get_field_name�get_field_defnr r rG �r" r6 r$ r r r% � s �
�z Layer.fields.<locals>.<listcomp>�r- rK r$ r r$ r �fields� s
�z Layer.fieldsc rQ )z�
Return a list of the types of fields in this Layer. For example,
return the list [OFTInteger, OFTReal, OFTString] for an OGR layer that
has an integer, a floating-point, and string fields.
c s$ g | ]}t t�t�� j|�� �qS r )r
r �get_field_typerS r rT r$ r r r% � s �z%Layer.field_types.<locals>.<listcomp>rU r$ r r$ r � field_types� s
�zLayer.field_typesc rQ )z;Return a list of the maximum field widths for the features.c � g | ] }t �t �� j|���qS r )r �get_field_widthrS r rT r$ r r r% � � �z&Layer.field_widths.<locals>.<listcomp>rU r$ r r$ r � field_widths� �
�zLayer.field_widthsc rQ )z-Return the field precisions for the features.c rY r )r �get_field_precisionrS r rT r$ r r r% � r[ z*Layer.field_precisions.<locals>.<listcomp>rU r$ r r$ r �field_precisions� r] zLayer.field_precisionsc C s. z t t�t�| j���W S ty Y d S w )N)r �geom_api�
clone_geomr �get_spatial_filterr r r$ r r r �_get_spatial_filter� s
�zLayer._get_spatial_filterc C s� t |t�rt�| j|j� d S t |ttf�r5t|�dks td��t t
|�\}}}}t� | j||||� d S |d u rBt�| jd � d S t d��)N� z/Spatial filter list/tuple must have 4 elements.zJSpatial filter must be either an OGRGeometry instance, a 4-tuple, or None.)
r' r r �set_spatial_filterr �tuple�list�len�
ValueError�mapr �set_spatial_filter_rectr. )r �filter�xmin�ymin�xmax�ymaxr r r �_set_spatial_filter� s
zLayer._set_spatial_filterc s( � | j vr td� ��� fdd�| D �S )zg
Return a list containing the given field name for every Feature
in the Layer.
zinvalid field name: %sc s g | ]}|� � ��qS r )�get�r" r? ��
field_namer r r% � r& z$Layer.get_fields.<locals>.<listcomp>)rV r )r ru r rt r �
get_fields� s
zLayer.get_fieldsFc s0 |rddl m� � fdd�| D �S dd� | D �S )zb
Return a list containing the OGRGeometry for every Feature in
the Layer.
r �� GEOSGeometryc s g | ]}� |j j��qS r )r �wkbrs rw r r r% � s z#Layer.get_geoms.<locals>.<listcomp>c S s g | ]}|j �qS r )r rs r r r r% � s )�django.contrib.gis.geosrx )r �geosr rw r � get_geoms� s zLayer.get_geomsc C s t t�| jt|���S )ah
Return a bool indicating whether the this Layer supports the given
capability (a string). Valid capability strings include:
'RandomRead', 'SequentialWrite', 'RandomWrite', 'FastSpatialFilter',
'FastFeatureCount', 'FastGetExtent', 'CreateField', 'Transactions',
'DeleteFeature', and 'FastSetNextByIndex'.
)�boolr r r r )r �
capabilityr r r r � s zLayer.test_capabilityN)r@ )F)�__name__�
__module__� __qualname__�__doc__r r3 r7 r9 r< r! �propertyrC r; r, rK rM r rV rX r\ r_ rc rq �spatial_filterrv r| r r r r r r sD
r N) �ctypesr r � django.contrib.gis.gdal.baser � django.contrib.gis.gdal.enveloper r � django.contrib.gis.gdal.errorr r � django.contrib.gis.gdal.featurer � django.contrib.gis.gdal.fieldr
�"django.contrib.gis.gdal.geometriesr � django.contrib.gis.gdal.geomtyper �"django.contrib.gis.gdal.prototypesr
r r r` r rO �django.contrib.gis.gdal.srsr �django.utils.encodingr r r r r r r �<module> s