API Reference¶

Complete API documentation for gdrift. All classes and functions are organized by functionality.

Core Data Loading¶

load_dataset¶

load_dataset ¶

load_dataset(dataset_name: str, table_names=[], return_metadata=False)

Load a dataset from local cache, downloading if necessary.

Args: dataset_name (str): Dataset name (without .h5 extension) table_names (list, optional): Specific tables to load. Defaults to []. return_metadata (bool, optional): Whether to return file-level metadata.

Returns: dict: dictionary with all the datasets (and optionally metadata tuple)

1D Radial Profiles¶

RadialEarthModel¶

RadialEarthModel ¶

RadialEarthModel(profiles: Union[AbstractProfile, List[AbstractProfile]])

Class representing reference Earth Models such as PREM or AK135 Composite object containing multiple radial profiles representing different Earth properties, such as shear wave velocity (Vs), primary wave velocity (Vp), and density. T

Attributes: depth_profiles (dict): A dictionary of RadialProfile instances.

Initialise the RadialEarthModel with a dictionary of radial profiles instances.

Args: profiles (dict of RadialProfile): Profiles for different properties, keyed by property name.

get_profile ¶

get_profile(property_name: str) -> AbstractProfile

Retrieve a profile by its name.

Args: property_name (str): The name of the property profile to retrieve.

Returns: AbstractProfile: The profile corresponding to the specified property name.

Raises: ValueError: If the specified property name does not exist in the model.

RadialEarthModelFromFile¶

RadialEarthModelFromFile ¶

RadialEarthModelFromFile(model_name: str, description: str = None)

Bases: RadialEarthModel

A class for loading radial profiles from a dataset.

This class extends SplineProfile to specifically handle loading, and utilizing available profiles related to profiles in the mantle.

Attributes: model_name (str): The name of the model/dataset from which profiles are loaded. description (str, optional): A brief description of the profile's purpose or characteristics.

Initialize a radial Earth model by loading profiles from an HDF5 dataset.

Loads all property profiles from a registered dataset file. The HDF5 file must contain a "depth" array and one or more property arrays (e.g., "density", "vs", "vp"). Each property is wrapped in a SplineProfile for interpolation.

Parameters:

Name	Type	Description	Default
`model_name`	`str`	Name of the registered dataset (e.g., "1d_prem", "1d_solidus_Andrault_et_al_2011_EPSL"). Must exist in the dataset registry.	required
`description`	`str`	Human-readable description of the model. If None, no description is set. Default is None.	`None`

Raises:

Type	Description
`ValueError`	If model_name is not in the dataset registry.
`KeyError`	If the HDF5 file does not contain a "depth" array.

Examples:

>>> import gdrift
>>> andrault = gdrift.RadialEarthModelFromFile(
...     "1d_solidus_Andrault_et_al_2011_EPSL")
>>> print(andrault.get_profile_names())
['solidus temperature']
>>> solidus = andrault.get_profile("solidus temperature")
>>> T = solidus.at_depth(410e3)

PreliminaryRefEarthModel¶

PreliminaryRefEarthModel ¶

PreliminaryRefEarthModel()

Bases: RadialEarthModelFromFile

Initialises the Preliminary Reference Earth Model (PREM). This model is based on the work by Dziewonski and Anderson (1981) and provides a reference Earth model that can be queried at specific depths for various profiles.

References: Dziewonski, Adam M., and Don L. Anderson. "Preliminary reference Earth model." Physics of the Earth and Planetary Interiors 25.4 (1981): 297-356.

The object is of type RadialEarthModel and is initialized by loading profiles from an existing dataset. Each profile is represented as a SplineProfile object.

Attributes: prem_profiles (list): A list of SplineProfile objects representing different profiles in the PREM dataset.

Initialize the Preliminary Reference Earth Model (PREM).

Loads the PREM dataset (Dziewonski & Anderson, 1981) containing reference profiles for density, seismic velocities, elastic moduli, pressure, and gravity as a function of radius/depth.

The model spans from Earth's center to the surface and includes discontinuities at major boundaries (e.g., core-mantle boundary, 410 km, 660 km discontinuities).

Notes

