def __getitem__(self, key):
self._update()
dx = c_double()
dy = c_double()
dz = c_double()
m = self.__len__()
has_z = self._ndim == 3
if isinstance(key, int):
if key + m < 0 or key >= m:
raise IndexError("index out of range")
if key < 0:
i = m + key
else:
i = key
lgeos.GEOSCoordSeq_getX(self._cseq, i, byref(dx))
lgeos.GEOSCoordSeq_getY(self._cseq, i, byref(dy))
if has_z:
lgeos.GEOSCoordSeq_getZ(self._cseq, i, byref(dz))
return (dx.value, dy.value, dz.value)
else:
return (dx.value, dy.value)
elif isinstance(key, slice):
res = []
start, stop, stride = key.indices(m)
for i in range(start, stop, stride):
lgeos.GEOSCoordSeq_getX(self._cseq, i, byref(dx))
lgeos.GEOSCoordSeq_getY(self._cseq, i, byref(dy))
if has_z:
lgeos.GEOSCoordSeq_getZ(self._cseq, i, byref(dz))
res.append((dx.value, dy.value, dz.value))
else:
res.append((dx.value, dy.value))
return res
else:
raise TypeError("key must be an index or slice")
def __iter__(self):
self._update()
dx = c_double()
dy = c_double()
dz = c_double()
for i in range(self.__len__()):
lgeos.GEOSCoordSeq_getX(self._cseq, i, byref(dx))
lgeos.GEOSCoordSeq_getY(self._cseq, i, byref(dy))
if self._ndim == 3: # TODO: use hasz
lgeos.GEOSCoordSeq_getZ(self._cseq, i, byref(dz))
yield (dx.value, dy.value, dz.value)
else:
yield (dx.value, dy.value)
def ctypes(self):
self._update()
n = self._ndim
m = self.__len__()
array_type = c_double * (m * n)
data = array_type()
temp = c_double()
for i in xrange(m):
lgeos.GEOSCoordSeq_getX(self._cseq, i, byref(temp))
data[n * i] = temp.value
lgeos.GEOSCoordSeq_getY(self._cseq, i, byref(temp))
data[n * i + 1] = temp.value
if n == 3: # TODO: use hasz
lgeos.GEOSCoordSeq_getZ(self._cseq, i, byref(temp))
data[n * i + 2] = temp.value
return data
def _ctypes(self):
self._update()
has_z = self._ndim == 3
n = self._ndim
m = self.__len__()
array_type = c_double * (m * n)
data = array_type()
temp = c_double()
for i in range(m):
lgeos.GEOSCoordSeq_getX(self._cseq, i, byref(temp))
data[n * i] = temp.value
lgeos.GEOSCoordSeq_getY(self._cseq, i, byref(temp))
data[n * i + 1] = temp.value
if has_z:
lgeos.GEOSCoordSeq_getZ(self._cseq, i, byref(temp))
data[n * i + 2] = temp.value
return data
def _ctypes(self):
if not self._ctypes_data:
temp = c_double()
n = self._ndim
m = len(self.geoms)
array_type = c_double * (m * n)
data = array_type()
for i in range(m):
g = self.geoms[i]._geom
cs = lgeos.GEOSGeom_getCoordSeq(g)
lgeos.GEOSCoordSeq_getX(cs, 0, byref(temp))
data[n*i] = temp.value
lgeos.GEOSCoordSeq_getY(cs, 0, byref(temp))
data[n*i+1] = temp.value
if n == 3: # TODO: use hasz
lgeos.GEOSCoordSeq_getZ(cs, 0, byref(temp))
data[n*i+2] = temp.value
self._ctypes_data = data
return self._ctypes_data
def __getitem__(self, i):
self._update()
M = self.__len__()
if i + M < 0 or i >= M:
raise IndexError("index out of range")
if i < 0:
ii = M + i
else:
ii = i
dx = c_double()
dy = c_double()
dz = c_double()
lgeos.GEOSCoordSeq_getX(self._cseq, ii, byref(dx))
lgeos.GEOSCoordSeq_getY(self._cseq, ii, byref(dy))
if self._ndim == 3: # TODO: use hasz
lgeos.GEOSCoordSeq_getZ(self._cseq, ii, byref(dz))
return (dx.value, dy.value, dz.value)
else:
return (dx.value, dy.value)
def _ctypes(self):
self._update()
has_z = self._ndim == 3
n = self._ndim or 0
if n == 0:
# ignore with NumPy 1.21 __array_interface__
raise AttributeError("empty geometry sequence")
m = self.__len__()
array_type = c_double * (m * n)
data = array_type()
temp = c_double()
for i in range(m):
lgeos.GEOSCoordSeq_getX(self._cseq, i, byref(temp))
data[n * i] = temp.value
lgeos.GEOSCoordSeq_getY(self._cseq, i, byref(temp))
data[n * i + 1] = temp.value
if has_z:
lgeos.GEOSCoordSeq_getZ(self._cseq, i, byref(temp))
data[n * i + 2] = temp.value
return data
def nearest_points(self, other):
"""Returns list of tuples like (x, y) of nearest points"""
_ndim = 2 # TODO: remove hardcoding
_cseq = self.impl['nearest_points'](self, other)
cs_len = c_uint(0)
lgeos.GEOSCoordSeq_getSize(_cseq, byref(cs_len))
llen = cs_len.value
dx = c_double()
dy = c_double()
dz = c_double()
has_z = _ndim == 3
res = []
for i in xrange(llen):
lgeos.GEOSCoordSeq_getX(_cseq, i, byref(dx))
lgeos.GEOSCoordSeq_getY(_cseq, i, byref(dy))
if has_z:
lgeos.GEOSCoordSeq_getZ(_cseq, i, byref(dz))
res.append((dx.value, dy.value, dz.value))
else:
res.append((dx.value, dy.value))
return res
