Changeset 2492 in lmdz_wrf

Timestamp:

May 1, 2019, 5:05:24 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `p_sinusiode': Function to get coordinates of a sinusoidal curve

File:

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

trunk/tools/geometry_tools.py

@@ -44 +44 @@
 # p_reg_polygon: Function to provide a regular polygon of Nv vertices
 # p_reg_star: Function to provide a regular star of Nv vertices
+# p_sinusiode: Function to get coordinates of a sinusoidal curve
 # p_square: Function to get a polygon square
 # p_spiral: Function to provide a polygon of an Archimedean spiral
@@ -620 +621 @@
 
     return reg_star
+
+def p_sinusiode(length=10., amp=5., lamb=3., ival=0., func='sin', N=100):
+    """ Function to get coordinates of a sinusoidal curve
+      length: length of the line (default 10.)
+      amp: amplitude of the peaks (default 5.)
+      lamb: wave longitude (defalult 3.)
+      ival: initial angle (default 0. in degree)
+      func: function to use: (default sinus)
+        'sin': sinus
+        'cos': cosinus
+      N: number of points (default 100)
+    """
+    fname = 'p_sinusiode'
+    availfunc = ['sin', 'cos']
+
+    dx = length/(N-1)
+    ia = ival*np.pi/180.
+    da = 2*np.pi*dx/3.
+
+    sinusoide = np.zeros((N,2), dtype=np.float)
+    if func == 'sin':
+        for ix in range(N):
+            sinusoide[ix,:] = [amp*np.sin(ia+da*ix),dx*ix]
+    elif func == 'cos':
+        for ix in range(N):
+            sinusoide[ix,:] = [amp*np.cos(ia+da*ix),dx*ix]
+    else:
+        print errormsg
+        print '  ' + fname + ": function '" + func + "' not ready !!"
+        print '    available ones:', availfunc
+        quit(-1)
+
+
+    return sinusoide
 
 # Combined objects