Source code for burnman.classes.perplex

# This file is part of BurnMan - a thermoelastic and thermodynamic toolkit for the Earth and Planetary Sciences
# Copyright (C) 2012 - 2023 by the BurnMan team, released under the GNU
# GPL v2 or later.

from __future__ import absolute_import
from __future__ import print_function

import time
import subprocess
from os import rename

import numpy as np
from scipy.interpolate import RegularGridInterpolator
from scipy.interpolate import griddata

from .material import Material, material_property

from ..utils.misc import copy_documentation


def create_perplex_table(
    werami_path,
    project_name,
    outfile,
    n_pressures,
    n_temperatures,
    pressure_range=None,
    temperature_range=None,
):
    """
    This function uses PerpleX's werami software to output a table file
    containing the following material properties.
    2 - Density (kg/m3)
    4 - Expansivity (1/K, for volume)
    5 - Compressibility (1/bar, for volume)
    10 - Adiabatic bulk modulus (bar)
    11 - Adiabatic shear modulus (bar)
    12 - Sound velocity (km/s)
    13 - P-wave velocity (Vp, km/s)
    14 - S-wave velocity (Vs, km/s)
    17 - Entropy (J/K/kg)
    18 - Enthalpy (J/kg)
    19 - Heat Capacity (J/K/kg)
    22 - Molar Volume (J/bar)

    The user must already have a PerpleX build file,
    and have run vertex on that build file.
    """

    print(
        "Creating a {0}x{1} P-T table file using werami. Please wait.\n".format(
            n_pressures, n_temperatures
        )
    )

    try:
        str2 = "y\n{0} {1}\n{2} {3}\n".format(
            pressure_range[0] / 1.0e5,
            pressure_range[1] / 1.0e5,
            temperature_range[0],
            temperature_range[1],
        )
    except TypeError:
        print("Keeping P-T range the same as the original project range.\n")
        str2 = "n\n"

    stdin = (
        "{0:s}\n2\n"
        "2\nn\n"
        "4\nn\n"
        "5\nn\n"
        "10\nn\n"
        "11\nn\n"
        "12\nn\n"
        "13\nn\n"
        "14\nn\n"
        "17\nn\n"
        "18\nn\n"
        "19\nn\n"
        "22\nn\n"
        "0\n"
        "{1:s}"
        "{2:d} {3:d}\n"
        "0\n".format(project_name, str2, n_pressures, n_temperatures)
    )

    with subprocess.Popen(
        werami_path,
        stdin=subprocess.PIPE,
        stdout=subprocess.PIPE,
        encoding="utf8",
    ) as process:
        process.stdin.write(stdin)
        process.stdin.flush()

        out = ""
        # Grab stdout line by line as it becomes available.
        # This will loop until the process terminates.
        stdoutput = ""
        while process.poll() is not None:
            line = process.stdout.readline()
            stdoutput += line

            # Check if vertex has been run on the build file
            if "missing *.tof file" in line:
                raise Exception(
                    "You must run Perple_X vertex "
                    f"({werami_path[0].split('werami')[0]}vertex) "
                    "using the PerpleX build file ({project_name}) "
                    "before running this script."
                )

        while process.poll() is None:
            line = process.stdout.readline()
            stdoutput += line

            # Check if werami is trying to create a standard resolution grid
            # Tell the user to modify their local perplex option file if so.
            if "Continue (y/n)?" in line:
                raise Exception(
                    "If you do not want to define your own P-T range for the grid,\n"
                    "you must set sample_on_grid to F in the perplex option file\n"
                    "(default is perplex_option.dat)."
                )

            # Get the output file name
            if "Output has been written to the" in line:
                out = line.split()[-1]

        # Print stdoutput
        print(stdoutput)
        print(process.stdout.read())

        # Rename the file to the user-specified filename
        rename(out, outfile)
        print("Output file renamed to {0:s}".format(outfile))
        print("Processing complete")


