add Vsv and Vsh to DataFrame

marcoalopez · Apr 24, 2024 · 99dd8a9 · 99dd8a9
1 parent 7fec979
commit 99dd8a9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/src/orthotropic_models.py b/src/orthotropic_models.py
@@ -68,7 +68,7 @@ def Tsvankin_params(cij: np.ndarray, density: float):
     epsilon2 = (c11 - c33) / (2 * c33)
     delta2 = ((c13 + c55)**2 - (c33 - c55)**2) / (2*c33 * (c33 - c55))
     gamma2 = (c66 - c44) / (2 * c44)
-    # VTI parameter in the XZ plane
+    # VTI parameter in the XY plane
     delta3 = (c12 + c66)**2 - (c11 - c66)**2 / (2*c11 * (c11 - c66))
 
     return Vp0, Vs0, epsilon1, delta1, gamma1, epsilon2, delta2, gamma2, delta3

diff --git a/src/polar_models.py b/src/polar_models.py
@@ -84,8 +84,9 @@ def weak_polar_anisotropy(elastic, wavevectors):
     -------
     pandas.DataFrame
         Tabular data object containing the propagation directions
-        and calculated Vp, Vs1, and Vs2 speeds and polarization
-        anisotropy using the weak polar anisotropy model.
+        and calculated Vp, Vsv, Vsh, Vs1, and Vs2 speeds and
+        polarization anisotropy using the weak polar anisotropy
+        model.
     """
 
     # extract azimuths and polar angles
@@ -111,6 +112,8 @@ def weak_polar_anisotropy(elastic, wavevectors):
     # reshape and store arrays
     data = {'polar_ang': polar,
             'azimuthal_ang': azimuths,
+            'Vsv': Vsv,
+            'Vsh': Vsh,
             'Vp': Vp,
             'Vs1': Vs1,
             'Vs2': Vs2,
@@ -136,8 +139,9 @@ def polar_anisotropy(elastic, wavevectors):
     -------
     pandas.DataFrame
         Tabular data object containing the propagation directions
-        and calculated Vp, Vs1, and Vs2 speeds and polarization
-        anisotropy using the Andreson polar anisotropy model.
+        and calculated Vp, Vsv, Vsh, Vs1, and Vs2 speeds and
+        polarization anisotropy using the weak polar anisotropy
+        model.
     """
 
     # extract azimuths and polar angles
@@ -180,6 +184,8 @@ def polar_anisotropy(elastic, wavevectors):
     # reshape and store arrays
     data = {'polar_ang': polar,
             'azimuthal_ang': azimuths,
+            'Vsv': Vsv,
+            'Vsh': Vsh,
             'Vp': Vp,
             'Vs1': Vs1,
             'Vs2': Vs2,