// add_uphi.cpp
// Run this by doing from the directory containing root files (nobackup)
// root -l
// .x ../root_macros/add_uphi.cpp("pizero_200_211_52_+0000","pizero_200_211_52_-3000","pizero_200_211_52_+3000")

// requires that Ntuples have been saved first to a root file "..._w_plot.root"
// using skewness.cpp macro

#include <string.h>

#include <iostream>// std::cout
#include <string>// std::string
#include <fstream>// std::ifstream
#include <sstream>// std::istringstream
#include <algorithm>// std::remove & std::erase
#include <boost/algorithm/string/replace.hpp>// replacing subtrings with other substrings in strings

using namespace std;

void add_uphi(string cent_shms, string minus_shms, string plus_shms){

  // Directories for LICHEN:
  TString hist_dir = "/home/huberg/r2d2/simc/simc_gfortran.omega/outfiles/"; // Directory for .hist files that contain normfac
  TString root_dir = "/home/huberg/r2d2/simc/simc_gfortran.omega/nobackup/"; // Directory for .root files that contain meson plots
  TString pr_dir = "/home/huberg/r2d2/pizero/simc_plots/plots_root_files/";      // Directory for saving plots as .root files
  TString png_dir = "/home/huberg/r2d2/pizero/simc_plots/plots_png_files/";  // Directory for saving plots as .png files
  TString eps_dir = "/home/huberg/r2d2/pizero/simc_plots/plots_eps_files/";  // Directory for saving plots as .eps files
  

  // central shms
  string cent_shms_root0 = cent_shms + "_w_plot.root";
  TString cent_shms_root = cent_shms_root0.c_str();
  TFile f0(cent_shms_root);

  // minus shms
  string minus_shms_root0 = minus_shms + "_w_plot.root";
  TString minus_shms_root = minus_shms_root0.c_str();
  TFile f1(minus_shms_root);

  // plus shms
  string plus_shms_root0 = plus_shms + "_w_plot.root";
  TString plus_shms_root = plus_shms_root0.c_str();
  TFile f2(plus_shms_root);

  /*
  //u plots
  f0.cd();
  TH2F *u_cent = (TH2F*)gDirectory->Get("u");

  f1.cd();
  TH2F *u_minus = (TH2F*)gDirectory->Get("u");

  f2.cd();
  TH2F *u_plus = (TH2F*)gDirectory->Get("u");

  TCanvas *d2 = new TCanvas();

  u_cent->SetLineColor(4);
  u_minus->SetLineColor(3);
  u_plus->SetLineColor(2);
  
  u_cent->Draw();
  u_minus->Draw("same");
  u_plus->Draw("same");
  */

  /*
  //phi plots
  f0.cd();
  TH1F *ph_cent = (TH1F*)gDirectory->Get("phipq");

  f1.cd();
  TH1F *ph_minus = (TH1F*)gDirectory->Get("phi_pq");

  f2.cd();
  TH1F *ph_plus = (TH1F*)gDirectory->Get("phi_pq");
  
  TCanvas *t2 = new TCanvas("t2", "t2");
  
  ph_cent->SetLineColor(4);
  ph_minus->SetLineColor(3);
  ph_plus->SetLineColor(2);
  
  ph_cent->Draw();
  ph_minus->Draw("same");
  ph_plus->Draw("same");
  */
  
  //polar plots
  f0.cd();
  TGraphPolar *cent_h0 = (TGraphPolar*)gDirectory->Get("uphiplot;1");
  cent_h0->Draw("AOP");

  f1.cd();
  TGraphPolar *minus_h0 = (TGraphPolar*)gDirectory->Get("uphiplot;1");
  minus_h0->Draw("AOP");

  f2.cd();
  TGraphPolar *plus_h0 = (TGraphPolar*)gDirectory->Get("uphiplot;1");
  plus_h0->Draw("AOP");
  
  TCanvas *uphi1 = new TCanvas("uphi", "uphi", 500, 500);
  cent_h0->Draw("AOP");
  uphi1->Update();

  /*
  cent_h0->GetPolargram()->SetNdivPolar(4);
  cent_h0->GetPolargram()->SetNdivRadial(4);
  cent_h0->GetPolargram()->SetRangeRadial(0, 0.4); 
  
  minus_h0->GetPolargram()->SetRangeRadial(0, 0.4); 
  plus_h0->GetPolargram()->SetRangeRadial(0, 0.4); 
  */
  
  minus_h0->Draw("OPsame");
  plus_h0->Draw("OPsame");
  
  uphi1->Update();

}

/***
  
  string uphi0 = cent_shms+"uphi.eps";
  TString uphifile = uphi0.c_str();
  c1->SaveAs(eps_dir + uphifile);

  string uphi01 = cent_shms+"uphi.png";
  TString uphifile1 = uphi01.c_str();
  c1->SaveAs(png_dir + uphifile1);

  string rootout0 = cent_shms+"uphi.root";
  TString rootout = rootout0.c_str();
  TFile* rootoutfile = new TFile(pr_dir + rootout, "recreate");
  c1->Write();

}
  */