Added remove_small_polygons operation for processor

bjornd · Jul 23, 2015 · 71761cf · 71761cf
1 parent 6dc3908
commit 71761cf
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/converter/processor.py b/converter/processor.py
@@ -551,6 +551,16 @@ def intersect_rect(self, config, data_source):
     for geometry in data_source.geometries:
       geometry.geom = geometry.geom.intersection(rect)
 
+  def remove_small_polygons(self, config, data_source):
+    for geometry in data_source.geometries:
+      if isinstance(geometry.geom, shapely.geometry.multipolygon.MultiPolygon):
+        polygons = geometry.geom.geoms
+      else:
+        polygons = [geometry.geom]
+      polygons = filter(lambda p: p.area > config['minimal_area'], polygons)
+      if len(polygons) > 0:
+        geometry.geom = shapely.geometry.multipolygon.MultiPolygon(polygons)
+
 
 args = {}
 if len(sys.argv) > 1: