SecondaryVertex/interface/GhostTrackComputer.h

0001 #ifndef RecoBTag_SecondaryVertex_GhostTrackComputer_h
0002 #define RecoBTag_SecondaryVertex_GhostTrackComputer_h
0003
0004 #include "DataFormats/BTauReco/interface/CandIPTagInfo.h"
0005 #include "DataFormats/BTauReco/interface/CandSecondaryVertexTagInfo.h"
0006 #include "DataFormats/BTauReco/interface/SecondaryVertexTagInfo.h"
0007 #include "DataFormats/BTauReco/interface/TaggingVariable.h"
0008 #include "DataFormats/BTauReco/interface/TrackIPTagInfo.h"
0009 #include "DataFormats/GeometryVector/interface/GlobalPoint.h"
0010 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0011 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0012 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
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   static void fillPSetDescription(edm::ParameterSetDescription &desc);
0026
0027 private:
0028   const reco::btag::TrackIPData &threshTrack(const reco::TrackIPTagInfo &trackIPTagInfo,
0029                                              const reco::btag::SortCriteria sort,
0030                                              const reco::Jet &jet,
0031                                              const GlobalPoint &pv) const;
0032   const reco::btag::TrackIPData &threshTrack(const reco::CandIPTagInfo &trackIPTagInfo,
0033                                              const reco::btag::SortCriteria sort,
0034                                              const reco::Jet &jet,
0035                                              const GlobalPoint &pv) const;
0036
0037   double charmCut;
0038   reco::btag::SortCriteria sortCriterium;
0039   reco::TrackSelector trackSelector;
0040   reco::TrackSelector trackNoDeltaRSelector;
0041   double minTrackWeight;
0042   bool vertexMassCorrection;
0043   reco::V0Filter trackPairV0Filter;
0044 };
0045
0046 #endif  // RecoBTag_SecondaryVertex_GhostTrackComputer_h