Changeset 2655 for inundation/utilities/polygon.py

Timestamp:

Apr 4, 2006, 3:49:42 PM (18 years ago)

Author:

duncan

Message:

Adding ability to remove points outside of the mesh when interpolating

File:

• inundation/utilities/polygon.py (modified) (2 diffs)

inundation/utilities/polygon.py

@@ -127 +127 @@
                                                 verbose=verbose)
 
+    
     if one_point:
         return count != 1
@@ -135 +136 @@
         else:
             return indices[count:][::-1]  #return reversed
+       
+
+def in_and_outside_polygon(points, polygon, closed = True, verbose = False):
+    """Determine points inside and outside a polygon
+
+       See separate_points_by_polygon for documentation
+
+       Returns an array of points inside and an array of points outside the polygon
+    """
+
+    if verbose: print 'Checking input to outside_polygon'
+    try:
+        points = ensure_numeric(points, Float)
+    except NameError, e:
+        raise NameError, e
+    except:
+        msg = 'Points could not be converted to Numeric array'
+        raise msg
+
+    try:
+        polygon = ensure_numeric(polygon, Float)
+    except NameError, e:
+        raise NameError, e
+    except:
+        msg = 'Polygon could not be converted to Numeric array'
+        raise msg
+
+
+
+    if len(points.shape) == 1:
+        one_point = True
+        points = reshape(points, (1,2))
+    else:
+        one_point = False
+
+    indices, count = separate_points_by_polygon(points, polygon,
+                                                closed=closed,
+                                                verbose=verbose)
+    # Returns an array of points inside and an array of points outside
+    # the polygon
+  
+    if count == len(indices):
+        # No points are outside
+        return indices[:count],[]
+    else:
+        return  indices[:count], indices[count:][::-1]  #return reversed