Changeset 2449 in lmdz_wrf for trunk/tools/geometry_tools.py

Timestamp:

Apr 19, 2019, 5:06:16 AM (6 years ago)

Author:

lfita

Message:

Adding:

• `p_circle': Function to get a polygon of a circle
• `p_square': Function to get a polygon square
• `read_join_poly': Function to read an ASCII file with the combination of polygons
• `write_join_poly': Function to write an ASCII file with the combination of polygons

File:

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

trunk/tools/geometry_tools.py

@@ -26 +26 @@
 # position_sphere: Function to tranform fom a point in lon, lat deg coordinates to 
 #   cartesian coordinates over an sphere
+# p_circle: Function to get a polygon of a circle
+# p_square: Function to get a polygon square
+# 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_line2D: Function to rotate a line given by 2 pairs of x,y coordinates by a 
@@ -33 +36 @@
 # spheric_line: Function to transform a series of locations in lon, lat coordinates 
 #   to x,y,z over an 3D spaceFunction to provide coordinates of a line  on a 3D space
+# write_join_poly: Function to write an ASCII file with the combination of polygons
 
 ## Shapes/objects
 # ellipse_polar: Function to determine an ellipse from its center and polar coordinates
 # surface_sphere: Function to provide an sphere as matrix of x,y,z coordinates
+
 
 ## Plotting
@@ -384 +389 @@
     return circ_sec
 
+def p_square(face, N=5):
+    """ Function to get a polygon square
+      face: length of the face of the square
+      N: number of points of the polygon
+    """
+    fname = 'p_square'
+
+    square = np.zeros((N,2), dtype=np.float)
+
+    f2 = face/2.
+    N4 = N/4
+    df = face/(N4)
+    # SW-NW
+    for ip in range(N4):
+        square[ip,:] = [-f2+ip*df,-f2]
+    # NW-NE
+    for ip in range(N4):
+        square[ip+N4,:] = [f2,-f2+ip*df]
+    # NE-SE
+    for ip in range(N4):
+        square[ip+2*N4,:] = [f2-ip*df,f2]
+    N42 = N-3*N4-1
+    df = face/(N42)
+    # SE-SW
+    for ip in range(N42):
+        square[ip+3*N4,:] = [-f2,f2-ip*df]
+    square[N-1,:] = [-f2,-f2]
+
+    return square
+
+def p_circle(radii, N=50):
+    """ Function to get a polygon of a circle
+      radii: length of the radii of the circle
+      N: number of points of the polygon
+    """
+    fname = 'p_circle'
+
+    circle = np.zeros((N,2), dtype=np.float)
+
+    dangle = 2.*np.pi/(N-1)
+
+    for ia in range(N):
+        circle[ia,:] = [radii*np.sin(ia*dangle), radii*np.cos(ia*dangle)]
+
+    circle[N-1,:] = [0., radii]
+
+    return circle
+
+# Combined objects
+##
+
 # FROM: http://www.photographers1.com/Sailing/NauticalTerms&Nomenclature.html
 def zsailing_boat(length=10., beam=1., lbeam=0.4, sternbp=0.5):
@@ -434 +490 @@
  
     return boat
+
+def write_join_poly(polys, flname='join_polygons.dat'):
+    """ Function to write an ASCII file with the combination of polygons
+      polys: dictionary with the names of the different polygons
+      flname: name of the ASCII file
+    """
+    fname = 'write_join_poly'
+
+    of = open(flname, 'w')
+
+    for polyn in polys.keys():
+        vertices = polys[polyn]
+        Npts = vertices.shape[0]
+        for ip in range(Npts):
+            of.write(polyn+' '+str(vertices[ip,1]) + ' ' + str(vertices[ip,0]) + '\n')
+
+    of.close()
+
+    return
+
+def read_join_poly(flname='join_polygons.dat'):
+    """ Function to read an ASCII file with the combination of polygons
+      flname: name of the ASCII file
+    """
+    fname = 'read_join_poly'
+
+    of = open(flname, 'r')
+
+    polys = {}
+    polyn = ''
+    poly = []
+    for line in of:
+        if len(line) > 1: 
+            linevals = line.replace('\n','').split(' ')
+            if polyn != linevals[0]:
+                if len(poly) > 1:
+                    polys[polyn] = np.array(poly)
+                polyn = linevals[0]
+                poly = []
+                poly.append([np.float(linevals[2]), np.float(linevals[1])])
+            else:
+                poly.append([np.float(linevals[2]), np.float(linevals[1])])
+
+    of.close()
+    polys[polyn] = np.array(poly)
+
+    return polys
 
 ####### ####### ##### #### ### ## #
@@ -599 +702 @@
 
     return fig, ax
-