Jungfraujoch/writer/HDF5DataFilePluginAzInt.h

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

#pragma once

#include "HDF5DataFilePlugin.h"

class HDF5DataFilePluginAzInt : public HDF5DataFilePlugin {
    const size_t azimuthal_bins;
    const size_t q_bins;

    std::vector<float> az_int_bin_to_q;
    std::vector<float> az_int_bin_to_two_theta;
    std::vector<float> az_int_bin_to_phi;
    std::vector<float> az_int_image;
    std::vector<float> az_int_image_std;
    std::vector<uint64_t> az_int_image_count;

    bool save_profile_count = false;
    bool save_profile_std = false;

    // std::unique_ptr<HDF5DataSet> dataset;
public:
    explicit HDF5DataFilePluginAzInt(const StartMessage &start);
    void OpenFile(HDF5File &data_file, const DataMessage& msg, size_t images_per_file) override;
    void Write(const DataMessage& msg, uint64_t image_number) override;
    void WriteFinal(HDF5File &data_file) override;
};