Correct streamer (Haavard)

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

#if !defined(__CINT__) || defined(__MAKECINT__)

#include <Riostream.h>
#include "TFile.h"
#include "TTree.h"

#include "AliAODEvent.h"
#include "AliAODHeader.h"
#include "AliAODVertex.h"
#include "AliAODTrack.h"
#include "AliAODCluster.h"

#endif

void ReadAOD(const char *fileName = "AliAOD.root") {

  // open input file and get the TTree
  TFile inFile(fileName, "READ");
  TTree *aodTree = (TTree*)inFile.Get("AOD");

  AliAODEvent *aod = (AliAODEvent*)aodTree->GetUserInfo()->FindObject("AliAODEvent");
  TIter next(aod->GetList());
  TObject *el;
  while((el=(TNamed*)next())) 
    aodTree->SetBranchAddress(el->GetName(),aod->GetList()->GetObjectRef(el));

  // loop over events
  Int_t nEvents = aodTree->GetEntries();
  for (Int_t nEv = 0; nEv < nEvents; nEv++) {
    cout << "Event: " << nEv+1 << "/" << nEvents << endl;

    // read events
    aodTree->GetEvent(nEv);
    
    // set pointers
    aod->GetStdContent();

    //print event info
    aod->GetHeader()->Print();

    // loop over tracks
    Int_t nTracks = aod->GetNTracks();
    for (Int_t nTr = 0; nTr < nTracks; nTr++) {
      
      AliAODTrack *tr = aod->GetTrack(nTr);

      // print track info
      cout << nTr+1 << "/" << nTracks << ": track pt: " << tr->Pt();
      if (tr->GetProdVertex()) {
	cout << ", vertex z of this track: " << tr->GetProdVertex()->GetZ();
      }
      cout << endl;
    }

    // loop over vertices
    Int_t nVtxs = aod->GetNVertices();
    for (Int_t nVtx = 0; nVtx < nVtxs; nVtx++) {
      
      // print track info
      cout << nVtx+1 << "/" << nVtxs << ": vertex z position: " << aod->GetVertex(nVtx)->GetZ() << endl;
    }
  }
  
  return;
}