parallel_streaming_framer.h

#pragma once
 
#include <iostream>
#include <vector>
#include <string>
#include <map>
#include <unordered_map>
#include <variant>
#include "TFile.h"
#include "TTree.h"
#include "TH1.h"
#include "TH2.h"
#include "TSystem.h"
#include <filesystem>
#include "alcor_spilldata.h"
#include "alcor_data_streamer.h"
 
 
#define _FRAME_SIZE_ 1024
 
#define _FRAME_LENGTH_NS_ (_FRAME_SIZE_ * 3.125)
 
#define _FIRST_FRAMES_TRIGGER_ 2500
 
#define _AFTERPULSE_DEADTIME_ 64
 
 
class parallel_streaming_framer
{
public:
    // -------------------------------------------------------------------------
    // Constructors
    // -------------------------------------------------------------------------
 
    parallel_streaming_framer() = default;
 
    parallel_streaming_framer(std::vector<std::string> filenames, uint16_t frame_size = _FRAME_SIZE_)
        : parallel_streaming_framer(filenames, "", "", frame_size) {}
 
    parallel_streaming_framer(std::vector<std::string> filenames, std::string trigger_config_file, uint16_t frame_size = _FRAME_SIZE_)
        : parallel_streaming_framer(filenames, trigger_config_file, "", frame_size) {}
 
    parallel_streaming_framer(std::vector<std::string> filenames, std::string trigger_config_file, std::string readout_config_file, uint16_t frame_size = _FRAME_SIZE_);
 
    // -------------------------------------------------------------------------
    // Getters
    // -------------------------------------------------------------------------
 
    alcor_spilldata get_spilldata() const;
 
    alcor_spilldata &get_spilldata_link();
 
    int get_registered_triggers();
 
    TH2F *get_fine_tune_distribution();
 
    // -------------------------------------------------------------------------
    // Setters
    // -------------------------------------------------------------------------
 
    void set_spilldata(alcor_spilldata v);
 
    void set_spilldata_link(alcor_spilldata &v);
 
    void set_parallel_cores(uint16_t v);
 
    // -------------------------------------------------------------------------
    // Progress bar assignment
    // -------------------------------------------------------------------------
 
    void assign_bar(mist::logger::progress_bar &bar);
 
    void assign_bar(mist::logger::subtask_progress_bar &bar);
 
    void clear_bar();
 
    // -------------------------------------------------------------------------
    // I/O Operations
    // -------------------------------------------------------------------------
 
    bool next_spill();
 
    void process(alcor_data_streamer &current_stream, int _frame_size);
 
private:
    // -------------------------------------------------------------------------
    // Internal progress helpers
    // -------------------------------------------------------------------------
 
    void _update_bar(int64_t current, int64_t total);
 
 
    uint16_t n_threads_requested;
 
    std::mutex frame_mutexes_access;
 
    std::mutex triggers_map_mutex;
 
    std::mutex spilldata_masks_mutex;
 
 
 
    uint8_t _current_spill;
 
    std::map<std::string, uint32_t> _next_spill;
 
    std::variant<std::monostate,
                 mist::logger::progress_bar *,
                 mist::logger::subtask_progress_bar *>
        assigned_bar_;
 
    alcor_spilldata spilldata;
 
    std::vector<alcor_data_streamer> data_streams;
 
 
 
    std::vector<trigger_config> triggers;
 
    std::unordered_map<uint8_t, trigger_config> triggers_map;
 
    readout_config_list readout_config;
 
    uint16_t _frame_size;
 
 
 
    TH2F *h2_fine_tune_distribution;
 
};