SeedingTrackInfoBuilder.h

Go to the documentation of this file.

#ifndef RecoBTag_FeatureTools_SeedingTrackInfoBuilder_h
#define RecoBTag_FeatureTools_SeedingTrackInfoBuilder_h

#include "TrackingTools/Records/interface/TransientTrackRecord.h"
#include "TrackingTools/IPTools/interface/IPTools.h"
#include "RecoBTag/TrackProbability/interface/HistogramProbabilityEstimator.h"

namespace btagbtvdeep{
    
class SeedingTrackInfoBuilder{
public:
    SeedingTrackInfoBuilder();

    void buildSeedingTrackInfo(const reco::TransientTrack * it , const reco::Vertex & pv,  const reco::Jet & jet,/*GlobalVector jetdirection,*/ 
                               float mass, const std::pair<bool,Measurement1D> & ip, const std::pair<bool,Measurement1D> & ip2d,
                               float jet_distance, float jaxis_dlength, HistogramProbabilityEstimator* m_probabilityEstimator, bool m_computeProbabilities);

    const float pt() const {return pt_;}
    const float eta() const {return eta_;}
    const float phi() const {return phi_;}
    const float mass() const {return mass_;}
    const float dz() const {return dz_;}
    const float dxy() const {return dxy_;}
    const float ip3d() const {return ip3D_;}
    const float sip3d() const {return sip3D_;}
    const float ip2d() const {return ip2D_;}
    const float sip2d() const {return sip2D_;}
    const float ip3d_Signed() const {return ip3D_signed_;}
    const float sip3d_Signed() const {return sip3D_signed_;}
    const float ip2d_Signed() const {return ip2D_signed_;}
    const float sip2d_Signed() const {return sip2D_signed_;}
    const float chi2reduced() const {return chi2reduced_;}
    const float nPixelHits() const {return nPixelHits_;}
    const float nHits() const {return nHits_;}
    const float jetAxisDistance() const {return jetAxisDistance_;}
    const float jetAxisDlength() const {return jetAxisDlength_;}
    const float trackProbability3D() const {return trackProbability3D_;}
    const float trackProbability2D() const {return trackProbability2D_;}   


private:

    float pt_;
    float eta_;
    float phi_;
    float mass_;
    float dz_;
    float dxy_;
    float ip3D_;
    float sip3D_;
    float ip2D_;
    float sip2D_;
    float ip3D_signed_;
    float sip3D_signed_;
    float ip2D_signed_;
    float sip2D_signed_;
    float chi2reduced_;
    float nPixelHits_;
    float nHits_;
    float jetAxisDistance_;
    float jetAxisDlength_;
    float trackProbability3D_;
    float trackProbability2D_;

    

};
}

#endif //RecoBTag_FeatureTools_SeedingTrackInfoBuilder_h