fading out dependent helper (AliPWG0depHelper)
[u/mrichter/AliRoot.git] / PWG0 / AliPWG0Helper.cxx
index 5466eec538dace8ceeb9d4bd0dc6e1a6a90bcab3..99c24a633ca554cd2cb247e7b1105b4d9961740e 100644 (file)
@@ -6,16 +6,34 @@
 #include <TParticlePDG.h>
 #include <TH1.h>
 #include <TH3.h>
 #include <TParticlePDG.h>
 #include <TH1.h>
 #include <TH3.h>
+#include <TList.h>
+
+#include <AliHeader.h>
+#include <AliStack.h>
+#include <AliLog.h>
 
 #include <AliLog.h>
 #include <AliESD.h>
 #include <AliESDVertex.h>
 
 
 #include <AliLog.h>
 #include <AliESD.h>
 #include <AliESDVertex.h>
 
+#include <AliGenEventHeader.h>
+#include <AliGenPythiaEventHeader.h>
+#include <AliGenCocktailEventHeader.h>
+
 //____________________________________________________________________
 ClassImp(AliPWG0Helper)
 
 //____________________________________________________________________
 //____________________________________________________________________
 ClassImp(AliPWG0Helper)
 
 //____________________________________________________________________
-Bool_t AliPWG0Helper::IsEventTriggered(AliESD* aEsd, Trigger trigger)
+Bool_t AliPWG0Helper::IsEventTriggered(const AliESD* aEsd, Trigger trigger)
+{
+  // see function with ULong64_t argument
+
+  ULong64_t triggerMask = aEsd->GetTriggerMask();
+  return IsEventTriggered(triggerMask, trigger);
+}
+
+//____________________________________________________________________
+Bool_t AliPWG0Helper::IsEventTriggered(ULong64_t triggerMask, Trigger trigger)
 {
   // check if the event was triggered
   //
 {
   // check if the event was triggered
   //
@@ -26,8 +44,6 @@ Bool_t AliPWG0Helper::IsEventTriggered(AliESD* aEsd, Trigger trigger)
   // VZERO_OR_LEFT   : 1
   // VZERO_OR_RIGHT  : 2
 
   // VZERO_OR_LEFT   : 1
   // VZERO_OR_RIGHT  : 2
 
-  ULong64_t triggerMask = aEsd->GetTriggerMask();
-
   switch (trigger)
   {
     case kMB1:
   switch (trigger)
   {
     case kMB1:
@@ -48,14 +64,23 @@ Bool_t AliPWG0Helper::IsEventTriggered(AliESD* aEsd, Trigger trigger)
 }
 
 //____________________________________________________________________
 }
 
 //____________________________________________________________________
-Bool_t AliPWG0Helper::IsVertexReconstructed(AliESD* aEsd)
+Bool_t AliPWG0Helper::IsVertexReconstructed(const AliESD* aEsd)
+{
+  // see function with AliESDVertex argument
+
+  const AliESDVertex* vtxESD = aEsd->GetVertex();
+  return IsVertexReconstructed(vtxESD);
+}
+
+//____________________________________________________________________
+Bool_t AliPWG0Helper::IsVertexReconstructed(const AliESDVertex* vtxESD)
 {
   // checks if the vertex is reasonable
   //
   // this function needs the branches fSPDVertex*
 
 {
   // checks if the vertex is reasonable
   //
   // this function needs the branches fSPDVertex*
 
-
-  const AliESDVertex* vtxESD = aEsd->GetVertex();
+  if (!vtxESD)
+    return kFALSE;
 
   // the vertex should be reconstructed
   if (strcmp(vtxESD->GetName(), "default")==0)
 
   // the vertex should be reconstructed
   if (strcmp(vtxESD->GetName(), "default")==0)
@@ -237,3 +262,95 @@ const char* AliPWG0Helper::GetAxisTitle(TH3* hist, const char axis)
 
   return 0;
 }
 
   return 0;
 }
+
+//____________________________________________________________________
+Int_t AliPWG0Helper::GetPythiaEventProcessType(AliHeader* aHeader, Bool_t adebug) {
+  //
+  // get the process type of the event.
+  //
+
+  // can only read pythia headers, either directly or from cocktalil header
+  AliGenPythiaEventHeader* pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(aHeader->GenEventHeader());
+
+  if (!pythiaGenHeader) {
+
+    AliGenCocktailEventHeader* genCocktailHeader = dynamic_cast<AliGenCocktailEventHeader*>(aHeader->GenEventHeader());
+    if (!genCocktailHeader) {
+      printf("AliPWG0Helper::GetProcessType : Unknown header type (not Pythia or Cocktail). \n");
+      return -1;
+    }
+
+    TList* headerList = genCocktailHeader->GetHeaders();
+    if (!headerList) {
+      return -1;
+    }
+
+    for (Int_t i=0; i<headerList->GetEntries(); i++) {
+      pythiaGenHeader = dynamic_cast<AliGenPythiaEventHeader*>(headerList->At(i));
+      if (pythiaGenHeader)
+        break;
+    }
+
+    if (!pythiaGenHeader) {
+      printf("AliPWG0Helper::GetProcessType : Could not find Pythia header. \n");
+      return -1;
+    }
+  }
+
+  if (adebug) {
+    printf("AliPWG0Helper::GetProcessType : Pythia process type found: %d \n",pythiaGenHeader->ProcessType());
+  }
+
+  return pythiaGenHeader->ProcessType();
+}
+
+//____________________________________________________________________
+TParticle* AliPWG0Helper::FindPrimaryMother(AliStack* stack, Int_t label)
+{
+  //
+  // Finds the first mother among the primary particles of the particle identified by <label>,
+  // i.e. the primary that "caused" this particle
+  //
+
+  Int_t motherLabel = FindPrimaryMotherLabel(stack, label);
+  if (motherLabel < 0)
+    return 0;
+
+  return stack->Particle(motherLabel);
+}
+
+//____________________________________________________________________
+Int_t AliPWG0Helper::FindPrimaryMotherLabel(AliStack* stack, Int_t label)
+{
+  //
+  // Finds the first mother among the primary particles of the particle identified by <label>,
+  // i.e. the primary that "caused" this particle
+  //
+  // returns its label
+  //
+
+  Int_t nPrim  = stack->GetNprimary();
+
+  while (label >= nPrim)
+  {
+    //printf("Particle %d (pdg %d) is not a primary. Let's check its mother %d\n", label, mother->GetPdgCode(), mother->GetMother(0));
+
+    TParticle* particle = stack->Particle(label);
+    if (!particle)
+    {
+      AliDebugGeneral("FindPrimaryMother", AliLog::kError, Form("UNEXPECTED: particle with label %d not found in stack.", label));
+      return -1;
+    }
+    // find mother
+    if (particle->GetMother(0) < 0)
+    {
+      AliDebugGeneral("FindPrimaryMother", AliLog::kError, Form("UNEXPECTED: Could not find mother of secondary particle %d.", label));
+      return -1;
+    }
+
+    label = particle->GetMother(0);
+  }
+
+  return label;
+}