NJetsMC.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:   NJetsMC
// Class:     NJetsMC
// 
//
// Original Author:  "Nathaniel Odell"
//         Created:  Thu Aug 12 09:24:46 CDT 2010
// then moved to more general N-jets purpose in GeneratorInterface/GenFilters
//


// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/JetReco/interface/GenJetCollection.h"
#include "DataFormats/JetReco/interface/GenJet.h"

#include "CommonTools/UtilAlgos/interface/TFileService.h"

#include "TROOT.h"
#include "TH1F.h"
#include "TFile.h"
#include "TSystem.h"
#include <iostream>

using namespace edm;
using namespace std;
using namespace reco;

//
// class declaration
//

class NJetsMC : public edm::EDFilter 
{
public:
  explicit NJetsMC(const edm::ParameterSet&);
  ~NJetsMC() override;
  
private:
  void beginJob() override ;
  bool filter(edm::Event&, const edm::EventSetup&) override;
  void endJob() override ;
  
  // ----------member data ---------------------------
  
  edm::EDGetTokenT<reco::GenJetCollection> GenToken_;
  Int_t  njets_;
  double minpt_;

};

NJetsMC::NJetsMC(const edm::ParameterSet& iConfig):
  GenToken_(consumes<reco::GenJetCollection>(iConfig.getUntrackedParameter<edm::InputTag>("GenTag"))),
  njets_(iConfig.getParameter<int32_t>("Njets")),
  minpt_(iConfig.getParameter<double>("MinPt"))
{
}


NJetsMC::~NJetsMC()
{
}

bool NJetsMC::filter(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
   using namespace edm;
   
   Handle<reco::GenJetCollection> GenJets;
   iEvent.getByToken(GenToken_, GenJets);

   Int_t count = 0;
   bool result = false;

   for(GenJetCollection::const_iterator iJet = GenJets->begin(); iJet != GenJets->end(); ++iJet)
     {
       reco::GenJet myJet = reco::GenJet(*iJet);

       if(myJet.pt() > minpt_) ++count;
     }

   if( count >= njets_ )
      result = true;

   return result;
}

void NJetsMC::beginJob()
{
}

void NJetsMC::endJob()
{
}

//define this as a plug-in
DEFINE_FWK_MODULE(NJetsMC);