#include "Riostream.h"

///////////////////////////////////////////////////////////////////
//////// Go4 GUI example script projectionX.C
//          J.Adamczewski, gsi, May 2006
// NOTE: to be run in Go4 GUI local command line only!
//       NEVER call this script in remote analysis process!!!
/////// Functionality:
// create projection of histogram name1 along x axis
/////// Usage:
// parameters of root TH2::ProjectionX method may be specified:
//   the range of projection concerning y bins(firstybin, lastybin);
// 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
///////
Bool_t projectionX(const char *name1, const char *polyname, Int_t firstybin, Int_t lastybin, Bool_t draw)
{
   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
   TH2 *his1 = nullptr;
   if(ob1 && ob1->InheritsFrom("TH2")) his1 = (TH2 *) ob1;
   if(!his1) {
      std::cout <<"projectionX could not get 2d histogram "<<fullname1 << std::endl;
      return kFALSE;
   }
   TGo4PolyCond *poly =nullptr;
   TCutG *cut = nullptr;
   if(polyname && strlen(polyname) > 0){
      TString fullname2 = go4->FindItem(polyname);
      TObject *ob2 = go4->GetObject(fullname2,1000); // 1000=timeout to get object from analysis in ms
      if(ob2 && ob2->InheritsFrom("TGo4Condition")) poly = (TGo4PolyCond*) ob2;
      if(!poly) {
         std::cout <<"projectionX could not get polygon condition "<<polyname << std::endl;
         return kFALSE;
      }
   }
   TString n1 = his1->GetName();
   TString t1 = his1->GetTitle();
   TString oper;
   oper.Form("(ylo=%d, yup=%d)",firstybin,lastybin);
   TString finalname = TString("X-Projection of ")+n1;
   TString finaltitle = TString("X-Proj.")+oper+" of "+t1;
   TString options("");
   if(poly){
      cut = (TCutG*) poly->GetCut(kFALSE)->Clone("tempcut");  // therefore create clone
      options.Form("[%s]",cut->GetName());
   }
   TH1 *result = his1->ProjectionX(finalname.Data(),firstybin,lastybin,options.Data());
   if(poly) delete cut; // clone
   result->SetTitle(finaltitle);
   result->SetDirectory(nullptr);
   TString rname = go4->SaveToMemory("Projections", result, kTRUE);
   std::cout << "Saved result histogram to " << rname.Data() <<std::endl;
   if(draw) {
      ViewPanelHandle panel1 = go4->StartViewPanel();
      go4->DivideViewPanel(panel1, 1, 2);
      go4->SelectPad(panel1, 1);
      go4->DrawItem(fullname1, panel1,"COLZ");
      go4->SelectPad(panel1, 2);
      go4->DrawItem(rname, panel1);
   }
   return kTRUE;
}