o g‰¼cþã@sRddlZddlZddlmZddlmZmZmZer ddl m Z Gdd„dƒZ dS) éNé)ÚNpImageé)ÚHAS_GDALÚHAS_PILÚ HAS_IMGIO)Úgdalc@sBeZdZdZdd„Zdddddddœfd d „Zd d „Zd d„ZdS)Ú BigTiffWriterao This class is designed to write a bigtif with jpeg compression writing a large tiff file without trigger a memory overflow is possible with the help of GDAL library jpeg compression allows to maintain a reasonable file size transparency or nodata are stored in an internal tiff mask because it's not possible to have an alpha channel when using jpg compression cCs d|_dS)N)Úds©Úself©r ú~C:\Users\olivi\AppData\Roaming\Blender Foundation\Blender\3.4\scripts\addons\BlenderGIS-master\core\georaster\bigtiffwriter.pyÚ__del__+s zBigTiffWriter.__del__ÚYESÚJPEGéPÚYCBCR)ÚTFWÚTILEDÚBIGTIFFÚCOMPRESSÚ JPEG_QUALITYÚ PHOTOMETRICc Cs0tstdƒ‚||_||_||f|_||_||_| dd¡dkr+d|_t   dd¡d}nd |_d }||_ d d „|  ¡Dƒ}t   d ¡}t j} d|_| ||||| |¡|_|jrl|j t j¡|j d¡ ¡|_|j d¡n |d kry|j d ¡ d¡|j |j ¡¡|jjdur–|j |jj ¡ ¡¡dSdS)z× path = fule system path for the ouput tiff w, h = width and height in pixels georef : a Georef object used to set georeferencing informations, optional geoTiffOptions : GDAL create option for tiff format zGDAL interface unavailablerNrTÚGDAL_TIFF_INTERNAL_MASKréFécSs$g|]\}}t|ƒdt|ƒ‘qS)ú=)Ústr)Ú.0ÚkÚvr r rÚ Os$z*BigTiffWriter.__init__..ÚGTiffÚuint8réÿ)rÚ ImportErrorÚwÚhÚsizeÚpathÚgeorefÚgetÚuseMaskrÚSetConfigOptionÚnbBandsÚitemsÚGetDriverByNameZGDT_ByteÚdtypeÚCreater ÚCreateMaskBandÚGMF_PER_DATASETÚ GetRasterBandÚ GetMaskBandÚmaskÚFillÚSetGeoTransformÚtoGDALÚcrsÚ SetProjectionÚgetOgrSpatialRefÚ ExportToWkt) r r*r'r(r+ÚgeoTiffOptionsÚnÚoptionsÚdriverZgdtyper r rÚ__init__0s:    ÿzBigTiffWriter.__init__c Cs°t|ƒ}|j}tdƒD]}|dd…dd…|f}|j |d¡ |||¡q |jddk}|rT|dd…dd…df}|jrG|j |||¡dS|j d¡ |||¡dS dS)zdata = numpy array or NpImgrNrrr) rÚdataÚranger r6Ú WriteArrayÚshaper-r8) r rEÚxÚyÚimgÚbandIdxÚ bandArrayÚhasAlphaÚalphar r rÚpasteds zBigTiffWriter.pastec Cs4d dd |j¡d |j¡d |j¡d |j¡g¡S)NÚ z* Data infos :z size {}z type {}z number of bands {}z* Georef & Geometry : {})ÚjoinÚformatr)r2r/r+r r r rÚ__repr__‚s    ûzBigTiffWriter.__repr__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrDrPrTr r r rr "s 4 r ) ÚosÚnumpyÚnpÚnpimgrÚ checkdepsrrrZosgeorr r r r rÚs