* provided "as is" without express or implied warranty. *
**************************************************************************/
+/* $Id$ */
+
///////////////////////////////////////////////////////////////////////////////
-//
-// This is a class for reading a raw data from a root file and providing
-// information about digits
-//
+///
+/// This is a class for reading raw data from a root file.
+///
+/// The root file is expected to contain a tree of name "RAW" with
+/// a branch of name "rawevent" which contains objects of type
+/// AliRawEvent.
+///
+/// The file name and the event number are arguments of the constructor
+/// of AliRawReaderRoot.
+///
///////////////////////////////////////////////////////////////////////////////
+#include <TFile.h>
+#include <TTree.h>
#include "AliRawReaderRoot.h"
#include "AliRawEvent.h"
+#include "AliRawEventHeaderBase.h"
+#include "AliRawEquipment.h"
+#include "AliRawEquipmentHeader.h"
+#include "AliRawData.h"
ClassImp(AliRawReaderRoot)
-AliRawReaderRoot::AliRawReaderRoot(const char* fileName, Int_t eventNumber)
+AliRawReaderRoot::AliRawReaderRoot(const char* fileName, Int_t eventNumber) :
+ fFile(NULL),
+ fBranch(NULL),
+ fEventIndex(eventNumber),
+ fEvent(NULL),
+ fSubEventIndex(0),
+ fSubEvent(NULL),
+ fEquipmentIndex(0),
+ fEquipment(NULL),
+ fRawData(NULL),
+ fPosition(NULL),
+ fEnd(NULL)
{
// create an object to read digits from the given input file for the
// event with the given number
Error("AliRawReaderRoot", "no raw data tree found");
return;
}
- TBranch* branch = tree->GetBranch("rawevent");
- if (!branch) {
+ fBranch = tree->GetBranch("rawevent");
+ if (!fBranch) {
Error("AliRawReaderRoot", "no raw data branch found");
return;
}
fEvent = new AliRawEvent;
- branch->SetAddress(&fEvent);
- if (branch->GetEntry(eventNumber) <= 0) {
- Error("AliRawReaderRoot", "no event with number %d found", eventNumber);
- return;
+ fBranch->SetAddress(&fEvent);
+ if (fEventIndex >= 0) {
+ if (fBranch->GetEntry(fEventIndex) <= 0) {
+ Error("AliRawReaderRoot", "no event with number %d found", fEventIndex);
+ return;
+ }
}
-
- fSubEventIndex = 0;
- fSubEvent = NULL;
- fRawData = NULL;
- fMiniHeader = NULL;
-
- fCount = 0;
- fPosition = fEnd = NULL;
}
-AliRawReaderRoot::AliRawReaderRoot(AliRawEvent* event)
+AliRawReaderRoot::AliRawReaderRoot(AliRawEvent* event) :
+ fFile(NULL),
+ fBranch(NULL),
+ fEventIndex(-1),
+ fEvent(event),
+ fSubEventIndex(0),
+ fSubEvent(NULL),
+ fEquipmentIndex(0),
+ fEquipment(NULL),
+ fRawData(NULL),
+ fPosition(NULL),
+ fEnd(NULL)
{
// create an object to read digits from the given raw event
- fFile = NULL;
- fEvent = event;
-
- fSubEventIndex = 0;
- fSubEvent = NULL;
- fRawData = NULL;
- fMiniHeader = NULL;
-
- fCount = 0;
- fPosition = fEnd = NULL;
}
AliRawReaderRoot::AliRawReaderRoot(const AliRawReaderRoot& rawReader) :
- AliRawReader(rawReader)
+ AliRawReader(rawReader),
+ fFile(NULL),
+ fBranch(NULL),
+ fEventIndex(rawReader.fEventIndex),
+ fEvent(NULL),
+ fSubEventIndex(rawReader.fSubEventIndex),
+ fSubEvent(NULL),
+ fEquipmentIndex(rawReader.fEquipmentIndex),
+ fEquipment(NULL),
+ fRawData(NULL),
+ fPosition(NULL),
+ fEnd(NULL)
{
// copy constructor
- fFile = NULL;
- fEvent = rawReader.fEvent;
-
- fSubEventIndex = rawReader.fSubEventIndex;
- fSubEvent = rawReader.fSubEvent;
- fRawData = rawReader.fRawData;
- fMiniHeader = rawReader.fMiniHeader;
-
- fCount = rawReader.fCount;
- fPosition = rawReader.fPosition;
- fEnd = rawReader.fEnd;
+ if (rawReader.fFile) {
+ TDirectory* dir = gDirectory;
+ fFile = TFile::Open(rawReader.fFile->GetName());
+ dir->cd();
+ if (!fFile || !fFile->IsOpen()) {
+ Error("AliRawReaderRoot", "could not open file %s",
+ rawReader.fFile->GetName());
+ return;
+ }
+ TTree* tree = (TTree*) fFile->Get("RAW");
+ if (!tree) {
+ Error("AliRawReaderRoot", "no raw data tree found");
+ return;
+ }
+ fBranch = tree->GetBranch("rawevent");
+ if (!fBranch) {
+ Error("AliRawReaderRoot", "no raw data branch found");
+ return;
+ }
+
+ fEvent = new AliRawEvent;
+ fBranch->SetAddress(&fEvent);
+ if (fEventIndex >= 0) {
+ if (fBranch->GetEntry(fEventIndex) <= 0) {
+ Error("AliRawReaderRoot", "no event with number %d found",
+ fEventIndex);
+ return;
+ }
+ }
+ } else {
+ fEvent = rawReader.fEvent;
+ }
+
+ if (fSubEventIndex > 0) {
+ fSubEvent = fEvent->GetSubEvent(fSubEventIndex-1);
+ fEquipment = fSubEvent->GetEquipment(fEquipmentIndex);
+ fRawData = fEquipment->GetRawData();
+ fCount = 0;
+ fHeader = (AliRawDataHeader*) ((UChar_t*) fRawData->GetBuffer() +
+ ((UChar_t*) rawReader.fHeader -
+ (UChar_t*) rawReader.fRawData->GetBuffer()));
+ fPosition = (UChar_t*) fRawData->GetBuffer() +
+ (rawReader.fPosition - (UChar_t*) rawReader.fRawData->GetBuffer());
+ fEnd = ((UChar_t*) fRawData->GetBuffer()) + fRawData->GetSize();
+ }
}
AliRawReaderRoot& AliRawReaderRoot::operator = (const AliRawReaderRoot&
// get the type from the event header
if (!fEvent) return 0;
- return fEvent->GetHeader()->GetType();
+ return fEvent->GetHeader()->Get("Type");
}
UInt_t AliRawReaderRoot::GetRunNumber() const
// get the run number from the event header
if (!fEvent) return 0;
- return fEvent->GetHeader()->GetRunNumber();
+ return fEvent->GetHeader()->Get("RunNb");
}
const UInt_t* AliRawReaderRoot::GetEventId() const
// get the event id from the event header
if (!fEvent) return NULL;
- return fEvent->GetHeader()->GetId();
+ return fEvent->GetHeader()->GetP("Id");
}
const UInt_t* AliRawReaderRoot::GetTriggerPattern() const
// get the trigger pattern from the event header
if (!fEvent) return NULL;
- return fEvent->GetHeader()->GetTriggerPattern();
+ return fEvent->GetHeader()->GetP("TriggerPattern");
}
const UInt_t* AliRawReaderRoot::GetDetectorPattern() const
// get the detector pattern from the event header
if (!fEvent) return NULL;
- return fEvent->GetHeader()->GetDetectorPattern();
+ return fEvent->GetHeader()->GetP("DetectorPattern");
}
const UInt_t* AliRawReaderRoot::GetAttributes() const
// get the type attributes from the event header
if (!fEvent) return NULL;
- return fEvent->GetHeader()->GetTypeAttribute();
+ return fEvent->GetHeader()->GetP("TypeAttribute");
+}
+
+const UInt_t* AliRawReaderRoot::GetSubEventAttributes() const
+{
+// get the type attributes from the sub event header
+
+ if (!fSubEvent) return NULL;
+ return fSubEvent->GetHeader()->GetP("TypeAttribute");
+}
+
+UInt_t AliRawReaderRoot::GetLDCId() const
+{
+// get the LDC Id from the event header
+
+ if (!fEvent || !fSubEvent) return 0;
+ return fSubEvent->GetHeader()->Get("LdcId");
}
UInt_t AliRawReaderRoot::GetGDCId() const
// get the GDC Id from the event header
if (!fEvent) return 0;
- return fEvent->GetHeader()->GetGDCId();
+ return fEvent->GetHeader()->Get("GdcId");
}
-Bool_t AliRawReaderRoot::ReadMiniHeader()
+Int_t AliRawReaderRoot::GetEquipmentSize() const
+{
+// get the size of the equipment
+
+ if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
+ return fEquipment->GetEquipmentHeader()->GetEquipmentSize();
+}
+
+Int_t AliRawReaderRoot::GetEquipmentType() const
{
-// read a mini header at the current position
-// returns kFALSE if the mini header could not be read
+// get the type from the equipment header
+ if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return -1;
+ return fEquipment->GetEquipmentHeader()->GetEquipmentType();
+}
+
+Int_t AliRawReaderRoot::GetEquipmentId() const
+{
+// get the ID from the equipment header
+
+ if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return -1;
+ return fEquipment->GetEquipmentHeader()->GetId();
+}
+
+const UInt_t* AliRawReaderRoot::GetEquipmentAttributes() const
+{
+// get the attributes from the equipment header
+
+ if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return NULL;
+ return fEquipment->GetEquipmentHeader()->GetTypeAttribute();
+}
+
+Int_t AliRawReaderRoot::GetEquipmentElementSize() const
+{
+// get the basic element size from the equipment header
+
+ if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
+ return fEquipment->GetEquipmentHeader()->GetBasicSizeType();
+}
+
+
+Bool_t AliRawReaderRoot::ReadHeader()
+{
+// read a data header at the current position
+// returns kFALSE if the data header could not be read
+
+ fErrorCode = 0;
if (!fEvent) return kFALSE;
+
do {
- if (fCount > 0) fPosition += fCount; // skip payload if event was not selected
- if (!fSubEvent || (fPosition >= fEnd)) { // new sub event
- if (fSubEventIndex >= fEvent->GetNSubEvents()) return kFALSE;
- fSubEvent = fEvent->GetSubEvent(fSubEventIndex++);
- fRawData = fSubEvent->GetRawData();
+ // skip payload (if event was not selected)
+ if (fCount > 0) fPosition += fCount;
+
+ // get the first or the next equipment if at the end of an equipment
+ if (!fEquipment || (fPosition >= fEnd)) {
+
+ // get the first or the next sub event if at the end of a sub event
+ if (!fSubEvent || (fEquipmentIndex >= fSubEvent->GetNEquipments())) {
+
+ // check for end of event data
+ if (fSubEventIndex >= fEvent->GetNSubEvents()) return kFALSE;
+ fSubEvent = fEvent->GetSubEvent(fSubEventIndex++);
+
+ // check the magic word of the sub event
+ if (!fSubEvent->GetHeader()->IsValid()) {
+ Error("ReadHeader", "wrong magic number in sub event!");
+ fSubEvent->GetHeader()->Dump();
+ fErrorCode = kErrMagic;
+ return kFALSE;
+ }
+
+ fEquipmentIndex = 0;
+ fEquipment = NULL;
+ fRawData = NULL;
+ }
+
+ // get the next equipment and raw data
fCount = 0;
+ fEquipment = fSubEvent->GetEquipment(fEquipmentIndex++);
+ if (!fEquipment) continue;
+ fRawData = fEquipment->GetRawData();
+ if (!fRawData) {
+ fPosition = fEnd;
+ continue;
+ }
fPosition = (UChar_t*) fRawData->GetBuffer();
fEnd = ((UChar_t*) fRawData->GetBuffer()) + fRawData->GetSize();
}
- if (fPosition >= fEnd) continue; // no data left in the payload
- if (fPosition + sizeof(AliMiniHeader) > fEnd) {
- Error("ReadMiniHeader", "could not read data!");
- return kFALSE;
- }
- fMiniHeader = (AliMiniHeader*) fPosition;
- fPosition += sizeof(AliMiniHeader);
- if (!CheckMiniHeader()) {
- Warning("ReadMiniHeader", "skipping %d bytes", fEnd - fPosition);
- fSubEvent->GetHeader()->Dump();
- fCount = 0;
- fPosition = fEnd;
- continue;
+
+ // continue with the next equipment if no data left in the payload
+ if (fPosition >= fEnd) continue;
+
+ if (fRequireHeader) {
+ // check that there are enough bytes left for the data header
+ if (fPosition + sizeof(AliRawDataHeader) > fEnd) {
+ Error("ReadHeader", "could not read data header!");
+ Warning("ReadHeader", "skipping %d bytes", fEnd - fPosition);
+ fEquipment->GetEquipmentHeader()->Dump();
+ fCount = 0;
+ fPosition = fEnd;
+ fErrorCode = kErrNoDataHeader;
+ continue;
+ }
+
+ // "read" the data header
+ fHeader = (AliRawDataHeader*) fPosition;
+ fPosition += sizeof(AliRawDataHeader);
}
- fCount = fMiniHeader->fSize;
- if (fPosition + fCount > fEnd) { // check data size in mini header and sub event
- Error("ReadMiniHeader", "size in mini header exceeds event size!");
- Warning("ReadMiniHeader", "skipping %d bytes", fEnd - fPosition);
- fSubEvent->GetHeader()->Dump();
- fCount = 0;
- fPosition = fEnd;
- continue;
+
+ if (fHeader && (fHeader->fSize != 0xFFFFFFFF)) {
+ fCount = fHeader->fSize - sizeof(AliRawDataHeader);
+
+ // check consistency of data size in the header and in the sub event
+ if (fPosition + fCount > fEnd) {
+ Error("ReadHeader", "size in data header exceeds event size!");
+ Warning("ReadHeader", "skipping %d bytes", fEnd - fPosition);
+ fEquipment->GetEquipmentHeader()->Dump();
+ fCount = 0;
+ fPosition = fEnd;
+ fErrorCode = kErrSize;
+ continue;
+ }
+
+ } else {
+ fCount = fEnd - fPosition;
}
+
} while (!IsSelected());
+
return kTRUE;
}
// reads the next payload at the current position
// returns kFALSE if the data could not be read
+ fErrorCode = 0;
while (fCount == 0) {
- if (!ReadMiniHeader()) return kFALSE;
+ if (!ReadHeader()) return kFALSE;
}
data = fPosition;
fPosition += fCount;
// reads the next block of data at the current position
// returns kFALSE if the data could not be read
+ fErrorCode = 0;
if (fPosition + size > fEnd) {
Error("ReadNext", "could not read data!");
+ fErrorCode = kErrOutOfBounds;
return kFALSE;
}
memcpy(data, fPosition, size);
fSubEventIndex = 0;
fSubEvent = NULL;
+ fEquipmentIndex = 0;
+ fEquipment = NULL;
fRawData = NULL;
- fMiniHeader = NULL;
+ fHeader = NULL;
fCount = 0;
fPosition = fEnd = NULL;
return kTRUE;
}
+
+Bool_t AliRawReaderRoot::NextEvent()
+{
+// go to the next event in the root file
+
+ if (!fFile) return kFALSE;
+
+ do {
+ if (fBranch->GetEntry(fEventIndex+1) <= 0) {
+ delete fEvent;
+ fEvent = new AliRawEvent;
+ fBranch->SetAddress(&fEvent);
+ return kFALSE;
+ }
+ fEventIndex++;
+ } while (!IsEventSelected());
+ return Reset();
+}
+
+Bool_t AliRawReaderRoot::RewindEvents()
+{
+// go back to the beginning of the root file
+
+ if (!fFile) return kFALSE;
+
+ fEventIndex = -1;
+ delete fEvent;
+ fEvent = new AliRawEvent;
+ fBranch->SetAddress(&fEvent);
+ return Reset();
+}
+
+
+Int_t AliRawReaderRoot::CheckData() const
+{
+// check the consistency of the data
+
+ if (!fEvent) return 0;
+
+ AliRawEvent* subEvent = NULL;
+ Int_t subEventIndex = 0;
+ AliRawEquipment* equipment = NULL;
+ Int_t equipmentIndex = 0;
+ UChar_t* position = 0;
+ UChar_t* end = 0;
+ Int_t result = 0;
+
+ while (kTRUE) {
+ // get the first or the next sub event if at the end of an equipment
+ if (!subEvent || (equipmentIndex >= subEvent->GetNEquipments())) {
+
+ // check for end of event data
+ if (subEventIndex >= fEvent->GetNSubEvents()) return result;
+ subEvent = fEvent->GetSubEvent(subEventIndex++);
+
+ // check the magic word of the sub event
+ if (!fSubEvent->GetHeader()->IsValid()) {
+ result |= kErrMagic;
+ return result;
+ }
+
+ equipmentIndex = 0;
+ }
+
+ // get the next equipment and raw data
+ equipment = subEvent->GetEquipment(equipmentIndex++);
+ if (!equipment) continue;
+ AliRawData* rawData = equipment->GetRawData();
+ if (!rawData) continue;
+ position = (UChar_t*) rawData->GetBuffer();
+ end = ((UChar_t*) rawData->GetBuffer()) + rawData->GetSize();
+
+ // continue with the next sub event if no data left in the payload
+ if (position >= end) continue;
+
+ if (fRequireHeader) {
+ // check that there are enough bytes left for the data header
+ if (position + sizeof(AliRawDataHeader) > end) {
+ result |= kErrNoDataHeader;
+ continue;
+ }
+
+ // check consistency of data size in the header and in the equipment
+ AliRawDataHeader* header = (AliRawDataHeader*) position;
+ if (header->fSize != 0xFFFFFFFF) {
+ if (position + header->fSize > end) result |= kErrSize;
+ }
+ }
+ };
+
+ return result;
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff