#include <TFile.h>
#include <TTree.h>
#include <TTreeIndex.h>
#include <TGrid.h>
#include "AliRawReaderRoot.h"
#include "AliRawVEvent.h"
#include "AliRawEventHeaderBase.h"
#include "AliRawVEquipment.h"
#include "AliRawEquipmentHeader.h"
#include "AliRawData.h"
ClassImp(AliRawReaderRoot)
Bool_t AliRawReaderRoot::fgUseOrder = kFALSE;
AliRawReaderRoot::AliRawReaderRoot() :
fFile(NULL),
fBranch(NULL),
fEventIndex(-1),
fEvent(NULL),
fEventHeader(NULL),
fSubEventIndex(0),
fSubEvent(NULL),
fEquipmentIndex(0),
fEquipment(NULL),
fRawData(NULL),
fPosition(NULL),
fEnd(NULL),
fIndex(0x0)
{
}
AliRawReaderRoot::AliRawReaderRoot(const char* fileName, Int_t eventNumber) :
fFile(NULL),
fBranch(NULL),
fEventIndex(eventNumber),
fEvent(NULL),
fEventHeader(NULL),
fSubEventIndex(0),
fSubEvent(NULL),
fEquipmentIndex(0),
fEquipment(NULL),
fRawData(NULL),
fPosition(NULL),
fEnd(NULL),
fIndex(0x0)
{
TDirectory* dir = gDirectory;
TString flStr = fileName;
if (flStr.BeginsWith("alien://") && !gGrid) TGrid::Connect("alien://");
fFile = TFile::Open(fileName);
dir->cd();
if (!fFile || !fFile->IsOpen()) {
Error("AliRawReaderRoot", "could not open file %s", fileName);
fIsValid = kFALSE;
return;
}
TTree* tree = (TTree*) fFile->Get("RAW");
if (!tree) {
Error("AliRawReaderRoot", "no raw data tree found");
fIsValid = kFALSE;
return;
}
fBranch = tree->GetBranch("rawevent");
if (!fBranch) {
Error("AliRawReaderRoot", "no raw data branch found");
fIsValid = kFALSE;
return;
}
fBranch->SetAddress(&fEvent);
if (fEventIndex >= 0) {
if (fBranch->GetEntry(fEventIndex) <= 0) {
Error("AliRawReaderRoot", "no event with number %d found", fEventIndex);
fIsValid = kFALSE;
return;
}
fEventHeader = fEvent->GetHeader();
}
}
AliRawReaderRoot::AliRawReaderRoot(AliRawVEvent* event) :
fFile(NULL),
fBranch(NULL),
fEventIndex(-1),
fEvent(event),
fEventHeader(event->GetHeader()),
fSubEventIndex(0),
fSubEvent(NULL),
fEquipmentIndex(0),
fEquipment(NULL),
fRawData(NULL),
fPosition(NULL),
fEnd(NULL),
fIndex(0x0)
{
if (!fEvent) fIsValid = kFALSE;
}
AliRawReaderRoot::AliRawReaderRoot(const AliRawReaderRoot& rawReader) :
AliRawReader(rawReader),
fFile(NULL),
fBranch(NULL),
fEventIndex(rawReader.fEventIndex),
fEvent(NULL),
fEventHeader(NULL),
fSubEventIndex(rawReader.fSubEventIndex),
fSubEvent(NULL),
fEquipmentIndex(rawReader.fEquipmentIndex),
fEquipment(NULL),
fRawData(NULL),
fPosition(NULL),
fEnd(NULL),
fIndex(0x0)
{
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());
fIsValid = kFALSE;
return;
}
TTree* tree = (TTree*) fFile->Get("RAW");
if (!tree) {
Error("AliRawReaderRoot", "no raw data tree found");
fIsValid = kFALSE;
return;
}
fBranch = tree->GetBranch("rawevent");
if (!fBranch) {
Error("AliRawReaderRoot", "no raw data branch found");
fIsValid = kFALSE;
return;
}
fBranch->SetAddress(&fEvent);
if (fEventIndex >= 0) {
if (fBranch->GetEntry(fEventIndex) <= 0) {
Error("AliRawReaderRoot", "no event with number %d found",
fEventIndex);
fIsValid = kFALSE;
return;
}
fEventHeader = fEvent->GetHeader();
}
} else {
fEvent = rawReader.fEvent;
fEventHeader = rawReader.fEventHeader;
}
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()));
UChar_t version = 2;
if (fHeader) version = fHeader->GetVersion();
if (version==3) {
fHeader = NULL;
fHeaderV3 = (AliRawDataHeaderV3*) ((UChar_t*) fRawData->GetBuffer() +
((UChar_t*) rawReader.fHeaderV3 -
(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&
rawReader)
{
this->~AliRawReaderRoot();
new(this) AliRawReaderRoot(rawReader);
return *this;
}
AliRawReaderRoot::~AliRawReaderRoot()
{
if (fFile) {
if (fEvent) delete fEvent;
fFile->Close();
delete fFile;
}
}
const AliRawEventHeaderBase* AliRawReaderRoot::GetEventHeader() const
{
return fEventHeader;
}
UInt_t AliRawReaderRoot::GetType() const
{
if (!fEventHeader) return 0;
return fEventHeader->Get("Type");
}
UInt_t AliRawReaderRoot::GetRunNumber() const
{
if (!fEventHeader) return 0;
return fEventHeader->Get("RunNb");
}
const UInt_t* AliRawReaderRoot::GetEventId() const
{
if (!fEventHeader) return NULL;
return fEventHeader->GetP("Id");
}
const UInt_t* AliRawReaderRoot::GetTriggerPattern() const
{
if (!fEventHeader) return NULL;
return fEventHeader->GetP("TriggerPattern");
}
const UInt_t* AliRawReaderRoot::GetDetectorPattern() const
{
if (!fEventHeader) return NULL;
return fEventHeader->GetP("DetectorPattern");
}
const UInt_t* AliRawReaderRoot::GetAttributes() const
{
if (!fEventHeader) return NULL;
return fEventHeader->GetP("TypeAttribute");
}
const UInt_t* AliRawReaderRoot::GetSubEventAttributes() const
{
if (!fSubEvent) return NULL;
return fSubEvent->GetHeader()->GetP("TypeAttribute");
}
UInt_t AliRawReaderRoot::GetLDCId() const
{
if (!fEvent || !fSubEvent) return 0;
return fSubEvent->GetHeader()->Get("LdcId");
}
UInt_t AliRawReaderRoot::GetGDCId() const
{
if (!fEventHeader) return 0;
return fEventHeader->Get("GdcId");
}
UInt_t AliRawReaderRoot::GetTimestamp() const
{
if (!fEventHeader) return 0;
return fEventHeader->Get("Timestamp");
}
Int_t AliRawReaderRoot::GetEquipmentSize() const
{
if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
return fEquipment->GetEquipmentHeader()->GetEquipmentSize();
}
Int_t AliRawReaderRoot::GetEquipmentType() const
{
if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return -1;
return fEquipment->GetEquipmentHeader()->GetEquipmentType();
}
Int_t AliRawReaderRoot::GetEquipmentId() const
{
if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return -1;
return fEquipment->GetEquipmentHeader()->GetId();
}
const UInt_t* AliRawReaderRoot::GetEquipmentAttributes() const
{
if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return NULL;
return fEquipment->GetEquipmentHeader()->GetTypeAttribute();
}
Int_t AliRawReaderRoot::GetEquipmentElementSize() const
{
if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
return fEquipment->GetEquipmentHeader()->GetBasicSizeType();
}
Int_t AliRawReaderRoot::GetEquipmentHeaderSize() const
{
if (!fEvent || !fEquipment || !fEquipment->GetEquipmentHeader()) return 0;
return fEquipment->GetEquipmentHeader()->HeaderSize();
}
void AliRawReaderRoot::SwapData(const void* inbuf, const void* outbuf, UInt_t size) {
UInt_t intCount = (size+3)/sizeof(UInt_t);
UInt_t* buf = (UInt_t*) inbuf;
for (UInt_t i=0; i<intCount; i++, buf++) {
UInt_t value = SwapWord(*buf);
if (i==(intCount-1))
memcpy((UInt_t*)outbuf+i, &value, size%sizeof(UInt_t));
else
memcpy((UInt_t*)outbuf+i, &value, sizeof(UInt_t));
}
}
Bool_t AliRawReaderRoot::ReadHeader()
{
fErrorCode = 0;
if (!fEvent) return kFALSE;
do {
if (fCount > 0) fPosition += fCount;
if (!fEquipment || (fPosition >= fEnd)) {
if (!fSubEvent || (fEquipmentIndex >= fSubEvent->GetNEquipments())) {
if (fSubEventIndex >= fEvent->GetNSubEvents()) return kFALSE;
fSubEvent = fEvent->GetSubEvent(fSubEventIndex++);
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;
}
fCount = 0;
if (fEquipmentIndex >= fSubEvent->GetNEquipments()) {
fEquipment = NULL;
continue;
}
fEquipment = fSubEvent->GetEquipment(fEquipmentIndex++);
if (!fEquipment) continue;
if (!IsSelected()) {
fPosition = fEnd;
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;
if (fRequireHeader) {
if (fPosition + sizeof(AliRawDataHeader) > fEnd) {
Error("ReadHeader", "could not read data header!");
Warning("ReadHeader", "skipping %ld bytes", fEnd - fPosition);
fEquipment->GetEquipmentHeader()->Dump();
fCount = 0;
fPosition = fEnd;
fErrorCode = kErrNoDataHeader;
continue;
}
fHeader = (AliRawDataHeader*) fPosition;
UChar_t version = 2;
if (fHeader) version=fHeader->GetVersion();
if (version==2) {
#ifndef R__BYTESWAP
SwapData((void*) fHeader, (void*) fHeaderSwapped, sizeof(AliRawDataHeader));
fHeader=fHeaderSwapped;
#endif
if ((fPosition + fHeader->fSize) != fEnd) {
if (fHeader->fSize != 0xFFFFFFFF)
Warning("ReadHeader",
"Equipment %d : raw data size found in the header is wrong (%d != %ld)! Using the equipment size instead !",
fEquipment->GetEquipmentHeader()->GetId(),fHeader->fSize, fEnd - fPosition);
fHeader->fSize = fEnd - fPosition;
}
fPosition += sizeof(AliRawDataHeader);
fHeaderV3 = 0;
} else if (version==3) {
fHeaderV3 = (AliRawDataHeaderV3*) fPosition;
#ifndef R__BYTESWAP
SwapData((void*) fHeaderV3, (void*) fHeaderSwapped, sizeof(AliRawDataHeaderV3));
fHeaderV3=fHeaderSwappedV3;
#endif
if ((fPosition + fHeaderV3->fSize) != fEnd) {
if (fHeaderV3->fSize != 0xFFFFFFFF)
Warning("ReadHeader",
"Equipment %d : raw data size found in the header is wrong (%d != %ld)! Using the equipment size instead !",
fEquipment->GetEquipmentHeader()->GetId(),fHeader->fSize, fEnd - fPosition);
fHeaderV3->fSize = fEnd - fPosition;
}
fPosition += sizeof(AliRawDataHeaderV3);
fHeader = 0;
}
}
if (fHeader && (fHeader->fSize != 0xFFFFFFFF)) {
fCount = fHeader->fSize - sizeof(AliRawDataHeader);
if (fPosition + fCount > fEnd) {
Error("ReadHeader", "size in data header exceeds event size!");
Warning("ReadHeader", "skipping %ld bytes", fEnd - fPosition);
fEquipment->GetEquipmentHeader()->Dump();
fCount = 0;
fPosition = fEnd;
fErrorCode = kErrSize;
continue;
}
} else if (fHeaderV3 && (fHeaderV3->fSize != 0xFFFFFFFF)) {
fCount = fHeaderV3->fSize - sizeof(AliRawDataHeaderV3);
if (fPosition + fCount > fEnd) {
Error("ReadHeader", "size in data header exceeds event size!");
Warning("ReadHeader", "skipping %ld bytes", fEnd - fPosition);
fEquipment->GetEquipmentHeader()->Dump();
fCount = 0;
fPosition = fEnd;
fErrorCode = kErrSize;
continue;
}
} else {
fCount = fEnd - fPosition;
}
} while (!IsSelected());
return kTRUE;
}
Bool_t AliRawReaderRoot::ReadNextData(UChar_t*& data)
{
fErrorCode = 0;
while (fCount == 0) {
if (!ReadHeader()) return kFALSE;
}
data = fPosition;
fPosition += fCount;
fCount = 0;
return kTRUE;
}
Bool_t AliRawReaderRoot::ReadNext(UChar_t* data, Int_t size)
{
fErrorCode = 0;
if (fPosition + size > fEnd) {
Error("ReadNext", "could not read data!");
fErrorCode = kErrOutOfBounds;
return kFALSE;
}
memcpy(data, fPosition, size);
fPosition += size;
fCount -= size;
return kTRUE;
}
Bool_t AliRawReaderRoot::Reset()
{
fSubEventIndex = 0;
fSubEvent = NULL;
fEquipmentIndex = 0;
fEquipment = NULL;
fRawData = NULL;
fHeader = NULL;
fHeaderV3 = NULL;
fCount = 0;
fPosition = fEnd = NULL;
return kTRUE;
}
Bool_t AliRawReaderRoot::NextEvent()
{
if (!fBranch) return kFALSE;
if (fgUseOrder && !fIndex) MakeIndex();
do {
delete fEvent;
fEvent = NULL;
fEventHeader = NULL;
fBranch->SetAddress(&fEvent);
Int_t entryToGet = fEventIndex + 1;
if (fgUseOrder
&& fIndex
&& entryToGet<fBranch->GetEntries()
&& entryToGet>-1 ) entryToGet = fIndex[entryToGet];
if (fBranch->GetEntry(entryToGet) <= 0)
return kFALSE;
fEventHeader = fEvent->GetHeader();
fEventIndex++;
} while (!IsEventSelected());
fEventNumber++;
return Reset();
}
Bool_t AliRawReaderRoot::RewindEvents()
{
if (!fBranch) return kFALSE;
fEventIndex = -1;
delete fEvent;
fEvent = NULL;
fEventHeader = NULL;
fBranch->SetAddress(&fEvent);
fEventNumber = -1;
return Reset();
}
Bool_t AliRawReaderRoot::GotoEvent(Int_t event)
{
if (!fBranch) return kFALSE;
if (fgUseOrder && !fIndex) MakeIndex();
delete fEvent;
fEvent = NULL;
fEventHeader = NULL;
fBranch->SetAddress(&fEvent);
Int_t entryToGet = event;
if (fgUseOrder
&& fIndex
&& entryToGet<fBranch->GetEntries()
&& entryToGet>-1 ) entryToGet = fIndex[entryToGet];
if (fBranch->GetEntry(entryToGet) <= 0)
return kFALSE;
fEventHeader = fEvent->GetHeader();
fEventIndex = event;
fEventNumber++;
return Reset();
}
Int_t AliRawReaderRoot::GetNumberOfEvents() const
{
if (!fBranch) return -1;
return fBranch->GetEntries();
}
Int_t AliRawReaderRoot::CheckData() const
{
if (!fEvent) return 0;
AliRawVEvent* subEvent = NULL;
Int_t subEventIndex = 0;
AliRawVEquipment* equipment = NULL;
Int_t equipmentIndex = 0;
UChar_t* position = 0;
UChar_t* end = 0;
Int_t result = 0;
while (kTRUE) {
if (!subEvent || (equipmentIndex >= subEvent->GetNEquipments())) {
if (subEventIndex >= fEvent->GetNSubEvents()) return result;
subEvent = fEvent->GetSubEvent(subEventIndex++);
if (fSubEvent && !fSubEvent->GetHeader()->IsValid()) {
result |= kErrMagic;
return result;
}
equipmentIndex = 0;
}
if (equipmentIndex >= subEvent->GetNEquipments()) {
equipment = NULL;
continue;
}
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();
if (position >= end) continue;
if (fRequireHeader) {
if (position + sizeof(AliRawDataHeader) > end) {
result |= kErrNoDataHeader;
continue;
}
AliRawDataHeader* header = (AliRawDataHeader*) position;
UChar_t version = header->GetVersion();
if (version==2) {
if ((position + header->fSize) != end) {
if (header->fSize != 0xFFFFFFFF)
Warning("CheckData",
"Equipment %d : raw data size found in the header V2 is wrong (%d != %ld)! Using the equipment size instead !",
equipment->GetEquipmentHeader()->GetId(),header->fSize, end - position);
header->fSize = end - position;
result |= kErrSize;
}
}
else if (version==3) {
AliRawDataHeaderV3 * headerV3 = (AliRawDataHeaderV3*) position;
if ((position + headerV3->fSize) != end) {
if (headerV3->fSize != 0xFFFFFFFF)
Warning("CheckData",
"Equipment %d : raw data size found in the header V3 is wrong (%d != %ld)! Using the equipment size instead !",
equipment->GetEquipmentHeader()->GetId(),headerV3->fSize, end - position);
headerV3->fSize = end - position;
result |= kErrSize;
}
}
}
position = end;
};
return result;
}
AliRawReader* AliRawReaderRoot::CloneSingleEvent() const
{
if (fEvent) {
AliRawVEvent *gdcRootEvent = (AliRawVEvent*)fEvent->Clone();
for (Int_t ldcCounter=0; ldcCounter < gdcRootEvent->GetNSubEvents(); ldcCounter++) {
AliRawVEvent *ldcRootEvent = gdcRootEvent->GetSubEvent(ldcCounter);
AliRawVEvent *subEvent = fEvent->GetSubEvent(ldcCounter);
for (Int_t eqCounter=0; eqCounter < ldcRootEvent->GetNEquipments(); eqCounter++) {
AliRawVEquipment *equipment=ldcRootEvent->GetEquipment(eqCounter);
AliRawVEquipment *eq = subEvent->GetEquipment(eqCounter);
equipment->CloneRawData(eq->GetRawData());
}
}
gdcRootEvent->GetSubEvent(-1);
fEvent->GetSubEvent(-1);
return new AliRawReaderRoot(gdcRootEvent);
}
return NULL;
}
void AliRawReaderRoot::MakeIndex() {
if (fBranch) {
TTree * rawTree = fBranch->GetTree();
if (rawTree) {
rawTree->BuildIndex("-fEvtHdrs[0].fSize");
TTreeIndex * treeInd = (TTreeIndex*)rawTree->GetTreeIndex();
if (treeInd) fIndex = treeInd->GetIndex();
}
}
}