// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#include <catch2/catch_all.hpp>
#include "../image_analysis/spot_finding/StrongPixelSet.h"

TEST_CASE("DiffractionSpot_AddOperator","[StrongPixelSet]") {
    DiffractionSpot spot1(4,1,10), spot2(3,4,5);

    spot1 += spot2;

    // Spot2 is not changed
    REQUIRE(spot2.Count() == 5);

    REQUIRE(spot1.Count() == 15);
    REQUIRE(spot1.RawCoord().x == Catch::Approx((4*10+3*5)/15.0));
    REQUIRE(spot1.RawCoord().y == Catch::Approx((1*10+4*5)/15.0));
    REQUIRE(spot1.MaxCount() == 10);
    REQUIRE(spot1.PixelCount() == 2);
}

TEST_CASE("StrongPixelSet_BuildSpots","[StrongPixelSet]") {
    DiffractionExperiment experiment(DetJF(1,1,0,0,false));
    experiment.Raw();

    SpotFindingSettings settings;
    settings.low_resolution_limit = 200.0;
    settings.high_resolution_limit = 0.5;
    settings.min_pix_per_spot = 3;
    settings.max_pix_per_spot = 200;

    // In raster order, which is what AddStrongPixel requires and what both callers produce.
    std::vector<DiffractionSpot> spots;
    StrongPixelSet strong_pixel_set;
    for (uint16_t line = 104; line <= 106; line++)
        for (uint16_t col = 6; col <= 8; col++)
            strong_pixel_set.AddStrongPixel(col, line);
    strong_pixel_set.FindSpots(experiment, settings, spots, 0);
    REQUIRE(spots.size() == 1);

    REQUIRE(spots[0].Count() == 9.0f);
    REQUIRE(spots[0].PixelCount() == 9);
    REQUIRE(spots[0].RawCoord().x == Catch::Approx(7.0));
    REQUIRE(spots[0].RawCoord().y == Catch::Approx(105.0));
}

TEST_CASE("StrongPixelSet_LargeSpotMustBeCompact","[StrongPixelSet]") {
    SpotFindingSettings settings;   // max_pix_per_spot 200, so all three pass on size alone

    // A filled 9x9 square: 81 pixels in a 9-wide box, so it fills all of it. A Bragg reflection is
    // this shape however bright it is, and this is what a raised size bound is meant to admit.
    StrongPixelSet square;
    for (uint16_t line = 100; line < 109; line++)
        for (uint16_t col = 200; col < 209; col++)
            square.AddStrongPixel(col, line);
    std::vector<DiffractionSpot> spots;
    square.FindComponentsImage(settings, spots);
    REQUIRE(spots.size() == 1);
    CHECK(spots[0].PixelCount() == 81);

    // The same 81 pixels drawn as a line: an ice arc, a cosmic-ray track or a lit detector row. It
    // fills 1/81 of its bounding square, so it is refused however many pixels it has.
    StrongPixelSet streak;
    for (uint16_t col = 200; col < 281; col++)
        streak.AddStrongPixel(col, 100);
    spots.clear();
    streak.FindComponentsImage(settings, spots);
    CHECK(spots.empty());

    // Shape is asked of large components only. A short line of the same one-pixel width is below
    // SPOT_SHAPE_FREE_PIXELS and is kept, exactly as it was before the bound was raised.
    StrongPixelSet short_streak;
    for (uint16_t col = 200; col < 210; col++)
        short_streak.AddStrongPixel(col, 100);
    spots.clear();
    short_streak.FindComponentsImage(settings, spots);
    REQUIRE(spots.size() == 1);
    CHECK(spots[0].PixelCount() == 10);
}

/*
TEST_CASE("StrongPixelSet_ReadFPGAOutput_1","[StrongPixelSet]") {
    DiffractionExperiment experiment(DetectorGeometry(8, 2, 8, 36, true));
    experiment.BeamX_pxl(1000).BeamY_pxl(1000).DetectorDistance_mm(100).PhotonEnergy_keV(WVL_1A_IN_KEV);

    StrongPixelSet strong_pixel_set(experiment);

    DeviceOutput output{};
    for (auto &i: output.spot_finding_result.strong_pixel)
        i = 0;
    output.pixels[0] = 7;
    output.pixels[1] = 7;
    output.pixels[1024] = 7;
    output.pixels[1025] = 7;
    output.pixels[7] = 345;

    // Pixels 0, 1, 7, 1024 and 1025 will be enabled
    output.spot_finding_result.strong_pixel[0] = (1U<<0) | (1U<<1) | (1U<<7);
    output.spot_finding_result.strong_pixel[RAW_MODULE_COLS/8] = (1<<0) | (1<<1);
    output.spot_finding_result.strong_pixel_count = 5;

    strong_pixel_set.ReadFPGAOutput(output);
    REQUIRE(strong_pixel_set.GetStrongPixelCount() == 5);
    SpotFindingSettings settings{
            .min_pix_per_spot = 2,
            .max_pix_per_spot = 55,
            .high_resolution_limit = 1.0,
            .low_resolution_limit = 50.0
    };
    std::vector<DiffractionSpot> spots;
    strong_pixel_set.FindSpots(experiment, settings, spots, 2);
    CHECK(spots.size() == 1);
    REQUIRE(spots[0].RawCoord().x == Approx(0.5));
    REQUIRE(spots[0].RawCoord().y == Approx(CONVERTED_MODULE_LINES * 3 + 36 * 2 - 1 - 0.5));
    REQUIRE(spots[0].PixelCount() == 4);
    REQUIRE(spots[0].MaxCount() == 7);
}

TEST_CASE("StrongPixelSet_ReadFPGAOutput_2","[StrongPixelSet]") {
    DiffractionExperiment experiment(DetectorGeometry(8, 2, 8, 36, true));
    experiment.BeamX_pxl(1000).BeamY_pxl(1000).DetectorDistance_mm(100).PhotonEnergy_keV(WVL_1A_IN_KEV);

    StrongPixelSet strong_pixel_set(experiment);

    DeviceOutput output{};
    for (auto &i: output.spot_finding_result.strong_pixel)
        i = 0;
    output.pixels[0] = 7;
    output.pixels[1] = 7;
    output.pixels[1024] = 7;
    output.pixels[1025] = 7;
    output.pixels[7] = 345;

    // Pixels 0, 1, 7, 1024 and 1025 will be enabled
    output.spot_finding_result.strong_pixel[0] = (1U<<0) | (1U<<1) | (1U<<7);
    output.spot_finding_result.strong_pixel[RAW_MODULE_COLS/8] = (1<<0) | (1<<1);
    output.spot_finding_result.strong_pixel_count = 5;

    strong_pixel_set.ReadFPGAOutput(output);
    REQUIRE(strong_pixel_set.GetStrongPixelCount() == 5);
    SpotFindingSettings settings{
            .min_pix_per_spot = 1,
            .max_pix_per_spot = 3,
            .high_resolution_limit = 1.0,
            .low_resolution_limit = 50.0
    };
    std::vector<DiffractionSpot> spots;
    strong_pixel_set.FindSpots(experiment, settings, spots, 2);
    CHECK(spots.size() == 1);
    REQUIRE(spots[0].RawCoord().x == Approx(7));
    REQUIRE(spots[0].RawCoord().y == Approx(CONVERTED_MODULE_LINES * 3 + 36 * 2 - 1));
    REQUIRE(spots[0].PixelCount() == 1);
    REQUIRE(spots[0].MaxCount() == 345);
} */