1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
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
24 // -- Author: Gustavo Conesa (INFN-LNF, LPSC-Grenoble)
30 #include <TInterpreter.h>
31 #include <TClonesArray.h>
32 //#include <Riostream.h>
33 //#include <TObjectTable.h>
36 #include "AliAnalysisTaskCaloTrackCorrelation.h"
37 #include "AliAnaCaloTrackCorrMaker.h"
38 #include "AliCaloTrackReader.h"
40 #include "AliAnalysisManager.h"
41 #include "AliInputEventHandler.h"
43 ClassImp(AliAnalysisTaskCaloTrackCorrelation)
45 //________________________________________________________________________
46 AliAnalysisTaskCaloTrackCorrelation::AliAnalysisTaskCaloTrackCorrelation() :
49 fOutputContainer(0x0),
53 // Default constructor
56 //________________________________________________________________________________________
57 AliAnalysisTaskCaloTrackCorrelation::AliAnalysisTaskCaloTrackCorrelation(const char* name) :
58 AliAnalysisTaskSE(name),
60 fOutputContainer(0x0),
64 // Default constructor
66 DefineOutput(1, TList::Class());
67 DefineOutput(2, TList::Class()); // will contain cuts or local params
70 //_________________________________________________________________________
71 AliAnalysisTaskCaloTrackCorrelation::~AliAnalysisTaskCaloTrackCorrelation()
73 // Remove all pointers
75 if (fOutputContainer && ! AliAnalysisManager::GetAnalysisManager()->IsProofMode()) {
76 fOutputContainer->Clear() ;
77 delete fOutputContainer ;
80 if (fAna && ! AliAnalysisManager::GetAnalysisManager()->IsProofMode()) delete fAna;
84 //_________________________________________________________________
85 void AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects()
87 // Create the output container
88 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - Begin\n");
90 //Get list of aod arrays, add each aod array to analysis frame
91 TClonesArray *array = 0;
92 TList * list = fAna->FillAndGetAODBranchList(); //Loop the analysis and create the list of branches
93 if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - n AOD branches %d\n",list->GetEntries());
95 //Put the delta AODs in output file, std or delta
96 if((fAna->GetReader())->WriteDeltaAODToFile()){
97 TString deltaAODName = (fAna->GetReader())->GetDeltaAODFileName();
98 for(Int_t iaod = 0; iaod < list->GetEntries(); iaod++){
99 array = (TClonesArray*) list->At(iaod);
100 if(deltaAODName!="") AddAODBranch("TClonesArray", &array, deltaAODName);//Put it in DeltaAOD file
101 else AddAODBranch("TClonesArray", &array);//Put it in standard AOD file
105 //Histograms container
107 fOutputContainer = fAna->GetOutputContainer();
109 if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - n histograms %d\n",fOutputContainer->GetEntries());
111 fOutputContainer->SetOwner(kTRUE);
113 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserCreateOutputObjects() - End\n");
115 PostData(1,fOutputContainer);
119 //___________________________________________________
120 void AliAnalysisTaskCaloTrackCorrelation::LocalInit()
122 // Local Initialization
124 //Call the Init to initialize the configuration of the analysis
127 // Create cuts/param objects and publish to slot
128 fCuts = fAna->GetListOfAnalysisCuts();
129 fCuts ->SetOwner(kTRUE);
135 //______________________________________________
136 void AliAnalysisTaskCaloTrackCorrelation::Init()
140 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::Init() - Begin\n");
142 // Call configuration file if specified
144 if (fConfigName.Length()) {
145 printf("AliAnalysisTaskCaloTrackCorrelation::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data());
146 gROOT->LoadMacro(fConfigName+".C");
147 fAna = (AliAnaCaloTrackCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()");
151 printf("AliAnalysisTaskCaloTrackCorrelation::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n");
155 // Add different generator particles to PDG Data Base
156 // to avoid problems when reading MC generator particles
157 AliPDG::AddParticlesToPdgDataBase();
159 //Set in the reader the name of the task in case is needed
160 (fAna->GetReader())->SetTaskName(GetName());
162 // Initialise analysis
166 if((fAna->GetReader())->GetDeltaAODFileName()!="")
167 AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName());
169 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::Init() - End\n");
174 //______________________________________________________________________
175 void AliAnalysisTaskCaloTrackCorrelation::UserExec(Option_t */*option*/)
177 // Execute analysis for current event
179 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - Begin\n");
181 //Get the type of data, check if type is correct
182 Int_t datatype = fAna->GetReader()->GetDataType();
183 if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD &&
184 datatype != AliCaloTrackReader::kMC){
185 printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - Wrong type of data\n");
189 fAna->GetReader()->SetInputOutputMCEvent(InputEvent(), AODEvent(), MCEvent());
192 fAna->ProcessEvent((Int_t) Entry(), CurrentFileName());
194 //printf("AliAnalysisTaskCaloTrackCorrelation::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName());
195 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelation::UserExec() - End\n");
197 PostData(1, fOutputContainer);
199 //gObjectTable->Print();
203 //_______________________________________________________________________
204 void AliAnalysisTaskCaloTrackCorrelation::Terminate(Option_t */*option*/)
206 // Terminate analysis
209 // Get merged histograms from the output container
210 // Propagate histagrams to maker
211 fAna->Terminate((TList*)GetOutputData(1));
215 //__________________________________________________________
216 void AliAnalysisTaskCaloTrackCorrelation::FinishTaskOutput()
218 // Put in the output some event summary histograms
219 AliAnalysisManager *am = AliAnalysisManager::GetAnalysisManager();
221 AliInputEventHandler *inputH = dynamic_cast<AliInputEventHandler*>(am->GetInputEventHandler());
225 TH2F *histStat = dynamic_cast<TH2F*>(inputH->GetStatistics());
226 TH2F *histBin0 = dynamic_cast<TH2F*>(inputH->GetStatistics("BIN0"));
228 if(histStat)fOutputContainer->Add(histStat);
229 else if(DebugLevel()>0)
230 printf("AliAnalysisTaskCaloTrackCorrelation::FinishTaskOutput() - Stat histogram not available check, \n if ESDs, that AliPhysicsSelection was on, \n if AODs, if EventStat_temp.root exists \n");
232 if(histBin0)fOutputContainer->Add(histBin0);