Just something left from v2...now clean
[u/mrichter/AliRoot.git] / RAW / AliRawReaderRoot.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 /* $Id$ */
17
18 ///////////////////////////////////////////////////////////////////////////////
19 ///
20 /// This is a class for reading raw data from a root file.
21 ///
22 /// The root file is expected to contain a tree of name "RAW" with
23 /// a branch of name "rawevent" which contains objects of type
24 /// AliRawEvent.
25 /// 
26 /// The file name and the event number are arguments of the constructor
27 /// of AliRawReaderRoot.
28 ///
29 ///////////////////////////////////////////////////////////////////////////////
30
31 #include <TFile.h>
32 #include <TTree.h>
33 #include "AliRawReaderRoot.h"
34 #include "AliRawEvent.h"
35 #include "AliRawEventHeaderBase.h"
36 #include "AliRawEquipment.h"
37 #include "AliRawEquipmentHeader.h"
38 #include "AliRawData.h"
39
40
41 ClassImp(AliRawReaderRoot)
42
43
44 AliRawReaderRoot::AliRawReaderRoot(const char* fileName, Int_t eventNumber) :
45   fFile(NULL),
46   fBranch(NULL),
47   fEventIndex(eventNumber),
48   fEvent(NULL),
49   fSubEventIndex(0),
50   fSubEvent(NULL),
51   fEquipmentIndex(0),
52   fEquipment(NULL),
53   fRawData(NULL),
54   fPosition(NULL),
55   fEnd(NULL)
56 {
57 // create an object to read digits from the given input file for the
58 // event with the given number
59
60   TDirectory* dir = gDirectory;
61   fFile = TFile::Open(fileName);
62   dir->cd();
63   if (!fFile || !fFile->IsOpen()) {
64     Error("AliRawReaderRoot", "could not open file %s", fileName);
65     return;
66   }
67   TTree* tree = (TTree*) fFile->Get("RAW");
68   if (!tree) {
69     Error("AliRawReaderRoot", "no raw data tree found");
70     return;
71   }
72   fBranch = tree->GetBranch("rawevent");
73   if (!fBranch) {
74     Error("AliRawReaderRoot", "no raw data branch found");
75     return;
76   }
77
78   fEvent = new AliRawEvent;
79   fBranch->SetAddress(&fEvent);
80   if (fEventIndex >= 0) {
81     if (fBranch->GetEntry(fEventIndex) <= 0) {
82       Error("AliRawReaderRoot", "no event with number %d found", fEventIndex);
83       return;
84     }
85   }
86 }
87
88 AliRawReaderRoot::AliRawReaderRoot(AliRawEvent* event) :
89   fFile(NULL),
90   fBranch(NULL),
91   fEventIndex(-1),
92   fEvent(event),
93   fSubEventIndex(0),
94   fSubEvent(NULL),
95   fEquipmentIndex(0),
96   fEquipment(NULL),
97   fRawData(NULL),
98   fPosition(NULL),
99   fEnd(NULL)
100 {
101 // create an object to read digits from the given raw event
102
103 }
104
105 AliRawReaderRoot::AliRawReaderRoot(const AliRawReaderRoot& rawReader) :
106   AliRawReader(rawReader),
107   fFile(NULL),
108   fBranch(NULL),
109   fEventIndex(rawReader.fEventIndex),
110   fEvent(NULL),
111   fSubEventIndex(rawReader.fSubEventIndex),
112   fSubEvent(NULL),
113   fEquipmentIndex(rawReader.fEquipmentIndex),
114   fEquipment(NULL),
115   fRawData(NULL),
116   fPosition(NULL),
117   fEnd(NULL)
118 {
119 // copy constructor
120
121   if (rawReader.fFile) {
122     TDirectory* dir = gDirectory;
123     fFile = TFile::Open(rawReader.fFile->GetName());
124     dir->cd();
125     if (!fFile || !fFile->IsOpen()) {
126       Error("AliRawReaderRoot", "could not open file %s", 
127             rawReader.fFile->GetName());
128       return;
129     }
130     TTree* tree = (TTree*) fFile->Get("RAW");
131     if (!tree) {
132       Error("AliRawReaderRoot", "no raw data tree found");
133       return;
134     }
135     fBranch = tree->GetBranch("rawevent");
136     if (!fBranch) {
137       Error("AliRawReaderRoot", "no raw data branch found");
138       return;
139     }
140
141     fEvent = new AliRawEvent;
142     fBranch->SetAddress(&fEvent);
143     if (fEventIndex >= 0) {
144       if (fBranch->GetEntry(fEventIndex) <= 0) {
145         Error("AliRawReaderRoot", "no event with number %d found", 
146               fEventIndex);
147         return;
148       }
149     }
150   } else {
151     fEvent = rawReader.fEvent;
152   }
153
154   if (fSubEventIndex > 0) {
155     fSubEvent = fEvent->GetSubEvent(fSubEventIndex-1);
156     fEquipment = fSubEvent->GetEquipment(fEquipmentIndex);
157     fRawData = fEquipment->GetRawData();
158       fCount = 0;
159     fHeader = (AliRawDataHeader*) ((UChar_t*) fRawData->GetBuffer() + 
160       ((UChar_t*) rawReader.fHeader - 
161        (UChar_t*) rawReader.fRawData->GetBuffer()));
162     fPosition = (UChar_t*) fRawData->GetBuffer() + 
163       (rawReader.fPosition - (UChar_t*) rawReader.fRawData->GetBuffer());
164     fEnd = ((UChar_t*) fRawData->GetBuffer()) + fRawData->GetSize();
165   }
166 }
167
168 AliRawReaderRoot& AliRawReaderRoot::operator = (const AliRawReaderRoot& 
169                                                 rawReader)
170 {
171 // assignment operator
172
173   this->~AliRawReaderRoot();
174   new(this) AliRawReaderRoot(rawReader);
175   return *this;
176 }
177
178 AliRawReaderRoot::~AliRawReaderRoot()
179 {
180 // delete objects and close root file
181
182   if (fFile) {
183     if (fEvent) delete fEvent;
184     fFile->Close();
185     delete fFile;
186   }
187 }
188
189 const AliRawEventHeaderBase* AliRawReaderRoot::GetEventHeader() const
190 {
191   // Get the even header
192   // Return NULL in case of failure
193   if (!fEvent) return NULL;
194   return fEvent->GetHeader();
195 }
196
197 UInt_t AliRawReaderRoot::GetType() const
198 {
199 // get the type from the event header
200
201   if (!fEvent) return 0;
202   return fEvent->GetHeader()->Get("Type");
203 }
204
205 UInt_t AliRawReaderRoot::GetRunNumber() const
206 {
207 // get the run number from the event header
208
209   if (!fEvent) return 0;
210   return fEvent->GetHeader()->Get("RunNb");
211 }
212
213 const UInt_t* AliRawReaderRoot::GetEventId() const
214 {
215 // get the event id from the event header
216
217   if (!fEvent) return NULL;
218   return fEvent->GetHeader()->GetP("Id");
219 }
220
221 const UInt_t* AliRawReaderRoot::GetTriggerPattern() const
222 {
223 // get the trigger pattern from the event header
224
225   if (!fEvent) return NULL;
226   return fEvent->GetHeader()->GetP("TriggerPattern");
227 }
228
229 const UInt_t* AliRawReaderRoot::GetDetectorPattern() const
230 {
231 // get the detector pattern from the event header
232
233   if (!fEvent) return NULL;
234   return fEvent->GetHeader()->GetP("DetectorPattern");
235 }
236
237 const UInt_t* AliRawReaderRoot::GetAttributes() const
238 {
239 // get the type attributes from the event header
240
241   if (!fEvent) return NULL;
242   return fEvent->GetHeader()->GetP("TypeAttribute");
243 }
244
245 const UInt_t* AliRawReaderRoot::GetSubEventAttributes() const
246 {
247 // get the type attributes from the sub event header
248
249   if (!fSubEvent) return NULL;
250   return fSubEvent->GetHeader()->GetP("TypeAttribute");
251 }
252
253 UInt_t AliRawReaderRoot::GetLDCId() const
254 {
255 // get the LDC Id from the event header
256
257   if (!fEvent || !fSubEvent) return 0;
258   return fSubEvent->GetHeader()->Get("LdcId");
259 }
260
261 UInt_t AliRawReaderRoot::GetGDCId() const
262 {
263 // get the GDC Id from the event header
264
265   if (!fEvent) return 0;
266   return fEvent->GetHeader()->Get("GdcId");
267 }
268
269 UInt_t AliRawReaderRoot::GetTimestamp() const
270 {
271   if (!fEvent) return 0;
272   return fEvent->GetHeader()->Get("Timestamp");
273 }
274
275 Int_t AliRawReaderRoot::GetEquipmentSize() const
276 {
277 // get the size of the equipment
278
279   if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
280   return fEquipment->GetEquipmentHeader()->GetEquipmentSize();
281 }
282
283 Int_t AliRawReaderRoot::GetEquipmentType() const
284 {
285 // get the type from the equipment header
286
287   if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return -1;
288   return fEquipment->GetEquipmentHeader()->GetEquipmentType();
289 }
290
291 Int_t AliRawReaderRoot::GetEquipmentId() const
292 {
293 // get the ID from the equipment header
294
295   if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return -1;
296   return fEquipment->GetEquipmentHeader()->GetId();
297 }
298
299 const UInt_t* AliRawReaderRoot::GetEquipmentAttributes() const
300 {
301 // get the attributes from the equipment header
302
303   if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return NULL;
304   return fEquipment->GetEquipmentHeader()->GetTypeAttribute();
305 }
306
307 Int_t AliRawReaderRoot::GetEquipmentElementSize() const
308 {
309 // get the basic element size from the equipment header
310
311   if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
312   return fEquipment->GetEquipmentHeader()->GetBasicSizeType();
313 }
314
315 Int_t AliRawReaderRoot::GetEquipmentHeaderSize() const
316 {
317 // get the size of the equipment header (28 bytes by default)
318
319   if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
320   return fEquipment->GetEquipmentHeader()->HeaderSize();
321 }
322
323 // _________________________________________________________________________
324 void AliRawReaderRoot::SwapData(const void* inbuf, const void* outbuf, UInt_t size) {
325   // The method swaps the contents of the
326   // raw-data event header
327   UInt_t  intCount = (size+3)/sizeof(UInt_t);
328
329   UInt_t* buf = (UInt_t*) inbuf;    // temporary integers buffer
330   for (UInt_t i=0; i<intCount; i++, buf++) {
331       UInt_t value = SwapWord(*buf);
332       if (i==(intCount-1))
333          memcpy((UInt_t*)outbuf+i, &value, size%sizeof(UInt_t));
334       else
335          memcpy((UInt_t*)outbuf+i, &value, sizeof(UInt_t));
336   }
337 }
338 // _________________________________________________________________________
339
340 Bool_t AliRawReaderRoot::ReadHeader()
341 {
342 // read a data header at the current position
343 // returns kFALSE if the data header could not be read
344
345   fErrorCode = 0;
346   if (!fEvent) return kFALSE;
347
348   do {
349     // skip payload (if event was not selected)
350     if (fCount > 0) fPosition += fCount;
351
352     // get the first or the next equipment if at the end of an equipment
353     if (!fEquipment || (fPosition >= fEnd)) {
354
355       // get the first or the next sub event if at the end of a sub event
356       if (!fSubEvent || (fEquipmentIndex >= fSubEvent->GetNEquipments())) {
357
358         // check for end of event data
359         if (fSubEventIndex >= fEvent->GetNSubEvents()) return kFALSE;
360         fSubEvent = fEvent->GetSubEvent(fSubEventIndex++);
361
362         // check the magic word of the sub event
363         if (!fSubEvent->GetHeader()->IsValid()) {
364           Error("ReadHeader", "wrong magic number in sub event!");
365           fSubEvent->GetHeader()->Dump();
366           fErrorCode = kErrMagic;
367           return kFALSE;
368         }
369
370         fEquipmentIndex = 0;
371         fEquipment = NULL;
372         fRawData = NULL;
373       }
374
375       // get the next equipment and raw data
376       fCount = 0;
377       fEquipment = fSubEvent->GetEquipment(fEquipmentIndex++);
378       if (!fEquipment) continue;
379       if (!IsSelected()) {
380         fPosition = fEnd;
381         continue;
382       }
383       fRawData = fEquipment->GetRawData();
384       if (!fRawData) {
385         fPosition = fEnd;
386         continue;
387       }
388       fPosition = (UChar_t*) fRawData->GetBuffer();
389       fEnd = ((UChar_t*) fRawData->GetBuffer()) + fRawData->GetSize();
390     }
391
392     // continue with the next equipment if no data left in the payload
393     if (fPosition >= fEnd) continue;
394
395     if (fRequireHeader) {
396       // check that there are enough bytes left for the data header
397       if (fPosition + sizeof(AliRawDataHeader) > fEnd) {
398         Error("ReadHeader", "could not read data header!");
399         Warning("ReadHeader", "skipping %d bytes", fEnd - fPosition);
400         fEquipment->GetEquipmentHeader()->Dump();
401         fCount = 0;
402         fPosition = fEnd;
403         fErrorCode = kErrNoDataHeader;
404         continue;
405       }
406
407       // "read" the data header
408       fHeader = (AliRawDataHeader*) fPosition;
409 #ifndef R__BYTESWAP
410       SwapData((void*) fHeader, (void*) fHeaderSwapped, sizeof(AliRawDataHeader));
411       fHeader=fHeaderSwapped;
412 #endif
413       if ((fPosition + fHeader->fSize) != fEnd) {
414         if (fHeader->fSize != 0xFFFFFFFF)
415           Warning("ReadHeader",
416                   "Equipment %d : raw data size found in the header is wrong (%d != %d)! Using the equipment size instead !",
417                   fEquipment->GetEquipmentHeader()->GetId(),fHeader->fSize, fEnd - fPosition);
418         fHeader->fSize = fEnd - fPosition;
419       }
420       fPosition += sizeof(AliRawDataHeader);
421     }
422
423     if (fHeader && (fHeader->fSize != 0xFFFFFFFF)) {
424       fCount = fHeader->fSize - sizeof(AliRawDataHeader);
425
426       // check consistency of data size in the header and in the sub event
427       if (fPosition + fCount > fEnd) {  
428         Error("ReadHeader", "size in data header exceeds event size!");
429         Warning("ReadHeader", "skipping %d bytes", fEnd - fPosition);
430         fEquipment->GetEquipmentHeader()->Dump();
431         fCount = 0;
432         fPosition = fEnd;
433         fErrorCode = kErrSize;
434         continue;
435       }
436
437     } else {
438       fCount = fEnd - fPosition;
439     }
440
441   } while (!IsSelected());
442
443   return kTRUE;
444 }
445
446 Bool_t AliRawReaderRoot::ReadNextData(UChar_t*& data)
447 {
448 // reads the next payload at the current position
449 // returns kFALSE if the data could not be read
450
451   fErrorCode = 0;
452   while (fCount == 0) {
453     if (!ReadHeader()) return kFALSE;
454   }
455   data = fPosition;
456   fPosition += fCount;  
457   fCount = 0;
458   return kTRUE;
459 }
460
461 Bool_t AliRawReaderRoot::ReadNext(UChar_t* data, Int_t size)
462 {
463 // reads the next block of data at the current position
464 // returns kFALSE if the data could not be read
465
466   fErrorCode = 0;
467   if (fPosition + size > fEnd) {
468     Error("ReadNext", "could not read data!");
469     fErrorCode = kErrOutOfBounds;
470     return kFALSE;
471   }
472   memcpy(data, fPosition, size);
473
474   fPosition += size;
475   fCount -= size;
476   return kTRUE;
477 }
478
479
480 Bool_t AliRawReaderRoot::Reset()
481 {
482 // reset the current position to the beginning of the event
483
484   fSubEventIndex = 0;
485   fSubEvent = NULL;
486   fEquipmentIndex = 0;
487   fEquipment = NULL;
488   fRawData = NULL;
489   fHeader = NULL;
490
491   fCount = 0;
492   fPosition = fEnd = NULL;
493   return kTRUE;
494 }
495
496
497 Bool_t AliRawReaderRoot::NextEvent()
498 {
499 // go to the next event in the root file
500
501   if (!fBranch) return kFALSE;
502
503   do {
504     delete fEvent;
505     fEvent = new AliRawEvent;
506     fBranch->SetAddress(&fEvent);
507     if (fBranch->GetEntry(fEventIndex+1) <= 0)
508       return kFALSE;
509     fEventIndex++;
510   } while (!IsEventSelected());
511   fEventNumber++;
512   return Reset();
513 }
514
515 Bool_t AliRawReaderRoot::RewindEvents()
516 {
517 // go back to the beginning of the root file
518
519   if (!fBranch) return kFALSE;
520
521   fEventIndex = -1;
522   delete fEvent;
523   fEvent = new AliRawEvent;
524   fBranch->SetAddress(&fEvent);
525   fEventNumber = -1;
526   return Reset();
527 }
528
529
530 Int_t AliRawReaderRoot::CheckData() const
531 {
532 // check the consistency of the data
533
534   if (!fEvent) return 0;
535
536   AliRawEvent* subEvent = NULL;
537   Int_t subEventIndex = 0;
538   AliRawEquipment* equipment = NULL;
539   Int_t equipmentIndex = 0;
540   UChar_t* position = 0;
541   UChar_t* end = 0;
542   Int_t result = 0;
543
544   while (kTRUE) {
545     // get the first or the next sub event if at the end of an equipment
546     if (!subEvent || (equipmentIndex >= subEvent->GetNEquipments())) {
547
548       // check for end of event data
549       if (subEventIndex >= fEvent->GetNSubEvents()) return result;
550       subEvent = fEvent->GetSubEvent(subEventIndex++);
551
552       // check the magic word of the sub event
553       if (!fSubEvent->GetHeader()->IsValid()) {
554         result |= kErrMagic;
555         return result;
556       }
557
558       equipmentIndex = 0;
559     }
560
561     // get the next equipment and raw data
562     equipment = subEvent->GetEquipment(equipmentIndex++);
563     if (!equipment) continue;
564     AliRawData* rawData = equipment->GetRawData();
565     if (!rawData) continue;
566     position = (UChar_t*) rawData->GetBuffer();
567     end = ((UChar_t*) rawData->GetBuffer()) + rawData->GetSize();
568
569     // continue with the next sub event if no data left in the payload
570     if (position >= end) continue;
571
572     if (fRequireHeader) {
573     // check that there are enough bytes left for the data header
574       if (position + sizeof(AliRawDataHeader) > end) {
575         result |= kErrNoDataHeader;
576         continue;
577       }
578
579       // check consistency of data size in the header and in the equipment
580       AliRawDataHeader* header = (AliRawDataHeader*) position;
581       if ((position + header->fSize) != end) {
582         if (header->fSize != 0xFFFFFFFF)
583           Warning("ReadHeader",
584                   "Equipment %d : raw data size found in the header is wrong (%d != %d)! Using the equipment size instead !",
585                   equipment->GetEquipmentHeader()->GetId(),header->fSize, end - position);
586         header->fSize = end - position;
587         result |= kErrSize;
588       }
589     }
590     position = end;
591   };
592
593   return result;
594 }