Changeset 2c8865f in git for modules/PdgCodeFilter.cc

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

File:

• modules/PdgCodeFilter.cc (modified) (3 diffs)

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;
+}
+