#include "CompareGrid.hh"
#include "RooAbsPdf.h"
#include "RooRealProxy.h"
#include "TMath.h"
#include "RooRealVar.h"
#include "RooArgSet.h"
#include "RooDataSet.h"
#include "RooCmdArg.h"

Double_t CompareGrid(RooAbsPdf *test_fit_model,RooRealVar *var,Double_t *mu,Double_t *mubest,Double_t *bg,Int_t nbins)
{
    Double_t chisquares[1000];

    Double_t expected =test_fit_model->expectedEvents(RooArgSet(*var));

    RooAbsPdf::GenSpec* genspec = test_fit_model->prepareMultiGen(RooArgSet(*var),RooFit::Verbose(kTRUE));

    for(Int_t k=0;k<nbins;k++){
        bg[k] -= mubest[k];
        }

    for (Int_t l=0;l<1000;l++){

        RooDataSet *data_model;
        data_model = test_fit_model->generate(*genspec);
        Int_t num = data_model->numEntries();
        Double_t alist[num];

        for (Int_t p=0;p<num;p++){
            alist[p] = data_model->get(p)->getRealValue("ee_energy1");
            }

        std::sort(alist,alist+num);
        delete data_model;

        Double_t chisquareprime = 0;
        for (Int_t k=0;k<nbins;k++){
            Int_t count = 0;
            for (Int_t p=0;p<num;p++){
                if (alist[p]>var->getMin()+(var->getMax()-var->getMin())*(k+1)/nbins) p=num;
                if (alist[p]>var->getMin()+(var->getMax()-var->getMin())*k/nbins) count = count+1;
                }
            std::cout<<count<<std::endl;
            chisquareprime += 2*((mu[k]-mubest[k])+count*TMath::Log((mubest[k]+bg[k])/(mu[k]+bg[k])));
            }
        chisquares[l] = chisquareprime;
        }

    delete genspec;

    Int_t index[1000];
    TMath::Sort(1000,chisquares,index,kFALSE);

    return chisquares[index[899]];
}