]>
Commit | Line | Data |
---|---|---|
3f2a1b1c | 1 | // @(#) $Id$ |
2 | ||
3 | #ifndef ALIHLTDATABUFFER_H | |
4 | #define ALIHLTDATABUFFER_H | |
5 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
6 | * See cxx source for full Copyright notice */ | |
7 | ||
b22e91eb | 8 | /** @file AliHLTDataBuffer.h |
9 | @author Matthias Richter | |
10 | @date | |
11 | @brief Handling of Data Buffers for HLT components. | |
12 | @note The class is used in Offline (AliRoot) context | |
13 | */ | |
3f2a1b1c | 14 | |
30338a30 | 15 | // see below for class documentation |
16 | // or | |
17 | // refer to README to build package | |
18 | // or | |
19 | // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt | |
20 | ||
8ede8717 | 21 | #include <vector> |
2be3f004 | 22 | #include "TObject.h" |
3f2a1b1c | 23 | #include "AliHLTLogging.h" |
24 | #include "AliHLTDataTypes.h" | |
2be3f004 | 25 | #include "AliHLTComponent.h" |
3f2a1b1c | 26 | |
6235cd38 | 27 | class AliHLTConsumerDescriptor; |
3f2a1b1c | 28 | |
2be3f004 | 29 | /** list of AliHLTConsumerDescriptor pointers */ |
30 | typedef vector<AliHLTConsumerDescriptor*> AliHLTConsumerDescriptorPList; | |
31 | ||
0c0c9d99 | 32 | /** |
bfccbf68 | 33 | * @class AliHLTDataBuffer |
34 | * @brief Handling of data buffers for the HLT. | |
b22e91eb | 35 | * |
70ed7d01 | 36 | * The class provides handling of data buffers for HLT tasks. Each task gets |
37 | * its own Data Buffer instance. The buffer is grouped into different data | |
38 | * segments according to the output of the component.<br> | |
39 | * The Data Buffer keeps control over the data requests of the 'child' | |
40 | * components. Each component can subscribe to a certain segment of the data | |
41 | * buffer. It's state is then changed from 'reserved' to 'active'. After the | |
42 | * data processing, the component has to release the segment and it's state is | |
43 | * set to 'processed'. If all components have requested and released their data, | |
44 | * the Raw Buffer is released and pushed back in the list of available buffers. | |
b22e91eb | 45 | * |
46 | * @note This class is only used for the @ref alihlt_system. | |
47 | * | |
48 | * @ingroup alihlt_system | |
0c0c9d99 | 49 | */ |
6235cd38 | 50 | class AliHLTDataBuffer : public TObject, public AliHLTLogging |
51 | { | |
3f2a1b1c | 52 | public: |
70ed7d01 | 53 | ////////////////////////////////////////////////////////////////////////////// |
6235cd38 | 54 | // constructors and destructors |
0c0c9d99 | 55 | |
56 | /* standard constructor | |
57 | */ | |
3f2a1b1c | 58 | AliHLTDataBuffer(); |
85869391 | 59 | /** destructor */ |
3f2a1b1c | 60 | virtual ~AliHLTDataBuffer(); |
61 | ||
70ed7d01 | 62 | ////////////////////////////////////////////////////////////////////////////// |
3f2a1b1c | 63 | // initialization |
64 | ||
0c0c9d99 | 65 | /** |
66 | * Add component to the list of consumers | |
67 | * @param pConsumer - a consumer of type AliHLTComponent | |
3f2a1b1c | 68 | */ |
0c0c9d99 | 69 | int SetConsumer(AliHLTComponent* pConsumer); |
3f2a1b1c | 70 | |
70ed7d01 | 71 | ////////////////////////////////////////////////////////////////////////////// |
3f2a1b1c | 72 | // component to component communication |
73 | ||
0c0c9d99 | 74 | /** |
70ed7d01 | 75 | * Determine the number of matching data blocks for the component and a |
76 | * consumer component. <br> | |
77 | * The first approach will support only one output data type for processing | |
78 | * components. | |
0c0c9d99 | 79 | * @param pConsumer the component which subscribes to the buffer |
80 | * @param tgtList (optional) the list to receive the data types | |
81 | * @return: number of data blocks which match the input data types | |
82 | * of the consumer, neg. error code if failed <br> | |
83 | * -EINVAL invalid parameter <br> | |
3f2a1b1c | 84 | */ |
70ed7d01 | 85 | int FindMatchingDataBlocks(const AliHLTComponent* pConsumer, |
2be3f004 | 86 | AliHLTComponentDataTypeList* tgtList=NULL); |
0c0c9d99 | 87 | |
88 | /** | |
89 | * Subscribe to a segment of the data buffer. | |
70ed7d01 | 90 | * The function prepares the block descriptor for subsequent use with the |
91 | * AliHLTComponent::ProcessEvent method, the method can prepare several block | |
92 | * descriptors up to the array size specified by iArraySize. The return value | |
93 | * is independent from the array size the number of block descriptors which | |
94 | * would have been prepared if there was enough space in the array<br> | |
0c0c9d99 | 95 | * The method is used by the consumer component. |
96 | * @param pConsumer the component which subscribes to the buffer | |
97 | * @param arrayBlockDesc pointer to block descriptor to be filled | |
98 | * @param iArraySize size of the block descriptor array | |
70ed7d01 | 99 | * @return: number of matching data blocks, neg. error code if failed<br> |
100 | * -EACCESS the consumer state can't be changed (activated) | |
0c0c9d99 | 101 | * -EBADF unresolved data segments <br> |
102 | * -ENOENT consumer component not found <br> | |
103 | * -ENODATA data buffer does not have raw data <br> | |
104 | * -EINVAL invalid parameter <br> | |
105 | */ | |
70ed7d01 | 106 | int Subscribe(const AliHLTComponent* pConsumer, |
107 | AliHLTComponentBlockData* arrayBlockDesc, | |
108 | int iArraySize); | |
0c0c9d99 | 109 | |
110 | /** | |
111 | * Release an instance of the data buffer. | |
112 | * Resets the variables of the block descriptor. | |
113 | * If all buffer segments are released, the Data Buffer is reseted | |
114 | * and the Raw Buffer released.<br> | |
115 | * The method is used by the consumer component. | |
116 | * @param pBlockDesc descriptor of the data segment | |
117 | * @param pConsumer the component which subscribes to the buffer | |
118 | * @return: >0 if success, negative error code if failed <br> | |
70ed7d01 | 119 | * -EACCESS the consumer state can not be changed (de-activated) |
120 | * -ENOENT consumer has not subscribed to the buffer <br> | |
0c0c9d99 | 121 | * -EINVAL invalid parameter <br> |
3f2a1b1c | 122 | */ |
8ede8717 | 123 | int Release(AliHLTComponentBlockData* pBlockDesc, const AliHLTComponent* pConsumer); |
3f2a1b1c | 124 | |
0c0c9d99 | 125 | /** |
126 | * Get a target buffer of minimum size iMinSize. | |
127 | * The method is used by the component which owns the Data Buffer to | |
128 | * allocate a buffer for the data it is going to produce. | |
129 | * @param iMinSize minumum size of the requested buffer | |
130 | * @return: pointer to target buffer if | |
3f2a1b1c | 131 | */ |
132 | AliHLTUInt8_t* GetTargetBuffer(int iMinSize); | |
133 | ||
0c0c9d99 | 134 | /** |
135 | * Set the segments for the data buffer. | |
136 | * This is usually done after the component has written the data to the buffer, | |
137 | * which was requested by the @ref GetTargetBuffer method. The component might | |
138 | * produce different types of data, for each type a segment has to be defined | |
2d7ff710 | 139 | * which describes the data inside the buffer.<br> |
70ed7d01 | 140 | * The @ref AliHLTComponentBlockData segment descriptor comes directly from |
141 | * the @ref AliHLTComponent::ProcessEvent method. | |
0c0c9d99 | 142 | * @param pTgt the target buffer which the segments refer to |
143 | * @param arraySegments the output block descriptors of the component | |
144 | * @param iSize size of the array | |
3f2a1b1c | 145 | */ |
8ede8717 | 146 | int SetSegments(AliHLTUInt8_t* pTgt, AliHLTComponentBlockData* arraySegments, int iSize); |
3f2a1b1c | 147 | |
0c0c9d99 | 148 | /** |
149 | * Check if the data buffer is empty. | |
150 | * @return 1 if empty, 0 if not | |
3f2a1b1c | 151 | */ |
152 | int IsEmpty(); | |
153 | ||
0c0c9d99 | 154 | /** |
155 | * Get the total and maximum size of the buffer. | |
156 | * Lets see if this is needed later | |
3f2a1b1c | 157 | */ |
158 | //int GetTotalSize(); | |
159 | ||
0c0c9d99 | 160 | /** |
161 | * Get the number of segments | |
162 | * @return number of segments | |
3f2a1b1c | 163 | */ |
164 | int GetNofSegments(); | |
165 | ||
0c0c9d99 | 166 | /** |
b426991e | 167 | * Get the total number of consumers. |
168 | * This gives the number of consumers regardless of their state. | |
0c0c9d99 | 169 | * @return number of consumers |
3f2a1b1c | 170 | */ |
171 | int GetNofConsumers(); | |
172 | ||
0c0c9d99 | 173 | /** |
b426991e | 174 | * Get the number of consumers which still need to be processed during |
175 | * the current event. | |
176 | * @return number of consumers | |
177 | */ | |
178 | int GetNofPendingConsumers(); | |
179 | ||
180 | /** | |
181 | * Get the number of consumers currently under processing. | |
0c0c9d99 | 182 | * @return number of active consumers |
3f2a1b1c | 183 | */ |
184 | int GetNofActiveConsumers(); | |
185 | ||
9ce4bf4a | 186 | /** |
187 | * Check if a consumer is already in the list | |
188 | * @param pConsumer pointer to consumer component | |
189 | * @param bAllLists search in all lists if 1 | |
190 | * search only in fConsumer list if 0 | |
191 | * @return 1 if found, 0 if not | |
192 | */ | |
193 | int FindConsumer(AliHLTComponent* pConsumer, int bAllLists=1); | |
194 | ||
195 | /** | |
196 | * Public method to reset the buffer. | |
197 | * Eventually with some additional checks. In normal operation, | |
198 | * an external reset should not be necessary. | |
199 | */ | |
200 | int Reset(); | |
201 | ||
dba03d72 | 202 | /** |
203 | * Set local logging level | |
204 | * logging filter for individual object | |
205 | */ | |
206 | void SetLocalLoggingLevel(AliHLTComponentLogSeverity level) | |
207 | {fgLogging.SetLocalLoggingLevel(level); AliHLTLogging::SetLocalLoggingLevel(level);} | |
208 | ||
6235cd38 | 209 | /** |
965919c8 | 210 | * @class AliHLTDataSegment |
6235cd38 | 211 | * @brief Descriptor of a data segment within the buffer. |
212 | */ | |
66043029 | 213 | class AliHLTDataSegment { |
5df0cbb9 | 214 | friend class AliHLTDataBuffer; |
215 | friend class AliHLTConsumerDescriptor; | |
66043029 | 216 | public: |
6235cd38 | 217 | AliHLTDataSegment() |
218 | : | |
219 | fDataType(), | |
220 | fSegmentOffset(0), | |
221 | fSegmentSize(0), | |
222 | fSpecification(0) | |
223 | { | |
224 | memset(&fDataType, 0, sizeof(AliHLTComponentDataType)); | |
225 | } | |
226 | AliHLTDataSegment(AliHLTUInt32_t offset, AliHLTUInt32_t size) | |
227 | : | |
228 | fDataType(), | |
229 | fSegmentOffset(offset), | |
230 | fSegmentSize(size), | |
231 | fSpecification(0) | |
232 | { | |
233 | memset(&fDataType, 0, sizeof(AliHLTComponentDataType)); | |
234 | } | |
2be3f004 | 235 | |
5df0cbb9 | 236 | private: |
6235cd38 | 237 | /** the data type of this segment */ |
238 | AliHLTComponentDataType fDataType; // see above | |
239 | /** offset in byte within the data buffer */ | |
240 | AliHLTUInt32_t fSegmentOffset; // see above | |
241 | /** size of the actual content */ | |
242 | AliHLTUInt32_t fSegmentSize; // see above | |
243 | /** data specification */ | |
244 | AliHLTUInt32_t fSpecification; // see above | |
245 | }; | |
246 | ||
247 | /** | |
965919c8 | 248 | * @class AliHLTRawBuffer |
6235cd38 | 249 | * @brief Descriptor of the raw data buffer which can host several segments. |
250 | */ | |
66043029 | 251 | class AliHLTRawBuffer { |
252 | public: | |
253 | /** standard constructor */ | |
254 | AliHLTRawBuffer() : fSize(0), fTotalSize(0), fPtr(NULL) {} | |
d6cbe999 | 255 | /** constructor */ |
256 | AliHLTRawBuffer(AliHLTUInt32_t size); | |
257 | /** destructor */ | |
258 | virtual ~AliHLTRawBuffer(); | |
259 | ||
260 | /** | |
261 | * Use a fraction of the buffer. | |
262 | * @param size size in bytes to be used | |
263 | * @return pointer to buffer | |
264 | */ | |
265 | AliHLTUInt8_t* UseBuffer(AliHLTUInt32_t size); | |
266 | ||
267 | /** | |
268 | * Check whether buffer fits for a request. | |
269 | * @param size size of the request in bytes | |
270 | * @return 1 if buffer is big enough, 0 if not | |
271 | */ | |
272 | int CheckSize(AliHLTUInt32_t size) const; | |
273 | ||
c0a2bfc2 | 274 | /** |
275 | * Get used size of the buffer | |
276 | */ | |
277 | AliHLTUInt32_t GetUsedSize() const {return fSize;} | |
278 | ||
d6cbe999 | 279 | /** |
280 | * Get total size of the buffer | |
281 | */ | |
282 | AliHLTUInt32_t GetTotalSize() const {return fTotalSize;} | |
283 | ||
c0a2bfc2 | 284 | /** |
285 | * Get pointer of data buffer | |
286 | */ | |
287 | AliHLTUInt8_t* GetPointer() const {return fPtr;} | |
288 | ||
d6cbe999 | 289 | /** |
290 | * Write check pattern | |
291 | */ | |
292 | int WritePattern(const char* pattern, int size); | |
293 | ||
294 | /** | |
295 | * Check pattern | |
296 | */ | |
297 | int CheckPattern(const char* pattern, int size) const; | |
298 | ||
299 | /** | |
300 | * Reset buffer. | |
301 | * Data buffer remains allocated, used size set to 0 | |
302 | */ | |
303 | int Reset(); | |
1e6e67ec | 304 | |
3a7c0444 | 305 | int operator==(void*) const; |
306 | int operator==(AliHLTUInt8_t* ptr) const {return fPtr==ptr;} | |
307 | int operator<=(void*) const; | |
308 | int operator>(void*) const; | |
309 | int operator-(void*) const; | |
1e6e67ec | 310 | |
3a7c0444 | 311 | operator void*() const {return fPtr;} |
312 | operator AliHLTUInt8_t*() const {return fPtr;} | |
1e6e67ec | 313 | |
5df0cbb9 | 314 | private: |
d6cbe999 | 315 | /** copy constructor prohibited */ |
316 | AliHLTRawBuffer(const AliHLTRawBuffer&); | |
317 | /** assignment operator prohibited */ | |
318 | AliHLTRawBuffer& operator=(const AliHLTRawBuffer&); | |
319 | ||
6235cd38 | 320 | /** size of the currently occupied partition of the buffer */ |
321 | AliHLTUInt32_t fSize; // see above | |
322 | /** total size of the buffer, including safety margin */ | |
323 | AliHLTUInt32_t fTotalSize; // see above | |
324 | /** the buffer */ | |
1e6e67ec | 325 | AliHLTUInt8_t* fPtr; //! transient |
6235cd38 | 326 | }; |
327 | ||
3f2a1b1c | 328 | private: |
b426991e | 329 | /** copy constructor prohibited */ |
330 | AliHLTDataBuffer(const AliHLTDataBuffer&); | |
331 | /** assignment operator prohibited */ | |
332 | AliHLTDataBuffer& operator=(const AliHLTDataBuffer&); | |
333 | ||
0c0c9d99 | 334 | /* lets see if this is needed |
6235cd38 | 335 | AliHLTDataSegment* FindDataSegment(AliHLTComponentDataType datatype); |
0c0c9d99 | 336 | */ |
337 | ||
338 | /** | |
339 | * Find those data segments which match the input types of a component. | |
340 | * @param pConsumer the component which subscribes to the buffer | |
341 | * @param tgtList the list to receive the data segment descriptors | |
342 | * @return: number of data blocks which match the input data types | |
343 | * of the consumer, neg. error code if failed <br> | |
344 | * -EINVAL invalid parameter <br> | |
345 | */ | |
90ebac25 | 346 | int FindMatchingDataSegments(const AliHLTComponent* pConsumer, |
347 | vector<AliHLTDataBuffer::AliHLTDataSegment>& tgtList); | |
3f2a1b1c | 348 | |
0c0c9d99 | 349 | /** |
350 | * Reset the data buffer. | |
9ce4bf4a | 351 | * Removes all consumers back to the @ref fConsumers list, deletes |
352 | * segments and releases the Raw Buffer. | |
3f2a1b1c | 353 | */ |
354 | int ResetDataBuffer(); | |
355 | ||
70ed7d01 | 356 | ////////////////////////////////////////////////////////////////////////////// |
357 | ||
3f2a1b1c | 358 | // the data description |
3f2a1b1c | 359 | |
0c0c9d99 | 360 | // the data segments within this buffer |
70ed7d01 | 361 | vector<AliHLTDataSegment> fSegments; // see above |
3f2a1b1c | 362 | |
0c0c9d99 | 363 | // the list of all consumers which are going to subscribe to the buffer |
2be3f004 | 364 | AliHLTConsumerDescriptorPList fConsumers; // see above |
0c0c9d99 | 365 | // the list of all consumers which are currently subscribed to the buffer |
2be3f004 | 366 | AliHLTConsumerDescriptorPList fActiveConsumers; // see above |
0c0c9d99 | 367 | // the list of all consumers which are already released for the current event |
2be3f004 | 368 | AliHLTConsumerDescriptorPList fReleasedConsumers; // see above |
3f2a1b1c | 369 | |
0c0c9d99 | 370 | // the buffer instance |
70ed7d01 | 371 | AliHLTRawBuffer* fpBuffer; //! transient |
3f2a1b1c | 372 | |
0c0c9d99 | 373 | // flags indicating the state of the buffer |
70ed7d01 | 374 | AliHLTUInt32_t fFlags; // see above |
3f2a1b1c | 375 | |
70ed7d01 | 376 | ////////////////////////////////////////////////////////////////////////////// |
0c0c9d99 | 377 | // global buffer handling, internal use only |
378 | ||
379 | /** | |
380 | * Create a raw buffer of a certain size. | |
381 | * The function tries to find a buffer of the given size (or a bit bigger by a | |
9ddaea75 | 382 | * certain margin @ref fgMargin) from the list of free buffers. |
0c0c9d99 | 383 | * If no buffer is available, a new one is created and added to the buffer handling. |
384 | * @param size min. size of the requested buffer | |
385 | * @return pointer to raw buffer | |
3f2a1b1c | 386 | */ |
387 | static AliHLTRawBuffer* CreateRawBuffer(AliHLTUInt32_t size); | |
388 | ||
0c0c9d99 | 389 | /** |
390 | * Mark a buffer as free. | |
70ed7d01 | 391 | * After the Data Buffer has finnished using the raw buffer, it is released |
392 | * and added to the list of available buffers. | |
0c0c9d99 | 393 | * @param pBuffer the raw buffer to release |
394 | * @return >=0 if succeeded, neg. error code if failed | |
3f2a1b1c | 395 | */ |
396 | static int ReleaseRawBuffer(AliHLTRawBuffer* pBuffer); | |
397 | ||
0c0c9d99 | 398 | /** |
399 | * Deletes all the raw buffers. | |
70ed7d01 | 400 | * When the last Data Buffer object is destructed, all raw data buffers are |
401 | * relesed. | |
3f2a1b1c | 402 | */ |
403 | static int DeleteRawBuffers(); | |
404 | ||
0c0c9d99 | 405 | /** |
406 | * Number of instances of AliHLTDataBuffer. | |
70ed7d01 | 407 | * The statice variable is incremented and decremented in the constructor/ |
408 | * destructor. All internal data structures are cleaned up when the last | |
409 | * instance is exiting. | |
0c0c9d99 | 410 | */ |
70ed7d01 | 411 | static int fgNofInstances; // see above |
0c0c9d99 | 412 | /** global list of free raw buffers */ |
70ed7d01 | 413 | static vector<AliHLTRawBuffer*> fgFreeBuffers; // see above |
0c0c9d99 | 414 | /** global list of currently active raw buffers */ |
70ed7d01 | 415 | static vector<AliHLTRawBuffer*> fgActiveBuffers; // see above |
0c0c9d99 | 416 | /** determines the raw buffer size margin at buffer requests */ |
70ed7d01 | 417 | static AliHLTUInt32_t fgMargin; // see above |
3f2a1b1c | 418 | |
0c0c9d99 | 419 | /** global instance to HLT logging class for static methods */ |
70ed7d01 | 420 | static AliHLTLogging fgLogging; // see above |
3f2a1b1c | 421 | |
8451168b | 422 | /** size of the safety pattern */ |
423 | static const Int_t fgkSafetyPatternSize; // see above | |
424 | ||
425 | /** the safety pattern */ | |
426 | static const char fgkSafetyPattern[]; //!transient | |
427 | ||
70ed7d01 | 428 | ////////////////////////////////////////////////////////////////////////////// |
0c0c9d99 | 429 | // internal helper functions |
430 | ||
431 | /** | |
432 | * Find the consumer descriptor for a certain component and data type in | |
433 | * a list of consumers.<br> | |
70ed7d01 | 434 | * <b>Note:</b> There are three lists which contain the consumers in the |
435 | * different states. | |
0c0c9d99 | 436 | * @param pConsumer pointer to consumer component |
437 | * @param list list where to search for the consumer | |
438 | */ | |
70ed7d01 | 439 | AliHLTConsumerDescriptor* FindConsumer(const AliHLTComponent* pConsumer, |
2be3f004 | 440 | AliHLTConsumerDescriptorPList &list) const; |
0c0c9d99 | 441 | |
442 | /** | |
443 | * Change the state of a consumer. | |
70ed7d01 | 444 | * The state of a consumer is determined by the list it is strored in, the |
445 | * method moves a consumer from the source to the target list. | |
0c0c9d99 | 446 | * @param pDesc pointer to consumer descriptor |
447 | * @param srcList list where the consumer is currently to be found | |
448 | * @param tgtList list where to move the consumer | |
449 | */ | |
70ed7d01 | 450 | int ChangeConsumerState(AliHLTConsumerDescriptor* pDesc, |
2be3f004 | 451 | AliHLTConsumerDescriptorPList &srcList, |
452 | AliHLTConsumerDescriptorPList &tgtList); | |
3f2a1b1c | 453 | |
0c0c9d99 | 454 | /** |
455 | * Cleanup a consumer list. | |
456 | * Release all allocated data structures. <b>Note:</b> Not the component itself! | |
457 | */ | |
3f2a1b1c | 458 | int CleanupConsumerList(); |
459 | ||
460 | ClassDef(AliHLTDataBuffer, 0) | |
461 | }; | |
6235cd38 | 462 | |
3f2a1b1c | 463 | #endif // ALIHLTDATABUFFER_H |