#include "AliRawReaderRoot.h"
#include "AliRawEvent.h"
#include "AliRawEventHeader.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
- fEvent = NULL;
TDirectory* dir = gDirectory;
fFile = TFile::Open(fileName);
dir->cd();
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;
- fHeader = 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;
- fHeader = 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;
- fHeader = rawReader.fHeader;
-
- 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&
return fEvent->GetHeader()->GetTypeAttribute();
}
+const UInt_t* AliRawReaderRoot::GetSubEventAttributes() const
+{
+// get the type attributes from the sub event header
+
+ if (!fSubEvent) return NULL;
+ return fSubEvent->GetHeader()->GetTypeAttribute();
+}
+
UInt_t AliRawReaderRoot::GetLDCId() const
{
// get the LDC Id from the event header
- if (!fEvent || !fEvent->GetSubEvent(fSubEventIndex)) return 0;
- return fEvent->GetSubEvent(fSubEventIndex)->GetHeader()->GetLDCId();
+ if (!fEvent || !fSubEvent) return 0;
+ return fSubEvent->GetHeader()->GetLDCId();
}
UInt_t AliRawReaderRoot::GetGDCId() const
{
// get the size of the equipment
- if (!fEvent || !fEvent->GetEquipmentHeader()) return 0;
- return fEvent->GetEquipmentHeader()->GetEquipmentSize();
+ if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
+ return fEquipment->GetEquipmentHeader()->GetEquipmentSize();
}
Int_t AliRawReaderRoot::GetEquipmentType() const
{
// get the type from the equipment header
- if (!fEvent || !fEvent->GetEquipmentHeader()) return -1;
- return fEvent->GetEquipmentHeader()->GetEquipmentType();
+ 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 || !fEvent->GetEquipmentHeader()) return -1;
- return fEvent->GetEquipmentHeader()->GetId();
+ 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 || !fEvent->GetEquipmentHeader()) return NULL;
- return fEvent->GetEquipmentHeader()->GetTypeAttribute();
+ 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 || !fEvent->GetEquipmentHeader()) return 0;
- return fEvent->GetEquipmentHeader()->GetBasicSizeType();
+ if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
+ return fEquipment->GetEquipmentHeader()->GetBasicSizeType();
}
// skip payload (if event was not selected)
if (fCount > 0) fPosition += fCount;
- // get the first or the next sub event if at the end of a sub event
- if (!fSubEvent || (fPosition >= fEnd)) {
+ // get the first or the next equipment if at the end of an equipment
+ if (!fEquipment || (fPosition >= fEnd)) {
- // check for end of event data
- if (fSubEventIndex >= fEvent->GetNSubEvents()) return kFALSE;
- fSubEvent = fEvent->GetSubEvent(fSubEventIndex++);
+ // get the first or the next sub event if at the end of a sub event
+ if (!fSubEvent || (fEquipmentIndex >= fSubEvent->GetNEquipments())) {
- // 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;
+ // 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;
}
- fRawData = fSubEvent->GetRawData();
+ // 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();
}
- // continue with the next sub event if no data left in the payload
+ // continue with the next equipment if no data left in the payload
if (fPosition >= fEnd) continue;
// 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);
- fSubEvent->GetHeader()->Dump();
+ fEquipment->GetEquipmentHeader()->Dump();
fCount = 0;
fPosition = fEnd;
fErrorCode = kErrNoDataHeader;
if (fPosition + fCount > fEnd) {
Error("ReadHeader", "size in data header exceeds event size!");
Warning("ReadHeader", "skipping %d bytes", fEnd - fPosition);
- fSubEvent->GetHeader()->Dump();
+ fEquipment->GetEquipmentHeader()->Dump();
fCount = 0;
fPosition = fEnd;
fErrorCode = kErrSize;
fSubEventIndex = 0;
fSubEvent = NULL;
+ fEquipmentIndex = 0;
+ fEquipment = NULL;
fRawData = NULL;
fHeader = NULL;
}
+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
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 a sub event
- if (!subEvent || (position >= end)) {
+ // 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;
return result;
}
- AliRawData* rawData = subEvent->GetRawData();
- position = (UChar_t*) rawData->GetBuffer();
- end = ((UChar_t*) rawData->GetBuffer()) + rawData->GetSize();
+ 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;
// check that there are enough bytes left for the data header
if (position + sizeof(AliRawDataHeader) > end) {
result |= kErrNoDataHeader;
- position = end;
continue;
}
- // check consistency of data size in the header and in the sub event
+ // check consistency of data size in the header and in the equipment
AliRawDataHeader* header = (AliRawDataHeader*) position;
- if (fHeader->fSize != 0xFFFFFFFF) {
+ if (header->fSize != 0xFFFFFFFF) {
if (position + header->fSize > end) {
result |= kErrSize;
- position = end;
- } else {
- position += header->fSize;
}
- } else {
- position = end;
}
};
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff