//#ifndef __CINT__
#include "RooGlobalFunc.h"
//#endif
#include "RooRealVar.h"
#include "RooArgList.h"
#include "RooFormulaVar.h"
#include "RooDataSet.h"
#include "RooGaussian.h"
#include "RooChebychev.h"
#include "RooExponential.h"
#include "RooBifurGauss.h"
#include "RooAddModel.h"
#include "RooProdPdf.h"
#include "TCanvas.h"
#include "RooPlot.h"
#include "RooHist.h"
#include "RooCBShape.h"
#include "RooPolynomial.h"
#include "RooBinning.h"
#include "TH1.h"
#include "TH2.h"
#include "RooAddPdf.h"
#include "RooProdPdf.h"
#include "RooFitResult.h"
#include "RooGenericPdf.h"
#include "RooLandau.h"
#include "TChain.h"
#include<cmath>
#include "TCanvas.h"
#include "TAxis.h"
#include "RooPlot.h"
#include "RooCategory.h"
#include "RooSuperCategory.h"
#include "RooSimultaneous.h"
#include "RooNLLVar.h"
#include "TLorentzVector.h"
#include "TVector3.h"


using namespace RooFit ;
using namespace std;

void ksks_fit_trial(){

   RooRealVar deltaM("deltam","#DeltaM", 0.14, 0.159);
  
    
  TFile *f = new TFile("forfit_proc11_bucket9to12_bcs.root");    
  
  TTree *tree = (TTree*)f->Get("after_bcs");
  Double_t  deltaM_after_bcs;
  
  tree->SetBranchAddress("deltaM_after_bcs",&deltaM_after_bcs);

  RooDataSet data("data","data", tree, RooArgSet(deltaM));
  cout << "Entries = " << tree->GetEntries() << endl;
  
  //signal
  RooRealVar mean('mean', 'mean_of_the_dist',0.1454, 0.14, 0.159, 'GeV/c^{2}');
  RooRealVar sigma('sigma', 'sigma_of_the_dist', 0.0003, 0, 10);
  RooGaussian sig('sig', 'signal component',deltaM, mean, sigma);

  //background
  RooRealVar alpha('alpha', 'co_efficient',20,-100,100);
  RooFormulaVar diff("diff","deltaM_after_bcs-mpi", r.RooArgList(deltaM,mpi));
  RooFormulaVar first_part("first_part", "pow(diff,1/2)", r.RooArgList(diff));
  RooFormulaVar second_part("second_part", "alpha*pow(diff,3/2)", r.RooArgList(alpha, diff));
  RooGenericPdf bkg("bkg","first_part + second_part",r.RooArgList(first_part, second_part));
  RooRealVar n_sig('n_sig', 'n_{s}', 2000, 0, 10000);
  RooRealVar n_bkg('n_bkg', 'n_{b}', 100, 0, 10000);
  RooAddPdf pdf('pdf', 'two component model',RooArgList(sig, bkg), RooArgList(n_sig, n_bkg));  
  RooFitResult *fitresult = pdf.fitTo(data, Extended(true));
  TCanvas *c1 =new TCanvas ("c1", "", 600, 500);
  RooPlot* deltaMplot = deltaM.frame();
  data.plotOn(deltaMplot);
  pdf.plotOn(deltaMplot);
  cout << "chi^2 = " << deltaMplot->chiSquare() << endl;

  pdf.plotOn(deltaMplot,Components(RooArgSet(sig)),LineColor(kRed),LineStyle(kDashed));
  pdf.plotOn(deltaMplot,Components(RooArgSet(bkg)),LineColor(kBlue),LineStyle(kDashed));
  deltaMplot->Draw();
  c1->SetFrameFillColor(kWhite);
  c1->Update();
  
}