Coding violations corrected.
[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJets.cxx
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
16 /* $Id$ */
17 #include <Riostream.h> 
18 #include <TROOT.h>
19 #include <TSystem.h>
20 #include <TInterpreter.h>
21 #include <TChain.h>
22 #include <TFile.h>
23 #include <TList.h>
24 #include <TH1.h>
25
26 #include "AliAnalysisTaskJets.h"
27 #include "AliAnalysisManager.h"
28 #include "AliJetFinder.h"
29 #include "AliJetHeader.h"
30 #include "AliJetReader.h"
31 #include "AliJetReaderHeader.h"
32 #include "AliJetHistos.h"
33 #include "AliESDEvent.h"
34 #include "AliESD.h"
35 #include "AliAODEvent.h"
36 #include "AliAODJetEventBackground.h"
37 #include "AliAODHandler.h"
38 #include "AliMCEventHandler.h"
39 #include "AliESDInputHandler.h"
40 #include "AliMCEvent.h"
41 #include "AliStack.h"
42
43
44 ClassImp(AliAnalysisTaskJets)
45
46 ////////////////////////////////////////////////////////////////////////
47
48 AliAnalysisTaskJets::AliAnalysisTaskJets():
49   AliAnalysisTaskSE(),
50   fConfigFile("ConfigJetAnalysis.C"),
51   fNonStdBranch(""), 
52   fNonStdFile(""),
53   fJetFinder(0x0),
54   fHistos(0x0),
55   fAODExtension(0x0),
56   fListOfHistos(0x0),
57   fChain(0x0),
58   fOpt(0),
59   fReadAODFromOutput(0)
60 {
61   // Default constructor
62 }
63
64 AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name):
65   AliAnalysisTaskSE(name),
66   fConfigFile("ConfigJetAnalysis.C"),
67   fNonStdBranch(""),
68   fNonStdFile(""),
69   fJetFinder(0x0),
70   fHistos(0x0),
71   fAODExtension(0x0),
72   fListOfHistos(0x0),
73   fChain(0x0),
74   fOpt(0),
75   fReadAODFromOutput(0)
76 {
77   // Default constructor
78   DefineOutput(1, TList::Class());
79 }
80
81 AliAnalysisTaskJets::AliAnalysisTaskJets(const char* name, TChain* chain):
82   AliAnalysisTaskSE(name),
83   fConfigFile("ConfigJetAnalysis.C"),
84   fNonStdBranch(""),
85   fNonStdFile(""),
86   fJetFinder(0x0),
87   fHistos(0x0),
88   fAODExtension(0x0),
89   fListOfHistos(0x0),
90   fChain(chain),
91   fOpt(0),
92   fReadAODFromOutput(0)
93 {
94   // Default constructor
95   DefineOutput(1, TList::Class());
96 }
97
98 //----------------------------------------------------------------
99 void AliAnalysisTaskJets::UserCreateOutputObjects()
100 {
101   // Create the output container
102   //
103   if (fDebug > 1) printf("AnalysisTaskJets::CreateOutPutData() \n");
104
105   if(fNonStdBranch.Length()==0)
106     {
107       //  Connec default AOD to jet finder
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       }
114       fJetFinder->ConnectAOD(AODEvent());
115     }
116   else
117     {
118       // Create a new branch for jets...
119       // how is this reset? -> cleared in the UserExec....
120       // Can this be handled by the framework?
121       // here we can also have the case that the brnaches are written to a separate file
122
123       TClonesArray *tca = new TClonesArray("AliAODJet", 0);
124       tca->SetName(fNonStdBranch.Data());
125       AddAODBranch("TClonesArray",&tca,fNonStdFile.Data());
126       if(!AODEvent() || !AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()))){
127         AliAODJetEventBackground* evBkg = new AliAODJetEventBackground();
128         evBkg->SetName(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data()));
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());
142             if(fDebug>10){
143               Printf("%s:%d Dumping..",(char*)__FILE__,__LINE__);
144               fAODExtension->GetAOD()->Dump();
145             }
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         if (fDebug > 1) printf("Connecting Non Std Branch AOD %p %s \n",AODEvent(),fNonStdBranch.Data());
157         fJetFinder->ConnectAODNonStd(AODEvent(), fNonStdBranch.Data()); 
158       }
159     }
160
161   // do not add the histograms in the directory
162   // all handled by the list
163   Bool_t oldStatus = TH1::AddDirectoryStatus();
164   TH1::AddDirectory(kFALSE);
165
166
167   // Histograms
168   fListOfHistos = new TList();
169   fHistos       = new AliJetHistos();
170   fHistos->AddHistosToList(fListOfHistos);
171   
172   // Add the JetFinderInformation to the Outputlist
173   AliJetHeader *fH = fJetFinder->GetHeader();
174   
175   // Compose a characteristic output name
176   // with the name of the output branch
177   if(fH) {
178     if(fNonStdBranch.Length()==0) {
179       fH->SetName("AliJetHeader_jets");
180     }
181     else {
182       fH->SetName(Form("AliJetHeader_%s",fNonStdBranch.Data()));
183     }
184   }
185
186   TH1::AddDirectory(oldStatus);
187   
188   
189   if(!fAODExtension)OutputTree()->GetUserInfo()->Add(fH);
190   else fAODExtension->GetTree()->GetUserInfo()->Add(fH);
191 }
192
193 //----------------------------------------------------------------
194 void AliAnalysisTaskJets::Init()
195 {
196   // Initialization
197   if (fDebug > 1) printf("AnalysisTaskJets::Init() \n");
198   
199   // Call configuration file
200   if (fConfigFile.Length()) {
201     gROOT->LoadMacro(fConfigFile);
202     fJetFinder = (AliJetFinder*) gInterpreter->ProcessLine("ConfigJetAnalysis()");
203   }
204   AliJetReaderHeader *header = (AliJetReaderHeader*)fJetFinder->GetReader()->GetReaderHeader();
205   fOpt = header->GetDetector();
206
207   // Initialise Jet Analysis
208   if(fOpt == 0) fJetFinder->Init();
209   else fJetFinder->InitTask(fChain); // V2
210 }
211
212
213 //----------------------------------------------------------------
214 void AliAnalysisTaskJets::UserExec(Option_t */*option*/)
215 {
216   // Execute analysis for current event
217   //
218   // Fill control histos
219   TClonesArray* jarray = 0;
220   AliAODJetEventBackground*  evBkg = 0;
221
222   if(fNonStdBranch.Length()==0) {
223     jarray = AODEvent()->GetJets();
224     evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(AliAODJetEventBackground::StdBranchName()));
225     evBkg->Reset();
226   }
227   else {
228     if(AODEvent())jarray = (TClonesArray*)(AODEvent()->FindListObject(fNonStdBranch.Data()));
229     if(!jarray)jarray = (TClonesArray*)(fAODExtension->GetAOD()->FindListObject(fNonStdBranch.Data()));
230     if(jarray)jarray->Delete();    // this is our responsibility, clear before filling again
231     if(AODEvent())evBkg = (AliAODJetEventBackground*)(AODEvent()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
232     if(!evBkg)  evBkg = (AliAODJetEventBackground*)(fAODExtension->GetAOD()->FindListObject(Form("%s_%s",AliAODJetEventBackground::StdBranchName(),fNonStdBranch.Data())));
233     if(evBkg)evBkg->Reset();
234   }
235
236   if (dynamic_cast<AliAODEvent*>(InputEvent()) !=  0 && !fReadAODFromOutput) {
237 // AOD is input event..........................................V                                       
238       fJetFinder->GetReader()->SetInputEvent(InputEvent(), InputEvent(), MCEvent());
239   } else {
240 // AOD is read from output ....................................V      
241       fJetFinder->GetReader()->SetInputEvent(InputEvent(), AODEvent(), MCEvent());
242   }
243   
244   
245
246   if(fOpt==0) fJetFinder->ProcessEvent();
247   else fJetFinder->ProcessEvent2();    // V2
248  
249   // Fill control histos
250   if(jarray)fHistos->FillHistos(jarray);
251
252   // Post the data
253   PostData(1, fListOfHistos);
254   return;
255 }
256
257
258 //*************************************************************
259
260 void AliAnalysisTaskJets::Terminate(Option_t */*option*/)
261 {
262 // Terminate analysis
263 //
264     if (fDebug > 1) printf("AnalysisJets: Terminate() \n");
265 //    if (fJetFinder) fJetFinder->FinishRun();
266 }
267