MuonAssociators/plugins/TriggerObjectFilterByCollection.cc

0001 //
0002 // $Id: TriggerObjectFilterByCollection.cc,v 1.1 2012/08/02 14:34:28 gpetrucc Exp $
0003 //
0004
0005 /**
0006   \class     TriggerObjectFilterByCollection.h "MuonAnalysis/TagAndProbe/plugins/TriggerObjectFilterByCollection.h"
0007   \brief    Creates a filtered list of TriggerObjectStandAlone objects selecting by collection label
0008             Inputs are:
0009                 - a list of TriggerObjectStandAlone (param. "src")
0010                 - a list of collections (param. "collections")
0011             Outputs are:
0012                 - a list of TriggerObjectStandAlone
0013
0014   \author   Giovanni Petrucciani
0015   \version  $Id: TriggerObjectFilterByCollection.cc,v 1.1 2012/08/02 14:34:28 gpetrucc Exp $
0016 */
0017
0018 #include <cstring>
0019 #include <string>
0020 #include <vector>
0021 #include <algorithm>
0022
0023 #include "FWCore/Framework/interface/global/EDProducer.h"
0024 #include "FWCore/Framework/interface/Event.h"
0025 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0026 #include "FWCore/Utilities/interface/InputTag.h"
0027 #include "FWCore/Utilities/interface/EDGetToken.h"
0028 #include "DataFormats/PatCandidates/interface/TriggerObjectStandAlone.h"
0029
0030 class TriggerObjectFilterByCollection : public edm::global::EDProducer<> {
0031 public:
0032   explicit TriggerObjectFilterByCollection(const edm::ParameterSet &iConfig);
0033   ~TriggerObjectFilterByCollection() override {}
0034
0035   void produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const override;
0036
0037 private:
0038   edm::EDGetTokenT<std::vector<pat::TriggerObjectStandAlone>> src_;
0039   std::vector<std::string> collections_;
0040 };
0041
0042 TriggerObjectFilterByCollection::TriggerObjectFilterByCollection(const edm::ParameterSet &iConfig)
0043     : src_(consumes<std::vector<pat::TriggerObjectStandAlone>>(iConfig.getParameter<edm::InputTag>("src"))),
0044       collections_(iConfig.getParameter<std::vector<std::string>>("collections")) {
0045   produces<std::vector<pat::TriggerObjectStandAlone>>();
0046   for (unsigned int i = 0, n = collections_.size(); i < n; ++i) {
0047     std::string &c = collections_[i];
0048     int numsc = std::count(c.begin(), c.end(), ':');
0049     if (numsc == 1)
0050       c.push_back(':');
0051     else if (numsc == 2)
0052       c.append("::");
0053   }
0054 }
0055
0056 void TriggerObjectFilterByCollection::produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const {
0057   using namespace edm;
0058
0059   Handle<std::vector<pat::TriggerObjectStandAlone>> src;
0060   iEvent.getByToken(src_, src);
0061
0062   std::unique_ptr<std::vector<pat::TriggerObjectStandAlone>> out(new std::vector<pat::TriggerObjectStandAlone>());
0063   out->reserve(src->size());
0064   for (std::vector<pat::TriggerObjectStandAlone>::const_iterator it = src->begin(), ed = src->end(); it != ed; ++it) {
0065     const std::string &coll = it->collection();
0066     bool found = false;
0067     for (std::vector<std::string>::const_iterator ic = collections_.begin(), ec = collections_.end(); ic != ec; ++ic) {
0068       if (strncmp(coll.c_str(), ic->c_str(), ic->size()) == 0) {
0069         found = true;
0070         break;
0071       }
0072     }
0073     if (found)
0074       out->push_back(*it);
0075   }
0076
0077   iEvent.put(std::move(out));
0078 }
0079
0080 #include "FWCore/Framework/interface/MakerMacros.h"
0081 DEFINE_FWK_MODULE(TriggerObjectFilterByCollection);