Changeset 2c8865f in git

Timestamp:

Feb 24, 2022, 6:01:00 AM (3 years ago)

Author:

youngkwon jo <cccpy98@…>

Children:

46b3e01

Parents:

dd263e4

Message:

add code for keeping the BHadrons

Files:

• cards/delphes_card_CMS.tcl (modified) (2 diffs)
• modules/PdgCodeFilter.cc (modified) (3 diffs)
• modules/PdgCodeFilter.h (modified) (1 diff)

cards/delphes_card_CMS.tcl

@@ -585 +585 @@
 module PdgCodeFilter NeutrinoFilter {
 
-  set InputArray Delphes/stableParticles
+  #set InputArray Delphes/stableParticles
+  set InputArray Delphes/allParticles
   set OutputArray filteredParticles
 
   set PTMin 0.0
+  set RequireStatus true
+  set RequireKeepGhostBHadron true
 
   add PdgCode {12}
@@ -783 +786 @@
 # add Branch InputArray BranchName BranchClass
   add Branch Delphes/allParticles Particle GenParticle
+#  add Branch NeutrinoFilter/filteredParticles Particle2 GenParticle
 
   add Branch TrackMerger/tracks Track Track

modules/PdgCodeFilter.cc

@@ -85 +85 @@
   fCharge = GetInt("Charge", 1);
 
+  // keep bhadron
+  fRequireKeepGhostBHadron = GetBool("RequireKeepGhostBHadron", false);
+
   // import input array
   fInputArray = ImportArray(GetString("InputArray", "Delphes/allParticles"));
@@ -124 +127 @@
   {
     pdgCode = candidate->PID;
+    //std::cout << "--" << pdgCode << std::endl;
+
+    if (fRequireKeepGhostBHadron) {
+      if (isBHadron(abs(pdgCode)) ){
+        candidate->PT = candidate->PT * 1e-18;
+        if (candidate->PT ==0) candidate->PT = 1e-18;
+        candidate->Momentum.SetPtEtaPhiM(candidate->PT, candidate->Momentum.Eta(), candidate->Phi, candidate->Momentum.M());
+        fOutputArray->Add(candidate);
+        continue;
+      }
+    } 
+
     const TLorentzVector &candidateMomentum = candidate->Momentum;
     pt = candidateMomentum.Pt();
@@ -139 +154 @@
   }
 }
+
+Bool_t PdgCodeFilter::isBHadron(const unsigned int absPdgId) {
+  if (absPdgId <= 100)
+    return false;  // Fundamental particles and MC internals
+  if (absPdgId >= 1000000000)
+    return false;  // Nuclei, +-10LZZZAAAI
+
+  // General form of PDG ID is 7 digit form
+  // +- n nr nL nq1 nq2 nq3 nJ
+  //const int nJ = absPdgId % 10; // Spin
+  const int nq3 = (absPdgId / 10) % 10;
+  const int nq2 = (absPdgId / 100) % 10;
+  const int nq1 = (absPdgId / 1000) % 10;
+
+  if (nq3 == 0)
+    return false;  // Diquarks
+  if (nq1 == 0 and nq2 == 5)
+    return true;  // B mesons
+  if (nq1 == 5)
+    return true;  // B baryons
+
+  return false;
+}
+

modules/PdgCodeFilter.h

@@ -55 +55 @@
   Bool_t fRequireNotPileup; //!
 
+  Bool_t fRequireKeepGhostBHadron; //!
+
   std::vector<Int_t> fPdgCodes;
+
+  Bool_t isBHadron(const unsigned int absPdgId); //!
 
   TIterator *fItInputArray; //!