GlobalTrackingTools/plugins/GlobalTrackQualityProducer.h

0001 #ifndef GlobalTrackingTools_GlobalTrackQualityProducer_h
0002 #define GlobalTrackingTools_GlobalTrackQualityProducer_h
0003
0004 #include <string>
0005 #include <vector>
0006
0007 #include "FWCore/Framework/interface/Frameworkfwd.h"
0008 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0009 #include "DataFormats/Common/interface/ValueMap.h"
0010 #include "DataFormats/Common/interface/Handle.h"
0011 #include "DataFormats/TrackReco/interface/Track.h"
0012
0013 #include "FWCore/Framework/interface/stream/EDProducer.h"
0014 #include "FWCore/Framework/interface/Event.h"
0015 #include "FWCore/Framework/interface/EventSetup.h"
0016 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0017 #include "DataFormats/MuonReco/interface/MuonQuality.h"
0018 #include "DataFormats/MuonReco/interface/MuonFwd.h"
0019 #include "RecoMuon/TrackingTools/interface/MuonServiceProxy.h"
0020 #include "RecoMuon/GlobalTrackingTools/interface/GlobalMuonRefitter.h"
0021 #include "RecoMuon/GlobalTrackingTools/interface/GlobalMuonTrackMatcher.h"
0022 #include "DataFormats/MuonReco/interface/MuonTrackLinks.h"
0023 #include "TrackingTools/DetLayers/interface/MeasurementEstimator.h"
0024 #include "TrackingTools/KalmanUpdators/interface/Chi2MeasurementEstimator.h"
0025
0026 class GlobalMuonRefitter;
0027
0028 class GlobalTrackQualityProducer : public edm::stream::EDProducer<> {
0029 public:
0030   explicit GlobalTrackQualityProducer(const edm::ParameterSet& iConfig);
0031
0032   ~GlobalTrackQualityProducer() override;  // {}
0033
0034   // describe the parameters it allows or requires to be in its configuration
0035   static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
0036
0037 private:
0038   void produce(edm::Event&, const edm::EventSetup&) override;
0039   virtual std::pair<double, double> kink(Trajectory& muon) const;
0040   virtual std::pair<double, double> newChi2(Trajectory& muon) const;
0041   virtual double trackProbability(Trajectory& track) const;
0042
0043   edm::InputTag inputCollection_;
0044   edm::InputTag inputLinksCollection_;
0045   edm::EDGetTokenT<reco::TrackCollection> glbMuonsToken;
0046   edm::EDGetTokenT<reco::MuonTrackLinksCollection> linkCollectionToken;
0047   const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> tTopoToken_;
0048   MuonServiceProxy* theService;
0049   GlobalMuonRefitter* theGlbRefitter;
0050   GlobalMuonTrackMatcher* theGlbMatcher;
0051   MeasurementEstimator* theEstimator;
0052   //muon::SelectionType selectionType_;
0053 };
0054 #endif