Changeset 2452 in lmdz_wrf

Timestamp:

Apr 21, 2019, 9:55:18 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `mirror_polygon': Function to reflex a polygon for a given axis

File:

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

trunk/tools/geometry_tools.py

@@ -23 +23 @@
 # dist_points: Function to provide the distance between two points
 # max_coords_poly: Function to provide the extremes of the coordinates of a polygon
-# rotate_polygon_2D: Function to rotate 2D plain the vertices of a polygon
+# mirror_polygon: Function to reflex a polygon for a given axis
 # position_sphere: Function to tranform fom a point in lon, lat deg coordinates to 
 #   cartesian coordinates over an sphere
 # read_join_poly: Function to read an ASCII file with the combination of polygons
 # 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
 # rotate_line2D: Function to rotate a line given by 2 pairs of x,y coordinates by a 
 #   certain angle in the plain
@@ -257 +258 @@
 
     return [nx, xx], [ny, xy], [ayx, axx], xyx
+
+def mirror_polygon(polygon,axis):
+    """ Function to reflex a polygon for a given axis
+      polygon: polygon to mirror
+      axis: axis at which mirror is located ('x' or 'y')
+    """
+    fname = 'mirror_polygon'
+
+    reflex = np.zeros(polygon.shape, dtype=np.float)
+
+    N = polygon.shape[0]
+    if axis == 'x':
+        for iv in range(N):
+            reflex[iv,:] = [-polygon[iv,1], polygon[iv,0]]
+    elif axis == 'y':
+        for iv in range(N):
+            reflex[iv,:] = [polygon[iv,1], -polygon[iv,0]]
+
+    return reflex
 
 ####### ###### ##### #### ### ## #
@@ -484 +504 @@
 
     dangle = 2.*np.pi*loops/(N-1)
-    dr = eradii/(N-1)
+    dr = eradii*1./(N-1)
 
     for ia in range(N):