Changeset 2533 in lmdz_wrf for trunk

Timestamp:

May 15, 2019, 3:42:57 AM (6 years ago)

Author:

lfita

Message:

Adding:

• `p_angle_triangle': Function to draw a triangle by an initial point and two consecutive angles and the first length of face. The third angle and 2 and 3rd face will be computed accordingly the provided values

File:

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

trunk/tools/geometry_tools.py

@@ -45 +45 @@
 # circ_sec: Function union of point A and B by a section of a circle
 # ellipse_polar: Function to determine an ellipse from its center and polar coordinates
+# p_angle_triangle: Function to draw a triangle by an initial point and two 
+#   consecutive angles and the first length of face. The third angle and 2 and 3rd 
+#   face will be computed accordingly the provided values
 # p_doubleArrow: Function to provide an arrow with double lines
 # p_circle: Function to get a polygon of a circle
@@ -868 +871 @@
 
     return doubleArrow, doublearrowsecs, doublearrowdic 
+
+def p_angle_triangle(pi=np.array([0.,0.]), angle1=60., length1=1., angle2=60., N=100):
+    """ Function to draw a triangle by an initial point and two consecutive angles 
+        and the first length of face. The third angle and 2 and 3rd face will be 
+        computed accordingly the provided values:
+           length1 / sin(angle1) = length2 / sin(angle2) = length3 / sin(angle3)
+           angle1 + angle2 + angle3 = 180.
+      pi: initial point ([0., 0.], default)
+      angle1: first angle from pi clockwise (60., default)
+      length1: length of face from pi by angle1 (1., default)
+      angle2: second angle from second point (60., default)
+      length2: length of face from p2 by angle2 (1., default)
+      N: number of points (100, default)
+    """
+    fname = 'p_angle_triange'
+
+    angle3 = 180. - angle1 - angle2
+    length2 = np.sin(angle2*np.pi/180.)*length1/np.sin(angle1*np.pi/180.)
+    length3 = np.sin(angle3*np.pi/180.)*length1/np.sin(angle1*np.pi/180.)
+
+    print 'lengths', length1, length2, length3
+
+    triangle = np.zeros((N,2), dtype=np.float)
+
+    N3 = int(N/3)
+    # first face
+    ix = pi[1]
+    iy = pi[0]
+    dx = length1*np.cos(angle1*np.pi/180.)/(N3-1)
+    dy = length1*np.sin(angle1*np.pi/180.)/(N3-1)
+    for ip in range(N3):
+        triangle[ip,:] = [iy+dy*ip, ix+dx*ip]
+
+    # second face
+    ia = -90. - (90.-angle1)
+    ix = triangle[N3-1,1]
+    iy = triangle[N3-1,0]
+    dx = length2*np.cos((ia+angle2)*np.pi/180.)/(N3-1)
+    dy = length2*np.sin((ia+angle2)*np.pi/180.)/(N3-1)
+    for ip in range(N3):
+        triangle[N3+ip,:] = [iy+dy*ip, ix+dx*ip]
+
+    # third face
+    N32 = N - 2*N3
+    ia = -180. - (90.-angle2)
+    ix = triangle[2*N3-1,1]
+    iy = triangle[2*N3-1,0]
+    angle3 = np.arctan2(pi[0]-iy, pi[1]-ix)
+    dx = (pi[1]-ix)/(N32-1)
+    dy = (pi[0]-iy)/(N32-1)
+    for ip in range(N32):
+        triangle[2*N3+ip,:] = [iy+dy*ip, ix+dx*ip]
+
+    return triangle
 
 # Combined objects
@@ -1123 +1180 @@
       bfrac: fraction of the ball above the prism (0.8, default)
       N: total number of points of the buoy (200, default)
-
     """
     fname = 'buoy1'
@@ -1350 +1406 @@
     return lighthouse, lighthousesecs, lighthousedic
 
+def north_buoy1(height=5., width=10., bradii=1.75, bfrac=0.8, hsings=0.2, N=300):
+    """ Function to draw a North danger 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)
+      hisngs: height of the sings [as reg. triangle] as percentage of the height 
+        (0.2, default)
+      N: total number of points of the buoy (300, default)
+    """
+    fname = 'north_buoy1'
+
+    Nbuoy = np.ones((N,2), dtype=np.float)*gen.fillValueF
+
+    # buoy
+    N2 = int(N/2)
+    buoy1v = buoy1(height=5., width=10., bradii=1.75, bfrac=0.8, hsings=0.2, N=N2)
+    Nbuoy[0:N2] = buoy1v
+
+    # signs
+    N3 = N - N2 - 1
+
+    lsign = height*hsigns
+    # First up
+    N32 = N3 / 2
+    N323 = N32 / 3
+
+    # left up arm
+    ix = -lsign/2.
+    iy = height
+    dx = -lsign/(2.*(N323-1))
+    dy = lsign*np.sin(np.pi/3.)/(N323-1)
+    for ip in range(N323):
+        Nbuoy[N2+1:N2+N323+1] = [iy + dy*ip, ix + dx*ip]
+
+    Nbuoy = ma.masked_equal(Nbuoy, gen.fillValueF)
+
+    return Nbuoy, Nbuoysecs, Nbuoydic
+
 ####### ####### ##### #### ### ## #
 # Plotting