Changeset 2494 in lmdz_wrf

Timestamp:

May 1, 2019, 11:46:47 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `p_doubleArrow': Function to provide an arrow with double lines

File:

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

trunk/tools/geometry_tools.py

@@ -40 +40 @@
 ## Shapes/objects
 # circ_sec: Function union of point A and B by a section of a circle
+# p_doubleArrow: Function to provide an arrow with double lines
 # ellipse_polar: Function to determine an ellipse from its center and polar coordinates
 # p_circle: Function to get a polygon of a circle
@@ -653 +654 @@
         quit(-1)
 
-
-    return sinusoide
+    sinusoidesecs = ['sinusoide']
+    sinusoidedic = {'sinusoide': [sinusoide, '-', '#000000', 1.]}
+
+    return sinusoide, sinusoidesecs, sinusoidedic
+
+def p_doubleArrow(length=5., angle=45., width=1., alength=0.10, N=50):
+    """ Function to provide an arrow with double lines
+      length: length of the arrow (5. default)
+      angle: angle of the head of the arrow (45., default)
+      width: separation between the two lines (2., default)
+      alength: length of the head (as percentage in excess of width, 0.1 default)
+      N: number of points (50, default)
+    """
+    import numpy.ma as ma
+    function = 'p_doubleArrow'
+
+    doubleArrow = np.zeros((50,2), dtype=np.float)
+    N4 = int((N-3)/4)
+
+    doublearrowdic = {}
+
+    # Arms
+    dx = length/(N4-1)
+    for ix in range(N4-1):
+        doubleArrow[ix,:] = [dx*ix,-width/2.]
+    doublearrowdic['leftarm'] = [doubleArrow[0:N4-1,:], '-', '#000000', 2.]
+    doubleArrow[N4-1,:] = [gen.fillValueF,gen.fillValueF]
+    for ix in range(N4-1):
+        doubleArrow[N4+ix,:] = [dx*ix,width/2.]
+    doublearrowdic['rightarm'] = [doubleArrow[N4:2*N4-1,:], '-', '#000000', 2.]
+    doubleArrow[2*N4-1,:] = [gen.fillValueF,gen.fillValueF]
+
+    # Head
+    N42 = int((N-2 - 2*N4)/2)
+    dx = width*(1.+alength)/(N42-1)
+    for ix in range(N42):
+        doubleArrow[2*N4+ix,:] = [length-dx*ix,-dx*ix]
+    doublearrowdic['lefthead'] = [doubleArrow[2*N4:2*N4+N42,:], '-', '#000000', 2.]
+    doubleArrow[2*N4+N42,:] = [gen.fillValueF,gen.fillValueF]
+
+    N43 = N-2 - 2*N4 - N42 + 1
+    dx = width*(1.+alength)/(N43-1)
+    for ix in range(N43):
+        doubleArrow[2*N4+N42+1+ix,:] = [length-dx*ix,dx*ix]
+    doublearrowdic['rightthead'] = [doubleArrow[2*N4+N42:51,:], '-', '#000000', 2.]
+
+    doubleArrow = ma.masked_equal(doubleArrow, gen.fillValueF)
+    doublearrowsecs = ['leftarm', 'rightarm', 'lefthead', 'righthead']
+
+    return doubleArrow, doublearrowsecs, doublearrowdic 
 
 # Combined objects