TrackPairInfoBuilder.h

Go to the documentation of this file.

#ifndef RecoBTag_FeatureTools_TrackPairInfoBuilder_h
#define RecoBTag_FeatureTools_TrackPairInfoBuilder_h

#include "DataFormats/GeometrySurface/interface/Line.h"
#include "TrackingTools/Records/interface/TransientTrackRecord.h"
#include "TrackingTools/IPTools/interface/IPTools.h"
#include "TrackingTools/PatternTools/interface/TwoTrackMinimumDistance.h"


namespace btagbtvdeep{

class TrackPairInfoBuilder{
public:
    TrackPairInfoBuilder();

    void buildTrackPairInfo(const reco::TransientTrack * it , const reco::TransientTrack * tt, const reco::Vertex & pv, float mass, GlobalVector jetdirection,
                            const std::pair<bool,Measurement1D> & t_ip, const std::pair<bool,Measurement1D> & t_ip2d );

    const float track_pt() const {return track_pt_;}
    const float track_eta() const {return track_eta_;}
    const float track_phi() const {return track_phi_;}
    const float track_dz() const {return track_dz_;}
    const float track_dxy() const {return track_dxy_;}
    const float pca_distance() const {return pca_distance_;}
    const float pca_significance() const {return pca_significance_; }   
    const float pcaSeed_x() const {return pcaSeed_x_;}
    const float pcaSeed_y() const {return pcaSeed_y_;}
    const float pcaSeed_z() const {return pcaSeed_z_;}
    const float pcaSeed_xerr() const {return pcaSeed_xerr_;}
    const float pcaSeed_yerr() const {return pcaSeed_yerr_;}
    const float pcaSeed_zerr() const {return pcaSeed_zerr_;}
    const float pcaTrack_x() const {return pcaTrack_x_;}
    const float pcaTrack_y() const {return pcaTrack_y_;}
    const float pcaTrack_z() const {return pcaTrack_z_;}
    const float pcaTrack_xerr() const {return pcaTrack_xerr_;}
    const float pcaTrack_yerr() const {return pcaTrack_yerr_;}
    const float pcaTrack_zerr() const {return pcaTrack_zerr_;}    
    const float dotprodTrack() const {return dotprodTrack_;}
    const float dotprodSeed() const {return dotprodSeed_;}
    const float pcaSeed_dist() const {return pcaSeed_dist_;}
    const float pcaTrack_dist() const {return pcaTrack_dist_;}    
    const float track_candMass() const {return track_candMass_;}
    const float track_ip2d() const {return track_ip2d_;}
    const float track_ip2dSig() const {return track_ip2dSig_;}
    const float track_ip3d() const {return track_ip3d_;}
    const float track_ip3dSig() const {return track_ip3dSig_; }   
    const float dotprodTrackSeed2D() const {return dotprodTrackSeed2D_;}
    const float dotprodTrackSeed2DV() const {return dotprodTrackSeed2DV_;}
    const float dotprodTrackSeed3D() const {return dotprodTrackSeed3D_;}
    const float dotprodTrackSeed3DV() const {return dotprodTrackSeed3DV_;}    
    const float pca_jetAxis_dist() const {return pca_jetAxis_dist_;}
    const float pca_jetAxis_dotprod() const {return pca_jetAxis_dotprod_;}
    const float pca_jetAxis_dEta() const {return pca_jetAxis_dEta_;}
    const float pca_jetAxis_dPhi() const {return pca_jetAxis_dPhi_;}
    
    


private:

    float track_pt_;
    float track_eta_;
    float track_phi_;
    float track_dz_;
    float track_dxy_;
    float pca_distance_;
    float pca_significance_;    
    float pcaSeed_x_;
    float pcaSeed_y_;
    float pcaSeed_z_;
    float pcaSeed_xerr_;
    float pcaSeed_yerr_;
    float pcaSeed_zerr_;
    float pcaTrack_x_;
    float pcaTrack_y_;
    float pcaTrack_z_;
    float pcaTrack_xerr_;
    float pcaTrack_yerr_;
    float pcaTrack_zerr_;    
    float dotprodTrack_;
    float dotprodSeed_;
    float pcaSeed_dist_;
    float pcaTrack_dist_;    
    float track_candMass_;
    float track_ip2d_;
    float track_ip2dSig_;
    float track_ip3d_;
    float track_ip3dSig_;    
    float dotprodTrackSeed2D_;
    float dotprodTrackSeed2DV_;
    float dotprodTrackSeed3D_;
    float dotprodTrackSeed3DV_;    
    float pca_jetAxis_dist_;
    float pca_jetAxis_dotprod_;
    float pca_jetAxis_dEta_;
    float pca_jetAxis_dPhi_;

};

}

#endif //RecoBTag_FeatureTools_TrackPairInfoBuilder_h