FeatureTools/interface/SeedingTrackInfoBuilder.h

0001 #ifndef RecoBTag_FeatureTools_SeedingTrackInfoBuilder_h
0002 #define RecoBTag_FeatureTools_SeedingTrackInfoBuilder_h
0003
0004 #include "TrackingTools/Records/interface/TransientTrackRecord.h"
0005 #include "TrackingTools/IPTools/interface/IPTools.h"
0006 #include "RecoBTag/TrackProbability/interface/HistogramProbabilityEstimator.h"
0007
0008 namespace btagbtvdeep {
0009
0010   class SeedingTrackInfoBuilder {
0011   public:
0012     SeedingTrackInfoBuilder();
0013
0014     void buildSeedingTrackInfo(const reco::TransientTrack* it,
0015                                const reco::Vertex& pv,
0016                                const reco::Jet& jet, /*GlobalVector jetdirection,*/
0017                                float mass,
0018                                const std::pair<bool, Measurement1D>& ip,
0019                                const std::pair<bool, Measurement1D>& ip2d,
0020                                float jet_distance,
0021                                float jaxis_dlength,
0022                                HistogramProbabilityEstimator* m_probabilityEstimator,
0023                                bool m_computeProbabilities);
0024
0025     const float pt() const { return pt_; }
0026     const float eta() const { return eta_; }
0027     const float phi() const { return phi_; }
0028     const float mass() const { return mass_; }
0029     const float dz() const { return dz_; }
0030     const float dxy() const { return dxy_; }
0031     const float ip3d() const { return ip3D_; }
0032     const float sip3d() const { return sip3D_; }
0033     const float ip2d() const { return ip2D_; }
0034     const float sip2d() const { return sip2D_; }
0035     const float ip3d_Signed() const { return ip3D_signed_; }
0036     const float sip3d_Signed() const { return sip3D_signed_; }
0037     const float ip2d_Signed() const { return ip2D_signed_; }
0038     const float sip2d_Signed() const { return sip2D_signed_; }
0039     const float chi2reduced() const { return chi2reduced_; }
0040     const float nPixelHits() const { return nPixelHits_; }
0041     const float nHits() const { return nHits_; }
0042     const float jetAxisDistance() const { return jetAxisDistance_; }
0043     const float jetAxisDlength() const { return jetAxisDlength_; }
0044     const float trackProbability3D() const { return trackProbability3D_; }
0045     const float trackProbability2D() const { return trackProbability2D_; }
0046
0047   private:
0048     float pt_;
0049     float eta_;
0050     float phi_;
0051     float mass_;
0052     float dz_;
0053     float dxy_;
0054     float ip3D_;
0055     float sip3D_;
0056     float ip2D_;
0057     float sip2D_;
0058     float ip3D_signed_;
0059     float sip3D_signed_;
0060     float ip2D_signed_;
0061     float sip2D_signed_;
0062     float chi2reduced_;
0063     float nPixelHits_;
0064     float nHits_;
0065     float jetAxisDistance_;
0066     float jetAxisDlength_;
0067     float trackProbability3D_;
0068     float trackProbability2D_;
0069   };
0070 }  // namespace btagbtvdeep
0071
0072 #endif  //RecoBTag_FeatureTools_SeedingTrackInfoBuilder_h