materials_for_mc 0.2.0

Material builder for Monte Carlo simulations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use crate::data::{ATOMIC_MASSES, ELEMENT_NAMES, ELEMENT_NUCLIDES, NATURAL_ABUNDANCE};
use pyo3::prelude::*;
use pyo3::types::PyDict;

#[pyfunction]
pub fn element_nuclides(py: Python) -> PyObject {
    let dict = PyDict::new(py);
    for (element, nuclides) in ELEMENT_NUCLIDES.iter() {
        let mut sorted_nuclides = nuclides.clone();
        sorted_nuclides.sort();
        dict.set_item(*element, sorted_nuclides).unwrap();
    }
    dict.into()
}

#[pyfunction]
pub fn natural_abundance(py: Python) -> PyObject {
    let dict = PyDict::new(py);
    for (k, v) in NATURAL_ABUNDANCE.iter() {
        dict.set_item(*k, v).unwrap();
    }
    dict.into()
}

#[pyfunction]
pub fn element_names(py: Python) -> PyObject {
    let dict = PyDict::new(py);
    for (symbol, name) in ELEMENT_NAMES.iter() {
        dict.set_item(*symbol, *name).unwrap();
    }
    dict.into()
}

#[pyfunction]
pub fn atomic_masses(py: Python) -> PyObject {
    let dict = PyDict::new(py);
    for (nuclide, mass) in ATOMIC_MASSES.iter() {
        dict.set_item(*nuclide, *mass).unwrap();
    }
    dict.into()
}