Changeset fa42514 in git for modules

Timestamp:

May 22, 2015, 6:00:15 PM (9 years ago)

Author:

Chase Shimmin <cshimmin@…>

Branches:

ImprovedOutputFile, Timing, dual_readout, llp, master

Children:

f3c4047

Parents:

e8070b6

git-author:

Chase Shimmin <cshimmin@…> (05/22/15 17:50:49)

git-committer:

Chase Shimmin <cshimmin@…> (05/22/15 18:00:15)

Message:

Add Invert and Status options to PdgCodeFilter

It is sometimes useful to output the truth record of a single
(non-stable) particle species, for example when considering an
exotic heavy resonance. Presently the only way to do so is to
output the entire Delphes/allParticles array, which is inefficient.

This patch adds the option InvertPdg to PdgCodeFilter, which causes
the PdgCodes specified to be *added* to the output array, rather
than removed. It also adds the options RequireStatus and Status,
for additional filtering based on the MC record status. If
RequireStatus is true, than only particles matching Status are
added to the output array.

This patch should be backwards compatible with existing usages of
PdgCodeFilter.

Location:

modules

Files:

• PdgCodeFilter.cc (modified) (3 diffs)
• PdgCodeFilter.h (modified) (1 diff)

modules/PdgCodeFilter.cc

@@ -74 +74 @@
   fPTMin = GetDouble("PTMin", 0.0);
 
+  fInvertPdg = GetBool("InvertPdg", false);
+
+  fRequireStatus = GetBool("RequireStatus", false);
+  fStatus = GetInt("Status", 1);
+
   // import input array
   fInputArray = ImportArray(GetString("InputArray", "Delphes/allParticles"));
@@ -109 +114 @@
   Double_t pt;
 
+  const Bool_t requireStatus = fRequireStatus;
+  const Bool_t invertPdg = fInvertPdg;
+  const int reqStatus = fStatus;
+
   fItInputArray->Reset();
   while((candidate = static_cast<Candidate*>(fItInputArray->Next())))
@@ -116 +125 @@
     pt = candidateMomentum.Pt();
 
+    if(pt < fPTMin) continue;
+    if(requireStatus && (candidate->Status != reqStatus)) continue;
+
     pass = kTRUE;
-
-    if(pt < fPTMin) pass = kFALSE;
     if(find(fPdgCodes.begin(), fPdgCodes.end(), pdgCode) != fPdgCodes.end()) pass = kFALSE;
 
+    if (invertPdg) pass = !pass;
     if(pass) fOutputArray->Add(candidate);
   }

modules/PdgCodeFilter.h

@@ -50 +50 @@
 
   Double_t fPTMin; //!
+  Bool_t fInvertPdg; //!
+  Bool_t fRequireStatus; //!
+  Int_t fStatus; //!
 
   std::vector<Int_t> fPdgCodes;