dfYddlmZesdevrddlmZnddlZ ddlZdZ dZ dZ d Z Gd d eZdd lmZGd deZej&edZdZdZdZdZdZGddeZdZdZd:dZej<ZdZdZ dZ!d;dZ"dd-ZPd.ZQddddddddejddd&df d/ZRdddd&ejddfd0ZSddddddddejddf d1ZTddejddfd2ZUd3ZVd4ZW d?d5ZX d:d6ZYd@d7ZZd;d8Z[dAd9Z\y#e $rddlZYwxYw)B) version_info.) _gdal_arrayNc d|jjz}d|jj d|jj d|dS#tj$rd}YKwxYw)Nz proxy of )this__repr__ __builtin__ Exception __class__ __module____name__)selfstrthiss 2/usr/lib/python3/dist-packages/osgeo/gdal_array.py _swig_reprrs_ 2 2 44"^^668O8OQX ZZ  sAA+*A+cfd}|S)Nc|dk(r |||y|dk(r|jj|yt||r.tt t ||t r |||ytd|z)Nr thisownz(You cannot add instance attributes to %s)r ownhasattr isinstancegetattrtypepropertyAttributeError)rnamevaluesets rset_instance_attrzE_swig_setattr_nondynamic_instance_variable..set_instance_attrsi 6> dE " Y  IIMM% T4 ZT D0I8%T dE " !Kd!RS S)r!r"s` r*_swig_setattr_nondynamic_instance_variabler%sT r#cfd}|S)Nct||r%tt||ts |||yt d|z)Nz%You cannot add class attributes to %s)rrrrr)clsrr r!s rset_class_attrz?_swig_setattr_nondynamic_class_variable..set_class_attr)s: 3 jd1CX&N T5 ! !H3!NO Or#r$)r!r)s` r'_swig_setattr_nondynamic_class_variabler*(sP r#cfd}|S)zlClass decorator for adding a metaclass to a SWIG wrapped class - a slimmed down version of six.add_metaclasscp|j|j|jjSN)r __bases____dict__copy)r( metaclasss rwrapperz$_swig_add_metaclass..wrapper3s's}}cll6G6G6IJJr#r$)r1r2s` r_swig_add_metaclassr31sK Nr#c4eZdZdZeej Zy)_SwigNonDynamicMetazKMeta class to enforce nondynamic attributes (no new attributes) for a classN)rr __qualname____doc__r*r __setattr__r$r#rr5r58sU9$:J:JKKr#r5gdalcZeZdZdZedddZdZeZe jZ dZ d dZ y ) VirtualMemz'Proxy of C++ CPLVirtualMemShadow class.c6|jjSr-r r)xs rzVirtualMem.Asr#c8|jj|Sr-r>)r?vs rr@zVirtualMem.AsAFFJJqMr#zThe membership flag)docctd)NzNo constructor defined)r)rargskwargss r__init__zVirtualMem.__init__Cs566r#c,tj|S)zGetAddr(VirtualMem self))rVirtualMem_GetAddr)rs rGetAddrzVirtualMem.GetAddrHs--d33r#c2tj||||S)zKPin(VirtualMem self, size_t start_offset=0, size_t nsize=0, int bWriteOp=0))rVirtualMem_Pin)r start_offsetnsizebWriteOps rPinzVirtualMem.PinLs))$ eXNNr#N)rrr)rrr6r7rrrGrr rdelete_VirtualMem__swig_destroy__rJrPr$r#rr<r<>s72-/IOdeG7H"444Or#r<c*tjS)zGetUseExceptions() -> int)rGetUseExceptionsr$r#rrTrTSs  ' ' ))r#c*tjS)z_GetExceptionsLocal() -> int)r_GetExceptionsLocalr$r#rrVrVWs  * * ,,r#c,tj|S)z_SetExceptionsLocal(int bVal))r_SetExceptionsLocal)bVals rrXrX[s  * *4 00r#c*tjS)z_UseExceptions())r_UseExceptionsr$r#rr[r[_s  % % ''r#c*tjS)z_DontUseExceptions())r_DontUseExceptionsr$r#rr]r]cs  ) ) ++r#c*tjS)z+_UserHasSpecifiedIfUsingExceptions() -> int)r"_UserHasSpecifiedIfUsingExceptionsr$r#rr_r_gs  9 9 ;;r#c$eZdZdZddZdZdZy) ExceptionMgra Context manager to manage Python Exception state for GDAL/OGR/OSR/GNM. Separate exception state is maintained for each module (gdal, ogr, etc), and this class appears independently in all of them. This is built in top of calls to the older UseExceptions()/DontUseExceptions() functions. Example:: >>> print(gdal.GetUseExceptions()) 0 >>> with gdal.ExceptionMgr(): ... # Exceptions are now in use ... print(gdal.GetUseExceptions()) 1 >>> >>> # Exception state has now been restored >>> print(gdal.GetUseExceptions()) 0 c||_y)zK Save whether or not this context will be using exceptions N)requestedUseExceptions)r useExceptionss rrGzExceptionMgr.__init__s '4#r#ct|_t|jtj dk(r& ddlm}|r|j|jyyy#t$rd}Y-wxYw)z On context entry, save the current GDAL exception state, and set it to the state requested for the context osgeo.gdalr gdal_arrayN) rVcurrentUseExceptionsrXrcrarrrh ImportError)rrhs r __enter__zExceptionMgr.__enter__sl %8$9!D778  " "l 2 "(..t/J/JK 3 "!  "sA A-,A-ct|jtjdk(r& ddlm}|r|j|jyyy#t $rd}Y-wxYw)zy On exit, restore the GDAL/OGR/OSR/GNM exception state which was current on entry to the context rfrrgN)rXrirarrrhrj)rexc_typeexc_valexc_tbrhs r__exit__zExceptionMgr.__exit__sa D556  " "l 2 "(..t/H/HI 3 "!  "sA AAN)T)rrr6r7rGrkrpr$r#rraraks.4 L Jr#rac ddlm}|j ddlm}|j ddlm}|j ddlm}|j ddlm}|jy#t$rYhwxYw#t$rY`wxYw#t$rYXwxYw#t$rYPwxYw#t$rYywxYw)z Enable exceptions in all GDAL related modules (osgeo.gdal, osgeo.ogr, osgeo.osr, osgeo.gnm). Note: prior to GDAL 3.7, this only affected the calling modulerr9rgogrosrgnmN)rr:r[rjrhrsrurwr:rhrsrurws r UseExceptionsrys   !!#      '                WA6BBB#B26 BB BB B B # B/.B/2 B>=B>c ddlm}|j ddlm}|j ddlm}|j ddlm}|j ddlm}|jy#t$rYhwxYw#t$rY`wxYw#t$rYXwxYw#t$rYPwxYw#t$rYywxYw)z Disable exceptions in all GDAL related modules (osgeo.gdal, osgeo.ogr, osgeo.osr, osgeo.gnm). Note: prior to GDAL 3.7, this only affected the calling modulerr9rgrrrtrvN)rr:r]rjrhrsrurwrxs rDontUseExceptionsr|s !  %%'         '                rzc0tj|||S)z\TermProgress_nocb(double dfProgress, char const * pszMessage=None, void * pData=None) -> int)rTermProgress_nocb) dfProgress pszMessagepDatas rr~r~s  ( (Z GGr#c.tj||S)zDOpenNumPyArray(PyArrayObject * psArray, bool binterleave) -> Dataset)rOpenNumPyArray)psArray binterleaves rrrs  % %g{ ;;r#c,tj|S)zBOpenMultiDimensionalNumPyArray(PyArrayObject * psArray) -> Dataset)rOpenMultiDimensionalNumPyArrayrs rrrs  5 5g >>r#c,tj|S)zBGetArrayFilename(PyArrayObject * psArray) -> retStringAndCPLFree *)rGetArrayFilenamers rrrs  ' ' 00r#c @tj|||||||||| | S)zBandRasterIONumPy(Band band, int bWrite, double xoff, double yoff, double xsize, double ysize, PyArrayObject * psArray, GDALDataType buf_type, GDALRIOResampleAlg resample_alg, GDALProgressFunc callback=0, void * callback_data=None) -> CPLErr)rBandRasterIONumPy) bandbWritexoffyoffxsizeysizerbuf_type resample_algcallback callback_datas rrrs6  ( (vtT5%QXZbdprz}J KKr#c Dtj|||||||||| | | | S)aDatasetIONumPy(Dataset ds, int bWrite, double xoff, double yoff, double xsize, double ysize, PyArrayObject * psArray, GDALDataType buf_type, GDALRIOResampleAlg resample_alg, GDALProgressFunc callback=0, void * callback_data=None, bool binterleave=True, int band_list=0) -> CPLErr)rDatasetIONumPy) dsrrrrrrrrrrr band_lists rrrsC  % %b&$eUGU]_kmuxEGRT] ^^r#c6tj||||||S)zMDArrayIONumPy(bool bWrite, GDALMDArrayHS * mdarray, PyArrayObject * psArray, int nDims1, int nDims3, GDALExtendedDataTypeHS * buffer_datatype) -> CPLErr)rMDArrayIONumPy)rmdarrayrnDims1nDims3buffer_datatypes rrrs  % %fgwP_ ``r#c4tj|||||S)zAddNumpyArrayToDict(PyObject * dict, ArrowSchema const * schemaField, ArrowArray const * arrayField, std::string const & osPrefix, PyObject * pointerArrayKeeper) -> bool)rAddNumpyArrayToDict)dict schemaField arrayFieldosPrefixpointerArrayKeepers rrrs  * *4j(Tf ggr#c0tj|||S)zw_RecordBatchAsNumpy(VoidPtrAsLong recordBatchPtr, VoidPtrAsLong schemaPtr, PyObject * pointerArrayKeeper) -> PyObject *)r_RecordBatchAsNumpy)recordBatchPtr schemaPtrrs rrr s  * *>9FX YYr#c,tj|S)z)VirtualMemGetArray(VirtualMem virtualmem))rVirtualMemGetArray) virtualmems rrrs  ) )* 55r#c2tj||||S)zlRATValuesIONumPyWrite(RasterAttributeTable poRAT, int nField, int nStart, PyArrayObject * psArray) -> CPLErr)rRATValuesIONumPyWrite)poRATnFieldnStartrs rrrs  , ,UFFG LLr#c2tj||||S)zcRATValuesIONumPyRead(RasterAttributeTable poRAT, int nField, int nStart, int nLength) -> PyObject *)rRATValuesIONumPyRead)rrrnLengths rrrs  + +E667 KKr#) gdalconstrc|j}|dk(rd}n|dk(rd}n tdt||}|=|;t|jdk(rt j |}| t|||S)NrTpixelF"Interleave should be band or pixelstr)lower ValueErrorrrrr:OpenCopyDatasetInfo)array prototype_ds interleavers r OpenArrayr0s!!#JV w  =>> z *B ~,2   & &% /99\2L  # L" - Ir#ct|tjtfrI|tjk(rt j StjD]\}}||k(s |cSy t|S#t$rYywxYwr-) rnumpydtyper complex64r GDT_CFloat32codesitemsKeyError)codekeyr s r flip_coderEsz$d+, 5?? ")) )++- JC}   ;   s+A44 B?Bcnt|tjtfs t dt |S)NzInput must be a type)rrrr TypeErrorr) numeric_types rNumericTypeCodeToGDALTypeCoderVs, lU[[$$7 8.// \ ""r#ct|Sr-)r) gdal_codes rGDALTypeCodeToNumericTypeCoder[s Y r#cftjrttjyr-)r:rT RuntimeErrorGetLastErrorMsgr$r#r_RaiseExceptionr^s' 4//122r#c tj|} | &td|zdztjzt | ||||||||| | | |  S)Nz Can't open z ) buf_xsize buf_ysizerrrrrr)r:rrrDatasetReadAsArray)filenamerrrrrrrrrrrrrs rLoadFilerbsg 8 B zx/6t7K7K7MMNN b$eU(1YQY+7'/})3(1  33r#ctj|}|td|z|j|t |||S)NzCan't find driver )r:GetDriverByNamer CreateCopyr) src_arrayrformat prototyperdrivers r SaveArrayrrsD  ! !& )F ~-f455   XyIz'R SSr#c||St|drdt|dr|jr t|St|ds t|St|}|t|k(r t|}|St|dr t|S|S)znConverts an object with a __int__ or __float__ method to the corresponding primitive type, or return x.__int__ is_integer __float__)rrintfloat)r?rets r_to_primitive_typerys yq) 1l # q6MK(q6M(Cc#h#hJ K Qx Hr#c>| |j}| |j}t|}t|}t|}t|}t|}t|}| "tt d|j dz} | j } | dk(r d} d}d}d}n| dk(r d} d}d}d}n td t| }|dk(ry|dk(r)t|j| d||||||||| | | S|5||}||}|c|j| dj}t d|D]6}| |}||j|jk7s'tj}8t|}|!tj}tj }n t#|}|tj$k(rX|jd}|j'd|j)d d d k(rtj*}|j'd| r|||fn|||f}tj,||}nt|j.dk7r td|j.|}|j.|}|||k7r td|||k7r td|j.||k7rtd||fzt#|j0j2}|s td|||k7r td|}t5|d||||||| | | | | dk7r t7y|S)zPure python implementation of reading a chunk of a GDAL file into a numpy array. Used by the gdal.Dataset.ReadAsArray method.NrrTrrFr)rrrbuf_objrrr PIXELTYPEIMAGE_STRUCTURE SIGNEDBYTErzArray should have 3 dimensions3Specified buf_xsize not consistent with array shape3Specified buf_ysize not consistent with array shape9Dimension %d of array should have size %d to store bands)0array does not have corresponding GDAL data type1Specified buf_type not consistent with array type) RasterXSize RasterYSizerlistrange RasterCountrrlenBandReadAsArray GetRasterBandDataTyper GDT_Float32rrfloat32rGDT_Byte!_EnablePixelTypeSignedByteWarningGetMetadataItemint8emptyshaperrrr)rrr win_xsize win_ysizerrrrrrrrrxdimydimbanddimnbandsidx band_indextypecoder buf_shapeshape_buf_xsizeshape_buf_ysizedatatypes rrrsjNN NN d #D d #D"9-I"9-I"9-I"9-Iq"..1"456 !!#JV  w  =>> ^F { {r// ! =tT9V_)2iRZ'.,8(0-: < <  !I  !I  '' ! 5>>HQ' 5&s^ r// ;DDD(44H 5 1:   ,,H}}H4X>H y)) )##A&D  2 25 9##K1BC|S ::  2 24 86@VY 2yR[]cFd ++ix8 w}}  "=> >!---!---  Y/%ARS S  Y/%ARS S == !V +X\cek[llm m01C1CDOP P  H$8PQ Qb!T4Ixx )-012  Nr#c t|}t|}|"ttd|jdz}|j }|dk(r d}d} d} d} n|dk(r d}d} d} d} n t dt |dk(r|0t |jdk7r#t |jd k7r t d t |jd k(r%|j| dk7rt d | z|| }t|j|d|||||| S|t |jd k7r t d |j| } |j| } | |z|jkDs| |z|jkDr t d|j| t |k7rt d| t |fzt|jj}|sTtj dd|j#t$j&}t|jj}|s t dt)|d||| | ||||||| }|dk7r t+|S)zPure python implementation of writing a chunk of a GDAL file from a numpy array. Used by the gdal.Dataset.WriteArray method.rrTrrrFrrzexpected array of dim 2 or 3z)expected size of dimension %d should be 1)rrrrrzexpected array of dim 31array larger than output file, or offset off edgerrhforce array to float64r)rrrrrrrr BandWriteArrayrrrrrrr:Debugastyperfloat64rr)rrrrrrrrrrrrrrrrs rDatasetWriteArrayr"sf d #D d #Dq"..1"456 !!#JV  w  =>> 9~ =S-2s5;;7G17L;< < u{{ q {{7#q( !Lw!VWW'NEb..y|<##'d'/}N N  }EKK(A-233 KK E KK E t|bnn$ r~~(ELMM {{7s9~-TX_adenaoWppqq,U[[-=-=>H  sCJJ JJ d #D d #D"9-I"9-I"9-I"9-I  !I  !I  }}H0:   ,,H}}H4X>H y)) )  2 25 9##K1BC|S ::  2 24 8++y)4HE w}} V +AB B w}}  "%mmA.O%mmA.O}}Q1$ !OPP%mmA.O%mmA.O  Y/%ARS S  Y/%ARS S01C1CDOP P  H$8PQ Qq$i (L(MSVWX Nr#c b|t|jdk7r tdt|}t|}|jd}|jd}||z|jkDs||z|j kDr tdt |jj} | sTtjdd|jtj}t |jj} | s tdt|d|||||| ||| } | dk7r t| S) z~Pure python implementation of writing a chunk of a GDAL file from a numpy array. Used by the gdal.Band.WriteArray method.rzexpected array of dim 2rrrrhrr)rr rrr$r%rrrr:rr rr!rr) rrrrrrrrrrrs rrrs  }EKK(A-233 d #D d #D KKNE KKNE t|djj EDL4::$=LMM,U[[-=-=>H  .>?B|&&--.KH.UVB| D'' '' ' EGG  ") T\\^$:4<<>J% DLLN "34 4x t~~'( ) ;;#*#*$&JJL2 357 77r#c"t|\}}|Sr-)r7)r;r_s rExtendedDataTypeToNumPyDataTyperAs226KHa Or#c |sdg|jz}|s+|jDcgc]}|j}}|sdg|jz}|8|s|j}t |\}}t j ||}nKt|jj}|s tdtjj|}td|||||} | dk7r t|Scc}w)NrrrrF)GetDimensionCount GetDimensionsr: GetDataTyper7rr rrrrr:r1r2rr) rarray_start_idxcount array_steprrdimrrrs rMDArrayReadAsArrayrJs # 9 9 ;; +2+@+@+BDC#++-DD S74466 %113O$D_$U!/++e8401C1CDOP P//66x@ /: _C ax N'EsDc2|sdg|jz}|sdg|jz}|j}t|}|j|k7rt |jj }|sTt jdd|jtj}t |jj }|s tdt jj|}td|||||}|dk7r t|S)NrrrhrrT)rCrErArrrr:rr rr!rr1r2rr)rrrFrHrrrrs rMDArrayWriteArrayrLs # 9 9 ;; S74466 ))+O.?H {{h01A1AB JJ|%= >LL/E4U[[5E5EFHOP P//66x@ w O \C ax Jr#c| tdt|tjstj|}|j dk7r td||j z|jkDr tdtj|jtjr |jtj}ntj|jtjr |jtj}nOtj|jtjr|jt }n tdt#||||}|dk7r t%|S)z Pure Python implementation of writing a chunk of the RAT from a numpy array. Type of array is coerced to one of the types (int, double, string) supported. Called from RasterAttributeTable.WriteArray zExpected array of dim 1rz1Array too big to fit into RAT from start positionz9Array not of a supported type (integer, double or string)r)rrrndarrayrndimsize GetRowCount issubdtyperintegerr int32floatingdouble characterbytesrr)ratrfieldstartrs r RATWriteArrayr\ s  }233 eU]] + E" zzQ233  coo//LMM  U]]3 U[[)   %++u~~ 6 U\\*   %++u 7 U#TUU UE5 9C ax Jr#cd||j|z }t||||}| t|S)z Pure Python implementation of reading a chunk of the RAT into a numpy array. Called from RasterAttributeTable.ReadAsArray )rQrr)rYrZr[lengthrs r RATReadArrayr_1s: ~"U* sE5& 9C { Jr#c J|j|j|j}|dk7r|j|j |dk(r|dk(r|j |y |d|d|d|d|d|dg}|d||dzz||dzz|d<|d||dzz||dzz|d<|j |d|d|d|d|d|dfy |j dkDr|dk(r4|dk(r/|j|j|jy |j}g}|D]}tj} |j| _ |j| _ |j| _|j|z | _|j |z | _|j"| _|j$| _|j'|  |j||jy y #t)dYy xYw) ac Copy georeferencing information and metadata from one dataset to another. src: input dataset dst: output dataset - It can be a ROI - xoff, yoff: dst's offset with respect to src in pixel/line. Notes: Destination dataset must have update access. Certain formats do not support creation of geotransforms and/or gcps. )rrrrrrrrrrzFailed to set GCPsN) SetMetadata GetMetadataGetGeoTransform SetProjectionGetProjectionRefSetGeoTransform GetGCPCountSetGCPsGetGCPsGetGCPProjectionr:GCPGCPXGCPYGCPZGCPPixelGCPLineInfoIdr5print) srcdstrrgtngtgcpsnew_gcpsgcpngcps rrr>sOOCOO%&   B  #..01 19    #B ?a5"Q%A1r!ube, );;=DH &xxzHH HH HH # t 3 "{{T1 HH &&% &  Hc&:&:&<=   *+s 1 H H")NN)rN)rNTr)Nr)GTiffNr)NNNNN)r)rr)]sysr_swig_python_version_info __package__rrrbuiltinsr rjrr%r*r3rr5r:objectr<VirtualMem_swigregisterrTrVrXr[r]r_raryr|r~ TermProgressrrrrrrrrrrrrosgeor AllRegisterruint8GDT_Int8r  GDT_UInt16uint16 GDT_Int16int16 GDT_UInt32uint32 GDT_Int32rT GDT_UInt64uint64 GDT_Int64int64rr GDT_Float64r! GDT_CInt16r GDT_CInt32r GDT_CFloat64 complex128rrrrrrGRIORA_NearestNeighbourrrrrr"rrr7rArJrLr\r_rr$r#rrs:#/"[ L$L OO&$ ##J/*-1(,<:J6:J~ < @H'' <?1K^ahZ6ML    U[[   UZZ   u||   ekk   u||   ekk   u||   ekk         u   u   %//   !1!1 3 *"#  3ATtd66$6 3 T &!"TTSW!%$($@$@ $DV!% gT'(a $!'#'#?#?#4 EP !qDD"dT4!%!=!=!FP&'Q $ < < "H7B +/ $%)*."& >+/%)<$L 7 ]sJ33 K?K