CandMCMatchTableProducer Class Reference

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
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)
virtual	~ProducerBase () noexcept(false)
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
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)

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

Member Enumeration Documentation

enum CandMCMatchTableProducer::MatchType

protected

Enumerator
MMuon
MElectron
MTau
MPhoton
MOther

Definition at line 145 of file CandMCMatchTableProducer.cc.

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

Constructor & Destructor Documentation

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

inline

Definition at line 17 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 48 of file CandMCMatchTableProducer.cc.

{}

Member Function Documentation

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

inlinestatic

Definition at line 128 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 110 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 50 of file CandMCMatchTableProducer.cc.

References funct::abs(), branchName_, candMap_, candMapVisTau_, 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->isPromptFinalState()                      && abs(match->pdgId()) == 11 ) flav[i] = 1;
                        else if ( match.isNonnull() && match->isPromptFinalState()                      && 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));
        }

Member Data Documentation

const std::string CandMCMatchTableProducer::branchName_

protected

Definition at line 141 of file CandMCMatchTableProducer.cc.

Referenced by produce().

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

protected

Definition at line 143 of file CandMCMatchTableProducer.cc.

Referenced by produce().

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

protected

Definition at line 144 of file CandMCMatchTableProducer.cc.

Referenced by CandMCMatchTableProducer(), and produce().

const std::string CandMCMatchTableProducer::doc_

protected

Definition at line 141 of file CandMCMatchTableProducer.cc.

Referenced by produce().

std::string CandMCMatchTableProducer::flavDoc_

protected

Definition at line 146 of file CandMCMatchTableProducer.cc.

Referenced by CandMCMatchTableProducer(), and produce().

const std::string CandMCMatchTableProducer::objName_

protected

Definition at line 141 of file CandMCMatchTableProducer.cc.

Referenced by produce().

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

protected

Definition at line 142 of file CandMCMatchTableProducer.cc.

Referenced by produce().

enum CandMCMatchTableProducer::MatchType CandMCMatchTableProducer::type_

protected

Referenced by Modules.Service::__str__(), Modules.Service::_placeImpl(), Modules.ESSource::_placeImpl(), Modules.ESProducer::_placeImpl(), CandMCMatchTableProducer(), Modules.ESPrefer::copy(), Mixins._TypedParameterizable::dumpPython(), Modules.ESPrefer::dumpPythonAs(), Modules.Service::dumpSequencePython(), RandomRunSource.RandomRunSource::insertInto(), Modules.Service::insertInto(), Mixins._TypedParameterizable::insertInto(), Modules.ESSource::moduleLabel_(), Modules.ESProducer::moduleLabel_(), Modules.ESSource::nameInProcessDesc_(), Modules.ESProducer::nameInProcessDesc_(), Modules.ESPrefer::nameInProcessDesc_(), and produce().