PREM is the standard 1D reference model for seismology and geodynamics. It represents a spherically symmetric, non-rotating, oceanless Earth.

Examples:

>>> import gdrift
>>> prem = gdrift.PreliminaryRefEarthModel()
>>> print(prem.get_profile_names())
['density', 'vs', 'vp', ...]
>>> rho_profile = prem.get_profile("density")
>>> rho_670km = rho_profile.at_depth(670e3)

References

Dziewonski, A. M., & Anderson, D. L. (1981). Preliminary reference Earth model. Physics of the Earth and Planetary Interiors, 25(4), 297-356. https://doi.org/10.1016/0031-9201(81)90046-7

HirschmannSolidus¶

HirschmannSolidus ¶

HirschmannSolidus()

Bases: RadialEarthModel

A class representing the solidus model based on the work of Hirschmann (2000).

Attributes: solidus_profile (HirschmannSolidusProfile): The solidus profile based on Hirschmann (2000).

Initialize a Hirschmann solidus radial Earth model.

Creates a RadialEarthModel containing a single profile: the Hirschmann (2000) solidus temperature. This is a convenience wrapper around HirschmannSolidusProfile that conforms to the RadialEarthModel interface.

The solidus can be accessed via get_profile("solidus temperature") or directly through the solidus_profile attribute.

Examples:

>>> import gdrift
>>> solidus_model = gdrift.HirschmannSolidus()
>>> solidus_profile = solidus_model.get_profile("solidus temperature")
>>> T_410km = solidus_profile.at_depth(410e3)
>>> print(f"Solidus at 410 km: {T_410km:.1f} K")

See Also

HirschmannSolidusProfile : The underlying solidus implementation

3D Seismic Models¶

SeismicModel¶

SeismicModel ¶

SeismicModel(model_name, nearest_neighbours: int = 8, default_max_distance: float = 200000.0, labels=[])

Bases: EarthModel3D

SeismicModel is a class for handling 3D seismic models.

This class inherits from EarthModel3D and is used to load and manage seismic models. It allows for the initialization of a seismic model with a specified name, number of nearest neighbours, and a default maximum distance. The model data is loaded from a dataset file, and quantities and coordinates are set accordingly.

Parameters:

model_name : str Name of the seismic model to load nearest_neighbours : int, optional Number of nearest neighbours to consider for interpolation. Defaults to 8. default_max_distance : float, optional The default maximum distance for the model in meters. Defaults to 200e3. labels : list, optional Specific labels to load from the dataset. If empty, loads all available fields.

Attributes: model_name (str): The name of the seismic model. nearest_neighbours (int): The number of nearest neighbours to consider. default_max_distance (float): The default maximum distance for the model in meters.

The full list of available tomography models is exposed as gdrift.AVAILABLE_SEISMIC_MODELS. See the Data Catalog for descriptions and citations.

Thermodynamics¶

ThermodynamicModel¶

ThermodynamicModel ¶

ThermodynamicModel(model: str, composition: str, temps=None, depths=None, extrapolate=False)

Bases: object

Thermodynamic lookup table for mantle mineral physics properties.

Provides 2D interpolation of pre-computed mineral physics properties (density, seismic velocities, elastic moduli) as a function of depth and temperature. Tables are based on self-consistent thermodynamic databases (Stixrude & Lithgow-Bertelloni 2011, 2021) and stored in HDF5 format with bivariate spline interpolation.

The model supports: - Forward queries: (depth, temperature) → property (e.g., Vs, Vp, rho) - Inverse queries: (depth, velocity) → temperature - Multiple compositions and database versions (see MODELS_AVAIL, COMPOSITIONS_AVAIL) - On-the-fly computation of Vs/Vp from elastic moduli

Parameters:

Name	Type	Description	Default
`model`	`str`	Thermodynamic database version (e.g. "SLB_21"). See MODELS_AVAIL for all available versions, derived from the datasets.json manifest.	required
`composition`	`str`	Mantle composition. Available options depend on the model. See COMPOSITIONS_AVAIL for all available compositions (e.g. "pyroliteCFMAS", "pyroliteNCMAS", "bulk-oceanic-crustCFMAS").	required
`temps`	`array_like`	Temperature grid for subsampling (Kelvin). If None, uses full temperature range from dataset. Default is None.	`None`
`depths`	`array_like`	Depth grid for subsampling (meters). If None, uses full depth range from dataset. Default is None.	`None`
`extrapolate`	`bool`	Whether to allow extrapolation outside the table bounds. If False, queries outside the range raise ValueError. Default is False.	`False`

