#include <TFile.h>
#include <TH1D.h>
#include <TCanvas.h>
#include <TMath.h>
#include <TLegend.h>
#include <TString.h>
#include <iostream>

TH1* hBB1;
TH1* hBB2;
TH1* hBB3;
void setMystyle(TH1D* h1,TH1D* h2,TH1D* h3);
TCanvas * DrawHistogramOnCanvas(TH1D* h1,TH1D* h2,TH1D* h3,TH1D* h4,TH1D* h5,TH1D* h6,TH1D* h7,TH1D* h8,TH1D* h9, const char* canvasName);
void CreateDir(const Char_t* dirName);
TH1D* Pol0functionppbPi(TH1D * obj,Int_t i);
TH1D* Pol0functionppbAS(TH1D * obj,Int_t i);
TH1D* Pol0functionppbNS(TH1D * obj,Int_t i);
TCanvas * Draw3HistogramOnCanvas2pads(TH1D* h1, TH1D* h2, TH1D* h3, TH1D* h4, TH1D* h5, TH1D* h6,TH1D* h21, TH1D* h22, TH1D* h23, TH1D* h24, TH1D* h25, TH1D* h26,TH1D* h31, TH1D* h32, TH1D* h33, TH1D* h34, TH1D* h35, TH1D* h36, const char* canvasName, int cen);
TCanvas * Draw3HistogramNoPads(TH1D* h1, TH1D* h2, TH1D* h3, TH1D* h4, TH1D* h5, TH1D* h6,TH1D* h21, TH1D* h22, TH1D* h23, TH1D* h24, TH1D* h25, TH1D* h26,TH1D* h31, TH1D* h32, TH1D* h33, TH1D* h34, TH1D* h35, TH1D* h36, const char* canvasName, int cen);
void SaveHistogramAsText(TH1D* histogram, const char* canvasName,const Char_t * outFile);
TCanvas *DrawHistogramsForCentrality1(int centrality, TH1D* h1, TH1D* h2, TH1D* h3,TH1D* h4, TH1D* h5, TH1D* h6,const char* canvasName);
void DrawHistogramsForCentrality8(int centrality, TH1D* h1, TH1D* h2, TH1D* h3,TH1D* h4, TH1D* h5, TH1D* h6,const char* canvasName);
TH1D * RebinNewHistograms(TH1D* MCPri,const char* canvasName);

struct FitResults {
            TH1D* hPiPlus_cent0;
            TH1D* hKPlus_cent0;
            TH1D* hProPlus_cent0;

            TH1D* hPiMinus_cent0;
            TH1D* hKMinus_cent0;
            TH1D* hProMinus_cent0;
};
// Function to normalize histograms
TH1D* NormKharra(TH1D* hist, double nEvt) {
    auto const dy = 0.2;

    TH1D* hist_Norm = static_cast<TH1D*>(hist->Clone(Form("%s_Norm", hist->GetName())));
    hist_Norm->Reset();
    
    for (int i = 1; i < hist_Norm->GetNbinsX(); i++) {
        //auto pT = hist->GetBinCenter(i);
        auto pT = hist->GetBinWidth(i);
        auto binCont = hist->GetBinContent(i);
        auto binErr = hist->GetBinError(i);

        if (binCont == 0 || binErr == 0) continue;

        hist_Norm->SetBinContent(i, binCont/pT );

        auto binErr_rel = sqrt(binErr) / binErr * binCont;
       
        hist_Norm->SetBinError(i, binErr_rel);
        
    }

    hist_Norm->Scale(1 / (nEvt *TMath::TwoPi() ), "width"); //*TMath::TwoPi()
    return hist_Norm;
}

