#include "aare/PixelMap.hpp"

#include <array>

namespace aare {
NDArray<ssize_t, 2> GenerateMoench03PixelMap() {
    std::array<int, 32> const adc_nr = {300, 325, 350, 375, 300, 325, 350, 375,
                                        200, 225, 250, 275, 200, 225, 250, 275,
                                        100, 125, 150, 175, 100, 125, 150, 175,
                                        0,   25,  50,  75,  0,   25,  50,  75};
    int const sc_width = 25;
    int const nadc = 32;
    int const pixels_per_sc = 5000;
    NDArray<ssize_t, 2> order_map({400, 400});

    int pixel = 0;
    for (int i = 0; i != pixels_per_sc; ++i) {
        for (int i_adc = 0; i_adc != nadc; ++i_adc) {
            int const col = adc_nr[i_adc] + (i % sc_width);
            int row = 0;
            if ((i_adc / 4) % 2 == 0)
                row = 199 - (i / sc_width);
            else
                row = 200 + (i / sc_width);

            order_map(row, col) = pixel;
            pixel++;
        }
    }
    return order_map;
}

NDArray<ssize_t, 2> GenerateMoench05PixelMap() {
    std::array<int, 3> adc_numbers = {9, 13, 1};
    NDArray<ssize_t, 2> order_map({160, 150});
    int n_pixel = 0;
    for (int row = 0; row < 160; row++) {
        for (int i_col = 0; i_col < 50; i_col++) {
            n_pixel = row * 50 + i_col;
            for (int i_sc = 0; i_sc < 3; i_sc++) {
                int col = 50 * i_sc + i_col;
                int adc_nr = adc_numbers[i_sc];
                int i_analog = n_pixel * 32 + adc_nr;


                // analog_frame[row * 150 + col] = analog_data[i_analog] & 0x3FFF;
                order_map(row, col) = i_analog;
                
            }
        }
    }
    return order_map;
}

NDArray<ssize_t, 2>GenerateMH02SingleCounterPixelMap(){
    NDArray<ssize_t, 2> order_map({48, 48});
    for(int row = 0; row < 48; row++){
        for(int col = 0; col < 48; col++){
            order_map(row, col) = row*48 + col;
        }
    }
    return order_map;
}

NDArray<ssize_t, 3> GenerateMH02FourCounterPixelMap(){
    NDArray<ssize_t, 3> order_map({4, 48, 48});
    for (int counter=0; counter<4; counter++){
        for(int row = 0; row < 48; row++){
            for(int col = 0; col < 48; col++){
                order_map(counter, row, col) = counter*48*48 + row*48 + col;
            }
        }
    }
    return order_map;
}

} // namespace aare