Use ESD objects from a tree (T.Kuhr)

[u/mrichter/AliRoot.git] / STEER / AliESDv0Analysis.C

diff --git a/STEER/AliESDv0Analysis.C b/STEER/AliESDv0Analysis.C
index 101dbd885c14c118595919cc2ee90b9bf61145f5..bf2dd7d2c7e648619da495c95bcb3c6ebb0e0792 100644 (file)
--- a/STEER/AliESDv0Analysis.C
+++ b/STEER/AliESDv0Analysis.C
@@ -8,7 +8,7 @@
 
 #if !defined( __CINT__) || defined(__MAKECINT__)
   #include <Riostream.h>
-  #include "TKey.h"
+  #include <TTree.h>
   #include "TFile.h"
   #include "TH1F.h"
   #include "TH2F.h"
@@ -30,22 +30,22 @@ Int_t AliESDv0Analysis(Int_t nev=1) {
 
    TFile *ef=TFile::Open("AliESDs.root");
    if (!ef->IsOpen()) {cerr<<"Can't AliESDs.root !\n"; return 1;}
+   AliESD* event = new AliESD;
+   TTree* tree = (TTree*) ef->Get("esdTree");
+   if (!tree) {cerr<<"no ESD tree found\n"; return 1;};
+   tree->SetBranchAddress("ESD", &event);
 
    TStopwatch timer;
    Int_t rc=0,n=0;
-   TKey *key=0;
-   TIter next(ef->GetListOfKeys());
 
    //****** Tentative particle type "concentrations"
    Double_t c[5]={0.0, 0.0, 0.1, 0.1, 0.1};
 
    //******* The loop over events
-   while ((key=(TKey*)next())!=0) {
+   while (tree->GetEvent(n)) {
 
      cerr<<"Processing event number : "<<n++<<endl;
 
-     AliESD *event=(AliESD*)key->ReadObj();
-
      Int_t nv0=event->GetNumberOfV0s();
      cerr<<"Number of ESD v0s : "<<nv0<<endl; 
 
@@ -74,9 +74,11 @@ Int_t AliESDv0Analysis(Int_t nev=1) {
        Double_t mass=v0->GetEffMass();
        hm->Fill(mass);
      } 
-     delete event;
    }
 
+   delete event;
+   ef->Close();
+
    timer.Stop(); timer.Print();
 
    hm->Draw();