#include <string.h>
#include <iostream>
#include <vector>
#include "TFile.h"
#include "TLorentzVector.h"
#include "TH1.h"
#include "TH2.h"
#include "TF1.h"
#include "TRandom3.h"
#include "TVector3.h"
#include <fstream>

using namespace std;

int main(int argc, char **argv) {


  TFile *inFile = new TFile("./AB_V1.root"); //Get the file that has histo
  TH1D *hTheta45=(TH1D*) inFile->Get("hMassKpStar"); //Get the histogram 

  double sMax = hTheta45->GetXaxis()->GetXmax();
  double sMin = hTheta45->GetXaxis()->GetXmin();
  int nBinsS     = hTheta45->GetNbinsX();

  int nBinsX = nBinsS;
  double xMin = sMin;
  double xMax = sMax;

  int nBinsY = nBinsS;
  double yMin = sMin;
  double yMax = sMax;

  TH2D *hXY = new TH2D("hXY","",nBinsX,xMin,xMax,nBinsY,yMin,yMax);

  for (int iBinX = 1; iBinX <= nBinsX; iBinX++) {
    for (int iBinY = 1; iBinY <= nBinsY; iBinY++) {
      double xVal = hXY->GetXaxis()->GetBinCenter(iBinX);
      double yVal = hXY->GetYaxis()->GetBinCenter(iBinY);
      //cout<<"ix,iy: x,y = "<<iBinX<<" , "<<iBinY<<" : "<<xVal<<" , "<<yVal<<endl;
      
      if (iBinX == iBinY) hXY->SetBinContent(iBinX,iBinY,34);

    }
  }

  TFile *outFile = new TFile("./myOutFile.root","RECREATE");
  outFile->cd();
  hXY->Write();
  outFile->Close();
  return 0;


}