void DrawAndNormalizeHistograms(TFile** f_energy, const int numFiles) {
    const int numCentralityBins = 9;  // Assuming there are 9 centrality bins (0 to 8)

    const Char_t * outFile="WS_test_20";
    CreateDir(outFile);

    FitResults results;
    // Arrays to store histograms
    TH1D* hPiPlus_energy[numFiles][numCentralityBins];
    TH1D* hKPlus_energy[numFiles][numCentralityBins];
    TH1D* hProPlus_energy[numFiles][numCentralityBins];
    TH1D* hPiMinus_energy[numFiles][numCentralityBins];
    TH1D* hKMinus_energy[numFiles][numCentralityBins];
    TH1D* hProMinus_energy[numFiles][numCentralityBins];
    // TH1D* hrefmult_energy[numFiles];
    // double nEvt_[numFiles];
    TH1D* hrefmult_energy[numFiles][numCentralityBins];
    double nEvt_[numFiles][numCentralityBins];
    double Int_Pi_[numFiles][numCentralityBins];
    double tInt_Pi_[numFiles][numCentralityBins];

    // Loop over files
    for (int i = 0; i < numFiles; ++i) {
        // Loop over centrality bins
        for (int cen = 0; cen < numCentralityBins; ++cen) {
            hPiPlus_energy[i][cen] = static_cast<TH1D*>(f_energy[i]->Get(Form("hPiPlus_Spectra_Cen%d", cen)));
            hKPlus_energy[i][cen] = static_cast<TH1D*>(f_energy[i]->Get(Form("hKPlus_Spectra_Cen%d", cen)));
            hProPlus_energy[i][cen] = static_cast<TH1D*>(f_energy[i]->Get(Form("hProton_Spectra_Cen%d", cen)));

            hPiMinus_energy[i][cen] = static_cast<TH1D*>(f_energy[i]->Get(Form("hPiMinus_Spectra_Cen%d", cen)));
            hKMinus_energy[i][cen] = static_cast<TH1D*>(f_energy[i]->Get(Form("hKMinus_Spectra_Cen%d", cen)));
            hProMinus_energy[i][cen] = static_cast<TH1D*>(f_energy[i]->Get(Form("hAProton_Spectra_Cen%d", cen)));

            hrefmult_energy[i][cen] = static_cast<TH1D*>(f_energy[i]->Get(Form("hEvnAnalysed_Cen%d", cen)));
            nEvt_[i][cen] = hrefmult_energy[i][cen]->GetEntries();

            std::cout << "Nmult " << i << ": " <<cen<< " centr"<< nEvt_[i][cen] << "\n";
            Int_Pi_[i][cen]=hPiPlus_energy[i][cen]->Integral();
            std::cout << "integraal " << i << ": " <<cen<< " inte"<< Int_Pi_[i][cen] << "\n";

            tInt_Pi_[i][cen] = Int_Pi_[i][cen]/nEvt_[i][cen];
            std::cout << "integraal " << i << ": " <<cen<< " tinte"<< tInt_Pi_[i][cen] << "\n";
        }

        // Loop over centrality bins for normalization and drawing
        for (int cen = 0; cen < numCentralityBins; ++cen) {
            // Normalize histograms
            hPiPlus_energy[i][cen] = NormKharra(hPiPlus_energy[i][cen], nEvt_[i][cen]);
            hKPlus_energy[i][cen] = NormKharra(hKPlus_energy[i][cen], nEvt_[i][cen]);
            hProPlus_energy[i][cen] = NormKharra(hProPlus_energy[i][cen], nEvt_[i][cen]);

            hPiMinus_energy[i][cen] = NormKharra(hPiMinus_energy[i][cen], nEvt_[i][cen]);
            hKMinus_energy[i][cen] = NormKharra(hKMinus_energy[i][cen], nEvt_[i][cen]);
            hProMinus_energy[i][cen] = NormKharra(hProMinus_energy[i][cen], nEvt_[i][cen]);

            const char* canvasName;

             SaveHistogramAsText(hPiPlus_energy[i][cen], Form("PiPlus_Cen%d",cen), outFile);
             SaveHistogramAsText(hPiMinus_energy[i][cen], Form("PiMinus_Cen%d",cen), outFile);
             SaveHistogramAsText(hKPlus_energy[i][cen], Form("KPlus_Cen%d",cen), outFile);
             SaveHistogramAsText(hKMinus_energy[i][cen], Form("KMinus_Cen%d",cen), outFile);
             SaveHistogramAsText(hProPlus_energy[i][cen], Form("ProPlus_Cen%d",cen), outFile);
             SaveHistogramAsText(hProMinus_energy[i][cen], Form("ProMinus_Cen%d",cen), outFile);

            TCanvas * cCent_0 =DrawHistogramsForCentrality1(0, hPiPlus_energy[i][0], hKPlus_energy[i][0], hProPlus_energy[i][0],hPiMinus_energy[i][0], hKMinus_energy[i][0], hProMinus_energy[i][0],Form("AllPlus_Cen%d",0));

             cCent_0->SaveAs(Form("%s/%s.pdf",outFile,"all_random_cent0"));

            TCanvas * cCent_8 =DrawHistogramsForCentrality1(8, hPiPlus_energy[i][7], hKPlus_energy[i][7], hProPlus_energy[i][7],hPiMinus_energy[i][7], hKMinus_energy[i][7], hProMinus_energy[i][7],Form("AllPlus_Cen%d",8));

             cCent_8->SaveAs(Form("%s/%s.pdf",outFile,"all_random_cent8"));

             TCanvas * cCent_4 =DrawHistogramsForCentrality1(4, hPiPlus_energy[i][4], hKPlus_energy[i][4], hProPlus_energy[i][4],hPiMinus_energy[i][4], hKMinus_energy[i][4], hProMinus_energy[i][4],Form("AllPlus_Cen%d",4));

             cCent_4->SaveAs(Form("%s/%s.pdf",outFile,"all_random_cent4"));
                    
        }
        
        TCanvas * cRatio_2pads =Draw3HistogramNoPads(hPiPlus_energy[i][0], hKPlus_energy[i][0], hProPlus_energy[i][0],hPiMinus_energy[i][0], hKMinus_energy[i][0], hProMinus_energy[i][0],hPiPlus_energy[i][4], hKPlus_energy[i][4], hProPlus_energy[i][4],hPiMinus_energy[i][4], hKMinus_energy[i][4], hProMinus_energy[i][4],hPiPlus_energy[i][7], hKPlus_energy[i][7], hProPlus_energy[i][7],hPiMinus_energy[i][7], hKMinus_energy[i][7], hProMinus_energy[i][7],Form("twopads_ratio_Cen%d_3",0),0);
    }

    // return results;
}

