SecondaryVertex/interface/GhostTrackComputer.h

0001 #ifndef RecoBTag_SecondaryVertex_GhostTrackComputer_h
0002 #define RecoBTag_SecondaryVertex_GhostTrackComputer_h
0003
0004 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0005
0006 #include "DataFormats/GeometryVector/interface/GlobalPoint.h"
0007 #include "DataFormats/BTauReco/interface/TrackIPTagInfo.h"
0008 #include "DataFormats/BTauReco/interface/CandIPTagInfo.h"
0009 #include "DataFormats/BTauReco/interface/SecondaryVertexTagInfo.h"
0010 #include "DataFormats/BTauReco/interface/CandSecondaryVertexTagInfo.h"
0011 #include "DataFormats/BTauReco/interface/TaggingVariable.h"
0012
0013 #include "RecoBTag/SecondaryVertex/interface/TrackSelector.h"
0014 #include "RecoBTag/SecondaryVertex/interface/V0Filter.h"
0015
0016 class GhostTrackComputer {
0017 public:
0018   GhostTrackComputer(const edm::ParameterSet &params);
0019   virtual ~GhostTrackComputer() = default;
0020   virtual reco::TaggingVariableList operator()(const reco::TrackIPTagInfo &ipInfo,
0021                                                const reco::SecondaryVertexTagInfo &svInfo) const;
0022   virtual reco::TaggingVariableList operator()(const reco::CandIPTagInfo &ipInfo,
0023                                                const reco::CandSecondaryVertexTagInfo &svInfo) const;
0024
0025 private:
0026   const reco::btag::TrackIPData &threshTrack(const reco::TrackIPTagInfo &trackIPTagInfo,
0027                                              const reco::btag::SortCriteria sort,
0028                                              const reco::Jet &jet,
0029                                              const GlobalPoint &pv) const;
0030   const reco::btag::TrackIPData &threshTrack(const reco::CandIPTagInfo &trackIPTagInfo,
0031                                              const reco::btag::SortCriteria sort,
0032                                              const reco::Jet &jet,
0033                                              const GlobalPoint &pv) const;
0034
0035   double charmCut;
0036   reco::btag::SortCriteria sortCriterium;
0037   reco::TrackSelector trackSelector;
0038   reco::TrackSelector trackNoDeltaRSelector;
0039   double minTrackWeight;
0040   bool vertexMassCorrection;
0041   reco::V0Filter trackPairV0Filter;
0042 };
0043
0044 #endif  // RecoBTag_SecondaryVertex_GhostTrackComputer_h