Changeset 2566 in lmdz_wrf

Timestamp:

May 29, 2019, 10:24:48 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `red_buoy1': Function to draw a red mark buoy using buoy1

File:

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

trunk/tools/geometry_tools.py

@@ -61 +61 @@
 # p_triangle: Function to provide the polygon of a triangle from its 3 vertices
 # band_lighthouse: Function to plot a lighthouse with spiral bands
+# red_buoy1: Function to draw a red mark buoy using buoy1
 # safewater_buoy1: Function to draw a safe water mark buoy using buoy1
 # surface_sphere: Function to provide an sphere as matrix of x,y,z coordinates
@@ -2116 +2117 @@
     return buoy, buoysecs, buoydic
 
+def red_buoy1(height=5., width=10., bradii=1.75, bfrac=0.8, hsigns=0.3, N=300):
+    """ Function to draw a red 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 = 'red_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