Changeset 2609 in lmdz_wrf

Timestamp:

Jun 17, 2019, 11:42:22 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `add_secpolygon_list': Function to add a range of points of a polygon into a list (previously add, but uncommented)
• `rm_consecpt_polygon': Function to remove consecutive same point of a polygon

File:

• trunk/tools/geometry_tools.py (modified) (4 diffs)

trunk/tools/geometry_tools.py

@@ -23 +23 @@
 
 ####### Contents:
+# add_secpolygon_list: Function to add a range of points of a polygon into a list
 # cut_between_[x/y]polygon: Function to cut a polygon between 2 given value of the [x/y]-axis
 # cut_[x/y]polygon: Function to cut a polygon from a given value of the [x/y]-axis
@@ -35 +36 @@
 #   cartesian coordinates over an sphere
 # read_join_poly: Function to read an ASCII file with the combination of polygons
+# rm_consecpt_polygon: Function to remove consecutive same point of a polygon
 # rotate_2D: Function to rotate a vector by a certain angle in the plain
 # rotate_polygon_2D: Function to rotate 2D plain the vertices of a polygon
@@ -483 +485 @@
 
     return
+
+def rm_consecpt_polygon(polygon):
+    """ Function to remove consecutive same point of a polygon
+      poly: polygon
+    >>> poly = np.ones((5,2), dtype=np.float)
+    >>> poly[2,:] = [2., 1.]
+    rm_consecpt_polygon(poly)
+    [[ 1.  1.]
+     [ 2.  1.]
+     [ 1.  1.]]
+    """
+    fname = 'rm_consecpt_polygon'
+
+    newpolygon = []
+    prevpt = polygon[0,:]
+    newpolygon.append(prevpt)
+    for ip in range(1,polygon.shape[0]-1):
+        if polygon[ip,0] != prevpt[0] or polygon[ip,1] != prevpt[1]:
+            prevpt = polygon[ip,:]
+            newpolygon.append(prevpt)
+
+    newpolygon = np.array(newpolygon)
+
+    return newpolygon
 
 def cut_ypolygon(polygon, yval, keep='below', Nadd=20):
@@ -961 +987 @@
                 rmpolygon.append(cutpolygon[ip,:])
     Npts = len(rmpolygon)
-    cutpolygon = np.array(rmpolygon)
+
+    rmpolygon = np.array(rmpolygon)
+    cutpolygon = rm_consecpt_polygon(rmpolygon)
 
     cutpolygon = ma.masked_equal(cutpolygon, gen.fillValueF)