mirror of
https://github.com/slsdetectorgroup/slsDetectorPackage.git
synced 2025-06-12 21:07:13 +02:00
Python binding for Pattern
This commit is contained in:
@ -2,6 +2,7 @@ import _slsdet
|
||||
|
||||
from _slsdet import Pattern
|
||||
|
||||
|
||||
class patternParameters(_slsdet.patternParameters):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
@ -15,13 +16,38 @@ class patternParameters(_slsdet.patternParameters):
|
||||
raise KeyError(f"Key: {name} not found")
|
||||
|
||||
def __setattr__(self, name, value):
|
||||
if name in ['view', 'names']:
|
||||
if name in ["view", "names"]:
|
||||
self.__dict__[name] = value
|
||||
elif name in self.names:
|
||||
self.view[name] = value
|
||||
else:
|
||||
else:
|
||||
raise KeyError(f"Key: {name} not found")
|
||||
|
||||
#Provide custom dir for tab completion
|
||||
# Provide custom dir for tab completion
|
||||
def __dir__(self):
|
||||
return self.names
|
||||
|
||||
|
||||
class Pattern(_slsdet.Pattern):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.view = self.data().numpy_view()
|
||||
self.names = self.view.dtype.names
|
||||
|
||||
def __getattr__(self, name):
|
||||
if name in self.names:
|
||||
return self.view[name][0]
|
||||
else:
|
||||
raise KeyError(f"Key: {name} not found")
|
||||
|
||||
def __setattr__(self, name, value):
|
||||
if name in ["view", "names"]:
|
||||
self.__dict__[name] = value
|
||||
elif name in self.names:
|
||||
self.view[name] = value
|
||||
else:
|
||||
raise KeyError(f"Key: {name} not found")
|
||||
|
||||
# Provide custom dir for tab completion
|
||||
def __dir__(self):
|
||||
return self.names
|
@ -20,23 +20,28 @@ void init_pattern(py::module &m) {
|
||||
pat &o = obj.cast<pat &>();
|
||||
return py::array_t<pat>(1, &o, obj);
|
||||
});
|
||||
//.def_readwrite("name", &Pet::name)
|
||||
// patternParameters.def_property(
|
||||
// "some",
|
||||
// [](py::object &obj) {
|
||||
// pat &o = obj.cast<pat &>();
|
||||
// return py::array_t<pat>(1, &o, obj);
|
||||
// },
|
||||
// [](py::object &obj) {
|
||||
// pat &o = obj.cast<pat &>();
|
||||
// return py::array_t<pat>(1, &o, obj);
|
||||
// });
|
||||
|
||||
// m.def("get_memoryview1d", []() {
|
||||
// return py::memoryview::from_memory(
|
||||
// buffer, // buffer pointer
|
||||
// sizeof(uint8_t) * 8 // buffer size
|
||||
// );
|
||||
// })
|
||||
// patternParameters.def("load", &pat::load);
|
||||
// patternParameters.def_property_readonly(
|
||||
// "loop",
|
||||
// [](py::object &obj) {
|
||||
// pat &o = obj.cast<pat &>();
|
||||
// return py::array_t<uint32_t>(6, &o.loop[0], 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);
|
||||
|
||||
// m.def("get_memoryview1d", []() {
|
||||
// return py::memoryview::from_memory(data(), // buffer pointer
|
||||
// sizeof(pat) // buffer size
|
||||
// );
|
||||
// });
|
||||
Pattern.def("data", (pat * (sls::Pattern::*)()) & sls::Pattern::data,
|
||||
py::return_value_policy::reference);
|
||||
}
|
||||
|
Reference in New Issue
Block a user