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

macro for bindings and better lookup

Browse Source
This commit is contained in:
froejdh_e 2025-06-03 09:33:39 +02:00
parent 69964e08d5
commit a0849feca7
2 changed files with 129 additions and 82 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

75
python/aare/ClusterFinder.py
View File

@ -1,22 +1,47 @@
from ._aare import ClusterFinder_Cluster3x3i, ClusterFinder_Cluster2x2i, ClusterFinderMT_Cluster3x3i, ClusterFinderMT_Cluster2x2i, ClusterCollector_Cluster3x3i, ClusterCollector_Cluster2x2i # from ._aare import ClusterFinder_Cluster3x3i, ClusterFinder_Cluster2x2i, ClusterFinderMT_Cluster3x3i, ClusterFinderMT_Cluster2x2i, ClusterCollector_Cluster3x3i, ClusterCollector_Cluster2x2i
from ._aare import ClusterFileSink_Cluster3x3i, ClusterFileSink_Cluster2x2i # from ._aare import ClusterFileSink_Cluster3x3i, ClusterFileSink_Cluster2x2i
from . import _aare
import numpy as np import numpy as np
_supported_cluster_sizes = [(2,2), (3,3), (5,5), (7,7), (9,9),]
# def _get_class()
def _type_to_char(dtype):
if dtype == np.int32:
return 'i'
elif dtype == np.float32:
return 'f'
elif dtype == np.float64:
return 'd'
else:
raise ValueError(f"Unsupported dtype: {dtype}. Only np.int32, np.float32, and np.float64 are supported.")
def _get_class(name, cluster_size, dtype):
"""
Helper function to get the class based on the name, cluster size, and dtype.
"""
try:
class_name = f"{name}_Cluster{cluster_size[0]}x{cluster_size[1]}{_type_to_char(dtype)}"
cls = getattr(_aare, class_name)
except AttributeError:
raise ValueError(f"Unsupported combination of type and cluster size: {dtype}/{cluster_size} when requesting {class_name}")
return cls
def ClusterFinder(image_size, cluster_size, n_sigma=5, dtype = np.int32, capacity = 1024): def ClusterFinder(image_size, cluster_size, n_sigma=5, dtype = np.int32, capacity = 1024):
""" """
Factory function to create a ClusterFinder object. Provides a cleaner syntax for Factory function to create a ClusterFinder object. Provides a cleaner syntax for
the templated ClusterFinder in C++. the templated ClusterFinder in C++.
""" """
if dtype == np.int32 and cluster_size == (3,3): cls = _get_class("ClusterFinder", cluster_size, dtype)
return ClusterFinder_Cluster3x3i(image_size, n_sigma = n_sigma, capacity=capacity) return cls(image_size, n_sigma=n_sigma, capacity=capacity)
elif dtype == np.int32 and cluster_size == (2,2):
return ClusterFinder_Cluster2x2i(image_size, n_sigma = n_sigma, capacity=capacity)
else:
#TODO! add the other formats
raise ValueError(f"Unsupported dtype: {dtype}. Only np.int32 is supported.")
def ClusterFinderMT(image_size, cluster_size = (3,3), dtype=np.int32, n_sigma=5, capacity = 1024, n_threads = 3): def ClusterFinderMT(image_size, cluster_size = (3,3), dtype=np.int32, n_sigma=5, capacity = 1024, n_threads = 3):
@ -25,15 +50,9 @@ def ClusterFinderMT(image_size, cluster_size = (3,3), dtype=np.int32, n_sigma=5,
the templated ClusterFinderMT in C++. the templated ClusterFinderMT in C++.
""" """
if dtype == np.int32 and cluster_size == (3,3): cls = _get_class("ClusterFinderMT", cluster_size, dtype)
return ClusterFinderMT_Cluster3x3i(image_size, n_sigma = n_sigma, return cls(image_size, n_sigma=n_sigma, capacity=capacity, n_threads=n_threads)
capacity = capacity, n_threads = n_threads)
elif dtype == np.int32 and cluster_size == (2,2):
return ClusterFinderMT_Cluster2x2i(image_size, n_sigma = n_sigma,
capacity = capacity, n_threads = n_threads)
else:
#TODO! add the other formats
raise ValueError(f"Unsupported dtype: {dtype}. Only np.int32 is supported.")
def ClusterCollector(clusterfindermt, cluster_size = (3,3), dtype=np.int32): def ClusterCollector(clusterfindermt, cluster_size = (3,3), dtype=np.int32):
@ -42,14 +61,8 @@ def ClusterCollector(clusterfindermt, cluster_size = (3,3), dtype=np.int32):
the templated ClusterCollector in C++. the templated ClusterCollector in C++.
""" """
if dtype == np.int32 and cluster_size == (3,3): cls = _get_class("ClusterCollector", cluster_size, dtype)
return ClusterCollector_Cluster3x3i(clusterfindermt) return cls(clusterfindermt)
elif dtype == np.int32 and cluster_size == (2,2):
return ClusterCollector_Cluster2x2i(clusterfindermt)
else:
#TODO! add the other formats
raise ValueError(f"Unsupported dtype: {dtype}. Only np.int32 is supported.")
def ClusterFileSink(clusterfindermt, cluster_file, dtype=np.int32): def ClusterFileSink(clusterfindermt, cluster_file, dtype=np.int32):
""" """
@ -57,11 +70,5 @@ def ClusterFileSink(clusterfindermt, cluster_file, dtype=np.int32):
the templated ClusterCollector in C++. the templated ClusterCollector in C++.
""" """
if dtype == np.int32 and clusterfindermt.cluster_size == (3,3): cls = _get_class("ClusterFileSink", clusterfindermt.cluster_size, dtype)
return ClusterFileSink_Cluster3x3i(clusterfindermt, cluster_file) return cls(clusterfindermt, cluster_file)
elif dtype == np.int32 and clusterfindermt.cluster_size == (2,2):
return ClusterFileSink_Cluster2x2i(clusterfindermt, cluster_file)
else:
#TODO! add the other formats
raise ValueError(f"Unsupported dtype: {dtype}. Only np.int32 is supported.")

136
python/src/module.cpp
View File

@ -28,6 +28,25 @@
namespace py = pybind11; namespace py = pybind11;
/* MACRO that defines Cluster bindings for a specific size and type
T - Storage type of the cluster data (int, float, double)
N - Number of rows in the cluster
M - Number of columns in the cluster
U - Type of the pixel data (e.g., uint16_t)
TYPE_CODE - A character representing the type code (e.g., 'i' for int, 'd' for double, 'f' for float)
*/
#define DEFINE_CLUSTER_BINDINGS(T, N, M, U, TYPE_CODE) \
define_ClusterFile<T, N, M, U>(m, "Cluster" #N "x" #M #TYPE_CODE); \
define_ClusterVector<T, N, M, U>(m, "Cluster" #N "x" #M #TYPE_CODE); \
define_ClusterFinder<T, N, M, U>(m, "Cluster" #N "x" #M #TYPE_CODE); \
define_ClusterFinderMT<T, N, M, U>(m, "Cluster" #N "x" #M #TYPE_CODE); \
define_ClusterFileSink<T, N, M, U>(m, "Cluster" #N "x" #M #TYPE_CODE); \
define_ClusterCollector<T, N, M, U>(m, "Cluster" #N "x" #M #TYPE_CODE); \
define_Cluster<T, N, M, U>(m, #N "x" #M #TYPE_CODE); \
register_calculate_eta<T, N, M, U>(m);
PYBIND11_MODULE(_aare, m) { PYBIND11_MODULE(_aare, m) {
define_file_io_bindings(m); define_file_io_bindings(m);
define_raw_file_io_bindings(m); define_raw_file_io_bindings(m);
@ -42,59 +61,80 @@ PYBIND11_MODULE(_aare, m) {
define_interpolation_bindings(m); define_interpolation_bindings(m);
define_jungfrau_data_file_io_bindings(m); define_jungfrau_data_file_io_bindings(m);
define_ClusterFile<int, 3, 3, uint16_t>(m, "Cluster3x3i"); DEFINE_CLUSTER_BINDINGS(int, 3, 3, uint16_t, i);
define_ClusterFile<double, 3, 3, uint16_t>(m, "Cluster3x3d"); DEFINE_CLUSTER_BINDINGS(double, 3, 3, uint16_t, d);
define_ClusterFile<float, 3, 3, uint16_t>(m, "Cluster3x3f"); DEFINE_CLUSTER_BINDINGS(float, 3, 3, uint16_t, f);
define_ClusterFile<int, 2, 2, uint16_t>(m, "Cluster2x2i");
define_ClusterFile<float, 2, 2, uint16_t>(m, "Cluster2x2f");
define_ClusterFile<double, 2, 2, uint16_t>(m, "Cluster2x2d");
define_ClusterVector<int, 3, 3, uint16_t>(m, "Cluster3x3i"); DEFINE_CLUSTER_BINDINGS(int, 2, 2, uint16_t, i);
define_ClusterVector<double, 3, 3, uint16_t>(m, "Cluster3x3d"); DEFINE_CLUSTER_BINDINGS(double, 2, 2, uint16_t, d);
define_ClusterVector<float, 3, 3, uint16_t>(m, "Cluster3x3f"); DEFINE_CLUSTER_BINDINGS(float, 2, 2, uint16_t, f);
define_ClusterVector<int, 2, 2, uint16_t>(m, "Cluster2x2i");
define_ClusterVector<double, 2, 2, uint16_t>(m, "Cluster2x2d");
define_ClusterVector<float, 2, 2, uint16_t>(m, "Cluster2x2f");
define_ClusterFinder<int, 3, 3, uint16_t>(m, "Cluster3x3i"); DEFINE_CLUSTER_BINDINGS(int, 5, 5, uint16_t, i);
define_ClusterFinder<double, 3, 3, uint16_t>(m, "Cluster3x3d"); DEFINE_CLUSTER_BINDINGS(double, 5, 5, uint16_t, d);
define_ClusterFinder<float, 3, 3, uint16_t>(m, "Cluster3x3f"); DEFINE_CLUSTER_BINDINGS(float, 5, 5, uint16_t, f);
define_ClusterFinder<int, 2, 2, uint16_t>(m, "Cluster2x2i");
define_ClusterFinder<double, 2, 2, uint16_t>(m, "Cluster2x2d");
define_ClusterFinder<float, 2, 2, uint16_t>(m, "Cluster2x2f");
define_ClusterFinderMT<int, 3, 3, uint16_t>(m, "Cluster3x3i"); DEFINE_CLUSTER_BINDINGS(int, 7, 7, uint16_t, i);
define_ClusterFinderMT<double, 3, 3, uint16_t>(m, "Cluster3x3d"); DEFINE_CLUSTER_BINDINGS(double, 7, 7, uint16_t, d);
define_ClusterFinderMT<float, 3, 3, uint16_t>(m, "Cluster3x3f"); DEFINE_CLUSTER_BINDINGS(float, 7, 7, uint16_t, f);
define_ClusterFinderMT<int, 2, 2, uint16_t>(m, "Cluster2x2i");
define_ClusterFinderMT<double, 2, 2, uint16_t>(m, "Cluster2x2d");
define_ClusterFinderMT<float, 2, 2, uint16_t>(m, "Cluster2x2f");
define_ClusterFileSink<int, 3, 3, uint16_t>(m, "Cluster3x3i"); DEFINE_CLUSTER_BINDINGS(int, 9, 9, uint16_t, i);
define_ClusterFileSink<double, 3, 3, uint16_t>(m, "Cluster3x3d"); DEFINE_CLUSTER_BINDINGS(double, 9, 9, uint16_t, d);
define_ClusterFileSink<float, 3, 3, uint16_t>(m, "Cluster3x3f"); DEFINE_CLUSTER_BINDINGS(float, 9, 9, uint16_t, f);
define_ClusterFileSink<int, 2, 2, uint16_t>(m, "Cluster2x2i");
define_ClusterFileSink<double, 2, 2, uint16_t>(m, "Cluster2x2d");
define_ClusterFileSink<float, 2, 2, uint16_t>(m, "Cluster2x2f");
define_ClusterCollector<int, 3, 3, uint16_t>(m, "Cluster3x3i"); // define_ClusterFile<int, 3, 3, uint16_t>(m, "Cluster3x3i");
define_ClusterCollector<double, 3, 3, uint16_t>(m, "Cluster3x3d"); // define_ClusterFile<double, 3, 3, uint16_t>(m, "Cluster3x3d");
define_ClusterCollector<float, 3, 3, uint16_t>(m, "Cluster3x3f"); // define_ClusterFile<float, 3, 3, uint16_t>(m, "Cluster3x3f");
define_ClusterCollector<int, 2, 2, uint16_t>(m, "Cluster2x2i"); // define_ClusterFile<int, 2, 2, uint16_t>(m, "Cluster2x2i");
define_ClusterCollector<double, 2, 2, uint16_t>(m, "Cluster2x2d"); // define_ClusterFile<float, 2, 2, uint16_t>(m, "Cluster2x2f");
define_ClusterCollector<float, 2, 2, uint16_t>(m, "Cluster2x2f"); // define_ClusterFile<double, 2, 2, uint16_t>(m, "Cluster2x2d");
define_Cluster<int, 3, 3, uint16_t>(m, "3x3i");
define_Cluster<float, 3, 3, uint16_t>(m, "3x3f");
define_Cluster<double, 3, 3, uint16_t>(m, "3x3d");
define_Cluster<int, 2, 2, uint16_t>(m, "2x2i");
define_Cluster<float, 2, 2, uint16_t>(m, "2x2f");
define_Cluster<double, 2, 2, uint16_t>(m, "2x2d");
register_calculate_eta<int, 3, 3, uint16_t>(m); // define_ClusterVector<int, 3, 3, uint16_t>(m, "Cluster3x3i");
register_calculate_eta<float, 3, 3, uint16_t>(m); // define_ClusterVector<double, 3, 3, uint16_t>(m, "Cluster3x3d");
register_calculate_eta<double, 3, 3, uint16_t>(m); // define_ClusterVector<float, 3, 3, uint16_t>(m, "Cluster3x3f");
register_calculate_eta<int, 2, 2, uint16_t>(m); // define_ClusterVector<int, 2, 2, uint16_t>(m, "Cluster2x2i");
register_calculate_eta<float, 2, 2, uint16_t>(m); // define_ClusterVector<double, 2, 2, uint16_t>(m, "Cluster2x2d");
register_calculate_eta<double, 2, 2, uint16_t>(m); // define_ClusterVector<float, 2, 2, uint16_t>(m, "Cluster2x2f");
// define_ClusterFinder<int, 3, 3, uint16_t>(m, "Cluster3x3i");
// define_ClusterFinder<double, 3, 3, uint16_t>(m, "Cluster3x3d");
// define_ClusterFinder<float, 3, 3, uint16_t>(m, "Cluster3x3f");
// define_ClusterFinder<int, 2, 2, uint16_t>(m, "Cluster2x2i");
// define_ClusterFinder<double, 2, 2, uint16_t>(m, "Cluster2x2d");
// define_ClusterFinder<float, 2, 2, uint16_t>(m, "Cluster2x2f");
// define_ClusterFinderMT<int, 3, 3, uint16_t>(m, "Cluster3x3i");
// define_ClusterFinderMT<double, 3, 3, uint16_t>(m, "Cluster3x3d");
// define_ClusterFinderMT<float, 3, 3, uint16_t>(m, "Cluster3x3f");
// define_ClusterFinderMT<int, 2, 2, uint16_t>(m, "Cluster2x2i");
// define_ClusterFinderMT<double, 2, 2, uint16_t>(m, "Cluster2x2d");
// define_ClusterFinderMT<float, 2, 2, uint16_t>(m, "Cluster2x2f");
// define_ClusterFileSink<int, 3, 3, uint16_t>(m, "Cluster3x3i");
// define_ClusterFileSink<double, 3, 3, uint16_t>(m, "Cluster3x3d");
// define_ClusterFileSink<float, 3, 3, uint16_t>(m, "Cluster3x3f");
// define_ClusterFileSink<int, 2, 2, uint16_t>(m, "Cluster2x2i");
// define_ClusterFileSink<double, 2, 2, uint16_t>(m, "Cluster2x2d");
// define_ClusterFileSink<float, 2, 2, uint16_t>(m, "Cluster2x2f");
// define_ClusterCollector<int, 3, 3, uint16_t>(m, "Cluster3x3i");
// define_ClusterCollector<double, 3, 3, uint16_t>(m, "Cluster3x3d");
// define_ClusterCollector<float, 3, 3, uint16_t>(m, "Cluster3x3f");
// define_ClusterCollector<int, 2, 2, uint16_t>(m, "Cluster2x2i");
// define_ClusterCollector<double, 2, 2, uint16_t>(m, "Cluster2x2d");
// define_ClusterCollector<float, 2, 2, uint16_t>(m, "Cluster2x2f");
// // define_Cluster<int, 3, 3, uint16_t>(m, "3x3i");
// define_Cluster<float, 3, 3, uint16_t>(m, "3x3f");
// define_Cluster<double, 3, 3, uint16_t>(m, "3x3d");
// define_Cluster<int, 2, 2, uint16_t>(m, "2x2i");
// define_Cluster<float, 2, 2, uint16_t>(m, "2x2f");
// define_Cluster<double, 2, 2, uint16_t>(m, "2x2d");
// // register_calculate_eta<int, 3, 3, uint16_t>(m);
// register_calculate_eta<float, 3, 3, uint16_t>(m);
// register_calculate_eta<double, 3, 3, uint16_t>(m);
// register_calculate_eta<int, 2, 2, uint16_t>(m);
// register_calculate_eta<float, 2, 2, uint16_t>(m);
// register_calculate_eta<double, 2, 2, uint16_t>(m);
} }