#include "TFile.h"
#include "TSystem.h"
#include "TH1.h"
#include "TRandom.h"
#include "TStopwatch.h"
#include "TKey.h"
#include "TTree.h"
#include "TROOT.h"

const Int_t N = 10000;       //number of events to be processed
TStopwatch timer;

void billw(const char *billname, Int_t compress)
{
   //write N histograms as keys
   timer.Start();
   TFile f(billname, "recreate", "bill benchmark with keys", compress);
   TH1F h("h", "h", 1000, -3, 3);
   h.FillRandom("gaus", 50000);

   for (Int_t i = 0; i < N; i++) {
      char name[20];
      sprintf(name, "h%d", i);
      h.SetName(name);
      h.Fill( 2 * gRandom->Rndm());
      h.Write();
   }
   timer.Stop();
   printf("billw%d  : RT=%7.3f s, Cpu=%7.3f s, File size= %9d bytes, CX= %g\n",
          compress, timer.RealTime(), timer.CpuTime(), (Int_t)f.GetBytesWritten(),
          f.GetCompressionFactor());
}

void billr(const char *billname, Int_t compress)
{
   TDirectory::TContext ctx{nullptr}; // Don't register histograms to the current directory

   //read N histograms from keys
   timer.Start();
   TFile f(billname);
   TIter next(f.GetListOfKeys());
   TH1F *h;

   TKey *key;
   Int_t i = 0;
   auto hmean = new TH1F("hmean", "hist mean from keys", 100, 0, 1);

   while ((key = (TKey*)next())) {
      h = (TH1F*)key->ReadObj();
      hmean->Fill(h->GetMean());
      delete h;
      i++;
   }
   timer.Stop();
   printf("billr%d  : RT=%7.3f s, Cpu=%7.3f s\n", compress, timer.RealTime(), timer.CpuTime());
}

void billtw(const char *billtname, Int_t compress)
{
   //write N histograms to a Tree
   timer.Start();
   TFile f(billtname,"recreate", "bill benchmark with trees", compress);
   auto h = new TH1F("h", "h", 1000, -3, 3);
   h->FillRandom("gaus", 50000);
   auto T = new TTree("T", "test bill");
   T->Branch("event", "TH1F", &h, 64000, 0);
   for (Int_t i = 0; i < N; i++) {
      char name[20];
      sprintf(name, "h%d", i);
      h->SetName(name);
      h->Fill(2 * gRandom->Rndm());
      T->Fill();
   }
   T->Write();
   delete T;
   timer.Stop();
   printf("billtw%d : RT=%7.3f s, Cpu=%7.3f s, File size= %9d bytes, CX= %g\n",
          compress, timer.RealTime(), timer.CpuTime(), (Int_t)f.GetBytesWritten(),
          f.GetCompressionFactor());
}

void billtr(const char *billtname, Int_t compress)
{
   //read N histograms from a tree
   timer.Start();
   TFile f(billtname);
   TH1F *h = nullptr;
   auto T = f.Get<TTree>("T");
   T->SetBranchAddress("event", &h);
   auto hmeant = new TH1F("hmeant", "hist mean from tree", 100, 0, 1);
   Long64_t nentries = T->GetEntries();
   for (Long64_t i = 0; i < nentries; i++) {
      T->GetEntry(i);
      hmeant->Fill(h->GetMean());
   }
   timer.Stop();
   printf("billtr%d : RT=%7.3f s, Cpu=%7.3f s\n", compress, timer.RealTime(), timer.CpuTime());
}

void tree202_benchmarks()
{
   TString dir = gSystem->GetDirName(gSystem->UnixPathName(__FILE__));
   TString bill = dir + "/bill.root";
   TString billt = dir + "/billt.root";

   TStopwatch totaltimer;
   totaltimer.Start();
   for (Int_t compress = 0; compress < 2; compress++) {
      billw(bill, compress);
      billr(bill, compress);
      billtw(billt, compress);
      billtr(billt, compress);
   }
   gSystem->Unlink(bill);
   gSystem->Unlink(billt);
   totaltimer.Stop();
   Double_t realtime = totaltimer.RealTime();
   Double_t cputime = totaltimer.CpuTime();
   printf("billtot : RT=%7.3f s, Cpu=%7.3f s\n", realtime, cputime);
   //reference is a P IV 2.4 GHz
   Float_t rootmarks = 600 * (16.98 + 14.40) / (realtime + cputime);
   printf("******************************************************************\n");
   printf("*  ROOTMARKS =%6.1f   *  Root%-8s  %d/%d\n", rootmarks,
          gROOT->GetVersion(), gROOT->GetVersionDate(), gROOT->GetVersionTime());
   printf("******************************************************************\n");
}