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 **************************************************************************/
17 #include <Riostream.h>
20 #include <TInterpreter.h>
25 #include "AliAnalysisTaskJets.h"
26 #include "AliAnalysisManager.h"
27 #include "AliJetFinder.h"
28 #include "AliJetHeader.h"
29 #include "AliJetReader.h"
30 #include "AliJetReaderHeader.h"
31 #include "AliJetHistos.h"
32 #include "AliESDEvent.h"
34 #include "AliAODEvent.h"
35 #include "AliAODJetEventBackground.h"
36 #include "AliAODHandler.h"
37 #include "AliMCEventHandler.h"
38 #include "AliESDInputHandler.h"
39 #include "AliMCEvent.h"
43 ClassImp(AliAnalysisTaskJets)
45 ////////////////////////////////////////////////////////////////////////
47 AliAnalysisTaskJets::AliAnalysisTaskJets():
49 fConfigFile("ConfigJetAnalysis.C"),
59 // Default constructor
62 AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name):
63 AliAnalysisTaskSE(name),
64 fConfigFile("ConfigJetAnalysis.C"),
74 // Default constructor
75 DefineOutput(1, TList::Class());
78 AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name, TChain* chain):
79 AliAnalysisTaskSE(name),
80 fConfigFile("ConfigJetAnalysis.C"),
90 // Default constructor
91 DefineOutput(1, TList::Class());
94 //----------------------------------------------------------------
95 void AliAnalysisTaskJets::UserCreateOutputObjects()
97 // Create the output container
99 if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
101 // we already have a non standard output, need not to book a new file...
102 if (!IsStandardAOD())fNonStdFile = "";
104 if(fNonStdBranch.Length()==0)
106 // Connec default AOD to jet finder
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);
113 fJetFinder->ConnectAOD(AODEvent());
117 // Create a new branch for jets...
118 // how is this reset? -> cleared in the UserExec....
119 // Can this be handled by the framework?
120 // here we can also have the case that the brnaches are written to a separate file
122 TClonesArray *tca = new TClonesArray("AliAODJet", 0);
123 tca->SetName(fNonStdBranch.Data());
124 AddAODBranch("TClonesArray",&tca,fNonStdFile.Data());
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()));
128 AddAODBranch("AliAODJetEventBackground",&evBkg,fNonStdFile.Data());
131 if(fNonStdFile.Length()!=0){
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();
138 TIter next(extArray);
139 while ((fAODExtension=(AliAODExtension*)next())){
140 TObject *obj = fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data());
141 fAODExtension->GetAOD()->Dump();
143 if(fDebug>1)Printf("AODExtension found for %s",fNonStdBranch.Data());
149 if(fAODExtension)fJetFinder->ConnectAODNonStd(fAODExtension->GetAOD(), fNonStdBranch.Data());
152 fJetFinder->ConnectAODNonStd(AODEvent(), fNonStdBranch.Data());
159 fListOfHistos = new TList();
160 fHistos = new AliJetHistos();
161 fHistos->AddHistosToList(fListOfHistos);
163 // Add the JetFinderInformation to the Outputlist
164 AliJetHeader *fH = fJetFinder->GetHeader();
166 // Compose a characteristic output name
167 // with the name of the output branch
169 if(fNonStdBranch.Length()==0) {
170 fH->SetName("AliJetHeader_jets");
173 fH->SetName(Form("AliJetHeader_%s",fNonStdBranch.Data()));
176 if(!fAODExtension)OutputTree()->GetUserInfo()->Add(fH);
177 else fAODExtension->GetTree()->GetUserInfo()->Add(fH);
180 //----------------------------------------------------------------
181 void AliAnalysisTaskJets::Init()
184 if (fDebug > 1) printf("AnalysisTaskJets::Init() \n");
186 // Call configuration file
187 if (fConfigFile.Length()) {
188 gROOT->LoadMacro(fConfigFile);
189 fJetFinder = (AliJetFinder*) gInterpreter->ProcessLine("ConfigJetAnalysis()");
191 AliJetReaderHeader *header = (AliJetReaderHeader*)fJetFinder->GetReader()->GetReaderHeader();
192 fOpt = header->GetDetector();
194 // Initialise Jet Analysis
195 if(fOpt == 0) fJetFinder->Init();
196 else fJetFinder->InitTask(fChain); // V2
198 // Write header information to local file
199 fJetFinder->WriteHeaders();
203 //----------------------------------------------------------------
204 void AliAnalysisTaskJets::UserExec(Option_t */*option*/)
206 // Execute analysis for current event
208 // Fill control histos
209 TClonesArray* jarray = 0;
210 AliAODJetEventBackground* evBkg;
212 if(fNonStdBranch.Length()==0) {
213 jarray = AODEvent()->GetJets();
214 evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName()));
218 jarray = (TClonesArray*)(AODEvent()->FindListObject(fNonStdBranch.Data()));
219 if(!jarray)jarray = (TClonesArray*)(fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data()));
220 jarray->Delete(); // this is our responsibility, clear before filling again
221 evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
222 if(!evBkg) evBkg = (AliAODJetEventBackground*)(fAODExtension->GetAOD()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
226 if (dynamic_cast<AliAODEvent*>(InputEvent()) != 0) {
227 fJetFinder->GetReader()->SetInputEvent(InputEvent(), InputEvent(), MCEvent());
229 fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent());
234 if(fOpt==0) fJetFinder->ProcessEvent();
235 else fJetFinder->ProcessEvent2(); // V2
237 // Fill control histos
238 fHistos->FillHistos(jarray);
241 PostData(1, fListOfHistos);
246 //*************************************************************
248 void AliAnalysisTaskJets::Terminate(Option_t */*option*/)
250 // Terminate analysis
252 if (fDebug > 1) printf("AnalysisJets: Terminate() \n");
253 // if (fJetFinder) fJetFinder->FinishRun();