alcor_recotrackdata.h

Go to the documentation of this file.

#pragma once
 
#include "alcor_recodata.h"
#include "tracking_altai.h"
#include <vector>
#include <cmath>
#include "TTree.h"
 
// =========================================================================
//  Plain data struct
// =========================================================================
 
struct alcor_recotrackdata_struct
{
    float det_plane_x;     
    float det_plane_y;     
    float traj_angcoeff_x; 
    float traj_angcoeff_y; 
    float chi2ndof;        
 
    alcor_recotrackdata_struct() = default;
 
    alcor_recotrackdata_struct(alcor_recodata &v);
};
 
// =========================================================================
//  Container class
// =========================================================================
 
class alcor_recotrackdata : public alcor_recodata
{
public:
    // -------------------------------------------------------------------------
 
 
    alcor_recotrackdata() = default;
 
    alcor_recotrackdata(alcor_recodata &v);
 
 
    // -------------------------------------------------------------------------
 
 
    alcor_recotrackdata_struct &recotrackdata_at(std::size_t idx);
 
    const alcor_recotrackdata_struct &recotrackdata_at(std::size_t idx) const { return recotrackdata.at(idx); }
 
    void add_recotrackdata(const alcor_recotrackdata_struct &entry) { recotrackdata.push_back(entry); }
 
    std::size_t n_recotrackdata() const { return recotrackdata.size(); }
 
    float get_det_plane_x(std::size_t idx) const { return recotrackdata.at(idx).det_plane_x; }
    float get_det_plane_y(std::size_t idx) const { return recotrackdata.at(idx).det_plane_y; }
    float get_traj_angcoeff_x(std::size_t idx) const { return recotrackdata.at(idx).traj_angcoeff_x; }
    float get_traj_angcoeff_y(std::size_t idx) const { return recotrackdata.at(idx).traj_angcoeff_y; }
    float get_chi2ndof(std::size_t idx) const { return recotrackdata.at(idx).chi2ndof; }
 
 
    // -------------------------------------------------------------------------
 
 
    double get_det_plane_r(std::size_t idx) const
    {
        return std::hypot(recotrackdata.at(idx).det_plane_x,
                          recotrackdata.at(idx).det_plane_y);
    }
 
    double get_det_plane_phi(std::size_t idx) const
    {
        return std::atan2(recotrackdata.at(idx).det_plane_y,
                          recotrackdata.at(idx).det_plane_x);
    }
 
    double get_traj_angcoeff_theta(std::size_t idx) const
    {
        return std::hypot(recotrackdata.at(idx).traj_angcoeff_x,
                          recotrackdata.at(idx).traj_angcoeff_y);
    }
 
    double get_traj_angcoeff_phi(std::size_t idx) const
    {
        return std::atan2(recotrackdata.at(idx).traj_angcoeff_y,
                          recotrackdata.at(idx).traj_angcoeff_x);
    }
 
 
    // -------------------------------------------------------------------------
 
 
    void set_det_plane_x(std::size_t idx, float val) { recotrackdata_at(idx).det_plane_x = val; }
    void set_det_plane_y(std::size_t idx, float val) { recotrackdata_at(idx).det_plane_y = val; }
    void set_traj_angcoeff_x(std::size_t idx, float val) { recotrackdata_at(idx).traj_angcoeff_x = val; }
    void set_traj_angcoeff_y(std::size_t idx, float val) { recotrackdata_at(idx).traj_angcoeff_y = val; }
    void set_chi2ndof(std::size_t idx, float val) { recotrackdata_at(idx).chi2ndof = val; }
 
 
    // -------------------------------------------------------------------------
 
 
    void clear();
 
    void link_to_tree(TTree *input_tree);
 
    void write_to_tree(TTree *output_tree);
 
 
    // -------------------------------------------------------------------------
 
 
    void import_event(std::vector<tracking_altai_struct> vec);
 
 
private:
    // -------------------------------------------------------------------------
 
 
    std::vector<alcor_recotrackdata_struct> recotrackdata;                       
    std::vector<alcor_recotrackdata_struct> *recotrackdata_ptr = &recotrackdata; 
 
};
 
// =========================================================================
//  Free functions & constants
// =========================================================================
 
inline double calculate_angle(double detector_to_telescope_plane, double pixel_position)
{
    return std::atan(pixel_position / detector_to_telescope_plane);
}
 
inline double calculate_angle_resolution(double detector_to_telescope_plane, double detector_to_telescope_plane_error,
                                         double pixel_position, double pixel_position_error)
{
    (void)detector_to_telescope_plane;
    (void)detector_to_telescope_plane_error;
    (void)pixel_position;
    (void)pixel_position_error;
    return -1.;
}
 
// --- Detector-to-telescope plane distances [cm] --------------------------
 
inline const double detector_to_telescope_plane_0 = 235.;                                 
inline const double detector_to_telescope_plane_1 = detector_to_telescope_plane_0 + 2.55; 
inline const double detector_to_telescope_plane_2 = detector_to_telescope_plane_1 + 22.5; 
inline const double detector_to_telescope_plane_3 = detector_to_telescope_plane_2 + 2.55; 
 
// --- Pixel resolutions per plane [cm] ------------------------------------
 
inline const double pixel_resolution_plane_0 = 3.7e-3; 
inline const double pixel_resolution_plane_1 = 3.4e-3; 
inline const double pixel_resolution_plane_2 = 3.4e-3; 
inline const double pixel_resolution_plane_3 = 3.7e-3;