SeedCreatorFromRegionHitsEDProducerT.h

Go to the documentation of this file.

#ifndef RecoTracker_TkSeedGenerator_SeedCreatorFromRegionHitsEDProducerT_H
#define RecoTracker_TkSeedGenerator_SeedCreatorFromRegionHitsEDProducerT_H

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "DataFormats/Common/interface/Handle.h"

#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "RecoTracker/TkSeedingLayers/interface/SeedComparitorFactory.h"
#include "RecoTracker/TkSeedingLayers/interface/SeedComparitor.h"

#include "RecoTracker/TkHitPairs/interface/RegionsSeedingHitSets.h"

template <typename T_SeedCreator>
class SeedCreatorFromRegionHitsEDProducerT : public edm::stream::EDProducer<> {
public:
  SeedCreatorFromRegionHitsEDProducerT(const edm::ParameterSet& iConfig);
  ~SeedCreatorFromRegionHitsEDProducerT() override = default;

  static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

  void produce(edm::Event& iEvent, const edm::EventSetup& iSetup) override;

private:
  edm::EDGetTokenT<RegionsSeedingHitSets> seedingHitSetsToken_;
  T_SeedCreator seedCreator_;
  std::unique_ptr<SeedComparitor> comparitor_;
};

template <typename T_SeedCreator>
SeedCreatorFromRegionHitsEDProducerT<T_SeedCreator>::SeedCreatorFromRegionHitsEDProducerT(
    const edm::ParameterSet& iConfig)
    : seedingHitSetsToken_(consumes<RegionsSeedingHitSets>(iConfig.getParameter<edm::InputTag>("seedingHitSets"))),
      seedCreator_(iConfig, consumesCollector()) {
  edm::ConsumesCollector iC = consumesCollector();
  edm::ParameterSet comparitorPSet = iConfig.getParameter<edm::ParameterSet>("SeedComparitorPSet");
  std::string comparitorName = comparitorPSet.getParameter<std::string>("ComponentName");
  if (comparitorName != "none") {
    comparitor_ = SeedComparitorFactory::get()->create(comparitorName, comparitorPSet, iC);
  }

  produces<TrajectorySeedCollection>();
}

template <typename T_SeedCreator>
void SeedCreatorFromRegionHitsEDProducerT<T_SeedCreator>::fillDescriptions(
    edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;

  desc.add<edm::InputTag>("seedingHitSets", edm::InputTag("hitPairEDProducer"));
  T_SeedCreator::fillDescriptions(desc);

  edm::ParameterSetDescription descComparitor;
  descComparitor.add<std::string>("ComponentName", "none");
  descComparitor.setAllowAnything();  // until we have moved SeedComparitor too to EDProducers
  desc.add<edm::ParameterSetDescription>("SeedComparitorPSet", descComparitor);

  auto label = std::string("seedCreatorFromRegion") + T_SeedCreator::fillDescriptionsLabel() + "EDProducer";
  descriptions.add(label, desc);
}

template <typename T_SeedCreator>
void SeedCreatorFromRegionHitsEDProducerT<T_SeedCreator>::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  edm::Handle<RegionsSeedingHitSets> hseedingHitSets;
  iEvent.getByToken(seedingHitSetsToken_, hseedingHitSets);
  const auto& seedingHitSets = *hseedingHitSets;

  auto seeds = std::make_unique<TrajectorySeedCollection>();
  seeds->reserve(seedingHitSets.size());

  if (comparitor_)
    comparitor_->init(iEvent, iSetup);

  for (const auto& regionSeedingHitSets : seedingHitSets) {
    const TrackingRegion& region = regionSeedingHitSets.region();
    seedCreator_.init(region, iSetup, comparitor_.get());

    for (const SeedingHitSet& hits : regionSeedingHitSets) {
      // TODO: do we really need a comparitor at this point? It is
      // used in triplet and quadruplet generators, as well as inside
      // seedCreator.
      if (!comparitor_ || comparitor_->compatible(hits)) {
        seedCreator_.makeSeed(*seeds, hits);
      }
    }
  }

  seeds->shrink_to_fit();
  iEvent.put(std::move(seeds));
}

#endif