Renaming AliHLTReconstructorBase to AliHLTPluginBase to reflect the
[u/mrichter/AliRoot.git] / HLT / rec / AliHLTOUTDigitReader.cxx
CommitLineData
c5123824 1// $Id$
2
3//**************************************************************************
4//* This file is property of and copyright by the ALICE HLT Project *
5//* ALICE Experiment at CERN, All rights reserved. *
6//* *
7//* Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no> *
8//* for The ALICE HLT Project. *
9//* *
10//* Permission to use, copy, modify and distribute this software and its *
11//* documentation strictly for non-commercial purposes is hereby granted *
12//* without fee, provided that the above copyright notice appears in all *
13//* copies and that both the copyright notice and this permission notice *
14//* appear in the supporting documentation. The authors make no claims *
15//* about the suitability of this software for any purpose. It is *
16//* provided "as is" without express or implied warranty. *
17//**************************************************************************
18
19/** @file AliHLTOUTDigitReader.cxx
20 @author Matthias Richter
21 @date
22 @brief HLTOUT data wrapper for simulated AliRoot HLT digit data.
23*/
24
25#include "AliHLTOUTDigitReader.h"
26#include "AliRawDataHeader.h"
27#include "AliDAQ.h"
28#include "TTree.h"
29#include "TFile.h"
30#include "TArrayC.h"
31#include "TBranch.h"
32#include "TString.h"
33#include <cassert>
34
35/** ROOT macro for the implementation of ROOT specific class methods */
36ClassImp(AliHLTOUTDigitReader)
37
032c5e5e 38AliHLTOUTDigitReader::AliHLTOUTDigitReader(int event, AliHLTEsdManager* pEsdManager, const char* digitFile)
c5123824 39 :
40 AliHLTOUTHomerCollection(event, pEsdManager),
032c5e5e 41 fDigitFileName(digitFile),
c5123824 42 fpDigitFile(NULL),
43 fpDigitTree(NULL),
44 fMinDDL(-1),
45 fMaxDDL(-1),
46 fppDigitArrays(NULL),
47 fpEquipments(NULL),
48 fNofDDLs(0),
49 fCurrent(-1)
50{
51 // see header file for class documentation
52 // or
53 // refer to README to build package
54 // or
55 // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
56}
57
58AliHLTOUTDigitReader::~AliHLTOUTDigitReader()
59{
60 // see header file for class documentation
61 CloseTree();
62
63 if (fpDigitFile) {
64 fpDigitFile->Close();
65 delete fpDigitFile;
66 fpDigitFile=NULL;
67 }
68}
69
70Bool_t AliHLTOUTDigitReader::ReadNextData(UChar_t*& data)
71{
72 // see header file for class documentation
73 if (!fppDigitArrays && (!ReadArrays() || !fppDigitArrays))
74 return kFALSE;
75
76 if (fCurrent>=fNofDDLs)
77 return kFALSE;
78
79 while (++fCurrent<fNofDDLs) {
80 if (fMinDDL>-1 && fMinDDL>fpEquipments[fCurrent]) continue;
81 if (fMaxDDL>-1 && fMaxDDL<fpEquipments[fCurrent]) continue;
82 if (fppDigitArrays[fCurrent]->GetSize()>(int)sizeof(AliRawDataHeader)) {
83 data=reinterpret_cast<UChar_t*>(fppDigitArrays[fCurrent]->GetArray());
84 data+=sizeof(AliRawDataHeader);
85 return kTRUE;
86 }
87 }
88 return kFALSE;
89}
90
91int AliHLTOUTDigitReader::Reset()
92{
93 // see header file for class documentation
94 fCurrent=-1;
95 return 0;
96}
97
98int AliHLTOUTDigitReader::GetDataSize()
99{
100 // see header file for class documentation
101 if (fCurrent>=0 && fCurrent<fNofDDLs && fppDigitArrays) {
102 return fppDigitArrays[fCurrent]->GetSize()-sizeof(AliRawDataHeader);
103 }
104 return 0;
105}
106
107const AliRawDataHeader* AliHLTOUTDigitReader::GetDataHeader()
108{
109 // see header file for class documentation
110 if (fCurrent>=0 && fCurrent<fNofDDLs && fppDigitArrays) {
111 return reinterpret_cast<AliRawDataHeader*>(fppDigitArrays[fCurrent]->GetArray());
112 }
113 return NULL;
114}
115
116void AliHLTOUTDigitReader::SelectEquipment(int /*equipmentType*/, int minEquipmentId, int maxEquipmentId)
117{
118 // see header file for class documentation
119 fMinDDL=minEquipmentId;
120 fMaxDDL=maxEquipmentId;
121}
122
123int AliHLTOUTDigitReader::GetEquipmentId()
124{
125 // see header file for class documentation
126 if (fCurrent>=0 && fCurrent<fNofDDLs && fpEquipments) {
127 return fpEquipments[fCurrent];
128 }
129 return -1;
130}
131
132bool AliHLTOUTDigitReader::ReadArrays()
133{
134 // see header file for class documentation
135 bool result=false;
136
137 if (GetCurrentEventNo()<0) {
138 HLTWarning("no event selected, no data available");
139 return false;
140 }
141
142 if (!fpDigitFile) {
032c5e5e 143 fpDigitFile=new TFile(fDigitFileName);
c5123824 144 }
145 if (!fpDigitFile) return false;
146 if (fpDigitFile->IsZombie()) return false;
147
148 fpDigitFile->GetObject("rawhltout", fpDigitTree);
149 if (!fpDigitTree) {
150 return false;
151 }
152
153 vector<int> equipments;
154 int iTotal=0;
155 TIter iter(fpDigitTree->GetListOfBranches());
156 while (TBranch* br=(TBranch *)iter.Next()) {
157 iTotal++;
158 TString bname=br->GetName();
159 if (bname.BeginsWith("HLT_") && bname.EndsWith(".ddl")) {
160 bname.ReplaceAll("HLT_", "");
161 bname.ReplaceAll(".ddl", "");
162 if (bname.IsDigit()) {
163 int equipment=bname.Atoi();
164 int index=equipment-AliDAQ::DdlIDOffset("HLT");
165 if (index>=0 && strcmp(br->GetName(), AliDAQ::DdlFileName("HLT", index))==0) {
166 equipments.push_back(equipment);
167 } else {
168 HLTWarning("equipment mismatch: skipping id %d (out of range of HLT equipments)", equipment);
169 }
170 }
171 }
172 }
173 HLTDebug("found %d HLT branche(s) out of %d", equipments.size(), iTotal);
174
175 if (equipments.size()>0) {
176 fNofDDLs=equipments.size();
177 fpEquipments=new int[fNofDDLs];
178 fppDigitArrays=new TArrayC*[fNofDDLs];
179 if (fpEquipments && fppDigitArrays) {
180 memcpy(fpEquipments, &equipments[0], fNofDDLs*sizeof(int));
181 int i=0;
182 for (i=0; i<fNofDDLs; i++) {
183 fppDigitArrays[i]=NULL;
184 fpDigitTree->SetBranchAddress(AliDAQ::DdlFileName("HLT", fpEquipments[i]-AliDAQ::DdlIDOffset("HLT")), &fppDigitArrays[i]);
185 }
186 fpDigitTree->GetEvent(GetCurrentEventNo());
187 for (i=0; i<fNofDDLs; i++) {
188 if (fppDigitArrays[i]) {
189 HLTDebug("branch %s: %d byte(s)", AliDAQ::DdlFileName("HLT", fpEquipments[i]-AliDAQ::DdlIDOffset("HLT")), fppDigitArrays[i]);
190 }
191 }
192 result=true;
193 } else {
194 fNofDDLs=0;
195 }
196 }
197 return result;
198}
199
200int AliHLTOUTDigitReader::CloseTree()
201{
202 // see header file for class documentation
203 int iResult=0;
204 if (fppDigitArrays) delete[] fppDigitArrays;
205 fppDigitArrays=NULL;
206 if (fpEquipments) delete[] fpEquipments;
207 fpEquipments=NULL;
208 if (fpDigitTree) delete fpDigitTree;
209 fpDigitTree=NULL;
210 fNofDDLs=0;
211 fCurrent=-1;
212
213 return iResult;
214}
c1292031 215
216void AliHLTOUTDigitReader::SetParam(TTree* /*pDigitTree*/, int event)
217{
218 // see header file for class documentation
219
220 // TODO: here we have the implemented to correct loading of
221 // the digit file from the run loader. At time of writing
222 // (Jun 2008) the AliLoader for the HLT is missing in the AliRoot
223 // framework.
224 fEvent=event;
225}