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