Move creation and filling of analysis parameters from LocalInit to Terminate
[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / AliAnalysisTaskCaloTrackCorrelationM.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 "AliAnalysisTaskCaloTrackCorrelationM.h"
37 #include "AliAnaCaloTrackCorrMaker.h"
38 #include "AliCaloTrackReader.h"
39 #include "AliPDG.h"
40 #include "AliAnalysisManager.h"
41 #include "AliMultiEventInputHandler.h"
42 #include "AliMixedEvent.h"
43 #include "AliAODEvent.h"
44 #include "AliESDEvent.h"
45 #include "AliAnalysisDataSlot.h"
46
47 ClassImp(AliAnalysisTaskCaloTrackCorrelationM)
48
49 //__________________________________________________________________________
50 AliAnalysisTaskCaloTrackCorrelationM::AliAnalysisTaskCaloTrackCorrelationM() :
51   AliAnalysisTaskME(),
52   fAna(0x0),
53   fOutputContainer(0x0),
54   fConfigName(""), 
55   fCuts(0x0), 
56   fInputEvent(NULL)
57 {
58   // Default constructor
59 }
60
61 //__________________________________________________________________________________________
62 AliAnalysisTaskCaloTrackCorrelationM::AliAnalysisTaskCaloTrackCorrelationM(const char* name) :
63   AliAnalysisTaskME(name),
64   fAna(0x0),
65   fOutputContainer(0x0),
66   fConfigName(""), 
67   fCuts(0x0), 
68   fInputEvent(NULL)
69 {
70   // Default constructor
71   
72   DefineOutput(1, TList::Class());
73   DefineOutput(2, TList::Class());  // will contain cuts or local params
74 }
75
76 //___________________________________________________________________________
77 AliAnalysisTaskCaloTrackCorrelationM::~AliAnalysisTaskCaloTrackCorrelationM() 
78 {
79   // Remove all pointers
80         
81   //  // Do not delete it here, already done somewhere else, need to understand where.
82   //  if(fOutputContainer && !AliAnalysisManager::GetAnalysisManager()->IsProofMode()){
83   //    fOutputContainer->Clear() ; 
84   //    delete fOutputContainer ;
85   //  }
86   
87   delete fInputEvent ; 
88   if(fAna) delete fAna;
89   
90 }
91
92 //__________________________________________________________________
93 void AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects()
94 {
95   // Create the output container
96   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - Begin\n");
97   
98   //Get list of aod arrays, add each aod array to analysis frame 
99   TClonesArray *array = 0;
100   TList * list = fAna->FillAndGetAODBranchList();
101   if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - n AOD branches %d\n",list->GetEntries());
102   
103   //Put the delta AODs in output file, std or delta
104   if((fAna->GetReader())->WriteDeltaAODToFile())
105   {
106     TString deltaAODName = (fAna->GetReader())->GetDeltaAODFileName();
107     for(Int_t iaod = 0; iaod < list->GetEntries(); iaod++)
108     {
109       array = (TClonesArray*) list->At(iaod);
110       if(deltaAODName!="") AddAODBranch("TClonesArray", &array, deltaAODName);//Put it in DeltaAOD file
111       else AddAODBranch("TClonesArray", &array);//Put it in standard AOD file
112     } 
113         }
114         
115   //Histograms container
116   OpenFile(1);
117   fOutputContainer = fAna->GetOutputContainer();
118   
119   if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - n histograms %d\n",fOutputContainer->GetEntries());
120   
121   fOutputContainer->SetOwner(kTRUE);
122   
123   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - End\n");
124   
125   PostData(1,fOutputContainer);
126         
127 }
128
129 //____________________________________________________
130 void AliAnalysisTaskCaloTrackCorrelationM::LocalInit()
131 {
132         // Local Initialization
133         
134         //Call the Init to initialize the configuration of the analysis
135         Init();
136         
137 }
138
139 //_______________________________________________
140 void AliAnalysisTaskCaloTrackCorrelationM::Init()
141 {
142   // Initialization
143   
144   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - Begin\n");
145   
146   fInputEvent = new AliMixedEvent() ; 
147   
148   // Call configuration file if specified
149   
150   if (fConfigName.Length())
151   {
152     printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data());
153     gROOT->LoadMacro(fConfigName+".C");
154     fAna = (AliAnaCaloTrackCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()");
155   }
156   
157   if(!fAna)
158   {
159     printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n");
160     abort();
161   }
162   
163   // Add different generator particles to PDG Data Base 
164   // to avoid problems when reading MC generator particles
165   AliPDG::AddParticlesToPdgDataBase();
166   
167   //Set in the reader the name of the task in case is needed
168   (fAna->GetReader())->SetTaskName(GetName());
169         
170   // Initialise analysis
171   fAna->Init();
172   
173   //Delta AOD
174   if((fAna->GetReader())->GetDeltaAODFileName()!="")
175           AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName());
176   
177   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - End\n");
178   
179 }
180
181
182 //_______________________________________________________________________
183 void AliAnalysisTaskCaloTrackCorrelationM::UserExec(Option_t */*option*/)
184 {
185   // Execute analysis for current event
186   
187   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - Begin\n");
188   
189   //Get the type of data, check if type is correct
190   Int_t  datatype = fAna->GetReader()->GetDataType();
191   if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD &&
192      datatype != AliCaloTrackReader::kMC){
193     printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - Wrong type of data\n");
194     return ;
195   }
196   
197   Int_t nev = fInputHandler->GetBufferSize();
198   fInputEvent->Reset();
199   
200   for (Int_t iev = 0; iev < nev; iev++) 
201   {
202     if (datatype == AliCaloTrackReader::kESD) 
203     {
204       AliESDEvent* esd = dynamic_cast<AliESDEvent*>(GetEvent(iev));
205       fInputEvent->AddEvent(esd);
206     } else if (datatype == AliCaloTrackReader::kAOD) 
207     {
208       AliAODEvent* aod = dynamic_cast<AliAODEvent*>(GetEvent(iev));
209       fInputEvent->AddEvent(aod);
210     } else 
211     {
212       AliFatal("need to implement mixed event for MC") ; 
213     }
214   }
215   
216   fInputEvent->Init();
217   
218   fAna->GetReader()->SetInputOutputMCEvent(fInputEvent, AODEvent(), MCEvent());
219   
220   //Process event
221   fAna->ProcessEvent((Int_t) Entry(), CurrentFileName());
222   
223   //printf("AliAnalysisTaskCaloTrackCorrelationM::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName());
224   if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - End\n");
225   
226   PostData(1, fOutputContainer);
227         
228   AliAnalysisDataSlot *out0 = GetOutputSlot(0);
229   if (out0 && out0->IsConnected()) PostData(0, fTreeA);  
230   
231   //gObjectTable->Print();
232   
233 }
234
235 //________________________________________________________________________
236 void AliAnalysisTaskCaloTrackCorrelationM::Terminate(Option_t */*option*/)
237 {
238   // Terminate analysis
239   // Do some plots
240   
241   // Get merged histograms from the output container
242   // Propagate histagrams to maker
243   fAna->Terminate((TList*)GetOutputData(1));
244
245   // Create cuts/param objects and publish to slot
246         fCuts = fAna->GetListOfAnalysisCuts();
247   fCuts ->SetOwner(kTRUE);
248   
249         // Post Data
250         PostData(2, fCuts);
251   
252 }
253