Inheritance diagram for TICLCandidateFromTrackstersProducer:

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

	TICLCandidateFromTrackstersProducer (const edm::ParameterSet &)

	~TICLCandidateFromTrackstersProducer () override

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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

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

Private Attributes
float	min_particle_prob_

std::unique_ptr< TracksterMomentumPluginBase >	momentum_algo_

std::unique_ptr< TracksterTrackPluginBase >	track_algo_

std::vector< edm::EDGetTokenT< std::vector< Trackster > > >	trackster_tokens_

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 24 of file TICLCandidateFromTrackstersProducer.cc.

Constructor & Destructor Documentation

◆ TICLCandidateFromTrackstersProducer()

TICLCandidateFromTrackstersProducer::TICLCandidateFromTrackstersProducer ( const edm::ParameterSet & ps )

Definition at line 63 of file TICLCandidateFromTrackstersProducer.cc.

     : min_particle_prob_(ps.getParameter<double>("minParticleProbability")) {
   trackster_tokens_ =
       edm::vector_transform(ps.getParameter<std::vector<edm::InputTag>>("tracksterCollections"),
                             [this](edm::InputTag const& tag) { return consumes<std::vector<Trackster>>(tag); });
   produces<std::vector<TICLCandidate>>();
   auto pset_momentum = ps.getParameter<edm::ParameterSet>("momentumPlugin");
   momentum_algo_ = TracksterMomentumPluginFactory::get()->create(
       pset_momentum.getParameter<std::string>("plugin"), pset_momentum, consumesCollector());
  
   auto pset_track = ps.getParameter<edm::ParameterSet>("trackPlugin");
   track_algo_ = TracksterTrackPluginFactory::get()->create(
       pset_track.getParameter<std::string>("plugin"), pset_track, consumesCollector());
 }

References get, edm::ParameterSet::getParameter(), momentum_algo_, AlCaHLTBitMon_QueryRunRegistry::string, GlobalPosition_Frontier_DevDB_cff::tag, track_algo_, trackster_tokens_, and edm::vector_transform().

◆ ~TICLCandidateFromTrackstersProducer()

TICLCandidateFromTrackstersProducer::~TICLCandidateFromTrackstersProducer ( )

inlineoverride

Definition at line 27 of file TICLCandidateFromTrackstersProducer.cc.

27 {}

Member Function Documentation

◆ fillDescriptions()

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

static

Definition at line 78 of file TICLCandidateFromTrackstersProducer.cc.

                                                                                                      {
   edm::ParameterSetDescription desc;
  
   std::vector<edm::InputTag> source_vector{edm::InputTag("trackstersTrk"),
                                            edm::InputTag("trackstersMIP"),
                                            edm::InputTag("trackstersEM"),
                                            edm::InputTag("trackstersHAD")};
   desc.add<std::vector<edm::InputTag>>("tracksterCollections", source_vector);
   desc.add<double>("minParticleProbability", 0.);
  
   edm::ParameterSetDescription desc_momentum;
   desc_momentum.add<std::string>("plugin", "TracksterP4FromEnergySum");
   desc_momentum.add<bool>("energyFromRegression", false);
   desc_momentum.add<edm::InputTag>("vertices", edm::InputTag("offlinePrimaryVertices"));
   desc_momentum.add<edm::InputTag>("layerClusters", edm::InputTag("hgcalLayerClusters"));
   desc_momentum.add<edm::InputTag>("tracks", edm::InputTag("generalTracks"));
   desc.add<edm::ParameterSetDescription>("momentumPlugin", desc_momentum);
  
   edm::ParameterSetDescription desc_track;
   desc_track.add<std::string>("plugin", "TracksterRecoTrackPlugin");
   desc.add<edm::ParameterSetDescription>("trackPlugin", desc_track);
  
   descriptions.add("ticlCandidateFromTrackstersProducer", desc);
 }

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), submitPVResolutionJobs::desc, HLT_FULL_cff::InputTag, and AlCaHLTBitMon_QueryRunRegistry::string.

◆ produce()

void TICLCandidateFromTrackstersProducer::produce	(	edm::Event &	evt,
		const edm::EventSetup &	es
	)

override

Definition at line 103 of file TICLCandidateFromTrackstersProducer.cc.

                                                                                         {
   auto result = std::make_unique<std::vector<TICLCandidate>>();
  
   std::vector<const Trackster*> trackster_ptrs;
  
   // adds one TICLCandidate for each trackster that has a minimum particle probability
   for (auto& trackster_token : trackster_tokens_) {
     edm::Handle<std::vector<Trackster>> trackster_h;
     evt.getByToken(trackster_token, trackster_h);
     for (size_t i_trackster = 0; i_trackster < trackster_h->size(); ++i_trackster) {
       auto const& trackster = trackster_h->at(i_trackster);
       auto id_prob_begin = std::begin(trackster.id_probabilities());
       auto max_id_prob_it = std::max_element(id_prob_begin, std::end(trackster.id_probabilities()));
       float max_id_prob = *max_id_prob_it;
       if (max_id_prob < min_particle_prob_)
         continue;
  
       trackster_ptrs.push_back(&trackster);
       auto& ticl_cand = result->emplace_back(edm::Ptr<ticl::Trackster>(trackster_h, i_trackster));
       auto max_index = std::distance(id_prob_begin, max_id_prob_it);
       auto pdg_id = pdg_id_from_particle_type(static_cast<ticl::Trackster::ParticleType>(max_index));
       ticl_cand.setPdgId(pdg_id);
     }
   }
  
   momentum_algo_->setP4(trackster_ptrs, *result, evt);
   track_algo_->setTrack(trackster_ptrs, *result, evt);
  
   // charge assignment
   for (size_t i = 0; i < result->size(); ++i) {
     auto& ticl_cand = result->at(i);
     auto pdg_id = ticl_cand.pdgId();
     if (pdg_id == -11 || pdg_id == -13 || pdg_id == 211) {
       if (ticl_cand.trackPtr().isNonnull()) {
         auto charge = ticl_cand.trackPtr()->charge();
         ticl_cand.setCharge(charge);
         ticl_cand.setPdgId(pdg_id * charge);
         // If this is a special 0 energy trackster that has been injected
         // starting from a seed that has not been linked to any shower,
         // re-assign the energy and momentum to be the ones of the track linked
         // to the original seeding region.
         if (ticl_cand.rawEnergy() == 0.) {
           auto const& three_mom = ticl_cand.trackPtr()->momentum();
           constexpr double mpion2 = 0.13957 * 0.13957;
           double energy = std::sqrt(three_mom.mag2() + mpion2);
           math::XYZTLorentzVector trk_p4(three_mom.x(), three_mom.y(), three_mom.z(), energy);
           ticl_cand.setP4(trk_p4);
           ticl_cand.setRawEnergy(energy);
         }
       } else {
         // Demote them to be neutral, since there's not track associated.
         ticl_cand.setCharge(0);
         if (pdg_id == -11) {
           ticl_cand.setPdgId(22);
         } else {
           ticl_cand.setPdgId(130);
         }
       }
     }
   }
  
   evt.put(std::move(result));
 }