Inheritance diagram for BTVMCFlavourTableProducer:

Public Member Functions
	BTVMCFlavourTableProducer (const edm::ParameterSet &iConfig)

int	jet_flavour (const pat::Jet &jet, const std::vector< reco::GenParticle > &gToBB, const std::vector< reco::GenParticle > &gToCC, const std::vector< reco::GenParticle > &neutrinosLepB, const std::vector< reco::GenParticle > &neutrinosLepB_C, const std::vector< reco::GenParticle > &alltaus, bool usePhysForLightAndUndefined)

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

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

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

Private Attributes
edm::EDGetTokenT< reco::GenParticleCollection >	genParticlesToken_

std::string	name_

edm::EDGetTokenT< std::vector< pat::Jet > >	src_

Static Private Attributes
static constexpr double	jetR_ = 0.4

static constexpr bool	usePhysForLightAndUndefined = false

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 23 of file BTVMCFlavourTableProducer.cc.

Constructor & Destructor Documentation

◆ BTVMCFlavourTableProducer()

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

inlineexplicit

Definition at line 25 of file BTVMCFlavourTableProducer.cc.

       : name_(iConfig.getParameter<std::string>("name")),
         src_(consumes<std::vector<pat::Jet> >(iConfig.getParameter<edm::InputTag>("src"))),
         genParticlesToken_(consumes<reco::GenParticleCollection>(iConfig.getParameter<edm::InputTag>("genparticles"))) {
     produces<nanoaod::FlatTable>();
   }

◆ ~BTVMCFlavourTableProducer()

BTVMCFlavourTableProducer::~BTVMCFlavourTableProducer ( )

inlineoverride

Definition at line 32 of file BTVMCFlavourTableProducer.cc.

32 {};

Member Function Documentation

◆ fillDescriptions()

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

inlinestatic

Definition at line 167 of file BTVMCFlavourTableProducer.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                            {
     edm::ParameterSetDescription desc;
     desc.add<edm::InputTag>("src")->setComment("input Jet collection");
     desc.add<edm::InputTag>("genparticles")->setComment("input genparticles info collection");
     desc.add<std::string>("name")->setComment("name of the genJet FlatTable we are extending with flavour information");
     descriptions.add("btvMCTable", desc);
   }

◆ jet_flavour()

int BTVMCFlavourTableProducer::jet_flavour	(	const pat::Jet &	jet,
		const std::vector< reco::GenParticle > &	gToBB,
		const std::vector< reco::GenParticle > &	gToCC,
		const std::vector< reco::GenParticle > &	neutrinosLepB,
		const std::vector< reco::GenParticle > &	neutrinosLepB_C,
		const std::vector< reco::GenParticle > &	alltaus,
		bool	usePhysForLightAndUndefined
	)

inline

Definition at line 33 of file BTVMCFlavourTableProducer.cc.

References funct::abs(), reco::deltaR2(), mps_fire::i, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, metsig::jet, jetR_, AlCaHLTBitMon_ParallelJobs::p, reco::Candidate::pdgId(), and usePhysForLightAndUndefined.

