Inheritance diagram for CandMCMatchTableProducer:

Public Member Functions
	CandMCMatchTableProducer (edm::ParameterSet const &params)

void	produce (edm::StreamID id, edm::Event &iEvent, const edm::EventSetup &iSetup) const override

	~CandMCMatchTableProducer () override

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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () const final

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsStreamLuminosityBlocks () const final

bool	wantsStreamRuns () const final

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

ModuleDescription const &	moduleDescription () const

	~EDProducerBase () override

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 ()

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

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, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

	~ProducerBase () noexcept(false) override

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

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

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

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

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

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

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

static int	getParentHadronFlag (const reco::GenParticleRef match)

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Types
enum	MatchType { MMuon, MElectron, MTau, MPhoton, MOther }

Protected Attributes
const std::string	branchName_

const edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > >	candMap_

edm::EDGetTokenT< edm::Association< reco::GenParticleCollection > >	candMapVisTau_

const std::string	doc_

std::string	flavDoc_

const std::string	objName_

const edm::EDGetTokenT< reco::CandidateView >	src_

enum CandMCMatchTableProducer::MatchType	type_

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType

Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex >>

typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Protected Member Functions inherited from edm::ProducerBase
ProducesCollector	producesCollector ()

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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

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 14 of file CandMCMatchTableProducer.cc.

Member Enumeration Documentation

enum CandMCMatchTableProducer::MatchType

protected

Enumerator
MMuon
MElectron
MTau
MPhoton
MOther

Definition at line 195 of file CandMCMatchTableProducer.cc.

195 { MMuon, MElectron, MTau, MPhoton, MOther } type_;

CandMCMatchTableProducer::MPhoton

Definition: CandMCMatchTableProducer.cc:195

CandMCMatchTableProducer::type_

enum CandMCMatchTableProducer::MatchType type_

CandMCMatchTableProducer::MElectron

Definition: CandMCMatchTableProducer.cc:195

CandMCMatchTableProducer::MOther

Definition: CandMCMatchTableProducer.cc:195

CandMCMatchTableProducer::MMuon

Definition: CandMCMatchTableProducer.cc:195

CandMCMatchTableProducer::MTau

Definition: CandMCMatchTableProducer.cc:195

Constructor & Destructor Documentation

CandMCMatchTableProducer::CandMCMatchTableProducer ( edm::ParameterSet const & params )

inline

Definition at line 16 of file CandMCMatchTableProducer.cc.

References candMapVisTau_, Exception, flavDoc_, edm::ParameterSet::getParameter(), MElectron, MMuon, MOther, MPhoton, MTau, AlCaHLTBitMon_QueryRunRegistry::string, and type_.

       : objName_(params.getParameter<std::string>("objName")),
         branchName_(params.getParameter<std::string>("branchName")),
         doc_(params.getParameter<std::string>("docString")),
         src_(consumes<reco::CandidateView>(params.getParameter<edm::InputTag>("src"))),
         candMap_(consumes<edm::Association<reco::GenParticleCollection>>(params.getParameter<edm::InputTag>("mcMap"))) {
     produces<nanoaod::FlatTable>();
     const std::string& type = params.getParameter<std::string>("objType");
     if (type == "Muon")
       type_ = MMuon;
     else if (type == "Electron")
       type_ = MElectron;
     else if (type == "Tau")
       type_ = MTau;
     else if (type == "Photon")
       type_ = MPhoton;
     else if (type == "Other")
       type_ = MOther;
     else
       throw cms::Exception("Configuration", "Unsupported objType '" + type + "'\n");
 
     switch (type_) {
       case MMuon:
         flavDoc_ =
             "1 = prompt muon (including gamma*->mu mu), 15 = muon from prompt tau, "  // continues below
             "5 = muon from b, 4 = muon from c, 3 = muon from light or unknown, 0 = unmatched";
         break;
       case MElectron:
         flavDoc_ =
             "1 = prompt electron (including gamma*->mu mu), 15 = electron from prompt tau, 22 = prompt photon (likely "
             "conversion), "  // continues below
             "5 = electron from b, 4 = electron from c, 3 = electron from light or unknown, 0 = unmatched";
         break;
       case MPhoton:
         flavDoc_ = "1 = prompt photon, 13 = prompt electron, 0 = unknown or unmatched";
         break;
       case MTau:
         flavDoc_ =
             "1 = prompt electron, 2 = prompt muon, 3 = tau->e decay, 4 = tau->mu decay, 5 = hadronic tau decay, 0 = "
             "unknown or unmatched";
         break;
       case MOther:
         flavDoc_ = "1 = from hard scatter, 0 = unknown or unmatched";
         break;
     }
 
     if (type_ == MTau) {
       candMapVisTau_ =
           consumes<edm::Association<reco::GenParticleCollection>>(params.getParameter<edm::InputTag>("mcMapVisTau"));
     }
   }

