function to reset stat error on corrections
[u/mrichter/AliRoot.git] / PWG0 / AliPWG0Helper.cxx
index 9593f8e..66ff632 100644 (file)
 
 #include <AliHeader.h>
 #include <AliStack.h>
-#include <AliLog.h>
 
 #include <AliLog.h>
 #include <AliESD.h>
 #include <AliESDEvent.h>
 #include <AliESDVertex.h>
 #include <AliVertexerTracks.h>
+#include <AliMultiplicity.h>
 
 #include <AliGenEventHeader.h>
 #include <AliGenPythiaEventHeader.h>
 #include <AliGenCocktailEventHeader.h>
 #include <AliGenDPMjetEventHeader.h>
+#include <AliESDVZERO.h>
 
 //____________________________________________________________________
 ClassImp(AliPWG0Helper)
@@ -33,53 +34,32 @@ ClassImp(AliPWG0Helper)
 Int_t AliPWG0Helper::fgLastProcessType = -1;
 
 //____________________________________________________________________
-Bool_t AliPWG0Helper::IsEventTriggered(const AliESD* aEsd, Trigger trigger)
+Bool_t AliPWG0Helper::TestVertex(const AliESDVertex* vertex, AnalysisMode analysisMode, Bool_t debug)
 {
-  // see function with ULong64_t argument
+    // Checks if a vertex meets the needed quality criteria
 
-  ULong64_t triggerMask = aEsd->GetTriggerMask();
-  return IsEventTriggered(triggerMask, trigger);
-}
+  Float_t requiredZResolution = -1;
+  if (analysisMode & kSPD || analysisMode & kTPCITS)
+  {
+    requiredZResolution = 0.1;
+  }
+  else if (analysisMode & kTPC)
+    requiredZResolution = 10.;
 
