#include <TracksterLinkingbyFastJet.h>

Inheritance diagram for ticl::TracksterLinkingbyFastJet:

Public Member Functions
void	initialize (const HGCalDDDConstants *hgcons, const hgcal::RecHitTools rhtools, const edm::ESHandle< MagneticField > bfieldH, const edm::ESHandle< Propagator > propH) override

void	linkTracksters (const Inputs &input, std::vector< Trackster > &resultTracksters, std::vector< std::vector< unsigned int >> &linkedResultTracksters, std::vector< std::vector< unsigned int >> &linkedTracksterIdToInputTracksterId) override

	TracksterLinkingbyFastJet (const edm::ParameterSet &conf, edm::ConsumesCollector iC, cms::Ort::ONNXRuntime const *onnxRuntime=nullptr)

	~TracksterLinkingbyFastJet () override

Public Member Functions inherited from ticl::TracksterLinkingAlgoBase
virtual void	setEvent (edm::Event &iEvent, edm::EventSetup const &iEventSetup)

	TracksterLinkingAlgoBase (const edm::ParameterSet &conf, edm::ConsumesCollector, cms::Ort::ONNXRuntime const *onnxRuntime=nullptr)

virtual	~TracksterLinkingAlgoBase ()

Static Public Member Functions
static void	fillPSetDescription (edm::ParameterSetDescription &iDesc)

Static Public Member Functions inherited from ticl::TracksterLinkingAlgoBase
static void	fillPSetDescription (edm::ParameterSetDescription &desc)

Private Attributes
fastjet::JetAlgorithm	algorithm_

const float	radius_

Additional Inherited Members
Protected Attributes inherited from ticl::TracksterLinkingAlgoBase
int	algo_verbosity_

cms::Ort::ONNXRuntime const *	onnxRuntime_

Detailed Description

Definition at line 14 of file TracksterLinkingbyFastJet.h.

Constructor & Destructor Documentation

◆ TracksterLinkingbyFastJet()

ticl::TracksterLinkingbyFastJet::TracksterLinkingbyFastJet	(	const edm::ParameterSet &	conf,
		edm::ConsumesCollector	iC,
		cms::Ort::ONNXRuntime const *	onnxRuntime = `nullptr`
	)

inline

Definition at line 16 of file TracksterLinkingbyFastJet.h.

References algorithm_, Exception, and edm::ParameterSet::getParameter().

         : TracksterLinkingAlgoBase(conf, iC), radius_(conf.getParameter<double>("radius")) {
       // Cluster tracksters into jets using FastJet with configurable algorithm
       auto algo = conf.getParameter<int>("jet_algorithm");
 
       switch (algo) {
         case 0:
           algorithm_ = fastjet::kt_algorithm;
           break;
         case 1:
           algorithm_ = fastjet::cambridge_algorithm;
           break;
         case 2:
           algorithm_ = fastjet::antikt_algorithm;
           break;
         default:
           throw cms::Exception("BadConfig") << "FastJet jet clustering algorithm not set correctly.";
       }
     }

◆ ~TracksterLinkingbyFastJet()

ticl::TracksterLinkingbyFastJet::~TracksterLinkingbyFastJet ( )

inlineoverride

Definition at line 38 of file TracksterLinkingbyFastJet.h.

38 {}

Member Function Documentation

◆ fillPSetDescription()

static void ticl::TracksterLinkingbyFastJet::fillPSetDescription ( edm::ParameterSetDescription & iDesc )

inlinestatic

Definition at line 50 of file TracksterLinkingbyFastJet.h.

References edm::ParameterSetDescription::add().

                                                                        {
       iDesc.add<int>("algo_verbosity", 0);
       iDesc.add<int>("jet_algorithm", 2)
           ->setComment("FastJet jet clustering algorithm: 0 = kt, 1 = Cambridge/Aachen, 2 = anti-kt");
       iDesc.add<double>("radius", 0.1);
     }

◆ initialize()

void ticl::TracksterLinkingbyFastJet::initialize	(	const HGCalDDDConstants *	hgcons,
		const hgcal::RecHitTools	rhtools,
		const edm::ESHandle< MagneticField >	bfieldH,
		const edm::ESHandle< Propagator >	propH
	)

inlineoverridevirtual

Implements ticl::TracksterLinkingAlgoBase.

Definition at line 45 of file TracksterLinkingbyFastJet.h.

48 {};

◆ linkTracksters()

void TracksterLinkingbyFastJet::linkTracksters	(	const Inputs &	input,
		std::vector< Trackster > &	resultTracksters,
		std::vector< std::vector< unsigned int >> &	linkedResultTracksters,
		std::vector< std::vector< unsigned int >> &	linkedTracksterIdToInputTracksterId
	)

overridevirtual

Implements ticl::TracksterLinkingAlgoBase.

Definition at line 8 of file TracksterLinkingbyFastJet.cc.

References algorithm_, mps_fire::i, input, metsig::jet, PDWG_EXODelayedJetMET_cff::jets, ticl::Trackster::mergeTracksters(), and radius_.

                                                                              {
   // Create jets of tracksters using FastJet
   std::vector<fastjet::PseudoJet> fjInputs;
   for (size_t i = 0; i < input.tracksters.size(); ++i) {
     // Convert Trackster information to PseudoJet
     fastjet::PseudoJet pj(input.tracksters[i].barycenter().x(),
                           input.tracksters[i].barycenter().y(),
                           input.tracksters[i].barycenter().z(),
                           input.tracksters[i].raw_energy());
     pj.set_user_index(i);
     fjInputs.push_back(pj);
   }
 
   // Cluster tracksters into jets using FastJet
   fastjet::ClusterSequence sequence(fjInputs, fastjet::JetDefinition(algorithm_, radius_));
   auto jets = fastjet::sorted_by_pt(sequence.inclusive_jets(0));
   linkedTracksterIdToInputTracksterId.resize(jets.size());
   // Link tracksters based on which ones are components of the same jet
   for (unsigned int i = 0; i < jets.size(); ++i) {
     const auto& jet = jets[i];
 
     std::vector<unsigned int> linkedTracksters;
     Trackster outTrackster;
     if (!jet.constituents().empty()) {
       // Check if a trackster is a component of the current jet
       for (const auto& constituent : jet.constituents()) {
         auto tracksterIndex = constituent.user_index();
         linkedTracksterIdToInputTracksterId[i].push_back(tracksterIndex);
         outTrackster.mergeTracksters(input.tracksters[tracksterIndex]);
       }
       linkedTracksters.push_back(resultTracksters.size());
       resultTracksters.push_back(outTrackster);
       // Store the linked tracksters
       linkedResultTracksters.push_back(linkedTracksters);
     }
   }
 }