U g7 @sddlmZmZmZmZddlmZddlmZ ddl m Z ddl m Z ddlmZddlmZmZmZmZGd d d e ZGd d d eZd S))byrefc_doublec_intc_void_p) GDALException)raster)GDALRasterBase)numpy) force_str)GDAL_COLOR_TYPESGDAL_INTEGER_TYPESGDAL_PIXEL_TYPESGDAL_TO_CTYPESc@seZdZdZddZddZeddZedd Zed d Z ed d Z dZ d#ddZ eddZ eddZeddZeddZeddZejddZd$ddZd%ddZd&d!d"Zd S)'GDALBandzQ Wrap a GDAL raster band, needs to be obtained from a GDALRaster object. cCs||_t|j||_dSN)sourcecapiZget_ds_raster_band_ptr)selfrindexrG/tmp/pip-unpacked-wheel-ye1bytdm/django/contrib/gis/gdal/raster/band.py__init__szGDALBand.__init__cCs|jd|_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_refreshrrrrrs zGDALBand._flushcCstt|jS)z< Return the description string of the band. )r rZget_band_descriptionrrrrr descriptionszGDALBand.descriptioncCs t|jS)z7 Width (X axis) in pixels of the band. )rZget_band_xsizerrrrrwidth%szGDALBand.widthcCs t|jS)z8 Height (Y axis) in pixels of the band. )rZget_band_ysizerrrrrheight,szGDALBand.heightcCs |j|jS)zA Return the total number of pixels in this band. )rrrrrr pixel_count3szGDALBand.pixel_countFc Csttttf\}}}}|jt|t|t|t|t|ttg}|sT|jr\tj}nd} |dt| tj }z |||j |j |j |j f} Wnt k rd} YnXd|_| 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) rrrrrrrZcompute_band_statisticsinsertZget_band_statisticsvaluer) rZrefreshZ approximateZsminZsmaxZsmeanZsstdZ stats_argsfuncforceresultrrr statistics<s,  zGDALBand.statisticscCs |dS)z? Return the minimum pixel value for this band. rr'rrrrminjsz GDALBand.mincCs |dS)z? Return the maximum pixel value for this band. r r(rrrrmaxqsz GDALBand.maxcCs |dS)zC Return the mean of all pixel values of this band. r!r(rrrrmeanxsz GDALBand.meancCs |dS)zQ Return the standard deviation of all pixel values of this band. r(rrrrstdsz GDALBand.stdcCs6t}t|j|}|sd}n|tkr2t|}|S)zQ Return the nodata value for this band, or None if it isn't set. N)rrZget_band_nodata_valuerdatatyper int)rZ nodata_existsr#rrr nodata_values zGDALBand.nodata_valuecCsV|dkr$tjstdt|jn&t|ttfs= 2.1 required to delete nodata values.z%Nodata value must be numeric or None.) rZdelete_band_nodata_value ValueErrorr isinstancer/floatZset_band_nodata_valuer)rr#rrrr0s cCst|j}|rt|}|S)z? Return the GDAL Pixel Datatype for this band. )rZget_band_datatyperr)r as_stringdtyperrrr.s zGDALBand.datatypecCst|j}|rt|}|S)z3Return the GDAL color interpretation for this band.)rZget_band_color_interprr )rr4colorrrr color_interps zGDALBand.color_interpNc Csl|pd}|p&|j|d|j|df}|p.|}tdd|DrJtd|d|jksf|d|jkrntdt||d|d}|dkrd}|}n6d}t|ttfst rt|t j r| |}n||}t |j||d|d|d|dt||d|d|dd |dkr`|r.t|St rVt j|t |d tt|St|Sn|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. )rrrr css|]}|dkVqdS)rNr).0xrrr sz GDALBand.data..zOffset too big for this raster.zSize is larger than raster.N)r5)rranyr1rr.r2bytes memoryviewr Zndarrayfrom_buffer_copyrZband_iorrZ frombufferr5Zreshapetuplereversedlistr) rdataoffsetsizeshapeZ as_memoryviewZ ctypes_arrayZ access_flagZ data_arrayrrrrBsJ      z GDALBand.data)FF)F)F)NNNNF)__name__ __module__ __qualname____doc__rrpropertyrrrr rr'r)r*r+r-r0setterr.r7rBrrrrrs8     .       rcs4eZdZfddZddZddZddZZS) BandListcs||_tdSr)rsuperr)rr __class__rrrszBandList.__init__ccs*tdt|dD]}t|j|VqdS)Nr )rangelenrr)ridxrrr__iter__szBandList.__iter__cCst|jjSr)rZget_ds_raster_countrrrrrr__len__szBandList.__len__cCs8zt|j|dWStk r2td|YnXdS)Nr zUnable to get band index %d)rrr)rrrrr __getitem__szBandList.__getitem__)rFrGrHrrSrTrU __classcell__rrrNrrLs rLN)ctypesrrrrZdjango.contrib.gis.gdal.errorrZ"django.contrib.gis.gdal.prototypesrrZ#django.contrib.gis.gdal.raster.baserZdjango.contrib.gis.shortcutsr Zdjango.utils.encodingr constr r rrrrArLrrrrs     _