FeatureTools/interface/TrackInfoBuilder.h

0001 #ifndef RecoBTag_FeatureTools_TrackInfoBuilder_h
0002 #define RecoBTag_FeatureTools_TrackInfoBuilder_h
0003
0004 #include "DataFormats/Candidate/interface/Candidate.h"
0005 #include "DataFormats/Candidate/interface/VertexCompositePtrCandidate.h"
0006 #include "DataFormats/VertexReco/interface/Vertex.h"
0007 #include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
0008
0009 namespace btagbtvdeep {
0010
0011   // adapted from DeepNtuples
0012   class TrackInfoBuilder {
0013   public:
0014     TrackInfoBuilder(edm::ESHandle<TransientTrackBuilder>& build);
0015
0016     void buildTrackInfo(const reco::Candidate* candidate,
0017                         const math::XYZVector& jetDir,
0018                         GlobalVector refjetdirection,
0019                         const reco::Vertex& pv);
0020     const float getTrackDeltaR() const { return trackDeltaR_; }
0021     const float getTrackEta() const { return trackEta_; }
0022     const float getTrackEtaRel() const { return trackEtaRel_; }
0023     const float getTrackJetDistSig() const { return trackJetDistSig_; }
0024     const float getTrackJetDistVal() const { return trackJetDistVal_; }
0025     const float getTrackMomentum() const { return trackMomentum_; }
0026     const float getTrackPPar() const { return trackPPar_; }
0027     const float getTrackPParRatio() const { return trackPParRatio_; }
0028     const float getTrackPtRatio() const { return trackPtRatio_; }
0029     const float getTrackPtRel() const { return trackPtRel_; }
0030     const float getTrackSip2dSig() const { return trackSip2dSig_; }
0031     const float getTrackSip2dVal() const { return trackSip2dVal_; }
0032     const float getTrackSip3dSig() const { return trackSip3dSig_; }
0033     const float getTrackSip3dVal() const { return trackSip3dVal_; }
0034     const float getTrackJetDecayLen() const { return trackJetDecayLen_; }
0035
0036   private:
0037     edm::ESHandle<TransientTrackBuilder> builder_;
0038
0039     float trackMomentum_;
0040     float trackEta_;
0041     float trackEtaRel_;
0042     float trackPtRel_;
0043     float trackPPar_;
0044     float trackDeltaR_;
0045     float trackPtRatio_;
0046     float trackPParRatio_;
0047     float trackSip2dVal_;
0048     float trackSip2dSig_;
0049     float trackSip3dVal_;
0050     float trackSip3dSig_;
0051
0052     float trackJetDistVal_;
0053     float trackJetDistSig_;
0054     float trackJetDecayLen_;
0055   };
0056
0057 }  // namespace btagbtvdeep
0058
0059 #endif  //RecoBTag_FeatureTools_TrackInfoBuilder_h