#include "Riostream.h"

///////////////////////////////////////////////////////////////////
//////// Go4 GUI example script divhistos.C
//       J.Adamczewski, gsi, May 2006
//       improvement for float ratios by S.Augustin, Sep 2018
// NOTE: to be run in Go4 GUI local command line only!
//       NEVER call this script in remote analysis process!!!
/////// Functionality:
// divide histogram of name 1 by histogram of name2
/////// Usage:
// The draw flag switches if the results are displayed each time this makro is called
// if display is switched off, the result histogram is just updated in browser and existing displays
///////


TH1F* CloneTHI2F(TH1 *horig)
{
   TAxis *ax = horig->GetXaxis();

   Int_t nbins   = ax->GetNbins();
   Double_t vmin = ax->GetXmin();
   Double_t vmax = ax->GetXmax();

   TH1F *hnew = new TH1F("tmp", "tmp", nbins, vmin, vmax);
   hnew->Add(horig, 1);

   return hnew;
}

TH2F* CloneTHI2F_2D(TH1 *horig)
{
   TAxis *ax = horig->GetXaxis();
   Int_t xnbins   = ax->GetNbins();
   Double_t xvmin = ax->GetXmin();
   Double_t xvmax = ax->GetXmax();
   TAxis *ay = horig->GetYaxis();
   Int_t ynbins   = ay->GetNbins();
   Double_t yvmin = ay->GetXmin();
   Double_t yvmax = ay->GetXmax();
   TH2F* hnew = new TH2F("tmp", "tmp", xnbins, xvmin, yvmax, ynbins, yvmin, yvmax);
   hnew->Add(horig, 1);

   return hnew;
}

Bool_t divhistos(const char *name1, const char *name2, Bool_t draw, Bool_t float_division=kFALSE)
{
   if(!TGo4AbstractInterface::Instance() || go4!=TGo4AbstractInterface::Instance()) {
      std::cout <<"FATAL: Go4 gui macro executed outside Go4 GUI!! returning." << std::endl;
      return kFALSE;
   }
   TString fullname1 = go4->FindItem(name1);
   TObject *ob1 = go4->GetObject(fullname1,1000); // 1000=timeout to get object from analysis in ms
   TH1 *his1 = nullptr, *his2 = nullptr;
   if(ob1 && ob1->InheritsFrom("TH1"))
      his1 = (TH1 *) ob1;
   if(!his1) {
      std::cout <<"divhistos could not get histogram "<<fullname1 << std::endl;
      return kFALSE;
   }
   TString fullname2 = go4->FindItem(name2);
   TObject *ob2 = go4->GetObject(fullname2,1000); // 1000=timeout to get object from analysis in ms
   if(ob2 && ob2->InheritsFrom("TH1"))
      his2 = (TH1 *) ob2;
   if(!his2) {
      std::cout <<"divhistos could not get histogram "<<fullname2 << std::endl;
      return kFALSE;
   }
   TH1 *result = nullptr;
   Int_t dim = his1->GetDimension();
   if (float_division) {
       if(dim==1)
         result = CloneTHI2F(his1);
       else if (dim==2)
         result = CloneTHI2F_2D(his1);
       else
         result = (TH1 *) his1->Clone();
   } else {
       result = (TH1 *) his1->Clone();
   }

   TString n1 = his1->GetName();
   TString n2 = his2->GetName();
   TString t1 = his1->GetTitle();
   TString t2 = his2->GetTitle();
   TString soper;
   soper.Form(") : (");
   TString finalname = TString("(")+n1+soper+n2+")";
   TString finaltitle = TString("(")+t1+soper+t2+")";
   result->SetName(finalname);
   result->SetTitle(finalname);
   result->Sumw2();
   result->Divide(his2);
   result->SetDirectory(nullptr);
   TString rname = go4->SaveToMemory("Divs", result, kTRUE);
   std::cout<< "Saved result histogram to " << rname.Data() <<std::endl;
   if(draw) go4->DrawItem(rname);
   return kTRUE;
}