rock-physics-open 0.0__py3-none-any.whl

rock_physics_open/equinor_utilities/various_utilities/vp_vs_rho_set_statistics.py

@@ -0,0 +1,154 @@
+import os
+
+import numpy as np
+import pandas as pd
+from sklearn.metrics import r2_score
+
+from .display_result_statistics import disp_result_stats
+
+
+def vp_vs_rho_stats(
+    vp_observed,
+    vs_observed,
+    rho_observed,
+    vp_estimated,
+    vs_estimated,
+    rho_estimated,
+    fname,
+    estimated_set_names,
+    well_names,
+    file_mode="a",
+    disp_results=True,
+):
+    """
+    Utility to estimate statistics between vp-vs-rho sets - observed and estimated values. The results are displayed
+    on screen (optional) and saved to a .csv file. If the file exists, the results will be appended.
+    The statistics consists of RAE (relative absolute error) [fraction], RRMSE (relative root mean squared error)
+    [fraction] and R^2 (coefficient of determination) [fraction].
+
+
+    Parameters
+    ----------
+    vp_observed : np.ndarray or list
+        Observed vp [m/s].
+    vs_observed : np.ndarray or list
+        Observed vs [m/s].
+    rho_observed : np.ndarray or list
+        Observed density [kg/m^3].
+    vp_estimated : np.ndarray or list
+        Estimated vp [m/s].
+    vs_estimated : np.ndarray or list
+        Estimated vs [m/s].
+    rho_estimated : np.ndarray or list
+        Estimated density [kg/m^3].
+    fname : str
+        File name for saved results.
+    estimated_set_names : str or list
+        Name of the estimated vp-vs-rho set(s).
+    well_names : str or list
+        Well name of the vp-vs-rho set(s).
+    file_mode : str
+        File open mode, default append.
+    disp_results : bool
+        Display results on screen.
+    """
+    if isinstance(estimated_set_names, str):
+        estimated_set_names = [estimated_set_names]
+    if isinstance(well_names, str):
+        well_names = [well_names]
+
+    _verify(
+        vp_observed,
+        vs_observed,
+        rho_observed,
+        vp_estimated,
+        vs_estimated,
+        rho_estimated,
+        set_names=estimated_set_names,
+        well_names=well_names,
+        file_mode=file_mode,
+    )
+
+    est_frame_columns = [
+        "Well",
+        "Vp RMAE",
+        "Vp RRMSE",
+        "Vp R2",
+        "Vs RMAE",
+        "Vs RRMSE",
+        "Vs R2",
+        "Rho RMAE",
+        "Rho RRMSE",
+        "Rho R2",
+    ]
+    est_frame = pd.DataFrame(columns=est_frame_columns, index=estimated_set_names)
+    est_frame.index.name = "Estimated set name"
+    est_frame.iloc[:, 0] = well_names
+
+    # If inputs are found to satisfy expectations in _verify, and they are numpy arrays, cast to lists, and run through
+    if isinstance(vp_observed, np.ndarray):
+        vp_observed = [vp_observed]
+        vs_observed = [vs_observed]
+        rho_observed = [rho_observed]
+        vp_estimated = [vp_estimated]
+        vs_estimated = [vs_estimated]
+        rho_estimated = [rho_estimated]
+
+    for i in range(len(vp_observed)):
+        res = []
+        for obs, est in zip(
+            [vp_observed[i], vs_observed[i], rho_observed[i]],
+            [vp_estimated[i], vs_estimated[i], rho_estimated[i]],
+        ):
+            res.append(np.mean(np.abs((est.flatten() - obs.flatten()) / obs.flatten())))
+            res.append(
+                np.sqrt(
+                    np.mean(np.square((est.flatten() - obs.flatten()) / obs.flatten()))
+                )
+            )
+            res.append(r2_score(obs.flatten(), est.flatten()))
+
+        res_dict = dict(zip(est_frame_columns[1:], res))
+        est_frame.iloc[i, 1:] = res_dict
+        if disp_results:
+            disp_result_stats(
+                estimated_set_names[i], res, est_frame_columns[1:], values_only=True
+            )
+    # Test if the file already exists. If so, and the file_mode is set to 'a', drop the column headers when writing
+    # the file
+    if os.path.exists(fname) and file_mode == "a":
+        est_frame.to_csv(fname, mode=file_mode, header=False)
+    else:
+        est_frame.to_csv(fname, mode=file_mode)
+
+
+def _verify(*args, set_names=None, well_names=None, file_mode=None):
+    """Verify that arguments are either numpy arrays or lists of numpy arrays.
+    Raises
+    ------
+    ValueError
+        If not np.ndarray or list of such.
+    ValueError
+        If an entry contains NaNs.
+    ValueError
+        If an entry contains Infs.
+    ValueError
+        If mismatch in argument lengths.
+    ValueError
+        If file mode not 'a' or 'w'.
+    """
+    # Verify that args are either numpy arrays or list of arrays
+    for arg in args:
+        if isinstance(arg, np.ndarray):
+            arg = [arg]
+        for this_arg in arg:
+            if not isinstance(this_arg, np.ndarray):
+                raise ValueError(f"{__file__}: input not numpy array: {type(arg)}")
+            if np.any(np.isnan(this_arg)):
+                raise ValueError(f"{__file__}: input contains NaNs")
+            if np.any(np.isinf(this_arg)):
+                raise ValueError(f"{__file__}: input contains Infs")
+        if not len(arg) == len(set_names) == len(well_names):
+            raise ValueError(f"{__file__}: mismatch in argument lengths")
+    if not (file_mode == "a" or file_mode == "w"):
+        raise ValueError(f'{__file__}: file_mode must be one of ["a", "w"]')

