]>
Commit | Line | Data |
---|---|---|
3f2a1b1c | 1 | // $Id$ |
2 | ||
3 | /************************************************************************** | |
9be2600f | 4 | * This file is property of and copyright by the ALICE HLT Project * |
5 | * ALICE Experiment at CERN, All rights reserved. * | |
3f2a1b1c | 6 | * * |
9be2600f | 7 | * Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no> * |
8 | * for The ALICE HLT Project. * | |
3f2a1b1c | 9 | * * |
10 | * Permission to use, copy, modify and distribute this software and its * | |
11 | * documentation strictly for non-commercial purposes is hereby granted * | |
12 | * without fee, provided that the above copyright notice appears in all * | |
13 | * copies and that both the copyright notice and this permission notice * | |
14 | * appear in the supporting documentation. The authors make no claims * | |
15 | * about the suitability of this software for any purpose. It is * | |
16 | * provided "as is" without express or implied warranty. * | |
17 | **************************************************************************/ | |
18 | ||
b22e91eb | 19 | /** @file AliHLTDataBuffer.cxx |
20 | @author Matthias Richter | |
21 | @date | |
22 | @brief Handling of Data Buffers for HLT components. | |
23 | */ | |
3f2a1b1c | 24 | |
30338a30 | 25 | // see header file for class documentation |
26 | // or | |
27 | // refer to README to build package | |
28 | // or | |
29 | // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt | |
30 | ||
0c0c9d99 | 31 | #if __GNUC__>= 3 |
3f2a1b1c | 32 | using namespace std; |
33 | #endif | |
34 | ||
35 | #include "AliHLTDataBuffer.h" | |
6235cd38 | 36 | #include "AliHLTConsumerDescriptor.h" |
b22e91eb | 37 | #include "AliHLTComponent.h" |
b46ca65e | 38 | #include "AliHLTTask.h" |
66043029 | 39 | #include <cerrno> |
d6cbe999 | 40 | #include <cassert> |
66043029 | 41 | //#include <string> |
42 | //#include "AliHLTSystem.h" | |
3f2a1b1c | 43 | |
2be3f004 | 44 | typedef vector<AliHLTDataBuffer::AliHLTDataSegment> AliHLTDataSegmentList; |
45 | typedef vector<AliHLTDataBuffer::AliHLTRawBuffer*> AliHLTRawBufferPList; | |
46 | ||
b22e91eb | 47 | /** ROOT macro for the implementation of ROOT specific class methods */ |
3f2a1b1c | 48 | ClassImp(AliHLTDataBuffer) |
49 | ||
3f2a1b1c | 50 | AliHLTDataBuffer::AliHLTDataBuffer() |
85869391 | 51 | : |
52 | fSegments(), | |
53 | fConsumers(), | |
54 | fActiveConsumers(), | |
55 | fReleasedConsumers(), | |
56 | fpBuffer(NULL), | |
b46ca65e | 57 | fFlags(0), |
58 | fForwardedSegmentSources(), | |
59 | fForwardedSegments() | |
3f2a1b1c | 60 | { |
70ed7d01 | 61 | // see header file for class documentation |
62 | // or | |
63 | // refer to README to build package | |
64 | // or | |
65 | // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt | |
85869391 | 66 | fSegments.empty(); |
67 | fConsumers.empty(); | |
68 | fActiveConsumers.empty(); | |
69 | fReleasedConsumers.empty(); | |
70ed7d01 | 70 | fgNofInstances++; |
3f2a1b1c | 71 | } |
72 | ||
70ed7d01 | 73 | int AliHLTDataBuffer::fgNofInstances=0; |
2be3f004 | 74 | AliHLTRawBufferPList AliHLTDataBuffer::fgFreeBuffers; |
75 | AliHLTRawBufferPList AliHLTDataBuffer::fgActiveBuffers; | |
70ed7d01 | 76 | AliHLTUInt32_t AliHLTDataBuffer::fgMargin=1024; |
b22e91eb | 77 | AliHLTLogging AliHLTDataBuffer::fgLogging; |
8451168b | 78 | const Int_t AliHLTDataBuffer::fgkSafetyPatternSize=16; |
79 | const char AliHLTDataBuffer::fgkSafetyPattern[]={0x28, 0x63, 0x29, 0x4d, 0x52, 0x49, 0x43, 0x48, 0x54, 0x45, 0x52, 0x20, 0x32, 0x30, 0x30, 0x37}; | |
b22e91eb | 80 | |
3f2a1b1c | 81 | AliHLTDataBuffer::~AliHLTDataBuffer() |
82 | { | |
70ed7d01 | 83 | // see header file for function documentation |
84 | if (--fgNofInstances<=0) { | |
3f2a1b1c | 85 | DeleteRawBuffers(); |
86 | } | |
87 | CleanupConsumerList(); | |
88 | } | |
89 | ||
0c0c9d99 | 90 | int AliHLTDataBuffer::SetConsumer(AliHLTComponent* pConsumer) |
3f2a1b1c | 91 | { |
70ed7d01 | 92 | // see header file for function documentation |
3f2a1b1c | 93 | int iResult=0; |
94 | if (pConsumer) { | |
9ce4bf4a | 95 | if (FindConsumer(pConsumer)) { |
96 | HLTWarning("consumer %s (%p) already set to data buffer %p", pConsumer->GetComponentID(), pConsumer, this); | |
97 | } | |
0c0c9d99 | 98 | AliHLTConsumerDescriptor* pDesc=new AliHLTConsumerDescriptor(pConsumer); |
3f2a1b1c | 99 | if (pDesc) { |
100 | fConsumers.push_back(pDesc); | |
9ce4bf4a | 101 | HLTDebug("set consumer %s (%p) to data buffer %p", pConsumer->GetComponentID(), pConsumer, this); |
3f2a1b1c | 102 | } else { |
103 | HLTError("memory allocation failed"); | |
104 | iResult=-ENOMEM; | |
105 | } | |
106 | } else { | |
9ce4bf4a | 107 | HLTError("invalid parameter: consumer component (nil)"); |
3f2a1b1c | 108 | iResult=-EINVAL; |
109 | } | |
110 | return iResult; | |
111 | } | |
112 | ||
2be3f004 | 113 | int AliHLTDataBuffer::FindMatchingDataBlocks(const AliHLTComponent* pConsumer, AliHLTComponentDataTypeList* tgtList) |
0c0c9d99 | 114 | { |
70ed7d01 | 115 | // see header file for function documentation |
0c0c9d99 | 116 | int iResult=0; |
117 | if (pConsumer) { | |
2be3f004 | 118 | AliHLTDataSegmentList segments; |
0c0c9d99 | 119 | if ((iResult=FindMatchingDataSegments(pConsumer, segments))>=0) { |
120 | if (tgtList) { | |
2be3f004 | 121 | AliHLTDataSegmentList::iterator segment=segments.begin(); |
0c0c9d99 | 122 | while (segment!=segments.end()) { |
123 | tgtList->push_back((*segment).fDataType); | |
124 | segment++; | |
125 | } | |
126 | } | |
127 | iResult=segments.size(); | |
128 | } | |
129 | } else { | |
130 | iResult=-EINVAL; | |
131 | } | |
132 | return iResult; | |
133 | } | |
134 | ||
6235cd38 | 135 | int AliHLTDataBuffer::FindMatchingDataSegments(const AliHLTComponent* pConsumer, vector<AliHLTDataBuffer::AliHLTDataSegment>& tgtList) |
0c0c9d99 | 136 | { |
70ed7d01 | 137 | // see header file for function documentation |
0c0c9d99 | 138 | int iResult=0; |
9b7fe12d | 139 | |
140 | // Matthias 26.09.2007 relax the restriction to matching data blocks | |
141 | // all blocks are passed to the consumer, which is the policy also in | |
142 | // PubSub | |
143 | tgtList.assign(fSegments.begin(), fSegments.end()); | |
b46ca65e | 144 | |
145 | // add all forwarded blocks | |
146 | tgtList.insert(tgtList.begin(), fForwardedSegments.begin(), fForwardedSegments.end()); | |
9b7fe12d | 147 | iResult=tgtList.size(); |
148 | return iResult; | |
149 | ||
0c0c9d99 | 150 | if (pConsumer) { |
2be3f004 | 151 | AliHLTComponentDataTypeList dtlist; |
0c0c9d99 | 152 | ((AliHLTComponent*)pConsumer)->GetInputDataTypes(dtlist); |
2be3f004 | 153 | AliHLTDataSegmentList::iterator segment=fSegments.begin(); |
0c0c9d99 | 154 | while (segment!=fSegments.end()) { |
2be3f004 | 155 | AliHLTComponentDataTypeList::iterator type=dtlist.begin(); |
0c0c9d99 | 156 | while (type!=dtlist.end()) { |
9ce4bf4a | 157 | if ((*segment).fDataType==(*type) || |
158 | (*type)==kAliHLTAnyDataType) { | |
0c0c9d99 | 159 | tgtList.push_back(*segment); |
160 | iResult++; | |
161 | break; | |
162 | } | |
163 | type++; | |
164 | } | |
165 | segment++; | |
166 | } | |
167 | } else { | |
168 | iResult=-EINVAL; | |
169 | } | |
170 | return iResult; | |
171 | } | |
172 | ||
8ede8717 | 173 | int AliHLTDataBuffer::Subscribe(const AliHLTComponent* pConsumer, AliHLTComponentBlockData* arrayBlockDesc, int iArraySize) |
3f2a1b1c | 174 | { |
70ed7d01 | 175 | // see header file for function documentation |
3f2a1b1c | 176 | int iResult=0; |
0c0c9d99 | 177 | if (pConsumer && arrayBlockDesc) { |
b46ca65e | 178 | if (1/*fpBuffer*/) { |
0c0c9d99 | 179 | AliHLTConsumerDescriptor* pDesc=FindConsumer(pConsumer, fConsumers); |
3f2a1b1c | 180 | if (pDesc) { |
2be3f004 | 181 | AliHLTDataSegmentList tgtList; |
9b7fe12d | 182 | // Matthias 26.07.2007 AliHLTSystem should behave the same way as PubSub |
183 | // so it does not matter if there are matching data types or not, unless | |
184 | // we implement such a check in PubSub | |
185 | if ((iResult=FindMatchingDataSegments(pConsumer, tgtList))>=0) { | |
0c0c9d99 | 186 | int i =0; |
2be3f004 | 187 | AliHLTDataSegmentList::iterator segment=tgtList.begin(); |
0c0c9d99 | 188 | while (segment!=tgtList.end() && i<iArraySize) { |
3f2a1b1c | 189 | // fill the block data descriptor |
8ede8717 | 190 | arrayBlockDesc[i].fStructSize=sizeof(AliHLTComponentBlockData); |
3f2a1b1c | 191 | // the shared memory key is not used in AliRoot |
8ede8717 | 192 | arrayBlockDesc[i].fShmKey.fStructSize=sizeof(AliHLTComponentShmData); |
193 | arrayBlockDesc[i].fShmKey.fShmType=gkAliHLTComponentInvalidShmType; | |
194 | arrayBlockDesc[i].fShmKey.fShmID=gkAliHLTComponentInvalidShmID; | |
3294f81a | 195 | // This models the behavior of PubSub. |
196 | // For incoming data blocks, fOffset must be ignored by the | |
197 | // processing component. It is set for bookkeeping in the framework. | |
198 | // fPtr always points to the beginning of the data. | |
b46ca65e | 199 | arrayBlockDesc[i].fOffset=0; |
200 | AliHLTUInt8_t* pTgt=*segment; | |
3294f81a | 201 | arrayBlockDesc[i].fPtr=reinterpret_cast<void*>(pTgt); |
0c0c9d99 | 202 | arrayBlockDesc[i].fSize=(*segment).fSegmentSize; |
203 | arrayBlockDesc[i].fDataType=(*segment).fDataType; | |
204 | arrayBlockDesc[i].fSpecification=(*segment).fSpecification; | |
b46ca65e | 205 | pDesc->SetActiveDataSegment(*segment); |
1e6e67ec | 206 | HLTDebug("component %p (%s) subscribed to segment #%d offset %d size %d data type %s %#x", |
207 | pConsumer, ((AliHLTComponent*)pConsumer)->GetComponentID(), i, arrayBlockDesc[i].fOffset, | |
208 | arrayBlockDesc[i].fSize, (AliHLTComponent::DataType2Text(arrayBlockDesc[i].fDataType)).c_str(), | |
209 | arrayBlockDesc[i].fSpecification); | |
0c0c9d99 | 210 | i++; |
211 | segment++; | |
212 | } | |
9b7fe12d | 213 | // check whether there was enough space for the segments |
298ef463 | 214 | if (i!=(int)tgtList.size()) { |
9b7fe12d | 215 | HLTError("too little space in block descriptor array: required %d, available %d", tgtList.size(), iArraySize); |
216 | iResult=-ENOSPC; | |
217 | } else { | |
0c0c9d99 | 218 | // move this consumer to the active list |
b46ca65e | 219 | if (i==0) { |
220 | ChangeConsumerState(pDesc, fConsumers, fReleasedConsumers); | |
221 | HLTDebug("no input data for component %p (%s) available", pConsumer, ((AliHLTComponent*)pConsumer)->GetComponentID()); | |
222 | } else if (ChangeConsumerState(pDesc, fConsumers, fActiveConsumers)>=0) { | |
0c0c9d99 | 223 | HLTDebug("component %p (%s) subscribed to data buffer %p", pConsumer, ((AliHLTComponent*)pConsumer)->GetComponentID(), this); |
3f2a1b1c | 224 | } else { |
0c0c9d99 | 225 | // TODO: cleanup the consumer descriptor correctly |
8ede8717 | 226 | memset(arrayBlockDesc, 0, iArraySize*sizeof(AliHLTComponentBlockData)); |
3f2a1b1c | 227 | HLTError("can not activate consumer %p for data buffer %p", pConsumer, this); |
228 | iResult=-EACCES; | |
229 | } | |
9b7fe12d | 230 | } |
3f2a1b1c | 231 | } else { |
0c0c9d99 | 232 | HLTError("unresolved data segment(s) for component %p (%s)", pConsumer, ((AliHLTComponent*)pConsumer)->GetComponentID()); |
3f2a1b1c | 233 | iResult=-EBADF; |
234 | } | |
235 | } else { | |
0c0c9d99 | 236 | HLTError("component %p is not a data consumer of data buffer %s", pConsumer, this); |
3f2a1b1c | 237 | iResult=-ENOENT; |
238 | } | |
239 | } else { | |
9b7fe12d | 240 | // Matthias 26.07.2007 until now, data had to be present for successful subscription |
241 | // in order to be consistent with the PubSub framework, this restiction has been | |
242 | // removed | |
243 | //HLTError("data buffer %p is empty", this); | |
244 | //iResult=-ENODATA; | |
3f2a1b1c | 245 | } |
246 | } else { | |
247 | HLTError("invalid parameter"); | |
248 | iResult=-EINVAL; | |
249 | } | |
250 | return iResult; | |
251 | } | |
252 | ||
b46ca65e | 253 | int AliHLTDataBuffer::Release(AliHLTComponentBlockData* pBlockDesc, |
254 | const AliHLTComponent* pConsumer, | |
255 | const AliHLTTask* pOwnerTask) | |
3f2a1b1c | 256 | { |
70ed7d01 | 257 | // see header file for function documentation |
3f2a1b1c | 258 | int iResult=0; |
259 | if (pBlockDesc && pConsumer) { | |
0c0c9d99 | 260 | AliHLTConsumerDescriptor* pDesc=FindConsumer(pConsumer, fActiveConsumers); |
261 | if (pDesc) { | |
b46ca65e | 262 | if ((iResult=pDesc->CheckActiveDataSegment(AliHLTDataSegment(pBlockDesc->fPtr, pBlockDesc->fOffset, pBlockDesc->fSize)))!=1) { |
0c0c9d99 | 263 | HLTWarning("data segment missmatch, component %p has not subscribed to a segment with offset %#x and size %d", pConsumer, pBlockDesc->fOffset, pBlockDesc->fSize); |
264 | // TODO: appropriate error handling, but so far optional | |
265 | iResult=0; | |
266 | } else { | |
b46ca65e | 267 | pDesc->ReleaseActiveDataSegment(AliHLTDataSegment(pBlockDesc->fPtr, pBlockDesc->fOffset, pBlockDesc->fSize)); |
268 | } | |
269 | if (GetNofPendingConsumers()==0 && fForwardedSegments.size()>0) { | |
270 | // last consumer, release forwarded segments | |
271 | assert(fForwardedSegments.size()==fForwardedSegmentSources.size()); | |
272 | AliHLTDataSegmentList::iterator segment=fForwardedSegments.begin(); | |
273 | AliHLTTaskPList::iterator src=fForwardedSegmentSources.begin(); | |
274 | //HLTDebug("%p checking forwarded segments", this); | |
275 | for (; segment!=fForwardedSegments.end(); segment++, src++) { | |
276 | //HLTDebug("segment ptr=%p offset=%d size=%d\n" | |
277 | // "block ptr=%p offset=%d size=%d", (*segment).fPtr, (*segment).fSegmentOffset, (*segment).fSegmentSize, pBlockDesc->fPtr, pBlockDesc->fOffset, pBlockDesc->fSize); | |
278 | if ((*segment)==AliHLTDataSegment(pBlockDesc->fPtr, pBlockDesc->fOffset, pBlockDesc->fSize)) { | |
279 | //HLTDebug("release segment of task %p", *src); | |
280 | assert((*src)!=NULL); | |
281 | if ((*src)!=NULL) { | |
282 | if ((*src)->Release(pBlockDesc, pOwnerTask)>=0) { | |
283 | HLTDebug("task %s (%p) released forwarded segment %p size %d of task %s (%p)", | |
284 | pOwnerTask->GetName(), pOwnerTask, (*segment).GetPtr(), (*segment).GetSize(), | |
285 | (*src)->GetName(), *src); | |
286 | } else { | |
287 | HLTError("task %s (%p) failed releasing forwarded segment %p size %d of task %s (%p)", | |
288 | pOwnerTask->GetName(), pOwnerTask, (*segment).GetPtr(), (*segment).GetSize(), | |
289 | (*src)->GetName(), *src); | |
290 | } | |
291 | } | |
292 | fForwardedSegments.erase(segment); | |
293 | fForwardedSegmentSources.erase(src); | |
294 | break; | |
295 | } | |
296 | } | |
0c0c9d99 | 297 | } |
b46ca65e | 298 | pBlockDesc->fOffset=0; |
299 | pBlockDesc->fPtr=NULL; | |
300 | pBlockDesc->fSize=0; | |
0c0c9d99 | 301 | if (pDesc->GetNofActiveSegments()==0) { |
3f2a1b1c | 302 | if ((iResult=ChangeConsumerState(pDesc, fActiveConsumers, fReleasedConsumers))>=0) { |
b426991e | 303 | if (GetNofActiveConsumers()==0 && GetNofPendingConsumers()==0) { |
0c0c9d99 | 304 | // this is the last consumer, reset the consumer list and release the raw buffer |
3f2a1b1c | 305 | ResetDataBuffer(); |
3f2a1b1c | 306 | } |
307 | } else { | |
308 | HLTError("can not deactivate consumer %p for data buffer %p", pConsumer, this); | |
309 | iResult=-EACCES; | |
310 | } | |
3f2a1b1c | 311 | } |
0c0c9d99 | 312 | } else { |
313 | HLTWarning("component %p has currently not subscribed to the data buffer %p", pConsumer, this); | |
314 | iResult=-ENOENT; | |
315 | } | |
3f2a1b1c | 316 | } else { |
317 | HLTError("inavalid parameter: pBlockDesc=%p pConsumer=%p", pBlockDesc, pConsumer); | |
318 | iResult=-EINVAL; | |
319 | } | |
320 | return iResult; | |
321 | } | |
322 | ||
b46ca65e | 323 | int AliHLTDataBuffer::Forward(AliHLTTask* pSrcTask, AliHLTComponentBlockData* pBlockDesc) |
324 | { | |
325 | // see header file for function documentation | |
326 | if (pSrcTask==NULL || pBlockDesc==NULL) return -EINVAL; | |
327 | assert(fForwardedSegments.size()==fForwardedSegmentSources.size()); | |
328 | if (fForwardedSegments.size()!=fForwardedSegmentSources.size()) return -EFAULT; | |
329 | fForwardedSegmentSources.push_back(pSrcTask); | |
330 | fForwardedSegments.push_back(AliHLTDataSegment(pBlockDesc->fPtr, pBlockDesc->fOffset, pBlockDesc->fSize)); | |
331 | return 0; | |
332 | } | |
333 | ||
3f2a1b1c | 334 | AliHLTUInt8_t* AliHLTDataBuffer::GetTargetBuffer(int iMinSize) |
335 | { | |
70ed7d01 | 336 | // see header file for function documentation |
3f2a1b1c | 337 | AliHLTUInt8_t* pTargetBuffer=NULL; |
b6800be0 | 338 | if (fpBuffer!=NULL) { |
339 | HLTWarning("data buffer not properly reset, possible memory leak\n"); | |
340 | } | |
3f2a1b1c | 341 | fpBuffer=CreateRawBuffer(iMinSize); |
9ce4bf4a | 342 | if (fpBuffer) { |
1e6e67ec | 343 | pTargetBuffer=*fpBuffer; |
9ce4bf4a | 344 | } else { |
345 | HLTError("can not create raw buffer"); | |
346 | } | |
3f2a1b1c | 347 | return pTargetBuffer; |
348 | } | |
349 | ||
8ede8717 | 350 | int AliHLTDataBuffer::SetSegments(AliHLTUInt8_t* pTgt, AliHLTComponentBlockData* arrayBlockData, int iSize) |
3f2a1b1c | 351 | { |
70ed7d01 | 352 | // see header file for function documentation |
3f2a1b1c | 353 | int iResult=0; |
354 | if (pTgt && arrayBlockData && iSize>=0) { | |
0c0c9d99 | 355 | if (fpBuffer) { |
1e6e67ec | 356 | if (*fpBuffer==pTgt) { |
6235cd38 | 357 | AliHLTDataBuffer::AliHLTDataSegment segment; |
0c0c9d99 | 358 | for (int i=0; i<iSize; i++) { |
3294f81a | 359 | // This function has to model the behavior of PubSub |
360 | // For output blocks only the fOffset value is used, this must be the offset | |
361 | // relative to the output pointer. fPtr must be either NULL or the output | |
b46ca65e | 362 | // pointer. In either case it is 'ignored' and set to the beginning of the |
363 | // data buffer | |
6434d28a | 364 | if (arrayBlockData[i].fPtr==NULL || |
3294f81a | 365 | arrayBlockData[i].fPtr==*fpBuffer) { |
b46ca65e | 366 | arrayBlockData[i].fPtr=*fpBuffer; |
c0a2bfc2 | 367 | if (arrayBlockData[i].fOffset+arrayBlockData[i].fSize<=fpBuffer->GetUsedSize()) { |
3294f81a | 368 | segment.fSegmentOffset=arrayBlockData[i].fOffset; |
b46ca65e | 369 | segment.fPtr=(AliHLTUInt8_t*)arrayBlockData[i].fPtr; |
1e6e67ec | 370 | segment.fSegmentSize=arrayBlockData[i].fSize; |
371 | segment.fDataType=arrayBlockData[i].fDataType; | |
372 | segment.fSpecification=arrayBlockData[i].fSpecification; | |
373 | fSegments.push_back(segment); | |
374 | HLTDebug("set segment %s with size %d at offset %d", AliHLTComponent::DataType2Text(segment.fDataType).data(), segment.fSegmentSize, segment.fSegmentOffset); | |
375 | } else { | |
376 | HLTError("block data specification %#d (%s) exceeds size of data buffer", i, AliHLTComponent::DataType2Text(arrayBlockData[i].fDataType).data()); | |
c0a2bfc2 | 377 | HLTError("block offset=%d, block size=%d, buffer size=%d", arrayBlockData[i].fOffset, arrayBlockData[i].fSize, fpBuffer->GetUsedSize()); |
1e6e67ec | 378 | iResult=-E2BIG; |
379 | } | |
0c0c9d99 | 380 | } else { |
3294f81a | 381 | HLTError("invalid pointer (%p) in block data specification (buffer %p size %d)." |
382 | "please note: for output blocks only the fOffset value is valid and must " | |
c0a2bfc2 | 383 | "be relative to the output buffer", arrayBlockData[i].fPtr, fpBuffer->GetPointer(), fpBuffer->GetUsedSize()); |
1e6e67ec | 384 | iResult=-ERANGE; |
0c0c9d99 | 385 | } |
386 | } | |
3f2a1b1c | 387 | } else { |
c0a2bfc2 | 388 | HLTError("this data buffer (%p) does not match the internal data buffer %p of raw buffer %p", pTgt, fpBuffer->GetPointer(), fpBuffer); |
8451168b | 389 | iResult=-EINVAL; |
3f2a1b1c | 390 | } |
0c0c9d99 | 391 | } else { |
392 | HLTFatal("internal data structur missmatch"); | |
393 | iResult=-EFAULT; | |
3f2a1b1c | 394 | } |
395 | } else { | |
0c0c9d99 | 396 | HLTError("invalid parameter: pTgtBuffer=%p arrayBlockData=%p", pTgt, arrayBlockData); |
3f2a1b1c | 397 | iResult=-EINVAL; |
398 | } | |
399 | return iResult; | |
400 | } | |
401 | ||
402 | int AliHLTDataBuffer::IsEmpty() | |
403 | { | |
70ed7d01 | 404 | // see header file for function documentation |
3f2a1b1c | 405 | int iResult=fpBuffer==NULL || GetNofSegments()==0; |
406 | return iResult; | |
407 | } | |
408 | ||
409 | int AliHLTDataBuffer::GetNofSegments() | |
410 | { | |
70ed7d01 | 411 | // see header file for function documentation |
3f2a1b1c | 412 | int iResult=fSegments.size(); |
413 | return iResult; | |
414 | } | |
415 | ||
416 | int AliHLTDataBuffer::GetNofConsumers() | |
417 | { | |
70ed7d01 | 418 | // see header file for function documentation |
3f2a1b1c | 419 | int iResult=fConsumers.size() + GetNofActiveConsumers() + fReleasedConsumers.size(); |
420 | return iResult; | |
421 | } | |
422 | ||
b426991e | 423 | int AliHLTDataBuffer::GetNofPendingConsumers() |
424 | { | |
425 | // see header file for function documentation | |
426 | int iResult=fConsumers.size(); | |
427 | return iResult; | |
428 | } | |
429 | ||
3f2a1b1c | 430 | int AliHLTDataBuffer::GetNofActiveConsumers() |
431 | { | |
70ed7d01 | 432 | // see header file for function documentation |
3f2a1b1c | 433 | int iResult=fActiveConsumers.size(); |
434 | return iResult; | |
435 | } | |
436 | ||
6235cd38 | 437 | AliHLTDataBuffer::AliHLTRawBuffer* AliHLTDataBuffer::CreateRawBuffer(AliHLTUInt32_t size) |
3f2a1b1c | 438 | { |
70ed7d01 | 439 | // see header file for function documentation |
3f2a1b1c | 440 | AliHLTRawBuffer* pRawBuffer=NULL; |
7a5ccd96 | 441 | unsigned int reqSize=size+fgkSafetyPatternSize; |
2be3f004 | 442 | AliHLTRawBufferPList::iterator buffer=fgFreeBuffers.begin(); |
70ed7d01 | 443 | while (buffer!=fgFreeBuffers.end() && pRawBuffer==NULL) { |
d6cbe999 | 444 | if ((*buffer)->CheckSize(reqSize)) { |
3f2a1b1c | 445 | // assign this element |
446 | pRawBuffer=*buffer; | |
d6cbe999 | 447 | pRawBuffer->UseBuffer(size); |
70ed7d01 | 448 | fgFreeBuffers.erase(buffer); |
c0a2bfc2 | 449 | 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()); |
70ed7d01 | 450 | fgActiveBuffers.push_back(pRawBuffer); |
0c0c9d99 | 451 | break; |
3f2a1b1c | 452 | } |
0c0c9d99 | 453 | buffer++; |
3f2a1b1c | 454 | } |
455 | if (pRawBuffer==NULL) { | |
0c0c9d99 | 456 | // no buffer found, create a new one |
d6cbe999 | 457 | pRawBuffer=new AliHLTRawBuffer(reqSize); |
3f2a1b1c | 458 | if (pRawBuffer) { |
c0a2bfc2 | 459 | if (pRawBuffer->GetPointer()) { |
d6cbe999 | 460 | pRawBuffer->UseBuffer(size); |
70ed7d01 | 461 | fgActiveBuffers.push_back(pRawBuffer); |
c0a2bfc2 | 462 | fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "new raw buffer %p of size %d created (container %p)", pRawBuffer->GetPointer(), pRawBuffer->GetTotalSize(), pRawBuffer); |
3f2a1b1c | 463 | } else { |
464 | delete pRawBuffer; | |
465 | pRawBuffer=NULL; | |
0c0c9d99 | 466 | fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "memory allocation failed"); |
3f2a1b1c | 467 | } |
468 | } else { | |
0c0c9d99 | 469 | fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "memory allocation failed"); |
3f2a1b1c | 470 | } |
471 | } | |
73ede1d3 | 472 | if (pRawBuffer!=NULL && fgkSafetyPatternSize>0) { |
c0a2bfc2 | 473 | //fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "writing safety pattern to %p offset %d", (*buffer)->GetPointer(), (*buffer)->GetUsedSize()); |
17e6313a | 474 | int res=pRawBuffer->WritePattern(fgkSafetyPattern, fgkSafetyPatternSize); |
d6cbe999 | 475 | assert(res>=0); |
73ede1d3 | 476 | } |
3f2a1b1c | 477 | return pRawBuffer; |
478 | } | |
479 | ||
480 | int AliHLTDataBuffer::ReleaseRawBuffer(AliHLTRawBuffer* pBuffer) | |
481 | { | |
70ed7d01 | 482 | // see header file for function documentation |
3f2a1b1c | 483 | int iResult=0; |
484 | if (pBuffer) { | |
2be3f004 | 485 | AliHLTRawBufferPList::iterator buffer=fgActiveBuffers.begin(); |
70ed7d01 | 486 | while (buffer!=fgActiveBuffers.end() && (*buffer)!=pBuffer) { |
3f2a1b1c | 487 | buffer++; |
488 | } | |
70ed7d01 | 489 | if (buffer!=fgActiveBuffers.end()) { |
8451168b | 490 | if (fgkSafetyPatternSize>0) { |
c0a2bfc2 | 491 | //fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "comparing safety pattern at %p offset %d", (*buffer)->GetPointer(), reinterpret_cast<AliHLTUInt32_t>(*buffer)); |
d6cbe999 | 492 | if ((*buffer)->CheckPattern(fgkSafetyPattern, fgkSafetyPatternSize)) { |
c0a2bfc2 | 493 | fgLogging.Logging(kHLTLogFatal, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "component has written beyond end of data buffer %p size %d", (*buffer)->GetPointer(), (*buffer)->GetUsedSize()); |
8451168b | 494 | } |
495 | } | |
d6cbe999 | 496 | (*buffer)->Reset(); |
70ed7d01 | 497 | fgFreeBuffers.push_back(*buffer); |
498 | fgActiveBuffers.erase(buffer); | |
3f2a1b1c | 499 | } else { |
0c0c9d99 | 500 | fgLogging.Logging(kHLTLogWarning, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "can not find raw buffer container %p in the list of active containers", pBuffer); |
3f2a1b1c | 501 | iResult=-ENOENT; |
502 | } | |
503 | } else { | |
0c0c9d99 | 504 | fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "invalid parameter"); |
3f2a1b1c | 505 | iResult=-EINVAL; |
506 | } | |
507 | return iResult; | |
508 | } | |
509 | ||
510 | ||
511 | int AliHLTDataBuffer::DeleteRawBuffers() | |
512 | { | |
70ed7d01 | 513 | // see header file for function documentation |
3f2a1b1c | 514 | int iResult=0; |
dba03d72 | 515 | // int iTotalSize=0; |
516 | // int iCount=fgFreeBuffers.size()+fgActiveBuffers.size(); | |
517 | AliHLTRawBufferPList::iterator buffer;; | |
518 | while ((buffer=fgFreeBuffers.begin())!=fgFreeBuffers.end()) { | |
519 | // iTotalSize+=(*buffer)->GetTotalSize(); | |
3f2a1b1c | 520 | delete *buffer; |
70ed7d01 | 521 | fgFreeBuffers.erase(buffer); |
3f2a1b1c | 522 | } |
dba03d72 | 523 | while ((buffer=fgActiveBuffers.begin())!=fgActiveBuffers.end()) { |
524 | // iTotalSize+=(*buffer)->GetTotalSize(); | |
c0a2bfc2 | 525 | fgLogging.Logging(kHLTLogWarning, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "request to delete active raw buffer container (raw buffer %p, size %d)", (*buffer)->GetPointer(), (*buffer)->GetTotalSize()); |
3f2a1b1c | 526 | delete *buffer; |
70ed7d01 | 527 | fgActiveBuffers.erase(buffer); |
3f2a1b1c | 528 | } |
dba03d72 | 529 | // fgLogging.Logging(kHLTLogInfo, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "Total memory allocation: %d byte in %d buffers", iTotalSize, iCount); |
3f2a1b1c | 530 | return iResult; |
531 | } | |
532 | ||
2be3f004 | 533 | AliHLTConsumerDescriptor* AliHLTDataBuffer::FindConsumer(const AliHLTComponent* pConsumer, AliHLTConsumerDescriptorPList &list) const |
3f2a1b1c | 534 | { |
70ed7d01 | 535 | // see header file for function documentation |
3f2a1b1c | 536 | AliHLTConsumerDescriptor* pDesc=NULL; |
2be3f004 | 537 | AliHLTConsumerDescriptorPList::iterator desc=list.begin(); |
3f2a1b1c | 538 | while (desc!=list.end() && pDesc==NULL) { |
0c0c9d99 | 539 | if ((pConsumer==NULL || (*desc)->GetComponent()==pConsumer)) { |
3f2a1b1c | 540 | pDesc=*desc; |
541 | } | |
0c0c9d99 | 542 | desc++; |
3f2a1b1c | 543 | } |
544 | return pDesc; | |
545 | } | |
546 | ||
0c0c9d99 | 547 | int AliHLTDataBuffer::ResetDataBuffer() |
548 | { | |
70ed7d01 | 549 | // see header file for function documentation |
3f2a1b1c | 550 | int iResult=0; |
0c0c9d99 | 551 | AliHLTRawBuffer* pBuffer=fpBuffer; |
3f2a1b1c | 552 | fpBuffer=NULL; |
9ce4bf4a | 553 | |
b46ca65e | 554 | // cleanup forwarded segment lists |
555 | assert(fForwardedSegments.size()==0); | |
556 | fForwardedSegments.clear(); | |
557 | fForwardedSegmentSources.clear(); | |
558 | ||
9ce4bf4a | 559 | // cleanup consumer states |
2be3f004 | 560 | AliHLTConsumerDescriptorPList::iterator desc; |
b426991e | 561 | // if (GetNofPendingConsumers()>0) { |
562 | // desc=fConsumers.begin(); | |
563 | // while (desc!=fConsumers.end()) { | |
564 | // AliHLTComponent* pComp=(*desc)->GetComponent(); | |
565 | // HLTError("internal error: consumer %p (%s %p) did not get data from data buffer %p", *desc, pComp?pComp->GetComponentID():"", pComp, this); | |
566 | // desc++; | |
567 | // } | |
568 | // } | |
569 | desc=fReleasedConsumers.begin(); | |
3f2a1b1c | 570 | while (desc!=fReleasedConsumers.end()) { |
571 | AliHLTConsumerDescriptor* pDesc=*desc; | |
572 | fReleasedConsumers.erase(desc); | |
573 | desc=fReleasedConsumers.begin(); | |
574 | fConsumers.push_back(pDesc); | |
575 | } | |
576 | desc=fActiveConsumers.begin(); | |
577 | while (desc!=fActiveConsumers.end()) { | |
578 | AliHLTConsumerDescriptor* pDesc=*desc; | |
579 | HLTWarning("consumer %p was not released", pDesc); | |
580 | fActiveConsumers.erase(desc); | |
581 | desc=fActiveConsumers.begin(); | |
582 | fConsumers.push_back(pDesc); | |
583 | } | |
9ce4bf4a | 584 | |
585 | // cleanup segments | |
2be3f004 | 586 | AliHLTDataSegmentList::iterator segment=fSegments.begin(); |
9ce4bf4a | 587 | while (segment!=fSegments.end()) { |
588 | fSegments.erase(segment); | |
589 | segment=fSegments.begin(); | |
590 | } | |
591 | ||
592 | // cleanup raw buffer | |
593 | if (pBuffer) { | |
594 | ReleaseRawBuffer(pBuffer); | |
595 | } | |
3f2a1b1c | 596 | return iResult; |
597 | } | |
598 | ||
9ce4bf4a | 599 | int AliHLTDataBuffer::Reset() |
600 | { | |
70ed7d01 | 601 | // see header file for function documentation |
9ce4bf4a | 602 | return ResetDataBuffer(); |
603 | } | |
604 | ||
0c0c9d99 | 605 | // this is the version which works on lists of components instead of consumer descriptors |
2be3f004 | 606 | // int AliHLTDataBuffer::ChangeConsumerState(AliHLTComponent* pConsumer, AliHLTComponentPList &srcList, AliHLTComponentPList &tgtList) |
0c0c9d99 | 607 | // { |
608 | // int iResult=0; | |
609 | // if (pDesc) { | |
2be3f004 | 610 | // AliHLTComponentPList::iterator desc=srcList.begin(); |
0c0c9d99 | 611 | // while (desc!=srcList.end()) { |
612 | // if ((*desc)==pConsumer) { | |
613 | // srcList.erase(desc); | |
614 | // tgtList.push_back(pConsumer); | |
615 | // break; | |
616 | // } | |
617 | // desc++; | |
618 | // } | |
619 | // if (desc==srcList.end()) { | |
620 | // HLTError("can not find consumer component %p in list", pConsumer); | |
621 | // iResult=-ENOENT; | |
622 | // } | |
623 | // } else { | |
624 | // HLTError("invalid parameter"); | |
625 | // iResult=-EINVAL; | |
626 | // } | |
627 | // return iResult; | |
628 | // } | |
629 | ||
2be3f004 | 630 | int AliHLTDataBuffer::ChangeConsumerState(AliHLTConsumerDescriptor* pDesc, AliHLTConsumerDescriptorPList &srcList, AliHLTConsumerDescriptorPList &tgtList) |
3f2a1b1c | 631 | { |
70ed7d01 | 632 | // see header file for function documentation |
9ce4bf4a | 633 | int iResult=-ENOENT; |
3f2a1b1c | 634 | if (pDesc) { |
2be3f004 | 635 | AliHLTConsumerDescriptorPList::iterator desc=srcList.begin(); |
3f2a1b1c | 636 | while (desc!=srcList.end()) { |
637 | if ((*desc)==pDesc) { | |
638 | srcList.erase(desc); | |
639 | tgtList.push_back(pDesc); | |
9ce4bf4a | 640 | iResult=0; |
3f2a1b1c | 641 | break; |
642 | } | |
0c0c9d99 | 643 | desc++; |
3f2a1b1c | 644 | } |
9ce4bf4a | 645 | if (iResult<0) { |
3f2a1b1c | 646 | HLTError("can not find consumer descriptor %p in list", pDesc); |
3f2a1b1c | 647 | } |
648 | } else { | |
649 | HLTError("invalid parameter"); | |
650 | iResult=-EINVAL; | |
651 | } | |
652 | return iResult; | |
653 | } | |
654 | ||
70ed7d01 | 655 | int AliHLTDataBuffer::CleanupConsumerList() |
656 | { | |
657 | // see header file for function documentation | |
3f2a1b1c | 658 | int iResult=0; |
659 | ResetDataBuffer(); | |
2be3f004 | 660 | AliHLTConsumerDescriptorPList::iterator desc=fConsumers.begin(); |
3f2a1b1c | 661 | while (desc!=fConsumers.end()) { |
662 | delete *desc; | |
663 | fConsumers.erase(desc); | |
664 | desc=fConsumers.begin(); | |
665 | } | |
666 | return iResult; | |
667 | } | |
9ce4bf4a | 668 | |
70ed7d01 | 669 | int AliHLTDataBuffer::FindConsumer(AliHLTComponent* pConsumer, int bAllLists) |
670 | { | |
671 | // see header file for function documentation | |
2be3f004 | 672 | AliHLTConsumerDescriptorPList::iterator desc=fConsumers.begin(); |
9ce4bf4a | 673 | while (desc!=fConsumers.end()) { |
674 | if ((*desc)->GetComponent()==pConsumer) | |
675 | return 1; | |
676 | desc++; | |
677 | } | |
678 | if (bAllLists==0) return 0; | |
679 | ||
680 | desc=fActiveConsumers.begin(); | |
681 | while (desc!=fActiveConsumers.end()) { | |
682 | if ((*desc)->GetComponent()==pConsumer) | |
683 | return 1; | |
684 | desc++; | |
685 | } | |
686 | desc=fReleasedConsumers.begin(); | |
687 | while (desc!=fReleasedConsumers.end()) { | |
688 | if ((*desc)->GetComponent()==pConsumer) | |
689 | return 1; | |
690 | desc++; | |
691 | } | |
692 | return 0; | |
693 | } | |
1e6e67ec | 694 | |
d6cbe999 | 695 | AliHLTDataBuffer::AliHLTRawBuffer::AliHLTRawBuffer(AliHLTUInt32_t size) |
696 | : | |
c0a2bfc2 | 697 | fSize(0), |
d6cbe999 | 698 | fTotalSize(size), |
699 | fPtr(static_cast<AliHLTUInt8_t*>(malloc(size))) | |
700 | { | |
701 | // see header file for class documentation | |
702 | // or | |
703 | // refer to README to build package | |
704 | // or | |
705 | // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt | |
706 | if (fPtr==NULL) { | |
707 | fSize=0; | |
708 | fTotalSize=0; | |
709 | } | |
710 | } | |
711 | ||
712 | AliHLTDataBuffer::AliHLTRawBuffer::~AliHLTRawBuffer() | |
713 | { | |
714 | if (fPtr) { | |
715 | free(fPtr); | |
716 | } | |
717 | fPtr=NULL; | |
718 | fSize=0; | |
719 | fTotalSize=0; | |
720 | } | |
721 | ||
3a7c0444 | 722 | int AliHLTDataBuffer::AliHLTRawBuffer::operator==(void* ptr) const |
1e6e67ec | 723 | { |
3a7c0444 | 724 | // see header file for function documentation |
1e6e67ec | 725 | return fPtr == static_cast<AliHLTUInt8_t*>(ptr); |
726 | } | |
727 | ||
3a7c0444 | 728 | int AliHLTDataBuffer::AliHLTRawBuffer::operator<=(void* ptr) const |
1e6e67ec | 729 | { |
3a7c0444 | 730 | // see header file for function documentation |
1e6e67ec | 731 | int iResult=fPtr <= static_cast<AliHLTUInt8_t*>(ptr); |
732 | //printf("%p: %p <= %p (%d)\n", this, fPtr, ptr, iResult); | |
733 | return iResult; | |
734 | } | |
735 | ||
3a7c0444 | 736 | int AliHLTDataBuffer::AliHLTRawBuffer::operator>(void* ptr) const |
1e6e67ec | 737 | { |
3a7c0444 | 738 | // see header file for function documentation |
1e6e67ec | 739 | int iResult=fPtr+fSize > static_cast<AliHLTUInt8_t*>(ptr); |
740 | //printf("%p: %p + %d > %p (%d)\n", this, fPtr, fSize, ptr, iResult); | |
741 | return iResult; | |
742 | } | |
743 | ||
3a7c0444 | 744 | int AliHLTDataBuffer::AliHLTRawBuffer::operator-(void* ptr) const |
1e6e67ec | 745 | { |
3a7c0444 | 746 | // see header file for function documentation |
1e6e67ec | 747 | return static_cast<int>(static_cast<AliHLTUInt8_t*>(ptr)-fPtr); |
748 | } | |
d6cbe999 | 749 | |
750 | AliHLTUInt8_t* AliHLTDataBuffer::AliHLTRawBuffer::UseBuffer(AliHLTUInt32_t size) | |
751 | { | |
752 | // see header file for function documentation | |
dba03d72 | 753 | if (size>0 && fTotalSize>=size) { |
d6cbe999 | 754 | fSize=size; |
755 | return fPtr; | |
756 | } | |
757 | return NULL; | |
758 | } | |
759 | ||
760 | int AliHLTDataBuffer::AliHLTRawBuffer::CheckSize(AliHLTUInt32_t size) const | |
761 | { | |
762 | // see header file for function documentation | |
763 | return fTotalSize>=size && ((fTotalSize-size)<fgMargin); | |
764 | } | |
765 | ||
766 | int AliHLTDataBuffer::AliHLTRawBuffer::Reset() | |
767 | { | |
768 | // see header file for function documentation | |
769 | fSize=0; | |
2be3f004 | 770 | return 0; |
d6cbe999 | 771 | } |
772 | ||
773 | int AliHLTDataBuffer::AliHLTRawBuffer::WritePattern(const char* pattern, int size) | |
774 | { | |
775 | // see header file for function documentation | |
776 | int iResult=0; | |
777 | if (pattern!=NULL && size>0) { | |
778 | if (fSize+size<=fTotalSize) { | |
779 | memcpy(((char*)fPtr)+fSize, pattern, size); | |
780 | iResult=size; | |
781 | } else { | |
782 | iResult=-ENOSPC; | |
783 | } | |
784 | } | |
785 | return iResult; | |
786 | } | |
787 | ||
788 | int AliHLTDataBuffer::AliHLTRawBuffer::CheckPattern(const char* pattern, int size) const | |
789 | { | |
790 | // see header file for function documentation | |
791 | int iResult=0; | |
792 | if (pattern!=NULL && size>0) { | |
793 | if (fSize+size<=fTotalSize) { | |
794 | iResult=memcmp(((char*)fPtr)+fSize, pattern, size)!=0; | |
795 | } else { | |
796 | iResult=-ENOSPC; | |
797 | } | |
798 | } | |
799 | return iResult; | |
800 | } |