CandMCMatchTableProducer::~CandMCMatchTableProducer ( )

inlineoverride

Definition at line 68 of file CandMCMatchTableProducer.cc.

68 {}

Member Function Documentation

static void CandMCMatchTableProducer::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

inlinestatic

Definition at line 171 of file CandMCMatchTableProducer.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), edm::ParameterSetDescription::addOptional(), and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                            {
     edm::ParameterSetDescription desc;
     desc.add<std::string>("objName")->setComment("name of the nanoaod::FlatTable to extend with this table");
     desc.add<std::string>("branchName")
         ->setComment(
             "name of the column to write (the final branch in the nanoaod will be <objName>_<branchName>Idx and "
             "<objName>_<branchName>Flav");
     desc.add<std::string>("docString")->setComment("documentation to forward to the output");
     desc.add<edm::InputTag>("src")->setComment(
         "physics object collection for the reconstructed objects (e.g. leptons)");
     desc.add<edm::InputTag>("mcMap")->setComment(
         "tag to an edm::Association<GenParticleCollection> mapping src to gen, such as the one produced by MCMatcher");
     desc.add<std::string>("objType")->setComment(
         "type of object to match (Muon, Electron, Tau, Photon, Other), taylors what's in t Flav branch");
     desc.addOptional<edm::InputTag>("mcMapVisTau")
         ->setComment("as mcMap, but pointing to the visible gen taus (only if objType == Tau)");
     descriptions.add("candMcMatchTable", desc);
   }

static int CandMCMatchTableProducer::getParentHadronFlag ( const reco::GenParticleRef match )

inlinestatic

Definition at line 148 of file CandMCMatchTableProducer.cc.

References funct::abs(), edm::Ref< C, T, F >::isAvailable(), edm::Ref< C, T, F >::isNonnull(), and edm::Ref< C, T, F >::key().

Referenced by produce().

                                                                  {
     bool has4 = false;
     for (unsigned int im = 0, nm = match->numberOfMothers(); im < nm; ++im) {
       reco::GenParticleRef mom = match->motherRef(im);
       assert(mom.isNonnull() && mom.isAvailable());  // sanity
       if (mom.key() >= match.key())
         continue;  // prevent circular refs
       int id = std::abs(mom->pdgId());
       if (id / 1000 == 5 || id / 100 == 5 || id == 5)
         return 5;
       if (id / 1000 == 4 || id / 100 == 4 || id == 4)
         has4 = true;
       if (mom->status() == 2) {
         id = getParentHadronFlag(mom);
         if (id == 5)
           return 5;
         else if (id == 4)
           has4 = true;
       }
     }
     return has4 ? 4 : 3;
   }

void CandMCMatchTableProducer::produce	(	edm::StreamID	id,
		edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)		const

inlineoverride

Definition at line 70 of file CandMCMatchTableProducer.cc.

