L3MuonCleaner.cc

Go to the documentation of this file.

// -*- C++ -*-
// Framework
#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/PluginManager/interface/ModuleDef.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

class L3MuonCleaner : public edm::global::EDProducer<> {
 public:
  L3MuonCleaner(const edm::ParameterSet&);
  ~L3MuonCleaner() override{}
  void produce(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
 private:
  edm::InputTag m_input; 
  int m_minTrkHits;
  int m_minMuonHits;
  double m_maxNormalizedChi2;
  edm::EDGetTokenT<reco::TrackCollection> inputToken_;



};

L3MuonCleaner::L3MuonCleaner(const edm::ParameterSet& parameterSet){
  m_input = parameterSet.getParameter<edm::InputTag>("input");
  m_minTrkHits = parameterSet.getParameter<int>("minTrkHits");
  m_minMuonHits = parameterSet.getParameter<int>("minMuonHits");
  m_maxNormalizedChi2 = parameterSet.getParameter<double>("maxNormalizedChi2");
  inputToken_ = consumes<reco::TrackCollection>(m_input);

  produces<reco::TrackCollection>();
}

void L3MuonCleaner::produce(edm::StreamID, edm::Event& event, const edm::EventSetup&) const{
  edm::Handle<reco::TrackCollection> tracks; 
  event.getByToken(inputToken_,tracks);
  auto outTracks = std::make_unique<reco::TrackCollection>();
  for ( reco::TrackCollection::const_iterator trk=tracks->begin(); trk!=tracks->end(); ++trk ){
    if (trk->normalizedChi2()>m_maxNormalizedChi2) continue;
    if (trk->hitPattern().numberOfValidTrackerHits()<m_minTrkHits) continue;
    if (trk->hitPattern().numberOfValidMuonHits()<m_minMuonHits) continue;
    outTracks->push_back(*trk);
  }
  event.put(std::move(outTracks));
}
DEFINE_FWK_MODULE(L3MuonCleaner);