Reset list of trees before resing new event.
[u/mrichter/AliRoot.git] / STEER / AliESDInputHandlerRP.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 ESD input reading the RecPoint Trees in parallel
20 //     Author: Andreas Morsch, CERN
21 //-------------------------------------------------------------------------
22
23 #include <TTree.h>
24 #include <TList.h>
25 #include <TFile.h>
26 #include <TArchiveFile.h>
27 #include <TSystemDirectory.h>
28 #include <TString.h>
29 #include <TObjString.h>
30 #include <TProcessID.h>
31
32 #include "AliESDInputHandlerRP.h"
33 #include "AliESDEvent.h"
34 #include "AliESD.h"
35 #include "AliLog.h"
36
37 ClassImp(AliESDInputHandlerRP)
38
39 //______________________________________________________________________________
40 AliESDInputHandlerRP::AliESDInputHandlerRP() :
41     AliESDInputHandler(),
42     fRTrees(   new TList()),
43     fRFiles(   new TList()),
44     fDetectors(new TList()),
45     fDirR(0),
46     fEventNumber(-1),
47     fNEvent(-1),
48     fFileNumber(0),
49     fEventsPerFile(0),
50     fExtension(""),
51     fPathName(new TString("./")),
52     fIsArchive(kFALSE)
53 {
54   // Default constructor
55 }
56
57
58 //______________________________________________________________________________
59 AliESDInputHandlerRP::AliESDInputHandlerRP(const char* name, const char* title):
60     AliESDInputHandler(name, title),
61     fRTrees(   new TList()),
62     fRFiles(   new TList()),
63     fDetectors(new TList()),
64     fDirR(0),
65     fEventNumber(-1),
66     fNEvent(-1),
67     fFileNumber(0),
68     fEventsPerFile(0),
69     fExtension(""),
70     fPathName(new TString("./")),
71     fIsArchive(kFALSE)
72 {
73     // Constructor
74 }
75
76 //______________________________________________________________________________
77 AliESDInputHandlerRP::~AliESDInputHandlerRP() 
78 {
79   // Destructor
80 }
81
82 Bool_t AliESDInputHandlerRP::Init(Option_t* opt)
83 {
84     //
85     // Initialize input
86     //
87     if (!(strcmp(opt, "proof")) || !(strcmp(opt, "local"))) return kTRUE;
88     //
89     TIter next(fDetectors);
90     TNamed* det;
91     TFile* file = 0;
92     while ((det = (TNamed*) next()))
93     {
94         if (!fIsArchive) {
95             file = TFile::Open(Form("%s%s.RecPoints.root", fPathName->Data(), det->GetName()));
96         } else {
97             file = TFile::Open(Form("%s#%s.RecPoints.root", fPathName->Data(), det->GetName()));
98         }
99         if (!file) AliFatal(Form("AliESDInputHandlerRP: %s.RecPoints.root not found in %s ! \n", det->GetName(), fPathName->Data()));
100         fRFiles->Add(file);
101     }
102     if (file) {
103         fEventsPerFile = file->GetNkeys() - file->GetNProcessIDs();
104     } else {
105         AliFatal(Form("AliESDInputHandlerRP: No file with RecPoints found in %s ! \n", fPathName->Data()));
106     }
107     
108     
109     // Reset the event number
110     fEventNumber      = -1;
111     fFileNumber       =  0;
112     // Get number of events from esd tree 
113     fNEvent           =  fTree->GetEntries();
114     
115     printf("AliESDInputHandler::Init() %d\n",__LINE__);
116     return kTRUE;
117 }
118
119 Bool_t AliESDInputHandlerRP::BeginEvent(Long64_t entry)
120 {
121     // Begin the next event
122     // Delegate first to base class
123     AliESDInputHandler::BeginEvent(entry);
124 //
125     if (entry == -1) {
126         fEventNumber++;
127         entry = fEventNumber;
128     } else {
129         fEventNumber = entry;
130     }
131     
132     if (entry >= fNEvent) {
133         AliWarning(Form("AliESDInputHandlerRP: Event number out of range %5d %5d\n", entry, fNEvent));
134         return kFALSE;
135     }
136     return LoadEvent(entry);
137 }
138
139 Bool_t AliESDInputHandlerRP::LoadEvent(Int_t iev)
140 {
141     // Load the event number iev
142     //
143     // Calculate the file number
144     Int_t inew  = iev / fEventsPerFile;
145     if (inew != fFileNumber) {
146         fFileNumber = inew;
147         if (!OpenFile(fFileNumber)){
148             return kFALSE;
149         }
150     }
151     // Folder name
152     char folder[20];
153     sprintf(folder, "Event%d", iev);
154     // Tree R
155     fRTrees->Clear();
156     
157     TIter next(fRFiles);
158     TFile* file;
159     while ((file = (TFile*) next()))
160     {
161         file->GetObject(folder, fDirR);
162         if (!fDirR) {
163             AliWarning(Form("AliESDInputHandlerRP: Event #%5d not found\n", iev));
164             return kFALSE;
165         }
166         TTree* tree;
167         fDirR ->GetObject("TreeR", tree);
168         fRTrees->Add(tree);
169         tree->ls();
170     }
171     return kTRUE;
172 }
173
174 Bool_t AliESDInputHandlerRP::OpenFile(Int_t i)
175 {
176     // Open file i
177     Bool_t ok = kTRUE;
178     if (i > 0) {
179         fExtension = Form("%d", i);
180     } else {
181         fExtension = "";
182     }
183     
184     fRFiles->Delete();
185     TIter next(fDetectors);
186     TNamed* det;
187     TFile* file;
188     while ((det = (TNamed*) next()))
189     {
190         if (!fIsArchive) {
191             file = TFile::Open(Form("%s%s.RecPoints%s.root", fPathName->Data(), det->GetName(), fExtension));
192         } else {
193             file = TFile::Open(Form("%s#%s.RecPoints%s.root", fPathName->Data(), det->GetName(), fExtension));
194         }
195         if (!file) AliFatal(Form("AliESDInputHandlerRP: RecPoints.root not found in %s ! \n", fPathName->Data()));
196         fRFiles->Add(file);
197     }
198     return ok;
199 }
200
201 Bool_t AliESDInputHandlerRP::Notify(const char *path)
202 {
203   // Notify about directory change
204   // The directory is taken from the 'path' argument
205   // 
206     // Get path to directory
207     TString fileName(path);
208     if(fileName.Contains("AliESDs.root")){
209         fileName.ReplaceAll("AliESDs.root", "");
210     }
211     // If this is an archive it will contain a # 
212     if(fileName.Contains("#")){
213         fileName.ReplaceAll("#", "");
214     }
215     //
216     // At this point we have a path to the directory or to the archive
217     *fPathName = fileName;
218     //
219     // Now filter the files containing RecPoints *.RecPoints.*
220     fIsArchive = kFALSE;
221     if (fPathName->Contains(".zip")) fIsArchive = kTRUE;
222
223     TSeqCollection* members;
224     
225     if (fIsArchive) {
226         // Archive
227         TFile* file = TFile::Open(fPathName->Data());
228         TArchiveFile* arch = file->GetArchive();
229         members = arch->GetMembers();
230     } else {
231         // Directory
232         TSystemDirectory dir(".", fPathName->Data());
233         members = dir.GetListOfFiles();
234     }
235     
236     TIter next(members);
237     TFile* entry;
238     Int_t ien = 0;
239     while ( (entry = (TFile*) next()) )
240     {
241         printf("File %s \n", entry->GetName());
242         TString name(entry->GetName());
243         TObjArray* tokens = name.Tokenize(".");
244         Int_t ntok = tokens->GetEntries();
245         if (ntok <= 1) continue;
246         TString str = ((TObjString*) tokens->At(1))->GetString();
247         if (!(strcmp(str.Data(), "RecPoints"))){
248             TString det = ((TObjString*) tokens->At(0))->GetString();
249             printf("Name  %s \n", det.Data());
250             TNamed* ent = new TNamed(det.Data(), det.Data());
251             ent->SetUniqueID(ien++);
252             fDetectors->Add(ent);
253         }
254     } // loop over files
255     
256
257     // Now we have the path and the list of detectors
258     
259     printf("AliESDInputHandlerRP::Notify() Path: %s\n", fPathName->Data());
260     //
261     ResetIO();
262     InitIO("");
263     // Some clean-up
264     members->Delete();
265     //
266     return kTRUE;
267 }
268
269 Bool_t AliESDInputHandlerRP::FinishEvent()
270 {
271     // Clean-up after each event
272     delete fDirR;  fDirR = 0;
273     AliESDInputHandler::FinishEvent();
274     return kTRUE;
275 }
276
277 void AliESDInputHandlerRP::ResetIO()
278 {
279 // Delete trees and files
280     fRTrees->Delete();
281     fRFiles->Delete();
282     fExtension="";
283 }
284
285 TTree* AliESDInputHandlerRP::GetTreeR(char* det)
286 {
287 // Return pointer to RecPoint tree for detector det
288     TNamed* entry = (TNamed*) (fDetectors->FindObject(det));
289     if (!entry) {
290         AliWarning(Form("AliESDInputHandlerRP: No RecPoints for detector %s available \n", det));
291         return 0;
292     } else {
293         Int_t ien = entry->GetUniqueID();
294         return ((TTree*) (fRTrees->At(ien)));
295     }
296 }