-int AliHLTTPCClusterAccessHLTOUT::ReadAliHLTTPCClusterMCData(AliHLTOUT* pHLTOUT, AliHLTTPCClusterMCDataList &tpcClusterLabels) const
-{
- // read cluster data from AliHLTTPCClusterData
- int iResult=0;
- if (!pHLTOUT) return -EINVAL;
- do {
- const AliHLTUInt8_t* pBuffer=NULL;
- AliHLTUInt32_t size=0;
- if ((iResult=pHLTOUT->GetDataBuffer(pBuffer, size))<0) {
- continue;
- }
- if (pBuffer==NULL || size<4) {
- AliError("invalid cluster mc data block");
- continue;
- }
- const AliHLTTPCClusterMCData* clusterMCData = reinterpret_cast<const AliHLTTPCClusterMCData*>(pBuffer);
- Int_t nLabels = (Int_t) clusterMCData->fCount;
- if (nLabels*sizeof(AliHLTTPCClusterMCLabel) + sizeof(AliHLTTPCClusterMCData) != size) {
- AliError("inconsistent cluster mc data block size, skipping block");
- continue;
- }
- // id of the cluster is
- AliHLTComponentDataType dt=kAliHLTVoidDataType;
- AliHLTUInt32_t specification=kAliHLTVoidDataSpec;
- if (pHLTOUT->GetDataBlockDescription(dt, specification)<0) {
- AliError("failed to retrieve data block description, skipping mc cluster data block ...");
- continue;
- }
- AliHLTUInt8_t slice = AliHLTTPCDefinitions::GetMinSliceNr(specification);
- AliHLTUInt8_t partition = AliHLTTPCDefinitions::GetMinPatchNr(specification);
- if (slice!=AliHLTTPCDefinitions::GetMaxSliceNr(specification) ||
- partition!=AliHLTTPCDefinitions::GetMaxPatchNr(specification)) {
- AliError(Form("can not read cluster mc data block with data of multiple partitions, skipping block %s %08x",
- AliHLTComponent::DataType2Text(dt).c_str(), specification));
- continue;
- }
- const AliHLTTPCClusterMCLabel *labels = clusterMCData->fLabels;
- for (int i=0; i<nLabels; i++) {
- AliHLTUInt32_t id=AliHLTTPCSpacePointData::GetID(slice, partition, i);
- if (tpcClusterLabels.find(id)==tpcClusterLabels.end()) {
- // new cluster
- tpcClusterLabels[id]=labels[i];
- } else {
- AliError(Form("cluster with ID 0x%08x already existing, skipping cluster %d of data block 0x%08x",
- id, i, specification));
- }
- }
- } while (pHLTOUT->SelectNextDataBlock()>=0);
- return iResult;
-}
-
-int AliHLTTPCClusterAccessHLTOUT::ReadAliHLTTPCClusterData(AliHLTOUT* pHLTOUT, TClonesArray* pClusters, const AliHLTTPCClusterMCDataList *tpcClusterLabels) const
-{
- // read cluster data from AliHLTTPCClusterData
- int iResult=0;
- if (!pHLTOUT || !pClusters) return -EINVAL;
- do {
- const AliHLTUInt8_t* pBuffer=NULL;
- AliHLTUInt32_t size=0;
- if ((iResult=pHLTOUT->GetDataBuffer(pBuffer, size))<0) {
- continue;
- }
- if (pBuffer==NULL || size<4) {
- AliError("invalid cluster data block");
- continue;
- }
- AliHLTComponentDataType dt=kAliHLTVoidDataType;
- AliHLTUInt32_t specification=kAliHLTVoidDataSpec;
- if (pHLTOUT->GetDataBlockDescription(dt, specification)<0) {
- AliError("failed to retrieve data block description, skipping mc cluster data block ...");
- continue;
- }
- const AliHLTTPCClusterData* clusterData = reinterpret_cast<const AliHLTTPCClusterData*>(pBuffer);
- Int_t nSpacepoints = (Int_t) clusterData->fSpacePointCnt;
- if (nSpacepoints*sizeof(AliHLTTPCSpacePointData) + sizeof(AliHLTTPCClusterData) != size) {
- AliError("inconsistent cluster data block size, skipping block");
- continue;
- }
- const AliHLTTPCSpacePointData *clusters = clusterData->fSpacePoints;
- int offset=pClusters->GetEntries();
- pClusters->ExpandCreate(offset+nSpacepoints);
- AliHLTUInt8_t slice = AliHLTTPCDefinitions::GetMinSliceNr(specification);
- AliHLTUInt8_t partition = AliHLTTPCDefinitions::GetMinPatchNr(specification);
- // FIXME: get first row number of outer sectors from a common definition instead using number
- unsigned rowOffset=partition<2?0:63;
- for (int i=0; i<nSpacepoints; i++) {
- if (!pClusters->At(offset+i)) continue;
- AliTPCclusterMI* pCluster=dynamic_cast<AliTPCclusterMI*>(pClusters->At(offset+i));
- if (!pCluster) {
- AliError("invalid object type, expecting AliTPCclusterMI");
- break; // this is a problem of all objects
- }
- if (clusters[i].fPadRow<rowOffset) {
- AliError(Form("invalid row number %d, expecting minimum row number %d for slice %d partition %d", clusters[i].fPadRow, rowOffset, slice, partition));
- } else {
- pCluster->SetRow(clusters[i].fPadRow-rowOffset);
- }
- pCluster->SetPad(clusters[i].fY);
- pCluster->SetTimeBin(clusters[i].fZ);
- pCluster->SetSigmaY2(clusters[i].fSigmaY2);
- pCluster->SetSigmaZ2(clusters[i].fSigmaZ2);
- pCluster->SetQ(clusters[i].fCharge);
- pCluster->SetMax(clusters[i].fQMax);
- if (tpcClusterLabels) {
- if (tpcClusterLabels->find(clusters[i].fID)!=tpcClusterLabels->end()) {
- const AliHLTTPCClusterMCWeight* mcWeights=tpcClusterLabels->find(clusters[i].fID)->second.fClusterID;
- for (int k=0; k<3; k++) {
- // TODO: sort the labels according to the weight in order to assign the most likely mc label
- // to the first component
- pCluster->SetLabel(mcWeights[k].fMCID, k);
- }
- } else {
- AliError(Form("can not find mc label of cluster with id %0x08x", clusters[i].fID));
- }
- }
- }
- if (fVerbosity>0) AliInfo(Form("converted %d cluster(s) from block %s 0x%08x", nSpacepoints, AliHLTComponent::DataType2Text(dt).c_str(), specification));
- } while (pHLTOUT->SelectNextDataBlock()>=0);
- return iResult;
-}
-
-int AliHLTTPCClusterAccessHLTOUT::ReadAliHLTTPCRawClusterData(AliHLTOUT* pHLTOUT, TClonesArray* pClusters, const AliHLTTPCClusterMCDataList *tpcClusterLabels)
-{
- // read cluster data from AliHLTTPCClusterData
-
- // FIXME: this is in large parts like ReadAliHLTTPCClusterData,
- // make a common method
- int iResult=0;
- if (!pHLTOUT || !pClusters) return -EINVAL;
- do {
- const AliHLTUInt8_t* pBuffer=NULL;
- AliHLTUInt32_t size=0;
- if ((iResult=pHLTOUT->GetDataBuffer(pBuffer, size))<0) {
- continue;
- }
- if (pBuffer==NULL || size<4) {
- AliError("invalid cluster data block");
- continue;
- }
- AliHLTComponentDataType dt=kAliHLTVoidDataType;
- AliHLTUInt32_t specification=kAliHLTVoidDataSpec;
- if (pHLTOUT->GetDataBlockDescription(dt, specification)<0) {
- AliError("failed to retrieve data block description, skipping mc cluster data block ...");
- continue;
- }
- const AliHLTTPCRawClusterData* clusterData = reinterpret_cast<const AliHLTTPCRawClusterData*>(pBuffer);
- Int_t nCount = (Int_t) clusterData->fCount;
- if (clusterData->fVersion!=0) {
- // this is encoded data of different formats
- switch (clusterData->fVersion) {
- case 1:
- iResult=ReadAliHLTTPCRawClusterDataDeflateSimple(reinterpret_cast<const AliHLTUInt8_t*>(clusterData->fClusters),
- size-sizeof(AliHLTTPCRawClusterData), nCount, specification,
- pClusters, tpcClusterLabels);
- break;
- default:
- iResult=-EPROTO;
- }
- return iResult;
- }
-
- if (nCount*sizeof(AliHLTTPCRawCluster) + sizeof(AliHLTTPCRawClusterData) != size) {
- AliError("inconsistent cluster data block size, skipping block");
- continue;
- }
- const AliHLTTPCRawCluster *clusters = clusterData->fClusters;
- int offset=pClusters->GetEntries();
- pClusters->ExpandCreate(offset+nCount);
- AliHLTUInt8_t slice = AliHLTTPCDefinitions::GetMinSliceNr(specification);
- AliHLTUInt8_t partition = AliHLTTPCDefinitions::GetMinPatchNr(specification);
- // FIXME: get first row number of outer sectors from a common definition instead using number
- int rowOffset=partition<2?0:63;
- for (int i=0; i<nCount; i++) {
- if (!pClusters->At(offset+i)) continue;
- AliTPCclusterMI* pCluster=dynamic_cast<AliTPCclusterMI*>(pClusters->At(offset+i));
- if (!pCluster) {
- AliError("invalid object type, expecting AliTPCclusterMI");
- break; // this is a problem of all objects
- }
- if (fVerbosity>1) AliInfo(Form("cluster padrow %d (slice %d partition %d)", clusters[i].GetPadRow(), slice, partition));
- if (clusters[i].GetPadRow()<rowOffset) {
- AliError(Form("invalid row number %d, expecting minimum row number %d for slice %d partition %d", clusters[i].GetPadRow(), rowOffset, slice, partition));
- } else {
- pCluster->SetRow(clusters[i].GetPadRow()-rowOffset);
- }
- pCluster->SetPad(clusters[i].GetPad());
- pCluster->SetTimeBin(clusters[i].GetTime());
- pCluster->SetSigmaY2(clusters[i].GetSigmaY2());
- pCluster->SetSigmaZ2(clusters[i].GetSigmaZ2());
- pCluster->SetQ(clusters[i].GetCharge());
- pCluster->SetMax(clusters[i].GetQMax());
- if (tpcClusterLabels) {
- UInt_t clusterID=AliHLTTPCSpacePointData::GetID(slice, partition, i);
- if (tpcClusterLabels->find(clusterID)!=tpcClusterLabels->end()) {
- const AliHLTTPCClusterMCWeight* mcWeights=tpcClusterLabels->find(clusterID)->second.fClusterID;
- for (int k=0; k<3; k++) {
- // TODO: sort the labels according to the weight in order to assign the most likely mc label
- // to the first component
- pCluster->SetLabel(mcWeights[k].fMCID, k);
- }
- } else {
- AliError(Form("can not find mc label of cluster with id %0x08x", clusterID));
- }
- }
- }
- if (fVerbosity>0) AliInfo(Form("converted %d cluster(s) from block %s 0x%08x", nCount, AliHLTComponent::DataType2Text(dt).c_str(), specification));
- } while (pHLTOUT->SelectNextDataBlock()>=0);
- return iResult;
-}
-
-int AliHLTTPCClusterAccessHLTOUT::ReadRemainingClustersCompressed(AliHLTOUT* pHLTOUT, TClonesArray* pClusters, const AliHLTTPCClusterMCDataList *tpcClusterLabels)
-{
- // read cluster data from AliHLTTPCClusterData
- int iResult=0;
- if (!pHLTOUT || !pClusters) return -EINVAL;
- do {
- const AliHLTUInt8_t* pBuffer=NULL;
- AliHLTUInt32_t size=0;
- if ((iResult=pHLTOUT->GetDataBuffer(pBuffer, size))<0) {
- continue;
- }
- if (pBuffer==NULL || size<4) {
- AliError("invalid cluster data block");
- continue;
- }
- AliHLTComponentDataType dt=kAliHLTVoidDataType;
- AliHLTUInt32_t specification=kAliHLTVoidDataSpec;
- if (pHLTOUT->GetDataBlockDescription(dt, specification)<0) {
- AliError("failed to retrieve data block description, skipping mc cluster data block ...");
- continue;
- }
- const AliHLTTPCRawClusterData* clusterData = reinterpret_cast<const AliHLTTPCRawClusterData*>(pBuffer);
- Int_t nCount = (Int_t) clusterData->fCount;
-
- // this is encoded data of different formats
- switch (clusterData->fVersion) {
- case 1:
- iResult=ReadAliHLTTPCRawClusterDataDeflateSimple(reinterpret_cast<const AliHLTUInt8_t*>(clusterData->fClusters),
- size-sizeof(AliHLTTPCRawClusterData), nCount, specification,
- pClusters, tpcClusterLabels);
- break;
- default:
- AliError(Form("invalid cluster format version %d", clusterData->fVersion));
- iResult=-EPROTO;
- }
-
- if (fVerbosity>0) AliInfo(Form("converted %d cluster(s) from block %s 0x%08x", nCount, AliHLTComponent::DataType2Text(dt).c_str(), specification));
- } while (pHLTOUT->SelectNextDataBlock()>=0 && iResult>=0);
-
- return iResult;
-}
-
-int AliHLTTPCClusterAccessHLTOUT::ReadAliHLTTPCRawClusterDataDeflateSimple(const AliHLTUInt8_t* pData, int dataSize,
- int nofClusters, AliHLTUInt32_t specification,
- TClonesArray* pClusters,
- const AliHLTTPCClusterMCDataList *tpcClusterLabels)
-{
- // read cluster data from AliHLTTPCClusterData
-
- // FIXME: quick implementation to read the compressed cluster data from HLTOUT
- // the data definition below is the same as in AliHLTTPCDataCompressionComponent
- // but needs to be moved to a common class (AliHLTTPCDefinitions?)
- // Think about a decoder class supporting iterator objects for various types
- // of cluster data
- int iResult=0;
- if (!pData || !pClusters) return -EINVAL;
- AliHLTDataInflater inflater;
- if ((iResult=inflater.InitBitDataInput(pData, dataSize))<0) {
- return iResult;
- }
-
- int offset=pClusters->GetEntries();
- pClusters->ExpandCreate(offset+nofClusters);
- AliHLTUInt8_t slice = AliHLTTPCDefinitions::GetMinSliceNr(specification);
- AliHLTUInt8_t partition = AliHLTTPCDefinitions::GetMinPatchNr(specification);
- // the compressed format stores the difference of the local row number in
- // the partition to the row of the last cluster
- // add the first row in the partition to get global row number
- // offline uses row number in physical sector, inner sector consists of
- // partitions 0 and 1, outer sector of partition 2-5
- int rowOffset=AliHLTTPCTransform::GetFirstRow(partition)-(partition<2?0:AliHLTTPCTransform::GetFirstRow(2));
-
- int parameterId=0;
- int outClusterCnt=0;
- AliHLTUInt8_t switchBit=0;
- AliHLTUInt64_t value=0;
- AliTPCclusterMI* pCluster=NULL;
- AliHLTUInt32_t lastPadRow=0;
- while (outClusterCnt<nofClusters && inflater.InputBit(switchBit)) {
- const AliHLTTPCDefinitions::AliClusterParameter& parameter
- =AliHLTTPCDefinitions::fgkClusterParameterDefinitions[parameterId];
- // in mode DeflaterSimple, the optional parameter of the cluster parameter definition
- // corresponds to the number bits of the reduced format
- if (!inflater.InputBits(value, switchBit?parameter.fBitLength:parameter.fOptional)) {
- break;
- }
-
- if (!pCluster) {
- if (!pClusters->At(offset+outClusterCnt)) {
- // here we should not get anymore because of the condition outClusterCnt<nofClusters
- return -ENOSPC;
- }
- pCluster=dynamic_cast<AliTPCclusterMI*>(pClusters->At(offset+outClusterCnt));
- if (!pCluster) {
- AliError("invalid object type, expecting AliTPCclusterMI");
- iResult=-EBADF; // this is a problem of all objects
- break;
- }
- }
- switch (parameterId) {
- case AliHLTTPCDefinitions::kPadRow:
- {pCluster->SetRow(value+lastPadRow+rowOffset); lastPadRow+=value;break;}
- case AliHLTTPCDefinitions::kPad:
- {float pad=value; pad/=parameter.fScale; pCluster->SetPad(pad); break;}
- case AliHLTTPCDefinitions::kTime:
- {float time=value; time/=parameter.fScale; pCluster->SetTimeBin(time); break;}
- case AliHLTTPCDefinitions::kSigmaY2:
- {float sigmaY2=value; sigmaY2/=parameter.fScale; pCluster->SetSigmaY2(sigmaY2); break;}
- case AliHLTTPCDefinitions::kSigmaZ2:
- {float sigmaZ2=value; sigmaZ2/=parameter.fScale; pCluster->SetSigmaZ2(sigmaZ2); break;}
- case AliHLTTPCDefinitions::kCharge:
- {pCluster->SetQ(value); break;}
- case AliHLTTPCDefinitions::kQMax:
- {pCluster->SetMax(value); break;}
- }
- if (parameterId>=AliHLTTPCDefinitions::kLast) {
- // switch to next cluster
- if (tpcClusterLabels) {
- UInt_t clusterID=AliHLTTPCSpacePointData::GetID(slice, partition, outClusterCnt);
- if (tpcClusterLabels->find(clusterID)!=tpcClusterLabels->end()) {
- const AliHLTTPCClusterMCWeight* mcWeights=tpcClusterLabels->find(clusterID)->second.fClusterID;
- for (int k=0; k<3; k++) {
- // TODO: sort the labels according to the weight in order to assign the most likely mc label
- // to the first component
- pCluster->SetLabel(mcWeights[k].fMCID, k);
- }
- } else {
- AliError(Form("can not find mc label of cluster with id 0x%08x", clusterID));
- }
- }
- outClusterCnt++;
- pCluster=NULL;
- parameterId=-1;
- }
- parameterId++;
- }
- inflater.Pad8Bits();
- if (inflater.InputBit(switchBit)) {
- AliWarning("format error of compressed clusters, there is more data than expected");
- }
- inflater.CloseBitDataInput();
- if (iResult>=0 && nofClusters!=outClusterCnt) {
- // is this a Fatal?
- AliError(Form("error reading compressed cluster format: expected %d, read only %d cluster(s)", nofClusters, outClusterCnt));
- return -EPROTO;
- }
- return iResult;
-}
-