Fixed serious bug in QA code
[u/mrichter/AliRoot.git] / HLT / rec / AliHLTEsdManagerImplementation.cxx
CommitLineData
c5123824 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
f1207f29 19/** @file AliHLTEsdManagerImplementation.cxx
c5123824 20 @author Matthias Richter
21 @date
22 @brief Manager for merging and writing of HLT ESDs
23*/
24
f1207f29 25#include "AliHLTEsdManagerImplementation.h"
c5123824 26#include "AliHLTComponent.h"
27#include "AliESDEvent.h"
28#include "AliHLTMessage.h"
29#include "AliESDEvent.h"
f527516f 30#include "AliESDtrack.h"
c5123824 31#include "TFile.h"
32#include "TTree.h"
33#include "TClass.h"
34#include "TObject.h"
62ff1e23 35#include "TObjectTable.h"
90c37647 36#include "TSystem.h"
37#include "TChain.h"
38#include "TList.h"
c5123824 39
40/** ROOT macro for the implementation of ROOT specific class methods */
f1207f29 41ClassImp(AliHLTEsdManagerImplementation)
c5123824 42
f1207f29 43AliHLTEsdManagerImplementation::AliHLTEsdManagerImplementation()
c5123824 44 :
90c37647 45 fESDs(),
46 fDirectory()
c5123824 47{
48 // see header file for class documentation
49 // or
50 // refer to README to build package
51 // or
52 // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
53}
54
f1207f29 55AliHLTEsdManagerImplementation::~AliHLTEsdManagerImplementation()
c5123824 56{
57 // see header file for class documentation
62ff1e23 58 for (unsigned int i=0; i<fESDs.size(); i++) {
59 if (fESDs[i]) {
60 delete fESDs[i];
61 }
62 fESDs[i]=NULL;
63 }
c5123824 64}
65
f1207f29 66AliHLTEsdManagerImplementation::AliHLTEsdListEntry* AliHLTEsdManagerImplementation::Find(AliHLTComponentDataType dt) const
c5123824 67{
68 // see header file for class documentation
69 AliHLTEsdListEntry* pEntry=NULL;
70 for (unsigned int i=0; i<fESDs.size(); i++) {
62ff1e23 71 if (fESDs[i] && *(fESDs[i])==dt) {
72 pEntry=const_cast<AliHLTEsdListEntry*>(fESDs[i]);
c5123824 73 }
74 }
75 return pEntry;
76}
77
f1207f29 78int AliHLTEsdManagerImplementation::WriteESD(const AliHLTUInt8_t* pBuffer, AliHLTUInt32_t size,
c5123824 79 AliHLTComponentDataType dt, AliESDEvent* tgtesd, int eventno)
80{
81 // see header file for class documentation
82 if (!pBuffer && size<=0) return -EINVAL;
83 int iResult=0;
84 AliHLTUInt32_t firstWord=*((AliHLTUInt32_t*)pBuffer);
85 if (firstWord==size-sizeof(AliHLTUInt32_t)) {
86 HLTDebug("create object from block %s size %d", AliHLTComponent::DataType2Text(dt).c_str(), size);
87 AliHLTMessage msg(const_cast<AliHLTUInt8_t*>(pBuffer), size);
88 TClass* objclass=msg.GetClass();
89 TObject* pObj=msg.ReadObject(objclass);
90 if (pObj && objclass) {
91 HLTDebug("object %p type %s created", pObj, objclass->GetName());
92 AliESDEvent* pESD=dynamic_cast<AliESDEvent*>(pObj);
93 TTree* pTree=NULL;
94 if (!pESD) {
95 pTree=dynamic_cast<TTree*>(pObj);
96 if (pTree) {
97 pESD=new AliESDEvent;
98 pESD->CreateStdContent();
99 if (pTree->GetEntries()>0) {
100 if (pTree->GetEntries()>1) {
101 HLTWarning("only one entry allowed for ESD embedded into tree, data block %s contains tree with %d entires, taking first entry",
102 AliHLTComponent::DataType2Text(dt).c_str(), pTree->GetEntries());
103 }
104 pESD->ReadFromTree(pTree);
105 pTree->GetEvent(0);
106 }
107 } else {
108 HLTWarning("tree of data block %s has no events, skipping data block", AliHLTComponent::DataType2Text(dt).c_str());
109 }
110 }
111 if (pESD) {
112 AliHLTEsdListEntry* entry=Find(dt);
113 if (!entry) {
57584811 114 if ((entry=new AliHLTEsdListEntry(dt))!=NULL) {
115 if (!fDirectory.IsNull()) {
116 entry->SetDirectory(fDirectory);
117 }
118 fESDs.push_back(entry);
90c37647 119 }
c5123824 120 }
57584811 121 if (entry) {
122 if (tgtesd) {
123#if !defined(HAVE_NOT_ESD_COPY)
124 Merge(tgtesd, pESD);
9877121c 125#else //HAVE_NOT_ESD_COPY
57584811 126 static bool warningPrinted=false;
127 if (!warningPrinted) {
128 HLTWarning("old version of AliESDEvent does not provide assignment operator, skip merging to global hltEsd");
129 }
130 warningPrinted=true;
9877121c 131#endif //HAVE_NOT_ESD_COPY
57584811 132 }
c5123824 133 entry->WriteESD(pESD, eventno);
134 } else {
135 HLTError("internal mismatch, can not create list entry");
136 iResult=-ENOMEM;
137 }
138 } else {
139 HLTWarning("data block %s is not of class type AliESDEvent, ignoring ...", AliHLTComponent::DataType2Text(dt).c_str());
140 }
141 if (pTree) {
142 // ESD has been created and must be cleaned up
242e6536 143 pESD->Reset();
c5123824 144 delete pESD;
145 pESD=NULL;
146 }
147 delete pObj;
148 pObj=NULL;
149 } else {
150 }
151 }
152 return iResult;
153}
154
f1207f29 155int AliHLTEsdManagerImplementation::PadESDs(int eventno)
c5123824 156{
157 // see header file for class documentation
90c37647 158 int iResult=0;
159 for (unsigned int i=0; i<fESDs.size(); i++) {
160 if (fESDs[i]) {
161 int res=fESDs[i]->WriteESD(NULL, eventno);
162 if (res<0 && iResult>=0) iResult=res;
163 }
164 }
165 return iResult;
c5123824 166}
167
f1207f29 168void AliHLTEsdManagerImplementation::SetDirectory(const char* directory)
c5123824 169{
170 // see header file for class documentation
90c37647 171 if (!directory) return;
172 fDirectory=directory;
173 for (unsigned int i=0; i<fESDs.size(); i++) {
174 if (fESDs[i]) {
175 fESDs[i]->SetDirectory(directory);
176 }
177 }
178}
179
f1207f29 180TString AliHLTEsdManagerImplementation::GetFileNames(AliHLTComponentDataType dt) const
90c37647 181{
182 TString result;
183 for (unsigned int i=0; i<fESDs.size(); i++) {
184 if (fESDs[i] && *(fESDs[i])==dt) {
185 if (!result.IsNull()) result+=" ";
186 result+=fESDs[i]->GetFileName();
187 }
c5123824 188 }
90c37647 189 return result;
190}
c5123824 191
f1207f29 192TTree* AliHLTEsdManagerImplementation::EmbedIntoTree(AliESDEvent* pESD, const char* name, const char* title)
90c37647 193{
194 // see header file for class documentation
195 int iResult=0;
196 TTree* pTree=new TTree(name, title);
197 if (pTree) {
198 pESD->WriteToTree(pTree);
199 pTree->Fill();
200 pTree->GetUserInfo()->Add(pESD);
201 } else {
202 iResult=-ENOMEM;
c5123824 203 }
204
90c37647 205 if (iResult<0) {
206 pTree->GetUserInfo()->Clear();
207 delete pTree;
c5123824 208 }
90c37647 209
210 return pTree;
211}
212
f1207f29 213AliHLTEsdManagerImplementation::AliHLTEsdListEntry::AliHLTEsdListEntry(AliHLTComponentDataType dt)
90c37647 214 :
215 fName(),
216 fDirectory(),
83cb7e1d 217 fDt(dt),
218 fpFile(NULL),
219 fpTree(NULL),
f527516f 220 fpEsd(NULL),
221 fPrefix()
90c37647 222{
223 // see header file for class documentation
224}
225
f1207f29 226AliHLTEsdManagerImplementation::AliHLTEsdListEntry::~AliHLTEsdListEntry()
90c37647 227{
228 // see header file for class documentation
242e6536 229 if (fpEsd) delete fpEsd;
230 fpEsd=NULL;
231
232 if (fpTree) delete fpTree;
233 fpTree=NULL;
234
ba93bdb0 235 if (fpFile) {
236 fpFile->Close();
237 delete fpFile;
238 }
242e6536 239 fpFile=NULL;
c5123824 240}
241
f1207f29 242bool AliHLTEsdManagerImplementation::AliHLTEsdListEntry::operator==(AliHLTComponentDataType dt) const
c5123824 243{
244 // see header file for class documentation
245 return fDt==dt;
246}
247
f1207f29 248int AliHLTEsdManagerImplementation::AliHLTEsdListEntry::WriteESD(AliESDEvent* pSrcESD, int eventno)
c5123824 249{
83cb7e1d 250 // see header file for class documentation
251 int iResult=0;
d0d53f68 252
253 // Matthias 2009-06-06: writing of individual ESD files for the different origins was a
254 // first attempt when functionality was missing in the AliRoot framework and remained as
255 // debugging feature. ESD merging is now implemented and data written to the hltEsd, so
256 // the feature is now disabled because it causes increasing memory consumption. Presumably
257 // not because of a memory leak but the way the internal TTree is used and kept in memory.
258 return 0;
259
9877121c 260#ifndef HAVE_NOT_ESD_COPY
83cb7e1d 261 if (fName.IsNull()) {
262 // this is the first event, create the file name
83cb7e1d 263 fName="";
264 if (!fDirectory.IsNull()) {
265 fName+=fDirectory; fName+="/";
266 }
f527516f 267 fName+="Ali"; fName+=GetPrefix();
83cb7e1d 268 if (fDt!=kAliHLTDataTypeESDObject &&
269 fDt!=kAliHLTDataTypeESDTree) {
270
271 HLTWarning("non-standard ESD type %s", AliHLTComponent::DataType2Text(fDt).c_str());
272 TString id;
273 id.Insert(0, fDt.fID, kAliHLTComponentDataTypefIDsize);
274 id.Remove(TString::kTrailing, ' ');
275 id.ToUpper();
276 fName+="_"; fName+=id; fName+=".root";
277 } else {
278 fName+="ESDs.root";
279 }
280
281 fpFile=new TFile(fName, "RECREATE");
282 fpTree=new TTree("esdTree", "Tree with HLT ESD objects");
b952ebda 283 fpTree->SetDirectory(0);
83cb7e1d 284 fpEsd=new AliESDEvent;
285 if (fpEsd) {
286 fpEsd->CreateStdContent();
b952ebda 287 *fpEsd=*pSrcESD;
83cb7e1d 288 if (fpTree) {
289 fpEsd->WriteToTree(fpTree);
290 }
291 }
292 }
293
294 if (fpFile && fpTree && fpEsd) {
295 // synchronize and add empty events
296 fpEsd->Reset();
297 int nofCurrentEvents=fpTree->GetEntries();
298 if (nofCurrentEvents<eventno) {
299 iResult=1; // indicate tree to be written
300 HLTDebug("adding %d empty events to file %s", eventno-nofCurrentEvents, fName.Data());
301 for (int i=nofCurrentEvents; i<eventno; i++) {
302 fpTree->Fill();
303 }
304 }
305
306 if (iResult>=0 && pSrcESD) {
b952ebda 307 int nofObjects=fpEsd->GetList()->GetEntries();
83cb7e1d 308 *fpEsd=*pSrcESD;
b952ebda 309 if (nofObjects!=fpEsd->GetList()->GetEntries()) {
310 // The source ESD contains object not present in the target ESD
311 // before. Those objects will not be written to the tree since
312 // the branch layout has been created earlier.
313 // Create new tree with the additional branches, copy the entries
314 // of the current tree into the new tree, and continue.
315 TTree* pNewTree=new TTree("esdTree", "Tree with HLT ESD objects");
316 pNewTree->SetDirectory(0);
317 AliESDEvent* readESD=new AliESDEvent;
318 readESD->CreateStdContent();
319 readESD->ReadFromTree(fpTree);
320 fpEsd->Reset();
321 fpEsd->WriteToTree(pNewTree);
322 HLTDebug("cloning tree with %d entries", fpTree->GetEntries());
323 for (int event=0; event<fpTree->GetEntries(); event++) {
324 fpTree->GetEntry(event);
325 *fpEsd=*readESD;
326 pNewTree->Fill();
327 fpEsd->Reset();
328 }
329 fpFile->Close();
330 delete fpFile;
331 delete readESD;
332 delete fpTree;
333 fpFile=new TFile(fName, "RECREATE");
334 fpTree=pNewTree;
335 *fpEsd=*pSrcESD;
336 HLTDebug("new ESD with %d objects", fpEsd->GetList()->GetEntries());
337 }
83cb7e1d 338 fpTree->Fill();
339 iResult=1; // indicate tree to be written
340 }
341
342 if (iResult>0) {
343 fpFile->cd();
344 fpTree->GetUserInfo()->Add(fpEsd);
345 fpTree->Write(fpTree->GetName(),TObject::kOverwrite);
346 fpTree->GetUserInfo()->Clear();
347 }
348 }
9877121c 349#else //HAVE_NOT_ESD_COPY
83cb7e1d 350 // this is the old workaround, necessary for older AliRoot versions
351 // version<=v4-12-Release
352
90c37647 353 // we need to copy the ESD, I did not find an approptiate
354 // method, the workaround is to save the ESD in a temporary
355 // tree, read the content back into the ESD structure
356 // used for filling.
357 // Unfortunately the following code crashes at the second event.
358 // The expert on the ESD (Christian Klein Boesig) does not have
359 // a solution either. It seems to be a problem in ROOT.
360 // TTree* dummy=new TTree("dummy","dummy");
361 // dummy->SetDirectory(0);
362 // pESD->WriteToTree(dummy);
363 // dummy->Fill();
364 // dummy->GetUserInfo()->Add(pESD);
365 // fpEsd->ReadFromTree(dummy);
366 // dummy->GetEvent(0);
367 // fpEsd->WriteToTree(fpTree);
368 // fpTree->Fill();
369 // dummy->GetUserInfo()->Clear();
370 // delete dummy;
371 //
372 // The only way is via TChain, which is working on files only at the
373 // time of writing.
374 // We use temporary files for the new event to be copied into the
375 // existing tree.
376 //
90c37647 377 if (fName.IsNull()) {
378 // this is the first event, create the file on disk and write ESD
c5123824 379 TString origin;
380 origin.Insert(0, fDt.fOrigin, kAliHLTComponentDataTypefOriginSize);
381 origin.Remove(TString::kTrailing, ' ');
382 origin.ToUpper();
90c37647 383 fName="";
384 if (!fDirectory.IsNull()) {
385 fName+=fDirectory; fName+="/";
386 }
387 fName+="AliHLT"; fName+=origin;
c5123824 388 if (fDt!=kAliHLTDataTypeESDObject &&
389 fDt!=kAliHLTDataTypeESDTree) {
390
391 HLTWarning("non-standard ESD type %s", AliHLTComponent::DataType2Text(fDt).c_str());
392 TString id;
393 id.Insert(0, fDt.fID, kAliHLTComponentDataTypefIDsize);
394 id.Remove(TString::kTrailing, ' ');
395 id.ToUpper();
396 fName+="_"; fName+=id; fName+=".root";
397 } else {
398 fName+="ESDs.root";
90c37647 399 }
400
401 if (!gSystem->AccessPathName(fName)) {
402 // file exists, delete
403 TString shellcmd="rm -f ";
404 shellcmd+=fName;
405 gSystem->Exec(shellcmd);
c5123824 406 }
407 }
90c37647 408
409 TChain chain("esdTree");
410 TList cleanup;
411 cleanup.SetOwner();
412
413 int nofCurrentEvents=0;
414 if (iResult>=0) {
415 if (!gSystem->AccessPathName(fName)) {
416 // these are the other events, use the target file and temporary files to merge
417 // with TChain
418 chain.Add(fName);
419
420 if (eventno>=0) {
421 TFile file(fName);
422 if (!file.IsZombie()) {
423 TTree* pSrcTree;
424 file.GetObject("esdTree", pSrcTree);
425 if (pSrcTree) {
426 nofCurrentEvents=pSrcTree->GetEntries();
427 }
428 file.Close();
429 }
c5123824 430 }
431 }
90c37647 432 }
433
434 // synchronize and add empty events
435 if (nofCurrentEvents<eventno) {
436 iResult=1; // indicate files to merge
437 TTree* pTgtTree=new TTree("esdTree", "Tree with HLT ESD objects");
438 if (pTgtTree) {
439 pTgtTree->SetDirectory(0);
440 AliESDEvent* pTmpESD=new AliESDEvent;
441 if (pTmpESD) {
442 TString tmpfilename;
443 FILE* pTmpFile=gSystem->TempFileName(tmpfilename);
444 if (pTmpFile) {
445 fclose(pTmpFile);
446 pTmpFile=NULL;
447 cleanup.Add(new TObjString(tmpfilename));
448 TFile emptyevents(tmpfilename, "RECREATE");
449 if (!emptyevents.IsZombie()) {
450 pTmpESD->CreateStdContent();
451 pTmpESD->WriteToTree(pTgtTree);
452 HLTDebug("adding %d empty events to file %s", eventno-nofCurrentEvents, fName.Data());
453 for (int i=nofCurrentEvents; i<eventno; i++) {
454 pTgtTree->Fill();
455 }
456 pTgtTree->GetUserInfo()->Add(pTmpESD);
457 emptyevents.cd();
458 pTgtTree->Write();
459 emptyevents.Close();
460 chain.Add(tmpfilename);
461 pTgtTree->GetUserInfo()->Clear();
462 }
c5123824 463 }
90c37647 464 delete pTmpESD;
c5123824 465 } else {
90c37647 466 iResult=-ENOMEM;
c5123824 467 }
90c37647 468 delete pTgtTree;
469 } else {
470 iResult=-ENOMEM;
471 }
472 }
473
474 if (iResult>=0 && pSrcESD) {
475 // add the new event to the chain
476 iResult=1; // indicate files to merge
477 TString tmpfilename=WriteTempFile(pSrcESD);
478 if (!tmpfilename.IsNull()) {
479 chain.Add(tmpfilename);
480 cleanup.Add(new TObjString(tmpfilename));
481 }
482 }
483
484 if (iResult>0) {
485 // build temporary file name for chain output
486 TString tgtName;
487 FILE* pTmpFile=gSystem->TempFileName(tgtName);
488 if (pTmpFile) {
489 fclose(pTmpFile);
490 pTmpFile=NULL;
491
41b5c1b6 492 // there have been problems with the memory consumption when using
493 // TChain::Merge
83cb7e1d 494 // but using a separate loop soemtimes crashes in AliESDEvent::ReadFromTree
495 // since this is for backward compatiblity only, we take the TChain::Merge
496 chain.Merge(tgtName);
497// TFile tgtFile(tgtName, "RECREATE");
498// TTree* pTgtTree=new TTree("esdTree", "Tree with HLT ESD objects");
499// AliESDEvent* pTgtESD=new AliESDEvent;
500// if (pTgtTree && pTgtESD) {
501// pTgtESD->ReadFromTree(&chain);
502// pTgtESD->WriteToTree(pTgtTree);
503
504// int nofEvents=chain.GetEntries();
505// for (int event=0; event<nofEvents; event++) {
506// chain.GetEntry(event);
507// pTgtTree->Fill();
508// }
509
510// pTgtTree->GetUserInfo()->Add(pTgtESD);
511// tgtFile.cd();
512// pTgtTree->Write();
513// pTgtTree->GetUserInfo()->Clear();
514// } else {
515// iResult=-ENOMEM;
516// }
517
518// if (pTgtTree) delete pTgtTree;
519// if (pTgtESD) delete pTgtESD;
520// tgtFile.Close();
41b5c1b6 521
90c37647 522 // rename the merged file to the original file
523 TString shellcmd="mv ";
524 shellcmd+=tgtName + " " + fName;
525 if (gSystem->Exec(shellcmd)==0) {
526 HLTDebug("renaming %s to %s", tgtName.Data(), fName.Data());
527 } else {
528 HLTError("can not rename temporary file %s to %s", tgtName.Data(), fName.Data());
c5123824 529 }
90c37647 530 } else {
531 HLTError("can not get temporary file name from system");
532 iResult=-EBADF;
533 }
534 }
535
536 // delete temporary files
41b5c1b6 537 // the list objects are cleaned up by the TList destructor as the
90c37647 538 // list is owner
539 TIter entry(&cleanup);
540 while (TObject* pObj=entry.Next()) {
541 if (dynamic_cast<TObjString*>(pObj)) {
542 TString shellcmd="rm -f ";
543 shellcmd+=(dynamic_cast<TObjString*>(pObj))->GetString();
544 gSystem->Exec(shellcmd);
545 }
546 }
9877121c 547#endif //HAVE_NOT_ESD_COPY
90c37647 548
549 return iResult;
550}
551
f1207f29 552TString AliHLTEsdManagerImplementation::AliHLTEsdListEntry::WriteTempFile(AliESDEvent* pESD) const
90c37647 553{
554 // see header file for class documentation
f7561f8d 555 int iResult=0;
90c37647 556 TString tmpfilename;
557 FILE* pTmpFile=gSystem->TempFileName(tmpfilename);
558 if (pTmpFile) {
559 fclose(pTmpFile);
560 pTmpFile=NULL;
561
562 TFile file(tmpfilename, "RECREATE");
563 if (!file.IsZombie()) {
f1207f29 564 TTree* pTree=AliHLTEsdManagerImplementation::EmbedIntoTree(pESD);
90c37647 565 if (pTree) {
566 file.cd();
567 if (pTree->Write()>0) {
c5123824 568 } else {
90c37647 569 HLTError("can not write esd tree to temporary file %s", tmpfilename.Data());
c5123824 570 }
90c37647 571
572 pTree->GetUserInfo()->Clear();
573 delete pTree;
c5123824 574 } else {
575 iResult=-ENOMEM;
576 }
90c37647 577 file.Close();
578 } else {
579 HLTError("can not open file %s", tmpfilename.Data());
3dd8541e 580 iResult=-EBADF;
c5123824 581 }
90c37647 582 } else {
583 HLTError("can not get temporary file name from system");
584 iResult=-EBADF;
c5123824 585 }
90c37647 586
587 if (iResult<0) {
588 if (gSystem->AccessPathName(tmpfilename)==0) {
589 TString shellcmd="rm -f ";
590 shellcmd+=tmpfilename;
591 gSystem->Exec(shellcmd);
592 }
593 tmpfilename="";
594 }
595 return tmpfilename;
596}
597
f1207f29 598void AliHLTEsdManagerImplementation::AliHLTEsdListEntry::SetDirectory(const char* directory)
90c37647 599{
600 // see header file for class documentation
601 if (!directory) return;
602 if (!fName.IsNull()) {
603 HLTWarning("ESD entry already in writing mode (%s), ignoring directory", fName.Data());
604 return;
605 }
606 fDirectory=directory;
607}
608
f1207f29 609void AliHLTEsdManagerImplementation::AliHLTEsdListEntry::Delete()
90c37647 610{
611 // see header file for class documentation
612 if (fName.IsNull()) return;
613 if (gSystem->AccessPathName(fName)!=0) return;
614
615 TString shellcmd="rm -f ";
616 shellcmd+=fName;
617 gSystem->Exec(shellcmd);
618 fName="";
619}
620
f1207f29 621const char* AliHLTEsdManagerImplementation::AliHLTEsdListEntry::GetFileName() const
90c37647 622{
623 // see header file for class documentation
624 return fName.Data();
c5123824 625}
f527516f 626
627const char* AliHLTEsdManagerImplementation::AliHLTEsdListEntry::GetPrefix()
628{
629 // see header file for class documentation
630 if (fPrefix.IsNull()) {
631 fPrefix.Insert(0, fDt.fOrigin, kAliHLTComponentDataTypefOriginSize);
632 fPrefix.Remove(TString::kTrailing, ' ');
633 fPrefix.ToUpper();
634 if (!fPrefix.Contains("HLT")) {
635 fPrefix.Insert(0, "HLT");
636 }
637 }
638 return fPrefix.Data();
639}
640
57584811 641int AliHLTEsdManagerImplementation::Merge(AliESDEvent* pTgt, AliESDEvent* pSrc) const
f527516f 642{
643 // see header file for class documentation
644 int iResult=0;
645 if (!pTgt || !pSrc) return -EINVAL;
646
f527516f 647 TIter next(pSrc->GetList());
648 TObject* pSrcObject=NULL;
649 TString name;
57584811 650 static int warningCount=0;
f527516f 651 while ((pSrcObject=next())) {
652 if(!pSrcObject->InheritsFrom("TCollection")){
653 // simple objects
18a9a5ad 654 if(pSrcObject->InheritsFrom("AliHLTTriggerDecision")){
e13512e4 655 //pSrcObject->Print();
656 // Matthias 2009-07-3: there is a problem with the additional object
657 // in the ESD when the tree is filled
18a9a5ad 658 TObject* pTgtObject=pTgt->GetList()->FindObject(name);
659 if (pTgtObject) {
e13512e4 660 //pSrcObject->Copy(*pTgtObject);
18a9a5ad 661 } else {
e13512e4 662 //pTgt->AddObject(pSrcObject->Clone());
18a9a5ad 663 }
664 } else {
665 // TODO: implement the handling of other objects, some kind of mapping
666 }
f527516f 667 } else if(pSrcObject->InheritsFrom("TClonesArray")){
668 TClonesArray* pTClA=dynamic_cast<TClonesArray*>(pSrcObject);
669 if (pTClA!=NULL && pTClA->GetEntriesFast()>0) {
57584811 670 name=pTClA->GetName();
f527516f 671 TObject* pTgtObject=pTgt->GetList()->FindObject(name);
672 TClonesArray* pTgtArray=NULL;
57584811 673 if (pTgtObject!=NULL && pTgtObject->InheritsFrom("TClonesArray")){
f527516f 674 pTgtArray=dynamic_cast<TClonesArray*>(pTgtObject);
675 if (pTgtArray) {
676 TString classType=pTClA->Class()->GetName();
677 if (classType.CompareTo(pTgtArray->Class()->GetName())==0) {
678 if (pTgtArray->GetEntries()==0) {
57584811 679 pTgtArray->ExpandCreate(pTClA->GetEntries());
680 for(int i=0; i<pTClA->GetEntriesFast(); ++i){
681 (*pTClA)[i]->Copy(*((*pTgtArray)[i]));
682 }
f527516f 683 } else {
57584811 684 if (warningCount++<10) {
685 HLTWarning("TClonesArray \"%s\" in target ESD %p is already filled with %d entries",
686 name.Data(), pTgt, pTgtArray->GetEntries());
687 }
688 iResult=-EBUSY;
f527516f 689 }
690 } else {
57584811 691 if (warningCount++<10) {
692 HLTWarning("TClonesArray \"%s\" exists in target ESD %p, but describes incompatible class type %s instead of %s",
693 name.Data(), pTgt, pTgtArray->GetClass()->GetName(), pTClA->GetClass()->GetName());
694 }
695 iResult=-EBUSY;
f527516f 696 }
697 } else {
57584811 698 if (warningCount++<10) {
699 HLTError("internal error: dynamic cast failed for object %s %p", pTgtObject->GetName(), pTgtObject);
700 }
701 iResult=-EBUSY;
f527516f 702 }
703 } else if (pTgtObject) {
57584811 704 if (warningCount++<10) {
705 HLTWarning("object \"%s\" does already exist in target ESD %p, but is %s rather than TClonesArray"
706 " skipping data",
707 name.Data(), pTgt, pTgtObject->Class()->GetName());
f527516f 708 }
57584811 709 iResult=-EBUSY;
f527516f 710 } else {
57584811 711 if (warningCount++<10) {
712 HLTWarning("object \"%s\" does not exist in target ESD %p, data can not be copied because it will be lost when filling the tree",
713 name.Data(), pTgt);
f527516f 714 }
57584811 715 iResult=-ENOENT;
f527516f 716 }
717 }
718 }
719 }
720 return iResult;
721}