int main() {
    const int numFiles = 1;  // Change this to the actual number of files you have
    gStyle->SetOptStat(0);
    
    // Open the ROOT file
    TFile* f_energy[numFiles];
    // f_energy[0] = new TFile("/home/ahmad/Downloads/ResultOut_0allPbPb5p02.root", "READ");

    f_energy[0] = new TFile("/home/ahmad/Downloads/ResultOut_0allPbPb5p02.root", "READ");  

    // Call the function
      DrawAndNormalizeHistograms(f_energy, numFiles);
      
    return 0;
}

TCanvas * DrawHistogramOnCanvas(TH1D* h1,TH1D* h2,TH1D* h3,TH1D* h4,TH1D* h5,TH1D* h6,TH1D* h7,TH1D* h8,TH1D* h9, const char* canvasName) {
    TCanvas* cBB1 = new TCanvas(canvasName, "", 800, 600);
    cBB1->SetLogy(1);
     hBB1 = cBB1->DrawFrame(0,0.001,2.6,10000);
     hBB1->SetXTitle("#it{p}_{T} (GeV/#it{c})");
    
     hBB1->SetYTitle("#frac{1}{2#pi#it{p}_{T}} #frac{d^{2}N}{d#it{p}_{T}dy}(GeV/#it{c})^{-2}");

    h1->Draw("same");    //70-100
     h2->Draw("same");    //60-70
     h3->Draw("same");    //50-60leg->AddEntry
    h4->Draw("same");      //40-50
    
     h5->Draw("same");    //30-40 
     h6->Draw("same");    //20-30
     h7->Draw("same");    //10-20
    h8->Draw("same");    //5-10
     h9->Draw("same");     //0-5

     h1->SetLineColor(kBlue);
    h1->SetMarkerColor(kBlue);
    h1->SetLineWidth(1);
    h1->SetMarkerSize(1.5);
    h1->SetMarkerStyle(20);

     h2->SetLineColor(kRed);
    h2->SetMarkerColor(kRed);
    h2->SetLineWidth(1);
    h2->SetMarkerSize(1.5);
    h2->SetMarkerStyle(20);

     h3->SetLineColor(kCyan+2);
    h3->SetMarkerColor(kCyan+2);
    h3->SetLineWidth(1);
    h3->SetMarkerSize(1.5);
    h3->SetMarkerStyle(20);


    h4->SetLineColor(kBlue+4);
    h4->SetMarkerColor(kBlue+4);
    h4->SetLineWidth(1);
    h4->SetMarkerSize(1.5);
    h4->SetMarkerStyle(20);

     h5->SetLineColor(kRed+2);
    h5->SetMarkerColor(kRed+2);
    h5->SetLineWidth(1);
    h5->SetMarkerSize(1.5);
    h5->SetMarkerStyle(20);

     h6->SetLineColor(kCyan+3);
    h6->SetMarkerColor(kCyan+3);
    h6->SetLineWidth(1);
    h6->SetMarkerSize(1.5);
    h6->SetMarkerStyle(20);


    h7->SetLineColor(kBlue+3);
    h7->SetMarkerColor(kBlue+3);
    h7->SetLineWidth(1);
    h7->SetMarkerSize(1.5);
    h7->SetMarkerStyle(20);

     h8->SetLineColor(kRed+3);
    h8->SetMarkerColor(kRed+3);
    h8->SetLineWidth(1);
    h8->SetMarkerSize(1.5);
    h8->SetMarkerStyle(20);

     h9->SetLineColor(kCyan-1);
    h9->SetMarkerColor(kCyan-1);
    h9->SetLineWidth(1);
    h9->SetMarkerSize(1.5);
    h9->SetMarkerStyle(20);



    cBB1->Draw();
//TLegend* legendPiPlus = new TLegend(0.1478697,0.12,0.3483709,0.4608696,NULL,"brNDC");
        TLegend* legendPiPlus = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legendPiPlus->SetFillStyle(0);
        legendPiPlus->SetTextSize(0.045);
        legendPiPlus->SetBorderSize(0);
        legendPiPlus->SetLineColor(0);
        legendPiPlus->SetFillColor(0);
        legendPiPlus->SetTextFont(42);

        legendPiPlus->SetHeader("Pb-Pb, #sqrt{#it{s}_{NN}} = 5.02 TeV");
       // if(canvasName=="Canvas_PiPlus_Cen0"){

        legendPiPlus->AddEntry(h1, "0-5 %", "pl");

        legendPiPlus->AddEntry(h2, "5-10 %", "pl");
        legendPiPlus->AddEntry(h3, "10-20 %", "pl");
        legendPiPlus->AddEntry(h4, "20-30 %", "pl");

        legendPiPlus->AddEntry(h5, "30-40 %", "pl");

        legendPiPlus->AddEntry(h6, "40-50 %", "pl");
        legendPiPlus->AddEntry(h7, "50-60 %", "pl");
        legendPiPlus->AddEntry(h8, "60-80 %", "pl");

        legendPiPlus->AddEntry(h9, "80-100 %", "pl");

        

        legendPiPlus->Draw();
    return cBB1;
}
TH1* scale(TH1*hBB2)
{
    
  
   hBB2->GetYaxis()->SetTitleSize(0.035);
    hBB2->GetYaxis()->SetTitleFont(42);
    hBB2->GetYaxis()->SetTitleOffset(1.65);
    // Absolute font size in pixel (precision 3)
    
    hBB2->GetYaxis()->SetLabelSize(0.03);
    hBB2->GetYaxis()->SetLabelFont(42);
    hBB2->GetYaxis()->SetLabelOffset(0.010);
     
    hBB2->GetXaxis()->SetLabelOffset(0.005);
    hBB2->GetXaxis()->SetLabelSize(0.03);
    hBB2->GetXaxis()->SetLabelFont(42);

    // hBB2->GetXaxis()->CenterTitle();
    
    hBB2->GetXaxis()->SetTitleSize(0.035);
    hBB2->GetXaxis()->SetTitleFont(42);
    hBB2->GetXaxis()->SetTitleOffset(0.91);
    
    return hBB2;
}
void setMystyle(TH1D* h1,TH1D* h2,TH1D* h3){

    h1->SetLineColor(kBlue);
    h1->SetMarkerColor(kBlue);
    h1->SetLineWidth(1);
    h1->SetMarkerSize(1.5);
    h1->SetMarkerStyle(20);
    //h1->GetXaxis()->SetRangeUser(0.512,39.98);
    
    h2->SetLineColor(kRed);
    h2->SetMarkerColor(kRed);
    h2->SetLineWidth(1);
    h2->SetMarkerSize(1.5);
    h2->SetMarkerStyle(24);
    
    h3->SetLineColor(kCyan+2);
    h3->SetMarkerColor(kCyan+2);
    h3->SetLineWidth(1);
    h3->SetMarkerSize(1.5);
    h3->SetMarkerStyle(21);

}

