_V@sddlmZmZddlmZddlmZddlmZ ddl m Z ddl m Z ddlmZddlmZd d lmZmZmZGd d d eZGd ddeZdS))byrefc_int)GDALBase) GDALException)raster)numpy)six) force_text)range)GDAL_INTEGER_TYPESGDAL_PIXEL_TYPESGDAL_TO_CTYPESc@seZdZdZddZeddZeddZedd Zed d Z ed d Z eddZ eddZ e j ddZ dddZddddddZdS)GDALBandzR Wraps a GDAL raster band, needs to be obtained from a GDALRaster object. cCs%||_tj|j||_dS)N)sourcecapiZget_ds_raster_band_ptr)selfrindexrE/tmp/pip-build-ghmbqnp_/Django/django/contrib/gis/gdal/raster/band.py__init__s zGDALBand.__init__cCsttj|jS)z= Returns the description string of the band. )r rZget_band_descriptionr)rrrr descriptionszGDALBand.descriptioncCstj|jS)z7 Width (X axis) in pixels of the band. )rZget_band_xsizer)rrrrwidthszGDALBand.widthcCstj|jS)z8 Height (Y axis) in pixels of the band. )rZget_band_ysizer)rrrrheight$szGDALBand.heightcCs|j|jS)zB Returns the total number of pixels in this band. )rr)rrrr pixel_count+szGDALBand.pixel_countcCstj|jttS)z@ Returns the minimum pixel value for this band. )rZget_band_minimumrrr)rrrrmin2sz GDALBand.mincCstj|jttS)z@ Returns the maximum pixel value for this band. )rZget_band_maximumrrr)rrrrmax9sz GDALBand.maxcCsRt}tj|j|}|s-d}n!|jtkrNt|}n|S)zR Returns the nodata value for this band, or None if it isn't set. N)rrZget_band_nodata_valuerdatatyper int)rZ nodata_existsvaluerrr nodata_value@s  zGDALBand.nodata_valuecCsHt|ttfs$tdntj|j||jjdS)z6 Sets the nodata value for this band. zNodata value must be numeric.N) isinstancerfloat ValueErrorrZset_band_nodata_valuerr_flush)rr rrrr!OsFcCs)tj|j}|r%t|}n|S)z@ Returns the GDAL Pixel Datatype for this band. )rZget_band_datatyperr )r as_stringdtyperrrrYs zGDALBand.datatypeNc Cs|sd }n|s:|j|d|j|df}ntdd|Drbtdn|d|jks|d|jkrtdnt|j|d|d}|dkrd}|}nTd}t|ttj fs t rt|t j r|j |}n ||}t j|j||d|d|d|dt||d|d|jdd |dkr|rt |St rt j|dt j|j|St|Sn |jjdS) af Reads 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. rr css|]}|dkVqdS)rNr).0xrrr qsz GDALBand.data..zOffset too big for this raster.zSize is larger than raster.Nr')rr)rranyr$rrr"bytesr memoryviewrZndarrayfrom_buffer_copyrZband_iorrZ frombufferr'Zreshapelistrr%)rdataoffsetsizeZ as_memoryviewZ ctypes_arrayZ access_flagZ data_arrayrrrr0bs6  %&   0    z GDALBand.data)__name__ __module__ __qualname____doc__rpropertyrrrrrrr!setterrr0rrrrrs    rc@s@eZdZddZddZddZddZd S) BandListcCs||_tj|dS)N)rr/r)rrrrrrs zBandList.__init__ccs9x2tdt|dD]}t|j|VqWdS)Nr )r lenrr)ridxrrr__iter__s zBandList.__iter__cCstj|jjS)N)rZget_ds_raster_countrr)rrrr__len__szBandList.__len__c CsAyt|j|dSWn"tk r<td|YnXdS)Nr zUnable to get band index %d)rrr)rrrrr __getitem__s zBandList.__getitem__N)r3r4r5rr<r=r>rrrrr9s    r9N)ctypesrrZdjango.contrib.gis.gdal.baserZdjango.contrib.gis.gdal.errorrZ"django.contrib.gis.gdal.prototypesrrZdjango.contrib.gis.shortcutsrZ django.utilsrZdjango.utils.encodingr Zdjango.utils.six.movesr constr r rrr/r9rrrrs