// ROOT files
#include "TROOT.h"
#include "TTree.h"
#include "TFile.h"
#include <TMath.h>
#include <TLorentzVector.h>
#include <TH1F.h>
#include <TH2.h>
#include <TProfile.h>
#include <TCanvas.h>
#include <TStyle.h>
#include <TF1.h>
#include <TCutG.h>
#include <TRandom.h>
// c++ header files
#include <iostream>
#include <fstream>
#include <stdlib.h>
#include <stdio.h>
#include <string>
#include <iomanip>
#include <math.h>
#include <cassert>
#include <vector>
#include <ctime>

using namespace std;


#include "clas_enable_fp_trap.h"
#include "clas_enable_fp_trap.c"

int main()
{
  //example illustrating how to fit a 2-d histogram of type y=f(x)
  //Author: Rene Brun
  
  clas_enable_fp_trap();
  
  // generate a 2-d histogram using a TCutG
  const Int_t n = 6;
  Float_t x[n] = {0.092,0.83,0.94,0.81,0.12,0.1};
  Float_t y[n] = {0.71,9.4,9,8,0.3,0.71};
  TCutG *cut = new TCutG("cut",n,x,y);
  TH2F *h2 = new TH2F("h2","h2",40,0,1,40,0,10);
  Float_t u,v;
  TRandom *trnd = new TRandom;
  for (Int_t i=0;i<100000;i++) {
    u = trnd->Rndm();
    v = 10*trnd->Rndm();
    if (cut->IsInside(u,v)) h2->Fill(u,v);
  }
  TCanvas *c1 = new TCanvas("c1","show profile",600,900);
  c1->Divide(1,2);
  c1->cd(1);
  h2->Draw();
  c1->cd(2);
   
  //use a TProfile to convert the 2-d to 1-d problem
  TProfile *prof = h2->ProfileX();
  prof->Draw();
  prof->Fit("pol1");
  
  string filename="fit2d.jpg";
  
  c1->Print(filename.c_str());
  
  return 0;
}