HLTMuonTrackSelector

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
#ifndef HLTrigger_Muon_HLTMuonTrackSelector_h
#define HLTrigger_Muon_HLTMuonTrackSelector_h

/*
* class HLTMuonTrackSelector
* 
* Select tracks matched to the reco::Muon
* 
* based on RecoTracker/FinalTrackSelectors/plugins/TrackCollectionFilterCloner.cc
* 
* Author: Kyeongpil Lee (kplee@cern.ch)
* 
*/

#include "RecoTracker/FinalTrackSelectors/interface/TrackCollectionCloner.h"

#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"

#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/MuonReco/interface/Muon.h"

#include <vector>

class HLTMuonTrackSelector : public edm::global::EDProducer<> {
public:
  explicit HLTMuonTrackSelector(const edm::ParameterSet &);
  ~HLTMuonTrackSelector() override = default;

  using MVACollection = std::vector<float>;

  static void fillDescriptions(edm::ConfigurationDescriptions &);

private:
  void produce(edm::StreamID, edm::Event &, const edm::EventSetup &) const override;

  TrackCollectionCloner collectionCloner;
  const TrackCollectionCloner::Tokens collectionClonerTokens;

  const edm::EDGetTokenT<std::vector<reco::Muon> > token_muon;
  const edm::EDGetTokenT<MVACollection> token_originalMVAVals;
  const bool flag_copyMVA;
};

#endif