style modifications (Markus)

[u/mrichter/AliRoot.git] / HLT / TRD / AliHLTTRDEsdWriterComponent.cxx

diff --git a/HLT/TRD/AliHLTTRDEsdWriterComponent.cxx b/HLT/TRD/AliHLTTRDEsdWriterComponent.cxx
index 47cc3a43a335565b1479b68c29e03b25d65a4189..2cda6af98f75854487610381cfe1de564770c7c6 100644 (file)
--- a/HLT/TRD/AliHLTTRDEsdWriterComponent.cxx
+++ b/HLT/TRD/AliHLTTRDEsdWriterComponent.cxx
@@ -37,9 +37,11 @@ ClassImp(AliHLTTRDEsdWriterComponent)
 AliHLTTRDEsdWriterComponent::AliHLTTRDEsdWriterComponent()
 :AliHLTRootFileWriterComponent()
   ,fTree(NULL)
+  ,fFrTree(NULL)
   ,fESD(NULL)
   ,fESDfriend(NULL)
   ,fFile(NULL)
+  ,fFrFile(NULL)
   ,fTracksArray(NULL)
 {
   // see header file for class documentation
@@ -52,9 +54,11 @@ AliHLTTRDEsdWriterComponent::AliHLTTRDEsdWriterComponent()
 AliHLTTRDEsdWriterComponent::AliHLTTRDEsdWriterComponent(const AliHLTTRDEsdWriterComponent&)
   :AliHLTRootFileWriterComponent()
   ,fTree(NULL)
+  ,fFrTree(NULL)
   ,fESD(NULL)
   ,fESDfriend(NULL)
   ,fFile(NULL)
+  ,fFrFile(NULL)
   ,fTracksArray(NULL)
 {
 }
@@ -77,15 +81,16 @@ int AliHLTTRDEsdWriterComponent::InitWriter()
   // see header file for class documentation
   
   fFile = new TFile("AliHLTTRDESDs.root", "recreate");
-  fFile->cd();
   fESD = new AliESDEvent;
   fESD->CreateStdContent();
-  fTree = new TTree("esdTree", "Tree with HLT ESD objects");
+  fTree = new TTree("esdTree", "Tree with HLT::TRD ESD objects");
   fESD->WriteToTree(fTree);
+  fFrFile = new TFile("AliHLTTRDESDfriends.root", "recreate");
   fESDfriend = new AliESDfriend();
-  /*TBranch* br=*/fTree->Branch("ESDfriend.","AliESDfriend", &fESDfriend);
-  //br->SetFile("AliHLTTRDESDfriends.root");
+  fFrTree = new TTree("esdFriendTree", "Tree with HLT::TRD ESD Friend objects");
+  fFrTree->Branch("ESDfriend.","AliESDfriend", &fESDfriend);
   fESD->AddObject(fESDfriend);
+  fFile->cd();
   fTree->GetUserInfo()->Add(fESD);
   fTracksArray = new TClonesArray("AliTRDtrackV1");
 
@@ -99,9 +104,16 @@ int AliHLTTRDEsdWriterComponent::CloseWriter()
   // see header file for class documentation
 
   //fTree->Print();
+  fFile->cd();
+  fTree->Write(fTree->GetName(),TObject::kOverwrite);
   fFile->Write();
   fFile->Close();
+  fFrFile->cd();
+  fFrTree->Write(fFrTree->GetName(),TObject::kOverwrite);
+  fFrFile->Write();
+  fFrFile->Close();
   delete fFile; fFile=0;
+  delete fFrFile; fFrFile=0;
   //delete fTree;
   delete fTracksArray; fTracksArray=0;
 
@@ -147,6 +159,7 @@ int AliHLTTRDEsdWriterComponent::DumpEvent( const AliHLTComponentEventData& /*ev
   fESD->SetOrbitNumber(GetOrbitNumber());
   fESD->SetBunchCrossNumber(GetBunchCrossNumber());
   fESD->SetTimeStamp(GetTimeStamp());
+  fESD->SetEventType(7);
 
   const AliHLTCTPData* pCTPData=CTPData();
   if (pCTPData) {
@@ -175,8 +188,10 @@ int AliHLTTRDEsdWriterComponent::DumpEvent( const AliHLTComponentEventData& /*ev
   }
   
   fESD->GetESDfriend(fESDfriend);
-  Int_t nb = fTree->Fill();  //endless-
-  printf("Tree filled with %i bytes\n", nb);
+  Int_t nb = fTree->Fill();
+  HLTInfo("Tree filled with %i bytes\n", nb);  
+  nb = fFrTree->Fill();
+  HLTInfo("FrTree filled with %i bytes\n", nb);
   fESD->Reset();
   fESDfriend->~AliESDfriend();
   new (fESDfriend) AliESDfriend();