Patch for AOD to keep the extrapolated pT at

[u/mrichter/AliRoot.git] / EVE / EveHLT / AliEveEventBufferOffline.cxx

diff --git a/EVE/EveHLT/AliEveEventBufferOffline.cxx b/EVE/EveHLT/AliEveEventBufferOffline.cxx
index 9f6403434ad2d5fce1f95fe51d78cafd7ff6d8ac..eacd88483662799c36825e6b2713da94f2d37ec6 100644 (file)
--- a/EVE/EveHLT/AliEveEventBufferOffline.cxx
+++ b/EVE/EveHLT/AliEveEventBufferOffline.cxx
@@ -1,3 +1,9 @@
+//-*- Mode: C++ -*-
+// $Id$
+//
+// 2010 Svein Lindal <slindal@fys.uio.no>                        *
+//         for The ALICE HLT Project.                                    *
+// Offline Buffer
 #if __GNUC__>= 3
    using namespace std;
 #endif
@@ -5,6 +11,8 @@
 #include "AliEveEventBufferOffline.h"
 
 #include "AliESDEvent.h"
+#include "AliEveEventBufferOffline.h"
+#include "AliEveEventBuffer.h"
 
 #include "TTimer.h"
 #include "TFile.h"
@@ -43,11 +51,12 @@ AliEveEventBufferOffline::AliEveEventBufferOffline(TString filename)  :
     return;
   }
 
-  fTree = dynamic_cast<TTree *>(fFile->Get("esdTree"));
-  cout << "File has " << fNEntries << "events" << endl;
-  fNEntries = fTree->GetEntries();
-  fEvent->ReadFromTree(fTree);
-  
+  fTree = dynamic_cast<TTree *>(fFile->Get("HLTesdTree"));
+  if(fTree) {
+    fNEntries = fTree->GetEntries();
+    cout << "File has " << fNEntries << "events" << endl;
+    fEvent->ReadFromTree(fTree);
+  }
 }
 
 
@@ -84,12 +93,12 @@ TObject * AliEveEventBufferOffline::GetEventFromSource() {
   }
 
   //Copy event into new event (must be deleted when no longer needed in list!!
-  AliESDEvent * event = new AliESDEvent();
-  fEvent->Copy(*event);
-  cout << event->GetNumberOfCaloClusters() << endl;
+  //AliESDEvent * event = new AliESDEvent();
+  //fEvent->Copy(*event);
+  //cout << event->GetNumberOfCaloClusters() << endl;
 
-  if (event) {
-    return dynamic_cast<TObject*>(event);
+  if (fEvent) {
+    return dynamic_cast<TObject*>(fEvent);
   } else {
     cout << "error getting event" << endl;
     return NULL;
@@ -105,9 +114,9 @@ void AliEveEventBufferOffline::ConnectToSource() {
 
 
 ///_____________________________________________________________________
-void AliEveEventBufferOffline::WriteToFile(){
+void AliEveEventBufferOffline::WriteToFile(Int_t runnumber){
   //Inherited from AliEveEventBuffer
-  TFile * file = TFile::Open(Form("Event_%d_ITS.root", 100), "RECREATE"); 
-  fEventBuffer->At(fBIndex[kTop])->Write("blockList", TObject::kSingleKey);
+  TFile * file = TFile::Open(Form("%d_0x%016llu_ESD.root", runnumber, GetEventId()), "RECREATE"); 
+  fEventBuffer->At(fBIndex[kCurrent])->Write("blockList", TObject::kSingleKey);
   file->Close();
 }