GenFilters/plugins/MCMultiSignatureFilter.cc

0001 // -*- C++ -*-
0002 //
0003 // Package:    MCMultiSignatureFilter
0004 // Class:      MCMultiSignatureFilter
0005 //
0006 /*
0007
0008  Description: select events satistify any or all of signatures
0009
0010  Implementation: derived from MCMultiParticleFilter
0011
0012 */
0013
0014 #include "DataFormats/Common/interface/Handle.h"
0015 #include "FWCore/Framework/interface/global/EDFilter.h"
0016 #include "FWCore/Framework/interface/Event.h"
0017 #include "FWCore/Framework/interface/Frameworkfwd.h"
0018 #include "FWCore/Framework/interface/MakerMacros.h"
0019 #include "FWCore/MessageLogger/interface/MessageLogger.h"
0020 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0021 #include "FWCore/Utilities/interface/EDGetToken.h"
0022 #include "FWCore/Utilities/interface/InputTag.h"
0023 #include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
0024
0025 #include <cmath>
0026 #include <cstdlib>
0027 #include <vector>
0028
0029 //
0030 // class declaration
0031 //
0032
0033 class MCMultiSignatureFilter : public edm::global::EDFilter<> {
0034 public:
0035   explicit MCMultiSignatureFilter(const edm::ParameterSet&);
0036
0037 private:
0038   bool filter(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
0039
0040   // ----------member data ---------------------------
0041
0042   const edm::EDGetTokenT<edm::HepMCProduct> token_;
0043   const bool selectAny_;               // if true select events that satistify any signature, otherwise all
0044   const std::vector<int> particleID_;  // particle IDs to look for
0045   std::vector<double> ptMin_;          // minimum Pt of particles
0046   std::vector<double> etaMax_;         // maximum fabs(eta) of particles
0047   std::vector<int> status_;            // status of particles
0048   std::vector<int> minN_;              // minimum number of particles
0049 };
0050
0051 MCMultiSignatureFilter::MCMultiSignatureFilter(const edm::ParameterSet& iConfig)
0052     : token_(consumes<edm::HepMCProduct>(
0053           iConfig.getUntrackedParameter<edm::InputTag>("src", edm::InputTag("generator", "unsmeared")))),
0054       selectAny_(iConfig.getParameter<bool>("SelectAny")),
0055       particleID_(iConfig.getParameter<std::vector<int> >("ParticleID")),
0056       ptMin_(iConfig.getParameter<std::vector<double> >("PtMin")),
0057       etaMax_(iConfig.getParameter<std::vector<double> >("EtaMax")),
0058       status_(iConfig.getParameter<std::vector<int> >("Status")),
0059       minN_(iConfig.getParameter<std::vector<int> >("MinN")) {
0060   if (particleID_.size() != ptMin_.size() || particleID_.size() != etaMax_.size() ||
0061       particleID_.size() != status_.size() || particleID_.size() != minN_.size())
0062     throw cms::Exception("BadConfig") << "MCMultiSignatureFilter: parameter vectors must have the same length";
0063 }
0064
0065 // ------------ method called to skim the data  ------------
0066 bool MCMultiSignatureFilter::filter(edm::StreamID, edm::Event& iEvent, const edm::EventSetup&) const {
0067   edm::Handle<edm::HepMCProduct> evt;
0068   iEvent.getByToken(token_, evt);
0069
0070   unsigned int nSignatures(particleID_.size());
0071   std::vector<int> searchResults(nSignatures, 0);
0072
0073   const HepMC::GenEvent* genEvent = evt->GetEvent();
0074
0075   for (HepMC::GenEvent::particle_const_iterator p = genEvent->particles_begin(); p != genEvent->particles_end(); ++p) {
0076     auto genParticle = *p;
0077     for (unsigned int i = 0; i < nSignatures; ++i) {
0078       if (abs(genParticle->pdg_id()) != abs(particleID_[i]) or genParticle->momentum().perp() < ptMin_[i] or
0079           fabs(genParticle->momentum().eta()) > etaMax_[i] or genParticle->status() != status_[i])
0080         continue;
0081
0082       searchResults[i] += 1;
0083     }
0084   }
0085
0086   bool passedAll(true);
0087   bool passedAny(false);
0088   for (unsigned int i = 0; i < nSignatures; ++i) {
0089     bool passed = searchResults[i] >= minN_[i];
0090     passedAll = passedAll and passed;
0091     passedAny = passedAny or passed;
0092   }
0093
0094   if (selectAny_)
0095     return passedAny;
0096   else
0097     return passedAll;
0098 }
0099
0100 //define this as a plug-in
0101 DEFINE_FWK_MODULE(MCMultiSignatureFilter);