/usr/lib/python2.7/dist-packages/osgeo/gdal_array.py is in python-gdal 1.11.3+dfsg-3build2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 | # This file was automatically generated by SWIG (http://www.swig.org).
# Version 3.0.8
#
# Do not make changes to this file unless you know what you are doing--modify
# the SWIG interface file instead.
from sys import version_info
if version_info >= (2, 6, 0):
def swig_import_helper():
from os.path import dirname
import imp
fp = None
try:
fp, pathname, description = imp.find_module('_gdal_array', [dirname(__file__)])
except ImportError:
import _gdal_array
return _gdal_array
if fp is not None:
try:
_mod = imp.load_module('_gdal_array', fp, pathname, description)
finally:
fp.close()
return _mod
_gdal_array = swig_import_helper()
del swig_import_helper
else:
import _gdal_array
del version_info
try:
_swig_property = property
except NameError:
pass # Python < 2.2 doesn't have 'property'.
def _swig_setattr_nondynamic(self, class_type, name, value, static=1):
if (name == "thisown"):
return self.this.own(value)
if (name == "this"):
if type(value).__name__ == 'SwigPyObject':
self.__dict__[name] = value
return
method = class_type.__swig_setmethods__.get(name, None)
if method:
return method(self, value)
if (not static):
if _newclass:
object.__setattr__(self, name, value)
else:
self.__dict__[name] = value
else:
raise AttributeError("You cannot add attributes to %s" % self)
def _swig_setattr(self, class_type, name, value):
return _swig_setattr_nondynamic(self, class_type, name, value, 0)
def _swig_getattr_nondynamic(self, class_type, name, static=1):
if (name == "thisown"):
return self.this.own()
method = class_type.__swig_getmethods__.get(name, None)
if method:
return method(self)
if (not static):
return object.__getattr__(self, name)
else:
raise AttributeError(name)
def _swig_getattr(self, class_type, name):
return _swig_getattr_nondynamic(self, class_type, name, 0)
def _swig_repr(self):
try:
strthis = "proxy of " + self.this.__repr__()
except Exception:
strthis = ""
return "<%s.%s; %s >" % (self.__class__.__module__, self.__class__.__name__, strthis,)
try:
_object = object
_newclass = 1
except AttributeError:
class _object:
pass
_newclass = 0
class VirtualMem(_object):
"""Proxy of C++ CPLVirtualMemShadow class."""
__swig_setmethods__ = {}
__setattr__ = lambda self, name, value: _swig_setattr(self, VirtualMem, name, value)
__swig_getmethods__ = {}
__getattr__ = lambda self, name: _swig_getattr(self, VirtualMem, name)
def __init__(self, *args, **kwargs):
raise AttributeError("No constructor defined")
__repr__ = _swig_repr
__swig_destroy__ = _gdal_array.delete_VirtualMem
__del__ = lambda self: None
def GetAddr(self):
"""GetAddr(VirtualMem self)"""
return _gdal_array.VirtualMem_GetAddr(self)
def Pin(self, start_offset=0, nsize=0, bWriteOp=0):
"""
Pin(VirtualMem self, size_t start_offset=0, size_t nsize=0, int bWriteOp=0)
Pin(VirtualMem self, size_t start_offset=0, size_t nsize=0)
Pin(VirtualMem self, size_t start_offset=0)
Pin(VirtualMem self)
"""
return _gdal_array.VirtualMem_Pin(self, start_offset, nsize, bWriteOp)
VirtualMem_swigregister = _gdal_array.VirtualMem_swigregister
VirtualMem_swigregister(VirtualMem)
def GetArrayFilename(psArray):
"""GetArrayFilename(PyArrayObject * psArray) -> retStringAndCPLFree *"""
return _gdal_array.GetArrayFilename(psArray)
def BandRasterIONumPy(band, bWrite, xoff, yoff, xsize, ysize, psArray, buf_type):
"""BandRasterIONumPy(Band band, int bWrite, int xoff, int yoff, int xsize, int ysize, PyArrayObject * psArray, int buf_type) -> CPLErr"""
return _gdal_array.BandRasterIONumPy(band, bWrite, xoff, yoff, xsize, ysize, psArray, buf_type)
def VirtualMemGetArray(virtualmem):
"""VirtualMemGetArray(VirtualMem virtualmem)"""
return _gdal_array.VirtualMemGetArray(virtualmem)
def RATValuesIONumPyWrite(poRAT, nField, nStart, psArray):
"""RATValuesIONumPyWrite(RasterAttributeTable poRAT, int nField, int nStart, PyArrayObject * psArray) -> CPLErr"""
return _gdal_array.RATValuesIONumPyWrite(poRAT, nField, nStart, psArray)
def RATValuesIONumPyRead(poRAT, nField, nStart, nLength):
"""RATValuesIONumPyRead(RasterAttributeTable poRAT, int nField, int nStart, int nLength) -> PyObject *"""
return _gdal_array.RATValuesIONumPyRead(poRAT, nField, nStart, nLength)
import numpy
import _gdal_array
import gdalconst
import gdal
gdal.AllRegister()
codes = { gdalconst.GDT_Byte : numpy.uint8,
gdalconst.GDT_UInt16 : numpy.uint16,
gdalconst.GDT_Int16 : numpy.int16,
gdalconst.GDT_UInt32 : numpy.uint32,
gdalconst.GDT_Int32 : numpy.int32,
gdalconst.GDT_Float32 : numpy.float32,
gdalconst.GDT_Float64 : numpy.float64,
gdalconst.GDT_CInt16 : numpy.complex64,
gdalconst.GDT_CInt32 : numpy.complex64,
gdalconst.GDT_CFloat32 : numpy.complex64,
gdalconst.GDT_CFloat64 : numpy.complex128
}
def OpenArray( array, prototype_ds = None ):
ds = gdal.Open( GetArrayFilename(array) )
if ds is not None and prototype_ds is not None:
if type(prototype_ds).__name__ == 'str':
prototype_ds = gdal.Open( prototype_ds )
if prototype_ds is not None:
CopyDatasetInfo( prototype_ds, ds )
return ds
def flip_code(code):
if isinstance(code, (numpy.dtype,type)):
# since several things map to complex64 we must carefully select
# the opposite that is an exact match (ticket 1518)
if code == numpy.int8:
return gdalconst.GDT_Byte
if code == numpy.complex64:
return gdalconst.GDT_CFloat32
for key, value in codes.items():
if value == code:
return key
return None
else:
try:
return codes[code]
except KeyError:
return None
def NumericTypeCodeToGDALTypeCode(numeric_type):
if not isinstance(numeric_type, (numpy.dtype,type)):
raise TypeError("Input must be a type")
return flip_code(numeric_type)
def GDALTypeCodeToNumericTypeCode(gdal_code):
return flip_code(gdal_code)
def LoadFile( filename, xoff=0, yoff=0, xsize=None, ysize=None ):
ds = gdal.Open( filename )
if ds is None:
raise ValueError("Can't open "+filename+"\n\n"+gdal.GetLastErrorMsg())
return DatasetReadAsArray( ds, xoff, yoff, xsize, ysize )
def SaveArray( src_array, filename, format = "GTiff", prototype = None ):
driver = gdal.GetDriverByName( format )
if driver is None:
raise ValueError("Can't find driver "+format)
return driver.CreateCopy( filename, OpenArray(src_array,prototype) )
def DatasetReadAsArray( ds, xoff=0, yoff=0, xsize=None, ysize=None, buf_obj=None ):
if xsize is None:
xsize = ds.RasterXSize
if ysize is None:
ysize = ds.RasterYSize
if ds.RasterCount == 1:
return BandReadAsArray( ds.GetRasterBand(1), xoff, yoff, xsize, ysize, buf_obj = buf_obj)
datatype = ds.GetRasterBand(1).DataType
for band_index in range(2,ds.RasterCount+1):
if datatype != ds.GetRasterBand(band_index).DataType:
datatype = gdalconst.GDT_Float32
typecode = GDALTypeCodeToNumericTypeCode( datatype )
if typecode == None:
datatype = gdalconst.GDT_Float32
typecode = numpy.float32
if buf_obj is not None:
for band_index in range(1,ds.RasterCount+1):
BandReadAsArray( ds.GetRasterBand(band_index),
xoff, yoff, xsize, ysize, buf_obj = buf_obj[band_index-1])
return buf_obj
array_list = []
for band_index in range(1,ds.RasterCount+1):
band_array = BandReadAsArray( ds.GetRasterBand(band_index),
xoff, yoff, xsize, ysize)
array_list.append( numpy.reshape( band_array, [1,ysize,xsize] ) )
return numpy.concatenate( array_list )
def BandReadAsArray( band, xoff = 0, yoff = 0, win_xsize = None, win_ysize = None,
buf_xsize=None, buf_ysize=None, buf_obj=None ):
"""Pure python implementation of reading a chunk of a GDAL file
into a numpy array. Used by the gdal.Band.ReadAsArray method."""
if win_xsize is None:
win_xsize = band.XSize
if win_ysize is None:
win_ysize = band.YSize
if buf_obj is None:
if buf_xsize is None:
buf_xsize = win_xsize
if buf_ysize is None:
buf_ysize = win_ysize
else:
if len(buf_obj.shape) == 2:
shape_buf_xsize = buf_obj.shape[1]
shape_buf_ysize = buf_obj.shape[0]
else:
shape_buf_xsize = buf_obj.shape[2]
shape_buf_ysize = buf_obj.shape[1]
if buf_xsize is not None and buf_xsize != shape_buf_xsize:
raise ValueError('Specified buf_xsize not consistent with array shape')
if buf_ysize is not None and buf_ysize != shape_buf_ysize:
raise ValueError('Specified buf_ysize not consistent with array shape')
buf_xsize = shape_buf_xsize
buf_ysize = shape_buf_ysize
if buf_obj is None:
datatype = band.DataType
typecode = GDALTypeCodeToNumericTypeCode( datatype )
if typecode == None:
datatype = gdalconst.GDT_Float32
typecode = numpy.float32
else:
datatype = NumericTypeCodeToGDALTypeCode( typecode )
if datatype == gdalconst.GDT_Byte and band.GetMetadataItem('PIXELTYPE', 'IMAGE_STRUCTURE') == 'SIGNEDBYTE':
typecode = numpy.int8
ar = numpy.empty([buf_ysize,buf_xsize], dtype = typecode)
if BandRasterIONumPy( band, 0, xoff, yoff, win_xsize, win_ysize,
ar, datatype ) != 0:
return None
return ar
else:
datatype = NumericTypeCodeToGDALTypeCode( buf_obj.dtype.type )
if not datatype:
raise ValueError("array does not have corresponding GDAL data type")
if BandRasterIONumPy( band, 0, xoff, yoff, win_xsize, win_ysize,
buf_obj, datatype ) != 0:
return None
return buf_obj
def BandWriteArray( band, array, xoff=0, yoff=0 ):
"""Pure python implementation of writing a chunk of a GDAL file
from a numpy array. Used by the gdal.Band.WriteArray method."""
if array is None or len(array.shape) != 2:
raise ValueError("expected array of dim 2")
xsize = array.shape[1]
ysize = array.shape[0]
if xsize + xoff > band.XSize or ysize + yoff > band.YSize:
raise ValueError("array larger than output file, or offset off edge")
datatype = NumericTypeCodeToGDALTypeCode( array.dtype.type )
# if we receive some odd type, like int64, try casting to a very
# generic type we do support (#2285)
if not datatype:
gdal.Debug( 'gdal_array', 'force array to float64' )
array = array.astype( numpy.float64 )
datatype = NumericTypeCodeToGDALTypeCode( array.dtype.type )
if not datatype:
raise ValueError("array does not have corresponding GDAL data type")
return BandRasterIONumPy( band, 1, xoff, yoff, xsize, ysize,
array, datatype )
def RATWriteArray(rat, array, field, start=0):
"""
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
"""
if array is None:
raise ValueError("Expected array of dim 1")
# if not the array type convert it to handle lists etc
if not isinstance(array, numpy.ndarray):
array = numpy.array(array)
if array.ndim != 1:
raise ValueError("Expected array of dim 1")
if (start + array.size) > rat.GetRowCount():
raise ValueError("Array too big to fit into RAT from start position")
if numpy.issubdtype(array.dtype, numpy.integer):
# is some type of integer - coerce to standard int
# TODO: must check this is fine on all platforms
# confusingly numpy.int 64 bit even if native type 32 bit
array = array.astype(numpy.int32)
elif numpy.issubdtype(array.dtype, numpy.floating):
# is some type of floating point - coerce to double
array = array.astype(numpy.double)
elif numpy.issubdtype(array.dtype, numpy.character):
# cast away any kind of Unicode etc
array = array.astype(numpy.character)
else:
raise ValueError("Array not of a supported type (integer, double or string)")
return RATValuesIONumPyWrite(rat, field, start, array)
def RATReadArray(rat, field, start=0, length=None):
"""
Pure Python implementation of reading a chunk of the RAT
into a numpy array. Called from RasterAttributeTable.ReadAsArray
"""
if length is None:
length = rat.GetRowCount() - start
return RATValuesIONumPyRead(rat, field, start, length)
def CopyDatasetInfo( src, dst, xoff=0, yoff=0 ):
"""
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.
"""
dst.SetMetadata( src.GetMetadata() )
#Check for geo transform
gt = src.GetGeoTransform()
if gt != (0,1,0,0,0,1):
dst.SetProjection( src.GetProjectionRef() )
if (xoff == 0) and (yoff == 0):
dst.SetGeoTransform( gt )
else:
ngt = [gt[0],gt[1],gt[2],gt[3],gt[4],gt[5]]
ngt[0] = gt[0] + xoff*gt[1] + yoff*gt[2];
ngt[3] = gt[3] + xoff*gt[4] + yoff*gt[5];
dst.SetGeoTransform( ( ngt[0], ngt[1], ngt[2], ngt[3], ngt[4], ngt[5] ) )
#Check for GCPs
elif src.GetGCPCount() > 0:
if (xoff == 0) and (yoff == 0):
dst.SetGCPs( src.GetGCPs(), src.GetGCPProjection() )
else:
gcps = src.GetGCPs()
#Shift gcps
new_gcps = []
for gcp in gcps:
ngcp = gdal.GCP()
ngcp.GCPX = gcp.GCPX
ngcp.GCPY = gcp.GCPY
ngcp.GCPZ = gcp.GCPZ
ngcp.GCPPixel = gcp.GCPPixel - xoff
ngcp.GCPLine = gcp.GCPLine - yoff
ngcp.Info = gcp.Info
ngcp.Id = gcp.Id
new_gcps.append(ngcp)
try:
dst.SetGCPs( new_gcps , src.GetGCPProjection() )
except:
print ("Failed to set GCPs")
return
return
# This file is compatible with both classic and new-style classes.
|