rock_physics_open/equinor_utilities/various_utilities/vrh_3_min.py

@@ -0,0 +1,61 @@
+from rock_physics_open.equinor_utilities import std_functions
+
+
+def min_3_voigt_reuss_hill(
+    vp1, vs1, rhob1, f1, vp2, vs2, rhob2, f2, vp3, vs3, rhob3, f3
+):
+    """
+    Mix of three phases by Voigt-Reuss-Hill model. The fractions should add up to 1 with input of vp, vs and rho.
+
+    Parameters
+    ----------
+    vp1 : np.ndarray
+        Pressure wave velocity of phase 1 [m/s].
+    vs1 : np.ndarray
+        Shear wave velocity of phase 1 [m/s].
+    rhob1 : np.ndarray
+        Bulk density of phase 1 [kg/m^3].
+    f1 : np.ndarray
+        Fraction of phase 1 [fraction].
+    vp2 : np.ndarray
+        Pressure wave velocity of phase 2 [m/s].
+    vs2 : np.ndarray
+        Shear wave velocity of phase 2 [m/s].
+    rhob2 : np.ndarray
+        Bulk density of phase 2 [kg/m^3].
+    f2 : np.ndarray
+        Fraction of phase 2 [fraction].
+    vp3 : np.ndarray
+        Pressure wave velocity of phase 3 [m/s].
+    vs3 : np.ndarray
+        Shear wave velocity of phase 3 [m/s].
+    rhob3 : np.ndarray
+        Bulk density of phase 3 [kg/m^3].
+    f3 : np.ndarray
+        Fraction of phase 3 [fraction].
+
+    Returns
+    -------
+    tuple
+        vp, vs, rhob, ai, vp_vs, k, mu : np.ndarray.
+        vp: pressure wave velocity [m/s], vs: shear wave velocity [m/s], rhob: bulk density [kg/m^3],
+        ai: acoustic impedance [m/s x kg/m^3], vp_vs: velocity ratio [ratio],
+        k: effective bulk modulus [Pa], mu: effective shear modulus [Pa].
+    """
+    k1, mu1 = std_functions.moduli(vp1, vs1, rhob1)
+    k2, mu2 = std_functions.moduli(vp2, vs2, rhob2)
+    k3, mu3 = std_functions.moduli(vp3, vs3, rhob3)
+
+    # Normalise the fractions to make sure they sum to one
+    tot = f1 + f2 + f3
+    f1 = f1 / tot
+    f2 = f2 / tot
+    f3 = f3 / tot
+
+    k, mu = std_functions.multi_voigt_reuss_hill(k1, mu1, f1, k2, mu2, f2, k3, mu3, f3)
+
+    rhob = rhob1 * f1 + rhob2 * f2 + rhob3 * f3
+
+    vp, vs, ai, vp_vs = std_functions.velocity(k, mu, rhob)
+
+    return vp, vs, rhob, ai, vp_vs, k, mu

