Changeset 11 for trunk/modules/MadGraphPartonSelector.cc

Timestamp:

Sep 11, 2008, 1:41:12 PM (16 years ago)

Author:

Pavel Demin

Message:

add SISCone jet algorithm and update names for ConeJetFinder modules

File:

• trunk/modules/MadGraphPartonSelector.cc (modified) (7 diffs)

trunk/modules/MadGraphPartonSelector.cc

@@ -33 +33 @@
 
   void InsertParticleID(Int_t pid);
-  void InsertAncestorID(Int_t pid);
+  void InsertExclAncestorID(Int_t pid);
+  void InsertInclAncestorID(Int_t pid);
 
 private:
 
   Bool_t hasBadAncestor(ExRootGenParticle *object);
+  Bool_t hasGoodAncestor(ExRootGenParticle *object);
 
   TClonesArray *fBranchParticle;
 
   set< Int_t > fParticleIDSet;
-  set< Int_t > fAncestorIDSet;
+  set< Int_t > fExclAncestorIDSet;
+  set< Int_t > fInclAncestorIDSet;
 };
 
@@ -61 +64 @@
 //------------------------------------------------------------------------------
 
-void MadGraphPartonClassifier::InsertAncestorID(Int_t pid)
-{
-  fAncestorIDSet.insert(pid);
+void MadGraphPartonClassifier::InsertExclAncestorID(Int_t pid)
+{
+  fExclAncestorIDSet.insert(pid);
+}
+
+//------------------------------------------------------------------------------
+
+void MadGraphPartonClassifier::InsertInclAncestorID(Int_t pid)
+{
+  fInclAncestorIDSet.insert(pid);
 }
 
@@ -76 +86 @@
   set< Int_t >::const_iterator itParticleIDSet;
 
+  // keep all particles if there is no pid in the list
+  if(fExclAncestorIDSet.empty())
+  {
+    return kFALSE;
+  }
+
   for(i = 0; i < kMaxAncestors; ++i)
   {
@@ -90 +106 @@
 
     // skip particles with pid included in list
-    itAncestorIDSet = fAncestorIDSet.find(pidAbs);
-
-    if(itAncestorIDSet != fAncestorIDSet.end()) return kTRUE;
+    itAncestorIDSet = fExclAncestorIDSet.find(pidAbs);
+
+    if(itAncestorIDSet != fExclAncestorIDSet.end()) return kTRUE;
+  }
+
+  return kFALSE;
+}
+
+//------------------------------------------------------------------------------
+
+Bool_t MadGraphPartonClassifier::hasGoodAncestor(ExRootGenParticle *object)
+{
+  const int kMaxAncestors = 10;
+  Int_t i, pidAbs;
+  ExRootGenParticle *particle = object;
+  set< Int_t >::const_iterator itAncestorIDSet;
+  set< Int_t >::const_iterator itParticleIDSet;
+
+  // keep all particles if there is no pid in the list
+  if(fInclAncestorIDSet.empty())
+  {
+    return kTRUE;
+  }
+
+  for(i = 0; i < kMaxAncestors; ++i)
+  {
+
+    if(particle->M1 < 0 || particle->M2 > -1) return kFALSE;
+
+//    if(particle->PID == 21) return kFALSE;
+
+    particle = static_cast<ExRootGenParticle*>(fBranchParticle->At(particle->M1));
+
+    if(particle->PID == 21) return kFALSE;
+
+    pidAbs = TMath::Abs(particle->PID);
+
+    // keep particles with pid included in list
+    itAncestorIDSet = fInclAncestorIDSet.find(pidAbs);
+
+    if(itAncestorIDSet != fInclAncestorIDSet.end()) return kTRUE;
   }
 
@@ -145 +199 @@
   if(hasBadAncestor(particle)) return -1;
 
+  if(!hasGoodAncestor(particle)) return -1;
+
   return 0;
 }
@@ -197 +253 @@
   {
     pid = param[i].GetInt();
-    fClassifier->InsertAncestorID(pid);
+    fClassifier->InsertExclAncestorID(pid);
+  }
+
+  // read ancestor IDs from configuration file and setup classifier
+
+  param = GetParam("IncludedAncestorIDs");
+  sizeParam = param.GetSize();
+
+  for(i = 0; i < sizeParam; ++i)
+  {
+    pid = param[i].GetInt();
+    fClassifier->InsertInclAncestorID(pid);
   }
 
@@ -209 +276 @@
 
 void MadGraphPartonSelector::Finish()
-{ 
+{
   if(fFilter) delete fFilter;
   if(fClassifier) delete fClassifier;