void SaveHistogramAsText(TH1D* histogram, const char* canvasName,const Char_t * outFile) {
    
    const char* directoryPath;
   // mkdir(directoryPath);
    
    TString fileName = Form("%s/%s.txt",outFile,canvasName);
    std::ofstream outputFile(fileName.Data());

    TString histogramName = histogram->GetName();

    // Print histogram name on terminal
    std::cout << "Saving histogram: " << histogramName.Data() << std::endl;


    outputFile << "HistogramName: " << histogramName << "\n";
   // outputFile << "Centrality: " << centrality << "\n";

    outputFile << "BinCenter\tBinContent\tBinError\n";
    for (int i = 1; i <= histogram->GetNbinsX(); ++i) {
        outputFile << histogram->GetBinCenter(i) << "\t"
                    << histogram->GetBinContent(i) << "\t"
                    << histogram->GetBinError(i) << "\n";
    }

    outputFile.close();
}

TH1D* Pol0functionppbPi(TH1D * obj,Int_t i)
{
    
    obj->Fit("expo", "0", "o", 1.4, 2.0);
    TH1D * nobj = (TH1D*)obj->Clone("nobj");

    for (Int_t bin=obj->GetXaxis()->FindBin(1.4); bin <= obj->GetNbinsX(); bin++){

      double bincount = nobj->GetBinError(bin);
      //std::cout << "bin error before : " << bincount << std::endl;
    
      obj->SetBinContent(bin, obj->GetFunction("expo")->Eval(obj->GetXaxis()->GetBinCenter(bin)));
      double bincount_obj = obj->GetBinError(bin);
      //std::cout << "bin error after: " << bincount_obj << std::endl;
      
      obj->SetBinError(bin, bincount/500);}
    
    return obj;
}

