#include <SeedGeneratorFromProtoTracksEDProducer.h>

Inheritance diagram for SeedGeneratorFromProtoTracksEDProducer:

Public Member Functions
void	produce (edm::Event &ev, const edm::EventSetup &es) override

	SeedGeneratorFromProtoTracksEDProducer (const edm::ParameterSet &cfg)

	~SeedGeneratorFromProtoTracksEDProducer () override

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () const final

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Attributes
const std::string	builderName

const bool	includeFourthHit_

const double	originHalfLength

const double	originRadius

const edm::ParameterSet	theConfig

const edm::EDGetTokenT< reco::TrackCollection >	theInputCollectionTag

const edm::EDGetTokenT< reco::VertexCollection >	theInputVertexCollectionTag

const bool	useEventsWithNoVertex

const bool	useProtoTrackKinematics

const bool	usePV_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 14 of file SeedGeneratorFromProtoTracksEDProducer.h.

Constructor & Destructor Documentation

SeedGeneratorFromProtoTracksEDProducer::SeedGeneratorFromProtoTracksEDProducer ( const edm::ParameterSet & cfg )

Definition at line 66 of file SeedGeneratorFromProtoTracksEDProducer.cc.

     : theConfig(cfg),
       originHalfLength(cfg.getParameter<double>("originHalfLength")),
       originRadius(cfg.getParameter<double>("originRadius")),
       useProtoTrackKinematics(cfg.getParameter<bool>("useProtoTrackKinematics")),
       useEventsWithNoVertex(cfg.getParameter<bool>("useEventsWithNoVertex")),
       builderName(cfg.getParameter<std::string>("TTRHBuilder")),
       usePV_(cfg.getParameter<bool>("usePV")),
       includeFourthHit_(cfg.getParameter<bool>("includeFourthHit")),
       theInputCollectionTag(consumes<reco::TrackCollection>(cfg.getParameter<InputTag>("InputCollection"))),
       theInputVertexCollectionTag(
           consumes<reco::VertexCollection>(cfg.getParameter<InputTag>("InputVertexCollection"))) {
   produces<TrajectorySeedCollection>();
 }

SeedGeneratorFromProtoTracksEDProducer::~SeedGeneratorFromProtoTracksEDProducer ( )

inlineoverride

Definition at line 17 of file SeedGeneratorFromProtoTracksEDProducer.h.

References ev, and fillDescriptions().

17 {}

Member Function Documentation

void SeedGeneratorFromProtoTracksEDProducer::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 40 of file SeedGeneratorFromProtoTracksEDProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), HLT_2018_cff::InputTag, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                                         {
   edm::ParameterSetDescription desc;
   desc.add<InputTag>("InputCollection", InputTag("pixelTracks"));
   desc.add<InputTag>("InputVertexCollection", InputTag(""));
   desc.add<double>("originHalfLength", 1E9);
   desc.add<double>("originRadius", 1E9);
   desc.add<bool>("useProtoTrackKinematics", false);
   desc.add<bool>("useEventsWithNoVertex", true);
   desc.add<std::string>("TTRHBuilder", "TTRHBuilderWithoutAngle4PixelTriplets");
   desc.add<bool>("usePV", false);
   desc.add<bool>("includeFourthHit", false);
 
   edm::ParameterSetDescription psd0;
   psd0.add<std::string>("ComponentName", std::string("SeedFromConsecutiveHitsCreator"));
   psd0.add<std::string>("propagator", std::string("PropagatorWithMaterial"));
   psd0.add<double>("SeedMomentumForBOFF", 5.0);
   psd0.add<double>("OriginTransverseErrorMultiplier", 1.0);
   psd0.add<double>("MinOneOverPtError", 1.0);
   psd0.add<std::string>("magneticField", std::string(""));
   psd0.add<std::string>("TTRHBuilder", std::string("WithTrackAngle"));
   psd0.add<bool>("forceKinematicWithRegionDirection", false);
   desc.add<edm::ParameterSetDescription>("SeedCreatorPSet", psd0);
 
   descriptions.add("SeedGeneratorFromProtoTracksEDProducer", desc);
 }

void SeedGeneratorFromProtoTracksEDProducer::produce	(	edm::Event &	ev,
		const edm::EventSetup &	es
	)

override

need optimization: all es stuff should go out of the loop

Definition at line 81 of file SeedGeneratorFromProtoTracksEDProducer.cc.

