PWG2/SPECTRA/AliAnalysisTaskProtonsQA.cxx

   1 #include "TChain.h"
   2 #include "TTree.h"
   3 #include "TList.h"
   4 #include "TH1F.h"
   5
   6 #include "AliAnalysisTask.h"
   7 #include "AliAnalysisManager.h"
   8
   9 #include "AliGenEventHeader.h"
  10 #include "AliESDEvent.h"
  11 #include "AliESDInputHandler.h"
  12 #include "AliMCEventHandler.h"
  13 #include "AliMCEvent.h"
  14 #include "AliStack.h"
  15 #include "AliESDVertex.h"
  16 #include "AliPhysicsSelection.h"
  17
  18 #include "AliProtonQAAnalysis.h"
  19 #include "AliProtonAnalysisBase.h"
  20 #include "AliAnalysisTaskProtonsQA.h"
  21
  22 //-----------------------------------------------------------------
  23 //                 AliAnalysisTakProtonsQA class
  24 //   This is the task to run the \bar{p}/p QA analysis
  25 //   Origin: Panos Christakoglou | Panos.Christakoglou@cern.ch
  26 //-----------------------------------------------------------------
  27
  28 ClassImp(AliAnalysisTaskProtonsQA)
  29
  30 //________________________________________________________________________
  31 AliAnalysisTaskProtonsQA::AliAnalysisTaskProtonsQA()
  32   : AliAnalysisTask(), fESD(0), fMC(0), fHistEventStats(0),
  33     fList0(0), fList1(0), fList2(0), fList3(0),
  34     fList4(0), fList5(0), fList6(0), fList7(0), fList8(0),
  35     fProtonQAAnalysis(0) {
  36   //Dummy constructor
  37 }
  38
  39 //________________________________________________________________________
  40 AliAnalysisTaskProtonsQA::AliAnalysisTaskProtonsQA(const char *name)
  41   : AliAnalysisTask(name, ""), fESD(0), fMC(0), fHistEventStats(0),
  42     fList0(0), fList1(0), fList2(0), fList3(0),
  43     fList4(0), fList5(0), fList6(0), fList7(0), fList8(0),
  44     fProtonQAAnalysis(0) {
  45   // Constructor
  46
  47   // Define input and output slots here
  48   // Input slot #0 works with a TChain
  49   DefineInput(0, TChain::Class());
  50   // Output slot #0 writes into a TList container
  51   DefineOutput(0, TList::Class());
  52   DefineOutput(1, TList::Class());
  53   DefineOutput(2, TList::Class());
  54   DefineOutput(3, TList::Class());
  55   DefineOutput(4, TList::Class());
  56   DefineOutput(5, TList::Class());
  57   DefineOutput(6, TList::Class());
  58   DefineOutput(7, TList::Class());
  59   DefineOutput(8, TList::Class());
  60   DefineOutput(9, TH1F::Class());
  61 }
  62
  63 //________________________________________________________________________
  64 void AliAnalysisTaskProtonsQA::ConnectInputData(Option_t *) {
  65   // Connect ESD or AOD here
  66   // Called once
  67
  68   TTree* tree = dynamic_cast<TTree*> (GetInputData(0));
  69   if (!tree) {
  70     Printf("ERROR: Could not read chain from input slot 0");
  71   } else {
  72     AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());
  73
  74     if (!esdH) {
  75       Printf("ERROR: Could not get ESDInputHandler");
  76     } else
  77       fESD = esdH->GetEvent();
  78   }
  79
  80   AliMCEventHandler* mcH = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
  81   if (!mcH) {
  82     Printf("ERROR: Could not retrieve MC event handler");
  83   }
  84   else
  85     fMC = mcH->MCEvent();
  86 }
  87
  88 //________________________________________________________________________
  89 void AliAnalysisTaskProtonsQA::CreateOutputObjects() {
  90   // Create histograms
  91   // Called once
  92   char *gCutName[5] = {"Total","Triggered","Offline trigger",
  93                        "Vertex","Analyzed"};
  94   fHistEventStats = new TH1F("fHistEventStats",
  95                              "Event statistics;;N_{events}",
  96                              5,0.5,5.5);
  97   for(Int_t i = 1; i <= 5; i++)
  98     fHistEventStats->GetXaxis()->SetBinLabel(i,gCutName[i-1]);
  99
 100   fList0 = new TList();
 101   fList0 = fProtonQAAnalysis->GetGlobalQAList();
 102
 103   fList1 = new TList();
 104   fList1 = fProtonQAAnalysis->GetPDGList();
 105
 106   fList2 = new TList();
 107   fList2 = fProtonQAAnalysis->GetMCProcessesList();
 108
 109   fList3 = new TList();
 110   fList3 = fProtonQAAnalysis->GetAcceptedCutList();
 111
 112   fList4 = new TList();
 113   fList4 = fProtonQAAnalysis->GetRejectedCutList();
 114
 115   fList5 = new TList();
 116   fList5 = fProtonQAAnalysis->GetAcceptedDCAList();
 117
 118   fList6 = new TList();
 119   fList6 = fProtonQAAnalysis->GetEfficiencyQAList();
 120
 121   fList7 = new TList();
 122   fList7 = fProtonQAAnalysis->GetVertexQAList();
 123
 124   fList8 = new TList();
 125   fList8 = fProtonQAAnalysis->GetCutEfficiencyList();
 126 }
 127
 128 //________________________________________________________________________
 129 void AliAnalysisTaskProtonsQA::Exec(Option_t *) {
 130   // Main loop
 131   // Called for each event
 132
 133   if (!fESD) {
 134     Printf("ERROR: fESD not available");
 135     return;
 136   }
 137
 138   if (!fMC) {
 139     Printf("ERROR: Could not retrieve MC event");
 140     return;
 141   }
 142
 143   AliGenEventHeader *header = fMC->GenEventHeader();
 144   if (!header) {
 145      Printf("ERROR: Could not retrieve the header");
 146      return;
 147   }
 148
 149   AliStack* stack = fMC->Stack();
 150   if (!stack) {
 151     Printf("ERROR: Could not retrieve the stack");
 152     return;
 153   }
 154
 155   fHistEventStats->Fill(1);
 156   //online trigger
 157   if(dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->IsEventTriggered(fESD,dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->GetTriggerMode())) {
 158     fHistEventStats->Fill(2);
 159
 160     //offline trigger
 161 if(dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->IsOfflineTriggerUsed()) {
 162       AliPhysicsSelection *gPhysicselection = dynamic_cast<AliPhysicsSelection *>(dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->GetPhysicsSelectionObject());
 163       if(gPhysicselection->IsCollisionCandidate(fESD)) {
 164         fHistEventStats->Fill(3);
 165
 166         fProtonQAAnalysis->RunVertexQA(header,
 167                                        fESD);
 168         const AliESDVertex *vertex = dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->GetVertex(fESD,dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->GetAnalysisMode(),dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->GetVxMax(),dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->GetVyMax(),dynamic_cast<AliProtonAnalysisBase*>(fProtonQAAnalysis->GetProtonAnalysisBaseObject())->GetVzMax());
 169         fHistEventStats->Fill(4);
 170
 171         if(vertex) {
 172           fHistEventStats->Fill(5);
 173           fProtonQAAnalysis->RunQAAnalysis(stack, fESD, vertex);
 174           fProtonQAAnalysis->RunMCAnalysis(stack);
 175           fProtonQAAnalysis->RunPIDEfficiencyAnalysis(stack, fESD, vertex);
 176           fProtonQAAnalysis->RunReconstructionEfficiencyAnalysis(fMC,fESD,vertex);
 177           fProtonQAAnalysis->RunCutEfficiencyAnalysis(stack, fESD, vertex);
 178         }//accepted vertex
 179       }//offline trigger
 180     }//offline trigger used
 181   }//online trigger
 182
 183   // Post output data.
 184   PostData(0, fList0);
 185   PostData(1, fList1);
 186   PostData(2, fList2);
 187   PostData(3, fList3);
 188   PostData(4, fList4);
 189   PostData(5, fList5);
 190   PostData(6, fList6);
 191   PostData(7, fList7);
 192   PostData(8, fList8);
 193   PostData(9, fHistEventStats);
 194 }
 195
 196 //________________________________________________________________________
 197 void AliAnalysisTaskProtonsQA::Terminate(Option_t *) {
 198   // Draw result to the screen
 199   // Called once at the end of the query
 200
 201   fList0 = dynamic_cast<TList*> (GetOutputData(0));
 202   if (!fList0) {
 203     Printf("ERROR: fList0 not available");
 204     return;
 205   }
 206   fList1 = dynamic_cast<TList*> (GetOutputData(1));
 207   if (!fList1) {
 208     Printf("ERROR: fList1 not available");
 209     return;
 210   }
 211   fList2 = dynamic_cast<TList*> (GetOutputData(2));
 212   if (!fList2) {
 213     Printf("ERROR: fList2 not available");
 214     return;
 215   }
 216   fList3 = dynamic_cast<TList*> (GetOutputData(3));
 217   if (!fList3) {
 218     Printf("ERROR: fList3 not available");
 219     return;
 220   }
 221   fList4 = dynamic_cast<TList*> (GetOutputData(4));
 222   if (!fList4) {
 223     Printf("ERROR: fList4 not available");
 224     return;
 225   }
 226   fList5 = dynamic_cast<TList*> (GetOutputData(5));
 227   if (!fList5) {
 228     Printf("ERROR: fList5 not available");
 229     return;
 230   }
 231   fList6 = dynamic_cast<TList*> (GetOutputData(6));
 232   if (!fList6) {
 233     Printf("ERROR: fList6 not available");
 234     return;
 235   }
 236   fList7 = dynamic_cast<TList*> (GetOutputData(7));
 237   if (!fList7) {
 238     Printf("ERROR: fList7 not available");
 239     return;
 240   }
 241   fList8 = dynamic_cast<TList*> (GetOutputData(8));
 242   if (!fList8) {
 243     Printf("ERROR: fList8 not available");
 244     return;
 245   }
 246   fHistEventStats = dynamic_cast<TH1F*> (GetOutputData(9));
 247   if (!fHistEventStats) {
 248     Printf("ERROR: fHistEventStats not available");
 249     return;
 250   }
 251 }
 252
 253