Changeset 2572 in lmdz_wrf

Timestamp:

May 31, 2019, 5:18:18 AM (5 years ago)

Author:

lfita

Message:

Adding:

• `p_cross_width': Function to draw a cross with arms with a given width and an angle

File:

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

trunk/tools/geometry_tools.py

@@ -57 +57 @@
 # p_doubleArrow: Function to provide an arrow with double lines
 # p_circle: Function to get a polygon of a circle
+# p_cross_width: Function to draw a cross with arms with a given width and an angle
 # p_prism: Function to get a polygon prism
 # p_reg_polygon: Function to provide a regular polygon of Nv vertices
@@ -1414 +1415 @@
 
     return triangle
+
+def p_cross_width(larm=5., width=1., Narms=4, N=200):
+    """ Function to draw a cross with arms with a given width and an angle
+      larm: legnth of the arms (5., default)
+      width: width of the arms (1., default)
+      Narms: Number of arms (4, default)
+      N: number of points to us (200, default)
+    """
+    fname = 'p_cross_width'
+
+    Narm = int((N-Narms)/Narms)
+
+    larm2 = larm/2.
+    width2 = width/2.
+
+    cross = np.ones((N,2), dtype=np.float)*gen.fillValueF
+    da = np.pi/Narms
+
+    N1 = int(Narm*3./8.)
+    N2 = int((Narm - 2*N1)/2.)
+    N21 = Narm - 2*N1 - N2
+
+    if N2 < 3:
+        print errormsg
+        print '  ' + fname + ": too few points for ", Narms, " arms !!"
+        print "    increase number 'N' at least up to '", 25*Narms
+        quit(-1)
+
+    crosssecs = []
+    crossdic = {}
+    Npot = int(np.log10(Narms))+1
+
+    iip = 0
+    for iarm in range(Narms-1):
+
+        a = da*iarm
+        iip0 = iip
+
+        # bottom coordinate
+        bx = larm*np.cos(a+np.pi)
+        by = larm*np.sin(a+np.pi)
+
+        # upper coordinate
+        ux = larm*np.cos(a)
+        uy = larm*np.sin(a)
+
+        rela = a+np.pi*3./2.
+        # SW-NW
+        ix = bx + width2*np.cos(rela)
+        iy = by + width2*np.sin(rela)
+        ex = ux + width2*np.cos(rela)
+        ey = uy + width2*np.sin(rela)
+        dx = (ex-ix)/(N1-1)
+        dy = (ey-iy)/(N1-1)
+        for ip in range(N1):
+            cross[iip+ip,:] = [iy+dy*ip,ix+dx*ip]
+        iip = iip + N1
+
+        # NW-NE
+        ix = ex + 0.
+        iy = ey + 0.
+        ex = ux - width2*np.cos(rela)
+        ey = uy - width2*np.sin(rela)
+        dx = (ex-ix)/(N2-1)
+        dy = (ey-iy)/(N2-1)
+        for ip in range(N2):
+            cross[iip+ip,:] = [iy+dy*ip,ix+dx*ip]
+        iip = iip + N2
+
+        # NW-SW
+        ix = ex + 0.
+        iy = ey + 0.
+        ex = bx - width2*np.cos(rela)
+        ey = by - width2*np.sin(rela)
+        dx = (ex-ix)/(N1-1)
+        dy = (ey-iy)/(N1-1)
+        for ip in range(N1):
+            cross[iip+ip,:] = [iy+dy*ip,ix+dx*ip]
+        iip = iip + N1
+
+        # SW-SE
+        ix = ex + 0.
+        iy = ey + 0.
+        ex = bx + width2*np.cos(rela)
+        ey = by + width2*np.sin(rela)
+        dx = (ex-ix)/(N21-1)
+        dy = (ey-iy)/(N21-1)
+        for ip in range(N21):
+            cross[iip+ip,:] = [iy+dy*ip,ix+dx*ip]
+        iip = iip + N21 + 1
+
+        iarmS = str(iarm).zfill(Npot)
+        crosssecs.append(iarmS)
+        crossdic[iarmS] = [cross[iip0:iip0+iip-1], '-', 'k', '1.']
+
+    iip0 = iip
+
+    Narm = N - Narm*(Narms-1) - Narms
+
+    N1 = int(Narm*3./8.)
+    N2 = int((Narm - 2*N1)/2.)
+    N21 = Narm - 2*N1 - N2
+
+    iarm = Narms-1
+    a = da*iarm
+
+    # bottom coordinate
+    bx = larm*np.cos(a+np.pi)
+    by = larm*np.sin(a+np.pi)
+
+    # upper coordinate
+    ux = larm*np.cos(a)
+    uy = larm*np.sin(a)
+
+    rela = a+np.pi*3./2.
+    # SW-NW
+    ix = bx + width2*np.cos(rela)
+    iy = by + width2*np.sin(rela)
+    ex = ux + width2*np.cos(rela)
+    ey = uy + width2*np.sin(rela)
+    dx = (ex-ix)/(N1-1)
+    dy = (ey-iy)/(N1-1)
+    for ip in range(N1):
+      cross[iip+ip,:] = [iy+dy*ip,ix+dx*ip]
+    iip = iip + N1
+
+    # NW-NE
+    ix = ex + 0.
+    iy = ey + 0.
+    ex = ux - width2*np.cos(rela)
+    ey = uy - width2*np.sin(rela)
+    dx = (ex-ix)/(N2-1)
+    dy = (ey-iy)/(N2-1)
+    for ip in range(N2):
+      cross[iip+ip,:] = [iy+dy*ip,ix+dx*ip]
+    iip = iip + N2
+
+    # NW-SW
+    ix = ex + 0.
+    iy = ey + 0.
+    ex = bx - width2*np.cos(rela)
+    ey = by - width2*np.sin(rela)
+    dx = (ex-ix)/(N1-1)
+    dy = (ey-iy)/(N1-1)
+    for ip in range(N1):
+      cross[iip+ip,:] = [iy+dy*ip,ix+dx*ip]
+    iip = iip + N1
+
+    # SW-SE
+    ix = ex + 0.
+    iy = ey + 0.
+    ex = bx + width2*np.cos(rela)
+    ey = by + width2*np.sin(rela)
+    dx = (ex-ix)/(N21-1)
+    dy = (ey-iy)/(N21-1)
+    for ip in range(N21):
+      cross[iip+ip,:] = [iy+dy*ip,ix+dx*ip]
+    iip = iip + N21
+
+    iarmS = str(iarm).zfill(Npot)
+    crosssecs.append(iarmS)
+    crossdic[iarmS] = [cross[iip0:iip0+iip-1], '-', 'k', '1.']
+
+    cross = ma.masked_equal(cross, gen.fillValueF)
+
+    return cross, crosssecs, crossdic
 
 # Combined objects