// $Id$
//**************************************************************************
-//* This file is property of and copyright by the ALICE HLT Project *
+//* This file is property of and copyright by the *
//* ALICE Experiment at CERN, All rights reserved. *
//* *
//* Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no> *
//* provided "as is" without express or implied warranty. *
//**************************************************************************
-/** @file AliHLTDataBuffer.cxx
- @author Matthias Richter
- @date
- @brief Handling of Data Buffers for HLT components.
-*/
-
-#if __GNUC__>= 3
-using namespace std;
-#endif
+// @file AliHLTDataBuffer.cxx
+// @author Matthias Richter
+// @date
+// @brief Handling of Data Buffers for HLT components.
+// @note Only used in the AliRoot framework
#include "AliHLTDataBuffer.h"
#include "AliHLTConsumerDescriptor.h"
//#include <string>
//#include "AliHLTSystem.h"
-typedef vector<AliHLTDataBuffer::AliHLTDataSegment> AliHLTDataSegmentList;
-typedef vector<AliHLTDataBuffer::AliHLTRawBuffer*> AliHLTRawBufferPList;
+using std::cout;
/** ROOT macro for the implementation of ROOT specific class methods */
ClassImp(AliHLTDataBuffer)
}
int AliHLTDataBuffer::fgNofInstances=0;
-AliHLTRawBufferPList AliHLTDataBuffer::fgFreeBuffers;
-AliHLTRawBufferPList AliHLTDataBuffer::fgActiveBuffers;
+AliHLTDataBuffer::AliHLTRawBufferPList AliHLTDataBuffer::fgFreeBuffers;
+AliHLTDataBuffer::AliHLTRawBufferPList AliHLTDataBuffer::fgActiveBuffers;
AliHLTUInt32_t AliHLTDataBuffer::fgMargin=1024;
AliHLTLogging AliHLTDataBuffer::fgLogging;
const Int_t AliHLTDataBuffer::fgkSafetyPatternSize=16;
AliHLTDataBuffer::~AliHLTDataBuffer()
{
// see header file for function documentation
+ CleanupConsumerList();
+
if (--fgNofInstances<=0) {
DeleteRawBuffers();
}
- CleanupConsumerList();
}
int AliHLTDataBuffer::SetConsumer(AliHLTComponent* pConsumer)
return iResult;
}
-int AliHLTDataBuffer::FindMatchingDataSegments(const AliHLTComponent* pConsumer, vector<AliHLTDataBuffer::AliHLTDataSegment>& tgtList)
+int AliHLTDataBuffer::FindMatchingDataSegments(const AliHLTComponent* /*pConsumer*/,
+ AliHLTDataSegmentList& tgtList)
{
// see header file for function documentation
int iResult=0;
tgtList.insert(tgtList.begin(), fForwardedSegments.begin(), fForwardedSegments.end());
iResult=tgtList.size();
return iResult;
-
- if (pConsumer) {
- AliHLTComponentDataTypeList dtlist;
- ((AliHLTComponent*)pConsumer)->GetInputDataTypes(dtlist);
- AliHLTDataSegmentList::iterator segment=fSegments.begin();
- while (segment!=fSegments.end()) {
- AliHLTComponentDataTypeList::iterator type=dtlist.begin();
- while (type!=dtlist.end()) {
- if ((*segment).fDataType==(*type)) {
- tgtList.push_back(*segment);
- iResult++;
- break;
- }
- type++;
- }
- segment++;
- }
- } else {
- iResult=-EINVAL;
- }
- return iResult;
}
int AliHLTDataBuffer::Subscribe(const AliHLTComponent* pConsumer, AliHLTComponentBlockDataList& blockDescList)
if (fpBuffer) {
if (*fpBuffer==pTgt) {
AliHLTDataBuffer::AliHLTDataSegment segment;
+ AliHLTUInt32_t maxSize=0;
for (int i=0; i<iSize; i++) {
// This function has to model the behavior of PubSub
// For output blocks only the fOffset value is used, this must be the offset
segment.fSpecification=arrayBlockData[i].fSpecification;
fSegments.push_back(segment);
HLTDebug("set segment %s with size %d at offset %d", AliHLTComponent::DataType2Text(segment.fDataType).data(), segment.fSegmentSize, segment.fSegmentOffset);
+
+ // find the actual size of the data
+ if ((arrayBlockData[i].fOffset!=~(AliHLTUInt32_t)0) &&
+ arrayBlockData[i].fOffset+arrayBlockData[i].fSize>maxSize) {
+ maxSize=arrayBlockData[i].fOffset+arrayBlockData[i].fSize;
+ }
} else {
HLTError("block data specification %#d (%s) exceeds size of data buffer", i, AliHLTComponent::DataType2Text(arrayBlockData[i].fDataType).data());
HLTError("block offset=%d, block size=%d, buffer size=%d", arrayBlockData[i].fOffset, arrayBlockData[i].fSize, fpBuffer->GetUsedSize());
iResult=-ERANGE;
}
}
+ // to be enabled if unit test is ready
+ iResult=SetRawBufferDataSize(fpBuffer, maxSize);
} else {
HLTError("this data buffer (%p) does not match the internal data buffer %p of raw buffer %p", pTgt, fpBuffer->GetPointer(), fpBuffer);
iResult=-EINVAL;
return iResult;
}
-int AliHLTDataBuffer::GetNofSegments()
+int AliHLTDataBuffer::GetNofSegments() const
{
// see header file for function documentation
int iResult=fSegments.size() + fForwardedSegments.size();
return iResult;
}
-int AliHLTDataBuffer::GetNofConsumers()
+int AliHLTDataBuffer::GetNofConsumers() const
{
// see header file for function documentation
int iResult=fConsumers.size() + GetNofActiveConsumers() + fReleasedConsumers.size();
return iResult;
}
-int AliHLTDataBuffer::GetNofPendingConsumers()
+int AliHLTDataBuffer::GetNofPendingConsumers() const
{
// see header file for function documentation
int iResult=fConsumers.size();
return iResult;
}
-int AliHLTDataBuffer::GetNofActiveConsumers()
+int AliHLTDataBuffer::GetNofActiveConsumers() const
{
// see header file for function documentation
int iResult=fActiveConsumers.size();
// see header file for function documentation
AliHLTRawBuffer* pRawBuffer=NULL;
unsigned int reqSize=size+fgkSafetyPatternSize;
- AliHLTRawBufferPList::iterator buffer=fgFreeBuffers.begin();
- while (buffer!=fgFreeBuffers.end() && pRawBuffer==NULL) {
- if ((*buffer)->CheckSize(reqSize)) {
- // assign this element
- pRawBuffer=*buffer;
- pRawBuffer->UseBuffer(size);
- fgFreeBuffers.erase(buffer);
- fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "raw buffer container %p provided for request of %d bytes (total %d available in buffer %p)", pRawBuffer, size, pRawBuffer->GetTotalSize(), pRawBuffer->GetPointer());
- fgActiveBuffers.push_back(pRawBuffer);
- break;
- }
- buffer++;
- }
- if (pRawBuffer==NULL) {
- // no buffer found, create a new one
- pRawBuffer=new AliHLTRawBuffer(reqSize);
- if (pRawBuffer) {
- if (pRawBuffer->GetPointer()) {
- pRawBuffer->UseBuffer(size);
- fgActiveBuffers.push_back(pRawBuffer);
- fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "new raw buffer %p of size %d created (container %p)", pRawBuffer->GetPointer(), pRawBuffer->GetTotalSize(), pRawBuffer);
- } else {
- delete pRawBuffer;
- pRawBuffer=NULL;
- fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "memory allocation failed");
- }
- } else {
- fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "memory allocation failed");
- }
+ pRawBuffer=AliHLTDataBuffer::AliHLTRawPage::GlobalAlloc(reqSize);
+ if (pRawBuffer) {
+ pRawBuffer->UseBuffer(size);
}
if (pRawBuffer!=NULL && fgkSafetyPatternSize>0) {
//fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "writing safety pattern to %p offset %d", (*buffer)->GetPointer(), (*buffer)->GetUsedSize());
return pRawBuffer;
}
+int AliHLTDataBuffer::SetRawBufferDataSize(AliHLTRawBuffer* pBuffer, AliHLTUInt32_t size) const
+{
+ // see header file for function documentation
+ int iResult=0;
+ if (!pBuffer) return -EINVAL;
+ if (size>pBuffer->GetUsedSize()) {
+ HLTError("indicated data size %d exceeds data buffer %p (%d)", size, pBuffer->GetPointer(), pBuffer->GetUsedSize());
+ return -ENOSPC;
+ }
+ if (fgkSafetyPatternSize>0) {
+ if (pBuffer->CheckPattern(fgkSafetyPattern, fgkSafetyPatternSize)) {
+ HLTError("potential memory corruption: component has written beyond end of data buffer %p size %d", pBuffer->GetPointer(), pBuffer->GetUsedSize());
+ }
+ }
+ // shrink the buffer and write new pattern at the end
+ AliHLTDataBuffer::AliHLTRawPage* rawpage=AliHLTDataBuffer::AliHLTRawPage::FindPage(pBuffer);
+ if (rawpage) {
+ pBuffer->UseBuffer(size);
+ if (rawpage->SetSize(pBuffer, size+fgkSafetyPatternSize)==0) {
+ // nothing to do
+ } else {
+ fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::SetRawBufferDataSize", "data buffer handling", "failed to set size for raw buffer %p", pBuffer);
+ iResult=-EFAULT;
+ }
+ } else {
+ fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::SetRawBufferDataSize", "data buffer handling", "can not find raw page for buffer %p", pBuffer);
+ iResult=-ENOENT;
+ }
+ if (fgkSafetyPatternSize>0) {
+ pBuffer->WritePattern(fgkSafetyPattern, fgkSafetyPatternSize);
+ }
+ return iResult;
+}
+
int AliHLTDataBuffer::ReleaseRawBuffer(AliHLTRawBuffer* pBuffer)
{
// see header file for function documentation
int iResult=0;
if (pBuffer) {
- AliHLTRawBufferPList::iterator buffer=fgActiveBuffers.begin();
- while (buffer!=fgActiveBuffers.end() && (*buffer)!=pBuffer) {
- buffer++;
- }
- if (buffer!=fgActiveBuffers.end()) {
+ AliHLTDataBuffer::AliHLTRawPage* rawpage=AliHLTDataBuffer::AliHLTRawPage::FindPage(pBuffer);
+ if (rawpage)
+ {
if (fgkSafetyPatternSize>0) {
- //fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "comparing safety pattern at %p offset %d", (*buffer)->GetPointer(), reinterpret_cast<AliHLTUInt32_t>(*buffer));
- if ((*buffer)->CheckPattern(fgkSafetyPattern, fgkSafetyPatternSize)) {
- fgLogging.Logging(kHLTLogFatal, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "component has written beyond end of data buffer %p size %d", (*buffer)->GetPointer(), (*buffer)->GetUsedSize());
+ //fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "comparing safety pattern at %p offset %d", pBuffer->GetPointer(), reinterpret_cast<AliHLTUInt32_t>(pBuffer));
+ if ((pBuffer)->CheckPattern(fgkSafetyPattern, fgkSafetyPatternSize)) {
+ fgLogging.Logging(kHLTLogFatal, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "component has written beyond end of data buffer %p size %d", pBuffer->GetPointer(), pBuffer->GetUsedSize());
}
}
- (*buffer)->Reset();
- fgFreeBuffers.push_back(*buffer);
- fgActiveBuffers.erase(buffer);
+ pBuffer->Reset();
+ if (rawpage->Free(pBuffer)==0) {
+ } else {
+ fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "failed to release raw buffer %p", pBuffer);
+ }
} else {
- fgLogging.Logging(kHLTLogWarning, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "can not find raw buffer container %p in the list of active containers", pBuffer);
+ fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "can not find raw page for buffer %p", pBuffer);
iResult=-ENOENT;
}
} else {
#ifdef ALIHLTSYSTEM_PROFILING
iTotalSize+=(*buffer)->GetTotalSize();
#endif //ALIHLTSYSTEM_PROFILING
- fgLogging.Logging(kHLTLogWarning, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "request to delete active raw buffer container (raw buffer %p, size %d)", (*buffer)->GetPointer(), (*buffer)->GetTotalSize());
+ fgLogging.Logging(kHLTLogWarning, "AliHLTDataBuffer::DeleteRawBuffer", "data buffer handling", "request to delete active raw buffer container (raw buffer %p, size %d)", (*buffer)->GetPointer(), (*buffer)->GetTotalSize());
delete *buffer;
fgActiveBuffers.erase(buffer);
}
#ifdef ALIHLTSYSTEM_PROFILING
- fgLogging.Logging(kHLTLogImportant, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "Total memory allocation: %d byte in %d buffers", iTotalSize, iCount);
+ fgLogging.Logging(kHLTLogImportant, "AliHLTDataBuffer::DeleteRawBuffer", "data buffer handling", "Total memory allocation: %d byte in %d buffers", iTotalSize, iCount);
#endif //ALIHLTSYSTEM_PROFILING
return iResult;
}
{
// see header file for function documentation
int iResult=0;
- int iFree=0;
- int iActive=0;
- AliHLTRawBufferPList::iterator buffer;;
- for (buffer=fgFreeBuffers.begin(); buffer!=fgFreeBuffers.end(); buffer++) {
- iFree+=(*buffer)->GetTotalSize();
- }
- for (buffer=fgActiveBuffers.begin(); buffer!=fgActiveBuffers.end(); buffer++) {
- iActive+=(*buffer)->GetTotalSize();
+ int nofPages=0;
+ AliHLTUInt32_t totalSize=0;
+ for (AliHLTDataBuffer::AliHLTRawPage* rawpage=AliHLTDataBuffer::AliHLTRawPage::NextPage(NULL);
+ rawpage!=NULL;
+ rawpage=AliHLTDataBuffer::AliHLTRawPage::NextPage(rawpage)) {
+ nofPages++;
+ totalSize+=rawpage->Size();
+ if (fgLogging.CheckFilter(kHLTLogDebug)) rawpage->Print("");
}
- fgLogging.Logging(kHLTLogInfo, "AliHLTDataBuffer::PrintStatistics", "data buffer handling", "Total memory allocation: %d byte; %d free buffers (%d byte) - %d active buffers (%d byte) ", iFree+iActive, fgFreeBuffers.size(), iFree, fgActiveBuffers.size(), iActive);
+ //if (rawpage) rawpage->Print("global");
+ fgLogging.Logging(kHLTLogInfo, "AliHLTDataBuffer::PrintStatistics", "data buffer handling", "total number of memory pages: %d total size %d", nofPages, totalSize);
+
return iResult;
}
}
AliHLTDataBuffer::AliHLTRawBuffer::AliHLTRawBuffer(AliHLTUInt32_t size)
- :
- fSize(0),
- fTotalSize(size),
- fPtr(static_cast<AliHLTUInt8_t*>(malloc(size))),
- fLastEventCount(0)
+ : fSize(0)
+ , fTotalSize(size)
+ , fExternalPtr(NULL)
+ , fPtr(static_cast<AliHLTUInt8_t*>(malloc(size)))
+ , fLastEventCount(0)
{
// see header file for class documentation
// or
}
}
+AliHLTDataBuffer::AliHLTRawBuffer::AliHLTRawBuffer(AliHLTUInt32_t size, AliHLTUInt8_t* buffer)
+ : fSize(0)
+ , fTotalSize(size)
+ , fExternalPtr(buffer)
+ , fPtr(fExternalPtr)
+ , fLastEventCount(0)
+{
+ // see header file for class documentation
+}
+
AliHLTDataBuffer::AliHLTRawBuffer::~AliHLTRawBuffer()
{
- if (fPtr) {
+ // see header file for class documentation
+ if (fExternalPtr==NULL && fPtr) {
free(fPtr);
}
fPtr=NULL;
return fPtr == static_cast<AliHLTUInt8_t*>(ptr);
}
+int AliHLTDataBuffer::AliHLTRawBuffer::operator<(void* ptr) const
+{
+ // see header file for function documentation
+ int iResult=fPtr < static_cast<AliHLTUInt8_t*>(ptr);
+ //printf("%p: %p <= %p (%d)\n", this, fPtr, ptr, iResult);
+ return iResult;
+}
+
int AliHLTDataBuffer::AliHLTRawBuffer::operator<=(void* ptr) const
{
// see header file for function documentation
return static_cast<int>(static_cast<AliHLTUInt8_t*>(ptr)-fPtr);
}
-AliHLTUInt8_t* AliHLTDataBuffer::AliHLTRawBuffer::UseBuffer(AliHLTUInt32_t size)
+int AliHLTDataBuffer::AliHLTRawBuffer::operator<(const AliHLTRawBuffer& op) const
+{
+ // see header file for function documentation
+ return (fPtr+fSize < op.fPtr);
+}
+
+int AliHLTDataBuffer::AliHLTRawBuffer::operator<=(const AliHLTRawBuffer& op) const
+{
+ // see header file for function documentation
+ return (fPtr+fSize <= op.fPtr);
+}
+
+int AliHLTDataBuffer::AliHLTRawBuffer::operator>(const AliHLTRawBuffer& op) const
{
// see header file for function documentation
- if (size>0 && fTotalSize>=size) {
+ return (fPtr >= op.fPtr+op.fSize);
+}
+
+AliHLTUInt8_t* AliHLTDataBuffer::AliHLTRawBuffer::UseBuffer(AliHLTUInt32_t size)
+{
+ // mark a portion of the buffer as used
+ if (fTotalSize>=size) {
fSize=size;
fLastEventCount=AliHLTDataBuffer::fgEventCount;
- return fPtr;
+ // only return pointer if there is a portion of the buffer used
+ if (size>0) return fPtr;
+ }
+ return NULL;
+}
+
+AliHLTDataBuffer::AliHLTRawBuffer* AliHLTDataBuffer::AliHLTRawBuffer::Split(AliHLTUInt32_t size)
+{
+ // split a buffer at specified size
+ // only possible for buffers with external memory
+ if (fTotalSize>size &&
+ (fSize==0 || fSize<=size) && // used size must fit into the first part
+ fExternalPtr!=NULL) {
+ AliHLTRawBuffer* part2=new AliHLTRawBuffer(fTotalSize-size, fPtr+size);
+ if (part2) {
+ fTotalSize=size;
+ }
+ return part2;
}
return NULL;
}
}
return iResult;
}
+
+int AliHLTDataBuffer::AliHLTRawBuffer::Merge(const AliHLTDataBuffer::AliHLTRawBuffer& neighbor)
+{
+ // Merge buffer with neighboring buffer.
+ // Only possible if the buffers are consecutive with out any gap.
+
+ if (!fExternalPtr || !neighbor.fExternalPtr) return -EPERM;
+
+ if (neighbor.fTotalSize==0 &&
+ fPtr < neighbor.fPtr &&
+ fPtr+fTotalSize > neighbor.fPtr) {
+ // special case for a buffer of zero size embedded into this buffer
+ // nothing to do
+ return 0;
+ }
+ if (fTotalSize==0 &&
+ neighbor.fPtr < fPtr &&
+ neighbor.fPtr+neighbor.fTotalSize > fPtr) {
+ // special case for this buffer of size zero embedded into another buffer
+ fPtr=neighbor.fPtr;
+ fExternalPtr=fPtr;
+ fTotalSize+=neighbor.fTotalSize;
+ fSize=0;
+ return 0;
+ }
+ if (fPtr+fTotalSize == neighbor.fPtr) {
+ fTotalSize+=neighbor.fTotalSize;
+ fSize=0;
+ return 0;
+ }
+ if (fPtr == neighbor.fPtr+neighbor.fTotalSize) {
+ fPtr=neighbor.fPtr;
+ fExternalPtr=fPtr;
+ fTotalSize+=neighbor.fTotalSize;
+ fSize=0;
+ return 0;
+ }
+ return -EINVAL;
+}
+
+void AliHLTDataBuffer::AliHLTRawBuffer::Print(const char* option) const
+{
+ /// print buffer information
+ if (strcmp(option, "min")!=0) {
+ cout << "************* AliHLTRawBuffer status ***********" << endl;
+ }
+ printf(" %p: buffer %p%s size %d used %d\n", this, fPtr, fExternalPtr?" (external)":"", fTotalSize, fSize); fflush(stdout);
+}
+
+AliHLTDataBuffer::AliHLTRawPage::AliHLTRawPage(AliHLTUInt32_t pagesize)
+ : fSize(pagesize)
+ , fPtr(static_cast<AliHLTUInt8_t*>(malloc(pagesize)))
+ , fFreeBuffers()
+ , fUsedBuffers()
+{
+ // constructor
+ if (fPtr) {
+ fFreeBuffers.push_back(new AliHLTRawBuffer(fSize, fPtr));
+ } else {
+ fSize=0;
+ }
+}
+
+AliHLTDataBuffer::AliHLTRawPage::~AliHLTRawPage()
+{
+ // destructor
+ if (IsUsed()) {
+ // do not free if the resources have not been completely freed
+ HLTError("memory mismatch: not all allocated intances have been released");
+ } else {
+ if (IsFragmented()) {
+ HLTWarning("page still fragmented");
+ }
+ AliHLTRawBufferPList::iterator element=fFreeBuffers.begin();
+ while (element!=fFreeBuffers.end()) {
+ if (*element) delete *element;
+ element=fFreeBuffers.erase(element);
+ }
+ if (fPtr) {
+ free(fPtr);
+ }
+ fPtr=NULL;
+ fSize=0;
+ }
+}
+
+AliHLTDataBuffer::AliHLTRawBuffer* AliHLTDataBuffer::AliHLTRawPage::Alloc(AliHLTUInt32_t size)
+{
+ /// alloc a buffer of specified size
+ if (fFreeBuffers.size()==0) return NULL;
+
+ for (AliHLTRawBufferPList::iterator iter=fFreeBuffers.begin();
+ iter!=fFreeBuffers.end();
+ iter++) {
+ if ((*iter)->GetTotalSize()==size) {
+ AliHLTRawBuffer* thisbuffer=*iter;
+ fFreeBuffers.erase(iter);
+ fUsedBuffers.push_back(thisbuffer);
+ return thisbuffer;
+ } else if ((*iter)->GetTotalSize()>size) {
+ AliHLTRawBuffer* thisbuffer=*iter;
+ AliHLTRawBuffer* newbuffer=thisbuffer->Split(size);
+ if (newbuffer) {
+ *iter=newbuffer;
+ fUsedBuffers.push_back(thisbuffer);
+ return thisbuffer;
+ } else {
+ HLTWarning("failed to alloc raw buffer: cannot split raw buffer %p of size %d (used %d) at size %d", *iter, (*iter)->GetTotalSize(), (*iter)->GetUsedSize(), size);
+ }
+ }
+ }
+ return NULL;
+}
+
+int AliHLTDataBuffer::AliHLTRawPage::Free(AliHLTRawBuffer* pBuffer)
+{
+ /// free a buffer and merge consecutive free buffers
+ int iResult=0;
+ for (AliHLTRawBufferPList::iterator iter=fUsedBuffers.begin();
+ iter!=fUsedBuffers.end() && iResult>=0;
+ iter++) {
+ if ((*iter)==pBuffer) {
+ fUsedBuffers.erase(iter);
+ AliHLTRawBufferPList::iterator prev=fFreeBuffers.begin();
+ for (; prev!=fFreeBuffers.end() && iResult>=0; prev++) {
+ if ((*pBuffer)<(*(*prev)) ||
+ ((*prev)->GetTotalSize()==0 && pBuffer->GetPointer()<=(*prev)->GetPointer() && (*prev)->GetPointer()<=pBuffer->GetPointer()+pBuffer->GetTotalSize())) {
+ // check consecutive buffers
+ if ((*(*prev)) == (pBuffer->GetPointer()+pBuffer->GetTotalSize()) ||
+ ((*prev)->GetTotalSize()==0 && pBuffer->GetPointer()<=(*prev)->GetPointer() && (*prev)->GetPointer()<=pBuffer->GetPointer()+pBuffer->GetTotalSize())) {
+ // the buffer to be released has a consecutive free buffer -> merge them
+ if ((iResult=pBuffer->Merge(*(*prev)))>=0) {
+ delete *prev;
+ *prev=pBuffer;
+ } else {
+ HLTError("failed to merge consecutive/overlapping buffers %p and %p", pBuffer, (*prev));
+ pBuffer->Print("");
+ (*prev)->Print("");
+ }
+ break;
+ }
+ fFreeBuffers.insert(prev, pBuffer);
+ break;
+ }
+ if ((*pBuffer)>(*(*prev)) ||
+ (pBuffer->GetTotalSize()==0 && (*prev)->GetPointer()<=pBuffer->GetPointer() && pBuffer->GetPointer()<=(*prev)->GetPointer()+(*prev)->GetTotalSize())) {
+ // check consecutive buffers
+ if ((*pBuffer) == ((*prev)->GetPointer()+(*prev)->GetTotalSize())||
+ (pBuffer->GetTotalSize()==0 && (*prev)->GetPointer()<=pBuffer->GetPointer() && pBuffer->GetPointer()<=(*prev)->GetPointer()+(*prev)->GetTotalSize())) {
+ // the buffer to be released is consecutive to a free buffer -> merge them
+ if ((iResult=pBuffer->Merge(*(*prev)))>=0) {
+ AliHLTRawBufferPList::iterator succ=prev+1;
+ delete *prev;
+ *prev=pBuffer;
+ // check if the buffer and the following one are consecutive
+ if (succ!=fFreeBuffers.end() &&
+ (*(*succ)) == (pBuffer->GetPointer()+pBuffer->GetTotalSize())) {
+ if ((iResult=pBuffer->Merge(*(*succ)))>=0) {
+ delete *succ;
+ fFreeBuffers.erase(succ);
+ }
+ }
+ }
+ break;
+ }
+ }
+ }
+ if (prev==fFreeBuffers.end()) {
+ fFreeBuffers.push_back(pBuffer);
+ }
+
+ // merge consecutive free buffers
+ prev=fFreeBuffers.begin();
+ for (AliHLTRawBufferPList::iterator current=prev+1; current!=fFreeBuffers.end() && iResult>=0; ) {
+ // check if the buffer is embedded into the previous one
+ if ((*current)->GetTotalSize()==0 && (*prev)->GetPointer()<=(*current)->GetPointer() && (*current)->GetPointer()<(*prev)->GetPointer()+(*prev)->GetTotalSize()) {
+ if ((iResult=(*prev)->Merge(*(*current)))>=0) {
+ current=fFreeBuffers.erase(current);
+ continue;
+ } else {
+ HLTError("failed to merge embedded zero length buffer into preceeding buffer");
+ Print("");
+ }
+ }
+ // check if the buffer is consecutive to the previous one
+ if ((*(*current)) == ((*prev)->GetPointer()+(*prev)->GetTotalSize())) {
+ if ((iResult=(*prev)->Merge(*(*current)))>=0) {
+ current=fFreeBuffers.erase(current);
+ continue;
+ } else {
+ HLTError("failed to merge consecutive free buffers");
+ Print("");
+ }
+ }
+ prev=current++;
+ }
+
+ // buffer was part of this page
+ return 0;
+ }
+ }
+ // buffer not found in this page
+ return 1;
+}
+
+int AliHLTDataBuffer::AliHLTRawPage::SetSize(const AliHLTDataBuffer::AliHLTRawBuffer* pBuffer, AliHLTUInt32_t size)
+{
+ /// set the size of a raw buffer and release the remaining part
+ int iResult=0;
+ for (AliHLTRawBufferPList::iterator iter=fUsedBuffers.begin();
+ iter!=fUsedBuffers.end() && iResult>=0;
+ iter++) {
+ if ((*iter)==pBuffer) { // buffer was part of this page
+ if ((*iter)->GetTotalSize()==size) return 0;
+ if ((*iter)->GetTotalSize()<size) {
+ HLTError("%d exceeds total size of buffer %p (%d used %d)\n", size, *iter, (*iter)->GetTotalSize(), (*iter)->GetUsedSize());
+ return -ENOSPC;
+ }
+ AliHLTDataBuffer::AliHLTRawBuffer* freespace=(*iter)->Split(size);
+ if (freespace) {
+ fUsedBuffers.push_back(freespace);
+ Free(freespace);
+ } else {
+ HLTWarning("failed to relase unused memory: cannot split raw buffer %p of size %d (used %d) at size %d", *iter, (*iter)->GetTotalSize(), (*iter)->GetUsedSize(), size);
+ }
+ return 0;
+ }
+ }
+ // buffer not found in this page
+ return 1;
+}
+
+bool AliHLTDataBuffer::AliHLTRawPage::HasBuffer(const AliHLTDataBuffer::AliHLTRawBuffer* pBuffer)
+{
+ /// check if the buffer is in this page
+ for (AliHLTRawBufferPList::iterator iter=fUsedBuffers.begin();
+ iter!=fUsedBuffers.end();
+ iter++) {
+ if ((*iter)==pBuffer) { // buffer was part of this page
+ return true;
+ }
+ }
+ // buffer not found in this page
+ return false;
+}
+
+AliHLTUInt32_t AliHLTDataBuffer::AliHLTRawPage::Capacity() const
+{
+ /// get max available contiguous buffer
+ AliHLTUInt32_t capacity=0;
+ for (unsigned i=0; i<fFreeBuffers.size(); i++) {
+ if (fFreeBuffers[i]->GetTotalSize()>capacity)
+ capacity=fFreeBuffers[i]->GetTotalSize();
+ }
+ return capacity;
+}
+
+void AliHLTDataBuffer::AliHLTRawPage::Print(const char* option)
+{
+ /// print page information
+ if (strcmp(option, "global")==0) {
+ cout << "number of global pages: " << fgGlobalPages.size() << endl;
+ for (AliHLTRawPage* rawpage=NextPage(NULL);
+ rawpage!=NULL;
+ rawpage=NextPage(rawpage)) {
+ rawpage->Print("");
+ }
+ return;
+ }
+ cout << "************* AliHLTRawPage status ***********" << endl;
+ cout << " instance " << this << endl;
+ printf(" buffer %p size %d", fPtr, fSize);
+ cout << " used buffers: " << fUsedBuffers.size() << endl;
+ AliHLTRawBufferPList::iterator iter=fUsedBuffers.begin();
+ for (; iter!=fUsedBuffers.end(); iter++) {
+ cout << " "; (*iter)->Print("min");
+ }
+ cout << " free buffers: " << fFreeBuffers.size() << endl;
+ iter=fFreeBuffers.begin();
+ for (; iter!=fFreeBuffers.end(); iter++) {
+ cout << " "; (*iter)->Print("min");
+ }
+}
+
+
+vector<AliHLTDataBuffer::AliHLTRawPage*> AliHLTDataBuffer::AliHLTRawPage::fgGlobalPages;
+
+AliHLTUInt32_t AliHLTDataBuffer::AliHLTRawPage::fgGlobalPageSize=30*1024*1024;
+
+unsigned int AliHLTDataBuffer::GetMaxBufferSize()
+{
+ return(30 * AliHLTRawPage::GetGlobalPageSize()); //Use a max data size of 30 * GlobalPageSize, as this is the maximum AliHLTRawPage::GlobalAlloc will allocate
+}
+
+AliHLTDataBuffer::AliHLTRawBuffer* AliHLTDataBuffer::AliHLTRawPage::GlobalAlloc(AliHLTUInt32_t size, int verbosity)
+{
+ // alloc a buffer of specified size from the global pages
+ AliHLTDataBuffer::AliHLTRawBuffer* rawbuffer=NULL;
+ vector<AliHLTDataBuffer::AliHLTRawPage*>::iterator page=fgGlobalPages.begin();
+ AliHLTLogging log;
+ for (page=fgGlobalPages.begin();page!=fgGlobalPages.end(); page++) {
+ if ((rawbuffer=(*page)->Alloc(size))!=NULL) {
+ if (verbosity>1) {
+ log.Logging(kHLTLogInfo, "AliHLTDataBuffer::AliHLTRawPage::GlobalAlloc", "data buffer handling", "allocated raw buffer %p from page %p\n", rawbuffer, *page);
+ rawbuffer->Print("min");
+ }
+ break;
+ }
+ }
+ if (!rawbuffer) {
+ AliHLTUInt32_t rawPageSize=fgGlobalPageSize;
+ if (rawPageSize<size) {
+ if (rawPageSize*30+fgkSafetyPatternSize<size) {
+ log.Logging(kHLTLogError, "AliHLTDataBuffer::AliHLTRawPage::GlobalAlloc", "data buffer handling", "refusing to allocate buffer of size %d", size);
+ return NULL;
+ }
+ rawPageSize=size;
+ }
+ AliHLTDataBuffer::AliHLTRawPage* rawpage=new AliHLTDataBuffer::AliHLTRawPage(rawPageSize);
+ if (!rawpage) {
+ log.Logging(kHLTLogError, "AliHLTDataBuffer::AliHLTRawPage::GlobalAlloc", "data buffer handling", "can not create raw page");
+ return NULL;
+ }
+
+ // check is there is at least one unused page which can be replaced by the newly created one
+ for (page=fgGlobalPages.begin(); page!=fgGlobalPages.end(); page++) {
+ if ((*page)->IsUsed()) continue;
+ delete *page;
+ fgGlobalPages.erase(page);
+ break; // delete only one page to be replaced by the new page
+ }
+ fgGlobalPages.push_back(rawpage);
+ if ((rawbuffer=rawpage->Alloc(size))!=NULL) {
+ if (verbosity>1) {
+ log.Logging(kHLTLogInfo, "AliHLTDataBuffer::AliHLTRawPage::GlobalAlloc", "data buffer handling", "allocated raw buffer %p from page %p\n", rawbuffer, rawpage);
+ rawbuffer->Print("min");
+ }
+ }
+ }
+
+ return rawbuffer;
+}
+
+AliHLTDataBuffer::AliHLTRawPage* AliHLTDataBuffer::AliHLTRawPage::FindPage(AliHLTDataBuffer::AliHLTRawBuffer* buffer)
+{
+ // find buffer in the global pages
+ vector<AliHLTDataBuffer::AliHLTRawPage*>::iterator page=fgGlobalPages.begin();
+ for (; page!=fgGlobalPages.end(); page++) {
+ if ((*page)->HasBuffer(buffer)) {
+ return *page;
+ }
+ }
+
+ return NULL;
+}
+
+int AliHLTDataBuffer::AliHLTRawPage::GlobalClean()
+{
+ // cleanup the global pages */
+ vector<AliHLTDataBuffer::AliHLTRawPage*>::iterator page=fgGlobalPages.begin();
+ while (page!=fgGlobalPages.end()) {
+ if (!(*page)->IsUsed()) {
+ delete *page;
+ page=fgGlobalPages.erase(page);
+ continue;
+ }
+ AliHLTLogging log;
+ log.Logging(kHLTLogError, "AliHLTDataBuffer::AliHLTRawPage::GlobalClean", "data buffer handling", "HLT memory page still in use, skipping cleanup, potential memory leak");
+
+ page++;
+ }
+
+ return 0;
+}
+
+AliHLTDataBuffer::AliHLTRawPage* AliHLTDataBuffer::AliHLTRawPage::NextPage(const AliHLTDataBuffer::AliHLTRawPage* prev)
+{
+ // get next global page
+ vector<AliHLTDataBuffer::AliHLTRawPage*>::iterator page=fgGlobalPages.begin();
+ for (; page!=fgGlobalPages.end(); page++) {
+ if (prev==NULL) return *page;
+ if (*page!=prev) continue;
+ if (++page!=fgGlobalPages.end()) return *page;
+ break;
+ }
+ return NULL;
+}
+
+void AliHLTDataBuffer::AliHLTDataSegment::Print(const char* /*option*/) const
+{
+ // print info for data segment
+ cout << "AliHLTDataSegment " << this
+ << " " << AliHLTComponent::DataType2Text(fDataType)
+ << " " << hex << fSpecification << dec
+ << " Ptr " << (void*)fPtr
+ << " offset " << fSegmentOffset
+ << " size " << fSegmentSize
+ << endl;
+}
+
+void AliHLTDataBuffer::AliHLTForwardedDataSegment::Print(const char* option) const
+{
+ // print info for data segment
+ cout << "AliHLTForwardeDataSegment " << this << endl;
+ cout << " my : "; AliHLTDataSegment::Print(option);
+ cout << " parent: "; fParentSegment.Print(option);
+ cout << " task : ";
+ if (fParentTask) fParentTask->Print("");
+ else cout << "nil" << endl;
+}
+
+void AliHLTDataBuffer::Print(const char* option) const
+{
+ // print info for data buffer
+ unsigned i=0;
+ cout << "AliHLTDataBuffer " << this << endl;
+ cout << " raw buffer " << fpBuffer << endl;
+ if (fpBuffer) {
+ cout << " ";
+ fpBuffer->Print(option);
+ }
+
+ cout << " total segments: " << GetNofSegments() << endl;
+ cout << " data segments: " << fSegments.size() << endl;
+ for (i=0; i<fSegments.size(); i++) {
+ cout << " ";
+ fSegments[i].Print(option);
+ }
+
+ cout << " forwarded segments: " << fForwardedSegments.size() << endl;
+ for (i=0; i<fForwardedSegments.size(); i++) {
+ cout << " ";
+ fForwardedSegments[i].Print(option);
+ }
+
+ cout << " consumers: " << GetNofConsumers() << endl;
+ for (i=0; i<fConsumers.size(); i++) {
+ cout << " ";
+ fConsumers[i]->Print(option);
+ }
+
+ cout << " active consumers: " << GetNofActiveConsumers() << endl;
+ for (i=0; i<fActiveConsumers.size(); i++) {
+ cout << " ";
+ fActiveConsumers[i]->Print(option);
+ }
+
+ cout << " released consumers: " << fReleasedConsumers.size() << endl;
+ for (i=0; i<fReleasedConsumers.size(); i++) {
+ cout << " ";
+ fReleasedConsumers[i]->Print(option);
+ }
+
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff