Emily Boudreaux
7fded59814
The python bindings now work with the polymorphic reaction class and the CVODE solver
84 lines
2.6 KiB
C++
84 lines
2.6 KiB
C++
#include <pybind11/pybind11.h>
|
|
#include <pybind11/stl_bind.h> // Needed for binding std::vector, std::map etc. if needed directly
|
|
|
|
#include "bindings.h"
|
|
|
|
#include "gridfire/utils/general_composition.h"
|
|
#include "gridfire/utils/hashing.h"
|
|
|
|
namespace py = pybind11;
|
|
|
|
#include "gridfire/utils/logging.h"
|
|
|
|
void register_utils_bindings(py::module &m) {
|
|
m.def("formatNuclearTimescaleLogString",
|
|
&gridfire::utils::formatNuclearTimescaleLogString,
|
|
py::arg("engine"),
|
|
py::arg("Y"),
|
|
py::arg("T9"),
|
|
py::arg("rho"),
|
|
"Format a string for logging nuclear timescales based on temperature, density, and energy generation rate."
|
|
);
|
|
|
|
m.def(
|
|
"massFractionFromMolarAbundanceAndComposition",
|
|
&gridfire::utils::massFractionFromMolarAbundanceAndComposition,
|
|
py::arg("composition"),
|
|
py::arg("species"),
|
|
py::arg("Yi"),
|
|
"Convert a specific species molar abundance into its mass fraction if it were present in a given composition."
|
|
);
|
|
|
|
m.def(
|
|
"massFractionFromMolarAbundanceAndMolarMass",
|
|
&gridfire::utils::massFractionFromMolarAbundanceAndMolarMass,
|
|
py::arg("molarAbundances"),
|
|
py::arg("molarMasses"),
|
|
"Convert a vector of molar abundances and a parallel vector of molar masses into a vector of mass fractions"
|
|
);
|
|
|
|
m.def(
|
|
"molarMassVectorFromComposition",
|
|
&gridfire::utils::molarMassVectorFromComposition,
|
|
py::arg("composition"),
|
|
"Extract vector of molar masses from a composition object, this will be sorted by species mass so that the lightest species are at the front of the list."
|
|
);
|
|
|
|
m.def(
|
|
"hash_atomic",
|
|
&gridfire::utils::hash_atomic,
|
|
py::arg("a"),
|
|
py::arg("z")
|
|
);
|
|
|
|
auto hashing_module = m.def_submodule("hashing", "module for gridfire hashing functions");
|
|
auto reaction_hashing_module = hashing_module.def_submodule("reaction", "utility module for hashing gridfire reaction functions");
|
|
|
|
reaction_hashing_module.def(
|
|
"splitmix64",
|
|
&gridfire::utils::hashing::reaction::splitmix64,
|
|
py::arg("x")
|
|
);
|
|
|
|
reaction_hashing_module.def(
|
|
"mix_species",
|
|
&gridfire::utils::hashing::reaction::mix_species,
|
|
py::arg("a"),
|
|
py::arg("z")
|
|
);
|
|
|
|
reaction_hashing_module.def(
|
|
"multiset_combine",
|
|
&gridfire::utils::hashing::reaction::multiset_combine,
|
|
py::arg("acc"),
|
|
py::arg("x")
|
|
);
|
|
|
|
m.def(
|
|
"hash_reaction",
|
|
&gridfire::utils::hash_reaction,
|
|
py::arg("reaction")
|
|
);
|
|
|
|
}
|