Move creation and filling of analysis parameters from LocalInit to Terminate
[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / AliAnalysisTaskCaloTrackCorrelation.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 //_________________________________________________________________________
17 // Analysis task that executes the analysis classes
18 // that depend on the CaloTrackCorr frame, frame for Particle identification 
19 // with calorimeters and tracks and correlations.
20 // Specially designed for calorimeters but also can be used for charged tracks
21 // Input of this task is a configuration file that contains all the settings 
22 // of the analysis
23 //
24 // -- Author: Gustavo Conesa (INFN-LNF, LPSC-Grenoble)
25
26 #include <cstdlib>
27
28 // --- Root ---
29 #include <TROOT.h>
30 #include <TInterpreter.h>
31 #include <TClonesArray.h>
32 //#include <Riostream.h>
33 //#include <TObjectTable.h>
34
35 // --- Analysis ---
36 #include "AliAnalysisTaskCaloTrackCorrelation.h"
37 #include "AliAnaCaloTrackCorrMaker.h"
38 #include "AliCaloTrackReader.h"
39 #include "AliPDG.h"
40 #include "AliAnalysisManager.h"
41 #include "AliInputEventHandler.h"
42
43 ClassImp(AliAnalysisTaskCaloTrackCorrelation)
44
45 //________________________________________________________________________
46 AliAnalysisTaskCaloTrackCorrelation::AliAnalysisTaskCaloTrackCorrelation() :
47   AliAnalysisTaskSE(),
48   fAna(0x0),
49   fOutputContainer(0x0),
50   fConfigName(""), 
51   fCuts(0x0)
52 {
53   // Default constructor
54 }
55
56 //________________________________________________________________________________________
57 AliAnalysisTaskCaloTrackCorrelation::AliAnalysisTaskCaloTrackCorrelation(const char* name) :
58   AliAnalysisTaskSE(name),
59   fAna(0x0),
60   fOutputContainer(0x0),
61   fConfigName(""), 
62   fCuts(0x0)
63 {
64   // Default constructor
65   
66   DefineOutput(1, TList::Class());
67   DefineOutput(2, TList::Class());  // will contain cuts or local params
68 }
69
70 //_________________________________________________________________________
71 AliAnalysisTaskCaloTrackCorrelation::~AliAnalysisTaskCaloTrackCorrelation() 
72 {
73   // Remove all pointers
74   
75   if (fOutputContainer && ! AliAnalysisManager::GetAnalysisManager()->IsProofMode()) 
76   {
77     fOutputContainer->Clear() ; 
78     delete fOutputContainer ;
79   }
80   
81   if (fAna && ! AliAnalysisManager::GetAnalysisManager()->IsProofMode()) delete fAna;
82   
83 }
84
85 //_________________________________________________________________
86 void AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects()
87 {
88   // Create the output container
89   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - Begin\n");
90   
91   //Get list of aod arrays, add each aod array to analysis frame 
92   TList * list = fAna->FillAndGetAODBranchList(); //Loop the analysis and create the list of branches
93   
94   if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - n AOD branches %d\n",list->GetEntries());
95   
96   //Put the delta AODs in output file, std or delta
97   if((fAna->GetReader())->WriteDeltaAODToFile())
98   {
99     TString deltaAODName = (fAna->GetReader())->GetDeltaAODFileName();
100     for(Int_t iaod = 0; iaod < list->GetEntries(); iaod++)
101     {
102       TClonesArray * array = (TClonesArray*) list->At(iaod);
103       if(deltaAODName!="") AddAODBranch("TClonesArray", &array, deltaAODName);//Put it in DeltaAOD file
104       else AddAODBranch("TClonesArray", &array);//Put it in standard AOD file
105     } 
106         }
107   
108   //Histograms container
109   OpenFile(1);
110   fOutputContainer = fAna->GetOutputContainer();
111   
112   if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - n histograms %d\n",fOutputContainer->GetEntries());
113   
114   fOutputContainer->SetOwner(kTRUE);
115   
116   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - End\n");
117   
118   PostData(1,fOutputContainer);
119   
120 }
121
122 //___________________________________________________
123 void AliAnalysisTaskCaloTrackCorrelation::LocalInit()
124 {
125         // Local Initialization
126         
127         //Call the Init to initialize the configuration of the analysis
128         Init();
129         
130 }
131
132 //______________________________________________
133 void AliAnalysisTaskCaloTrackCorrelation::Init()
134 {
135   // Initialization
136   
137   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::Init() - Begin\n");
138   
139   // Call configuration file if specified
140   
141   if (fConfigName.Length()) 
142   {
143     printf("AliAnalysisTaskCaloTrackCorrelation::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data());
144     gROOT->LoadMacro(fConfigName+".C");
145     fAna = (AliAnaCaloTrackCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()");
146   }
147   
148   if(!fAna) {
149     printf("AliAnalysisTaskCaloTrackCorrelation::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n");
150     abort();
151   }
152   
153   // Add different generator particles to PDG Data Base 
154   // to avoid problems when reading MC generator particles
155   AliPDG::AddParticlesToPdgDataBase();
156   
157   //Set in the reader the name of the task in case is needed
158   (fAna->GetReader())->SetTaskName(GetName());
159         
160   // Initialise analysis
161   fAna->Init();
162   
163   //Delta AOD
164   if((fAna->GetReader())->GetDeltaAODFileName()!="")
165           AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName());
166   
167   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::Init() - End\n");
168   
169 }
170
171
172 //______________________________________________________________________
173 void AliAnalysisTaskCaloTrackCorrelation::UserExec(Option_t */*option*/)
174 {
175   // Execute analysis for current event
176   
177   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - Begin\n");
178   
179   //Get the type of data, check if type is correct
180   Int_t  datatype = fAna->GetReader()->GetDataType();
181   if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD &&
182      datatype != AliCaloTrackReader::kMC)
183   {
184     printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - Wrong type of data\n");
185     return ;
186   }
187   
188   fAna->GetReader()->SetInputOutputMCEvent(InputEvent(), AODEvent(), MCEvent());
189   
190   //Process event
191   fAna->ProcessEvent((Int_t) Entry(), CurrentFileName());
192   
193   //printf("AliAnalysisTaskCaloTrackCorrelation::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName());
194   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - End\n");
195         
196   PostData(1, fOutputContainer);
197         
198   //gObjectTable->Print();
199   
200 }
201
202 //_______________________________________________________________________
203 void AliAnalysisTaskCaloTrackCorrelation::Terminate(Option_t */*option*/)
204 {
205   // Terminate analysis
206   // Do some plots
207   
208   // Get merged histograms from the output container
209   // Propagate histagrams to maker
210   fAna->Terminate((TList*)GetOutputData(1));
211         
212   // Create cuts/param objects and publish to slot
213         fCuts = fAna->GetListOfAnalysisCuts();
214         fCuts ->SetOwner(kTRUE);
215         // Post Data
216         PostData(2, fCuts);
217   
218 }
219
220 //__________________________________________________________
221 void AliAnalysisTaskCaloTrackCorrelation::FinishTaskOutput()
222 {
223   // Put in the output some event summary histograms
224   AliAnalysisManager *am = AliAnalysisManager::GetAnalysisManager();
225   
226   AliInputEventHandler *inputH = dynamic_cast<AliInputEventHandler*>(am->GetInputEventHandler());
227   
228   if (!inputH) return; 
229   
230   TH2F *histStat = dynamic_cast<TH2F*>(inputH->GetStatistics()); 
231   TH2F *histBin0 = dynamic_cast<TH2F*>(inputH->GetStatistics("BIN0"));
232   
233   if(histStat)fOutputContainer->Add(histStat); 
234   else if(DebugLevel()>0) 
235     printf("AliAnalysisTaskCaloTrackCorrelation::FinishTaskOutput() - Stat histogram not available check, \n if ESDs, that AliPhysicsSelection was on, \n if AODs, if EventStat_temp.root exists \n");
236   
237   if(histBin0)fOutputContainer->Add(histBin0); 
238   
239 }
240