Attributes:

Name	Type	Description
`model`	`str`	Database version (e.g., "SLB_21").
`composition`	`str`	Mantle composition (e.g., "pyroliteCFMAS").
`extrapolate`	`bool`	Extrapolation flag.
`_tables`	`dict`	Dictionary mapping property names to Table objects. Keys are the property names from the HDF5 file (e.g., "rho", "bulk_mod", "shear_mod", "v_s", "v_p").

Examples:

>>> import gdrift
>>> # Load SLB 2021 pyrolite model
>>> tm = gdrift.ThermodynamicModel("SLB_21", "pyroliteCFMAS")
>>> # List available properties
>>> print(tm.available_tables())
['rho', 'bulk_mod', 'shear_mod', 'v_s', 'v_p', ...]
>>> # Forward query: temperature to Vs
>>> vs = tm.temperature_to_vs(temperature=1600, depth=500e3)
>>> print(f"Vs = {vs:.1f} m/s")
>>> # Inverse query: Vs to temperature
>>> T = tm.vs_to_temperature(vs=4500, depth=500e3)
>>> print(f"Temperature = {T:.1f} K")
>>> # Generic property lookup
>>> rho = tm.temperature_to_property("rho", temperature=1600, depth=500e3)

Notes

All depths are in meters from the surface
All temperatures are in Kelvin
Velocities are in m/s, densities in kg/m³, moduli in Pa
Not all model/composition combinations are available
Use available_tables() to see which properties are loaded
Vs and Vp are computed from moduli if not present in HDF5 file

See Also

regularise_thermodynamic_table : Smooth phase transitions apply_anelastic_correction : Convert to seismic-frequency velocities compute_swave_speed : Calculate Vs from shear modulus and density compute_pwave_speed : Calculate Vp from bulk/shear moduli and density

References

Stixrude, L., & Lithgow-Bertelloni, C. (2011). Thermodynamics of mantle minerals—II. Phase equilibria. Geophysical Journal International, 184(3), 1180-1213. https://doi.org/10.1111/j.1365-246X.2010.04890.x

Stixrude, L., & Lithgow-Bertelloni, C. (2021). Thermal expansivity, heat capacity and bulk modulus of the mantle. Geophysical Journal International. (In preparation for SLB_21 parameters)

available_tables ¶

available_tables()

Return a list of available table names in this model.

temperature_to_property ¶

temperature_to_property(property_name, temperature, depth)

Convert temperature and depth to a material property value.

Args: property_name (str): Name of the property (e.g. 'rho', 'alpha', 'Cp', 'vs', 'vp'). temperature: Temperature value(s). depth: Depth value(s).

Returns: Interpolated property value(s) at the given temperature and depth. Returns NaN if extrapolate=False and inputs are out of bounds.

temperature_to_vs ¶

temperature_to_vs(temperature, depth)

Convert temperature and depth to shear wave velocity (Vs).

Convenience wrapper for temperature_to_property("vs", ...). Computes Vs from shear modulus and density using the relation: Vs = sqrt(shear_modulus / density).

Parameters:

Name	Type	Description	Default
`temperature`	`float or array_like`	Temperature in Kelvin.	required
`depth`	`float or array_like`	Depth in meters from the surface.	required

Returns:

Type	Description
`float or ndarray`	Shear wave velocity in m/s. Returns NaN for out-of-bounds queries if extrapolate=False.

Examples:

>>> import gdrift
>>> tm = gdrift.ThermodynamicModel("SLB_21", "pyroliteCFMAS")
>>> vs = tm.temperature_to_vs(temperature=1600, depth=500e3)
>>> print(f"Vs = {vs:.1f} m/s")

temperature_to_vp ¶

temperature_to_vp(temperature, depth)

Convert temperature and depth to compressional wave velocity (Vp).