References funct::abs(), branchName_, candMap_, candMapVisTau_, HLT_2018_cff::cands, doc_, flavDoc_, edm::Event::getByToken(), getParentHadronFlag(), mps_fire::i, nanoaod::FlatTable::IntColumn, edm::Ref< C, T, F >::isNonnull(), crabWrapper::key, edm::Ref< C, T, F >::key(), genParticles_cff::map, match(), MElectron, MMuon, eostools::move(), MPhoton, MTau, objName_, edm::View< T >::ptrAt(), edm::Event::put(), edm::View< T >::size(), src_, type_, and nanoaod::FlatTable::UInt8Column.

                                                                                              {
     edm::Handle<reco::CandidateView> cands;
     iEvent.getByToken(src_, cands);
     unsigned int ncand = cands->size();
 
     auto tab = std::make_unique<nanoaod::FlatTable>(ncand, objName_, false, true);
 
     edm::Handle<edm::Association<reco::GenParticleCollection>> map;
     iEvent.getByToken(candMap_, map);
 
     edm::Handle<edm::Association<reco::GenParticleCollection>> mapVisTau;
     if (type_ == MTau) {
       iEvent.getByToken(candMapVisTau_, mapVisTau);
     }
 
     std::vector<int> key(ncand, -1), flav(ncand, 0);
     for (unsigned int i = 0; i < ncand; ++i) {
       //std::cout << "cand #" << i << ": pT = " << cands->ptrAt(i)->pt() << ", eta = " << cands->ptrAt(i)->eta() << ", phi = " << cands->ptrAt(i)->phi() << std::endl;
       reco::GenParticleRef match = (*map)[cands->ptrAt(i)];
       reco::GenParticleRef matchVisTau;
       if (type_ == MTau) {
         matchVisTau = (*mapVisTau)[cands->ptrAt(i)];
       }
       if (match.isNonnull())
         key[i] = match.key();
       else if (matchVisTau.isNonnull())
         key[i] = matchVisTau.key();
       else
         continue;
       switch (type_) {
         case MMuon:
           if (match->isPromptFinalState())
             flav[i] = 1;  // prompt
           else if (match->isDirectPromptTauDecayProductFinalState())
             flav[i] = 15;  // tau
           else
             flav[i] = getParentHadronFlag(match);  // 3 = light, 4 = charm, 5 = b
           break;
         case MElectron:
           if (match->isPromptFinalState())
             flav[i] = (match->pdgId() == 22 ? 22 : 1);  // prompt electron or photon
           else if (match->isDirectPromptTauDecayProductFinalState())
             flav[i] = 15;  // tau
           else
             flav[i] = getParentHadronFlag(match);  // 3 = light, 4 = charm, 5 = b
           break;
         case MPhoton:
           if (match->isPromptFinalState())
             flav[i] = (match->pdgId() == 22 ? 1 : 13);  // prompt electron or photon
           break;
         case MTau:
           // CV: assignment of status codes according to https://twiki.cern.ch/twiki/bin/viewauth/CMS/HiggsToTauTauWorking2016#MC_Matching
           if (match.isNonnull() && match->statusFlags().isPrompt() && abs(match->pdgId()) == 11)
             flav[i] = 1;
           else if (match.isNonnull() && match->statusFlags().isPrompt() && abs(match->pdgId()) == 13)
             flav[i] = 2;
           else if (match.isNonnull() && match->isDirectPromptTauDecayProductFinalState() && abs(match->pdgId()) == 11)
             flav[i] = 3;
           else if (match.isNonnull() && match->isDirectPromptTauDecayProductFinalState() && abs(match->pdgId()) == 13)
             flav[i] = 4;
           else if (matchVisTau.isNonnull())
             flav[i] = 5;
           break;
         default:
           flav[i] = match->statusFlags().fromHardProcess();
       };
     }
 
     tab->addColumn<int>(
         branchName_ + "Idx", key, "Index into genParticle list for " + doc_, nanoaod::FlatTable::IntColumn);
     tab->addColumn<uint8_t>(branchName_ + "Flav",
                             flav,
                             "Flavour of genParticle for " + doc_ + ": " + flavDoc_,
                             nanoaod::FlatTable::UInt8Column);
 
     iEvent.put(std::move(tab));
   }