Updating example reco macro for real data
[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJets.cxx
CommitLineData
e5b96aa0 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 **************************************************************************/
f1a3e20a 15
16/* $Id$ */
8838ab7a 17#include <Riostream.h>
e5b96aa0 18#include <TROOT.h>
19#include <TSystem.h>
20#include <TInterpreter.h>
21#include <TChain.h>
76c48857 22#include <TFile.h>
22003a60 23#include <TList.h>
e5b96aa0 24
25#include "AliAnalysisTaskJets.h"
1d27ecd2 26#include "AliAnalysisManager.h"
e5b96aa0 27#include "AliJetFinder.h"
99c8d2ad 28#include "AliJetHeader.h"
8838ab7a 29#include "AliJetReader.h"
30#include "AliJetReaderHeader.h"
22003a60 31#include "AliJetHistos.h"
1b307662 32#include "AliESDEvent.h"
7f7091b8 33#include "AliESD.h"
1d27ecd2 34#include "AliAODEvent.h"
bcec7a80 35#include "AliAODJetEventBackground.h"
1d27ecd2 36#include "AliAODHandler.h"
691685d6 37#include "AliMCEventHandler.h"
9ff7671d 38#include "AliESDInputHandler.h"
7f7091b8 39#include "AliMCEvent.h"
691685d6 40#include "AliStack.h"
1d27ecd2 41
e5b96aa0 42
43ClassImp(AliAnalysisTaskJets)
44
45////////////////////////////////////////////////////////////////////////
46
19e6695b 47AliAnalysisTaskJets::AliAnalysisTaskJets():
8838ab7a 48 AliAnalysisTaskSE(),
49 fConfigFile("ConfigJetAnalysis.C"),
50 fNonStdBranch(""),
931c23a1 51 fNonStdFile(""),
4aa71b6f 52 fJetFinder(0x0),
53 fHistos(0x0),
931c23a1 54 fAODExtension(0x0),
8838ab7a 55 fListOfHistos(0x0),
56 fChain(0x0),
08c28025 57 fOpt(0),
58 fReadAODFromOutput(0)
19e6695b 59{
60 // Default constructor
61}
62
e5b96aa0 63AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name):
8838ab7a 64 AliAnalysisTaskSE(name),
65 fConfigFile("ConfigJetAnalysis.C"),
66 fNonStdBranch(""),
931c23a1 67 fNonStdFile(""),
8838ab7a 68 fJetFinder(0x0),
69 fHistos(0x0),
931c23a1 70 fAODExtension(0x0),
8838ab7a 71 fListOfHistos(0x0),
72 fChain(0x0),
0f4a88df 73 fOpt(0),
74 fReadAODFromOutput(0)
e5b96aa0 75{
76 // Default constructor
8838ab7a 77 DefineOutput(1, TList::Class());
eaabc21f 78}
79
8838ab7a 80AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name, TChain* chain):
81 AliAnalysisTaskSE(name),
82 fConfigFile("ConfigJetAnalysis.C"),
83 fNonStdBranch(""),
931c23a1 84 fNonStdFile(""),
8838ab7a 85 fJetFinder(0x0),
86 fHistos(0x0),
931c23a1 87 fAODExtension(0x0),
8838ab7a 88 fListOfHistos(0x0),
89 fChain(chain),
0f4a88df 90 fOpt(0),
91 fReadAODFromOutput(0)
eaabc21f 92{
8838ab7a 93 // Default constructor
94 DefineOutput(1, TList::Class());
95}
4aa71b6f 96
8838ab7a 97//----------------------------------------------------------------
98void AliAnalysisTaskJets::UserCreateOutputObjects()
99{
100 // Create the output container
101 //
102 if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
931c23a1 103
8838ab7a 104 if(fNonStdBranch.Length()==0)
105 {
4aa71b6f 106 // Connec default AOD to jet finder
bcec7a80 107 // create a new branch for the background
108 if(!AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName())){
109 AliAODJetEventBackground* evBkg = new AliAODJetEventBackground();
110 evBkg->SetName(AliAODJetEventBackground::StdBranchName());
111 AddAODBranch("AliAODJetEventBackground",&evBkg);
112 }
4aa71b6f 113 fJetFinder->ConnectAOD(AODEvent());
114 }
8838ab7a 115 else
116 {
4aa71b6f 117 // Create a new branch for jets...
8838ab7a 118 // how is this reset? -> cleared in the UserExec....
4aa71b6f 119 // Can this be handled by the framework?
931c23a1 120 // here we can also have the case that the brnaches are written to a separate file
121
4aa71b6f 122 TClonesArray *tca = new TClonesArray("AliAODJet", 0);
931c23a1 123 tca->SetName(fNonStdBranch.Data());
124 AddAODBranch("TClonesArray",&tca,fNonStdFile.Data());
bcec7a80 125 if(!AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()))){
126 AliAODJetEventBackground* evBkg = new AliAODJetEventBackground();
127 evBkg->SetName(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()));
931c23a1 128 AddAODBranch("AliAODJetEventBackground",&evBkg,fNonStdFile.Data());
129 }
130
131 if(fNonStdFile.Length()!=0){
132 //
133 // case that we have an AOD extension we need to fetch the jets from the extended output
134 // we identifay the extension aod event by looking for the branchname
135 AliAODHandler *aodH = dynamic_cast<AliAODHandler*>(AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler());
136 TObjArray* extArray = aodH->GetExtensions();
137 if (extArray) {
138 TIter next(extArray);
139 while ((fAODExtension=(AliAODExtension*)next())){
140 TObject *obj = fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data());
8f9a4721 141 if(fDebug>10){
142 Printf("%s:%d Dumping..",(char*)__FILE__,__LINE__);
143 fAODExtension->GetAOD()->Dump();
144 }
931c23a1 145 if(obj){
146 if(fDebug>1)Printf("AODExtension found for %s",fNonStdBranch.Data());
147 break;
148 }
149 fAODExtension = 0;
150 }
151 }
152 if(fAODExtension)fJetFinder->ConnectAODNonStd(fAODExtension->GetAOD(), fNonStdBranch.Data());
153 }
154 else{
155 fJetFinder->ConnectAODNonStd(AODEvent(), fNonStdBranch.Data());
bcec7a80 156 }
8838ab7a 157 }
bcec7a80 158
952b368c 159
8838ab7a 160 // Histograms
161 OpenFile(1);
162 fListOfHistos = new TList();
163 fHistos = new AliJetHistos();
164 fHistos->AddHistosToList(fListOfHistos);
165
166 // Add the JetFinderInformation to the Outputlist
167 AliJetHeader *fH = fJetFinder->GetHeader();
168
169 // Compose a characteristic output name
170 // with the name of the output branch
171 if(fH) {
172 if(fNonStdBranch.Length()==0) {
173 fH->SetName("AliJetHeader_jets");
4aa71b6f 174 }
8838ab7a 175 else {
176 fH->SetName(Form("AliJetHeader_%s",fNonStdBranch.Data()));
99c8d2ad 177 }
8838ab7a 178 }
931c23a1 179 if(!fAODExtension)OutputTree()->GetUserInfo()->Add(fH);
180 else fAODExtension->GetTree()->GetUserInfo()->Add(fH);
e5b96aa0 181}
182
8838ab7a 183//----------------------------------------------------------------
19e6695b 184void AliAnalysisTaskJets::Init()
e5b96aa0 185{
8838ab7a 186 // Initialization
187 if (fDebug > 1) printf("AnalysisTaskJets::Init() \n");
188
189 // Call configuration file
190 if (fConfigFile.Length()) {
191 gROOT->LoadMacro(fConfigFile);
192 fJetFinder = (AliJetFinder*) gInterpreter->ProcessLine("ConfigJetAnalysis()");
193 }
194 AliJetReaderHeader *header = (AliJetReaderHeader*)fJetFinder->GetReader()->GetReaderHeader();
195 fOpt = header->GetDetector();
196
197 // Initialise Jet Analysis
198 if(fOpt == 0) fJetFinder->Init();
199 else fJetFinder->InitTask(fChain); // V2
19e6695b 200}
201
931c23a1 202
8838ab7a 203//----------------------------------------------------------------
ae24a5a1 204void AliAnalysisTaskJets::UserExec(Option_t */*option*/)
e5b96aa0 205{
4aa71b6f 206 // Execute analysis for current event
207 //
4aa71b6f 208 // Fill control histos
209 TClonesArray* jarray = 0;
bcec7a80 210 AliAODJetEventBackground* evBkg;
8838ab7a 211
212 if(fNonStdBranch.Length()==0) {
4aa71b6f 213 jarray = AODEvent()->GetJets();
bcec7a80 214 evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName()));
215 evBkg->Reset();
4aa71b6f 216 }
8838ab7a 217 else {
931c23a1 218 jarray = (TClonesArray*)(AODEvent()->FindListObject(fNonStdBranch.Data()));
219 if(!jarray)jarray = (TClonesArray*)(fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data()));
4aa71b6f 220 jarray->Delete(); // this is our responsibility, clear before filling again
bcec7a80 221 evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
931c23a1 222 if(!evBkg) evBkg = (AliAODJetEventBackground*)(fAODExtension->GetAOD()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
bcec7a80 223 evBkg->Reset();
4aa71b6f 224 }
931c23a1 225
0f4a88df 226 if (dynamic_cast<AliAODEvent*>(InputEvent()) != 0 && !fReadAODFromOutput) {
227// AOD is input event..........................................V
c98859dd 228 fJetFinder->GetReader()->SetInputEvent(InputEvent(), InputEvent(), MCEvent());
229 } else {
0f4a88df 230// AOD is read from output ....................................V
c98859dd 231 fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent());
232 }
233
234
4aa71b6f 235
8838ab7a 236 if(fOpt==0) fJetFinder->ProcessEvent();
237 else fJetFinder->ProcessEvent2(); // V2
238
239 // Fill control histos
4aa71b6f 240 fHistos->FillHistos(jarray);
8838ab7a 241
4aa71b6f 242 // Post the data
243 PostData(1, fListOfHistos);
931c23a1 244 return;
e5b96aa0 245}
246
8838ab7a 247
248//*************************************************************
249
e5b96aa0 250void AliAnalysisTaskJets::Terminate(Option_t */*option*/)
251{
252// Terminate analysis
253//
19e6695b 254 if (fDebug > 1) printf("AnalysisJets: Terminate() \n");
9ff7671d 255// if (fJetFinder) fJetFinder->FinishRun();
e5b96aa0 256}
257