#[cfg(feature = "gc_pcsaft")]

use crate::gc_pcsaft::python::gc_pcsaft as gc_pcsaft_module;

#[cfg(feature = "pcsaft")]

use crate::pcsaft::python::pcsaft as pcsaft_module;

#[cfg(feature = "pets")]

use crate::pets::python::pets as pets_module;

#[cfg(feature = "saftvrqmie")]

use crate::saftvrqmie::python::saftvrqmie as saftvrqmie_module;

#[cfg(feature = "uvtheory")]

use crate::uvtheory::python::uvtheory as uvtheory_module;

use pyo3::prelude::*;

use pyo3::wrap_pymodule;

use quantity::python::quantity as quantity_module;

mod cubic;

mod eos;

mod ideal_gas;

use cubic::cubic as cubic_module;

use eos::eos as eos_module;

use ideal_gas::ideal_gas as ideal_gas_module;

#[cfg(feature = "dft")]

mod dft;

#[cfg(feature = "dft")]

use dft::dft as dft_module;

#[pymodule]

pub fn feos(py: Python<'_>, m: &PyModule) -> PyResult<()> {

m.add("__version__", env!("CARGO_PKG_VERSION"))?;

m.add_wrapped(wrap_pymodule!(quantity_module))?;

m.add_wrapped(wrap_pymodule!(eos_module))?;

#[cfg(feature = "dft")]

m.add_wrapped(wrap_pymodule!(dft_module))?;

m.add_wrapped(wrap_pymodule!(ideal_gas_module))?;

m.add_wrapped(wrap_pymodule!(cubic_module))?;

#[cfg(feature = "pcsaft")]

m.add_wrapped(wrap_pymodule!(pcsaft_module))?;

#[cfg(feature = "gc_pcsaft")]

m.add_wrapped(wrap_pymodule!(gc_pcsaft_module))?;

#[cfg(feature = "pets")]

m.add_wrapped(wrap_pymodule!(pets_module))?;

#[cfg(feature = "uvtheory")]

m.add_wrapped(wrap_pymodule!(uvtheory_module))?;

#[cfg(feature = "saftvrqmie")]

m.add_wrapped(wrap_pymodule!(saftvrqmie_module))?;

// set_path(py, m, "feos.si", "quantity")?;

// set_path(py, m, "feos.eos", "eos")?;

// #[cfg(feature = "estimator")]

// set_path(py, m, "feos.eos.estimator", "eos.estimator_eos")?;

// #[cfg(feature = "dft")]

// set_path(py, m, "feos.dft", "dft")?;

// #[cfg(all(feature = "dft", feature = "estimator"))]

// set_path(py, m, "feos.dft.estimator", "dft.estimator_dft")?;

// set_path(py, m, "feos.ideal_gas", "ideal_gas")?;

// set_path(py, m, "feos.cubic", "cubic")?;

// #[cfg(feature = "pcsaft")]

// set_path(py, m, "feos.pcsaft", "pcsaft")?;

// #[cfg(feature = "gc_pcsaft")]

// set_path(py, m, "feos.gc_pcsaft", "gc_pcsaft")?;

// #[cfg(feature = "pets")]

// set_path(py, m, "feos.pets", "pets")?;

// #[cfg(feature = "uvtheory")]

// set_path(py, m, "feos.uvtheory", "uvtheory")?;

// #[cfg(feature = "saftvrqmie")]

// set_path(py, m, "feos.saftvrqmie", "saftvrqmie")?;

// py.run(

// "\

// import sys

// quantity.SINumber.__module__ = 'feos.si'

// quantity.SIArray1.__module__ = 'feos.si'

// quantity.SIArray2.__module__ = 'feos.si'

// quantity.SIArray3.__module__ = 'feos.si'

// quantity.SIArray4.__module__ = 'feos.si'

// ",

// None,

// Some(m.dict()),

// )?;

Ok(())

}

fn set_path(py: Python<'_>, m: &PyModule, path: &str, module: &str) -> PyResult<()> {

py.run(

&format!(

"\

import sys

sys.modules['{path}'] = {module}

"

),

None,

Some(m.dict()),

)

}