c5123824 |
1 | //-*- Mode: C++ -*- |
2 | // $Id$ |
3 | |
f1207f29 |
4 | #ifndef ALIHLTESDMANAGERIMPLEMENTATION_H |
5 | #define ALIHLTESDMANAGERIMPLEMENTATION_H |
c5123824 |
6 | //* This file is property of and copyright by the ALICE HLT Project * |
7 | //* ALICE Experiment at CERN, All rights reserved. * |
8 | //* See cxx source for full Copyright notice * |
9 | |
04df9166 |
10 | /// @file AliHLTEsdManagerImplementation.h |
11 | /// @author Matthias Richter |
12 | /// @date |
13 | /// @brief Implementation of the AliHLTEsdManager |
14 | /// This is an implementation of the abstract interface AliHLTEsdManager |
c5123824 |
15 | |
f1207f29 |
16 | #include "AliHLTEsdManager.h" |
f5ea7ce2 |
17 | #include "AliESDEvent.h" |
90c37647 |
18 | #include "TString.h" |
c5123824 |
19 | #include <vector> |
20 | |
c5123824 |
21 | class TTree; |
22 | class TFile; |
23 | |
24 | /** |
f1207f29 |
25 | * @class AliHLTEsdManagerImplementation |
26 | * Implementation of the AliHLTEsdManager. |
83cb7e1d |
27 | * |
f1207f29 |
28 | * For the sake of library (in)dependencies, the concrete implementation of |
29 | * the AliHLTEsdManager is separated from the libHLTbase class as this would |
30 | * introduce dependencies to AliRoot libraries. See AliHLTEsdManager for |
31 | * usage. |
83cb7e1d |
32 | * |
9d4cf2ab |
33 | * @ingroup alihlt_aliroot_reconstruction |
c5123824 |
34 | */ |
f1207f29 |
35 | class AliHLTEsdManagerImplementation : public AliHLTEsdManager { |
c5123824 |
36 | public: |
37 | /** constructor */ |
f1207f29 |
38 | AliHLTEsdManagerImplementation(); |
c5123824 |
39 | /** destructor */ |
f1207f29 |
40 | virtual ~AliHLTEsdManagerImplementation(); |
c5123824 |
41 | |
794de106 |
42 | /** |
43 | * Inherited from base class, see AliHLTEsdManager::SetOption() for |
44 | * documentation. |
45 | */ |
46 | virtual int SetOption(const char* option); |
47 | |
c5123824 |
48 | /** |
49 | * Convert data buffer to ESD. |
50 | * The buffer is supposed to describe a streamed AliESDEvent object. |
51 | * If no target object is specified, the ESD is written to a file AliHLTdetESDs.root, |
52 | * where 'det' is derived from the data type origin. Each time the function is invoked |
53 | * a new event is created. Dummy events are added if the previous events did not contain |
54 | * |
55 | * @param pBuffer [in] the data buffer |
56 | * @param size [in] data buffer size |
57 | * @param dt [in] data type of the block |
58 | * @param tgtesd [out] optional target |
59 | * @param eventno [in] optional event no |
60 | */ |
61 | int WriteESD(const AliHLTUInt8_t* pBuffer, AliHLTUInt32_t size, AliHLTComponentDataType dt, |
62 | AliESDEvent* tgtesd=NULL, int eventno=-1); |
63 | |
57584811 |
64 | /** |
65 | * Merge content of source ESD into the target ESD. |
66 | * Merging is done on the level of objects in the ESD and for the |
67 | * moment it's only implemented for the TClonesArrays. In that case it's |
68 | * easy to detect whether the object is empty or not. |
69 | * |
70 | * \b Note: The function can not match entries of the same type, like e.g. |
71 | * tracks from the 'Tracks' member. |
72 | */ |
73 | int Merge(AliESDEvent* pTgt, AliESDEvent* pSrc) const; |
74 | |
90c37647 |
75 | /** |
76 | * Align all ESD to the same number of events. |
77 | * The function adds empty events to all ESD files if their event number |
78 | * does not match the specified one. |
79 | * @param eventno the desired event no |
80 | * @return neg. error code if failed |
81 | */ |
82 | int PadESDs(int eventno); |
83 | |
84 | /** |
85 | * Set the target directory for the ESD files. |
86 | */ |
87 | void SetDirectory(const char* directory); |
88 | |
89 | /** |
90 | * Get the list of the internally created files. |
91 | * Returns a blank separated list of the file names. |
92 | */ |
93 | TString GetFileNames(AliHLTComponentDataType dt=kAliHLTAnyDataType) const; |
94 | |
95 | /** |
96 | * Embed an ESD into a TTree object. |
97 | * The tree object needs to be deleted by the caller. |
98 | */ |
99 | static TTree* EmbedIntoTree(AliESDEvent* pESD, const char* name="esdTree", const char* title="Tree with HLT ESD objects"); |
100 | |
f5ea7ce2 |
101 | class AliHLTESDEventHelper : public AliESDEvent { |
102 | public: |
103 | static bool IsStdContent(const char* key); |
104 | }; |
105 | |
5b687e5d |
106 | int CheckClassConditions() const; |
107 | |
c5123824 |
108 | protected: |
109 | |
110 | private: |
111 | /** copy constructor prohibited */ |
f1207f29 |
112 | AliHLTEsdManagerImplementation(const AliHLTEsdManagerImplementation&); |
c5123824 |
113 | /** assignment operator prohibited */ |
f1207f29 |
114 | AliHLTEsdManagerImplementation& operator=(const AliHLTEsdManagerImplementation&); |
c5123824 |
115 | |
116 | class AliHLTEsdListEntry : public AliHLTLogging { |
117 | public: |
118 | /** constructor */ |
119 | AliHLTEsdListEntry(AliHLTComponentDataType dt); |
c5123824 |
120 | /** destructor */ |
121 | ~AliHLTEsdListEntry(); |
122 | |
123 | /** |
124 | * Write the ESD to the corresponding file. |
125 | * The tree is first synchronized with the eventno and additional empty |
126 | * events might be inserted if there was a gap. Since we are writing |
127 | * several files in parallel, we have to make sure that those files contain |
128 | * the same number of events. |
129 | * @param pESD ESD to write |
130 | * @param eventno optional event no for tree synchronization |
131 | */ |
132 | int WriteESD(AliESDEvent* pESD, int eventno=-1); |
133 | |
90c37647 |
134 | /** |
135 | * Set the target directory for the ESD file. |
136 | */ |
137 | void SetDirectory(const char* directory); |
138 | |
90c37647 |
139 | /** |
140 | * Get name of the ESD file. |
141 | */ |
142 | const char* GetFileName() const; |
143 | |
f527516f |
144 | /** |
145 | * Get the object name prefix generated from the data origin |
146 | * The prefix is added to the names of the ESD objects when copied to the |
147 | * master ESD. |
148 | */ |
149 | const char* GetPrefix(); |
150 | |
c5123824 |
151 | bool operator==(AliHLTComponentDataType dt) const; |
152 | |
153 | private: |
62ff1e23 |
154 | /** copy constructor prohibited */ |
155 | AliHLTEsdListEntry(const AliHLTEsdListEntry& src); |
156 | /** assignment operator prohibited */ |
157 | AliHLTEsdListEntry& operator=(const AliHLTEsdListEntry& src); |
c5123824 |
158 | |
159 | /** root file name */ |
160 | TString fName; //!transient |
90c37647 |
161 | /** target directory */ |
162 | TString fDirectory; //!transient |
c5123824 |
163 | /** data type of the corresponding block */ |
164 | AliHLTComponentDataType fDt; //!transient |
83cb7e1d |
165 | /** the root file for this esd */ |
166 | TFile* fpFile; //!transient |
167 | /** the tree for this esd */ |
168 | TTree* fpTree; //!transient |
169 | /** the esd to fill into the tree */ |
170 | AliESDEvent* fpEsd; //!transient |
f527516f |
171 | /** Prefix for generated ESD objects in the master ESD */ |
172 | TString fPrefix; //!transient |
c5123824 |
173 | }; |
174 | |
62ff1e23 |
175 | typedef vector<AliHLTEsdListEntry*> AliHLTEsdPList; |
c5123824 |
176 | |
177 | /** |
178 | * Find list entry for given data type |
179 | */ |
180 | AliHLTEsdListEntry* Find(AliHLTComponentDataType dt) const; |
181 | |
182 | /** the list of the ESDs */ |
62ff1e23 |
183 | AliHLTEsdPList fESDs; //!transient |
c5123824 |
184 | |
90c37647 |
185 | /** target directory */ |
186 | TString fDirectory; //!transient |
794de106 |
187 | /** write local files */ |
188 | bool fWriteLocal; //!transient |
90c37647 |
189 | |
794de106 |
190 | ClassDef(AliHLTEsdManagerImplementation, 2) |
c5123824 |
191 | }; |
90c37647 |
192 | |
c5123824 |
193 | #endif |