TH1D* Pol0functionppbAS(TH1D * obj,Int_t i)
{
    
   obj->Fit("expo", "0", "o", 0.6, 6.4);
    TH1D * nobj = (TH1D*)obj->Clone("nobj");

    for (Int_t bin=obj->GetXaxis()->FindBin(0.5); bin <= obj->GetXaxis()->FindBin(2.6); bin++){

      double bincount = nobj->GetBinError(bin);
    
      obj->SetBinContent(bin, obj->GetFunction("expo")->Eval(obj->GetXaxis()->GetBinCenter(bin)));
      double bincount_obj = obj->GetBinError(bin);
      
      obj->SetBinError(bin, bincount/500); }
    
    return obj;
}
TH1D* Pol0functionppbNS(TH1D * obj,Int_t i)
{
    
   obj->Fit("expo", "0", "o", 0.4, 6.5);
    TH1D * nobj = (TH1D*)obj->Clone("nobj");

    for (Int_t bin=obj->GetXaxis()->FindBin(0.7); bin <= obj->GetXaxis()->FindBin(2.6); bin++){

      double bincount = nobj->GetBinError(bin);
    
      obj->SetBinContent(bin, obj->GetFunction("expo")->Eval(obj->GetXaxis()->GetBinCenter(bin)));
      double bincount_obj = obj->GetBinError(bin);
      
      obj->SetBinError(bin, bincount/500); }
    
    return obj;
}
TCanvas * DrawHistogramsForCentrality1(int centrality, TH1D* h11, TH1D* h22, TH1D* h33,TH1D* h44, TH1D* h55, TH1D* h66,const char* canvasName) {
    // Create a canvas
    TCanvas* cBB1 = new TCanvas(canvasName, "", 800, 600);
    cBB1->SetLogy(1);
    cBB1->SetLeftMargin(0.1378446);
        hBB2 = cBB1->DrawFrame(0,10e-4,3,10000);
     hBB2->SetXTitle("#it{p}_{T} (GeV/#it{c})");
    
     hBB2->SetYTitle("#frac{1}{#it{N}_{ev}2#pi#it{p}_{T}} #frac{d^{2}N}{d#it{p}_{T}dy}(GeV/#it{c})^{-2}"); //2#pi

    hBB2=scale(hBB2);
  
    h11->Draw("same");
    
    h44->Draw("same");
    h11->SetMarkerSize(1.5); 
    h11->SetLineColor(kBlue);
    h11->SetMarkerColor(kBlue);
    h11->SetMarkerStyle(20);  

    h44->SetLineColor(kBlue);
    h44->SetMarkerColor(kBlue);
    h44->SetMarkerSize(1.5);
    h44->SetMarkerStyle(24);

   // hKPlus->SetLineColor(kBlue);
    h22->Draw("same");

    h55->Draw("same");

    h22->SetMarkerSize(1.5);  
    h22->SetLineColor(kRed);
    h22->SetMarkerColor(kRed);
    h22->SetMarkerStyle(20); 

    h55->SetLineColor(kRed);
    h55->SetMarkerColor(kRed);
    h55->SetMarkerSize(1.5);
    h55->SetMarkerStyle(24);

   // hProPlus->SetLineColor(kCyan);
    h33->Draw("same");
    h66->Draw("same");

    h33->SetMarkerSize(1.5);   

    h33->SetLineColor(kCyan+2);
    h33->SetMarkerColor(kCyan+2);

    h33->SetMarkerStyle(20);

    h66->SetLineColor(kCyan+2);
    h66->SetMarkerColor(kCyan+2);
    h66->SetMarkerSize(1.5);
    h66->SetMarkerStyle(24);
   
   if (centrality == 0) {
    // Adjusted position to match the image layout
    TLegend* legend = new TLegend(0.50, 0.75, 0.72, 0.85, NULL, "brNDC");

    legend->SetFillStyle(0);
    legend->SetTextSize(0.035);  // Slightly reduced for clarity
    legend->SetBorderSize(0);
    legend->SetLineColor(0);
    legend->SetFillColor(0);
    legend->SetTextFont(42);
    legend->SetNColumns(1); // Ensures a vertical list

    legend->SetHeader("#it{Pb-Pb}, #sqrt{#it{s}_{NN}} = 5.02 TeV, 0#minus5%");
    legend->AddEntry(h11, "#pi^{+} + #pi^{-}", "pl");
    legend->AddEntry(h22, "K^{+} + K^{-}", "pl");
    legend->AddEntry(h33, "p^{+} + p^{-}", "pl");
    
    legend->Draw();
}

    else if (centrality == 4) {
    // Carefully adjusted legend position
    TLegend* legend = new TLegend(0.50, 0.75, 0.72, 0.85, NULL, "brNDC");

    legend->SetFillStyle(0);
    legend->SetTextSize(0.035);  // Keeping original size for readability
    legend->SetBorderSize(0);
    legend->SetLineColor(0);
    legend->SetFillColor(0);
    legend->SetTextFont(42);
    legend->SetNColumns(1);  // Keeping two-column format

    legend->SetHeader("#it{Pb-Pb}, #sqrt{#it{s}_{NN}} = 5.02 TeV, 30#minus40%");
    legend->AddEntry(h11, "#pi^{+} + #pi^{-}", "pl");
    legend->AddEntry(h22, "K^{+} + K^{-}", "pl");
    legend->AddEntry(h33, "p^{+} + p^{-}", "pl");

    legend->Draw();
}


    else if (centrality == 8) {
    // Adjusted legend position for precise alignment
    TLegend* legend = new TLegend(0.50, 0.75, 0.72, 0.85, NULL, "brNDC");

    legend->SetFillStyle(0);
    legend->SetTextSize(0.045);  // Keeping the original text size for clarity
    legend->SetBorderSize(0);
    legend->SetLineColor(0);
    legend->SetFillColor(0);
    legend->SetTextFont(42);
    legend->SetNColumns(2);  // Keeping two-column format for alignment

    legend->SetHeader("#it{Pb-Pb}, #sqrt{#it{s}_{NN}} = 5.02 TeV, 60#minus80%");
    legend->AddEntry(h11, "#pi^{+} + #pi^{-}", "pl");
    legend->AddEntry(h22, "K^{+} + K^{-}", "pl");
    legend->AddEntry(h33, "p^{+} + p^{-}", "pl");

    legend->Draw();
}

//TLegend* legend2 = new TLegend(0.1716792,0.7426087,0.4536341,0.9426087,NULL,"brNDC");
     TLegend* legend2 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");

        legend2->SetFillStyle(0);
        legend2->SetTextSize(0.045);
        legend2->SetBorderSize(0);
        legend2->SetLineColor(0);
        legend2->SetFillColor(0);
        legend2->SetTextFont(42);
        legend2->SetNColumns(2);

        legend2->Draw();
    return cBB1;
}
void DrawHistogramsForCentrality8(int centrality, TH1D* h1, TH1D* h2, TH1D* h3,TH1D* h4, TH1D* h5, TH1D* h6,const char* canvasName) {
    // Create a canvas
    TCanvas* cBB1 = new TCanvas(canvasName, "", 800, 600);
    cBB1->SetLogy(1);
        hBB2 = cBB1->DrawFrame(0,0.0001,2.6,10000);
     hBB2->SetXTitle("#it{p}_{T} (GeV/#it{c})");
    
     hBB2->SetYTitle("#frac{1}{#it{N}_{ev}2#pi#it{p}_{T}} #frac{d^{2}N}{d#it{p}_{T}dy}(GeV/#it{c})^{-2}");

    h1->Draw("same");
    
    h4->Draw("same");
    h1->SetMarkerSize(1.5);   

    h4->SetLineColor(kBlue);
    h4->SetMarkerColor(kBlue);
    h4->SetMarkerSize(1.5);
    h4->SetMarkerStyle(24);

   // hKPlus->SetLineColor(kBlue);
    h2->Draw("same");

    h5->Draw("same");

    h2->SetMarkerSize(1.5);   

    h5->SetLineColor(kRed);
    h5->SetMarkerColor(kRed);
    h5->SetMarkerSize(1.5);
    h5->SetMarkerStyle(24);

   // hProPlus->SetLineColor(kCyan);
    h3->Draw("same");
    h6->Draw("same");

    h3->SetMarkerSize(1.5);   

    h6->SetLineColor(kCyan+2);
    h6->SetMarkerColor(kCyan+2);
    h6->SetMarkerSize(1.5);
    h6->SetMarkerStyle(24);

    // Add a legend
//TLegend* legend = new TLegend(0.4749373,0.2173913,0.7568922,0.4173913,NULL,"brNDC");
        TLegend* legend = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend->SetFillStyle(0);
        legend->SetTextSize(0.045);
        legend->SetBorderSize(0);
        legend->SetLineColor(0);
        legend->SetFillColor(0);
        legend->SetTextFont(42);
        legend->SetNColumns(2);
   if(centrality ==0){
    legend->SetHeader("Pb-Pb, #sqrt{#it{s}_{NN}} = 5.02 TeV, 0-5%");
    legend->AddEntry(h1, "#pi^{+} + #pi^{-}", "pl");
    //legend->AddEntry(h4, "#pi^{-}", "pl");
    
    
    legend->AddEntry(h2, "K^{+} + K^{-}", "pl");
    //legend->AddEntry(h5, "K^{-}", "pl");

    legend->AddEntry(h3, "p^{+} + p^{-}", "pl");
    //legend->AddEntry(h6, "p^{-}", "pl");
    
    
    }
    else if(centrality ==8){
    legend->SetHeader("Pb-Pb, #sqrt{#it{s}_{NN}} = 5.02 TeV, 60-80%");
    legend->AddEntry(h1, "#pi^{+} + #pi^{-}", "pl");
    //legend->AddEntry(h4, "#pi^{-}", "pl");
    
    
    legend->AddEntry(h2, "K^{+} + K^{-}", "pl");
    //legend->AddEntry(h5, "K^{-}", "pl");

    legend->AddEntry(h3, "p^{+} + p^{-}", "pl");
    //legend->AddEntry(h6, "p^{-}", "pl");
    }
    
    legend->Draw();

}
void CreateDir(const Char_t* dirName)
{
  TString pwd(gSystem->pwd());
  gSystem->cd(pwd.Data());
  
  if(gSystem->cd(dirName)) {
    gSystem->cd(pwd.Data());
  } else {
    gSystem->mkdir(dirName, kTRUE); // kTRUE means recursive
  }
}

