]>
Commit | Line | Data |
---|---|---|
6923e953 | 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 | /// | |
18 | /// This is a class for reading raw data from a root chain. | |
19 | /// There are two constructors available - one from a text file containing the | |
20 | /// list of root raw-data files to be processed and one directly from | |
21 | /// TFileCollection. | |
22 | /// | |
23 | /// cvetan.cheshkov@cern.ch 29/07/2008 | |
24 | /// | |
25 | /////////////////////////////////////////////////////////////////////////////// | |
26 | ||
27 | #include <TChain.h> | |
28 | #include <TFileCollection.h> | |
29 | ||
30 | #include "AliRawReaderChain.h" | |
31 | #include "AliRawEvent.h" | |
32 | ||
33 | ClassImp(AliRawReaderChain) | |
34 | ||
35 | AliRawReaderChain::AliRawReaderChain() : | |
36 | AliRawReaderRoot(), | |
37 | fChain(NULL) | |
38 | { | |
39 | // default constructor | |
40 | } | |
41 | ||
acce5036 | 42 | AliRawReaderChain::AliRawReaderChain(const char* listFileName) : |
6923e953 | 43 | AliRawReaderRoot(), |
44 | fChain(NULL) | |
45 | { | |
46 | // create raw-reader objects which takes as an input a root chain | |
47 | // from the file list found in 'listFileName' | |
48 | ||
49 | TFileCollection collection("RAW", | |
50 | "Collection with raw-data files", | |
acce5036 | 51 | listFileName); |
6923e953 | 52 | |
acce5036 | 53 | fChain = new TChain("RAW"); |
6923e953 | 54 | if (!fChain->AddFileInfoList((TCollection*)(collection.GetList()))) { |
55 | Error("AliRawReaderChain","Bad file list in collection, the chain is empty"); | |
a97af23d | 56 | fIsValid = kFALSE; |
6923e953 | 57 | return; |
58 | } | |
59 | ||
636c1780 | 60 | fChain->SetBranchStatus("*",1); |
4ebdfbcc | 61 | fChain->SetBranchAddress("rawevent",&fEvent,&fBranch); |
6923e953 | 62 | } |
63 | ||
64 | AliRawReaderChain::AliRawReaderChain(TFileCollection *collection) : | |
65 | AliRawReaderRoot(), | |
66 | fChain(NULL) | |
67 | { | |
68 | // create raw-reader objects which takes as an input a root chain | |
69 | // from a root file collection | |
70 | ||
acce5036 | 71 | fChain = new TChain("RAW"); |
6923e953 | 72 | if (!fChain->AddFileInfoList((TCollection*)(collection->GetList()))) { |
73 | Error("AliRawReaderChain","Bad file list in collection, the chain is empty"); | |
a97af23d | 74 | fIsValid = kFALSE; |
6923e953 | 75 | return; |
76 | } | |
77 | ||
636c1780 | 78 | fChain->SetBranchStatus("*",1); |
4ebdfbcc | 79 | fChain->SetBranchAddress("rawevent",&fEvent,&fBranch); |
636c1780 | 80 | } |
81 | ||
82 | AliRawReaderChain::AliRawReaderChain(TChain *chain) : | |
83 | AliRawReaderRoot(), | |
84 | fChain(chain) | |
85 | { | |
86 | // create raw-reader objects which takes as an input a root chain | |
87 | // from a root file collection | |
88 | ||
a97af23d | 89 | if (!fChain) fIsValid = kFALSE; |
90 | ||
636c1780 | 91 | fChain->SetBranchStatus("*",1); |
4ebdfbcc | 92 | fChain->SetBranchAddress("rawevent",&fEvent,&fBranch); |
6923e953 | 93 | } |
94 | ||
95 | AliRawReaderChain::AliRawReaderChain(const AliRawReaderChain& rawReader) : | |
96 | AliRawReaderRoot(rawReader), | |
97 | fChain(rawReader.fChain) | |
98 | { | |
99 | // copy constructor | |
100 | } | |
101 | ||
102 | AliRawReaderChain& AliRawReaderChain::operator = (const AliRawReaderChain& | |
103 | rawReader) | |
104 | { | |
105 | // assignment operator | |
106 | ||
107 | this->~AliRawReaderChain(); | |
108 | new(this) AliRawReaderChain(rawReader); | |
109 | return *this; | |
110 | } | |
111 | ||
112 | AliRawReaderChain::~AliRawReaderChain() | |
113 | { | |
114 | // delete objects and close root file | |
115 | ||
116 | if (fChain) { | |
117 | delete fChain; | |
118 | fChain = NULL; | |
119 | } | |
120 | } | |
121 | ||
122 | Bool_t AliRawReaderChain::NextEvent() | |
123 | { | |
124 | // go to the next event in the root file | |
125 | ||
126 | if (!fChain || !fChain->GetListOfFiles()->GetEntriesFast()) return kFALSE; | |
127 | ||
128 | do { | |
129 | delete fEvent; | |
130 | fEvent = new AliRawEvent; | |
4ebdfbcc | 131 | Long64_t treeEntry = fChain->LoadTree(fEventIndex+1); |
132 | if (!fBranch) | |
636c1780 | 133 | return kFALSE; |
4ebdfbcc | 134 | if (fBranch->GetEntry(treeEntry) <= 0) |
6923e953 | 135 | return kFALSE; |
136 | fEventIndex++; | |
137 | } while (!IsEventSelected()); | |
138 | fEventNumber++; | |
139 | return Reset(); | |
140 | } | |
141 | ||
142 | Bool_t AliRawReaderChain::RewindEvents() | |
143 | { | |
144 | // go back to the beginning of the root file | |
145 | ||
146 | fEventIndex = -1; | |
147 | delete fEvent; | |
148 | fEvent = new AliRawEvent; | |
149 | fEventNumber = -1; | |
150 | return Reset(); | |
151 | } | |
636c1780 | 152 | |
153 | Bool_t AliRawReaderChain::GotoEvent(Int_t event) | |
154 | { | |
155 | // go to a particular event | |
156 | // Uses the absolute event index inside the | |
157 | // chain with raw data | |
158 | ||
159 | if (!fChain || !fChain->GetListOfFiles()->GetEntriesFast()) return kFALSE; | |
160 | ||
161 | delete fEvent; | |
162 | fEvent = new AliRawEvent; | |
4ebdfbcc | 163 | Long64_t treeEntry = fChain->LoadTree(event); |
164 | if (!fBranch) | |
636c1780 | 165 | return kFALSE; |
4ebdfbcc | 166 | if (fBranch->GetEntry(treeEntry) <= 0) |
636c1780 | 167 | return kFALSE; |
168 | fEventIndex = event; | |
169 | fEventNumber++; | |
170 | return Reset(); | |
171 | } | |
25e82ff5 | 172 | |
173 | Int_t AliRawReaderChain::GetNumberOfEvents() const | |
174 | { | |
175 | // Get the total number of events in the chain | |
176 | // of raw-data files | |
177 | ||
178 | if (!fChain) return -1; | |
179 | ||
180 | return fChain->GetEntries(); | |
181 | } |