tracking_altai.h

#pragma once
 
#include <string>
#include <vector>
#include <map>
#include "utility.h"
 
struct tracking_altai_struct
{
    uint32_t event_id;  
    float zero_plane_x; 
    float zero_plane_y; 
    float zero_plane_z; 
    float angcoeff_dx;  
    float angcoeff_dy;  
    float angcoeff_dz;  
    float chi2;         
    float chi2ndof;     
    int ndof;           
    double timestamp;   
};
 
class tracking_altai
{
public:
    tracking_altai() = default;
    explicit tracking_altai(const std::string &file_name);
    ~tracking_altai() = default;
 
 
    std::map<uint32_t, std::vector<tracking_altai_struct>> get_data_map() const;
    uint32_t get_number_of_events() const;
 
    std::vector<tracking_altai_struct> get_event_tracks(uint32_t event_id) const;
 
    int get_event_tracks_size(uint32_t event_id) const;
 
    float get_zero_plane_x(uint32_t event_id, std::size_t idx) const;
    float get_zero_plane_y(uint32_t event_id, std::size_t idx) const;
    float get_zero_plane_z(uint32_t event_id, std::size_t idx) const;
    float get_angcoeff_dx(uint32_t event_id, std::size_t idx) const;
    float get_angcoeff_dy(uint32_t event_id, std::size_t idx) const;
    float get_angcoeff_dz(uint32_t event_id, std::size_t idx) const;
    float get_chi2(uint32_t event_id, std::size_t idx) const;
    float get_chi2ndof(uint32_t event_id, std::size_t idx) const;
    int get_ndof(uint32_t event_id, std::size_t idx) const;
    double get_timestamp(uint32_t event_id, std::size_t idx) const;
 
 
    void add_event_track(uint32_t event_id, const tracking_altai_struct &track);
 
    void set_event_tracks(uint32_t event_id, const std::vector<tracking_altai_struct> &tracks);
 
    bool event_has_one_track(uint32_t event_id) const;
    bool event_has_at_least_one_track(uint32_t event_id) const;
 
    void load_tracking_file(const std::string &input_file);
 
private:
    std::map<uint32_t, std::vector<tracking_altai_struct>> data_map; 
};