Referenced by produce().

                                                     {
     int hflav = abs(jet.hadronFlavour());
     int pflav = abs(jet.partonFlavour());
     int physflav = 0;
     if (!(jet.genJet())) {
       if (pflav == 0)
         return 999;
       else
         return 1000;
     }
     if (jet.genParton())
       physflav = abs(jet.genParton()->pdgId());
     std::size_t nbs = jet.jetFlavourInfo().getbHadrons().size();
     std::size_t ncs = jet.jetFlavourInfo().getcHadrons().size();
 
     unsigned int nbFromGSP(0);
     for (const reco::GenParticle& p : gToBB) {
       double dr2(reco::deltaR2(jet, p));
       if (dr2 < jetR_ * jetR_)
         ++nbFromGSP;
     }
 
     unsigned int ncFromGSP(0);
     for (const reco::GenParticle& p : gToCC) {
       double dr2(reco::deltaR2(jet, p));
       if (dr2 < jetR_ * jetR_)
         ++ncFromGSP;
     }
 
     //std::cout << " jet pt = " << jet.pt() << " hfl = " << hflav << " pfl = " << pflav << " genpart = " << physflav
     //  << " nbFromGSP = " << nbFromGSP << " ncFromGSP = " << ncFromGSP
     //  << " nBhadrons " << nbs << " nCHadrons " << ncs << std::endl;
     if (hflav == 5) {  //B jet
       if (nbs > 1) {
         if (nbFromGSP > 0)
           return 511;
         else
           return 510;
       } else if (nbs == 1) {
         for (std::vector<reco::GenParticle>::const_iterator it = neutrinosLepB.begin(); it != neutrinosLepB.end();
              ++it) {
           if (reco::deltaR2(it->eta(), it->phi(), jet.eta(), jet.phi()) < 0.4 * 0.4) {
             return 520;
           }
         }
         for (std::vector<reco::GenParticle>::const_iterator it = neutrinosLepB_C.begin(); it != neutrinosLepB_C.end();
              ++it) {
           if (reco::deltaR2(it->eta(), it->phi(), jet.eta(), jet.phi()) < 0.4 * 0.4) {
             return 521;
           }
         }
         return 500;
       } else {
         if (usePhysForLightAndUndefined) {
           if (physflav == 21)
             return 0;
           else if (physflav == 3)
             return 2;
           else if (physflav == 2 || physflav == 1)
             return 1;
           else
             return 1000;
         } else
           return 1000;
       }
     } else if (hflav == 4) {  //C jet
       if (ncs > 1) {
         if (ncFromGSP > 0)
           return 411;
         else
           return 410;
       } else
         return 400;
     } else {                   //not a heavy jet
       if (!alltaus.empty()) {  //check for tau in a simplistic way
         bool ishadrtaucontained = true;
         for (const auto& p : alltaus) {
           size_t ndau = p.numberOfDaughters();
           for (size_t i = 0; i < ndau; i++) {
             const reco::Candidate* dau = p.daughter(i);
             int daupid = std::abs(dau->pdgId());
             if (daupid == 13 || daupid == 11) {
               ishadrtaucontained = false;
               break;
             }
             if (daupid != 12 && daupid != 14 && daupid != 16 && reco::deltaR2(*dau, jet) > jetR_ * jetR_) {
               ishadrtaucontained = false;
               break;
             }
           }
         }
         if (ishadrtaucontained)
           return 600;
       }
       if (std::abs(pflav) == 4 || std::abs(pflav) == 5 || nbs || ncs) {
         if (usePhysForLightAndUndefined) {
           if (physflav == 21)
             return 0;
           else if (physflav == 3)
             return 2;
           else if (physflav == 2 || physflav == 1)
             return 1;
           else
             return 1000;
         } else
           return 1000;
       } else if (usePhysForLightAndUndefined) {
         if (physflav == 21)
           return 0;
         else if (physflav == 3)
           return 2;
         else if (physflav == 2 || physflav == 1)
           return 1;
         else
           return 1000;
       } else {
         if (pflav == 21)
           return 0;
         else if (pflav == 3)
           return 2;
         else if (pflav == 2 || pflav == 1)
           return 1;
         else
           return 1000;
       }
     }
   }

◆ produce()

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

overrideprivate

Definition at line 188 of file BTVMCFlavourTableProducer.cc.

