voly 0.0.62__tar.gz → 0.0.63__tar.gz

{voly-0.0.62/src/voly.egg-info → voly-0.0.63}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: voly
-Version: 0.0.62
+Version: 0.0.63
 Summary: Options & volatility research package
 Author-email: Manu de Cara <manu.de.cara@gmail.com>
 License: MIT

{voly-0.0.62 → voly-0.0.63}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "voly"
-version = "0.0.62"
+version = "0.0.63"
 description = "Options & volatility research package"
 readme = "README.md"
 authors = [
@@ -60,7 +60,7 @@ line_length = 100
 multi_line_output = 3
 
 [tool.mypy]
-python_version = "0.0.62"
+python_version = "0.0.63"
 warn_return_any = true
 warn_unused_configs = true
 disallow_untyped_defs = true

{voly-0.0.62 → voly-0.0.63}/src/voly/client.py

@@ -352,27 +352,29 @@ class VolyClient:
 
     @staticmethod
     def get_iv_surface(fit_results: Dict[str, Any],
-                       moneyness_params: Tuple[float, float, int] = (-2, 2, 500)
+                       log_moneyness_params: Tuple[float, float, int] = (-2, 2, 500)
                        ) -> Dict[str, Any]:
         """
         Generate implied volatility surface using optimized SVI parameters.
 
         Parameters:
-        - param_matrix: Matrix of optimized SVI parameters from fit_results
-        - moneyness_params: Tuple of (min, max, num_points) for the moneyness grid
+            - fit_results: DataFrame from fit_model()
+            - log_moneyness_params: Tuple of (min, max, num_points) for the moneyness grid
+            - return_domain: Domain for x-axis values ('log_moneyness', 'moneyness', 'strikes', 'delta')
 
         Returns:
-        - Tuple of (moneyness_array, iv_surface)
+        - Tuple of (iv_surface, x_surface)
         """
         # Generate the surface
-        moneyness_array, iv_surface = get_iv_surface(
+        iv_surface, x_surface = get_iv_surface(
             fit_results=fit_results,
-            moneyness_params=moneyness_params
+            log_moneyness_params=log_moneyness_params,
+            return_domain=return_domain
         )
 
         return {
-            'moneyness_array': moneyness_array,
-            'iv_surface': iv_surface
+            'iv_surface': iv_surface,
+            'x_surface': x_surface
         }
 
     @staticmethod

{voly-0.0.62 → voly-0.0.63}/src/voly/core/fit.py

@@ -11,6 +11,7 @@ from typing import List, Tuple, Dict, Optional, Union, Any
 from scipy.optimize import least_squares
 from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score
 from voly.utils.logger import logger, catch_exception
+from voly.formulas import get_x_domain
 from voly.exceptions import VolyError
 from voly.models import SVIModel
 import warnings
@@ -136,19 +137,30 @@ def fit_model(option_chain: pd.DataFrame,
 
 @catch_exception
 def get_iv_surface(fit_results: pd.DataFrame,
-                   log_moneyness_params: Tuple[float, float, int] = (-2, 2, 500)
-                   ) -> Tuple[np.ndarray, Dict[str, np.ndarray]]:
-    """Generate implied volatility surface using optimized SVI parameters."""
-    # Extract moneyness parameters
+                   log_moneyness_params: Tuple[float, float, int] = (-1.5, 1.5, 1000),
+                   return_domain: str = 'log_moneyness') -> Tuple[Dict[str, np.ndarray], np.ndarray]:
+    """
+    Generate implied volatility surface using optimized SVI parameters.
+
+    Parameters:
+    - fit_results: DataFrame from fit_model()
+    - log_moneyness_params: Tuple of (min, max, num_points) for the moneyness grid
+    - return_domain: Domain for x-axis values ('log_moneyness', 'moneyness', 'strikes', 'delta')
+
+    Returns:
+    - Tuple of (iv_surface, x_surface)
+      iv_surface: Dictionary mapping maturity names to IV arrays
+      x_surface: Dictionary mapping maturity names to requested x domain arrays
+    """
+
+    # Generate implied volatility surface in log-moneyness domain
     min_m, max_m, num_points = log_moneyness_params
+    log_moneyness_array = np.linspace(min_m, max_m, num=num_points)
 
-    # Generate moneyness array and initialize surface dictionary
-    moneyness_array = np.linspace(min_m, max_m, num=num_points)
     iv_surface = {}
-
-    # Generate implied volatility for each maturity
+    x_surface = {}
     for maturity in fit_results.columns:
-        # Extract parameters and YTM
+        # Calculate SVI total implied variance and convert to IV
         params = [
             fit_results.loc['a', maturity],
             fit_results.loc['b', maturity],
@@ -158,8 +170,19 @@ def get_iv_surface(fit_results: pd.DataFrame,
         ]
         ytm = fit_results.loc['ytm', maturity]
 
-        # Calculate SVI total implied variance and convert to IV
-        w_svi = np.array([SVIModel.svi(x, *params) for x in moneyness_array])
-        iv_surface[maturity] = np.sqrt(w_svi / ytm)
+        w_svi = np.array([SVIModel.svi(x, *params) for x in log_moneyness_array])
+        iv_array = np.sqrt(w_svi / ytm)
+        iv_surface[maturity] = iv_array
+
+        x_domain = get_x_domain(
+            log_moneyness_params=log_moneyness_params,
+            return_domain=return_domain,
+            s=fit_results.loc['s', maturity],
+            r=fit_results.loc['r', maturity],
+            iv_array=iv_array,
+            ytm=fit_results.loc['ytm', maturity]
+        )
+
+        x_surface[maturity] = x_domain
 
-    return moneyness_array, iv_surface
+    return iv_surface, x_surface

{voly-0.0.62 → voly-0.0.63}/src/voly/formulas.py

@@ -7,7 +7,7 @@ from scipy.stats import norm
 from typing import Tuple, Dict, Union, List, Optional
 from voly.utils.logger import catch_exception
 from voly.models import SVIModel
-from math import exp
+from typing import Tuple
 
 
 @catch_exception
@@ -276,3 +276,86 @@ def iv(option_price: float, s: float, k: float, r: float, t: float,
 
     # If we reach here, we didn't converge
     return np.nan
+
+
+@catch_exception
+def get_x_domain(log_moneyness_params: Tuple[float, float, int] = (-1.5, 1.5, 1000),
+                 return_domain: str = 'log_moneyness',
+                 s: float = None,
+                 r: float = None,
+                 iv_array: np.ndarray = None,
+                 ytm: float = None) -> np.ndarray:
+    """
+    Compute the x-domain for a given return type (log-moneyness, moneyness, strikes, or delta).
+
+    Parameters:
+    -----------
+    log_moneyness_params : Tuple[float, float, int],
+        Parameters for log-moneyness domain: (min_log_moneyness, max_log_moneyness, num_points).
+        Default is (-1.5, 1.5, 1000).
+    return_domain : str, optional
+        The desired domain to return. Options are 'log_moneyness', 'moneyness', 'strikes', or 'delta'.
+        Default is 'log_moneyness'.
+    s : float, optional
+        Spot price of the underlying asset. Required for 'strikes' and 'delta' domains.
+    r : float, optional
+        Risk-free interest rate. Required for 'delta' domain.
+    iv_array : np.ndarray, optional
+        Array of implied volatilities. Required for 'delta' domain.
+    ytm : float, optional
+        Time to maturity in years. Required for 'delta' domain.
+
+    Returns:
+    --------
+    np.ndarray
+        The x-domain array corresponding to the specified return_domain.
+
+    Raises:
+    -------
+    ValueError
+        If required parameters are missing for the specified return_domain.
+    """
+
+    # Extract log-moneyness parameters
+    min_m, max_m, num_points = log_moneyness_params
+
+    # Generate the base log-moneyness array
+    log_moneyness = np.linspace(min_m, max_m, num_points)
+
+    # Handle different return domains
+    if return_domain == 'log_moneyness':
+        return log_moneyness
+
+    elif return_domain == 'moneyness':
+        return np.exp(log_moneyness)
+
+    elif return_domain == 'strikes':
+        if s is None:
+            raise ValueError("Spot price 's' is required for return_domain='strikes'.")
+        return s / np.exp(log_moneyness)  # K = S/exp(log_moneyness)
+
+    elif return_domain == 'delta':
+        # Check for required parameters
+        required_params = {'s': s, 'r': r, 'iv_array': iv_array, 'ytm': ytm}
+        missing_params = [param for param, value in required_params.items() if value is None]
+        if missing_params:
+            raise ValueError(f"The following parameters are required for return_domain='delta': {missing_params}")
+
+        if len(iv_array) != len(log_moneyness):
+            raise ValueError(
+                f"iv_array must have the same length as the log-moneyness array ({len(log_moneyness)}).")
+
+        # Compute strikes
+        strikes = s / np.exp(log_moneyness)
+
+        # Compute deltas
+        delta_values = np.array([
+            delta(s, k, r, vol, ytm, 'call')
+            for k, vol in zip(strikes, iv_array)
+        ])
+
+        return delta_values
+
+    else:
+        raise ValueError(
+            f"Invalid return_domain: {return_domain}. Must be one of ['log_moneyness', 'moneyness', 'strikes', 'delta'].")

{voly-0.0.62 → voly-0.0.63/src/voly.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: voly
-Version: 0.0.62
+Version: 0.0.63
 Summary: Options & volatility research package
 Author-email: Manu de Cara <manu.de.cara@gmail.com>
 License: MIT