| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- # -*- coding: utf-8 -*-
- """
- Created on Fri Jun 8 18:46:34 2012
- @author: pietro
- """
- from grass.pygrass.raster.raster_type import TYPE as RTYPE
- import ctypes
- import numpy as np
- CELL = (np.int, np.int0, np.int8, np.int16, np.int32, np.int64)
- FCELL = (np.float, np.float16, np.float32)
- DCELL = (np.float64, np.float128)
- class Buffer(np.ndarray):
- """shape, mtype='FCELL', buffer=None, offset=0,
- strides=None, order=None
- """
- @property
- def mtype(self):
- if self.dtype in CELL:
- return 'CELL'
- elif self.dtype in FCELL:
- return 'FCELL'
- elif self.dtype in DCELL:
- return DCELL
- else:
- err = "Raster type: %r not supported by GRASS."
- raise TypeError(err % self.dtype)
- def __new__(cls, shape, mtype='FCELL', buffer=None, offset=0,
- strides=None, order=None):
- #import pdb; pdb.set_trace()
- obj = np.ndarray.__new__(cls, shape, RTYPE[mtype]['numpy'],
- buffer, offset, strides, order)
- obj.pointer_type = ctypes.POINTER(RTYPE[mtype]['ctypes'])
- obj.p = obj.ctypes.data_as(obj.pointer_type)
- return obj
- def __array_finalize__(self, obj):
- if obj is None:
- return
- self.pointer_type = getattr(obj, 'pointer_type', None)
- self.p = getattr(obj, 'p', None)
- def __array_wrap__(self, out_arr, context=None):
- """See:
- http://docs.scipy.org/doc/numpy/user/
- basics.subclassing.html#array-wrap-for-ufuncs"""
- if out_arr.dtype == np.bool:
- # there is not support for boolean maps, so convert into integer
- out_arr = out_arr.astype(np.int32)
- out_arr.p = out_arr.ctypes.data_as(out_arr.pointer_type)
- return np.ndarray.__array_wrap__(self, out_arr, context)
|
