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