Changeset 2567 in lmdz_wrf

Timestamp:

May 29, 2019, 10:31:52 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `p_prism': Function to get a polygon prism

File:

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

trunk/tools/geometry_tools.py

@@ -54 +54 @@
 # p_doubleArrow: Function to provide an arrow with double lines
 # p_circle: Function to get a polygon of a circle
+# p_prism: Function to get a polygon prism
 # p_reg_polygon: Function to provide a regular polygon of Nv vertices
 # p_reg_star: Function to provide a regular star of Nv vertices
@@ -1062 +1063 @@
 
     return square
+
+
+def p_prism(base, height, N=5):
+    """ Function to get a polygon prism
+      base: length of the base of the prism
+      height: length of the height of the prism
+      N: number of points of the polygon
+    """
+    fname = 'p_prism'
+
+    prism = np.zeros((N,2), dtype=np.float)
+
+    b2 = base/2.
+    h2 = height/2.
+    N4 = N/4
+    dh = height/(N4)
+    db = base/(N4)
+
+    # SW-NW
+    for ip in range(N4):
+        prism[ip,:] = [-h2+ip*dh,-b2]
+    # NW-NE
+    for ip in range(N4):
+        prism[ip+N4,:] = [h2,-b2+ip*db]
+    # NE-SE
+    for ip in range(N4):
+        prism[ip+2*N4,:] = [h2-ip*dh,b2]
+    N42 = N-3*N4-1
+    db = base/(N42)
+    # SE-SW
+    for ip in range(N42):
+        prism[ip+3*N4,:] = [-h2,b2-ip*db]
+    prism[N-1,:] = [-h2,-b2]
+
+    return prism
 
 def p_circle(radii, N=50):
@@ -2155 +2191 @@
     return buoy, buoysecs, buoydic
 
+
+def green_buoy1(height=5., width=10., bradii=1.75, bfrac=0.8, hsigns=0.3, N=300):
+    """ Function to draw a green mark buoy using buoy1
+      height: height of the prism (5., default) 
+      width: width of the prism (10., default)
+      bradii: radii of the ball (1.75, default)
+      bfrac: fraction of the ball above the prism (0.8, default)
+      hisgns: height of the signs [as reg. triangle] as percentage of the height 
+        (0.3, default)
+      N: total number of points of the buoy (300, default)
+    """
+    fname = 'green_buoy1'
+
+    buoy = np.ones((N,2), dtype=np.float)*gen.fillValueF
+
+    # buoy
+    N2 = int(N/2)
+    buoy1v, buoy1vsecs, buoy1vdic = buoy1(height=5., width=10., bradii=1.75,         \
+      bfrac=0.8, N=N2)
+    buoy[0:N2,:] = buoy1v
+
+    # signs
+    N3 = N - N2 - 1
+    lsign = height*hsigns*2.
+    
+    Height = np.max(buoy1v[:,0])
+    triu = p_angle_triangle(N=N3)
+    sign = triu*lsign
+    buoy[N2+1:N2+2+N3,:] = sign + [Height+0.2*lsign,-lsign/2.]
+
+    # painting it
+    buoy = ma.masked_equal(buoy, gen.fillValueF)
+
+    buoysecs = ['buoy', 'sign', 'quarter1', 'quarter2', 'quarter3', 'quarter4']
+    buoydic = {'buoy': [buoy[0:N2,:],'-','r',1.5],                                   \
+      'sign': [buoy[N2+1:N2+N3+1,:],'-','r',1.5]}
+
+    return buoy, buoysecs, buoydic
+
 ####### ####### ##### #### ### ## #
 # Plotting