CalorimeterUtils: Only if pointer with bad channels exist, access to its content
[u/mrichter/AliRoot.git] / PWG4 / PartCorrBase / AliAnalysisTaskParticleCorrelation.cxx
CommitLineData
1c5acb87 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/* $Id: $ */
16
17//_________________________________________________________________________
18// Analysis task that executes the analysis classes
19// that depend on the PartCorr frame, frame for Particle identification 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 of the analyis
22//
23// -- Author: Gustavo Conesa (INFN-LNF)
24
268934f9 25#include <cstdlib>
fbeaf916 26
27// --- Root ---
1c5acb87 28#include <TROOT.h>
1c5acb87 29#include <TInterpreter.h>
1cd71065 30#include <TClonesArray.h>
1c5acb87 31//#include <Riostream.h>
1cd71065 32//#include <TObjectTable.h>
1c5acb87 33
fbeaf916 34// --- Analysis ---
1c5acb87 35#include "AliAnalysisTaskParticleCorrelation.h"
1c5acb87 36#include "AliAnaPartCorrMaker.h"
37#include "AliCaloTrackReader.h"
7175a03a 38#include "AliPDG.h"
42dc8e7d 39#include "AliAnalysisManager.h"
1c5acb87 40
41ClassImp(AliAnalysisTaskParticleCorrelation)
42
43////////////////////////////////////////////////////////////////////////
fbeaf916 44AliAnalysisTaskParticleCorrelation::AliAnalysisTaskParticleCorrelation():
45 AliAnalysisTaskSE(),
46 fAna(0x0),
47 fOutputContainer(0x0),
0c1383b5 48 fConfigName(""), fCuts(0x0)
1c5acb87 49{
50 // Default constructor
51}
52
53//_____________________________________________________
54AliAnalysisTaskParticleCorrelation::AliAnalysisTaskParticleCorrelation(const char* name):
477d6cee 55 AliAnalysisTaskSE(name),
56 fAna(0x0),
57 fOutputContainer(0x0),
0c1383b5 58 fConfigName(""), fCuts(0x0)
1c5acb87 59{
60 // Default constructor
477d6cee 61
1c5acb87 62 DefineOutput(1, TList::Class());
0c1383b5 63 DefineOutput(2, TList::Class()); // will contain cuts or local params
1c5acb87 64}
65
66//_____________________________________________________
67AliAnalysisTaskParticleCorrelation::~AliAnalysisTaskParticleCorrelation()
68{
69 // Remove all pointers
7787a778 70
71// printf("********** Delete Task\n");
72// // Do not delete it here, already done somewhere else, need to understand where.
73// if(fOutputContainer){
74// fOutputContainer->Clear() ;
75// delete fOutputContainer ;
76// }
1c5acb87 77
1cd71065 78 if(fAna) delete fAna;
7787a778 79
80// printf("********** Task deleted \n");
1c5acb87 81}
82
83//_____________________________________________________
84void AliAnalysisTaskParticleCorrelation::UserCreateOutputObjects()
85{
477d6cee 86 // Create the output container
fbeaf916 87 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelation::UserCreateOutputObjects() - Begin\n");
477d6cee 88
89 //Get list of aod arrays, add each aod array to analysis frame
1cd71065 90 TClonesArray *array = 0;
477d6cee 91 TList * list = fAna->GetAODBranchList();
42dc8e7d 92 TString deltaAODName = (fAna->GetReader())->GetDeltaAODFileName();
477d6cee 93 for(Int_t iaod = 0; iaod < list->GetEntries(); iaod++){
94 array = (TClonesArray*) list->At(iaod);
42dc8e7d 95 if(deltaAODName!="") AddAODBranch("TClonesArray", &array, deltaAODName);//Put it in DeltaAOD file
96 else AddAODBranch("TClonesArray", &array);//Put it in standard AOD file
477d6cee 97 }
1cd71065 98
477d6cee 99 //Histograms container
100 OpenFile(1);
101 fOutputContainer = fAna->GetOutputContainer();
4a745797 102 fOutputContainer->SetOwner(kTRUE);
fbeaf916 103 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelation::UserCreateOutputObjects() - End\n");
609de1b2 104
105 PostData(1,fOutputContainer);
106
1c5acb87 107}
0c1383b5 108//_____________________________________________________
109void AliAnalysisTaskParticleCorrelation::LocalInit()
110{
111 // Local Initialization
112
37c0bcde 113 //Call the Init to initialize the configuration of the analysis
114 Init();
115
0c1383b5 116 // Create cuts/param objects and publish to slot
0c1383b5 117 fCuts = fAna->GetListOfAnalysisCuts();
37c0bcde 118
0c1383b5 119 // Post Data
37c0bcde 120 PostData(2, fCuts);
0c1383b5 121
0c1383b5 122}
1c5acb87 123
124//_____________________________________________________
125void AliAnalysisTaskParticleCorrelation::Init()
126{
127 // Initialization
fbeaf916 128
129 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelation::Init() - Begin\n");
477d6cee 130
131 // Call configuration file if specified
132
133 if (fConfigName.Length()) {
fbeaf916 134 printf("AliAnalysisTaskParticleCorrelation::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data());
477d6cee 135 gROOT->LoadMacro(fConfigName+".C");
136 fAna = (AliAnaPartCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()");
1c5acb87 137 }
1c5acb87 138
fbeaf916 139 if(!fAna) {
ee7d35e1 140 printf("AliAnalysisTaskParticleCorrelation::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n");
fbeaf916 141 abort();
477d6cee 142 }
1c5acb87 143
7175a03a 144 // Add different generator particles to PDG Data Base
145 // to avoid problems when reading MC generator particles
146 AliPDG::AddParticlesToPdgDataBase();
147
486258c9 148 //Set in the reader the name of the task in case is needed
149 (fAna->GetReader())->SetTaskName(GetName());
150
1c5acb87 151 // Initialise analysis
152 fAna->Init();
42dc8e7d 153
154 //Delta AOD
155 if((fAna->GetReader())->GetDeltaAODFileName()!="")
156 AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName());
157
fbeaf916 158 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelation::Init() - End\n");
1c5acb87 159
160}
161
162
163//_____________________________________________________
164void AliAnalysisTaskParticleCorrelation::UserExec(Option_t */*option*/)
165{
166 // Execute analysis for current event
167 //
fbeaf916 168 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelation::UserExec() - Begin\n");
1c5acb87 169
fbeaf916 170 //Get the type of data, check if type is correct
1c5acb87 171 Int_t datatype = fAna->GetReader()->GetDataType();
172 if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD &&
173 datatype != AliCaloTrackReader::kMC){
fbeaf916 174 printf("AliAnalysisTaskParticleCorrelation::UserExec() - Wrong type of data\n");
1c5acb87 175 return ;
176 }
1cd71065 177
477d6cee 178 fAna->GetReader()->SetInputOutputMCEvent(InputEvent(), AODEvent(), MCEvent());
1c5acb87 179
180 //Process event
a79a2424 181 fAna->ProcessEvent((Int_t) Entry(), CurrentFileName());
182 //printf("AliAnalysisTaskParticleCorrelation::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName());
fbeaf916 183 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelation::UserExec() - End\n");
1cd71065 184
1c5acb87 185 PostData(1, fOutputContainer);
1cd71065 186
187 //gObjectTable->Print();
188
1c5acb87 189
190}
191
192//_____________________________________________________
193void AliAnalysisTaskParticleCorrelation::Terminate(Option_t */*option*/)
194{
195 // Terminate analysis
6639984f 196 // Do some plots
6639984f 197
a5cc4f03 198 // Get merged histograms from the output container
a5cc4f03 199 // Propagate histagrams to maker
1cd71065 200 fAna->Terminate((TList*)GetOutputData(1));
a5cc4f03 201
1c5acb87 202}
203