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

#pragma once

#include <string>
#include <optional>
#include <vector>
#include <map>

#include "../common/DiffractionGeometry.h"
#include "../common/DiffractionExperiment.h"
#include "../common/PixelMask.h"
#include "../common/AzimuthalIntegrationMapping.h"

struct JFJochReaderDataset {
    std::string arm_date;

    DiffractionExperiment experiment;
    PixelMask pixel_mask;

    std::optional<int64_t> error_value;

    std::string jfjoch_release;

    std::vector<float> az_int_bin_to_q;
    std::vector<float> az_int_bin_to_phi;
    size_t azimuthal_bins = 0;
    size_t q_bins = 0;

    std::vector<float> spot_count;
    std::vector<float> spot_count_indexed;
    std::vector<float> spot_count_low_res;
    std::vector<float> spot_count_ice_rings;

    std::vector<float> indexing_result;
    std::vector<float> indexing_lattice_count;
    std::vector<float> bkg_estimate;
    std::vector<float> resolution_estimate;
    std::vector<float> efficiency;
    std::vector<float> profile_radius;
    std::vector<float> mosaicity_deg;
    std::vector<float> b_factor;
    std::vector<float> integrated_reflections;
    std::vector<float> image_scale_factor;
    std::vector<float> image_scale_cc;
    std::vector<float> image_scale_b;
    std::vector<int64_t> max_value;

    // Maps this dataset's image index -> the original/collected image number it came from.
    // Empty means identity (image i == original image i). Lets a dataset be a subset (or strided
    // selection) of the truly collected images: reprocessing snapshots over a sub-range, and (in
    // future) a main dataset that was filtered on-the-fly during collection.
    std::vector<int> source_image_number;

    std::vector<std::string> roi;
    std::vector<std::vector<int64_t>> roi_sum;
    std::vector<std::vector<int64_t>> roi_sum_sq;
    std::vector<std::vector<int64_t>> roi_max;
    std::vector<std::vector<int64_t>> roi_npixel;
    std::vector<std::vector<float>> roi_x;
    std::vector<std::vector<float>> roi_y;

    // ROI definitions stored in the master file. The logical definitions populate
    // experiment.ROI() (they re-derive with the current geometry); roi_map is the
    // per-pixel bitmask as written (constant footprint), with roi_bit_index mapping
    // each ROI name to its bit.
    std::vector<uint16_t> roi_map;
    std::map<std::string, uint16_t> roi_bit_index;

    std::vector<std::string> calibration_data;

    JFJochReaderDataset() = default;
    JFJochReaderDataset(const JFJochReaderDataset &other) = default;
};