References builderName, edm::EventSetup::get(), edm::Event::getByToken(), edm::ParameterSet::getParameter(), hfClusterShapes_cfi::hits, includeFourthHit_, SeedFromProtoTrack::isValid(), reco::TrackBase::momentum(), eostools::move(), SeedingHitSet::nullPtr(), originHalfLength, originRadius, edm::Event::put(), reco::Track::recHit(), reco::Track::recHitsSize(), rpcPointValidation_cfi::refHit, HLT_2018_cff::region, mps_fire::result, SeedFromConsecutiveHitsCreator_cfi::SeedFromConsecutiveHitsCreator, sqr(), mathSSE::sqrt(), theConfig, theInputCollectionTag, theInputVertexCollectionTag, SeedFromProtoTrack::trajectorySeed(), useEventsWithNoVertex, useProtoTrackKinematics, usePV_, reco::TrackBase::vertex(), pwdgSkimBPark_cfi::vertices, and badGlobalMuonTaggersAOD_cff::vtx.

                                                                                           {
   auto result = std::make_unique<TrajectorySeedCollection>();
   Handle<reco::TrackCollection> trks;
   ev.getByToken(theInputCollectionTag, trks);
 
   const TrackCollection& protos = *(trks.product());
 
   edm::Handle<reco::VertexCollection> vertices;
   bool foundVertices = ev.getByToken(theInputVertexCollectionTag, vertices);
   //const reco::VertexCollection & vertices = *(h_vertices.product());
 
   for (TrackCollection::const_iterator it = protos.begin(); it != protos.end(); ++it) {
     const Track& proto = (*it);
     GlobalPoint vtx(proto.vertex().x(), proto.vertex().y(), proto.vertex().z());
 
     // check the compatibility with a primary vertex
     bool keepTrack = false;
     if ((!foundVertices) || vertices->empty()) {
       if (useEventsWithNoVertex)
         keepTrack = true;
     } else if (usePV_) {
       GlobalPoint aPV(
           vertices->begin()->position().x(), vertices->begin()->position().y(), vertices->begin()->position().z());
       double distR2 = sqr(vtx.x() - aPV.x()) + sqr(vtx.y() - aPV.y());
       double distZ = fabs(vtx.z() - aPV.z());
       if (distR2 < sqr(originRadius) && distZ < originHalfLength) {
         keepTrack = true;
       }
     } else {
       for (reco::VertexCollection::const_iterator iv = vertices->begin(); iv != vertices->end(); ++iv) {
         GlobalPoint aPV(iv->position().x(), iv->position().y(), iv->position().z());
         double distR2 = sqr(vtx.x() - aPV.x()) + sqr(vtx.y() - aPV.y());
         double distZ = fabs(vtx.z() - aPV.z());
         if (distR2 < sqr(originRadius) && distZ < originHalfLength) {
           keepTrack = true;
           break;
         }
       }
     }
     if (!keepTrack)
       continue;
 
     if (useProtoTrackKinematics) {
       SeedFromProtoTrack seedFromProtoTrack(proto, es);
       if (seedFromProtoTrack.isValid())
         (*result).push_back(seedFromProtoTrack.trajectorySeed());
     } else {
       edm::ESHandle<TransientTrackingRecHitBuilder> ttrhbESH;
       es.get<TransientRecHitRecord>().get(builderName, ttrhbESH);
       std::vector<Hit> hits;
       for (unsigned int iHit = 0, nHits = proto.recHitsSize(); iHit < nHits; ++iHit) {
         TrackingRecHitRef refHit = proto.recHit(iHit);
         if (refHit->isValid())
           hits.push_back((Hit) & (*refHit));
       }
       sort(hits.begin(), hits.end(), HitLessByRadius());
 
       if (hits.size() > 1) {
         double mom_perp =
             sqrt(proto.momentum().x() * proto.momentum().x() + proto.momentum().y() * proto.momentum().y());
         GlobalTrackingRegion region(mom_perp, vtx, 0.2, 0.2);
 
         edm::ParameterSet seedCreatorPSet = theConfig.getParameter<edm::ParameterSet>("SeedCreatorPSet");
         SeedFromConsecutiveHitsCreator seedCreator(seedCreatorPSet);
         seedCreator.init(region, es, nullptr);
         seedCreator.makeSeed(
             *result,
             SeedingHitSet(hits[0],
                           hits[1],
                           hits.size() > 2 ? hits[2] : SeedingHitSet::nullPtr(),
                           (includeFourthHit_ && hits.size() > 3) ? hits[3] : SeedingHitSet::nullPtr()));
       }
     }
   }
 
   ev.put(std::move(result));
 }