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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

// First, import any modules and re-export the types for Rust usage
mod config;
mod element;
mod material;
mod materials;
mod nuclide;
mod reaction;
mod utilities;
mod url_cache;

pub use config::Config;
pub use element::Element;
pub use material::Material;
pub use materials::Materials;
pub use reaction::Reaction;
pub use nuclide::Nuclide;
pub use utilities::{interpolate_linear, interpolate_log_log};

// Import PyO3 items conditionally
#[cfg(feature = "pyo3")]
use pyo3::prelude::*;
#[cfg(feature = "pyo3")]
use pyo3::pymodule;
#[cfg(feature = "pyo3")]
use pyo3::wrap_pyfunction;

// Conditionally include the Python modules
#[cfg(feature = "pyo3")]
mod config_python;
#[cfg(feature = "pyo3")]
mod element_python;
#[cfg(feature = "pyo3")]
mod material_python;
#[cfg(feature = "pyo3")]
mod materials_python;
#[cfg(feature = "pyo3")]
mod nuclide_python;
#[cfg(feature = "pyo3")]
mod reaction_python;

// Re-export Python modules for Maturin to find
#[cfg(feature = "pyo3")]
pub use config_python::*;
#[cfg(feature = "pyo3")]
pub use element_python::*;
#[cfg(feature = "pyo3")]
pub use material_python::*;
#[cfg(feature = "pyo3")]
pub use materials_python::*;
#[cfg(feature = "pyo3")]
pub use nuclide_python::*;
#[cfg(feature = "pyo3")]
pub use reaction_python::*;

// Declare the data module
pub mod data;

#[cfg(feature = "wasm")]
mod data_wasm;
#[cfg(feature = "wasm")]
pub use data_wasm::*;

#[cfg(feature = "pyo3")]
mod data_python;
#[cfg(feature = "pyo3")]
pub use data_python::*;

// WASM Modules
#[cfg(feature = "wasm")]
pub mod config_wasm;
#[cfg(feature = "wasm")]
pub mod element_wasm;
#[cfg(feature = "wasm")]
pub mod material_wasm;
#[cfg(feature = "wasm")]
pub mod nuclide_wasm;
#[cfg(feature = "wasm")]
pub mod reaction_wasm;

// WASM setup
#[cfg(feature = "wasm")]
use wasm_bindgen::prelude::*;

#[cfg(feature = "wasm")]
#[wasm_bindgen(start)]
pub fn wasm_start() {
    console_error_panic_hook::set_once();
}

// Export WASM modules
#[cfg(feature = "wasm")]
pub use config_wasm::*;
#[cfg(feature = "wasm")]
pub use element_wasm::*;
#[cfg(feature = "wasm")]
pub use material_wasm::*;
#[cfg(feature = "wasm")]
pub use nuclide_wasm::*;
#[cfg(feature = "wasm")]
pub use reaction_wasm::*;

// If you have a main Python module entry point, update it to include PyMaterials:
#[cfg(feature = "pyo3")]
#[pymodule]
fn materials_for_mc(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
    m.add_class::<material_python::PyMaterial>()?;
    m.add_class::<materials_python::PyMaterials>()?;
    m.add_class::<nuclide_python::PyNuclide>()?;
    m.add_class::<reaction_python::PyReaction>()?; // Exposed as Reaction in Python
    m.add_class::<config_python::PyConfig>()?;
    m.add_class::<element_python::PyElement>()?;
    m.add_function(wrap_pyfunction!(
        nuclide_python::py_read_nuclide_from_json,
        m
    )?)?;
    m.add_function(wrap_pyfunction!(nuclide_python::clear_nuclide_cache, m)?)?;
    m.add_function(wrap_pyfunction!(crate::data_python::natural_abundance, m)?)?;
    m.add_function(wrap_pyfunction!(crate::data_python::element_nuclides, m)?)?;
    m.add_function(wrap_pyfunction!(crate::data_python::element_names, m)?)?;
    m.add_function(wrap_pyfunction!(crate::data_python::atomic_masses, m)?)?;
    Ok(())
}