TICLSeedingRegionProducer.cc

Go to the documentation of this file.

// Authors: Felice Pantaleo, Marco Rovere
// Emails: felice.pantaleo@cern.ch, marco.rovere@cern.ch
// Date: 06/2019
 
#include <vector>
 
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "RecoHGCal/TICL/plugins/SeedingRegionAlgoBase.h"
#include "SeedingRegionByTracks.h"
#include "SeedingRegionGlobal.h"
 
using namespace ticl;
 
class TICLSeedingRegionProducer : public edm::stream::EDProducer<> {
public:
  TICLSeedingRegionProducer(const edm::ParameterSet&);
  ~TICLSeedingRegionProducer() override {}
  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
 
  void beginRun(edm::Run const& iEvent, edm::EventSetup const& es) override;
 
  void produce(edm::Event&, const edm::EventSetup&) override;
 
private:
  std::unique_ptr<SeedingRegionAlgoBase> myAlgo_;
  int algoId_;
  std::string seedingId_;
};
DEFINE_FWK_MODULE(TICLSeedingRegionProducer);
 
TICLSeedingRegionProducer::TICLSeedingRegionProducer(const edm::ParameterSet& ps)
    : algoId_(ps.getParameter<int>("algoId")) {
  auto sumes = consumesCollector();
 
  switch (algoId_) {
    case 1:
      myAlgo_ = std::make_unique<SeedingRegionByTracks>(ps, sumes);
      break;
    case 2:
      myAlgo_ = std::make_unique<SeedingRegionGlobal>(ps, sumes);
      break;
    default:
      break;
  }
  produces<std::vector<TICLSeedingRegion>>();
}
 
void TICLSeedingRegionProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<int>("algo_verbosity", 0);
  desc.add<edm::InputTag>("tracks", edm::InputTag("generalTracks"));
  desc.add<std::string>("cutTk",
                        "1.48 < abs(eta) < 3.0 && pt > 1. && quality(\"highPurity\") && "
                        "hitPattern().numberOfLostHits(\"MISSING_OUTER_HITS\") < 5");
  desc.add<std::string>("propagator", "PropagatorWithMaterial");
  desc.add<int>("algoId", 1);
  descriptions.add("ticlSeedingRegionProducer", desc);
}
 
void TICLSeedingRegionProducer::beginRun(edm::Run const& iEvent, edm::EventSetup const& es) { myAlgo_->initialize(es); }
 
void TICLSeedingRegionProducer::produce(edm::Event& evt, const edm::EventSetup& es) {
  auto result = std::make_unique<std::vector<TICLSeedingRegion>>();
  myAlgo_->makeRegions(evt, es, *result);
 
  evt.put(std::move(result));
}