Convenience wrapper for temperature_to_property("vp", ...). Computes Vp from bulk modulus, shear modulus, and density using: Vp = sqrt((bulk_modulus + 4/3 * shear_modulus) / density).

Parameters:

Name	Type	Description	Default
`temperature`	`float or array_like`	Temperature in Kelvin.	required
`depth`	`float or array_like`	Depth in meters from the surface.	required

Returns:

Type	Description
`float or ndarray`	Compressional wave velocity in m/s. Returns NaN for out-of-bounds queries if extrapolate=False.

Examples:

>>> import gdrift
>>> tm = gdrift.ThermodynamicModel("SLB_21", "pyroliteCFMAS")
>>> vp = tm.temperature_to_vp(temperature=1600, depth=500e3)
>>> print(f"Vp = {vp:.1f} m/s")

temperature_to_rho ¶

temperature_to_rho(temperature, depth)

Convert temperature and depth to density.

Convenience wrapper for temperature_to_property("rho", ...). Queries the pre-computed density lookup table.

Parameters:

Name	Type	Description	Default
`temperature`	`float or array_like`	Temperature in Kelvin.	required
`depth`	`float or array_like`	Depth in meters from the surface.	required

Returns:

Type	Description
`float or ndarray`	Density in kg/m³. Returns NaN for out-of-bounds queries if extrapolate=False.

Examples:

>>> import gdrift
>>> tm = gdrift.ThermodynamicModel("SLB_21", "pyroliteCFMAS")
>>> rho = tm.temperature_to_rho(temperature=1600, depth=500e3)
>>> print(f"Density = {rho:.1f} kg/m³")

vs_to_temperature ¶

vs_to_temperature(vs: Number, depth: Number, bounds: Optional[Union[Tuple[float, float], Tuple[ndarray, ndarray]]] = (300, 7000)) -> Number

Convert S-wave velocity (vs) to temperature at a given depth.

Parameters:

vs : Number The S-wave velocity. depth : Number The depth at which the temperature is to be calculated. bounds : Optional[Union[Tuple[float, float], Tuple[numpy.ndarray, numpy.ndarray]]], default=(300, 7000) The bounds for the temperature calculation. It can be a tuple of floats or numpy arrays.

Returns:

Number The temperature corresponding to the given S-wave velocity and depth.

vp_to_temperature ¶

vp_to_temperature(vp: Number, depth: Number, bounds: Optional[Union[Tuple[float, float], Tuple[ndarray, ndarray]]] = (300, 7000)) -> Number

Convert P-wave velocity (vp) to temperature at a given depth.

Parameters:

vp : Number The P-wave velocity. depth : Number The depth at which the temperature is to be calculated. bounds : Optional[Union[Tuple[float, float], Tuple[numpy.ndarray, numpy.ndarray]]], default=(300, 7000) The bounds for the temperature calculation. It can be a tuple of floats or numpy arrays.

Returns:

Number The temperature corresponding to the given P-wave velocity and depth.

Phase Transition Regularization¶

regularise_thermodynamic_table ¶

regularise_thermodynamic_table(slb_pyrolite: ThermodynamicModel, temperature_profile: AbstractProfile, regular_range: Dict[str, Tuple] = default_regular_range)

Regularises the thermodynamic table by creating a regularised thermodynamic model that uses precomputed regular tables for S-wave and P-wave speeds.

Args: slb_pyrolite (ThermodynamicModel): The original thermodynamic model. temperature_profile (AbstractProfile): The temperature profile to be used for regularisation. This is supposed to be a 1D profile of average temperature profiles. regular_range (Dict[str, Tuple], optional): Dictionary specifying the regularisation range for each parameter. Defaults to gdrift.mineralogy.default_regular_range.

Returns: RegularisedThermodynamicModel: A regularised thermodynamic model with precomputed tables for S-wave and P-wave speeds.

Anelasticity¶

CammaranoAnelasticityModel¶

CammaranoAnelasticityModel ¶

CammaranoAnelasticityModel(B: Callable, g: Callable, a: Callable, solidus: SplineProfile, Q_bulk: Callable = lambda x: 10000, omega: Callable = lambda x: 1.0)

