// Copyright (2019-2024) Paul Scherrer Institute

#include "HDF5DataFilePluginAzInt.h"

HDF5DataFilePluginAzInt::HDF5DataFilePluginAzInt(const std::vector<float> &in_rad_int_bin_to_q) :
rad_int_bin_to_q(in_rad_int_bin_to_q) {}

void HDF5DataFilePluginAzInt::OpenFile(HDF5File &data_file, const DataMessage &msg) {
    if (rad_int_bin_to_q.empty())
        return;

    HDF5Group(data_file, "/entry/azint").NXClass("NXcollection");

    data_file.SaveVector("/entry/azint/binToQ", rad_int_bin_to_q);

    HDF5DataSpace data_space_rad_int({1, rad_int_bin_to_q.size()}, {H5S_UNLIMITED, rad_int_bin_to_q.size()});
    HDF5Dcpl dcpl_rad_int;
    dcpl_rad_int.SetChunking({1, rad_int_bin_to_q.size()});

    data_set_az_int = std::make_unique<HDF5DataSet>(data_file, "/entry/azint/result", HDF5DataType(0.0f),
                                                    data_space_rad_int, dcpl_rad_int);
}

void HDF5DataFilePluginAzInt::Write(const DataMessage &msg) {
    if (rad_int_bin_to_q.empty())
        return;

    size_t image_number = msg.number;

    if (image_number >= max_image_number) {
        max_image_number = image_number;
        data_set_az_int->SetExtent({max_image_number + 1, rad_int_bin_to_q.size()});
    }

    if (!msg.az_int_profile.empty() && (msg.az_int_profile.size() == rad_int_bin_to_q.size()))
        data_set_az_int->WriteVec(msg.az_int_profile, {image_number, 0}, {1, rad_int_bin_to_q.size()});
}

void HDF5DataFilePluginAzInt::WriteFinal(HDF5File &data_file) {}