JetSubstructurePacker Class Reference

#include <JetSubstructurePacker.h>

Inheritance diagram for JetSubstructurePacker:

Public Member Functions
	JetSubstructurePacker (const edm::ParameterSet &)
	~JetSubstructurePacker () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

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

Private Attributes
std::vector< std::string >	algoLabels_
std::vector< edm::InputTag >	algoTags_
std::vector< edm::EDGetTokenT< edm::View< pat::Jet > > >	algoTokens_
float	distMax_
bool	fixDaughters_
edm::EDGetTokenT< edm::View< pat::Jet > >	jetToken_
edm::EDGetTokenT< edm::Association< reco::PFCandidateCollection > >	pc2pf_
edm::EDGetTokenT< edm::Association< pat::PackedCandidateCollection > >	pf2pc_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 34 of file JetSubstructurePacker.h.

Constructor & Destructor Documentation

JetSubstructurePacker()

JetSubstructurePacker::JetSubstructurePacker ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 5 of file JetSubstructurePacker.cc.

References algoTags_, algoTokens_, fixDaughters_, edm::ParameterSet::getParameter(), pc2pf_, pf2pc_, makeGlobalPositionRcd_cfg::tag, and edm::vector_transform().

    : distMax_(iConfig.getParameter<double>("distMax")),
      jetToken_(consumes<edm::View<pat::Jet>>(iConfig.getParameter<edm::InputTag>("jetSrc"))),
      algoLabels_(iConfig.getParameter<std::vector<std::string>>("algoLabels")),
      algoTags_(iConfig.getParameter<std::vector<edm::InputTag>>("algoTags")),
      fixDaughters_(iConfig.getParameter<bool>("fixDaughters")) {
  algoTokens_ =
      edm::vector_transform(algoTags_, [this](edm::InputTag const& tag) { return consumes<edm::View<pat::Jet>>(tag); });
  if (fixDaughters_) {
    pf2pc_ = consumes<edm::Association<pat::PackedCandidateCollection>>(
        iConfig.getParameter<edm::InputTag>("packedPFCandidates"));
    pc2pf_ = consumes<edm::Association<reco::PFCandidateCollection>>(
        iConfig.getParameter<edm::InputTag>("packedPFCandidates"));
  }
  //register products
  produces<std::vector<pat::Jet>>();
}

~JetSubstructurePacker()

JetSubstructurePacker::~JetSubstructurePacker ( )

override

Definition at line 23 of file JetSubstructurePacker.cc.

{}

Member Function Documentation

produce()

void JetSubstructurePacker::produce ( edm::Event &  iEvent, const edm::EventSetup &    )

overrideprivate

Definition at line 26 of file JetSubstructurePacker.cc.