TCanvas * Draw3HistogramOnCanvas2pads(TH1D* h1, TH1D* h2, TH1D* h3, TH1D* h4, TH1D* h5, TH1D* h6,TH1D* h21, TH1D* h22, TH1D* h23, TH1D* h24, TH1D* h25, TH1D* h26,TH1D* h31, TH1D* h32, TH1D* h33, TH1D* h34, TH1D* h35, TH1D* h36, const char* canvasName, int cen) {
    TCanvas* cBB1 = new TCanvas(canvasName, "", 650, 850);
    // cBB1->SetMargin(0, 0, 0, 0);
    cBB1->SetLeftMargin(0);

     TPad* pad1 = new TPad("pad1", "pad1", 0.01, 0.01,1.0, 0.367);
   // pad1->SetMargin(0, 0, 0, 0);
    pad1->SetLeftMargin(0.12);
    pad1->SetRightMargin(0.1);
    pad1->SetBottomMargin(0.14);
    pad1->SetTopMargin(0);
    pad1->SetLogy(1);
    pad1->Draw();
    pad1->cd();
    hBB1 = pad1->DrawFrame(0,1e-3,3.0,9e3);
    hBB1=scale(hBB1);
    hBB1->SetXTitle("#it{p}_{T} (GeV/#it{c})");

    h31->Draw("same");
    h32->Draw("same");
    h33->Draw("same");
    h34->Draw("same");
    h35->Draw("same");
    h36->Draw("same");
//TLegend* legend1 = new TLegend(0.2767178,0.785335,0.4746851,0.985936,NULL,"brNDC");
    TLegend* legend1 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend1->SetFillStyle(0);
        legend1->SetTextSize(0.045);
        legend1->SetBorderSize(0);
        legend1->SetLineColor(0);
        legend1->SetFillColor(0);
        legend1->SetTextFont(42);
        legend1->SetNColumns(2);
        legend1->AddEntry(h2, "c)", "");
    legend1->AddEntry(h1, "60#minus80%", "");
   // legend1->AddEntry(h2, "Label2", "l");
    legend1->Draw();

    // Define Pad2 (Middle)
    cBB1->cd();

    TPad* pad2 = new TPad("pad2", "pad2",0.01, 0.367,1.0, 0.665);
   // pad2->SetMargin(0, 0, 0, 0);
    pad2->SetLeftMargin(0.12);
    pad2->SetRightMargin(0.1);
    pad2->SetBottomMargin(0);
    pad2->SetTopMargin(0);
    pad2->SetLogy(1);
    pad2->Draw();
    pad2->cd();
    hBB2 = pad2->DrawFrame(0.01,7e-2,2.999,9e3);
    hBB2=scale(hBB2);

    h21->Draw("same");
    h22->Draw("same");
    h23->Draw("same");
    h24->Draw("same");
    h25->Draw("same");
    h26->Draw("same");

    TLegend* legend2 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend2->SetFillStyle(0);
        legend2->SetTextSize(0.045);
        legend2->SetBorderSize(0);
        legend2->SetLineColor(0);
        legend2->SetFillColor(0);
        legend2->SetTextFont(42);
        legend2->SetNColumns(2);
    legend2->AddEntry(h2, "b)", "");    
    legend2->AddEntry(h1, "30#minus40%", "");
   // legend1->AddEntry(h2, "Label2", "l");
    legend2->Draw();

    // Define Pad3 (upper)
    cBB1->cd();

    TPad* pad3 = new TPad("pad3", "pad3",0.01, 0.665, 1.0, 0.983);
   // pad3->SetMargin(0, 0, 0, 0);
    pad3->SetLeftMargin(0.12);
    pad3->SetRightMargin(0.1);
    pad3->SetBottomMargin(0);
    pad3->SetTopMargin(0);
    pad3->SetLogy(1);
    pad3->Draw();
    pad3->cd();
    hBB3 = pad3->DrawFrame(0.01,7e-2,2.991,9e3);
    hBB3=scale(hBB3);
     hBB3->SetXTitle("#it{p}_{T} (GeV/#it{c})");
     hBB3->SetYTitle("#frac{1}{#it{N}_{ev}2#pi#it{p}_{T}} #frac{d^{2}N}{d#it{p}_{T}dy}(GeV/#it{c})^{-2}  ");

    h1->Draw("same");
    h2->Draw("same");
    h3->Draw("same");
    h4->Draw("same");
    h5->Draw("same");
    h6->Draw("same");
    
    TLegend* legend3 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend3->SetFillStyle(0);
        legend3->SetTextSize(0.045);
        legend3->SetBorderSize(0);
        legend3->SetLineColor(0);
        legend3->SetFillColor(0);
        legend3->SetTextFont(42);
        legend3->SetNColumns(2);
  
    legend3->SetHeader("Pb-Pb, #sqrt{#it{s}_{NN}} = 5.02 TeV, |y| < 0.5");
    legend3->AddEntry(h1, "#pi^{+} + #pi^{-}", "pl");
    //legend3->AddEntry(h4, "#pi^{-}", "pl");
    
    
    legend3->AddEntry(h2, "K^{+} + K^{-}", "pl");
    //legend3->AddEntry(h5, "K^{-}", "pl");

    legend3->AddEntry(h3, "p^{+} + p^{-}", "pl");
    //legend3->AddEntry(h6, "p^{-}", "pl");

    legend3->Draw();

    TLegend* legend4 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend4->SetFillStyle(0);
        legend4->SetTextSize(0.045);
        legend4->SetBorderSize(0);
        legend4->SetLineColor(0);
        legend4->SetFillColor(0);
        legend4->SetTextFont(42);
        legend4->SetNColumns(2);
    legend4->AddEntry(h2, "a)", "");    
    legend4->AddEntry(h1, "0#minus5%", "");
   // legend1->AddEntry(h2, "Label2", "l");
    legend4->Draw();

    return cBB1;
}

