// 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");
}