compute mu

news/mu.rst

@@ -0,0 +1,23 @@
+**Added:**
+
+* function to compute x-ray attenuation coefficient (mu) using XrayDB
+
+**Changed:**
+
+* <news item>
+
+**Deprecated:**
+
+* <news item>
+
+**Removed:**
+
+* <news item>
+
+**Fixed:**
+
+* <news item>
+
+**Security:**
+
+* <news item>

requirements/conda.txt

@@ -1,2 +1,3 @@
 numpy
+xraydb
 scipy

requirements/pip.txt

@@ -1,2 +1,3 @@
 numpy
+xraydb
 scipy

src/diffpy/utils/tools.py

@@ -6,6 +6,7 @@
 import numpy as np
 from scipy.optimize import dual_annealing
 from scipy.signal import convolve
+from xraydb import material_mu
 
 from diffpy.utils.parsers.loaddata import loadData
 
@@ -194,6 +195,57 @@ def get_package_info(package_names, metadata=None):
  return metadata
 
 
+def get_density_from_cloud(sample_composition, mp_token=""):
+ """Function to get material density from the MP or COD database.
+
+ It is not implemented yet.
+ """
+ raise NotImplementedError(
+ "So sorry, density computation from composition is not implemented right now. "
+ "We hope to have this implemented in the next release. "
+ "Please rerun specifying a sample mass density."
+ )
+
+
+def compute_mu_using_xraydb(sample_composition, energy, sample_mass_density=None, packing_fraction=None):
+ """Compute the attenuation coefficient (mu) using the XrayDB database.
+
+ Computes mu based on the sample composition and energy.
+ User should provide a sample mass density or a packing fraction.
+ If neither density nor packing fraction is specified, or if both are specified, a ValueError will be raised.
+ Reference: https://xraypy.github.io/XrayDB/python.html#xraydb.material_mu.
+
+ Parameters
+ ----------
+ sample_composition : str
+ The chemical formula of the material.
+ energy : float
+ The energy of the incident x-rays in keV.
+ sample_mass_density : float, optional, Default is None
+ The mass density of the packed powder/sample in g/cm*3.
+ packing_fraction : float, optional, Default is None
+ The fraction of sample in the capillary (between 0 and 1).
+ Specify either sample_mass_density or packing_fraction but not both.
+
+ Returns
+ -------
+ mu : float
+ The attenuation coefficient mu in mm^{-1}.
+ """
+ if (sample_mass_density is None and packing_fraction is None) or (
+ sample_mass_density is not None and packing_fraction is not None
+ ):
+ raise ValueError(
+ "You must specify either sample_mass_density or packing_fraction, but not both. "
+ "Please rerun specifying only one."
+ )
+ if packing_fraction is not None:
+ sample_mass_density = get_density_from_cloud(sample_composition) * packing_fraction
+ energy_eV = energy * 1000
+ mu = material_mu(sample_composition, energy_eV, density=sample_mass_density, kind="total") / 10
+ return mu
+
+
 def _top_hat(z, half_slit_width):
  """Create a top-hat function, return 1.0 for values within the specified
  slit width and 0 otherwise."""

tests/test_tools.py

@@ -9,6 +9,7 @@
 from diffpy.utils.tools import (
  _extend_z_and_convolve,
  check_and_build_global_config,
+ compute_mu_using_xraydb,
  compute_mud,
  get_package_info,
  get_user_info,
@@ -169,6 +170,35 @@ def test_get_package_info(monkeypatch, inputs, expected):
  assert actual_metadata == expected
 
 
+@pytest.mark.parametrize(
+ "inputs",
+ [
+ # Test when the function has invalid inputs
+ ( # C1: Both mass density and packing fraction are provided, expect ValueError exception
+ {
+ "sample_composition": "SiO2",
+ "energy": 10,
+ "sample_mass_density": 2.65,
+ "packing_fraction": 1,
+ }
+ ),
+ ( # C2: None of mass density or packing fraction are provided, expect ValueError exception
+ {
+ "sample_composition": "SiO2",
+ "energy": 10,
+ }
+ ),
+ ],
+)
+def test_compute_mu_using_xraydb_bad(inputs):
+ with pytest.raises(
+ ValueError,
+ match="You must specify either sample_mass_density or packing_fraction, but not both. "
+ "Please rerun specifying only one.",
+ ):
+ compute_mu_using_xraydb(**inputs)
+
+
 def test_compute_mud(tmp_path):
  diameter, slit_width, z0, I0, mud, slope = 1, 0.1, 0, 1e5, 3, 0
  z_data = np.linspace(-1, 1, 50)