CosmicSeedGenerator.cc

Go to the documentation of this file.

//
// Package:         RecoTracker/TkSeedGenerator
// Class:           GlobalPixelLessSeedGenerator
// 

#include <iostream>
#include <memory>
#include <string>

#include "RecoTracker/SpecialSeedGenerators/interface/CosmicSeedGenerator.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;
CosmicSeedGenerator::CosmicSeedGenerator(edm::ParameterSet const& conf) : 
  cosmic_seed(conf),
  check(conf,consumesCollector())

 {
  edm::LogInfo ("CosmicSeedGenerator")<<"Enter the CosmicSeedGenerator";
  // get Inputs
  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.
CosmicSeedGenerator::~CosmicSeedGenerator() { }  

// Functions that gets called by framework every event
void CosmicSeedGenerator::produce(edm::Event& ev, const edm::EventSetup& es)
{
  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>();

  //check on the number of clusters
  size_t clustsOrZero = check.tooManyClusters(ev);
  if (!clustsOrZero){
    cosmic_seed.init(*stereorecHits,*rphirecHits,*matchedrecHits, es);
    
    // invoke the seed finding algorithm
    cosmic_seed.run(*output,es);
  } else edm::LogError("TooManyClusters") << "Found too many clusters (" << clustsOrZero << "), bailing out.\n";

  // write output to file
  LogDebug("CosmicSeedGenerator")<<" number of seeds = "<< output->size();


  ev.put(std::move(output));
}