Changeset 2565 in lmdz_wrf

Timestamp:

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

Author:

lfita

Message:

Adding:

• `safewater_buoy1': Function to draw a safe water 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
+# 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
 # z_boat: Function to define an schematic boat from the z-plane
@@ -2070 +2071 @@
     return buoy, buoysecs, buoydic
 
+def safewater_buoy1(height=5., width=10., bradii=1.75, bfrac=0.8, hsigns=0.3, N=300):
+    """ Function to draw a safe water 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 = 'safewater_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
+    
+    Height = np.max(buoy1v[:,0])
+    sign = p_circle(lsign, N3)
+    buoy[N2+1:N2+2+N3,:] = sign + [Height+1.2*lsign,0.]
+
+    # painting it
+    ix = -width/2.
+    Ncut, quarter1 = cut_xpolygon(buoy1v, xval=ix+width/4., keep='left')
+    Ncut, quarter2 = cut_between_xpolygon(buoy1v, xval1=ix+width/4., xval2=ix+width/2.)
+    Ncut, quarter3 = cut_between_xpolygon(buoy1v, xval1=ix+width/2., xval2=ix+3.*width/4.)
+    Ncut, quarter4 = cut_xpolygon(buoy1v, xval=ix+3.*width/4., keep='right')
+
+    buoy = ma.masked_equal(buoy, gen.fillValueF)
+
+    buoysecs = ['buoy', 'sign', 'quarter1', 'quarter2', 'quarter3', 'quarter4']
+    buoydic = {'buoy': [buoy[0:N2,:],'-','k',1.5],                                   \
+      'sign': [buoy[N2+1:N2+N3+1,:],'-','r',1.5], 'quarter1': [quarter1,'-','r',1.], \
+      'quarter2': [quarter2,'-','#FFFFFF',1.], 'quarter3': [quarter3,'-','r',1.],    \
+      'quarter4': [quarter4,'-','#FFFFFF',1.]}
+
+    return buoy, buoysecs, buoydic
+
 ####### ####### ##### #### ### ## #
 # Plotting