Bug correction
[u/mrichter/AliRoot.git] / STEER / AliStream.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 /*
17 $Log$
18 Revision 1.2  2001/10/04 15:58:52  jchudoba
19 Option to open the stream in READ or UPDATE mode
20
21 Revision 1.1  2001/09/19 06:20:50  jchudoba
22 Class to manage input filenames, used by AliRunDigitizer
23
24 */
25
26 ////////////////////////////////////////////////////////////////////////
27 //
28 // AliStream.cxx
29 //
30 // - store file names associated with a given stream
31 // - open and close files
32 // - return serial event number of the next event in the stream
33 // and the TFile pointer for a proper file
34 //
35 ////////////////////////////////////////////////////////////////////////
36
37 #include <iostream.h>
38
39 #include "TTree.h"
40
41 #include "AliStream.h"
42
43 #include "AliRun.h"
44
45 ClassImp(AliStream)
46
47 AliStream::AliStream()
48 {
49 // default ctor
50   fLastEventSerialNr = -1;
51   fLastEventNr = 0;
52   fCurrentFileIndex = -1;
53   fCurrentFile = 0;
54   fEvents = 0;
55   fFileNames = new TObjArray(1);
56   fMode = "READ";
57 }
58
59 ////////////////////////////////////////////////////////////////////////
60 AliStream::AliStream(Option_t *option)
61 {
62 // ctor
63   fLastEventSerialNr = -1;
64   fLastEventNr = 0;
65   fCurrentFileIndex = -1;
66   fCurrentFile = 0;
67   fEvents = 0;
68   fFileNames = new TObjArray(1);
69   fMode = option;
70 }
71
72 ////////////////////////////////////////////////////////////////////////
73 AliStream::~AliStream()
74 {
75 // default dtor
76   if (fFileNames) delete fFileNames;
77 }
78
79 ////////////////////////////////////////////////////////////////////////
80 void AliStream::AddFile(char *fileName)
81 {
82 // stores the name of the file
83   TObjString *name = new TObjString(fileName);
84   fFileNames->Add(name);
85 }
86
87 ////////////////////////////////////////////////////////////////////////
88 Bool_t AliStream::NextEventInStream(Int_t &serialNr)
89 {
90 // returns kFALSE if no more events
91 // returns kTRUE and the serial nr of the next event
92 // fCurrentFile points to the file containing offered event
93
94 // no files given:
95   if (fFileNames->GetLast() < 0) return kFALSE;
96
97   if (!fCurrentFile) {
98     if (!OpenNextFile()) return kFALSE;
99   }
100   
101   if (fLastEventSerialNr+1 >= fEvents) {
102     if (!OpenNextFile()) return kFALSE;
103   }
104   serialNr = ++fLastEventSerialNr;
105   return kTRUE;
106 }
107
108 ////////////////////////////////////////////////////////////////////////
109 void AliStream::ChangeMode(Option_t* option)
110 // set the mode to READ or UPDATE, reopen file with the new mode
111 // only change from UPDATE to READ have sense in the current scheme,
112 // other changes are possible but not usefull
113 {
114   fMode = option;
115   if (fCurrentFile) {
116     fCurrentFile->Close();
117     fCurrentFileIndex--;
118     OpenNextFile();
119   }
120 }
121
122 ////////////////////////////////////////////////////////////////////////
123 Bool_t AliStream::OpenNextFile()
124 {
125   if (++fCurrentFileIndex > fFileNames->GetLast()) {
126     cerr<<"No more files in the stream"<<endl;
127     return kFALSE;
128   }
129
130   if (fCurrentFile) {
131     if (fCurrentFile->IsOpen()) {
132       fCurrentFile->Close();
133     }
134   }
135
136   const char * filename = 
137     static_cast<TObjString*>(fFileNames->At(fCurrentFileIndex))->GetName();
138   fCurrentFile = TFile::Open(filename,fMode.Data());
139   if (!fCurrentFile) {
140 // cannot open file specified on input. Do not skip it silently.
141     cerr<<"Cannot open file "<<filename<<endl;
142     return kFALSE;
143   }
144 // find nr of events in the given file  
145   TTree * te = (TTree *) fCurrentFile->Get("TE") ;
146   if (!te) {
147     Error("OpenNextFile", "input file does not contain TE");
148     return kFALSE;
149   }
150   fEvents = static_cast<Int_t>(te->GetEntries());
151   fLastEventSerialNr = -1;
152   return kTRUE;
153 }
154
155 ////////////////////////////////////////////////////////////////////////
156 Bool_t AliStream::ImportgAlice()
157 {
158   if (fFileNames->GetLast() < 0) return kFALSE;
159   if (!fCurrentFile) {
160     if (!OpenNextFile()) return kFALSE;
161   }
162   gAlice = (AliRun*)fCurrentFile->Get("gAlice");
163   if (!gAlice)  return kFALSE;
164   return kTRUE;
165 }