MuonSeedMerger.cc

Go to the documentation of this file.

#include "RecoMuon/MuonSeedGenerator/plugins/MuonSeedMerger.h"
 
// Framework
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
using namespace std;
using namespace edm;
 
MuonSeedMerger::MuonSeedMerger(const ParameterSet& parameterSet) {
  const string metname = "Muon|RecoMuon|MuonSeedMerger";
 
  theSeedCollectionLabels = parameterSet.getParameter<vector<InputTag> >("SeedCollections");
 
  LogTrace(metname) << "MuonSeedMerger will Merge the following seed collections:";
  for (vector<InputTag>::const_iterator label = theSeedCollectionLabels.begin(); label != theSeedCollectionLabels.end();
       ++label)
    LogTrace(metname) << *label;
 
  for (vector<InputTag>::const_iterator label = theSeedCollectionLabels.begin(); label != theSeedCollectionLabels.end();
       ++label) {
    seedTokens.push_back(consumes<edm::View<TrajectorySeed> >(*label));
  }
 
  produces<TrajectorySeedCollection>();
}
 
MuonSeedMerger::~MuonSeedMerger() {}
 
void MuonSeedMerger::produce(Event& event, const EventSetup& eventSetup) {
  const string metname = "Muon|RecoMuon|MuonSeedMerger";
 
  auto output = std::make_unique<TrajectorySeedCollection>();
 
  Handle<View<TrajectorySeed> > seeds;
 
  for (unsigned int i = 0; i < theSeedCollectionLabels.size(); ++i) {
    event.getByToken(seedTokens.at(i), seeds);
 
    LogTrace(metname) << theSeedCollectionLabels.at(i) << " has " << seeds->size() << " seeds";
    for (View<TrajectorySeed>::const_iterator seed = seeds->begin(); seed != seeds->end(); ++seed)
      output->push_back(*seed);
  }
 
  event.put(std::move(output));
}