TH1D * RebinNewHistograms(TH1D* MCPri,const char* canvasName){
    
    
    TH1D* RawPi=(TH1D*)MCPri->Clone(Form("%s", canvasName));

    double dca_bins[100];
        for (int i = 0; i <= 30; i++)
          dca_bins[i] = 0.1 * i;
        for (int i = 1; i <= 30; i++)
          dca_bins[30 + i] = 3 + 0.12 * i;
        for (int i = 1; i <= 40; i++)
          dca_bins[60 + i] = 6.6 + 0.15 * i;
        int nbins_dca = sizeof(dca_bins) / sizeof(*dca_bins) - 1;

        TH1D *hNewbin = new TH1D("hNewbin", "pT; pT ; Events",  nbins_dca, dca_bins);
        
    for (int j=1 ; j <=RawPi->GetNbinsX();j++)
     {
        double x = MCPri->GetBinContent(j);
       
       // double z = MCMaterial->GetBinContent(j);
         double xerror = MCPri->GetBinError(j);
                
        hNewbin->SetBinContent(j,x);
        hNewbin->SetBinError(j,xerror);
     }
        
    return hNewbin;
}

TCanvas * Draw3HistogramNoPads(TH1D* h1, TH1D* h2, TH1D* h3, TH1D* h4, TH1D* h5, TH1D* h6,TH1D* h21, TH1D* h22, TH1D* h23, TH1D* h24, TH1D* h25, TH1D* h26,TH1D* h31, TH1D* h32, TH1D* h33, TH1D* h34, TH1D* h35, TH1D* h36, const char* canvasName, int cen) {
    TCanvas* cBB1 = new TCanvas(canvasName, "", 600, 800);
    cBB1->SetLogy(1);

    cBB1->SetRightMargin(0.01003344);  
    cBB1->SetTopMargin(0.01003344);  
    cBB1->SetBottomMargin(0.1068744);
    cBB1->SetLeftMargin(0.1688963);

    hBB1 = cBB1->DrawFrame(0,3e-6,3.199,9e7);
     hBB1=scale(hBB1);
    hBB1->SetXTitle("#it{p}_{T} (GeV/#it{c})");
    // hBB1->SetYTitle("Ratio");
     hBB1->SetYTitle("#frac{1}{2#pi#it{p}_{T}} #frac{d^{2}#it{N}}{d#it{p}_{T}d#it{y}} (GeV/#it{c})^{-2}  ");

    h31->Draw("same");
    h32->Draw("same");
    h33->Draw("same");
    h34->Draw("same");
    h35->Draw("same");
    h36->Draw("same");

     h31->GetXaxis()->SetRangeUser(0,3);
     h32->GetXaxis()->SetRangeUser(0,3);
     h33->GetXaxis()->SetRangeUser(0,3);
     h34->GetXaxis()->SetRangeUser(0,3);
     h35->GetXaxis()->SetRangeUser(0,3);
     h36->GetXaxis()->SetRangeUser(0,3);

     h31->Scale(0.001);
    h32->Scale(0.001);
    h33->Scale(0.001);
    h34->Scale(0.001);
    h35->Scale(0.001);
    h36->Scale(0.001);

    TLegend* legend1 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend1->SetFillStyle(0);
        legend1->SetTextSize(0.035);
        legend1->SetBorderSize(0);
        legend1->SetLineColor(0);
        legend1->SetFillColor(0);
        legend1->SetTextFont(42);
      
    legend1->AddEntry(h1, "60#minus80% (#times 10^{-3})", "");
   // legend1->AddEntry(h2, "Label2", "l");
     legend1->Draw();

    h21->Draw("same");
    h22->Draw("same");
    h23->Draw("same");
    h24->Draw("same");
    h25->Draw("same");
    h26->Draw("same");

    h21->GetXaxis()->SetRangeUser(0,3);
     h22->GetXaxis()->SetRangeUser(0,3);
     h23->GetXaxis()->SetRangeUser(0,3);
     h24->GetXaxis()->SetRangeUser(0,3);
     h25->GetXaxis()->SetRangeUser(0,3);
     h26->GetXaxis()->SetRangeUser(0,3);

    TLegend* legend2 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend2->SetFillStyle(0);
        legend2->SetTextSize(0.035);
        legend2->SetBorderSize(0);
        legend2->SetLineColor(0);
        legend2->SetFillColor(0);
        legend2->SetTextFont(42);
         
    legend2->AddEntry(h1, "30#minus40%", "");
   // legend1->AddEntry(h2, "Label2", "l");
     legend2->Draw();

    h1->Draw("same");
    h2->Draw("same");
    h3->Draw("same");
    h4->Draw("same");
    h5->Draw("same");
    h6->Draw("same");

     h1->Scale(1000);
    h2->Scale(1000);
    h3->Scale(1000);
    h4->Scale(1000);
    h5->Scale(1000);
    h6->Scale(1000);

    h1->GetXaxis()->SetRangeUser(0,3);
     h2->GetXaxis()->SetRangeUser(0,3);
     h3->GetXaxis()->SetRangeUser(0,3);
     h4->GetXaxis()->SetRangeUser(0,3);
     h5->GetXaxis()->SetRangeUser(0,3);
     h6->GetXaxis()->SetRangeUser(0,3);
    
    TLegend* legend3 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend3->SetFillStyle(0);
        legend3->SetTextSize(0.035);
        legend3->SetBorderSize(0);
        legend3->SetLineColor(0);
        legend3->SetFillColor(0);
        legend3->SetTextFont(42);
        legend3->SetNColumns(2);
  
    legend3->SetHeader("#it{Pb-Pb}, #sqrt{#it{s}_{NN}} = 5.02 TeV, #left|#it{y}#right| < 0.5");  //#left|#eta#right|
    legend3->AddEntry(h1, "#pi^{+} + #pi^{-}", "pl");
    //legend3->AddEntry(h4, "#pi^{-}", "pl");
        
    legend3->AddEntry(h2, "K^{+} + K^{-}", "pl");
    //legend3->AddEntry(h5, "K^{-}", "pl");

    legend3->AddEntry(h3, "p^{+} + p^{-}", "pl");
    //legend3->AddEntry(h6, "p^{-}", "pl");

     legend3->Draw();

    TLegend* legend4 = new TLegend(0.50, 0.75, 0.72, 0.85,NULL,"brNDC");
        legend4->SetFillStyle(0);
        legend4->SetTextSize(0.035);
        legend4->SetBorderSize(0);
        legend4->SetLineColor(0);
        legend4->SetFillColor(0);
        legend4->SetTextFont(42);
        
    legend4->AddEntry(h1, "0#minus5% (#times 10^{3})", "");
   // legend1->AddEntry(h2, "Label2", "l");
     legend4->Draw();

    return cBB1;
}