#include <JetSubstructurePacker.h>

Inheritance diagram for JetSubstructurePacker:

Public Member Functions
	JetSubstructurePacker (const edm::ParameterSet &)

	~JetSubstructurePacker ()

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

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
virtual 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<>
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

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 36 of file JetSubstructurePacker.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 5 of file JetSubstructurePacker.cc.

References algoTags_, algoTokens_, fixDaughters_, edm::ParameterSet::getParameter(), pc2pf_, pf2pc_, GlobalPosition_Frontier_DevDB_cff::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 ( )

Definition at line 22 of file JetSubstructurePacker.cc.

23 {

24 }

Member Function Documentation

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

overrideprivatevirtual

Definition at line 29 of file JetSubstructurePacker.cc.

References algoLabels_, algoTags_, algoTokens_, DEFINE_FWK_MODULE, reco::deltaR(), distMax_, allElectronIsolations_cfi::dRMin, spr::find(), fixDaughters_, edm::Event::getByToken(), mps_fire::i, diffTreeTool::index, jetToken_, eostools::move(), PatBasicFWLiteJetAnalyzer_Selector_cfg::outputs, pc2pf_, pf2pc_, edm::Event::put(), 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 ( 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> & jdaus = outputs->back().daughterPtrVector();
         //std::cout << "Jet with pt " << outputs->back().pt() << ", " << outputs->back().numberOfDaughters() << " daughters, " << outputs->back().subjets().size() << ", subjets" << std::endl;
         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) {
                 reco::CandidatePtr rekeyed = edm::refToPtr((*pc2pf)[dau]);
                 if (std::find(jdaus.begin(), jdaus.end(), rekeyed) == jdaus.end()) {
                     skipSubjet = true;
                     break;
                 }
             }
             if (skipSubjet) continue;
 
             daughtersInSubjets.insert(daughtersInSubjets.end(), sjdaus.begin(), sjdaus.end());
             daughtersNew.push_back( reco::CandidatePtr(subjet) );
             //std::cout << "     found  " << subjet->numberOfDaughters() << " daughters in a subjet" << std::endl;
         }
         //if (!daughtersInSubjets.empty()) std::cout << "     subjet daughters are from collection " << daughtersInSubjets.front().id() << std::endl;
         //std::cout << "     in total,  " << daughtersInSubjets.size() << " daughters from subjets" << std::endl;
         for (const reco::CandidatePtr & dau : jdaus) {
             //if (!pf2pc->contains(dau.id())) {
             //    std::cout << "     daughter from collection " << dau.id() << " not in the value map!" << std::endl;
             //    std::cout << "     map expects collection " << pf2pc->ids().front().first << std::endl;
             //    continue;
             //}
             reco::CandidatePtr rekeyed = edm::refToPtr((*pf2pc)[dau]);
             if (std::find(daughtersInSubjets.begin(), daughtersInSubjets.end(), rekeyed) == daughtersInSubjets.end()) {
                 daughtersNew.push_back( rekeyed );
             }
         }
         //std::cout << "     in total,  " << daughtersNew.size() << " daughters including subjets" << std::endl;
         //if (daughtersNew.size() + daughtersInSubjets.size() - outputs->back().subjets().size() == outputs->back().numberOfDaughters()) {
         //    std::cout << "     it all adds up to the original number of daughters" << std::endl;
         //}
         outputs->back().clearDaughters();
         for (const auto & dau : daughtersNew) outputs->back().addDaughter(dau);
     }
   }
 
   iEvent.put(std::move(outputs));
 
 }