btau/plugins/HLTmumutktkVtxProducer.h

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