]>
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" |
66043029 | 38 | #include <cerrno> |
39 | //#include <string> | |
40 | //#include "AliHLTSystem.h" | |
3f2a1b1c | 41 | |
b22e91eb | 42 | /** ROOT macro for the implementation of ROOT specific class methods */ |
3f2a1b1c | 43 | ClassImp(AliHLTDataBuffer) |
44 | ||
3f2a1b1c | 45 | AliHLTDataBuffer::AliHLTDataBuffer() |
85869391 | 46 | : |
47 | fSegments(), | |
48 | fConsumers(), | |
49 | fActiveConsumers(), | |
50 | fReleasedConsumers(), | |
51 | fpBuffer(NULL), | |
52 | fFlags(0) | |
3f2a1b1c | 53 | { |
70ed7d01 | 54 | // see header file for class documentation |
55 | // or | |
56 | // refer to README to build package | |
57 | // or | |
58 | // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt | |
85869391 | 59 | fSegments.empty(); |
60 | fConsumers.empty(); | |
61 | fActiveConsumers.empty(); | |
62 | fReleasedConsumers.empty(); | |
70ed7d01 | 63 | fgNofInstances++; |
3f2a1b1c | 64 | } |
65 | ||
70ed7d01 | 66 | int AliHLTDataBuffer::fgNofInstances=0; |
6235cd38 | 67 | vector<AliHLTDataBuffer::AliHLTRawBuffer*> AliHLTDataBuffer::fgFreeBuffers; |
68 | vector<AliHLTDataBuffer::AliHLTRawBuffer*> AliHLTDataBuffer::fgActiveBuffers; | |
70ed7d01 | 69 | AliHLTUInt32_t AliHLTDataBuffer::fgMargin=1024; |
b22e91eb | 70 | AliHLTLogging AliHLTDataBuffer::fgLogging; |
8451168b | 71 | const Int_t AliHLTDataBuffer::fgkSafetyPatternSize=16; |
72 | const char AliHLTDataBuffer::fgkSafetyPattern[]={0x28, 0x63, 0x29, 0x4d, 0x52, 0x49, 0x43, 0x48, 0x54, 0x45, 0x52, 0x20, 0x32, 0x30, 0x30, 0x37}; | |
b22e91eb | 73 | |
3f2a1b1c | 74 | AliHLTDataBuffer::~AliHLTDataBuffer() |
75 | { | |
70ed7d01 | 76 | // see header file for function documentation |
77 | if (--fgNofInstances<=0) { | |
3f2a1b1c | 78 | DeleteRawBuffers(); |
79 | } | |
80 | CleanupConsumerList(); | |
81 | } | |
82 | ||
0c0c9d99 | 83 | int AliHLTDataBuffer::SetConsumer(AliHLTComponent* pConsumer) |
3f2a1b1c | 84 | { |
70ed7d01 | 85 | // see header file for function documentation |
3f2a1b1c | 86 | int iResult=0; |
87 | if (pConsumer) { | |
9ce4bf4a | 88 | if (FindConsumer(pConsumer)) { |
89 | HLTWarning("consumer %s (%p) already set to data buffer %p", pConsumer->GetComponentID(), pConsumer, this); | |
90 | } | |
0c0c9d99 | 91 | AliHLTConsumerDescriptor* pDesc=new AliHLTConsumerDescriptor(pConsumer); |
3f2a1b1c | 92 | if (pDesc) { |
93 | fConsumers.push_back(pDesc); | |
9ce4bf4a | 94 | HLTDebug("set consumer %s (%p) to data buffer %p", pConsumer->GetComponentID(), pConsumer, this); |
3f2a1b1c | 95 | } else { |
96 | HLTError("memory allocation failed"); | |
97 | iResult=-ENOMEM; | |
98 | } | |
99 | } else { | |
9ce4bf4a | 100 | HLTError("invalid parameter: consumer component (nil)"); |
3f2a1b1c | 101 | iResult=-EINVAL; |
102 | } | |
103 | return iResult; | |
104 | } | |
105 | ||
8ede8717 | 106 | int AliHLTDataBuffer::FindMatchingDataBlocks(const AliHLTComponent* pConsumer, vector<AliHLTComponentDataType>* tgtList) |
0c0c9d99 | 107 | { |
70ed7d01 | 108 | // see header file for function documentation |
0c0c9d99 | 109 | int iResult=0; |
110 | if (pConsumer) { | |
6235cd38 | 111 | vector<AliHLTDataBuffer::AliHLTDataSegment> segments; |
0c0c9d99 | 112 | if ((iResult=FindMatchingDataSegments(pConsumer, segments))>=0) { |
113 | if (tgtList) { | |
6235cd38 | 114 | vector<AliHLTDataBuffer::AliHLTDataSegment>::iterator segment=segments.begin(); |
0c0c9d99 | 115 | while (segment!=segments.end()) { |
116 | tgtList->push_back((*segment).fDataType); | |
117 | segment++; | |
118 | } | |
119 | } | |
120 | iResult=segments.size(); | |
121 | } | |
122 | } else { | |
123 | iResult=-EINVAL; | |
124 | } | |
125 | return iResult; | |
126 | } | |
127 | ||
6235cd38 | 128 | int AliHLTDataBuffer::FindMatchingDataSegments(const AliHLTComponent* pConsumer, vector<AliHLTDataBuffer::AliHLTDataSegment>& tgtList) |
0c0c9d99 | 129 | { |
70ed7d01 | 130 | // see header file for function documentation |
0c0c9d99 | 131 | int iResult=0; |
9b7fe12d | 132 | |
133 | // Matthias 26.09.2007 relax the restriction to matching data blocks | |
134 | // all blocks are passed to the consumer, which is the policy also in | |
135 | // PubSub | |
136 | tgtList.assign(fSegments.begin(), fSegments.end()); | |
137 | iResult=tgtList.size(); | |
138 | return iResult; | |
139 | ||
0c0c9d99 | 140 | if (pConsumer) { |
8ede8717 | 141 | vector<AliHLTComponentDataType> dtlist; |
0c0c9d99 | 142 | ((AliHLTComponent*)pConsumer)->GetInputDataTypes(dtlist); |
6235cd38 | 143 | vector<AliHLTDataBuffer::AliHLTDataSegment>::iterator segment=fSegments.begin(); |
0c0c9d99 | 144 | while (segment!=fSegments.end()) { |
8ede8717 | 145 | vector<AliHLTComponentDataType>::iterator type=dtlist.begin(); |
0c0c9d99 | 146 | while (type!=dtlist.end()) { |
9ce4bf4a | 147 | if ((*segment).fDataType==(*type) || |
148 | (*type)==kAliHLTAnyDataType) { | |
0c0c9d99 | 149 | tgtList.push_back(*segment); |
150 | iResult++; | |
151 | break; | |
152 | } | |
153 | type++; | |
154 | } | |
155 | segment++; | |
156 | } | |
157 | } else { | |
158 | iResult=-EINVAL; | |
159 | } | |
160 | return iResult; | |
161 | } | |
162 | ||
8ede8717 | 163 | int AliHLTDataBuffer::Subscribe(const AliHLTComponent* pConsumer, AliHLTComponentBlockData* arrayBlockDesc, int iArraySize) |
3f2a1b1c | 164 | { |
70ed7d01 | 165 | // see header file for function documentation |
3f2a1b1c | 166 | int iResult=0; |
0c0c9d99 | 167 | if (pConsumer && arrayBlockDesc) { |
3f2a1b1c | 168 | if (fpBuffer) { |
0c0c9d99 | 169 | AliHLTConsumerDescriptor* pDesc=FindConsumer(pConsumer, fConsumers); |
3f2a1b1c | 170 | if (pDesc) { |
6235cd38 | 171 | vector<AliHLTDataBuffer::AliHLTDataSegment> tgtList; |
9b7fe12d | 172 | // Matthias 26.07.2007 AliHLTSystem should behave the same way as PubSub |
173 | // so it does not matter if there are matching data types or not, unless | |
174 | // we implement such a check in PubSub | |
175 | if ((iResult=FindMatchingDataSegments(pConsumer, tgtList))>=0) { | |
0c0c9d99 | 176 | int i =0; |
6235cd38 | 177 | vector<AliHLTDataBuffer::AliHLTDataSegment>::iterator segment=tgtList.begin(); |
0c0c9d99 | 178 | while (segment!=tgtList.end() && i<iArraySize) { |
3f2a1b1c | 179 | // fill the block data descriptor |
8ede8717 | 180 | arrayBlockDesc[i].fStructSize=sizeof(AliHLTComponentBlockData); |
3f2a1b1c | 181 | // the shared memory key is not used in AliRoot |
8ede8717 | 182 | arrayBlockDesc[i].fShmKey.fStructSize=sizeof(AliHLTComponentShmData); |
183 | arrayBlockDesc[i].fShmKey.fShmType=gkAliHLTComponentInvalidShmType; | |
184 | arrayBlockDesc[i].fShmKey.fShmID=gkAliHLTComponentInvalidShmID; | |
3294f81a | 185 | // This models the behavior of PubSub. |
186 | // For incoming data blocks, fOffset must be ignored by the | |
187 | // processing component. It is set for bookkeeping in the framework. | |
188 | // fPtr always points to the beginning of the data. | |
0c0c9d99 | 189 | arrayBlockDesc[i].fOffset=(*segment).fSegmentOffset; |
3294f81a | 190 | AliHLTUInt8_t* pTgt=*fpBuffer; |
191 | pTgt+=(*segment).fSegmentOffset; | |
192 | arrayBlockDesc[i].fPtr=reinterpret_cast<void*>(pTgt); | |
0c0c9d99 | 193 | arrayBlockDesc[i].fSize=(*segment).fSegmentSize; |
194 | arrayBlockDesc[i].fDataType=(*segment).fDataType; | |
195 | arrayBlockDesc[i].fSpecification=(*segment).fSpecification; | |
196 | pDesc->SetActiveDataSegment(arrayBlockDesc[i].fOffset, arrayBlockDesc[i].fSize); | |
1e6e67ec | 197 | HLTDebug("component %p (%s) subscribed to segment #%d offset %d size %d data type %s %#x", |
198 | pConsumer, ((AliHLTComponent*)pConsumer)->GetComponentID(), i, arrayBlockDesc[i].fOffset, | |
199 | arrayBlockDesc[i].fSize, (AliHLTComponent::DataType2Text(arrayBlockDesc[i].fDataType)).c_str(), | |
200 | arrayBlockDesc[i].fSpecification); | |
0c0c9d99 | 201 | i++; |
202 | segment++; | |
203 | } | |
9b7fe12d | 204 | // check whether there was enough space for the segments |
298ef463 | 205 | if (i!=(int)tgtList.size()) { |
9b7fe12d | 206 | HLTError("too little space in block descriptor array: required %d, available %d", tgtList.size(), iArraySize); |
207 | iResult=-ENOSPC; | |
208 | } else { | |
0c0c9d99 | 209 | // move this consumer to the active list |
210 | if (ChangeConsumerState(pDesc, fConsumers, fActiveConsumers)>=0) { | |
211 | HLTDebug("component %p (%s) subscribed to data buffer %p", pConsumer, ((AliHLTComponent*)pConsumer)->GetComponentID(), this); | |
3f2a1b1c | 212 | } else { |
0c0c9d99 | 213 | // TODO: cleanup the consumer descriptor correctly |
8ede8717 | 214 | memset(arrayBlockDesc, 0, iArraySize*sizeof(AliHLTComponentBlockData)); |
3f2a1b1c | 215 | HLTError("can not activate consumer %p for data buffer %p", pConsumer, this); |
216 | iResult=-EACCES; | |
217 | } | |
9b7fe12d | 218 | } |
3f2a1b1c | 219 | } else { |
0c0c9d99 | 220 | HLTError("unresolved data segment(s) for component %p (%s)", pConsumer, ((AliHLTComponent*)pConsumer)->GetComponentID()); |
3f2a1b1c | 221 | iResult=-EBADF; |
222 | } | |
223 | } else { | |
0c0c9d99 | 224 | HLTError("component %p is not a data consumer of data buffer %s", pConsumer, this); |
3f2a1b1c | 225 | iResult=-ENOENT; |
226 | } | |
227 | } else { | |
9b7fe12d | 228 | // Matthias 26.07.2007 until now, data had to be present for successful subscription |
229 | // in order to be consistent with the PubSub framework, this restiction has been | |
230 | // removed | |
231 | //HLTError("data buffer %p is empty", this); | |
232 | //iResult=-ENODATA; | |
3f2a1b1c | 233 | } |
234 | } else { | |
235 | HLTError("invalid parameter"); | |
236 | iResult=-EINVAL; | |
237 | } | |
238 | return iResult; | |
239 | } | |
240 | ||
8ede8717 | 241 | int AliHLTDataBuffer::Release(AliHLTComponentBlockData* pBlockDesc, const AliHLTComponent* pConsumer) |
3f2a1b1c | 242 | { |
70ed7d01 | 243 | // see header file for function documentation |
3f2a1b1c | 244 | int iResult=0; |
245 | if (pBlockDesc && pConsumer) { | |
0c0c9d99 | 246 | AliHLTConsumerDescriptor* pDesc=FindConsumer(pConsumer, fActiveConsumers); |
247 | if (pDesc) { | |
248 | if ((iResult=pDesc->CheckActiveDataSegment(pBlockDesc->fOffset, pBlockDesc->fSize))!=1) { | |
249 | HLTWarning("data segment missmatch, component %p has not subscribed to a segment with offset %#x and size %d", pConsumer, pBlockDesc->fOffset, pBlockDesc->fSize); | |
250 | // TODO: appropriate error handling, but so far optional | |
251 | iResult=0; | |
252 | } else { | |
253 | pDesc->ReleaseActiveDataSegment(pBlockDesc->fOffset, pBlockDesc->fSize); | |
3f2a1b1c | 254 | pBlockDesc->fOffset=0; |
255 | pBlockDesc->fPtr=NULL; | |
256 | pBlockDesc->fSize=0; | |
0c0c9d99 | 257 | } |
258 | if (pDesc->GetNofActiveSegments()==0) { | |
3f2a1b1c | 259 | if ((iResult=ChangeConsumerState(pDesc, fActiveConsumers, fReleasedConsumers))>=0) { |
b426991e | 260 | if (GetNofActiveConsumers()==0 && GetNofPendingConsumers()==0) { |
0c0c9d99 | 261 | // this is the last consumer, reset the consumer list and release the raw buffer |
3f2a1b1c | 262 | ResetDataBuffer(); |
3f2a1b1c | 263 | } |
264 | } else { | |
265 | HLTError("can not deactivate consumer %p for data buffer %p", pConsumer, this); | |
266 | iResult=-EACCES; | |
267 | } | |
3f2a1b1c | 268 | } |
0c0c9d99 | 269 | } else { |
270 | HLTWarning("component %p has currently not subscribed to the data buffer %p", pConsumer, this); | |
271 | iResult=-ENOENT; | |
272 | } | |
3f2a1b1c | 273 | } else { |
274 | HLTError("inavalid parameter: pBlockDesc=%p pConsumer=%p", pBlockDesc, pConsumer); | |
275 | iResult=-EINVAL; | |
276 | } | |
277 | return iResult; | |
278 | } | |
279 | ||
280 | AliHLTUInt8_t* AliHLTDataBuffer::GetTargetBuffer(int iMinSize) | |
281 | { | |
70ed7d01 | 282 | // see header file for function documentation |
3f2a1b1c | 283 | AliHLTUInt8_t* pTargetBuffer=NULL; |
b6800be0 | 284 | if (fpBuffer!=NULL) { |
285 | HLTWarning("data buffer not properly reset, possible memory leak\n"); | |
286 | } | |
3f2a1b1c | 287 | fpBuffer=CreateRawBuffer(iMinSize); |
9ce4bf4a | 288 | if (fpBuffer) { |
1e6e67ec | 289 | pTargetBuffer=*fpBuffer; |
9ce4bf4a | 290 | } else { |
291 | HLTError("can not create raw buffer"); | |
292 | } | |
3f2a1b1c | 293 | return pTargetBuffer; |
294 | } | |
295 | ||
8ede8717 | 296 | int AliHLTDataBuffer::SetSegments(AliHLTUInt8_t* pTgt, AliHLTComponentBlockData* arrayBlockData, int iSize) |
3f2a1b1c | 297 | { |
70ed7d01 | 298 | // see header file for function documentation |
3f2a1b1c | 299 | int iResult=0; |
300 | if (pTgt && arrayBlockData && iSize>=0) { | |
0c0c9d99 | 301 | if (fpBuffer) { |
1e6e67ec | 302 | if (*fpBuffer==pTgt) { |
6235cd38 | 303 | AliHLTDataBuffer::AliHLTDataSegment segment; |
0c0c9d99 | 304 | for (int i=0; i<iSize; i++) { |
3294f81a | 305 | // This function has to model the behavior of PubSub |
306 | // For output blocks only the fOffset value is used, this must be the offset | |
307 | // relative to the output pointer. fPtr must be either NULL or the output | |
308 | // pointer | |
6434d28a | 309 | if (arrayBlockData[i].fPtr==NULL || |
3294f81a | 310 | arrayBlockData[i].fPtr==*fpBuffer) { |
311 | if (arrayBlockData[i].fOffset+arrayBlockData[i].fSize<=fpBuffer->fSize) { | |
312 | segment.fSegmentOffset=arrayBlockData[i].fOffset; | |
1e6e67ec | 313 | segment.fSegmentSize=arrayBlockData[i].fSize; |
314 | segment.fDataType=arrayBlockData[i].fDataType; | |
315 | segment.fSpecification=arrayBlockData[i].fSpecification; | |
316 | fSegments.push_back(segment); | |
317 | HLTDebug("set segment %s with size %d at offset %d", AliHLTComponent::DataType2Text(segment.fDataType).data(), segment.fSegmentSize, segment.fSegmentOffset); | |
318 | } else { | |
319 | HLTError("block data specification %#d (%s) exceeds size of data buffer", i, AliHLTComponent::DataType2Text(arrayBlockData[i].fDataType).data()); | |
320 | HLTError("block offset=%d, block size=%d, buffer size=%d", arrayBlockData[i].fOffset, arrayBlockData[i].fSize, fpBuffer->fSize); | |
321 | iResult=-E2BIG; | |
322 | } | |
0c0c9d99 | 323 | } else { |
3294f81a | 324 | HLTError("invalid pointer (%p) in block data specification (buffer %p size %d)." |
325 | "please note: for output blocks only the fOffset value is valid and must " | |
326 | "be relative to the output buffer", arrayBlockData[i].fPtr, fpBuffer->fPtr, fpBuffer->fSize); | |
1e6e67ec | 327 | iResult=-ERANGE; |
0c0c9d99 | 328 | } |
329 | } | |
3f2a1b1c | 330 | } else { |
0c0c9d99 | 331 | HLTError("this data buffer (%p) does not match the internal data buffer %p of raw buffer %p", pTgt, fpBuffer->fPtr, fpBuffer); |
8451168b | 332 | iResult=-EINVAL; |
3f2a1b1c | 333 | } |
0c0c9d99 | 334 | } else { |
335 | HLTFatal("internal data structur missmatch"); | |
336 | iResult=-EFAULT; | |
3f2a1b1c | 337 | } |
338 | } else { | |
0c0c9d99 | 339 | HLTError("invalid parameter: pTgtBuffer=%p arrayBlockData=%p", pTgt, arrayBlockData); |
3f2a1b1c | 340 | iResult=-EINVAL; |
341 | } | |
342 | return iResult; | |
343 | } | |
344 | ||
345 | int AliHLTDataBuffer::IsEmpty() | |
346 | { | |
70ed7d01 | 347 | // see header file for function documentation |
3f2a1b1c | 348 | int iResult=fpBuffer==NULL || GetNofSegments()==0; |
349 | return iResult; | |
350 | } | |
351 | ||
352 | int AliHLTDataBuffer::GetNofSegments() | |
353 | { | |
70ed7d01 | 354 | // see header file for function documentation |
3f2a1b1c | 355 | int iResult=fSegments.size(); |
356 | return iResult; | |
357 | } | |
358 | ||
359 | int AliHLTDataBuffer::GetNofConsumers() | |
360 | { | |
70ed7d01 | 361 | // see header file for function documentation |
3f2a1b1c | 362 | int iResult=fConsumers.size() + GetNofActiveConsumers() + fReleasedConsumers.size(); |
363 | return iResult; | |
364 | } | |
365 | ||
b426991e | 366 | int AliHLTDataBuffer::GetNofPendingConsumers() |
367 | { | |
368 | // see header file for function documentation | |
369 | int iResult=fConsumers.size(); | |
370 | return iResult; | |
371 | } | |
372 | ||
3f2a1b1c | 373 | int AliHLTDataBuffer::GetNofActiveConsumers() |
374 | { | |
70ed7d01 | 375 | // see header file for function documentation |
3f2a1b1c | 376 | int iResult=fActiveConsumers.size(); |
377 | return iResult; | |
378 | } | |
379 | ||
6235cd38 | 380 | AliHLTDataBuffer::AliHLTRawBuffer* AliHLTDataBuffer::CreateRawBuffer(AliHLTUInt32_t size) |
3f2a1b1c | 381 | { |
70ed7d01 | 382 | // see header file for function documentation |
3f2a1b1c | 383 | AliHLTRawBuffer* pRawBuffer=NULL; |
7a5ccd96 | 384 | unsigned int reqSize=size+fgkSafetyPatternSize; |
70ed7d01 | 385 | vector<AliHLTRawBuffer*>::iterator buffer=fgFreeBuffers.begin(); |
386 | while (buffer!=fgFreeBuffers.end() && pRawBuffer==NULL) { | |
8451168b | 387 | if ((*buffer)->fTotalSize>=reqSize && ((*buffer)->fTotalSize-reqSize)<fgMargin) { |
3f2a1b1c | 388 | // assign this element |
389 | pRawBuffer=*buffer; | |
390 | pRawBuffer->fSize=size; | |
70ed7d01 | 391 | fgFreeBuffers.erase(buffer); |
0c0c9d99 | 392 | 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->fTotalSize, pRawBuffer->fPtr); |
70ed7d01 | 393 | fgActiveBuffers.push_back(pRawBuffer); |
0c0c9d99 | 394 | break; |
3f2a1b1c | 395 | } |
0c0c9d99 | 396 | buffer++; |
3f2a1b1c | 397 | } |
398 | if (pRawBuffer==NULL) { | |
0c0c9d99 | 399 | // no buffer found, create a new one |
3f2a1b1c | 400 | pRawBuffer=new AliHLTRawBuffer; |
401 | if (pRawBuffer) { | |
1e6e67ec | 402 | pRawBuffer->fPtr=static_cast<AliHLTUInt8_t*>(malloc(reqSize)); |
3f2a1b1c | 403 | if (pRawBuffer->fPtr) { |
404 | pRawBuffer->fSize=size; | |
8451168b | 405 | pRawBuffer->fTotalSize=reqSize; |
70ed7d01 | 406 | fgActiveBuffers.push_back(pRawBuffer); |
0c0c9d99 | 407 | fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "new raw buffer %p of size %d created (container %p)", pRawBuffer->fPtr, pRawBuffer->fTotalSize, pRawBuffer); |
3f2a1b1c | 408 | } else { |
409 | delete pRawBuffer; | |
410 | pRawBuffer=NULL; | |
0c0c9d99 | 411 | fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "memory allocation failed"); |
3f2a1b1c | 412 | } |
413 | } else { | |
0c0c9d99 | 414 | fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::CreateRawBuffer", "data buffer handling", "memory allocation failed"); |
3f2a1b1c | 415 | } |
416 | } | |
73ede1d3 | 417 | if (pRawBuffer!=NULL && fgkSafetyPatternSize>0) { |
418 | //fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "writing safety pattern to %p offset %d", pRawBuffer->fPtr, pRawBuffer->fSize); | |
419 | memcpy(((char*)pRawBuffer->fPtr)+pRawBuffer->fSize, fgkSafetyPattern, fgkSafetyPatternSize); | |
420 | } | |
3f2a1b1c | 421 | return pRawBuffer; |
422 | } | |
423 | ||
424 | int AliHLTDataBuffer::ReleaseRawBuffer(AliHLTRawBuffer* pBuffer) | |
425 | { | |
70ed7d01 | 426 | // see header file for function documentation |
3f2a1b1c | 427 | int iResult=0; |
428 | if (pBuffer) { | |
70ed7d01 | 429 | vector<AliHLTRawBuffer*>::iterator buffer=fgActiveBuffers.begin(); |
430 | while (buffer!=fgActiveBuffers.end() && (*buffer)!=pBuffer) { | |
3f2a1b1c | 431 | buffer++; |
432 | } | |
70ed7d01 | 433 | if (buffer!=fgActiveBuffers.end()) { |
8451168b | 434 | if (fgkSafetyPatternSize>0) { |
73ede1d3 | 435 | //fgLogging.Logging(kHLTLogDebug, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "comparing safety pattern at %p offset %d", (*buffer)->fPtr, (*buffer)->fSize); |
8451168b | 436 | if (memcmp(((char*)(*buffer)->fPtr)+(*buffer)->fSize, fgkSafetyPattern, fgkSafetyPatternSize)!=0) { |
437 | fgLogging.Logging(kHLTLogFatal, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "component has written beyond end of data buffer %p size %d", (*buffer)->fPtr, (*buffer)->fSize); | |
438 | } | |
439 | } | |
3f2a1b1c | 440 | (*buffer)->fSize=0; |
70ed7d01 | 441 | fgFreeBuffers.push_back(*buffer); |
442 | fgActiveBuffers.erase(buffer); | |
3f2a1b1c | 443 | } else { |
0c0c9d99 | 444 | fgLogging.Logging(kHLTLogWarning, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "can not find raw buffer container %p in the list of active containers", pBuffer); |
3f2a1b1c | 445 | iResult=-ENOENT; |
446 | } | |
447 | } else { | |
0c0c9d99 | 448 | fgLogging.Logging(kHLTLogError, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "invalid parameter"); |
3f2a1b1c | 449 | iResult=-EINVAL; |
450 | } | |
451 | return iResult; | |
452 | } | |
453 | ||
454 | ||
455 | int AliHLTDataBuffer::DeleteRawBuffers() | |
456 | { | |
70ed7d01 | 457 | // see header file for function documentation |
3f2a1b1c | 458 | int iResult=0; |
70ed7d01 | 459 | vector<AliHLTRawBuffer*>::iterator buffer=fgFreeBuffers.begin(); |
460 | while (buffer!=fgFreeBuffers.end()) { | |
3f2a1b1c | 461 | free((*buffer)->fPtr); |
462 | delete *buffer; | |
70ed7d01 | 463 | fgFreeBuffers.erase(buffer); |
464 | buffer=fgFreeBuffers.begin(); | |
3f2a1b1c | 465 | } |
70ed7d01 | 466 | buffer=fgActiveBuffers.begin(); |
467 | while (buffer!=fgActiveBuffers.end()) { | |
0c0c9d99 | 468 | fgLogging.Logging(kHLTLogWarning, "AliHLTDataBuffer::ReleaseRawBuffer", "data buffer handling", "request to delete active raw buffer container (raw buffer %p, size %d)", (*buffer)->fPtr, (*buffer)->fTotalSize); |
3f2a1b1c | 469 | free((*buffer)->fPtr); |
470 | delete *buffer; | |
70ed7d01 | 471 | fgActiveBuffers.erase(buffer); |
472 | buffer=fgActiveBuffers.begin(); | |
3f2a1b1c | 473 | } |
474 | return iResult; | |
475 | } | |
476 | ||
70ed7d01 | 477 | AliHLTConsumerDescriptor* AliHLTDataBuffer::FindConsumer(const AliHLTComponent* pConsumer, vector<AliHLTConsumerDescriptor*> &list) const |
3f2a1b1c | 478 | { |
70ed7d01 | 479 | // see header file for function documentation |
3f2a1b1c | 480 | AliHLTConsumerDescriptor* pDesc=NULL; |
481 | vector<AliHLTConsumerDescriptor*>::iterator desc=list.begin(); | |
482 | while (desc!=list.end() && pDesc==NULL) { | |
0c0c9d99 | 483 | if ((pConsumer==NULL || (*desc)->GetComponent()==pConsumer)) { |
3f2a1b1c | 484 | pDesc=*desc; |
485 | } | |
0c0c9d99 | 486 | desc++; |
3f2a1b1c | 487 | } |
488 | return pDesc; | |
489 | } | |
490 | ||
0c0c9d99 | 491 | int AliHLTDataBuffer::ResetDataBuffer() |
492 | { | |
70ed7d01 | 493 | // see header file for function documentation |
3f2a1b1c | 494 | int iResult=0; |
0c0c9d99 | 495 | AliHLTRawBuffer* pBuffer=fpBuffer; |
3f2a1b1c | 496 | fpBuffer=NULL; |
9ce4bf4a | 497 | |
498 | // cleanup consumer states | |
b426991e | 499 | vector<AliHLTConsumerDescriptor*>::iterator desc; |
500 | // if (GetNofPendingConsumers()>0) { | |
501 | // desc=fConsumers.begin(); | |
502 | // while (desc!=fConsumers.end()) { | |
503 | // AliHLTComponent* pComp=(*desc)->GetComponent(); | |
504 | // HLTError("internal error: consumer %p (%s %p) did not get data from data buffer %p", *desc, pComp?pComp->GetComponentID():"", pComp, this); | |
505 | // desc++; | |
506 | // } | |
507 | // } | |
508 | desc=fReleasedConsumers.begin(); | |
3f2a1b1c | 509 | while (desc!=fReleasedConsumers.end()) { |
510 | AliHLTConsumerDescriptor* pDesc=*desc; | |
511 | fReleasedConsumers.erase(desc); | |
512 | desc=fReleasedConsumers.begin(); | |
513 | fConsumers.push_back(pDesc); | |
514 | } | |
515 | desc=fActiveConsumers.begin(); | |
516 | while (desc!=fActiveConsumers.end()) { | |
517 | AliHLTConsumerDescriptor* pDesc=*desc; | |
518 | HLTWarning("consumer %p was not released", pDesc); | |
519 | fActiveConsumers.erase(desc); | |
520 | desc=fActiveConsumers.begin(); | |
521 | fConsumers.push_back(pDesc); | |
522 | } | |
9ce4bf4a | 523 | |
524 | // cleanup segments | |
6235cd38 | 525 | vector<AliHLTDataBuffer::AliHLTDataSegment>::iterator segment=fSegments.begin(); |
9ce4bf4a | 526 | while (segment!=fSegments.end()) { |
527 | fSegments.erase(segment); | |
528 | segment=fSegments.begin(); | |
529 | } | |
530 | ||
531 | // cleanup raw buffer | |
532 | if (pBuffer) { | |
533 | ReleaseRawBuffer(pBuffer); | |
534 | } | |
3f2a1b1c | 535 | return iResult; |
536 | } | |
537 | ||
9ce4bf4a | 538 | int AliHLTDataBuffer::Reset() |
539 | { | |
70ed7d01 | 540 | // see header file for function documentation |
9ce4bf4a | 541 | return ResetDataBuffer(); |
542 | } | |
543 | ||
0c0c9d99 | 544 | // this is the version which works on lists of components instead of consumer descriptors |
545 | // int AliHLTDataBuffer::ChangeConsumerState(AliHLTComponent* pConsumer, vector<AliHLTComponent*> &srcList, vector<AliHLTComponent*> &tgtList) | |
546 | // { | |
547 | // int iResult=0; | |
548 | // if (pDesc) { | |
549 | // vector<AliHLTComponent*>::iterator desc=srcList.begin(); | |
550 | // while (desc!=srcList.end()) { | |
551 | // if ((*desc)==pConsumer) { | |
552 | // srcList.erase(desc); | |
553 | // tgtList.push_back(pConsumer); | |
554 | // break; | |
555 | // } | |
556 | // desc++; | |
557 | // } | |
558 | // if (desc==srcList.end()) { | |
559 | // HLTError("can not find consumer component %p in list", pConsumer); | |
560 | // iResult=-ENOENT; | |
561 | // } | |
562 | // } else { | |
563 | // HLTError("invalid parameter"); | |
564 | // iResult=-EINVAL; | |
565 | // } | |
566 | // return iResult; | |
567 | // } | |
568 | ||
3f2a1b1c | 569 | int AliHLTDataBuffer::ChangeConsumerState(AliHLTConsumerDescriptor* pDesc, vector<AliHLTConsumerDescriptor*> &srcList, vector<AliHLTConsumerDescriptor*> &tgtList) |
570 | { | |
70ed7d01 | 571 | // see header file for function documentation |
9ce4bf4a | 572 | int iResult=-ENOENT; |
3f2a1b1c | 573 | if (pDesc) { |
574 | vector<AliHLTConsumerDescriptor*>::iterator desc=srcList.begin(); | |
575 | while (desc!=srcList.end()) { | |
576 | if ((*desc)==pDesc) { | |
577 | srcList.erase(desc); | |
578 | tgtList.push_back(pDesc); | |
9ce4bf4a | 579 | iResult=0; |
3f2a1b1c | 580 | break; |
581 | } | |
0c0c9d99 | 582 | desc++; |
3f2a1b1c | 583 | } |
9ce4bf4a | 584 | if (iResult<0) { |
3f2a1b1c | 585 | HLTError("can not find consumer descriptor %p in list", pDesc); |
3f2a1b1c | 586 | } |
587 | } else { | |
588 | HLTError("invalid parameter"); | |
589 | iResult=-EINVAL; | |
590 | } | |
591 | return iResult; | |
592 | } | |
593 | ||
70ed7d01 | 594 | int AliHLTDataBuffer::CleanupConsumerList() |
595 | { | |
596 | // see header file for function documentation | |
3f2a1b1c | 597 | int iResult=0; |
598 | ResetDataBuffer(); | |
599 | vector<AliHLTConsumerDescriptor*>::iterator desc=fConsumers.begin(); | |
600 | while (desc!=fConsumers.end()) { | |
601 | delete *desc; | |
602 | fConsumers.erase(desc); | |
603 | desc=fConsumers.begin(); | |
604 | } | |
605 | return iResult; | |
606 | } | |
9ce4bf4a | 607 | |
70ed7d01 | 608 | int AliHLTDataBuffer::FindConsumer(AliHLTComponent* pConsumer, int bAllLists) |
609 | { | |
610 | // see header file for function documentation | |
9ce4bf4a | 611 | vector<AliHLTConsumerDescriptor*>::iterator desc=fConsumers.begin(); |
612 | while (desc!=fConsumers.end()) { | |
613 | if ((*desc)->GetComponent()==pConsumer) | |
614 | return 1; | |
615 | desc++; | |
616 | } | |
617 | if (bAllLists==0) return 0; | |
618 | ||
619 | desc=fActiveConsumers.begin(); | |
620 | while (desc!=fActiveConsumers.end()) { | |
621 | if ((*desc)->GetComponent()==pConsumer) | |
622 | return 1; | |
623 | desc++; | |
624 | } | |
625 | desc=fReleasedConsumers.begin(); | |
626 | while (desc!=fReleasedConsumers.end()) { | |
627 | if ((*desc)->GetComponent()==pConsumer) | |
628 | return 1; | |
629 | desc++; | |
630 | } | |
631 | return 0; | |
632 | } | |
1e6e67ec | 633 | |
634 | int AliHLTDataBuffer::AliHLTRawBuffer::operator==(void* ptr) | |
635 | { | |
636 | return fPtr == static_cast<AliHLTUInt8_t*>(ptr); | |
637 | } | |
638 | ||
639 | int AliHLTDataBuffer::AliHLTRawBuffer::operator<=(void* ptr) | |
640 | { | |
641 | int iResult=fPtr <= static_cast<AliHLTUInt8_t*>(ptr); | |
642 | //printf("%p: %p <= %p (%d)\n", this, fPtr, ptr, iResult); | |
643 | return iResult; | |
644 | } | |
645 | ||
646 | int AliHLTDataBuffer::AliHLTRawBuffer::operator>(void* ptr) | |
647 | { | |
648 | int iResult=fPtr+fSize > static_cast<AliHLTUInt8_t*>(ptr); | |
649 | //printf("%p: %p + %d > %p (%d)\n", this, fPtr, fSize, ptr, iResult); | |
650 | return iResult; | |
651 | } | |
652 | ||
653 | int AliHLTDataBuffer::AliHLTRawBuffer::operator-(void* ptr) | |
654 | { | |
655 | return static_cast<int>(static_cast<AliHLTUInt8_t*>(ptr)-fPtr); | |
656 | } |