-//____________________________________________________________________
-Bool_t AliPWG0Helper::IsEventTriggered(ULong64_t triggerMask, Trigger trigger)
-{
-  // check if the event was triggered
-  //
-  // this function needs the branch fTriggerMask
-  
-  // definitions from p-p.cfg
-  ULong64_t spdFO = (1 << 14);
-  ULong64_t v0left = (1 << 11);
-  ULong64_t v0right = (1 << 12);
+  // check resolution
+  Double_t zRes = vertex->GetZRes();
 
-  switch (trigger)
-  {
-    case kMB1:
-    {
-      if (triggerMask & spdFO || ((triggerMask & v0left) || (triggerMask & v0right)))
-        return kTRUE;
-      break;
-    }
-    case kMB2:
-    {
-      if (triggerMask & spdFO && ((triggerMask & v0left) || (triggerMask & v0right)))
-        return kTRUE;
-      break;
-    }
-    case kSPDFASTOR:
-    {
-      if (triggerMask & spdFO)
-        return kTRUE;
-      break;
-    }
+  if (zRes > requiredZResolution) {
+    if (debug)
+      Printf("AliPWG0Helper::TestVertex: Resolution too poor %f (required: %f", zRes, requiredZResolution);
+    return kFALSE;
   }
 
-  return kFALSE;
+  return kTRUE;
 }
 
 //____________________________________________________________________
-const AliESDVertex* AliPWG0Helper::GetVertex(AliESDEvent* aEsd, AnalysisMode analysisMode, Bool_t debug,Bool_t bRedoTPC)
+const AliESDVertex* AliPWG0Helper::GetVertex(AliESDEvent* aEsd, AnalysisMode analysisMode, Bool_t debug, Bool_t bRedoTPC)
 {
   // Get the vertex from the ESD and returns it if the vertex is valid
   //
@@ -87,31 +67,30 @@ const AliESDVertex* AliPWG0Helper::GetVertex(AliESDEvent* aEsd, AnalysisMode ana
   // also the quality criteria that are applied)
 
   const AliESDVertex* vertex = 0;
-  Float_t requiredZResolution = -1;
-  if (analysisMode == kSPD || analysisMode == kTPCITS)
+  if (analysisMode & kSPD || analysisMode & kTPCITS)
   {
     vertex = aEsd->GetPrimaryVertexSPD();
-    requiredZResolution = 0.1;
     if (debug)
       Printf("AliPWG0Helper::GetVertex: Returning SPD vertex");
   }
-  else if (analysisMode == kTPC) 
+  else if (analysisMode & kTPC)
   {
     if(bRedoTPC){
-      Double_t kBz = aEsd->GetMagneticField(); 
+      if (debug)
+        Printf("AliPWG0Helper::GetVertex: Redoing vertex");
+      Double_t kBz = aEsd->GetMagneticField();
       AliVertexerTracks vertexer(kBz);
-      vertexer.SetTPCMode(); 
+      vertexer.SetTPCMode();
       AliESDVertex *vTPC = vertexer.FindPrimaryVertex(aEsd);
       aEsd->SetPrimaryVertexTPC(vTPC);
       for (Int_t i=0; i<aEsd->GetNumberOfTracks(); i++) {
        AliESDtrack *t = aEsd->GetTrack(i);
        t->RelateToVertexTPC(vTPC, kBz, kVeryBig);
-      } 
+      }
       delete vTPC;
     }
 
     vertex = aEsd->GetPrimaryVertexTPC();
-    requiredZResolution = 10.;
     if (debug)
       Printf("AliPWG0Helper::GetVertex: Returning vertex from tracks");
   }
@@ -129,17 +108,15 @@ const AliESDVertex* AliPWG0Helper::GetVertex(AliESDEvent* aEsd, AnalysisMode ana
     if (debug){
       Printf("AliPWG0Helper::GetVertex: NContributors() <= 0: %d",vertex->GetNContributors());
       Printf("AliPWG0Helper::GetVertex: NIndices(): %d",vertex->GetNIndices());
-
+      vertex->Print();
     }
     return 0;
   }
 
   // check resolution
   Double_t zRes = vertex->GetZRes();
-
-  if (zRes == 0 || zRes > requiredZResolution) {
-    if (debug)
-      Printf("AliPWG0Helper::GetVertex: Resolution too poor %f (required: %f", zRes, requiredZResolution);
+  if (zRes == 0) {
+    Printf("AliPWG0Helper::GetVertex: UNEXPECTED: resolution is 0.");
     return 0;
   }
 
@@ -527,32 +504,33 @@ void AliPWG0Helper::NormalizeToBinWidth(TH2* hist)
 }
 
 //____________________________________________________________________
-void AliPWG0Helper::PrintConf(AnalysisMode analysisMode, Trigger trigger)
+void AliPWG0Helper::PrintConf(AnalysisMode analysisMode, AliTriggerAnalysis::Trigger trigger)
 {
   //
   // Prints the given configuration
   //
 
-  TString str(">>>> Running with ");
-
-  switch (analysisMode)
-  {
-    case kInvalid: str += "invalid setting"; break;
-    case kSPD : str += "SPD-only"; break;
-    case kTPC : str += "TPC-only"; break;
-    case kTPCITS : str += "Global tracking"; break;
-  }
+  TString str(">>>> Running with >");
 
-  str += " and trigger ";
+  if (analysisMode & kSPD)
+    str += "SPD-only";
+    
+  if (analysisMode & kTPC)
+     str += "TPC-only";
+    
+  if (analysisMode & kTPCITS)
+     str += "Global tracking";
 
-  switch (trigger)
+  if (analysisMode & kFieldOn)
   {
-    case kMB1 : str += "MB1"; break;
-    case kMB2 : str += "MB2"; break;
-    case kSPDFASTOR : str += "SPD FASTOR"; break;
+     str += " (with field)";
   }
-
-  str += " <<<<";
+  else
+     str += " (WITHOUT field)";
+  
+  str += "< and trigger >";
+  str += AliTriggerAnalysis::GetTriggerName(trigger);
+  str += "< <<<<";
 
   Printf("%s", str.Data());
 }