Changeset 11 for trunk/modules

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

Location:

trunk/modules

Files:

• MadGraphCDFConeJetFinder.cc (added)
• MadGraphCDFConeJetFinder.h (added)
• MadGraphConeJetFinder.cc (deleted)
• MadGraphConeJetFinder.h (deleted)
• MadGraphPartonSelector.cc (modified) (7 diffs)
• MadGraphSISConeJetFinder.cc (added)
• MadGraphSISConeJetFinder.h (added)
• MadGraphShowerPartonSelector.cc (modified) (8 diffs)
• ModulesLinkDef.h (modified) (2 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;

trunk/modules/MadGraphShowerPartonSelector.cc

@@ -35 +35 @@
   void SetEtaMax(Double_t eta);
   void InsertParticleID(Int_t pid);
-  void InsertAncestorID(Int_t pid);
+  void InsertExclAncestorID(Int_t pid);
+  void InsertInclAncestorID(Int_t pid);
   void SetHadronizationInfo(Bool_t info);
 
@@ -41 +42 @@
 
   Bool_t hasBadAncestor(ExRootGenParticle *object);
+  Bool_t hasGoodAncestor(ExRootGenParticle *object);
 
   Double_t fEtaMax;
@@ -47 +49 @@
 
   set< Int_t > fParticleIDSet;
-  set< Int_t > fAncestorIDSet;
+  set< Int_t > fExclAncestorIDSet;
+  set< Int_t > fInclAncestorIDSet;
 };
 
@@ -73 +76 @@
 //------------------------------------------------------------------------------
 
-void MadGraphShowerPartonClassifier::InsertAncestorID(Int_t pid)
-{
-  fAncestorIDSet.insert(pid);
+void MadGraphShowerPartonClassifier::InsertExclAncestorID(Int_t pid)
+{
+  fExclAncestorIDSet.insert(pid);
+}
+
+//------------------------------------------------------------------------------
+
+void MadGraphShowerPartonClassifier::InsertInclAncestorID(Int_t pid)
+{
+  fInclAncestorIDSet.insert(pid);
 }
 
@@ -96 +106 @@
   if(particle->PID == 21) return kFALSE;
 
+  // keep all particles if there is no pid in the list
+  if(fExclAncestorIDSet.empty())
+  {
+    return kFALSE;
+  }
+
   pidAbs = TMath::Abs(particle->PID);
 
   // 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;
   if(particle->M2 > -1) return kFALSE;
 
@@ -115 +131 @@
 
     // 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;
+    if(particle->M2 > -1) return kFALSE;
+  }
+
+  return kFALSE;
+}
+
+//------------------------------------------------------------------------------
+
+Bool_t MadGraphShowerPartonClassifier::hasGoodAncestor(ExRootGenParticle *object)
+{
+  const int kMaxAncestors = 10;
+  Int_t i, pidAbs;
+  ExRootGenParticle *particle = object;
+  set< Int_t >::const_iterator itAncestorIDSet;
+
+  for(i = 0; i < kMaxAncestors && particle->Status != 3; ++i)
+  {
+    if(particle->M1 < 0) return kFALSE;
+
+    particle = static_cast<ExRootGenParticle*>(fBranchParticle->At(particle->M1));
+  }
+
+  if(particle->PID == 21) return kFALSE;
+
+  // keep all particles if there is no pid in the list
+  if(fInclAncestorIDSet.empty())
+  {
+    return kTRUE;
+  }
+
+  pidAbs = TMath::Abs(particle->PID);
+
+  // keep particles with pid included in list
+  itAncestorIDSet = fInclAncestorIDSet.find(pidAbs);
+
+  if(itAncestorIDSet != fInclAncestorIDSet.end()) return kTRUE;
+  if(particle->M2 > -1) return kFALSE;
+
+  for(i = 0; i < kMaxAncestors; ++i)
+  {
+    if(particle->M1 < 0) return kFALSE;
+
+    if(particle->PID == 21) return kFALSE;
+
+    particle = static_cast<ExRootGenParticle*>(fBranchParticle->At(particle->M1));
+
+    pidAbs = TMath::Abs(particle->PID);
+
+    // keep particles with pid included in list
+    itAncestorIDSet = fInclAncestorIDSet.find(pidAbs);
+
+    if(itAncestorIDSet != fInclAncestorIDSet.end()) return kTRUE;
     if(particle->M2 > -1) return kFALSE;
   }
@@ -159 +227 @@
   if(hasBadAncestor(particle)) return -1;
 
+  if(!hasGoodAncestor(particle)) return -1;
+
   return 0;
 }
@@ -214 +284 @@
   {
     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);
   }
 

trunk/modules/ModulesLinkDef.h

@@ -7 +7 @@
 #include "modules/MadGraphMatchingTreeWriter.h"
 #include "modules/MadGraphKtJetFinder.h"
-#include "modules/MadGraphConeJetFinder.h"
+#include "modules/MadGraphCDFConeJetFinder.h"
+#include "modules/MadGraphSISConeJetFinder.h"
 #include "modules/MadGraphIsolatedLeptonFinder.h"
 #include "modules/PythiaFix.h"
@@ -29 +30 @@
 #pragma link C++ class MadGraphMatchingTreeWriter+;
 #pragma link C++ class MadGraphKtJetFinder+;
-#pragma link C++ class MadGraphConeJetFinder+;
+#pragma link C++ class MadGraphCDFConeJetFinder+;
+#pragma link C++ class MadGraphSISConeJetFinder+;
 #pragma link C++ class MadGraphIsolatedLeptonFinder+;
 #pragma link C++ class PythiaFix+;