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) : cosmic_seed(conf, consumesCollector()) {
  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));
}