References algoLabels_, algoTags_, algoTokens_, reco::deltaR(), distMax_, HLT_2023v12_cff::dRMin, spr::find(), fixDaughters_, mps_fire::i, iEvent, jetToken_, eostools::move(), PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, pc2pf_, pf2pc_, and edm::refToPtr().

                                                                          {
  auto outputs = std::make_unique<std::vector<pat::Jet>>();

  edm::Handle<edm::View<pat::Jet>> jetHandle;
  std::vector<edm::Handle<edm::View<pat::Jet>>> algoHandles;

  edm::Handle<edm::Association<pat::PackedCandidateCollection>> pf2pc;
  edm::Handle<edm::Association<reco::PFCandidateCollection>> pc2pf;
  if (fixDaughters_) {
    iEvent.getByToken(pf2pc_, pf2pc);
    iEvent.getByToken(pc2pf_, pc2pf);
  }

  iEvent.getByToken(jetToken_, jetHandle);
  algoHandles.resize(algoTags_.size());
  for (size_t i = 0; i < algoTags_.size(); ++i) {
    iEvent.getByToken(algoTokens_[i], algoHandles[i]);
  }

  // Loop over the input jets that will be modified.
  for (auto const& ijet : *jetHandle) {
    // Copy the jet.
    outputs->push_back(ijet);
    // Loop over the substructure collections
    unsigned int index = 0;

    for (auto const& ialgoHandle : algoHandles) {
      std::vector<edm::Ptr<pat::Jet>> nextSubjets;
      float dRMin = distMax_;
      for (auto const& jjet : *ialgoHandle) {
        if (reco::deltaR(ijet, jjet) < dRMin) {
          for (auto const& userfloatstr : jjet.userFloatNames()) {
            outputs->back().addUserFloat(userfloatstr, jjet.userFloat(userfloatstr));
          }
          for (auto const& userintstr : jjet.userIntNames()) {
            outputs->back().addUserInt(userintstr, jjet.userInt(userintstr));
          }
          for (auto const& usercandstr : jjet.userCandNames()) {
            outputs->back().addUserCand(usercandstr, jjet.userCand(usercandstr));
          }
          for (size_t ida = 0; ida < jjet.numberOfDaughters(); ++ida) {
            reco::CandidatePtr candPtr = jjet.daughterPtr(ida);
            nextSubjets.push_back(edm::Ptr<pat::Jet>(candPtr));
          }
          break;
        }
      }
      outputs->back().addSubjets(nextSubjets, algoLabels_[index]);
      ++index;
    }

    // fix daughters
    if (fixDaughters_) {
      std::vector<reco::CandidatePtr> daughtersInSubjets;
      std::vector<reco::CandidatePtr> daughtersNew;
      const std::vector<reco::CandidatePtr>& jdausPF = outputs->back().daughterPtrVector();
      std::vector<reco::CandidatePtr> jdaus;
      jdaus.reserve(jdausPF.size());
      // Convert the daughters to packed candidates. This is easier than ref-navigating through PUPPI or CHS to particleFlow.
      for (auto const& jdau : jdausPF) {
        jdaus.push_back(edm::refToPtr((*pf2pc)[jdau]));
      }

      for (const edm::Ptr<pat::Jet>& subjet : outputs->back().subjets()) {
        const std::vector<reco::CandidatePtr>& sjdaus = subjet->daughterPtrVector();
        // check that the subjet does not contain any extra constituents not contained in the jet
        bool skipSubjet = false;
        for (const reco::CandidatePtr& dau : sjdaus) {
          if (std::find(jdaus.begin(), jdaus.end(), dau) == jdaus.end()) {
            skipSubjet = true;
            break;
          }
        }
        if (skipSubjet)
          continue;

        daughtersInSubjets.insert(daughtersInSubjets.end(), sjdaus.begin(), sjdaus.end());
        daughtersNew.push_back(reco::CandidatePtr(subjet));
      }
      for (const reco::CandidatePtr& dau : jdaus) {
        if (std::find(daughtersInSubjets.begin(), daughtersInSubjets.end(), dau) == daughtersInSubjets.end()) {
          daughtersNew.push_back(dau);
        }
      }
      outputs->back().clearDaughters();
      for (const auto& dau : daughtersNew)
        outputs->back().addDaughter(dau);
    }
  }

  iEvent.put(std::move(outputs));
}

Member Data Documentation

algoLabels_

std::vector<std::string> JetSubstructurePacker::algoLabels_

private

Definition at line 47 of file JetSubstructurePacker.h.

Referenced by produce().

algoTags_

std::vector<edm::InputTag> JetSubstructurePacker::algoTags_

private

Definition at line 48 of file JetSubstructurePacker.h.

Referenced by JetSubstructurePacker(), and produce().

algoTokens_

std::vector<edm::EDGetTokenT<edm::View<pat::Jet> > > JetSubstructurePacker::algoTokens_

private

Definition at line 49 of file JetSubstructurePacker.h.

Referenced by JetSubstructurePacker(), and produce().

distMax_

float JetSubstructurePacker::distMax_

private

Definition at line 45 of file JetSubstructurePacker.h.

Referenced by produce().

fixDaughters_

bool JetSubstructurePacker::fixDaughters_

private

Definition at line 50 of file JetSubstructurePacker.h.

Referenced by JetSubstructurePacker(), and produce().

jetToken_

edm::EDGetTokenT<edm::View<pat::Jet> > JetSubstructurePacker::jetToken_

private

Definition at line 46 of file JetSubstructurePacker.h.

Referenced by produce().

pc2pf_

edm::EDGetTokenT<edm::Association<reco::PFCandidateCollection> > JetSubstructurePacker::pc2pf_

private

Definition at line 52 of file JetSubstructurePacker.h.

Referenced by JetSubstructurePacker(), and produce().

pf2pc_

edm::EDGetTokenT<edm::Association<pat::PackedCandidateCollection> > JetSubstructurePacker::pf2pc_

private

Definition at line 51 of file JetSubstructurePacker.h.

Referenced by JetSubstructurePacker(), and produce().