CRackSeedGenerator.cc

Go to the documentation of this file.

//
// Package:         RecoTracker/TkSeedGenerator
// Class:           GlobalPixelLessSeedGenerator
//
 
#include <iostream>
#include <memory>
#include <string>
 
#include "RecoTracker/SpecialSeedGenerators/interface/CRackSeedGenerator.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Utilities/interface/InputTag.h"
 
using namespace std;
CRackSeedGenerator::CRackSeedGenerator(edm::ParameterSet const& conf) : conf_(conf), cosmic_seed(conf) {
  edm::LogInfo("CRackSeedGenerator") << "Enter the CRackSeedGenerator";
  matchedrecHitsToken_ =
      consumes<SiStripMatchedRecHit2DCollection>(conf_.getParameter<edm::InputTag>("matchedRecHits"));
  rphirecHitsToken_ = consumes<SiStripRecHit2DCollection>(conf_.getParameter<edm::InputTag>("rphirecHits"));
  stereorecHitsToken_ = consumes<SiStripRecHit2DCollection>(conf_.getParameter<edm::InputTag>("stereorecHits"));
 
  produces<TrajectorySeedCollection>();
}
 
// Virtual destructor needed.
CRackSeedGenerator::~CRackSeedGenerator() {}
 
// Functions that gets called by framework every event
void CRackSeedGenerator::produce(edm::Event& ev, const edm::EventSetup& es) {
  // get Inputs
  edm::Handle<SiStripRecHit2DCollection> rphirecHits;
  ev.getByToken(rphirecHitsToken_, rphirecHits);
  edm::Handle<SiStripRecHit2DCollection> stereorecHits;
  ev.getByToken(stereorecHitsToken_, stereorecHits);
  edm::Handle<SiStripMatchedRecHit2DCollection> matchedrecHits;
  ev.getByToken(matchedrecHitsToken_, matchedrecHits);
 
  auto output = std::make_unique<TrajectorySeedCollection>();
  //
 
  cosmic_seed.init(*stereorecHits, *rphirecHits, *matchedrecHits, es);
 
  // invoke the seed finding algorithm
  cosmic_seed.run(*output, es);
 
  // write output to file
  LogDebug("CRackSeedGenerator") << " number of seeds = " << output->size();
 
  ev.put(std::move(output));
}