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

#pragma once

#include "HDF5DataFilePlugin.h"
#include "../common/AutoIncrVector.h"

class HDF5DataFilePluginPerformance : public HDF5DataFilePlugin {
    AutoIncrVector<float> spot_finding_time{NAN};
    AutoIncrVector<float> indexing_time{NAN};
    AutoIncrVector<float> refinement_time{NAN};
    AutoIncrVector<float> integration_time{NAN};
    AutoIncrVector<float> processing_time{NAN};
    AutoIncrVector<float> bragg_prediction_time{NAN};
    AutoIncrVector<float> preprocessing_time{NAN};
    AutoIncrVector<float> compression_time{NAN};
    AutoIncrVector<float> azint_time{NAN};
    AutoIncrVector<float> index_analysis_time{NAN};

public:
    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;
};