mirror of
https://github.com/slsdetectorgroup/slsDetectorPackage.git
synced 2025-04-25 15:50:03 +02:00
30 lines
927 B
C++
30 lines
927 B
C++
#include <pybind11/chrono.h>
|
|
#include <pybind11/numpy.h>
|
|
#include <pybind11/operators.h>
|
|
#include <pybind11/pybind11.h>
|
|
#include <pybind11/stl.h>
|
|
|
|
#include "sls/Pattern.h"
|
|
#include "sls/sls_detector_defs.h"
|
|
namespace py = pybind11;
|
|
void init_pattern(py::module &m) {
|
|
|
|
using pat = sls::patternParameters;
|
|
py::class_<pat> patternParameters(m, "patternParameters");
|
|
|
|
PYBIND11_NUMPY_DTYPE(pat, word, ioctrl, limits, loop, nloop, wait,
|
|
waittime);
|
|
|
|
patternParameters.def(py::init());
|
|
patternParameters.def("numpy_view", [](py::object &obj) {
|
|
pat &o = obj.cast<pat &>();
|
|
return py::array_t<pat>(1, &o, obj);
|
|
});
|
|
|
|
py::class_<sls::Pattern> Pattern(m, "Pattern");
|
|
Pattern.def(py::init());
|
|
Pattern.def("load", &sls::Pattern::load);
|
|
Pattern.def("data", (pat * (sls::Pattern::*)()) & sls::Pattern::data,
|
|
py::return_value_policy::reference);
|
|
}
|