#include <Root/Headers.h>
#include <TTreePerfStats.h>
#include <vector>

TH1D *h_Vertex_z, *h_Vertex_nTracks;
TTreePerfStats *ps;

void TTreePerfStatsTest() {
  

  h_Vertex_z = new TH1D("Vertex_z",";PVertex_z [mm];events", 100, -230, 235);
  h_Vertex_nTracks = new TH1D("Vertex_nTracks", ";PVertex_nTracks;events", 200, 0, 200);

  vector<int>   *Vertex_nTracks = new vector<int>;
  vector<float> *Vertex_z     = new vector<float>;


  TFile *f = new TFile("/data_atlas/eckweiler/MonteCarlo/UnfoldingSystematics/user10.SebastianEckweiler.Skimmed.user10.DavidWilkinsMiller.AprilRep.v01.mc09_7TeV.105015.J6_pythia_jetjet.recon.ESD.e468_s766_s767_r1251_AANT.20100601_EXT0/user10.SebastianEckweiler.Skimmed.user10.DavidWilkinsMiller.AprilRep.v01.mc09_7TeV.105015.J6_pythia_jetjet.recon.ESD.e468_s766_s767_r1251_AANT.20100601.EXT0._00091.SkimmedD3PD.root");

/*   TTree *c = (TTree*) f->Get("CollectionTree"); */
  TChain *c = new TChain("CollectionTree");
  c->Add( "/data_atlas/eckweiler/MonteCarlo/UnfoldingSystematics/user10.SebastianEckweiler.Skimmed.user10.DavidWilkinsMiller.AprilRep.v01.mc09_7TeV.105015.J6_pythia_jetjet.recon.ESD.e468_s766_s767_r1251_AANT.20100601_EXT0/*root" );
  //  c->Add( f->GetName() );
  std::cout << "chain has " << c->GetListOfFiles()->GetEntries() << " files " << std::endl;

  ps = new TTreePerfStats("ioperf",c);

  c->SetBranchStatus("*", 0);
  c->SetBranchStatus("Vertex_nTracks", 1);
  c->SetBranchStatus("Vertex_z", 1);
  
  c->SetBranchAddress("Vertex_nTracks", &Vertex_nTracks);
  c->SetBranchAddress("Vertex_z", &Vertex_z);
    
  c->SetCacheSize(10000000);
  c->AddBranchToCache("*");

  int n = c->GetEntries();
  for (int i = 0;i < n; ++i) {
    c->GetEntry(i);

    h_Vertex_z      ->Fill( Vertex_z->at(0) );
    h_Vertex_nTracks->Fill( Vertex_nTracks->at(0) );
  }
  ps->SaveAs("perfstats.root");
}