]>
Commit | Line | Data |
---|---|---|
a1408c4b | 1 | // $Id$ |
2 | ||
3 | //************************************************************************** | |
4 | //* This file is property of and copyright by the ALICE HLT Project * | |
5 | //* ALICE Experiment at CERN, All rights reserved. * | |
6 | //* * | |
7 | //* Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no> * | |
8 | //* for The ALICE HLT Project. * | |
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 | ||
a0b62fd1 | 19 | // @file AliHLTGlobalEsdConverterComponent.cxx |
20 | // @author Matthias Richter | |
21 | // @date | |
22 | // @brief Global ESD converter component. | |
23 | // | |
a1408c4b | 24 | |
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 | ||
31 | #include <cassert> | |
32 | #include "AliHLTGlobalEsdConverterComponent.h" | |
33 | #include "AliHLTGlobalBarrelTrack.h" | |
34 | #include "AliHLTExternalTrackParam.h" | |
093b64dc | 35 | #include "AliHLTTrackMCLabel.h" |
6117ef63 | 36 | #include "AliHLTCTPData.h" |
331d51c5 | 37 | #include "AliHLTErrorGuard.h" |
a1408c4b | 38 | #include "AliESDEvent.h" |
39 | #include "AliESDtrack.h" | |
f064ef44 | 40 | #include "AliESDMuonTrack.h" |
a1408c4b | 41 | #include "AliCDBEntry.h" |
42 | #include "AliCDBManager.h" | |
18ada816 | 43 | #include "AliPID.h" |
a1408c4b | 44 | #include "TTree.h" |
45 | #include "TList.h" | |
f064ef44 | 46 | #include "TClonesArray.h" |
9a9467ed | 47 | #include "AliHLTESDCaloClusterMaker.h" |
48 | #include "AliHLTCaloClusterDataStruct.h" | |
49 | #include "AliHLTCaloClusterReader.h" | |
50 | #include "AliESDCaloCluster.h" | |
331d51c5 | 51 | #include "AliESDVZERO.h" |
d9386025 | 52 | #include "AliHLTGlobalVertexerComponent.h" |
a1408c4b | 53 | |
54 | /** ROOT macro for the implementation of ROOT specific class methods */ | |
55 | ClassImp(AliHLTGlobalEsdConverterComponent) | |
56 | ||
57 | AliHLTGlobalEsdConverterComponent::AliHLTGlobalEsdConverterComponent() | |
58 | : AliHLTProcessor() | |
a1408c4b | 59 | , fWriteTree(0) |
093b64dc | 60 | , fVerbosity(0) |
8b7a0f3c | 61 | , fESD(NULL) |
a8714ffa | 62 | , fSolenoidBz(-5.00668) |
57a4102f | 63 | , fBenchmark("EsdConverter") |
a1408c4b | 64 | { |
65 | // see header file for class documentation | |
66 | // or | |
67 | // refer to README to build package | |
68 | // or | |
69 | // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt | |
70 | } | |
71 | ||
72 | AliHLTGlobalEsdConverterComponent::~AliHLTGlobalEsdConverterComponent() | |
73 | { | |
74 | // see header file for class documentation | |
75 | if (fESD) delete fESD; | |
76 | fESD=NULL; | |
77 | } | |
78 | ||
79 | int AliHLTGlobalEsdConverterComponent::Configure(const char* arguments) | |
80 | { | |
81 | // see header file for class documentation | |
82 | int iResult=0; | |
83 | if (!arguments) return iResult; | |
84 | ||
85 | TString allArgs=arguments; | |
86 | TString argument; | |
87 | int bMissingParam=0; | |
88 | ||
89 | TObjArray* pTokens=allArgs.Tokenize(" "); | |
90 | if (pTokens) { | |
91 | for (int i=0; i<pTokens->GetEntries() && iResult>=0; i++) { | |
fe0af255 | 92 | argument=((TObjString*)pTokens->At(i))->GetString(); |
a1408c4b | 93 | if (argument.IsNull()) continue; |
94 | ||
95 | if (argument.CompareTo("-solenoidBz")==0) { | |
96 | if ((bMissingParam=(++i>=pTokens->GetEntries()))) break; | |
75970b8d | 97 | HLTWarning("argument -solenoidBz is deprecated, magnetic field set up globally (%f)", GetBz()); |
a1408c4b | 98 | continue; |
99 | } else { | |
100 | HLTError("unknown argument %s", argument.Data()); | |
101 | iResult=-EINVAL; | |
102 | break; | |
103 | } | |
104 | } | |
105 | delete pTokens; | |
106 | } | |
107 | if (bMissingParam) { | |
108 | HLTError("missing parameter for argument %s", argument.Data()); | |
109 | iResult=-EINVAL; | |
110 | } | |
111 | ||
112 | return iResult; | |
113 | } | |
114 | ||
115 | int AliHLTGlobalEsdConverterComponent::Reconfigure(const char* cdbEntry, const char* chainId) | |
116 | { | |
117 | // see header file for class documentation | |
118 | int iResult=0; | |
75970b8d | 119 | const char* path=NULL; |
a1408c4b | 120 | const char* defaultNotify=""; |
121 | if (cdbEntry) { | |
122 | path=cdbEntry; | |
123 | defaultNotify=" (default)"; | |
124 | } | |
125 | if (path) { | |
126 | HLTInfo("reconfigure from entry %s%s, chain id %s", path, defaultNotify,(chainId!=NULL && chainId[0]!=0)?chainId:"<none>"); | |
127 | AliCDBEntry *pEntry = AliCDBManager::Instance()->Get(path/*,GetRunNo()*/); | |
128 | if (pEntry) { | |
129 | TObjString* pString=dynamic_cast<TObjString*>(pEntry->GetObject()); | |
130 | if (pString) { | |
131 | HLTInfo("received configuration object string: \'%s\'", pString->GetString().Data()); | |
132 | iResult=Configure(pString->GetString().Data()); | |
133 | } else { | |
134 | HLTError("configuration object \"%s\" has wrong type, required TObjString", path); | |
135 | } | |
136 | } else { | |
137 | HLTError("can not fetch object \"%s\" from CDB", path); | |
138 | } | |
139 | } | |
140 | ||
141 | return iResult; | |
142 | } | |
143 | ||
144 | void AliHLTGlobalEsdConverterComponent::GetInputDataTypes(AliHLTComponentDataTypeList& list) | |
145 | { | |
146 | // see header file for class documentation | |
147 | list.push_back(kAliHLTDataTypeTrack); | |
148 | list.push_back(kAliHLTDataTypeTrackMC); | |
9a9467ed | 149 | list.push_back(kAliHLTDataTypeCaloCluster); |
0973c527 | 150 | list.push_back(kAliHLTDataTypedEdx ); |
7f167a74 | 151 | list.push_back(kAliHLTDataTypeESDVertex ); |
f064ef44 | 152 | list.push_back(kAliHLTDataTypeESDObject); |
153 | list.push_back(kAliHLTDataTypeTObject); | |
d9963894 | 154 | list.push_back(kAliHLTDataTypeGlobalVertexer); |
331d51c5 | 155 | list.push_back(kAliHLTDataTypeESDContent); |
a1408c4b | 156 | } |
157 | ||
158 | AliHLTComponentDataType AliHLTGlobalEsdConverterComponent::GetOutputDataType() | |
159 | { | |
160 | // see header file for class documentation | |
7a9de6d1 | 161 | return kAliHLTDataTypeESDObject|kAliHLTDataOriginOut; |
a1408c4b | 162 | } |
163 | ||
164 | void AliHLTGlobalEsdConverterComponent::GetOutputDataSize(unsigned long& constBase, double& inputMultiplier) | |
165 | { | |
166 | // see header file for class documentation | |
167 | constBase=2000000; | |
168 | inputMultiplier=10.0; | |
169 | } | |
170 | ||
171 | int AliHLTGlobalEsdConverterComponent::DoInit(int argc, const char** argv) | |
172 | { | |
173 | // see header file for class documentation | |
174 | int iResult=0; | |
175 | TString argument=""; | |
176 | int bMissingParam=0; | |
a1408c4b | 177 | |
d57fc872 | 178 | // default list of skiped ESD objects |
179 | TString skipObjects= | |
180 | // "AliESDRun," | |
181 | // "AliESDHeader," | |
421e1221 | 182 | // "AliESDZDC," |
d57fc872 | 183 | "AliESDFMD," |
184 | // "AliESDVZERO," | |
185 | // "AliESDTZERO," | |
186 | // "TPCVertex," | |
187 | // "SPDVertex," | |
188 | // "PrimaryVertex," | |
189 | // "AliMultiplicity," | |
190 | // "PHOSTrigger," | |
191 | // "EMCALTrigger," | |
192 | // "SPDPileupVertices," | |
193 | // "TrkPileupVertices," | |
194 | "Cascades," | |
195 | "Kinks," | |
196 | "AliRawDataErrorLogs," | |
197 | "AliESDACORDE"; | |
198 | ||
f167b631 | 199 | iResult=Reconfigure(NULL, NULL); |
200 | TString allArgs = ""; | |
201 | for ( int i = 0; i < argc; i++ ) { | |
202 | if ( !allArgs.IsNull() ) allArgs += " "; | |
203 | allArgs += argv[i]; | |
204 | } | |
a1408c4b | 205 | |
f167b631 | 206 | TObjArray* pTokens=allArgs.Tokenize(" "); |
207 | if (pTokens) { | |
208 | for (int i=0; i<pTokens->GetEntries() && iResult>=0; i++) { | |
209 | argument=((TObjString*)pTokens->At(i))->GetString(); | |
210 | if (argument.IsNull()) continue; | |
211 | ||
212 | // -notree | |
213 | if (argument.CompareTo("-notree")==0) { | |
214 | fWriteTree=0; | |
215 | ||
216 | // -tree | |
217 | } else if (argument.CompareTo("-tree")==0) { | |
218 | fWriteTree=1; | |
219 | } else if (argument.CompareTo("-solenoidBz")==0) { | |
220 | if ((bMissingParam=(++i>=pTokens->GetEntries()))) break; | |
221 | HLTInfo("Magnetic Field set to: %s", ((TObjString*)pTokens->At(i))->GetString().Data()); | |
d57fc872 | 222 | HLTWarning("argument '-solenoidBz' is deprecated, solenoid field initiaized from CDB settings"); |
f167b631 | 223 | continue; |
d57fc872 | 224 | } else if (argument.Contains("-skipobject=")) { |
225 | argument.ReplaceAll("-skipobject=", ""); | |
226 | skipObjects=argument; | |
f167b631 | 227 | } else { |
228 | HLTError("unknown argument %s", argument.Data()); | |
d57fc872 | 229 | iResult=-EINVAL; |
f167b631 | 230 | break; |
231 | } | |
a1408c4b | 232 | } |
233 | } | |
234 | if (bMissingParam) { | |
235 | HLTError("missing parameter for argument %s", argument.Data()); | |
236 | iResult=-EINVAL; | |
237 | } | |
238 | ||
75970b8d | 239 | fSolenoidBz=GetBz(); |
240 | ||
a1408c4b | 241 | if (iResult>=0) { |
242 | fESD = new AliESDEvent; | |
243 | if (fESD) { | |
244 | fESD->CreateStdContent(); | |
d57fc872 | 245 | |
246 | // remove some of the objects which are not needed | |
247 | if (fESD->GetList() && !skipObjects.IsNull()) { | |
248 | pTokens=skipObjects.Tokenize(","); | |
249 | if (pTokens) { | |
250 | const char* id=NULL; | |
251 | TIter next(pTokens); | |
252 | TObject* pObject=NULL; | |
253 | while ((pObject=next())!=NULL) { | |
254 | id=((TObjString*)pObject)->GetString().Data(); | |
a0b62fd1 | 255 | TObject* pObj=fESD->GetList()->FindObject(id); |
256 | if (pObj) { | |
d57fc872 | 257 | HLTDebug("removing object %s", id); |
a0b62fd1 | 258 | fESD->GetList()->Remove(pObj); |
259 | delete pObj; | |
d57fc872 | 260 | } else { |
261 | HLTWarning("failed to remove object '%s' from ESD", id); | |
262 | } | |
263 | } | |
d57fc872 | 264 | fESD->GetStdContent(); |
265 | delete pTokens; | |
266 | } | |
267 | } | |
a1408c4b | 268 | } else { |
269 | iResult=-ENOMEM; | |
270 | } | |
6117ef63 | 271 | |
272 | SetupCTPData(); | |
a1408c4b | 273 | } |
274 | ||
57a4102f | 275 | fBenchmark.SetTimer(0,"total"); |
276 | ||
a1408c4b | 277 | return iResult; |
278 | } | |
279 | ||
280 | int AliHLTGlobalEsdConverterComponent::DoDeinit() | |
281 | { | |
282 | // see header file for class documentation | |
283 | if (fESD) delete fESD; | |
284 | fESD=NULL; | |
285 | ||
286 | return 0; | |
287 | } | |
288 | ||
289 | int AliHLTGlobalEsdConverterComponent::DoEvent(const AliHLTComponentEventData& /*evtData*/, | |
6117ef63 | 290 | AliHLTComponentTriggerData& trigData) |
a1408c4b | 291 | { |
292 | // see header file for class documentation | |
293 | int iResult=0; | |
294 | if (!fESD) return -ENODEV; | |
295 | ||
57a4102f | 296 | if (IsDataEvent()) fBenchmark.StartNewEvent(); |
297 | fBenchmark.Start(0); | |
298 | ||
a1408c4b | 299 | AliESDEvent* pESD = fESD; |
300 | ||
301 | pESD->Reset(); | |
302 | pESD->SetMagneticField(fSolenoidBz); | |
6117ef63 | 303 | pESD->SetRunNumber(GetRunNo()); |
6700298e | 304 | pESD->SetPeriodNumber(GetPeriodNumber()); |
305 | pESD->SetOrbitNumber(GetOrbitNumber()); | |
306 | pESD->SetBunchCrossNumber(GetBunchCrossNumber()); | |
307 | pESD->SetTimeStamp(GetTimeStamp()); | |
a1408c4b | 308 | |
6117ef63 | 309 | const AliHLTCTPData* pCTPData=CTPData(); |
310 | if (pCTPData) { | |
311 | AliHLTUInt64_t mask=pCTPData->ActiveTriggers(trigData); | |
312 | for (int index=0; index<gkNCTPTriggerClasses; index++) { | |
313 | if ((mask&((AliHLTUInt64_t)0x1<<index)) == 0) continue; | |
314 | pESD->SetTriggerClass(pCTPData->Name(index), index); | |
315 | } | |
316 | pESD->SetTriggerMask(mask); | |
317 | } | |
318 | ||
a1408c4b | 319 | TTree* pTree = NULL; |
320 | if (fWriteTree) | |
321 | pTree = new TTree("esdTree", "Tree with HLT ESD objects"); | |
322 | ||
323 | if (pTree) { | |
324 | pTree->SetDirectory(0); | |
325 | } | |
326 | ||
81509fd3 | 327 | if ((iResult=ProcessBlocks(pTree, pESD))>=0) { |
a1408c4b | 328 | // TODO: set the specification correctly |
329 | if (pTree) { | |
330 | // the esd structure is written to the user info and is | |
331 | // needed in te ReadFromTree method to read all objects correctly | |
332 | pTree->GetUserInfo()->Add(pESD); | |
333 | pESD->WriteToTree(pTree); | |
7a9de6d1 | 334 | iResult=PushBack(pTree, kAliHLTDataTypeESDTree|kAliHLTDataOriginOut, 0); |
a1408c4b | 335 | } else { |
7a9de6d1 | 336 | iResult=PushBack(pESD, kAliHLTDataTypeESDObject|kAliHLTDataOriginOut, 0); |
a1408c4b | 337 | } |
57a4102f | 338 | fBenchmark.AddOutput(GetLastObjectSize()); |
a1408c4b | 339 | } |
340 | if (pTree) { | |
341 | // clear user info list to prevent objects from being deleted | |
342 | pTree->GetUserInfo()->Clear(); | |
343 | delete pTree; | |
344 | } | |
57a4102f | 345 | |
346 | fBenchmark.Stop(0); | |
347 | HLTInfo( fBenchmark.GetStatistics() ); | |
348 | ||
a1408c4b | 349 | return iResult; |
350 | } | |
351 | ||
352 | int AliHLTGlobalEsdConverterComponent::ProcessBlocks(TTree* pTree, AliESDEvent* pESD) | |
353 | { | |
354 | // see header file for class documentation | |
355 | ||
356 | int iResult=0; | |
357 | int iAddedDataBlocks=0; | |
358 | ||
359 | // Barrel tracking | |
e85f2e31 | 360 | // tracks are based on the TPC tracks, and only updated from the ITS information |
361 | // Sequence: | |
362 | // 1) extract MC information for TPC and ITS from specific data blocks and store in | |
363 | // intermediate vector arrays | |
364 | // 2) extract TPC tracks, update with MC labels if available, the track parameters | |
365 | // are estimated at the first cluster position | |
366 | // 2.1) propagate to last cluster position and update kTPCout, sets also outer param (fOp) | |
367 | // 2.2) update kTPCin, sets also inner param (fIp) and TPC inner param (fTPCInner) | |
368 | // 2.3) update kTPCrefit using the same parameters at the first cluster position | |
369 | // HLT has strictly spoking no refit, but we want the flag to be set | |
370 | // can be changed to be done after all the individual barrel detector parameters | |
371 | // have been updated by looping over the tracks again | |
372 | // 3) extract ITS tracks, the tracks are actually TPC tracks updated from the ITS | |
373 | // tracking information | |
374 | // 3.1) TODO 2010-07-12: handle ITS standalone tracks by updating kITSout before kITSin | |
375 | // 3.2) update with kITSin | |
376 | // TODO 2010-07-12 find out if the kITSrefit has to be set as well | |
377 | // 4) extract TRD tracks and add to ESD | |
378 | // TODO 2010-07-12 at the moment there is no matching or merging of TPC and TRD tracks | |
331d51c5 | 379 | // 5) Add Trigger Detectors |
380 | // VZERO, ZDC | |
e85f2e31 | 381 | |
382 | // 1) first read MC information (if present) | |
eb60b0c1 | 383 | std::map<int,int> mcLabelsTPC; |
d72d9d99 | 384 | std::map<int,int> mcLabelsITS; |
093b64dc | 385 | |
386 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeTrackMC|kAliHLTDataOriginTPC); | |
387 | pBlock!=NULL; pBlock=GetNextInputBlock()) { | |
57a4102f | 388 | |
389 | fBenchmark.AddInput(pBlock->fSize); | |
390 | ||
093b64dc | 391 | AliHLTTrackMCData* dataPtr = reinterpret_cast<AliHLTTrackMCData*>( pBlock->fPtr ); |
392 | if (sizeof(AliHLTTrackMCData)+dataPtr->fCount*sizeof(AliHLTTrackMCLabel)==pBlock->fSize) { | |
393 | for( unsigned int il=0; il<dataPtr->fCount; il++ ){ | |
394 | AliHLTTrackMCLabel &lab = dataPtr->fLabels[il]; | |
eb60b0c1 | 395 | mcLabelsTPC[lab.fTrackID] = lab.fMCLabel; |
093b64dc | 396 | } |
397 | } else { | |
398 | HLTWarning("data mismatch in block %s (0x%08x): count %d, size %d -> ignoring track MC information", | |
399 | DataType2Text(pBlock->fDataType).c_str(), pBlock->fSpecification, | |
400 | dataPtr->fCount, pBlock->fSize); | |
401 | } | |
402 | } | |
d72d9d99 | 403 | |
404 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeTrackMC|kAliHLTDataOriginITS); | |
405 | pBlock!=NULL; pBlock=GetNextInputBlock()) { | |
406 | ||
407 | fBenchmark.AddInput(pBlock->fSize); | |
408 | ||
409 | AliHLTTrackMCData* dataPtr = reinterpret_cast<AliHLTTrackMCData*>( pBlock->fPtr ); | |
410 | if (sizeof(AliHLTTrackMCData)+dataPtr->fCount*sizeof(AliHLTTrackMCLabel)==pBlock->fSize) { | |
411 | for( unsigned int il=0; il<dataPtr->fCount; il++ ){ | |
412 | AliHLTTrackMCLabel &lab = dataPtr->fLabels[il]; | |
413 | mcLabelsITS[lab.fTrackID] = lab.fMCLabel; | |
414 | } | |
415 | } else { | |
416 | HLTWarning("data mismatch in block %s (0x%08x): count %d, size %d -> ignoring track MC information", | |
417 | DataType2Text(pBlock->fDataType).c_str(), pBlock->fSpecification, | |
418 | dataPtr->fCount, pBlock->fSize); | |
419 | } | |
420 | } | |
421 | ||
093b64dc | 422 | |
0973c527 | 423 | // read dEdx information (if present) |
e85f2e31 | 424 | // TODO 2010-07-12 this needs to be verified |
0973c527 | 425 | |
426 | AliHLTFloat32_t *dEdxTPC = 0; | |
427 | Int_t ndEdxTPC = 0; | |
428 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypedEdx|kAliHLTDataOriginTPC); | |
429 | pBlock!=NULL; pBlock=GetNextInputBlock()) { | |
57a4102f | 430 | fBenchmark.AddInput(pBlock->fSize); |
0973c527 | 431 | dEdxTPC = reinterpret_cast<AliHLTFloat32_t*>( pBlock->fPtr ); |
72a916d9 | 432 | ndEdxTPC = pBlock->fSize / (3*sizeof(AliHLTFloat32_t)); |
0973c527 | 433 | break; |
434 | } | |
435 | ||
e85f2e31 | 436 | // 2) convert the TPC tracks to ESD tracks |
a1408c4b | 437 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeTrack|kAliHLTDataOriginTPC); |
438 | pBlock!=NULL; pBlock=GetNextInputBlock()) { | |
57a4102f | 439 | fBenchmark.AddInput(pBlock->fSize); |
a1408c4b | 440 | vector<AliHLTGlobalBarrelTrack> tracks; |
81509fd3 | 441 | if ((iResult=AliHLTGlobalBarrelTrack::ConvertTrackDataArray(reinterpret_cast<const AliHLTTracksData*>(pBlock->fPtr), pBlock->fSize, tracks))>=0) { |
a1408c4b | 442 | for (vector<AliHLTGlobalBarrelTrack>::iterator element=tracks.begin(); |
443 | element!=tracks.end(); element++) { | |
444 | Float_t points[4] = { | |
445 | element->GetX(), | |
446 | element->GetY(), | |
447 | element->GetLastPointX(), | |
448 | element->GetLastPointY() | |
449 | }; | |
093b64dc | 450 | |
451 | Int_t mcLabel = -1; | |
eb60b0c1 | 452 | if( mcLabelsTPC.find(element->TrackID())!=mcLabelsTPC.end() ) |
453 | mcLabel = mcLabelsTPC[element->TrackID()]; | |
093b64dc | 454 | element->SetLabel( mcLabel ); |
455 | ||
a1408c4b | 456 | AliESDtrack iotrack; |
550ecaab | 457 | |
458 | // for the moment, the number of clusters are not set when processing the | |
459 | // kTPCin update, only at kTPCout | |
460 | // there ar emainly three parameters updated for kTPCout | |
461 | // number of clusters | |
462 | // chi2 | |
463 | // pid signal | |
464 | // The first one can be updated already at that stage here, while the two others | |
465 | // eventually require to update from the ITS tracks before. The exact scheme | |
ab700b08 | 466 | // needs to be checked |
c3261a7d | 467 | iotrack.SetID( element->TrackID() ); |
e85f2e31 | 468 | |
469 | // 2.1 set kTPCout | |
470 | // TODO 2010-07-12 disabled for the moment because of bug | |
471 | // https://savannah.cern.ch/bugs/index.php?69875 | |
472 | // the propagation does not work, there is an offset in z | |
473 | // propagate to last cluster and update parameters with flag kTPCout | |
474 | // Note: updating with flag kITSout further down will overwrite the outer | |
475 | // parameter again so this can be done only for ITS standalone tracks, meaning | |
476 | // tracks in the ITS not associated with any TPC track | |
477 | // HLT does not provide such standalone tracking | |
c3261a7d | 478 | { |
f167b631 | 479 | AliHLTGlobalBarrelTrack outPar(*element); |
114e074b | 480 | //outPar.AliExternalTrackParam::PropagateTo( element->GetLastPointX(), fSolenoidBz ); |
481 | const Int_t N=10; // number of steps. | |
482 | const Float_t xRange = element->GetLastPointX() - element->GetX(); | |
483 | const Float_t xStep = xRange / N ; | |
484 | for(int i = 1; i <= N; ++i) { | |
485 | if(!outPar.AliExternalTrackParam::PropagateTo(element->GetX() + xStep * i, fSolenoidBz)) break; | |
486 | } | |
b9838c05 | 487 | outPar.SetLabel(element->GetLabel()); |
114e074b | 488 | iotrack.UpdateTrackParams(&outPar,AliESDtrack::kTPCout); |
c3261a7d | 489 | } |
e85f2e31 | 490 | // 2.2 TPC tracking estimates parameters at first cluster |
ab700b08 | 491 | iotrack.UpdateTrackParams(&(*element),AliESDtrack::kTPCin); |
e85f2e31 | 492 | |
493 | // 2.3 use the same parameters also for kTPCrefit, there is no proper refit in HLT | |
494 | // maybe this can be done later, however also the inner parameter is changed which | |
495 | // is used as a reference point in the display. That point should be at the first | |
496 | // TPC cluster | |
497 | iotrack.UpdateTrackParams(&(*element),AliESDtrack::kTPCrefit); | |
a1408c4b | 498 | iotrack.SetTPCPoints(points); |
0973c527 | 499 | if( element->TrackID()<ndEdxTPC ){ |
72a916d9 | 500 | AliHLTFloat32_t *val = &(dEdxTPC[3*element->TrackID()]); |
5c2dca91 | 501 | iotrack.SetTPCsignal( val[0], val[1], (UChar_t) val[2] ); |
7e32fedf | 502 | //AliTPCseed s; |
503 | //s.Set( element->GetX(), element->GetAlpha(), | |
504 | //element->GetParameter(), element->GetCovariance() ); | |
72a916d9 | 505 | //s.SetdEdx( val[0] ); |
7e32fedf | 506 | //s.CookPID(); |
507 | //iotrack.SetTPCpid(s.TPCrPIDs() ); | |
0973c527 | 508 | } else { |
509 | if( dEdxTPC ) HLTWarning("Wrong number of dEdx TPC labels"); | |
510 | } | |
b9838c05 | 511 | iotrack.SetLabel(mcLabel); |
7fd2e09c | 512 | pESD->AddTrack(&iotrack); |
093b64dc | 513 | if (fVerbosity>0) element->Print(); |
a1408c4b | 514 | } |
093b64dc | 515 | HLTInfo("converted %d track(s) to AliESDtrack and added to ESD", tracks.size()); |
a1408c4b | 516 | iAddedDataBlocks++; |
517 | } else if (iResult<0) { | |
093b64dc | 518 | HLTError("can not extract tracks from data block of type %s (specification %08x) of size %d: error %d", |
519 | DataType2Text(pBlock->fDataType).c_str(), pBlock->fSpecification, pBlock->fSize, iResult); | |
a1408c4b | 520 | } |
521 | } | |
522 | ||
7f167a74 | 523 | |
524 | // Get ITS SPD vertex | |
57a4102f | 525 | for( const AliHLTComponentBlockData *i= GetFirstInputBlock(kAliHLTDataTypeESDVertex|kAliHLTDataOriginITS); i!=NULL; i=GetNextInputBlock() ){ |
526 | fBenchmark.AddInput(i->fSize); | |
527 | } | |
7f167a74 | 528 | |
529 | for ( const TObject *iter = GetFirstInputObject(kAliHLTDataTypeESDVertex|kAliHLTDataOriginITS); iter != NULL; iter = GetNextInputObject() ) { | |
530 | AliESDVertex *vtx = dynamic_cast<AliESDVertex*>(const_cast<TObject*>( iter ) ); | |
531 | pESD->SetPrimaryVertexSPD( vtx ); | |
532 | } | |
533 | ||
e85f2e31 | 534 | // 3.1. now update ESD tracks with the ITSOut info |
535 | // updating track parameters with flag kITSout will overwrite parameter set above with flag kTPCout | |
536 | // TODO 2010-07-12 there are some issues with this updating sequence, for the moment update with | |
537 | // flag kITSout is disabled, would require | |
538 | // a) to update kTPCout after kITSout | |
539 | // b) update only for ITS standalone tracks. The HLT ITS tracker does not provide standalone | |
540 | // tracking, every track is related to a TPC track | |
541 | // Furthermore there seems to be a bug as the data blocks describe track parameters around the | |
542 | // vertex, not at the outer ITS | |
543 | // bug https://savannah.cern.ch/bugs/index.php?69872 | |
f167b631 | 544 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeTrack|kAliHLTDataOriginITSOut); |
cd60a73f | 545 | pBlock!=NULL; pBlock=GetNextInputBlock()) { |
57a4102f | 546 | fBenchmark.AddInput(pBlock->fSize); |
cd60a73f | 547 | vector<AliHLTGlobalBarrelTrack> tracks; |
548 | if ((iResult=AliHLTGlobalBarrelTrack::ConvertTrackDataArray(reinterpret_cast<const AliHLTTracksData*>(pBlock->fPtr), pBlock->fSize, tracks))>0) { | |
549 | for (vector<AliHLTGlobalBarrelTrack>::iterator element=tracks.begin(); | |
550 | element!=tracks.end(); element++) { | |
cd60a73f | 551 | int tpcID=element->TrackID(); |
552 | // the ITS tracker assigns the TPC track used as seed for a certain track to | |
553 | // the trackID | |
554 | if( tpcID<0 || tpcID>=pESD->GetNumberOfTracks()) continue; | |
cd60a73f | 555 | AliESDtrack *tESD = pESD->GetTrack( tpcID ); |
b9838c05 | 556 | element->SetLabel(tESD->GetLabel()); |
e85f2e31 | 557 | // 2010-07-12 disabled, see above, bugfix https://savannah.cern.ch/bugs/index.php?69872 |
558 | //if( tESD ) tESD->UpdateTrackParams( &(*element), AliESDtrack::kITSout ); | |
a1408c4b | 559 | } |
cd60a73f | 560 | } |
561 | } | |
18ada816 | 562 | |
e85f2e31 | 563 | // 3.2. now update ESD tracks with the ITS info |
f167b631 | 564 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeTrack|kAliHLTDataOriginITS); |
b7ed2eb4 | 565 | pBlock!=NULL; pBlock=GetNextInputBlock()) { |
57a4102f | 566 | fBenchmark.AddInput(pBlock->fSize); |
b7ed2eb4 | 567 | vector<AliHLTGlobalBarrelTrack> tracks; |
568 | if ((iResult=AliHLTGlobalBarrelTrack::ConvertTrackDataArray(reinterpret_cast<const AliHLTTracksData*>(pBlock->fPtr), pBlock->fSize, tracks))>0) { | |
569 | for (vector<AliHLTGlobalBarrelTrack>::iterator element=tracks.begin(); | |
570 | element!=tracks.end(); element++) { | |
b7ed2eb4 | 571 | int tpcID=element->TrackID(); |
572 | // the ITS tracker assigns the TPC track used as seed for a certain track to | |
573 | // the trackID | |
574 | if( tpcID<0 || tpcID>=pESD->GetNumberOfTracks()) continue; | |
d72d9d99 | 575 | Int_t mcLabel = -1; |
576 | if( mcLabelsITS.find(element->TrackID())!=mcLabelsITS.end() ) | |
577 | mcLabel = mcLabelsITS[element->TrackID()]; | |
b7ed2eb4 | 578 | AliESDtrack *tESD = pESD->GetTrack( tpcID ); |
b9838c05 | 579 | if (!tESD) continue; |
eb60b0c1 | 580 | // the labels for the TPC and ITS tracking params can be different, e.g. |
581 | // there can be a decay. The ITS label should then be the better one, the | |
582 | // TPC label is saved in a member of AliESDtrack | |
583 | if (mcLabel>=0) { | |
584 | // upadte only if the ITS label is available, otherwise keep TPC label | |
585 | element->SetLabel( mcLabel ); | |
e85f2e31 | 586 | } else { |
587 | // bugfix https://savannah.cern.ch/bugs/?69713 | |
588 | element->SetLabel( tESD->GetLabel() ); | |
b9838c05 | 589 | } |
b9838c05 | 590 | tESD->UpdateTrackParams( &(*element), AliESDtrack::kITSin ); |
591 | ||
592 | // TODO: add a proper refit | |
593 | //tESD->UpdateTrackParams( &(*element), AliESDtrack::kTPCrefit ); | |
b7ed2eb4 | 594 | } |
595 | } | |
596 | } | |
597 | ||
d9386025 | 598 | // update with vertices and vertex-fitted tracks |
599 | ||
600 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeGlobalVertexer); | |
601 | pBlock!=NULL; pBlock=GetNextInputBlock()) { | |
57a4102f | 602 | fBenchmark.AddInput(pBlock->fSize); |
d9386025 | 603 | AliHLTGlobalVertexerComponent::FillESD( pESD, reinterpret_cast<AliHLTGlobalVertexerComponent::AliHLTGlobalVertexerData* >(pBlock->fPtr) ); |
604 | } | |
605 | ||
0d2e4e41 | 606 | // Fill DCA parameters for TPC tracks |
e85f2e31 | 607 | // TODO 2010-07-12 this propagates also the TPC inner param to beamline |
608 | // sounds not very reasonable | |
609 | // https://savannah.cern.ch/bugs/index.php?69873 | |
7fd2e09c | 610 | for (int i=0; i<pESD->GetNumberOfTracks(); i++) { |
0d2e4e41 | 611 | if (!pESD->GetTrack(i) || |
612 | !pESD->GetTrack(i)->GetTPCInnerParam() ) continue; | |
7fd2e09c | 613 | pESD->GetTrack(i)->RelateToVertexTPC(pESD->GetPrimaryVertexTracks(), fSolenoidBz, 1000 ); |
0d2e4e41 | 614 | } |
615 | ||
b9838c05 | 616 | // loop over all tracks and set the TPC refit flag by updating with the |
617 | // original TPC inner parameter if not yet set | |
618 | // TODO: replace this by a proper refit | |
619 | // code is comented for the moment as it does not fully solve the problems with | |
620 | // the display | |
e85f2e31 | 621 | // - would set the main parameters to the TPC inner wall again, or |
622 | // - changes the inner param if the parameters are propagated, so we loose the track | |
623 | // reference point for the display | |
624 | // with the current sequence we have the latter case as the DCA operations above | |
625 | // change the TPC inner parameters | |
b9838c05 | 626 | /* |
627 | for (int i=0; i<pESD->GetNumberOfTracks(); i++) { | |
628 | if (!pESD->GetTrack(i) || | |
629 | !pESD->GetTrack(i)->GetTPCInnerParam() || | |
630 | pESD->GetTrack(i)->IsOn(AliESDtrack::kTPCrefit)) continue; | |
631 | AliESDtrack* tESD=pESD->GetTrack(i); | |
632 | AliHLTGlobalBarrelTrack inner(*tESD->GetTPCInnerParam()); | |
633 | inner.SetLabel(tESD->GetLabel()); | |
634 | tESD->UpdateTrackParams(&inner, AliESDtrack::kTPCrefit); | |
635 | } | |
636 | */ | |
b7ed2eb4 | 637 | |
e85f2e31 | 638 | // 4. convert the HLT TRD tracks to ESD tracks |
18ada816 | 639 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeTrack | kAliHLTDataOriginTRD); |
640 | pBlock!=NULL; pBlock=GetNextInputBlock()) { | |
57a4102f | 641 | fBenchmark.AddInput(pBlock->fSize); |
18ada816 | 642 | vector<AliHLTGlobalBarrelTrack> tracks; |
643 | if ((iResult=AliHLTGlobalBarrelTrack::ConvertTrackDataArray(reinterpret_cast<const AliHLTTracksData*>(pBlock->fPtr), pBlock->fSize, tracks))>0) { | |
644 | for (vector<AliHLTGlobalBarrelTrack>::iterator element=tracks.begin(); | |
645 | element!=tracks.end(); element++) { | |
646 | ||
647 | Double_t TRDpid[AliPID::kSPECIES], eProb(0.2), restProb((1-eProb)/(AliPID::kSPECIES-1)); //eprob(element->GetTRDpid...); | |
648 | for(Int_t i=0; i<AliPID::kSPECIES; i++){ | |
649 | switch(i){ | |
650 | case AliPID::kElectron: TRDpid[AliPID::kElectron]=eProb; break; | |
651 | default: TRDpid[i]=restProb; break; | |
652 | } | |
653 | } | |
654 | ||
655 | AliESDtrack iotrack; | |
0d096bba | 656 | iotrack.UpdateTrackParams(&(*element),AliESDtrack::kTRDout); |
0b745718 | 657 | iotrack.SetStatus(AliESDtrack::kTRDin); |
18ada816 | 658 | iotrack.SetTRDpid(TRDpid); |
659 | ||
660 | pESD->AddTrack(&iotrack); | |
661 | if (fVerbosity>0) element->Print(); | |
662 | } | |
663 | HLTInfo("converted %d track(s) to AliESDtrack and added to ESD", tracks.size()); | |
664 | iAddedDataBlocks++; | |
665 | } else if (iResult<0) { | |
666 | HLTError("can not extract tracks from data block of type %s (specification %08x) of size %d: error %d", | |
667 | DataType2Text(pBlock->fDataType).c_str(), pBlock->fSpecification, pBlock->fSize, iResult); | |
668 | } | |
669 | } | |
b4479a87 | 670 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeCaloCluster | kAliHLTDataOriginAny); pBlock!=NULL; pBlock=GetNextInputBlock()) |
9a9467ed | 671 | { |
57a4102f | 672 | fBenchmark.AddInput(pBlock->fSize); |
9a9467ed | 673 | AliHLTCaloClusterHeaderStruct *caloClusterHeaderPtr = reinterpret_cast<AliHLTCaloClusterHeaderStruct*>(pBlock->fPtr); |
674 | ||
675 | HLTDebug("%d HLT clusters from spec: 0x%X", caloClusterHeaderPtr->fNClusters, pBlock->fSpecification); | |
676 | ||
2a24cbbe | 677 | //AliHLTCaloClusterReader reader; |
678 | //reader.SetMemory(caloClusterHeaderPtr); | |
9a9467ed | 679 | |
680 | AliHLTESDCaloClusterMaker clusterMaker; | |
681 | ||
682 | int nClusters = clusterMaker.FillESD(pESD, caloClusterHeaderPtr); | |
2a24cbbe | 683 | |
9a9467ed | 684 | HLTInfo("converted %d cluster(s) to AliESDCaloCluster and added to ESD", nClusters); |
685 | iAddedDataBlocks++; | |
686 | } | |
f064ef44 | 687 | |
331d51c5 | 688 | // 5) Add Trigger Detectors |
689 | // VZERO, ZDC | |
690 | ||
691 | // FIXME: the size of all input blocks can be added in one loop | |
692 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeESDContent|kAliHLTDataOriginVZERO); | |
693 | pBlock!=NULL; pBlock=GetNextInputBlock()) { | |
694 | fBenchmark.AddInput(pBlock->fSize); | |
695 | } | |
696 | ||
697 | for ( const TObject *pObject = GetFirstInputObject(kAliHLTDataTypeESDContent|kAliHLTDataOriginVZERO); | |
698 | pObject != NULL; pObject = GetNextInputObject() ) { | |
699 | AliESDVZERO *esdVZERO = dynamic_cast<AliESDVZERO*>(const_cast<TObject*>( pObject ) ); | |
700 | if (esdVZERO) { | |
701 | pESD->SetVZEROData( esdVZERO ); | |
702 | break; | |
703 | } else { | |
704 | ALIHLTERRORGUARD(1, "input object of data type %s is not of class AliESDVZERO", | |
705 | DataType2Text(kAliHLTDataTypeESDContent|kAliHLTDataOriginVZERO).c_str()); | |
706 | } | |
707 | } | |
708 | ||
421e1221 | 709 | // FIXME: the size of all input blocks can be added in one loop |
710 | for (const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeESDContent|kAliHLTDataOriginZDC); | |
711 | pBlock!=NULL; pBlock=GetNextInputBlock()) { | |
712 | fBenchmark.AddInput(pBlock->fSize); | |
713 | } | |
714 | for ( const TObject *pObject = GetFirstInputObject(kAliHLTDataTypeESDContent|kAliHLTDataOriginZDC); | |
715 | pObject != NULL; pObject = GetNextInputObject() ) { | |
716 | AliESDZDC *esdZDC = dynamic_cast<AliESDZDC*>(const_cast<TObject*>( pObject ) ); | |
717 | if (esdZDC) { | |
8cd67fda | 718 | #ifndef HAVE_NOT_ALIZDCRECONSTRUCTOR_r43770 |
421e1221 | 719 | pESD->SetZDCData( esdZDC ); |
8cd67fda | 720 | #else |
721 | ALIHLTERRORGUARD(1, "Processing of ZDC data requires AliRoot r43770m skipping data block of type %s", | |
722 | DataType2Text(kAliHLTDataTypeESDContent|kAliHLTDataOriginZDC).c_str()); | |
723 | #endif | |
421e1221 | 724 | break; |
725 | } else { | |
726 | ALIHLTERRORGUARD(1, "input object of data type %s is not of class AliESDZDC", | |
727 | DataType2Text(kAliHLTDataTypeESDContent|kAliHLTDataOriginZDC).c_str()); | |
728 | } | |
729 | } | |
730 | ||
f064ef44 | 731 | // Add tracks from MUON. |
57a4102f | 732 | for( const AliHLTComponentBlockData *i= GetFirstInputBlock(kAliHLTAnyDataType | kAliHLTDataOriginMUON); i!=NULL; i=GetNextInputBlock() ){ |
733 | fBenchmark.AddInput(i->fSize); | |
734 | } | |
735 | ||
f064ef44 | 736 | for (const TObject* obj = GetFirstInputObject(kAliHLTAnyDataType | kAliHLTDataOriginMUON); |
737 | obj != NULL; | |
738 | obj = GetNextInputObject() | |
739 | ) | |
740 | { | |
741 | const TClonesArray* tracklist = NULL; | |
742 | if (obj->IsA() == AliESDEvent::Class()) | |
743 | { | |
744 | const AliESDEvent* event = static_cast<const AliESDEvent*>(obj); | |
745 | HLTDebug("Received a MUON ESD with specification: 0x%X", GetSpecification(obj)); | |
746 | if (event->GetList() == NULL) continue; | |
747 | tracklist = dynamic_cast<const TClonesArray*>(event->GetList()->FindObject("MuonTracks")); | |
748 | if (tracklist == NULL) continue; | |
749 | } | |
750 | else if (obj->IsA() == TClonesArray::Class()) | |
751 | { | |
752 | tracklist = static_cast<const TClonesArray*>(obj); | |
753 | HLTDebug("Received a MUON TClonesArray of tracks with specification: 0x%X", GetSpecification(obj)); | |
754 | } | |
755 | else | |
756 | { | |
757 | // Cannot handle this object type. | |
758 | continue; | |
759 | } | |
760 | HLTDebug("Received %d MUON tracks.", tracklist->GetEntriesFast()); | |
761 | if (tracklist->GetEntriesFast() > 0) | |
762 | { | |
763 | const AliESDMuonTrack* track = dynamic_cast<const AliESDMuonTrack*>(tracklist->UncheckedAt(0)); | |
764 | if (track == NULL) | |
765 | { | |
766 | HLTError(Form("%s from MUON does not contain AliESDMuonTrack objects.", obj->ClassName())); | |
767 | continue; | |
a1408c4b | 768 | } |
f064ef44 | 769 | } |
770 | for (Int_t i = 0; i < tracklist->GetEntriesFast(); ++i) | |
771 | { | |
772 | const AliESDMuonTrack* track = static_cast<const AliESDMuonTrack*>(tracklist->UncheckedAt(i)); | |
773 | pESD->AddMuonTrack(track); | |
774 | } | |
775 | } | |
776 | ||
777 | if (iAddedDataBlocks>0 && pTree) { | |
778 | pTree->Fill(); | |
779 | } | |
a1408c4b | 780 | |
781 | if (iResult>=0) iResult=iAddedDataBlocks; | |
782 | return iResult; | |
783 | } |