mirror of
https://github.com/slsdetectorgroup/aare.git
synced 2026-02-19 08:08:42 +01:00
173 lines
6.7 KiB
C++
173 lines
6.7 KiB
C++
// SPDX-License-Identifier: MPL-2.0
|
|
// Files with bindings to the different classes
|
|
|
|
// New style file naming
|
|
#include "bind_Cluster.hpp"
|
|
#include "bind_ClusterCollector.hpp"
|
|
#include "bind_ClusterFile.hpp"
|
|
#include "bind_ClusterFileSink.hpp"
|
|
#include "bind_ClusterFinder.hpp"
|
|
#include "bind_ClusterFinderMT.hpp"
|
|
#include "bind_ClusterVector.hpp"
|
|
#include "bind_Eta.hpp"
|
|
#include "bind_Interpolator.hpp"
|
|
#include "bind_calibration.hpp"
|
|
|
|
// TODO! migrate the other names
|
|
#include "ctb_raw_file.hpp"
|
|
#include "file.hpp"
|
|
#include "fit.hpp"
|
|
#include "jungfrau_data_file.hpp"
|
|
#include "pedestal.hpp"
|
|
#include "pixel_map.hpp"
|
|
#include "raw_file.hpp"
|
|
#include "raw_master_file.hpp"
|
|
#include "raw_sub_file.hpp"
|
|
#include "var_cluster.hpp"
|
|
|
|
// Pybind stuff
|
|
#include <pybind11/pybind11.h>
|
|
#include <pybind11/stl.h>
|
|
|
|
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_Cluster<T, N, M, U>(m, #N "x" #M #TYPE_CODE); \
|
|
register_calculate_2x2eta<T, N, M, U>(m); \
|
|
define_2x2_reduction<T, N, M, U>(m); \
|
|
reduce_to_2x2<T, N, M, U>(m);
|
|
|
|
#define DEFINE_BINDINGS_CLUSTERFINDER(T, N, M, U, 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);
|
|
|
|
PYBIND11_MODULE(_aare, m) {
|
|
define_file_io_bindings(m);
|
|
define_raw_file_io_bindings(m);
|
|
define_raw_sub_file_io_bindings(m);
|
|
define_ctb_raw_file_io_bindings(m);
|
|
define_raw_master_file_bindings(m);
|
|
define_var_cluster_finder_bindings(m);
|
|
define_pixel_map_bindings(m);
|
|
define_pedestal_bindings<double>(m, "Pedestal_d");
|
|
define_pedestal_bindings<float>(m, "Pedestal_f");
|
|
define_fit_bindings(m);
|
|
define_interpolation_bindings(m);
|
|
define_jungfrau_data_file_io_bindings(m);
|
|
|
|
bind_calibration(m);
|
|
|
|
DEFINE_CLUSTER_BINDINGS(int, 3, 3, uint16_t, i);
|
|
DEFINE_CLUSTER_BINDINGS(double, 3, 3, uint16_t, d);
|
|
DEFINE_CLUSTER_BINDINGS(float, 3, 3, uint16_t, f);
|
|
|
|
DEFINE_CLUSTER_BINDINGS(int, 2, 2, uint16_t, i);
|
|
DEFINE_CLUSTER_BINDINGS(double, 2, 2, uint16_t, d);
|
|
DEFINE_CLUSTER_BINDINGS(float, 2, 2, uint16_t, f);
|
|
|
|
DEFINE_CLUSTER_BINDINGS(int, 5, 5, uint16_t, i);
|
|
DEFINE_CLUSTER_BINDINGS(double, 5, 5, uint16_t, d);
|
|
DEFINE_CLUSTER_BINDINGS(float, 5, 5, uint16_t, f);
|
|
|
|
DEFINE_CLUSTER_BINDINGS(int, 7, 7, uint16_t, i);
|
|
DEFINE_CLUSTER_BINDINGS(double, 7, 7, uint16_t, d);
|
|
DEFINE_CLUSTER_BINDINGS(float, 7, 7, uint16_t, f);
|
|
|
|
DEFINE_CLUSTER_BINDINGS(int, 9, 9, uint16_t, i);
|
|
DEFINE_CLUSTER_BINDINGS(double, 9, 9, uint16_t, d);
|
|
DEFINE_CLUSTER_BINDINGS(float, 9, 9, uint16_t, f);
|
|
|
|
DEFINE_CLUSTER_BINDINGS(int, 21, 21, uint16_t, i);
|
|
DEFINE_CLUSTER_BINDINGS(double, 21, 21, uint16_t, d);
|
|
DEFINE_CLUSTER_BINDINGS(float, 21, 21, uint16_t, f);
|
|
|
|
DEFINE_CLUSTER_BINDINGS(int16_t, 3, 3, uint16_t, i16);
|
|
|
|
DEFINE_BINDINGS_CLUSTERFINDER(int, 3, 3, uint16_t, i);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(double, 3, 3, uint16_t, d);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(float, 3, 3, uint16_t, f);
|
|
|
|
DEFINE_BINDINGS_CLUSTERFINDER(int, 5, 5, uint16_t, i);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(double, 5, 5, uint16_t, d);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(float, 5, 5, uint16_t, f);
|
|
|
|
DEFINE_BINDINGS_CLUSTERFINDER(int, 7, 7, uint16_t, i);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(double, 7, 7, uint16_t, d);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(float, 7, 7, uint16_t, f);
|
|
|
|
DEFINE_BINDINGS_CLUSTERFINDER(int, 9, 9, uint16_t, i);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(double, 9, 9, uint16_t, d);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(float, 9, 9, uint16_t, f);
|
|
|
|
DEFINE_BINDINGS_CLUSTERFINDER(int, 21, 21, uint16_t, i);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(double, 21, 21, uint16_t, d);
|
|
DEFINE_BINDINGS_CLUSTERFINDER(float, 21, 21, uint16_t, f);
|
|
|
|
define_3x3_reduction<int, 3, 3, uint16_t>(m);
|
|
define_3x3_reduction<double, 3, 3, uint16_t>(m);
|
|
define_3x3_reduction<float, 3, 3, uint16_t>(m);
|
|
define_3x3_reduction<int, 5, 5, uint16_t>(m);
|
|
define_3x3_reduction<double, 5, 5, uint16_t>(m);
|
|
define_3x3_reduction<float, 5, 5, uint16_t>(m);
|
|
define_3x3_reduction<int, 7, 7, uint16_t>(m);
|
|
define_3x3_reduction<double, 7, 7, uint16_t>(m);
|
|
define_3x3_reduction<float, 7, 7, uint16_t>(m);
|
|
define_3x3_reduction<int, 9, 9, uint16_t>(m);
|
|
define_3x3_reduction<double, 9, 9, uint16_t>(m);
|
|
define_3x3_reduction<float, 9, 9, uint16_t>(m);
|
|
|
|
reduce_to_3x3<int, 3, 3, uint16_t>(m);
|
|
reduce_to_3x3<double, 3, 3, uint16_t>(m);
|
|
reduce_to_3x3<float, 3, 3, uint16_t>(m);
|
|
reduce_to_3x3<int, 5, 5, uint16_t>(m);
|
|
reduce_to_3x3<double, 5, 5, uint16_t>(m);
|
|
reduce_to_3x3<float, 5, 5, uint16_t>(m);
|
|
reduce_to_3x3<int, 7, 7, uint16_t>(m);
|
|
reduce_to_3x3<double, 7, 7, uint16_t>(m);
|
|
reduce_to_3x3<float, 7, 7, uint16_t>(m);
|
|
reduce_to_3x3<int, 9, 9, uint16_t>(m);
|
|
reduce_to_3x3<double, 9, 9, uint16_t>(m);
|
|
reduce_to_3x3<float, 9, 9, uint16_t>(m);
|
|
|
|
register_calculate_3x3eta<int, 3, 3, uint16_t>(m);
|
|
register_calculate_3x3eta<double, 3, 3, uint16_t>(m);
|
|
register_calculate_3x3eta<float, 3, 3, uint16_t>(m);
|
|
register_calculate_3x3eta<int16_t, 3, 3, uint16_t>(m);
|
|
|
|
using Sum_index_pair_d = Sum_index_pair<double, corner>;
|
|
PYBIND11_NUMPY_DTYPE(Sum_index_pair_d, sum, index);
|
|
using Sum_index_pair_f = Sum_index_pair<float, corner>;
|
|
PYBIND11_NUMPY_DTYPE(Sum_index_pair_f, sum, index);
|
|
using Sum_index_pair_i = Sum_index_pair<int, corner>;
|
|
PYBIND11_NUMPY_DTYPE(Sum_index_pair_i, sum, index);
|
|
|
|
using eta_d = Eta2<double>;
|
|
PYBIND11_NUMPY_DTYPE(eta_d, x, y, c, sum);
|
|
using eta_i = Eta2<int>;
|
|
PYBIND11_NUMPY_DTYPE(eta_i, x, y, c, sum);
|
|
using eta_f = Eta2<float>;
|
|
PYBIND11_NUMPY_DTYPE(eta_f, x, y, c, sum);
|
|
using eta_i16 = Eta2<int16_t>;
|
|
PYBIND11_NUMPY_DTYPE(eta_i16, x, y, c, sum);
|
|
|
|
define_corner_enum(m);
|
|
define_eta<float>(m, "f");
|
|
define_eta<double>(m, "d");
|
|
define_eta<int>(m, "i");
|
|
define_eta<int16_t>(m, "i16");
|
|
}
|