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

#include "HDF5DataFilePluginDetector.h"

HDF5DataFilePluginDetector::HDF5DataFilePluginDetector(const StartMessage &msg)
: beam_center_x_default(msg.beam_center_x),
beam_center_y_default(msg.beam_center_y),
geom_refinement(msg.geom_refinement_algorithm != GeomRefinementAlgorithmEnum::None){}

void HDF5DataFilePluginDetector::OpenFile(HDF5File &in_data_file, const DataMessage &msg) {}

void HDF5DataFilePluginDetector::Write(const DataMessage &msg, uint64_t image_number) {
    if (msg.jf_info.has_value())
        jf_info[image_number] = msg.jf_info.value();
    if (msg.storage_cell.has_value())
        storage_cell[image_number] = msg.storage_cell.value();
    if (msg.receiver_aq_dev_delay.has_value())
        receiver_aq_dev_delay[image_number] = msg.receiver_aq_dev_delay.value();
    if (msg.receiver_free_send_buf.has_value())
        receiver_free_buffers[image_number] = msg.receiver_free_send_buf.value();

    if (msg.packets_received.has_value())
        packets_received[image_number] = msg.packets_received.value();
    if (msg.packets_expected.has_value())
        packets_expected[image_number] = msg.packets_expected.value();
    if (msg.pixel_sum.has_value())
        pixel_sum[image_number] = msg.pixel_sum.value();

    if (msg.image_collection_efficiency.has_value())
        efficiency[image_number] = msg.image_collection_efficiency.value();
    else
        efficiency[image_number] = 1.0;

    if (msg.processing_time_s.has_value())
        processing_time[image_number] = msg.processing_time_s.value();

    if (geom_refinement) {
        if (msg.beam_center_x.has_value())
            beam_center_x[image_number] = msg.beam_center_x.value();
        else
            beam_center_x[image_number] = beam_center_x_default;

        if (msg.beam_center_y.has_value())
            beam_center_y[image_number] = msg.beam_center_y.value();
        else
            beam_center_y[image_number] = beam_center_y_default;
    }
}

void HDF5DataFilePluginDetector::WriteFinal(HDF5File &data_file) {
    if (!jf_info.empty())
        data_file.SaveVector("/entry/detector/det_info", jf_info.vec());
    if (!storage_cell.empty())
        data_file.SaveVector("/entry/detector/storage_cell_image", storage_cell.vec());
    if (!receiver_aq_dev_delay.empty())
        data_file.SaveVector("/entry/detector/rcv_delay", receiver_aq_dev_delay.vec());
    if (!receiver_free_buffers.empty())
        data_file.SaveVector("/entry/detector/rcv_free_send_buffers", receiver_free_buffers.vec());
    if (!packets_received.empty())
        data_file.SaveVector("/entry/detector/packets_received", packets_received.vec());
    if (!packets_expected.empty())
        data_file.SaveVector("/entry/detector/packets_expected", packets_expected.vec());
    if (!pixel_sum.empty())
        data_file.SaveVector("/entry/detector/pixel_sum", pixel_sum.vec());
    if (!processing_time.empty())
        data_file.SaveVector("/entry/detector/processing_time", processing_time.vec())->Units("s");
    if (!beam_center_x.empty())
        data_file.SaveVector("/entry/detector/beam_center_x", beam_center_x.vec());
    if (!beam_center_y.empty())
        data_file.SaveVector("/entry/detector/beam_center_y", beam_center_y.vec());

    data_file.SaveVector("/entry/detector/data_collection_efficiency_image", efficiency.vec());
}