Changeset 2527 in lmdz_wrf

Timestamp:

May 11, 2019, 5:46:54 AM (6 years ago)

Author:

lfita

Message:

Adding:

• `band_lighthouse': Function to plot a lighthouse with spiral bands

File:

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

trunk/tools/geometry_tools.py

@@ -53 +53 @@
 # p_spiral: Function to provide a polygon of an Archimedean spiral
 # p_triangle: Function to provide the polygon of a triangle from its 3 vertices
+# band_lighthouse: Function to plot a lighthouse with spiral bands
 # 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
@@ -1196 +1197 @@
     return buoy, buoysecs, buoydic
 
+def band_lighthouse(height=10., width=2., hlight=3., bands=3, N=300):
+    """ Function to plot a lighthouse with spiral bands
+      height: height of the tower (10., default)
+      width: width of the tower (2., default)
+      hlight: height of the light (3., default)
+      bands: number of spiral bands (3, default)
+      N: number of points (300, default)
+    """
+    import numpy.ma as ma
+    fname = 'band_lighthouse'
+
+    lighthouse = np.ones((N,2), dtype=np.float)*gen.fillValueF
+    lighthousesecs = []
+    lighthousedic = {}
+
+    # base Tower
+    Nsec = int(0.30*N/7)
+    p1=np.array([0., width/2.])
+    p2=np.array([0., -width/2.])
+    iip = 0
+    lighthouse[0:Nsec,:] = circ_sec(p1, p2, 3*width, pos='left', Nang=Nsec)
+    iip = iip + Nsec
+
+    # left side
+    ix=-width/2.
+    iy=0.
+    dx = 0.
+    dy = height/(Nsec-1)
+    for ip in range(Nsec):
+        lighthouse[iip+ip,:] = [iy+dy*ip, ix+dx*ip]
+    iip = iip + Nsec
+
+    # Top Tower
+    p1=np.array([height, width/2.])
+    p2=np.array([height, -width/2.])
+    lighthouse[iip:iip+Nsec,:] = circ_sec(p1, p2, 3*width, pos='left', Nang=Nsec)
+    iip = iip + Nsec
+
+    # right side
+    ix=width/2.
+    iy=height
+    dx = 0.
+    dy = -height/(Nsec-1)
+    for ip in range(Nsec):
+        lighthouse[iip+ip,:] = [iy+dy*ip, ix+dx*ip]
+    iip = iip + Nsec + 1
+
+    Ntower = iip-1
+    lighthousesecs = ['tower']
+    lighthousedic['tower'] = [lighthouse[0:iip-1], '-', 'k', 1.5]
+
+    # Left light
+    p1 = np.array([height, -width/2.])
+    p2 = np.array([height+hlight, 0.])
+    lighthouse[iip:iip+Nsec,:] = circ_sec(p1, p2, 3*hlight, Nang=Nsec)
+    iip = iip + Nsec
+    
+    # Right light
+    p1 = np.array([height+hlight, 0.])
+    p2 = np.array([height, width/2.])
+    lighthouse[iip:iip+Nsec,:] = circ_sec(p1, p2, 3*hlight, Nang=Nsec)
+    iip = iip + Nsec
+
+    # Base Light
+    p1=np.array([height, width/2.])
+    p2=np.array([height, -width/2.])
+    lighthouse[iip:iip+Nsec,:] = circ_sec(p1, p2, 3*width, pos='left', Nang=Nsec)
+    iip = iip + Nsec + 1
+    lighthousesecs = ['light']
+    lighthousedic['light'] = [lighthouse[Ntower+1:iip-1], '-', '#AAAA00', 1.5]
+
+    # Spiral bands
+    hb = height/(2.*bands)
+    Nsec2 = (N - Nsec*7 - 3)/bands
+    for ib in range(bands-1):
+        iband = iip
+        Nsec = Nsec2/4
+        bandS = 'band' + str(ib).zfill(2)
+        # hband
+        ix = -width/2.
+        iy = hb*ib*2
+        dx = 0.
+        dy = hb/(Nsec-1)
+        for ip in range(Nsec):
+            lighthouse[iip+ip,:] = [iy+dy*ip, ix+dx*ip]
+        iip = iip + Nsec
+        # uband
+        p1 = np.array([hb*(ib*2+1), -width/2.])
+        p2 = np.array([hb*(ib*2+2), width/2.])
+        lighthouse[iip:iip+Nsec,:] = circ_sec(p1, p2, 3*width, pos='right', Nang=Nsec)
+        iip = iip + Nsec
+        # dband
+        ix = width/2.
+        iy = hb*(ib*2+2)
+        dx = 0.
+        dy = -hb/(Nsec-1)
+        for ip in range(Nsec):
+            lighthouse[iip+ip,:] = [iy+dy*ip, ix+dx*ip]
+        iip = iip + Nsec
+        # dband
+        p1 = np.array([hb*(ib*2+1), width/2.])
+        p2 = np.array([hb*ib*2, -width/2.])
+        lighthouse[iip:iip+Nsec,:] = circ_sec(p1, p2, 3*width, pos='left', Nang=Nsec)
+        iip = iip + Nsec + 1
+        lighthousesecs = [bandS]
+        lighthousedic[bandS] = [lighthouse[iband:iip-1], '-', '#160016', 2.]
+
+    ib = bands-1
+    Nsec3 = (N - iip - 1)
+    Nsec = int(Nsec3/4)
+    bandS = 'band' + str(ib).zfill(2)
+    # hband
+    iband = iip
+    ix = -width/2.
+    iy = hb*ib*2
+    dx = 0.
+    dy = hb/(Nsec-1)
+    for ip in range(Nsec):
+        lighthouse[iip+ip,:] = [iy+dy*ip, ix+dx*ip]
+    iip = iip + Nsec
+    # uband
+    p1 = np.array([hb*(ib*2+1), -width/2.])
+    p2 = np.array([hb*(ib*2+2), width/2.])
+    lighthouse[iip:iip+Nsec,:] = circ_sec(p1, p2, 3*width, pos='right', Nang=Nsec)
+    iip = iip + Nsec
+    # dband
+    ix = width/2.
+    iy = hb*(2+ib*2)
+    dx = 0.
+    dy = -hb/(Nsec-1)
+    for ip in range(Nsec):
+        lighthouse[iip+ip,:] = [iy+dy*ip, ix+dx*ip]
+    iip = iip + Nsec
+    # dband
+    Nsec = N - iip
+    p1 = np.array([hb*(1+ib*2), width/2.])
+    p2 = np.array([hb*ib*2, -width/2.])
+    lighthouse[iip:iip+Nsec,:] = circ_sec(p1, p2, 3*width, pos='left', Nang=Nsec)        
+    lighthousesecs = [bandS]
+    lighthousedic[bandS] = [lighthouse[iband:iip-1], '-', '#160016', 2.]
+
+    lighthouse = ma.masked_equal(lighthouse, gen.fillValueF)
+
+    return lighthouse, lighthousesecs, lighthousedic
+
 ####### ####### ##### #### ### ## #
 # Plotting