[docs]class PerplexMaterial(Material): """ This is the base class for a PerpleX material. States of the material can only be queried after setting the pressure and temperature using set_state(). Instances of this class are initialised with a 2D PerpleX tab file. This file should be in the standard format (as output by werami), and should have columns with the following names: 'rho,kg/m3', 'alpha,1/K', 'beta,1/bar', 'Ks,bar', 'Gs,bar', 'v0,km/s', 'vp,km/s', 'vs,km/s', 's,J/K/kg', 'h,J/kg', 'cp,J/K/kg', 'V,J/bar/mol'. The order of these names is not important. Properties of the material are determined by linear interpolation from the PerpleX grid. They are all returned in SI units on a molar basis, even though the PerpleX tab file is not in these units. This class is available as ``burnman.PerplexMaterial``. """ def __init__(self, tab_file, name="Perple_X material"): self.name = name self.params = {"name": name} ( self._property_interpolators, self.params["molar_mass"], self.bounds, ) = self._read_2D_perplex_file(tab_file) Material.__init__(self) def _read_2D_perplex_file(self, filename): with open(filename, "r") as f: datastream = f.read() lines = [ line.strip().split() for line in datastream.split("\n") if line.strip() ] if lines[2][0] != "2": raise Exception("This is not a 2D PerpleX table") bounds = [ (float(lines[4][0]), float(lines[5][0]), int(lines[6][0])), (float(lines[8][0]), float(lines[9][0]), int(lines[10][0])), ] if lines[3][0] == "P(bar)" and lines[7][0] == "T(K)": Pmin, Pint, nP = bounds[0] Tmin, Tint, nT = bounds[1] elif lines[3][0] == "T(K)" and lines[7][0] == "P(bar)": Pmin, Pint, nP = bounds[1] Tmin, Tint, nT = bounds[0] else: raise Exception( "This file does not have a recognised PerpleX structure.\n" "Are the independent variables P(bar) and T(K)?" ) Pmin = Pmin * 1.0e5 # bar to Pa Pint = Pint * 1.0e5 # bar to Pa Pmax = Pmin + Pint * (nP - 1.0) Tmax = Tmin + Tint * (nT - 1.0) pressures = np.linspace(Pmin, Pmax, nP) temperatures = np.linspace(Tmin, Tmax, nT) n_properties = int(lines[11][0]) property_list = lines[12] # property_table[i][j][k] returns the kth property at the ith pressure and jth temperature table = np.array( [[float(string) for string in line] for line in lines[13 : 13 + nP * nT]] ) if lines[3][0] == "P(bar)": property_table = np.swapaxes(table.reshape(nT, nP, n_properties), 0, 1) else: property_table = table.reshape(nP, nT, n_properties) ordered_property_list = [ "rho,kg/m3", "alpha,1/K", "beta,1/bar", "Ks,bar", "Gs,bar", "v0,km/s", "vp,km/s", "vs,km/s", "s,J/K/kg", "h,J/kg", "cp,J/K/kg", "V,J/bar/mol", ] p_indices = [ i for i, p in enumerate(property_list) for ordered_p in ordered_property_list if p == ordered_p ] properties = {} for i, p_idx in enumerate(p_indices): # Fill in NaNs as long as they aren't in the corners of the P-T grid a = np.array(property_table[:, :, [p_idx]][:, :, 0]) x, y = np.indices(a.shape) a[np.isnan(a)] = griddata( (x[~np.isnan(a)], y[~np.isnan(a)]), # points we know a[~np.isnan(a)], # values we know (x[np.isnan(a)], y[np.isnan(a)]), ) # Fill any remaining NaNs with zeros properties[ordered_property_list[i]] = np.nan_to_num(a, 0.0) densities = properties["rho,kg/m3"] volumes = 1.0e-5 * properties["V,J/bar/mol"] molar_masses = densities * volumes molar_mass = np.mean(molar_masses) property_interpolators = { "rho": RegularGridInterpolator( (pressures, temperatures), densities, bounds_error=True ), "alpha": RegularGridInterpolator( (pressures, temperatures), properties["alpha,1/K"], bounds_error=True ), "K_T": RegularGridInterpolator( (pressures, temperatures), 1.0e5 / properties["beta,1/bar"], bounds_error=True, ), "K_S": RegularGridInterpolator( (pressures, temperatures), 1.0e5 * properties["Ks,bar"], bounds_error=True, ), "G_S": RegularGridInterpolator( (pressures, temperatures), 1.0e5 * properties["Gs,bar"], bounds_error=True, ), "bulk_sound_velocity": RegularGridInterpolator( (pressures, temperatures), 1.0e3 * properties["v0,km/s"], bounds_error=True, ), "p_wave_velocity": RegularGridInterpolator( (pressures, temperatures), 1.0e3 * properties["vp,km/s"], bounds_error=True, ), "s_wave_velocity": RegularGridInterpolator( (pressures, temperatures), 1.0e3 * properties["vs,km/s"], bounds_error=True, ), "S": RegularGridInterpolator( (pressures, temperatures), properties["s,J/K/kg"] * molar_masses, bounds_error=True, ), "H": RegularGridInterpolator( (pressures, temperatures), properties["h,J/kg"] * molar_masses, bounds_error=True, ), "C_p": RegularGridInterpolator( (pressures, temperatures), properties["cp,J/K/kg"] * molar_masses, bounds_error=True, ), "V": RegularGridInterpolator( (pressures, temperatures), volumes, bounds_error=True ), } bounds = [[Pmin, Pmax], [Tmin, Tmax]] return property_interpolators, molar_mass, bounds @copy_documentation(Material.set_state) def set_state(self, pressure, temperature): if not np.logical_and( np.all(self.bounds[0][0] <= pressure), np.all(pressure <= self.bounds[0][1]) ): raise ValueError( "The set_state pressure ({0:.4f}) is outside the bounds of this rock ({1:.4f}-{2:.4f} GPa)".format( pressure, self.bounds[0][0] / 1.0e9, self.bounds[0][1] / 1.0e9 ) ) if not np.logical_and( np.all(self.bounds[1][0] <= temperature), np.all(temperature <= self.bounds[1][1]), ): raise ValueError( "The set_state temperature ({0:.1f}) is outside the bounds of this rock ({1:.1f}-{2:.1f} K)".format( temperature, self.bounds[1][0], self.bounds[1][1] ) ) Material.set_state(self, pressure, temperature) """ Properties by linear interpolation of Perple_X output """ @material_property @copy_documentation(Material.molar_volume) def molar_volume(self): return self._property_interpolators["V"]([self.pressure, self.temperature])[0] @material_property @copy_documentation(Material.molar_enthalpy) def molar_enthalpy(self): return self._property_interpolators["H"]([self.pressure, self.temperature])[0] @material_property @copy_documentation(Material.molar_entropy) def molar_entropy(self): return self._property_interpolators["S"]([self.pressure, self.temperature])[0] @material_property @copy_documentation(Material.isothermal_bulk_modulus) def isothermal_bulk_modulus(self): return self._property_interpolators["K_T"]([self.pressure, self.temperature])[0] @material_property @copy_documentation(Material.adiabatic_bulk_modulus) def adiabatic_bulk_modulus(self): return self._property_interpolators["K_S"]([self.pressure, self.temperature])[0] @material_property @copy_documentation(Material.molar_heat_capacity_p) def molar_heat_capacity_p(self): return self._property_interpolators["C_p"]([self.pressure, self.temperature])[0] @material_property @copy_documentation(Material.thermal_expansivity) def thermal_expansivity(self): return self._property_interpolators["alpha"]([self.pressure, self.temperature])[ 0 ] @material_property @copy_documentation(Material.shear_modulus) def shear_modulus(self): return self._property_interpolators["G_S"]([self.pressure, self.temperature])[0] @material_property @copy_documentation(Material.p_wave_velocity) def p_wave_velocity(self): return self._property_interpolators["p_wave_velocity"]( [self.pressure, self.temperature] )[0] @material_property @copy_documentation(Material.bulk_sound_velocity) def bulk_sound_velocity(self): return self._property_interpolators["bulk_sound_velocity"]( [self.pressure, self.temperature] )[0] @material_property @copy_documentation(Material.shear_wave_velocity) def shear_wave_velocity(self): return self._property_interpolators["s_wave_velocity"]( [self.pressure, self.temperature] )[0] """ Properties from mineral parameters, Legendre transformations or Maxwell relations """ @material_property @copy_documentation(Material.molar_gibbs) def molar_gibbs(self): return self.molar_enthalpy - self.temperature * self.molar_entropy @material_property @copy_documentation(Material.molar_mass) def molar_mass(self): if "molar_mass" in self.params: return self.params["molar_mass"] else: raise ValueError( "No molar_mass parameter for mineral " + self.to_string + "." ) @material_property @copy_documentation(Material.density) def density(self): return self._property_interpolators["rho"]([self.pressure, self.temperature])[0] @material_property @copy_documentation(Material.molar_internal_energy) def molar_internal_energy(self): return ( self.molar_gibbs - self.pressure * self.molar_volume + self.temperature * self.molar_entropy ) @material_property @copy_documentation(Material.molar_helmholtz) def molar_helmholtz(self): return self.molar_gibbs - self.pressure * self.molar_volume @material_property @copy_documentation(Material.isothermal_compressibility) def isothermal_compressibility(self): return 1.0 / self.isothermal_bulk_modulus @material_property @copy_documentation(Material.adiabatic_compressibility) def adiabatic_compressibility(self): return 1.0 / self.adiabatic_bulk_modulus @material_property @copy_documentation(Material.molar_heat_capacity_v) def molar_heat_capacity_v(self): return ( self.molar_heat_capacity_p - self.molar_volume * self.temperature * self.thermal_expansivity * self.thermal_expansivity * self.isothermal_bulk_modulus ) @material_property @copy_documentation(Material.grueneisen_parameter) def grueneisen_parameter(self): return ( self.thermal_expansivity * self.molar_volume * self.adiabatic_bulk_modulus / self.molar_heat_capacity_p )