Changeset 2413 in lmdz_wrf for trunk

Timestamp:

Mar 24, 2019, 6:55:57 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `dist_points': Function to provide the distance between two points

File:

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

trunk/tools/geometry_tools.py

@@ -15 +15 @@
 import matplotlib.pyplot as plt
 
+errormsg = 'ERROR -- error -- ERROR -- error'
+
 ####### Contents:
 # deg_deci: Function to pass from degrees [deg, minute, sec] to decimal angles [rad]
+# dist_points: Function to provide the distance between two points
 # multi_rotate_2D: Function to rotate multiple vectors by a certain angle in the plain
 # position_sphere: Function to tranform fom a point in lon, lat deg coordinates to 
@@ -25 +28 @@
 # rotate_lines2D: Function to rotate multiple lines given by mulitple pars of x,y 
 #   coordinates by a certain angle in the plain
-# surface_sphere: Function to provide an sphere as matrix of x,y,z coordinates
 # spheric_line: Function to transform a series of locations in lon, lat coordinates 
 #   to x,y,z over an 3D spaceFunction to provide coordinates of a line  on a 3D space
@@ -31 +33 @@
 ## Shapes/objects
 # ellipse_polar: Function to determine an ellipse from its center and polar coordinates
+# surface_sphere: Function to provide an sphere as matrix of x,y,z coordinates
 
 ## Plotting
@@ -67 +70 @@
 
     return xpt, ypt, zpt
-
-def surface_sphere(radii,Npts):
-    """ Function to provide an sphere as matrix of x,y,z coordinates
-      radii: radii of the sphere
-      Npts: number of points to discretisize longitues (half for latitudes)
-    """
-    fname = 'surface_sphere'
-
-    sphereup = np.zeros((3,Npts/2,Npts), dtype=np.float)
-    spheredown = np.zeros((3,Npts/2,Npts), dtype=np.float)
-    for ia in range(Npts):
-        alpha = ia*2*np.pi/(Npts-1)
-        for ib in range(Npts/2):
-            beta = ib*np.pi/(2.*(Npts/2-1))
-            sphereup[:,ib,ia] = position_sphere(radii, alpha, beta)
-        for ib in range(Npts/2):
-            beta = -ib*np.pi/(2.*(Npts/2-1))
-            spheredown[:,ib,ia] = position_sphere(radii, alpha, beta)
-
-    return sphereup, spheredown
 
 def spheric_line(radii,lon,lat):
@@ -225 +208 @@
 # Shapes/objects
 
+def surface_sphere(radii,Npts):
+    """ Function to provide an sphere as matrix of x,y,z coordinates
+      radii: radii of the sphere
+      Npts: number of points to discretisize longitues (half for latitudes)
+    """
+    fname = 'surface_sphere'
+
+    sphereup = np.zeros((3,Npts/2,Npts), dtype=np.float)
+    spheredown = np.zeros((3,Npts/2,Npts), dtype=np.float)
+    for ia in range(Npts):
+        alpha = ia*2*np.pi/(Npts-1)
+        for ib in range(Npts/2):
+            beta = ib*np.pi/(2.*(Npts/2-1))
+            sphereup[:,ib,ia] = position_sphere(radii, alpha, beta)
+        for ib in range(Npts/2):
+            beta = -ib*np.pi/(2.*(Npts/2-1))
+            spheredown[:,ib,ia] = position_sphere(radii, alpha, beta)
+
+    return sphereup, spheredown
+
 def ellipse_polar(c, a, b, Nang=100):
     """ Function to determine an ellipse from its center and polar coordinates
@@ -249 +252 @@
     return ellipse
 
+def hyperbola_polar(a, b, Nang=100):
+    """ Fcuntion to determine an hyperbola in polar coordinates
+        FROM: https://en.wikipedia.org/wiki/Hyperbola#Polar_coordinates
+          x^2/a^2 - y^2/b^2 = 1
+      a= x-parameter
+      y= y-parameter
+      Nang= number of angles to use
+      DOES NOT WORK!!!!
+    """
+    fname = 'hyperbola_polar'
+
+    dtheta = 2.*np.pi/(Nang-1)
+
+    # Positive branch
+    hyperbola_p = np.zeros((Nang,2), dtype=np.float)
+    for ia in range(Nang):
+        theta = dtheta*ia
+        x = a*np.cosh(theta)
+        y = b*np.sinh(theta)
+        hyperbola_p[ia,:] = [y,x]
+
+    # Negative branch
+    hyperbola_n = np.zeros((Nang,2), dtype=np.float)
+    for ia in range(Nang):
+        theta = dtheta*ia
+        x = -a*np.cosh(theta)
+        y = b*np.sinh(theta)
+        hyperbola_n[ia,:] = [y,x]
+
+    return hyperbola_p, hyperbola_n
+
+def dist_points(ptA, ptB):
+    """ Function to provide the distance between two points
+      ptA: coordinates of the point A [yA, xA]
+      ptB: coordinates of the point B [yB, xB]
+    >>> dist_points([1.,1.], [-1.,-1.])
+    2.82842712475
+    """
+    fname = 'dist_points'
+
+    dist = np.sqrt( (ptA[0]-ptB[0])**2 + (ptA[1]-ptB[1])**2)
+
+    return dist
+
+def circ_sec(ptA, ptB, radii, Nang=100):
+    """ Function union of point A and B by a section of a circle
+      ptA= coordinates od the point A [yA, xA]
+      ptB= coordinates od the point B [yB, xB]
+      radii= radi of the circle to use to unite the points
+      Nang= amount of angles to use
+    """
+    fname = 'circ_sec'
+
+    distAB = dist_points(ptA,ptB)
+
+    if distAB > radii:
+        print errormsg
+        print '  ' + fname + ': radii=', radii, " too small for the distance " +     \
+          "between points !!"
+        print '    distance between points:', distAB
+        quit(-1)
+
+    if ptA[0] > ptB[0]:
+        tottheta = np.atan2(ptA[0]-ptB[0],ptA[1]-ptB[1])
+        yc=ptA[0]
+        if ptA[1] < ptB[0]: 
+            sign = -1
+            xc=ptA[1]+radii
+        else: 
+            sign = 1
+            xc=ptA[1]-radii
+    else:
+        tottheta = np.atan2(ptB[0]-ptA[0],ptB[1]-ptA[1])
+        yc=ptA[0]
+        if ptA[1] < ptB[0]: 
+            sign = 1
+            xc=ptA[1]+radii
+        else: 
+            sign = -1
+            xc=ptA[1]-radii
+
+    dtheta = np.abs(tottheta)/(Nang-1)
+    if sign == 1:
+        dtheta = dtheta*(-1.)
+
+    circ_sec = np.zeros((Nang,2), dtype=np.float)
+    for ia in range(Nang):
+        theta = dtheta*ia
+        x = radii*np.cos(theta)
+        y = radii*np.sin(theta)
+
+        circ_sec[ia,:] = [x+xc,y+yc]
+    
+    return circ_sec
+
+# FROM: http://www.photographers1.com/Sailing/NauticalTerms&Nomenclature.html
+def zsailing_boat(length=10., beam=3., sternbp=0.5):
+    """ Function to define an schematic boat from the z-plane
+      length: length of the boat
+      beam: beam of the boat
+      sternbp: beam at stern as percentage of beam
+    """
+    fname = 'zsailing_boat'
+
+    
+
+    return boat
 
 ####### ####### ##### #### ### ## #