btau/plugins/HLTmumutktkVtxProducer.h

0001 #ifndef HLTrigger_btau_HLTmumutktkVtxProducer_h
0002 #define HLTrigger_btau_HLTmumutktkVtxProducer_h
0003 //
0004 // Package:    HLTrigger/btau
0005 // Class:      HLTmumutktkVtxProducer
0006 //
0007 /**\class HLTmumutktkVtxProducer
0008 */
0009
0010 #include <vector>
0011 #include <memory>
0012
0013 #include "FWCore/Framework/interface/stream/EDProducer.h"
0014 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0015 #include "FWCore/Framework/interface/Event.h"
0016 #include "FWCore/Framework/interface/EventSetup.h"
0017 #include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"
0018 #include "DataFormats/TrackReco/interface/TrackFwd.h"
0019 #include "DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h"
0020 #include "DataFormats/HLTReco/interface/TriggerRefsCollections.h"
0021 #include "DataFormats/VertexReco/interface/VertexFwd.h"
0022 #include "DataFormats/BeamSpot/interface/BeamSpot.h"
0023 #include "TrackingTools/Records/interface/TransientTrackRecord.h"
0024 #include "TrackingTools/TransientTrack/interface/TransientTrackBuilder.h"
0025 #include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
0026 #include "MagneticField/Engine/interface/MagneticField.h"
0027
0028 namespace edm {
0029   class ConfigurationDescriptions;
0030 }
0031
0032 // ----------------------------------------------------------------------
0033
0034 namespace reco {
0035   class Candidate;
0036   class Track;
0037 }  // namespace reco
0038
0039 class FreeTrajectoryState;
0040 class MagneticField;
0041
0042 class HLTmumutktkVtxProducer : public edm::stream::EDProducer<> {
0043 public:
0044   explicit HLTmumutktkVtxProducer(const edm::ParameterSet&);
0045   ~HLTmumutktkVtxProducer() override = default;
0046
0047   static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
0048
0049   void produce(edm::Event&, const edm::EventSetup&) override;
0050
0051 private:
0052   bool overlap(const reco::TrackRef& trackref1, const reco::TrackRef& trackref2);
0053   static FreeTrajectoryState initialFreeState(const reco::Track&, const MagneticField*);
0054   bool checkPreviousCand(const reco::TrackRef& trackref, const std::vector<reco::RecoChargedCandidateRef>& ref2) const;
0055
0056   const edm::ESGetToken<TransientTrackBuilder, TransientTrackRecord> transientTrackRecordToken_;
0057
0058   const edm::InputTag muCandTag_;
0059   const edm::EDGetTokenT<reco::RecoChargedCandidateCollection> muCandToken_;
0060   const edm::InputTag trkCandTag_;
0061   const edm::EDGetTokenT<reco::RecoChargedCandidateCollection> trkCandToken_;
0062   const edm::InputTag previousCandTag_;
0063   const edm::EDGetTokenT<trigger::TriggerFilterObjectWithRefs> previousCandToken_;
0064
0065   const std::string mfName_;
0066   const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> idealMagneticFieldRecordToken_;
0067
0068   const double thirdTrackMass_;
0069   const double fourthTrackMass_;
0070   const double maxEta_;
0071   const double minPt_;
0072   const double minInvMass_;
0073   const double maxInvMass_;
0074   const double minTrkTrkMass_;
0075   const double maxTrkTrkMass_;
0076   const double minD0Significance_;
0077   const bool oppositeSign_;
0078   const double overlapDR2_;
0079   const edm::InputTag beamSpotTag_;
0080   const edm::EDGetTokenT<reco::BeamSpot> beamSpotToken_;
0081 };
0082
0083 #endif