rock_physics_open/fluid_models/__init__.py

@@ -0,0 +1,9 @@
+from .brine_model import brine_properties
+from .gas_model import gas_properties
+from .oil_model import oil_properties
+
+__all__ = [
+    "brine_properties",
+    "gas_properties",
+    "oil_properties",
+]

rock_physics_open/fluid_models/brine_model/__init__.py

@@ -0,0 +1,5 @@
+from .brine_properties import brine_properties
+
+__all__ = [
+    "brine_properties",
+]

rock_physics_open/fluid_models/brine_model/brine_properties.py

@@ -0,0 +1,143 @@
+import warnings
+
+import numpy as np
+from numpy import sqrt
+from numpy.polynomial.polynomial import polyval2d, polyval3d
+
+
+def brine_properties(
+    temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    salinity: np.ndarray | float,
+    p_nacl: np.ndarray | float | None = None,
+    p_kcl: np.ndarray | float | None = None,
+    p_cacl: np.ndarray | float | None = None,
+) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
+    """
+    :param salinity: Salinity of solution as ppm of NaCl.
+    :param pressure: Pressure (Pa) of oil
+    :param temperature: Temperature (Celsius) of oil.
+    :param p_nacl: NaCl percentage, for future use
+    :param p_kcl: KCl percentage, for future use
+    :param p_cacl: CaCl percentage, for future use
+    :return: vel_b [m/s], den_b [kg/m^3], k_b [Pa]
+    """
+    vel_b = brine_primary_velocity(temperature, pressure * 1e-6, salinity * 1e-6)
+    den_b = brine_density(temperature, pressure * 1e-6, salinity * 1e-6) * 1000
+    k_b = vel_b**2 * den_b
+    return vel_b, den_b, k_b
+
+
+def brine_density(
+    temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    salinity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    density of sodium chloride solutions, equation 27 in Batzle & Wang [1].
+    :param salinity: Salinity of solution as weight fraction (ppm/1000000) of
+        sodium chloride.
+    :param pressure: Pressure (MPa) of oil
+    :param temperature: Temperature (Celsius) of oil.
+    :return: density of solution in g/cc.
+    """
+    coefficients = [
+        [[0.668, 3e-4], [8e-5, -13e-6], [3e-6, 0.0]],
+        [[0.44, -24e-4], [-33e-4, 47e-6], [0.0, 0.0]],
+    ]
+    return water_density(temperature, pressure) + salinity * polyval3d(
+        salinity, temperature, pressure, coefficients
+    )
+
+
+def brine_primary_velocity(
+    temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    salinity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    Primary wave velocity of sodium chloride solutions, equation 29 in Batzle & Wang [1]
+
+    :param salinity: Salinity of solution as weight fraction (ppm/1000000) of
+        sodium chloride.
+    :param pressure: Pressure (MPa) of oil
+    :param temperature: Temperature (Celsius) of oil.
+    :return: velocity of solution in m/s.
+    """
+    coefficients = np.zeros((3, 4, 3))
+    coefficients[0, 0, 0] = 1170
+    coefficients[0, 1, 0] = -9.6
+    coefficients[0, 2, 0] = 0.055
+    coefficients[0, 3, 0] = -8.5e-5
+    coefficients[0, 0, 1] = 2.6
+    coefficients[0, 1, 1] = -29e-4
+    coefficients[0, 0, 2] = -0.0476
+    coefficients[1, 0, 0] = 780
+    coefficients[1, 0, 1] = -10
+    coefficients[1, 0, 2] = 0.16
+    coefficients[2, 0, 0] = -820
+
+    return water_primary_velocity(temperature, pressure) + salinity * polyval3d(
+        sqrt(salinity), temperature, pressure, coefficients
+    )
+
+
+def water_density(
+    temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    Density of water,, equation 27a in Batzle & Wang [1].
+    :param pressure: Pressure (MPa) of oil
+    :param temperature: Temperature (Celsius) of oil.
+    :return: Density of water in g/cc.
+    """
+    coefficients = [
+        [1.0, 489e-6, -333e-9],
+        [-8e-5, -2e-6, -2e-09],
+        [-33e-7, 16e-9, 0.0],
+        [1.75e-9, -13e-12, 0.0],
+    ]
+    return polyval2d(temperature, pressure, coefficients)
+
+
+def water_primary_velocity(
+    temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    Primary wave velocity of water, table 1 and equation 28 in Batzle & Wang [1].
+    :param pressure: Pressure (MPa) of oil
+    :param temperature: Temperature (Celsius) of oil.
+    :return: primary wave velocity of water in m/s.
+    """
+    if np.any(pressure > 100):
+        warnings.warn(
+            "Calculations for water velocity is not precise for\n"
+            + "pressure outside [0,100]MPa"
+            + f"pressure given: {pressure}MPa",
+            stacklevel=1,
+        )
+    coefficients = [
+        [1402.85, 1.524, 3.437e-3, -1.197e-5],
+        [4.871, -1.11e-2, 1.739e-4, -1.628e-6],
+        [-4.783e-2, 2.747e-4, -2.135e-6, 1.237e-8],
+        [1.487e-4, -6.503e-7, -1.455e-8, 1.327e-10],
+        [-2.197e-7, 7.987e-10, 5.23e-11, -4.614e-13],
+    ]
+    return polyval2d(temperature, pressure, coefficients)
+
+
+def water(
+    temperature: np.ndarray | float, pressure: np.ndarray | float
+) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
+    """
+    :param pressure: Pressure (Pa) of oil
+    :param temperature: Temperature (Celsius) of oil.
+    :return: water_density [kg/m^3], water_velocity [m/s], water_bulk_modulus [Pa]
+    """
+    pressure_mpa = pressure * 1.0e-6
+    water_den = water_density(temperature, pressure_mpa)
+    water_vel = water_primary_velocity(temperature, pressure_mpa)
+    water_k = water_vel**2 * water_den * 1000.0
+    return water_den, water_vel, water_k

rock_physics_open/fluid_models/gas_model/__init__.py

@@ -0,0 +1,5 @@
+from .gas_properties import gas_properties
+
+__all__ = [
+    "gas_properties",
+]

rock_physics_open/fluid_models/gas_model/gas_properties.py

@@ -0,0 +1,277 @@
+import numpy as np
+from numpy import exp, sqrt
+from scipy.constants import gas_constant
+
+from rock_physics_open.equinor_utilities.conversions import celsius_to_kelvin
+
+AIR_WEIGHT = 28.8  # g/mol
+
+
+def gas_properties(
+    temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+    model: str | None = None,
+) -> tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray]:
+    """
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param pressure: Confining pressure (Pa)
+    :param temperature: Temperature (Celsius).
+    :param model: for future use
+    :return: vel_gas [m/s], den_gas [kg/m^3], k_gas [Pa], eta_gas [cP]
+    """
+    den_gas = gas_density(celsius_to_kelvin(temperature), pressure * 1e-6, gas_gravity)
+    k_gas = gas_bulk_modulus(
+        celsius_to_kelvin(temperature), pressure * 1e-6, gas_gravity
+    )
+    vel_gas = (k_gas / den_gas) ** 0.5
+
+    eta_gas = lee_gas_viscosity(celsius_to_kelvin(temperature), pressure, gas_gravity)
+
+    return vel_gas, den_gas, k_gas, eta_gas
+
+
+def molecular_weight(gas_gravity: np.ndarray | float) -> np.ndarray | float:
+    """
+    calculates molecluar weight of a gas from gas gravity.
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :return: The volume of the gas in g/mol.
+    """
+    return gas_gravity * AIR_WEIGHT
+
+
+def molar_volume(
+    absolute_temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    calculates molar volume using the ideal gas law.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :param pressure: Confining pressure in MPa.
+    :return: The volume of the gas in cc/mol.
+    """
+    return gas_constant * absolute_temperature / pressure
+
+
+def ideal_gas_density(
+    absolute_temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    calculates molar volume using the ideal gas law.
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :param pressure: Confining pressure in MPa.
+    :return: The density of the gas in g/cc
+    """
+    return molecular_weight(gas_gravity) / molar_volume(absolute_temperature, pressure)
+
+
+def ideal_gas_primary_velocity(
+    absolute_temperature: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :return: The compressional wave velocity of the gas in m/s.
+    """
+    return sqrt(gas_constant * absolute_temperature / molecular_weight(gas_gravity))
+
+
+def ideal_gas(
+    absolute_temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> tuple[np.ndarray | float, np.ndarray | float]:
+    """
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :param pressure: Confining pressure in Pa.
+    :return: ideal_gas_density, ideal_gas_velocity
+    """
+    ideal_gas_den = 1000 * ideal_gas_density(
+        absolute_temperature, pressure * 1e6, gas_gravity
+    )
+    ideal_gas_vel = ideal_gas_primary_velocity(absolute_temperature, gas_gravity)
+    return ideal_gas_den, ideal_gas_vel
+
+
+def pseudoreduced_temperature(
+    absolute_temperature: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    calculates pseudoreduced temperature, equation 9a from Batzle & Wang [1].
+
+    Uses relationship from
+
+    Thomas, L. K., Hankinson, R. W., and Phillips, K. A., 1970,
+    Determination of acoustic velocities for natural gas: J. Petr.
+    Tech., 22, 889-892.
+
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :return: Pseudoreduced temperature in kelvin.
+    """
+    return absolute_temperature / (94.72 + 170.75 * gas_gravity)
+
+
+def pseudoreduced_pressure(
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    calculates pseudoreduced pressure, equation 9a from Batzle & Wang [1].
+
+    Uses relationship from
+
+    Thomas, L. K., Hankinson, R. W., and Phillips, K. A., 1970,
+    Determination of acoustic velocities for natural gas: J. Petr.
+    Tech., 22, 889-892.
+
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param pressure: Confining pressure in MPa.
+    :return: Pseudoreduced pressure in MPa.
+    """
+    return pressure / (4.892 - 0.4048 * gas_gravity)
+
+
+def compressability_factor(
+    absolute_temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    calculates compressability hydro-carbon gas, equation 10b and 10c from
+    Batzle & Wang [1].
+
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :param pressure: Confining pressure in MPa.
+    :return: The density of the gas in g/cc
+    """
+    tpr = pseudoreduced_temperature(absolute_temperature, gas_gravity)
+    ppr = pseudoreduced_pressure(pressure, gas_gravity)
+
+    return (
+        (0.03 + 0.00527 * (3.5 - tpr) ** 3) * ppr
+        + 0.642 * tpr
+        - 0.007 * tpr**4
+        - 0.52
+        + 0.109
+        * (3.85 - tpr) ** 2
+        / exp((0.45 + 8 * (0.56 - 1 / tpr) ** 2) * ppr**1.2 / tpr)
+    )
+
+
+def gas_density(
+    absolute_temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    The density of hydro-carbon gas, using equation 10 from Batzle & Wang [1].
+
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :param pressure: Confining pressure in MPa.
+    :return: The density of the gas in g/cc
+    """
+    ideal_gas_den, ideal_gas_vel = ideal_gas(
+        absolute_temperature, pressure * 1e-6, gas_gravity
+    )
+    return ideal_gas_den / compressability_factor(
+        absolute_temperature, pressure, gas_gravity
+    )
+
+
+def compressability_rate_per_pseudoreduced_pressure(
+    absolute_temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    Derivate of compressability_factor with respect to pressure.
+
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :param pressure: Confining pressure in MPa.
+    :return: The density of the gas in g/cc
+    """
+    tpr = pseudoreduced_temperature(absolute_temperature, gas_gravity)
+    ppr = pseudoreduced_pressure(pressure, gas_gravity)
+
+    return (
+        0.03
+        + 0.00527 * (3.5 - tpr) ** 3
+        - (
+            0.1308
+            * (0.45 + 8 * (0.56 - tpr ** (-1)) ** 2)
+            * (3.85 - tpr) ** 2
+            * ppr**0.2
+        )
+        / (exp(((0.45 + 8 * (0.56 - tpr ** (-1)) ** 2) * ppr**1.2) / tpr) * tpr)
+    )
+
+
+def gas_bulk_modulus(
+    absolute_temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray | float:
+    """
+    The bulk modulus of hydro-carbon gas, using equation 11 from Batzle & Wang [1].
+
+    :param gas_gravity: molar mass of gas relative to air molar mas.
+    :param absolute_temperature: The absolute temperature of the gas in kelvin.
+    :param pressure: Confining pressure in MPa.
+    :return: The bulk modulus of the gas in MPa.
+    """
+    z = compressability_factor(absolute_temperature, pressure, gas_gravity)
+    dz_dppr = compressability_rate_per_pseudoreduced_pressure(
+        absolute_temperature, pressure, gas_gravity
+    )
+
+    ppr = pseudoreduced_pressure(pressure, gas_gravity)
+
+    # Equation 11b
+    gamma_0 = (
+        0.85
+        + 5.6 / (ppr + 2)
+        + 27.1 / ((ppr + 3.5) ** 2)
+        - 8.7 * exp(-0.65 * (ppr + 1))
+    )
+
+    return gamma_0 * pressure / (1 - dz_dppr * ppr / z)
+
+
+def lee_gas_viscosity(
+    absolute_temperature: np.ndarray | float,
+    pressure: np.ndarray | float,
+    gas_gravity: np.ndarray | float,
+) -> np.ndarray:
+    """
+    :param absolute_temperature: Absolute temperature of the gas in kelvin.
+    :param pressure: Confining pressure in Pa.
+    :param gas_gravity: specific gravity of gas relative to air.
+    :return: gas viscosity in cP
+
+    Reference
+    ---------
+    Lee, J. D., et al. (1966). "Viscosity of Natural Gas." In The American Institute of
+    Chemical Engineers Journal, Volume 12, Issue 6, pp. 1058-1062.
+
+    Original equation is given in imperial units. Inputs are transformed to temperature
+    in Farenheit and pressure in psi
+    """
+    temp_far = (absolute_temperature - 273.15) * 9.0 / 5.0 + 32.0
+    pres_psi = pressure / 6894.757
+    return (
+        0.001
+        * (temp_far + 459.67) ** 0.5
+        / pres_psi
+        * (0.7 + 1.5 * gas_gravity)
+        / (gas_gravity + 1) ** 1.5
+    )

rock_physics_open/fluid_models/oil_model/__init__.py

@@ -0,0 +1,5 @@
+from .oil_properties import oil_properties
+
+__all__ = [
+    "oil_properties",
+]