1 /**************************************************************************
2 * Copyright(c) 1998-2007, 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 **************************************************************************/
18 //-------------------------------------------------------------------------
19 // Event handler for AOD input
20 // Author: Andreas Morsch, CERN
21 //-------------------------------------------------------------------------
30 #include "AliAODInputHandler.h"
31 #include "AliAODEvent.h"
33 #include "AliMCEvent.h"
34 #include "AliAODpidUtil.h"
35 #include "AliAODMCHeader.h"
37 ClassImp(AliAODInputHandler)
39 static Option_t *gAODDataType = "AOD";
41 //______________________________________________________________________________
42 AliAODInputHandler::AliAODInputHandler() :
43 AliInputEventHandler(),
50 fMergeEMCALClusters(kTRUE),
51 fMergePHOSClusters(kTRUE),
52 fMergeEMCALCells(kTRUE),
53 fMergePHOSCells(kTRUE),
54 fMergeEMCALTrigger(kTRUE),
55 fMergePHOSTrigger(kTRUE),
56 fMergeHMPIDrings(kTRUE),
57 fFriendsConnected(kFALSE),
63 // Default constructor
64 fHistStatistics[0] = fHistStatistics[1] = NULL;
67 //______________________________________________________________________________
68 AliAODInputHandler::AliAODInputHandler(const char* name, const char* title):
69 AliInputEventHandler(name, title),
71 fMCEvent(new AliMCEvent()),
72 fFriends(new TList()),
76 fMergeEMCALClusters(kTRUE),
77 fMergePHOSClusters(kTRUE),
78 fMergeEMCALCells(kTRUE),
79 fMergePHOSCells(kTRUE),
80 fMergeEMCALTrigger(kTRUE),
81 fMergePHOSTrigger(kTRUE),
82 fMergeHMPIDrings(kTRUE),
83 fFriendsConnected(kFALSE),
90 fHistStatistics[0] = fHistStatistics[1] = NULL;
93 //______________________________________________________________________________
94 AliAODInputHandler::~AliAODInputHandler()
97 if (fFriends) fFriends->Delete();
99 delete fHistStatistics[0];
100 delete fHistStatistics[1];
104 //______________________________________________________________________________
105 Bool_t AliAODInputHandler::Init(TTree* tree, Option_t* opt)
107 // Initialisation necessary for each new tree
109 if (!fTree) return kFALSE;
116 // Get pointer to AOD event
117 if (!fEvent) fEvent = new AliAODEvent();
119 fEvent->ReadFromTree(fTree);
121 if (fMixingHandler) fMixingHandler->Init(tree, opt);
126 //______________________________________________________________________________
127 Bool_t AliAODInputHandler::BeginEvent(Long64_t entry)
130 static Int_t prevRunNumber = -1;
131 if (prevRunNumber != fEvent->GetRunNumber() && NeedField()) {
132 fEvent->InitMagneticField();
133 prevRunNumber = fEvent->GetRunNumber();
136 AliAODMCHeader* mcHeader = (AliAODMCHeader*) fEvent->GetList()->FindObject(AliAODMCHeader::StdBranchName());
138 fMCEvent->SetExternalHeader(mcHeader);}
140 TClonesArray* mcParticles = (TClonesArray*) (fEvent->FindListObject("mcparticles"));
142 if (!fMCEvent) fMCEvent = new AliMCEvent();
143 fMCEvent->SetParticleArray(mcParticles);
146 // When merging, get current event number from GetReadEntry(),
147 // entry gives the events in the current file
148 if (fTreeToMerge) fTreeToMerge->GetEntry(GetReadEntry() + fMergeOffset);
151 fIsSelectedResult = fEventCuts->GetSelectionMask(fEvent);
153 fIsSelectedResult = fEvent->GetHeader()->GetOfflineTrigger();
155 if (fMixingHandler) fMixingHandler->BeginEvent(entry);
157 // set transient pointer to event inside tracks
158 fEvent->ConnectTracks();
163 //______________________________________________________________________________
164 Bool_t AliAODInputHandler::Notify(const char* path)
166 // Notifaction of directory change
167 if (fMixingHandler) fMixingHandler->Notify(path);
168 if (!fFriendsConnected) {
170 fEvent->ReadFromTree(fTree, "reconnect");
172 fFriendsConnected = kFALSE;
173 fUserInfo=fTree->GetTree()->GetUserInfo();
175 TTree *ttree = fTree->GetTree();
176 if (!ttree) ttree = fTree;
177 TString statFname(ttree->GetCurrentFile()->GetName());
178 AliInfo(Form("Moving to file %s", statFname.Data()));
179 Int_t indarchive = statFname.Index("#");
181 statFname = gSystem->DirName(statFname);
184 statFname.Remove(indarchive+1);
186 statFname += "EventStat_temp.root";
188 if (IsCheckStatistics()) statFile = TFile::Open(statFname, "READ");
190 TList *list = (TList*)statFile->Get("cstatsout");
192 AliVCuts *physSel = (AliVCuts*)list->At(0);
194 TH2F *hAll = dynamic_cast<TH2F*>(physSel->GetStatistics("ALL"));
195 TH2F *hBin0 = dynamic_cast<TH2F*>(physSel->GetStatistics("BIN0"));
196 if (fHistStatistics[0] && hAll) {
199 fHistStatistics[0]->Merge(&tmplist);
202 if (fHistStatistics[1] && hBin0) fHistStatistics[1]->Merge(&tmplist);
205 fHistStatistics[0] = static_cast<TH2F*>(hAll->Clone());
206 fHistStatistics[1] = static_cast<TH2F*>(hBin0->Clone());
207 fHistStatistics[0]->SetDirectory(0);
208 fHistStatistics[1]->SetDirectory(0);
219 //______________________________________________________________________________
220 Bool_t AliAODInputHandler::FinishEvent()
223 if (fMixingHandler) fMixingHandler->FinishEvent();
224 if (fEvent) fEvent->Reset();
228 //______________________________________________________________________________
229 void AliAODInputHandler::AddFriend(char* filename)
232 TNamed* obj = new TNamed(filename, filename);
233 if (!fFriends) fFriends = new TList();
237 //______________________________________________________________________________
238 Option_t *AliAODInputHandler::GetDataType() const
240 // Returns handled data type.
244 //______________________________________________________________________________
245 TObject *AliAODInputHandler::GetStatistics(Option_t *option) const
247 // Get the statistics histogram(s) from the physics selection object. This
248 // should be called during FinishTaskOutput(). Option can be empty (default
249 // statistics histogram) or BIN0.
252 if (opt=="BIN0") return fHistStatistics[1];
253 return fHistStatistics[0];
256 void AliAODInputHandler::ConnectFriends()
258 // Connect the friend trees
259 if (!fFriends) return;
261 TIter next(fFriends);
263 TString aodTreeFName,aodFriendTreeFName;
264 TTree *ttree = fTree->GetTree();
265 if (!ttree) ttree = fTree;
266 if(!ttree->GetCurrentFile()){
267 AliWarning("Couldn't get current AOD file, not connecting friends");
270 aodTreeFName = ttree->GetCurrentFile()->GetName();
272 while((obj = (TNamed*)next())) {
273 aodFriendTreeFName = aodTreeFName;
274 if (strlen(GetInputFileName())) aodFriendTreeFName.ReplaceAll(GetInputFileName(),obj->GetName());
275 aodFriendTreeFName.ReplaceAll("AliAOD.root",obj->GetName());
276 aodFriendTreeFName.ReplaceAll("AliAODs.root",obj->GetName());
277 ttree->AddFriend("aodTree", aodFriendTreeFName.Data());
280 // Friends have to be merged
281 TNamed* filename = (TNamed*) (fFriends->At(0));
282 fFileToMerge = TFile::Open(filename->GetName());
284 fFileToMerge->GetObject("aodTree", fTreeToMerge);
285 if (!fAODEventToMerge) fAODEventToMerge = new AliAODEvent();
286 fAODEventToMerge->ReadFromTree(fTreeToMerge);
289 fFriendsConnected = kTRUE;
292 //______________________________________________________________________________
293 void AliAODInputHandler::CreatePIDResponse(Bool_t isMC/*=kFALSE*/)
296 // create the pid response object if it does not exist yet
298 if (fAODpidUtil) return;
299 fAODpidUtil=new AliAODpidUtil(isMC);