Bases: BaseAnelasticityModel

A specific implementation of an anelasticity model following the approach by Cammarano et al.

Initialize the model with the given parameters.

Args: B (Callable): Scaling factor for the Q model. g (Callable): Activation energy parameter. a (Callable): Frequency dependency parameter. solidus (SplineProfile): Solidus temperature profile for mantle. Q_bulk (Callable): Bulk quality factor (default is 10000). omega (Callable): Seismic frequency (default is 1).

from_q_profile `classmethod` ¶

from_q_profile(q_profile: str) -> CammaranoAnelasticityModel

Create a CammaranoAnelasticityModel from a predefined Q-profile.

Uses the six Q-profiles (Q1-Q6) from Cammarano et al. (2003), with upper/lower mantle parameter switching at 660 km depth. The Q factor is computed as:

Q = B * omega^a * exp(a * g * T_solidus / T)

where B controls the overall attenuation amplitude (related to grain size), g is a dimensionless activation energy parameter, a is the frequency exponent, and T_solidus is the solidus temperature at the given depth.

Q-profiles represent different assumptions about grain size, water content, and attenuation mechanisms in the mantle:

======= ============= ============= ====================================== Profile B (UM / LM) g (UM / LM) Physical Interpretation ======= ============= ============= ====================================== Q1 0.5 / 10 20 / 10 Fine grain size, low water content Q2 0.8 / 15 20 / 10 Medium grain size Q3 1.1 / 20 20 / 10 Coarse grain, higher water content Q4 0.035 / 2.25 30 / 15 SLB activation energy, fine grain Q5 0.056 / 3.6 30 / 15 SLB activation energy, medium grain Q6 0.077 / 4.95 30 / 15 SLB activation energy, coarse grain ======= ============= ============= ======================================

UM = upper mantle (< 660 km), LM = lower mantle (>= 660 km). Q1-Q3 use Cammarano et al. (2003) activation energy parameterization. Q4-Q6 use Stixrude & Lithgow-Bertelloni activation energy values. All profiles use a = 0.2 (frequency exponent) and omega = 1.0 Hz.

Args: q_profile (str): One of "Q1" through "Q6".

Returns: CammaranoAnelasticityModel: The configured model.

Raises: ValueError: If q_profile is not one of Q1-Q6.

GoesAnelasticityModel¶

GoesAnelasticityModel ¶

GoesAnelasticityModel(A, H_star, V_star, a, omega=1.0, Q_bulk=1000.0, max_depth=660000.0)

Bases: BaseAnelasticityModel

Anelasticity model following Goes et al. (2000, JGR).

Uses activation energy and activation volume to compute the shear quality factor Q_mu via:

Q_mu = A * omega^a * exp(a * (H* + P * V*) / (R * T))

where A is a pre-exponential factor, H is activation energy (J/mol), V is activation volume (m^3/mol), P is pressure from PREM, R is the gas constant, and T is temperature (K).

This model was calibrated for the upper mantle (50-200 km depth). Below 660 km depth, Q is set to a very large value (effectively no attenuation).

Initialize the Goes anelasticity model.

Args: A (float): Pre-exponential scaling factor. H_star (float): Activation energy in J/mol. V_star (float): Activation volume in m^3/mol. a (float): Frequency exponent. omega (float): Seismic frequency in Hz (default 1.0). Q_bulk (float): Constant bulk Q (default 1000, Durek & Ekstrom 1996). max_depth (float): Depth below which Q is set very high (default 660e3 m).

from_q_profile `classmethod` ¶

from_q_profile(q_profile: str) -> GoesAnelasticityModel

Create a GoesAnelasticityModel from a predefined Q-profile.

Uses the Goes et al. (2000) parameterization from Table A2 where the Q factor is computed as:

Q_mu = A * omega^a * exp(a * (H* + P * V*) / (R * T))

======= ====== ========== ============= =============== Profile a A H (kJ/mol) V (cm^3/mol) ======= ====== ========== ============= =============== Q1 0.15 0.148 500 20 Q2 0.25 2.0e-4 584 21 ======= ====== ========== ============= ===============

Args: q_profile (str): One of "Q1" or "Q2".

