Inheritance diagram for RecoTauJetRegionProducer:

Public Types
typedef edm::AssociationMap< edm::OneToMany< std::vector< reco::PFJet >, std::vector< reco::PFCandidate >, unsigned int > >	JetToPFCandidateAssociation

typedef edm::Association< reco::PFJetCollection >	PFJetMatchMap

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 Member Functions
void	produce (edm::Event &evt, const edm::EventSetup &es) override

	RecoTauJetRegionProducer (const edm::ParameterSet &pset)

	~RecoTauJetRegionProducer () override

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

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Private Attributes
double	deltaR2_

edm::InputTag	inputJets_

edm::EDGetTokenT< reco::CandidateView >	Jets_token

double	maxJetAbsEta_

double	minJetPt_

std::string	moduleLabel_

edm::EDGetTokenT< reco::PFCandidateCollection >	pf_token

edm::EDGetTokenT< JetToPFCandidateAssociation >	pfCandAssocMap_token

edm::InputTag	pfCandAssocMapSrc_

edm::InputTag	pfCandSrc_

int	verbosity_

Detailed Description

Definition at line 32 of file RecoTauJetRegionProducer.cc.

Member Typedef Documentation

typedef edm::AssociationMap<edm::OneToMany<std::vector<reco::PFJet>, std::vector<reco::PFCandidate>, unsigned int> > RecoTauJetRegionProducer::JetToPFCandidateAssociation

Definition at line 36 of file RecoTauJetRegionProducer.cc.

typedef edm::Association<reco::PFJetCollection> RecoTauJetRegionProducer::PFJetMatchMap

Definition at line 35 of file RecoTauJetRegionProducer.cc.

Constructor & Destructor Documentation

RecoTauJetRegionProducer::RecoTauJetRegionProducer ( const edm::ParameterSet & pset )

explicit

Definition at line 60 of file RecoTauJetRegionProducer.cc.

References deltaR(), deltaR2_, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), inputJets_, Jets_token, maxJetAbsEta_, minJetPt_, pf_token, pfCandAssocMap_token, pfCandAssocMapSrc_, pfCandSrc_, and verbosity_.

   : moduleLabel_(cfg.getParameter<std::string>("@module_label"))
 {
   inputJets_ = cfg.getParameter<edm::InputTag>("src");
   pfCandSrc_ = cfg.getParameter<edm::InputTag>("pfCandSrc");
   pfCandAssocMapSrc_ = cfg.getParameter<edm::InputTag>("pfCandAssocMapSrc");
 
   pf_token = consumes<reco::PFCandidateCollection>(pfCandSrc_); 
   Jets_token = consumes<reco::CandidateView>(inputJets_);
   pfCandAssocMap_token =  consumes<JetToPFCandidateAssociation>(pfCandAssocMapSrc_);
   
   double deltaR = cfg.getParameter<double>("deltaR"); 
   deltaR2_ = deltaR*deltaR;
   minJetPt_ = ( cfg.exists("minJetPt") ) ? cfg.getParameter<double>("minJetPt") : -1.0;
   maxJetAbsEta_ = ( cfg.exists("maxJetAbsEta") ) ? cfg.getParameter<double>("maxJetAbsEta") : 99.0;
   
   verbosity_ = ( cfg.exists("verbosity") ) ?
     cfg.getParameter<int>("verbosity") : 0;
   
   produces<reco::PFJetCollection>("jets");
   produces<PFJetMatchMap>();
 }

RecoTauJetRegionProducer::~RecoTauJetRegionProducer ( )

inlineoverride

Definition at line 38 of file RecoTauJetRegionProducer.cc.

References produce().

38 {}

Member Function Documentation

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

override

Definition at line 83 of file RecoTauJetRegionProducer.cc.

References funct::abs(), reco::CompositePtrCandidate::addDaughter(), reco::CompositePtrCandidate::clearDaughters(), gather_cfg::cout, DEFINE_FWK_MODULE, reco::deltaR2(), deltaR2_, MillePedeFileConverter_cfg::e, edm::AssociationMap< Tag >::end(), edm::helper::Filler< Map >::fill(), objects.autophobj::filler, edm::AssociationMap< Tag >::find(), edm::Event::get(), edm::Event::getByToken(), reco::PFJet::getPFConstituents(), edm::RefVector< C, T, F >::id(), training_settings::idx, inputJets_, edm::helper::Filler< Map >::insert(), edm::HandleBase::isValid(), fwrapper::jets, Jets_token, edm::Ref< C, T, F >::key(), edm::InputTag::label(), maxJetAbsEta_, minJetPt_, moduleLabel_, eostools::move(), pf_token, pfCandAssocMap_token, pfCandAssocMapSrc_, pfCandSrc_, edm::Event::put(), edm::RefVector< C, T, F >::size(), groupFilesInBlocks::temp, and verbosity_.

