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