References funct::abs(), gather_cfg::cout, d0, d1, reco::deltaR2(), genParticlesToken_, l1ctLayer2EG_cff::id, iEvent, metsig::jet, jet_flavour(), jetR_, PDWG_EXODelayedJetMET_cff::jets, eostools::move(), name_, nJets, reco::LeafCandidate::pdgId(), src_, mps_update::status, and usePhysForLightAndUndefined.

                                                                                      {
   auto jets = iEvent.getHandle(src_);
   // const auto& jetFlavourInfosProd = iEvent.get(genParticlesToken_);
   edm::Handle<reco::GenParticleCollection> genParticlesHandle;
   iEvent.getByToken(genParticlesToken_, genParticlesHandle);
   std::vector<reco::GenParticle> neutrinosLepB;
   std::vector<reco::GenParticle> neutrinosLepB_C;
 
   std::vector<reco::GenParticle> gToBB;
   std::vector<reco::GenParticle> gToCC;
   std::vector<reco::GenParticle> alltaus;
 
   unsigned int nJets = jets->size();
 
   std::vector<unsigned> jet_FlavSplit(nJets);
   for (const reco::Candidate& genC : *genParticlesHandle) {
     const reco::GenParticle& gen = static_cast<const reco::GenParticle&>(genC);
     if (abs(gen.pdgId()) == 12 || abs(gen.pdgId()) == 14 || abs(gen.pdgId()) == 16) {
       const reco::GenParticle* mother = static_cast<const reco::GenParticle*>(gen.mother());
       if (mother != nullptr) {
         if ((abs(mother->pdgId()) > 500 && abs(mother->pdgId()) < 600) ||
             (abs(mother->pdgId()) > 5000 && abs(mother->pdgId()) < 6000)) {
           neutrinosLepB.emplace_back(gen);
         }
         if ((abs(mother->pdgId()) > 400 && abs(mother->pdgId()) < 500) ||
             (abs(mother->pdgId()) > 4000 && abs(mother->pdgId()) < 5000)) {
           neutrinosLepB_C.emplace_back(gen);
         }
       } else {
         std::cout << "No mother" << std::endl;
       }
     }
 
     int id(std::abs(gen.pdgId()));
     int status(gen.status());
 
     if (id == 21 && status >= 21 && status <= 59) {  
       if (gen.numberOfDaughters() == 2) {
         const reco::Candidate* d0 = gen.daughter(0);
         const reco::Candidate* d1 = gen.daughter(1);
         if (std::abs(d0->pdgId()) == 5 && std::abs(d1->pdgId()) == 5 && d0->pdgId() * d1->pdgId() < 0 &&
             reco::deltaR2(*d0, *d1) < jetR_ * jetR_)
           gToBB.push_back(gen);
         if (std::abs(d0->pdgId()) == 4 && std::abs(d1->pdgId()) == 4 && d0->pdgId() * d1->pdgId() < 0 &&
             reco::deltaR2(*d0, *d1) < jetR_ * jetR_)
           gToCC.push_back(gen);
       }
     }
 
     if (id == 15 && false) {
       alltaus.push_back(gen);
     }
   }
   for (unsigned i_jet = 0; i_jet < nJets; ++i_jet) {
     // from DeepNTuples
     const auto& jet = jets->at(i_jet);
 
     jet_FlavSplit[i_jet] =
         jet_flavour(jet, gToBB, gToCC, neutrinosLepB, neutrinosLepB_C, alltaus, usePhysForLightAndUndefined);
   }
   auto newtab = std::make_unique<nanoaod::FlatTable>(nJets, name_, false, true);
   newtab->addColumn<int>("FlavSplit",
                          jet_FlavSplit,
                          "Flavour of the jet, numerical codes: "
                          "isG: 0, "
                          "isUD: 1, "
                          "isS: 2, "
                          "isC: 400, "
                          "isCC: 410, "
                          "isGCC: 411, "
                          "isB: 500, "
                          "isBB: 510, "
                          "isGBB: 511, "
                          "isLeptonicB: 520, "
                          "isLeptonicB_C: 521, "
                          "isTAU: 600, "
                          "isPU: 999,"
                          "isUndefined: 1000. "
                          "May be combined to form coarse labels for tagger training and flavour dependent attacks "
                          "using the loss surface.");
   iEvent.put(std::move(newtab));
 }