Returns: GoesAnelasticityModel: The configured model.

Raises: ValueError: If q_profile is not Q1 or Q2.

apply_anelastic_correction¶

apply_anelastic_correction ¶

apply_anelastic_correction(thermo_model, anelastic_model: BaseAnelasticityModel)

Apply anelastic corrections to seismic velocity data using the provided "anelastic_model" within the low attenuation limit. The corrections are based on the equation: \(1 - \frac{V(anelastic)}{V(elastic)} = \frac{1}{2} \cot(\frac{\alpha \pi}{2}) Q^{-1}\) as described by Stixrude & Lithgow-Bertelloni (doi:10.1029/2004JB002965, Eq-10).

thermo_model (ThermodynamicModel): The thermodynamic model containing temperature and depth data.

ThermodynamicModel: A new thermodynamic model with anelastically corrected seismic velocities.

The returned model includes the following methods with anelastic corrections:

compute_swave_speed: Calculates the anelastic effect on shear wave speed.
compute_pwave_speed: Calculates the anelastic effect on compressional wave speed.

The compute_swave_speed method applies the anelastic correction to the shear wave speed using the provided anelastic model. The compute_pwave_speed method applies the anelastic correction to the compressional wave speed using the quality factor derived from the equations provided by Don L. Anderson & R. S. Hart (1978, PEPI eq 1-3).

The corrections are applied by meshing the depths and temperatures from the thermodynamic model and computing the quality factor matrices for shear and bulk moduli. The corrected seismic velocities are then calculated and returned as new tables with the corrected values.

Utility Functions¶

Coordinate Transformations¶

geodetic_to_cartesian ¶

geodetic_to_cartesian(lat, lon, depth, earth_radius=R_earth)

Convert geographic coordinates to Cartesian coordinates.

Parameters: lat (float or numpy.ndarray): Latitude in degrees. lon (float or numpy.ndarray): Longitude in degrees. depth (float or numpy.ndarray): Depth below Earth's surface in km. earth_radius (float): Radius of the Earth in km. Default is 6371 km.

Returns: tuple: Cartesian coordinates (x, y, z).

cartesian_to_geodetic ¶

cartesian_to_geodetic(x, y, z, earth_radius=R_earth)

Convert Cartesian coordinates to geographic coordinates.

Parameters: x (float or numpy.ndarray): x coordinate in km. y (float or numpy.ndarray): y coordinate in km. z (float or numpy.ndarray): z coordinate in km. earth_radius (float): Radius of the Earth in km. Default is 6371e3 m.

Returns: tuple: Geographic coordinates (lat, lon, depth).

nondimensionalise_coords ¶

nondimensionalise_coords(x, y, z, R_nd_earth=2.22, R_nd_cmb=1.22)

Convert dimensional Cartesian coordinates to nondimensional form.

Applies linear radial scaling to map physical coordinates (meters) to a nondimensional reference frame. Commonly used in geodynamic simulations to improve numerical conditioning.

Parameters:

Name	Type	Description	Default
`x`	`float or array_like`	Cartesian coordinates in meters (origin at Earth's center).	required
`y`	`float or array_like`	Cartesian coordinates in meters (origin at Earth's center).	required
`z`	`float or array_like`	Cartesian coordinates in meters (origin at Earth's center).	required
`R_nd_earth`	`float`	Nondimensional radius for Earth's surface. Default is 2.22.	`2.22`
`R_nd_cmb`	`float`	Nondimensional radius for core-mantle boundary. Default is 1.22.	`1.22`

Returns:

Type	Description
`tuple of (float or ndarray)`	Nondimensionalized (x', y', z') coordinates.

Notes

Uses linear scaling: r' = a*r + b, where a and b are determined by mapping R_earth → R_nd_earth and R_cmb → R_nd_cmb. Angular coordinates (theta, phi) are preserved.

See Also

dimensionalise_coords : Inverse transformation

dimensionalise_coords ¶

dimensionalise_coords(x, y, z, R_nd_cmb=1.22, R_nd_earth=2.22)

Convert nondimensional Cartesian coordinates back to meters.

Inverse of nondimensionalise_coords. Maps nondimensional coordinates back to physical units (meters) using linear radial scaling.

Parameters:

Name	Type	Description	Default
`x`	`float or array_like`	Nondimensional Cartesian coordinates.	required
`y`	`float or array_like`	Nondimensional Cartesian coordinates.	required
`z`	`float or array_like`	Nondimensional Cartesian coordinates.	required
`R_nd_cmb`	`float`	Nondimensional radius for core-mantle boundary. Must match the value used in nondimensionalisation. Default is 1.22.	`1.22`
`R_nd_earth`	`float`	Nondimensional radius for Earth's surface. Must match the value used in nondimensionalisation. Default is 2.22.	`2.22`

Returns:

Type	Description
`tuple of (float or ndarray)`	Dimensional (x, y, z) coordinates in meters.

Notes

Uses inverse linear scaling: r = a*r' + b, where a and b are determined by mapping R_nd_earth → R_earth and R_nd_cmb → R_cmb.

See Also

nondimensionalise_coords : Forward transformation

Gravity and Pressure¶

compute_gravity ¶

compute_gravity(radius, mass_enclosed)

Compute gravitational acceleration at each radius based on the enclosed mass.

Args: radius (numpy.ndarray): Array of radii from the center. mass_enclosed (numpy.ndarray): Array of cumulative mass enclosed up to each radius.

Returns: numpy.ndarray: Array of gravitational acceleration at each radius.

compute_pressure ¶

compute_pressure(radius, density, gravity)

Calculate the hydrostatic pressure at each radius based on the density and gravitational acceleration.

Args: radius (numpy.ndarray): Array of radii from the center to the surface. density (numpy.ndarray): Array of densities at each radius. gravity (numpy.ndarray): Array of gravitational accelerations at each radius.

Returns: numpy.ndarray: Array of pressures calculated from the surface inward to each radius.

compute_mass ¶

compute_mass(radius, density)

Compute the mass enclosed within each radius using the cumulative trapezoidal rule.

Args: radius (numpy.ndarray): Array of radii from the center of the Earth or other celestial body. density (numpy.ndarray): Array of densities corresponding to each radius.

Returns: numpy.ndarray: Array of cumulative mass enclosed up to each radius.

Constants¶

R_earth `module-attribute` ¶

R_earth = 6371000.0

R_cmb `module-attribute` ¶

R_cmb = R_earth - 2890000.0

Usage Examples¶

See the Examples section for Jupyter notebooks demonstrating these APIs in action.

API Reference¶

Core Data Loading¶

load_dataset¶

load_dataset ¶

1D Radial Profiles¶

RadialEarthModel¶

RadialEarthModel ¶

get_profile ¶

RadialEarthModelFromFile¶

RadialEarthModelFromFile ¶

PreliminaryRefEarthModel¶

PreliminaryRefEarthModel ¶

HirschmannSolidus¶

HirschmannSolidus ¶

3D Seismic Models¶

SeismicModel¶

SeismicModel ¶

Thermodynamics¶

ThermodynamicModel¶

ThermodynamicModel ¶

available_tables ¶

temperature_to_property ¶

temperature_to_vs ¶

temperature_to_vp ¶

temperature_to_rho ¶

vs_to_temperature ¶

vp_to_temperature ¶

Phase Transition Regularization¶

regularise_thermodynamic_table ¶

Anelasticity¶

CammaranoAnelasticityModel¶

CammaranoAnelasticityModel ¶

from_q_profile classmethod ¶

GoesAnelasticityModel¶

GoesAnelasticityModel ¶

from_q_profile classmethod ¶

apply_anelastic_correction¶

apply_anelastic_correction ¶

Utility Functions¶

Coordinate Transformations¶

geodetic_to_cartesian ¶

cartesian_to_geodetic ¶

nondimensionalise_coords ¶

dimensionalise_coords ¶

Gravity and Pressure¶

compute_gravity ¶

compute_pressure ¶

compute_mass ¶

Constants¶

R_earth module-attribute ¶

R_cmb module-attribute ¶

Usage Examples¶

from_q_profile `classmethod` ¶

from_q_profile `classmethod` ¶

R_earth `module-attribute` ¶

R_cmb `module-attribute` ¶