detectors/aare
25
0
Fork 0
mirror of https://github.com/slsdetectorgroup/aare.git synced 2025-06-11 06:47:14 +02:00
Code Issues Actions Packages Releases Activity

frame reading for cluster file

Browse Source
This commit is contained in:
Erik Fröjdh
2024-11-15 16:13:46 +01:00
parent e77b615293
commit 17f8d28019
8 changed files with 84 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/examples/play.py
View File

@ -6,8 +6,10 @@ from aare import ClusterFile
base = Path('~/data/aare_test_data/clusters').expanduser()
# f = ClusterFile(base / 'beam_En700eV_-40deg_300V_10us_d0_f0_100.clust')
f = ClusterFile(base / 'single_frame_97_clustrers.clust')
f = ClusterFile(base / 'beam_En700eV_-40deg_300V_10us_d0_f0_100.clust')
# f = ClusterFile(base / 'single_frame_97_clustrers.clust')
for i in range(10):
fn, cl = f.read_frame()
print(fn, cl.size)

6
python/src/cluster_file.hpp
View File

@ -24,6 +24,12 @@ void define_cluster_file_io_bindings(py::module &m) {
auto* vec = new std::vector<Cluster>(self.read_clusters(n_clusters));
return return_vector(vec);
})
.def("read_frame",
[](ClusterFile &self) {
int32_t frame_number;
auto* vec = new std::vector<Cluster>(self.read_frame(frame_number));
return py::make_tuple(frame_number, return_vector(vec));
})
.def("read_cluster_with_cut",
[](ClusterFile &self, size_t n_clusters, py::array_t<double> noise_map, int nx, int ny) {
auto view = make_view_2d(noise_map);