Referenced by ~RecoTauJetRegionProducer().

 {
   if ( verbosity_ ) {
     std::cout << "<RecoTauJetRegionProducer::produce (moduleLabel = " << moduleLabel_ << ")>:" << std::endl;
     std::cout << " inputJets = " << inputJets_ << std::endl;
     std::cout << " pfCandSrc = " << pfCandSrc_ << std::endl;
     std::cout << " pfCandAssocMapSrc_ = " << pfCandAssocMapSrc_ << std::endl;
   }
 
   edm::Handle<reco::PFCandidateCollection> pfCandsHandle;
   evt.getByToken(pf_token, pfCandsHandle);
 
   // Build Ptrs for all the PFCandidates
   typedef edm::Ptr<reco::PFCandidate> PFCandPtr;
   std::vector<PFCandPtr> pfCands;
   pfCands.reserve(pfCandsHandle->size());
   for ( size_t icand = 0; icand < pfCandsHandle->size(); ++icand ) {
     pfCands.push_back(PFCandPtr(pfCandsHandle, icand));
   }
 
   // Get the jets
   edm::Handle<reco::CandidateView> jetView;
   evt.getByToken(Jets_token, jetView);
   // Convert to a vector of PFJetRefs
   reco::PFJetRefVector jets = reco::tau::castView<reco::PFJetRefVector>(jetView);
   size_t nJets = jets.size();
 
   // Get the association map matching jets to PFCandidates
   // (needed for recinstruction of boosted taus)
   edm::Handle<JetToPFCandidateAssociation> jetToPFCandMap;
   std::vector<std::unordered_set<unsigned> > fastJetToPFCandMap;
   if ( pfCandAssocMapSrc_.label() != "" ) {
     evt.getByToken(pfCandAssocMap_token, jetToPFCandMap);
     fastJetToPFCandMap.resize(nJets);
     for ( size_t ijet = 0; ijet < nJets; ++ijet ) {
       // Get a ref to jet
       const reco::PFJetRef& jetRef = jets[ijet];
       const auto& pfCandsMappedToJet = (*jetToPFCandMap)[jetRef];
       for ( const auto& pfCandMappedToJet : pfCandsMappedToJet ) {
     fastJetToPFCandMap[ijet].emplace(pfCandMappedToJet.key());
       }
     }
   }
 
   // Get the original product, so we can match against it - otherwise the
   // indices don't match up.
   edm::ProductID originalId = jets.id();
   edm::Handle<reco::PFJetCollection> originalJets;
   size_t nOriginalJets = 0;
   // We have to make sure that we have some selected jets, otherwise we don't
   // actually have a valid product ID to the original jets.
   if ( nJets ) {
     try {
       evt.get(originalId, originalJets);
     } catch(const cms::Exception &e) {
       edm::LogError("MissingOriginalCollection")
         << "Can't get the original jets that made: " << inputJets_
         << " that have product ID: " << originalId
         << " from the event!!";
       throw e;
     }
     nOriginalJets = originalJets->size();
   }
 
   auto newJets = std::make_unique<reco::PFJetCollection>();
 
   // Keep track of the indices of the current jet and the old (original) jet
   // -1 indicates no match.
   std::vector<int> matchInfo(nOriginalJets, -1);
   newJets->reserve(nJets);
   size_t nNewJets = 0;
   for ( size_t ijet = 0; ijet < nJets; ++ijet ) {
     // Get a ref to jet
     const reco::PFJetRef& jetRef = jets[ijet];
     if(jetRef->pt() - minJetPt_ < 1e-5) continue;
     if(std::abs(jetRef->eta()) - maxJetAbsEta_ > -1e-5) continue;
     // Make an initial copy.
     newJets->emplace_back(*jetRef);
     reco::PFJet& newJet = newJets->back();
     // Clear out all the constituents
     newJet.clearDaughters();
     // Loop over all the PFCands
     for ( const auto& pfCand : pfCands ) {
       bool isMappedToJet = false;
       if ( jetToPFCandMap.isValid() ) {
     auto temp = jetToPFCandMap->find(jetRef);
     if( temp == jetToPFCandMap->end() ) {
       edm::LogWarning("WeirdCandidateMap") << "Candidate map for jet " << jetRef.key() << " is empty!";
       continue;
     }
     isMappedToJet = fastJetToPFCandMap[ijet].count(pfCand.key());
       } else {
     isMappedToJet = true;
       }
       if ( reco::deltaR2(*jetRef, *pfCand) < deltaR2_ && isMappedToJet ) newJet.addDaughter(pfCand);
     }
     if ( verbosity_ ) {
       std::cout << "jet #" << ijet << ": Pt = " << jetRef->pt() << ", eta = " << jetRef->eta() << ", phi = " << jetRef->eta() << ","
         << " mass = " << jetRef->mass() << ", area = " << jetRef->jetArea() << std::endl;
       std::vector<reco::PFCandidatePtr> jetConstituents = newJet.getPFConstituents();
       int idx = 0;
       for ( std::vector<reco::PFCandidatePtr>::const_iterator jetConstituent = jetConstituents.begin();
         jetConstituent != jetConstituents.end(); ++jetConstituent ) {
     std::cout << " constituent #" << idx << ": Pt = " << (*jetConstituent)->pt() << ", eta = " << (*jetConstituent)->eta() << ", phi = " << (*jetConstituent)->phi() << std::endl;
       }
     }
     // Match the index of the jet we just made to the index into the original
     // collection.
     //matchInfo[jetRef.key()] = ijet;
     matchInfo[jetRef.key()] = nNewJets;
     nNewJets++;
   }
 
   // Put our new jets into the event
   edm::OrphanHandle<reco::PFJetCollection> newJetsInEvent = evt.put(std::move(newJets), "jets");
 
   // Create a matching between original jets -> extra collection
   auto matching = std::make_unique<PFJetMatchMap>(newJetsInEvent);
   if ( nJets ) {
     PFJetMatchMap::Filler filler(*matching);
     filler.insert(originalJets, matchInfo.begin(), matchInfo.end());
     filler.fill();
   }
   evt.put(std::move(matching));
 }