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));
}