roland/xo-alloc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

xopyunit: + multiply + a few mass units

Browse source
This commit is contained in:
Roland Conybeare 2024-05-01 08:11:16 -05:00
commit 9bea04bfa2
2 changed files with 52 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

53
src/pyunit/pyunit.cpp
View file

@ -3,6 +3,7 @@
// note: need pyreflect/ here bc pyreflect.hpp is generated, located in build directory
#include "pyunit.hpp"
#include "xo/unit/xquantity.hpp"
#include "xo/unit/xquantity_iostream.hpp"
#include "xo/pyutil/pyutil.hpp"
//#include <pybind11/pybind11.h>
//#include <pybind11/stl.h>
@ -11,17 +12,63 @@
namespace xo {
namespace py = pybind11;
using Unit = xo::qty::natural_unit<std::int64_t>;
using XoQuantity = xo::qty::Quantity<double, std::int64_t>;
namespace qty {
PYBIND11_MODULE(PYUNIT_MODULE_NAME(), m) {
py::class_<Quantity>(m, "Quantity")
m.doc() = "pybind11 plugin for xo.unit";
py::class_<Unit>(m, "Unit")
.def("__repr__",
[](Quantity & x)
[](Unit & x)
{
/* e.g. "1g" for xo::qty::nu::gram */
return tostr(1, x.abbrev());
})
;
py::module unit = m.def_submodule("unit");
unit.attr("microgram") = &xo::qty::nu::microgram;
unit.attr("milligram") = &xo::qty::nu::milligram;
unit.attr("gram") = &xo::qty::nu::gram;
unit.attr("kilogram") = &xo::qty::nu::kilogram;
unit.attr("tonne") = &xo::qty::nu::tonne;
py::module qty = m.def_submodule("qty");
qty.def("micrograms", [](double x) { return XoQuantity(x, nu::microgram); });
qty.def("milligrams", [](double x) { return XoQuantity(x, nu::milligram); });
qty.def("grams", [](double x) { return XoQuantity(x, nu::gram); });
qty.def("kilograms", [](double x) { return XoQuantity(x, nu::kilogram); });
qty.def("tonnes", [](double x) { return XoQuantity(x, nu::tonne); });
py::class_<XoQuantity>(m, "Quantity")
.def(py::init<double, const Unit &>(),
py::arg("scale"), py::arg("unit"))
.def("__mul__",
[](const XoQuantity & x, const XoQuantity & y)
{
return x * y;
})
.def("__mul__",
[](const XoQuantity & x, double y)
{
return x * y;
})
.def("__rmul__",
[](const XoQuantity & y, double x)
{
return x * y;
})
.def("__repr__",
[](const XoQuantity & x)
{
return tostr(x);
})
py::arg("qty"));
;
}
}
} /*namespace xo*/