4422d8f709897302f5a19b6d99c64fd3cdd8521b
[u/mrichter/AliRoot.git] / STEER / AOD / AliAODInputHandler.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-2007, 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
18 //-------------------------------------------------------------------------
19 //     Event handler for AOD input 
20 //     Author: Andreas Morsch, CERN
21 //-------------------------------------------------------------------------
22
23 #include <TSystem.h>
24 #include <TTree.h>
25 #include <TList.h>
26 #include <TNamed.h>
27 #include <TFile.h>
28 #include <TH2.h>
29
30 #include "AliAODInputHandler.h"
31 #include "AliAODEvent.h"
32 #include "AliVCuts.h"
33 #include "AliMCEvent.h"
34 #include "AliAODpidUtil.h"
35
36 ClassImp(AliAODInputHandler)
37
38 static Option_t *gAODDataType = "AOD";
39
40 //______________________________________________________________________________
41 AliAODInputHandler::AliAODInputHandler() :
42     AliInputEventHandler(),
43     fEvent(0),
44     fMCEvent(0),
45     fFriends(0),
46     fAODpidUtil(0x0),
47     fMergeEvents(kFALSE),
48     fMergeTracks(kTRUE), 
49     fMergeEMCALClusters(kTRUE), 
50     fMergePHOSClusters(kTRUE), 
51     fMergeEMCALCells(kTRUE), 
52     fMergePHOSCells(kTRUE),
53     fFriendsConnected(kFALSE),
54     fFileToMerge(0),
55     fTreeToMerge(0),
56     fAODEventToMerge(0),
57     fMergeOffset(0)
58 {
59   // Default constructor
60   fHistStatistics[0] = fHistStatistics[1] = NULL;
61 }
62
63 //______________________________________________________________________________
64 AliAODInputHandler::AliAODInputHandler(const char* name, const char* title):
65   AliInputEventHandler(name, title),
66   fEvent(0),
67   fMCEvent(new AliMCEvent()),
68   fFriends(new TList()),
69   fAODpidUtil(0x0),
70   fMergeEvents(kFALSE),
71   fMergeTracks(kTRUE), 
72   fMergeEMCALClusters(kTRUE), 
73   fMergePHOSClusters(kTRUE), 
74   fMergeEMCALCells(kTRUE), 
75   fMergePHOSCells(kTRUE),
76   fFriendsConnected(kFALSE),
77   fFileToMerge(0),
78   fTreeToMerge(0),
79   fAODEventToMerge(0),
80   fMergeOffset(0)
81 {
82     // Constructor
83   fHistStatistics[0] = fHistStatistics[1] = NULL;
84 }
85
86 //______________________________________________________________________________
87 AliAODInputHandler::~AliAODInputHandler() 
88 {
89 // Destructor
90   fFriends->Delete();
91   delete fHistStatistics[0];
92   delete fHistStatistics[1];
93   delete fAODpidUtil;
94 }
95
96 //______________________________________________________________________________
97 Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* opt)
98 {
99     // Initialisation necessary for each new tree
100     fTree = tree;
101     if (!fTree) return kFALSE;
102     fTree->GetEntries();
103     ConnectFriends();
104
105     SwitchOffBranches();
106     SwitchOnBranches();
107     
108     // Get pointer to AOD event
109     if (!fEvent) fEvent = new AliAODEvent();
110     
111     fEvent->ReadFromTree(fTree);
112     
113     if (fMixingHandler) fMixingHandler->Init(tree, opt);
114     
115     return kTRUE;
116 }
117
118 //______________________________________________________________________________
119 Bool_t AliAODInputHandler::BeginEvent(Long64_t entry)
120 {
121     // Begin event
122     TClonesArray* mcParticles = (TClonesArray*) (fEvent->FindListObject("mcparticles"));
123     if (mcParticles) fMCEvent->SetParticleArray(mcParticles);
124
125     // When merging, get current event number from GetReadEntry(), 
126     // entry gives the events in the current file
127     if (fTreeToMerge) fTreeToMerge->GetEntry(GetReadEntry() + fMergeOffset);
128   
129     fIsSelectedResult = fEvent->GetHeader()->GetOfflineTrigger();
130
131     if (fMixingHandler) fMixingHandler->BeginEvent(entry);
132     
133     return kTRUE;
134 }
135
136 //______________________________________________________________________________
137 Bool_t AliAODInputHandler::Notify(const char* path)
138 {
139   // Notifaction of directory change
140   if (fMixingHandler) fMixingHandler->Notify(path);
141   if (!fFriendsConnected) {
142       ConnectFriends();
143       fEvent->ReadFromTree(fTree, "reconnect");
144   }
145   fFriendsConnected = kFALSE;
146     
147   TTree *ttree = fTree->GetTree();
148   if (!ttree) ttree = fTree;
149   TString statFname(ttree->GetCurrentFile()->GetName());
150   AliInfo(Form("Moving to file %s", statFname.Data()));
151   Int_t indarchive = statFname.Index("#");
152   if (indarchive<0) {
153      statFname = gSystem->DirName(statFname);
154      statFname += "/";
155   } else {
156      statFname.Remove(indarchive+1);
157   }   
158   statFname += "EventStat_temp.root";
159   TFile *statFile = 0;
160   if (IsCheckStatistics()) statFile = TFile::Open(statFname, "READ");
161   if (statFile) {
162      TList *list = (TList*)statFile->Get("cstatsout");
163      if (list) {
164         AliVCuts *physSel = (AliVCuts*)list->At(0);
165         if (physSel) {
166            TH2F *hAll = dynamic_cast<TH2F*>(physSel->GetStatistics("ALL"));
167            TH2F *hBin0 = dynamic_cast<TH2F*>(physSel->GetStatistics("BIN0"));
168            if (fHistStatistics[0] && hAll) {
169               TList tmplist;
170               tmplist.Add(hAll);
171               fHistStatistics[0]->Merge(&tmplist);
172               tmplist.Clear();
173               tmplist.Add(hBin0);
174               if (fHistStatistics[1] && hBin0) fHistStatistics[1]->Merge(&tmplist);
175            } else {
176              if (hAll && hBin0) {
177                 fHistStatistics[0] = static_cast<TH2F*>(hAll->Clone());
178                 fHistStatistics[1] = static_cast<TH2F*>(hBin0->Clone());
179                 fHistStatistics[0]->SetDirectory(0);
180                 fHistStatistics[1]->SetDirectory(0);
181              }   
182            }   
183         }
184         delete list;
185      }
186      delete statFile;
187   }
188   return kTRUE;
189 }
190
191 //______________________________________________________________________________
192 Bool_t AliAODInputHandler::FinishEvent()
193 {
194   // Finish event
195   if (fMixingHandler) fMixingHandler->FinishEvent();
196   if (fEvent) fEvent->Reset();
197   return kTRUE;
198 }
199
200 //______________________________________________________________________________
201 void AliAODInputHandler::AddFriend(char* filename)
202 {
203     // Add a friend tree 
204     TNamed* obj = new TNamed(filename, filename);
205     fFriends->Add(obj);
206 }
207
208 //______________________________________________________________________________
209 Option_t *AliAODInputHandler::GetDataType() const
210 {
211 // Returns handled data type.
212    return gAODDataType;
213 }
214
215 //______________________________________________________________________________
216 TObject *AliAODInputHandler::GetStatistics(Option_t *option) const
217 {
218 // Get the statistics histogram(s) from the physics selection object. This
219 // should be called during FinishTaskOutput(). Option can be empty (default
220 // statistics histogram) or BIN0.
221    TString opt(option);
222    opt.ToUpper();
223    if (opt=="BIN0") return fHistStatistics[1];
224    return fHistStatistics[0];
225 }   
226
227 void AliAODInputHandler::ConnectFriends()
228 {
229     // Connect the friend trees 
230     if (!fMergeEvents) {
231         TIter next(fFriends);
232         TNamed* obj;
233         TString aodTreeFName,aodFriendTreeFName;
234         TTree *ttree = fTree->GetTree();
235         if (!ttree) ttree = fTree;
236         aodTreeFName = ttree->GetCurrentFile()->GetName();
237         
238         while((obj = (TNamed*)next())) {
239             aodFriendTreeFName = aodTreeFName;
240             aodFriendTreeFName.ReplaceAll("AliAOD.root",obj->GetName());
241             aodFriendTreeFName.ReplaceAll("AliAODs.root",obj->GetName());
242             ttree->AddFriend("aodTree", aodFriendTreeFName.Data());
243         }
244     } else {
245         // Friends have to be merged
246         TNamed* filename = (TNamed*) (fFriends->At(0));
247         fFileToMerge = TFile::Open(filename->GetName());
248         if (fFileToMerge) {
249             fFileToMerge->GetObject("aodTree", fTreeToMerge);
250             if (!fAODEventToMerge) fAODEventToMerge = new AliAODEvent();
251             fAODEventToMerge->ReadFromTree(fTreeToMerge);
252         }
253     }
254     fFriendsConnected = kTRUE;
255 }
256
257 //______________________________________________________________________________
258 void AliAODInputHandler::CreatePIDResponse(Bool_t isMC/*=kFALSE*/)
259 {
260   //
261   // create the pid response object if it does not exist yet
262   //
263   if (fAODpidUtil) return;
264   fAODpidUtil=new AliAODpidUtil(isMC);
265   
266 }
267