forked from s_ranjbar/city_retrofit
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
"""
|
|
Iterative forms of operations
|
|
"""
|
|
|
|
from shapely.topology import Delegating
|
|
|
|
|
|
class IterOp(Delegating):
|
|
|
|
"""A generating non-data descriptor.
|
|
"""
|
|
|
|
def __call__(self, context, iterator, value=True):
|
|
if context._geom is None:
|
|
raise ValueError("Null geometry supports no operations")
|
|
for item in iterator:
|
|
try:
|
|
this_geom, ob = item
|
|
except TypeError:
|
|
this_geom = item
|
|
ob = this_geom
|
|
if not this_geom._geom:
|
|
raise ValueError("Null geometry supports no operations")
|
|
try:
|
|
retval = self.fn(context._geom, this_geom._geom)
|
|
except Exception as err:
|
|
self._check_topology(err, context, this_geom)
|
|
if bool(retval) == value:
|
|
yield ob
|
|
|
|
|
|
# utilities
|
|
disjoint = IterOp('disjoint')
|
|
touches = IterOp('touches')
|
|
intersects = IterOp('intersects')
|
|
crosses = IterOp('crosses')
|
|
within = IterOp('within')
|
|
contains = IterOp('contains')
|
|
overlaps = IterOp('overlaps')
|
|
equals = IterOp('equals')
|