#define MakeDataset_cxx
#include "MakeDataset.h"
#include <TH2.h>
#include <TStyle.h>
#include <TCanvas.h>


using namespace RooFit;
using namespace RooStats;

void MakeDataset::Loop()
{
   
   //p1_P,p2_p,p3_P
   RooRealVar r_m("D_MM", "M(pipipi)", 1.91200000000000000e+003, 2.02800000000000000e+003,"MeV/c^{2}"); //ok
  /* RooRealVar r_VTXCHI2("D_ENDVERTEX_CHI2","VTXCHI2", 0.,3.25000000000000000e+001, ""); //ok
   RooRealVar r_IPCHI2("D_IPCHI2_OWNPV","IPCHI2", 0.,1.32000000000000010e+001, "");//ok
   RooRealVar r_IP("D_IP_OWNPV","IP",0.0,3.76000000000000020e+000);//ok
   RooRealVar r_FDCHI2("D_FDCHI2_OWNPV","FDCHI2",0,1.06000000000000000e+007);//ok
   RooRealVar r_FD("D_FD_OWNPV","FD", 0.,6.70000000000000000e+002, ""); //ok
   RooRealVar r_DIRA("D_DIRA_OWNPV","DIRA", 9.83299999999999840e-001,1.00070000000000010e+000, ""); //ok
   RooRealVar r_BPVTRGPOINTING("D_BPVTRGPOINTING","POINTING",0,0.6);//ok
   RooRealVar r_tCM("tCM","tCM", 0,125, "");//ok
   RooRealVar r_PT("D_PT","PT", 0.,5.60000000000000000e+004, ""); //ok
   RooRealVar r_P("D_P","P",0,3.15000000000000000e+005);//ok
   RooRealVar r_PTsum("PTsum","PTsum", 0.,5.60000000000000000e+004, ""); //ok
   RooRealVar r_logIP("logIP","logIP", 1.00000000000000000e+000,4.60000000000000000e+001, ""); //ok
   RooRealVar r_p1_IPCHI2("p1_IPCHI2_OWNPV","p1_IPCHI2", 0.,4.30000000000000000e+005, "");//ok
   RooRealVar r_p2_IPCHI2("p2_IPCHI2_OWNPV","p2_IPCHI2", 0.,4.00000000000000000e+005, "");//ok
   RooRealVar r_p3_IPCHI2("p3_IPCHI2_OWNPV","p3_IPCHI2", 0.,5.30000000000000000e+005, "");//ok
   RooRealVar r_p1_IP("p1_IP_OWNPV","p1_IP_OWNPV",0,4.80000000000000000e+001);//ok
   RooRealVar r_p2_IP("p2_IP_OWNPV","p2_IP_OWNPV",0,5.00000000000000000e+001);//ok
   RooRealVar r_p3_IP("p3_IP_OWNPV","p3_IP_OWNPV",0,4.80000000000000000e+001);//ok
   RooRealVar r_p1_PT("p1_PT","p1_PT", 0.,2.80000000000000000e+004, "");//ok
   RooRealVar r_p2_PT("p2_PT","p2_PT", 0.,2.60000000000000000e+004, "");//ok
   RooRealVar r_p3_PT("p3_PT","p3_PT", 0.,5.10000000000000000e+004, "");//ok
   RooRealVar r_s12_pipi_DTF("s12_pipi_DTF","s12_pipi_DTF", 0.,3.80000000000000030e+000, "");//ok
   RooRealVar r_s13_pipi_DTF("s13_pipi_DTF","s13_pipi_DTF", 0.,3.85000000000000050e+000, "");//ok
   RooRealVar r_p1_PIDK("p1_PIDK","p1_PIDK",-1.10000000000000000e+003,1.00000000000000000e+002); //ok
   RooRealVar r_p2_PIDK("p2_PIDK","p2_PIDK",-1.10000000000000000e+003,1.00000000000000000e+002); //ok
   RooRealVar r_p3_PIDK("p3_PIDK","p3_PIDK",-1.10000000000000000e+003,1.00000000000000000e+002); //ok
   RooRealVar r_p1_P("p1_P","p1_P",0,1.08000000000000000e+005);//ok
   RooRealVar r_p2_P("p2_P","p2_P",0,1.08000000000000000e+005);//ok
   RooRealVar r_p3_P("p3_P","p3_P",0,1.08000000000000000e+005);//ok
   RooRealVar r_nTracks("nTracks","nTracks",0,600);//ok

   
   RooArgList vars(r_m, r_VTXCHI2, r_IPCHI2,r_IP,r_FDCHI2,r_FD,r_DIRA,r_BPVTRGPOINTING,r_tCM);
   
   vars.add(r_PT);
   vars.add(r_P);
   vars.add(r_PTsum);
   vars.add(r_logIP);
   vars.add(r_p1_IPCHI2);
   vars.add(r_p2_IPCHI2);
   vars.add(r_p3_IPCHI2);
   vars.add(r_p1_IP);
   vars.add(r_p2_IP);
   vars.add(r_p3_IP);
   vars.add(r_p1_PT);
   vars.add(r_p2_PT);
   vars.add(r_p3_PT);
   vars.add(r_s12_pipi_DTF);
   vars.add(r_s13_pipi_DTF);
   vars.add(r_p1_PIDK);
   vars.add(r_p2_PIDK);
   vars.add(r_p3_PIDK);
   vars.add(r_p1_P);
   vars.add(r_p2_P);
   vars.add(r_p3_P);
   vars.add(r_nTracks);*/
   

RooArgList vars(r_m);

 	  TFile *outFile=new TFile("accdataset.root","RECREATE");  

   RooDataSet data("data", "", vars);
   

   Long64_t nentries = fChain->GetEntriesFast();
   Long64_t nbytes = 0, nb = 0;
    // loop over  events
   for (Long64_t jentry=0; jentry<nentries;jentry++) {
      Long64_t ientry = LoadTree(jentry);
      if (ientry < 0) break;
      nb = fChain->GetEntry(jentry);   nbytes += nb;
 
	// fill dataset
              r_m = D_MM;
             /* r_VTXCHI2 = D_ENDVERTEX_CHI2;
              r_IPCHI2 = D_IPCHI2_OWNPV;
              r_IP = D_IP_OWNPV;
              r_FDCHI2 = D_FDCHI2_OWNPV;
              r_FD = D_FD_OWNPV;
              r_DIRA = D_DIRA_OWNPV;
              r_BPVTRGPOINTING = D_BPVTRGPOINTING;
              r_tCM = tCM;
              r_PT = D_PT;
              r_P = D_P;
              r_PTsum = PTsum;
              r_logIP = logIP;
              r_p1_IPCHI2 = p1_IPCHI2_OWNPV;
              r_p2_IPCHI2 = p2_IPCHI2_OWNPV;
              r_p3_IPCHI2 = p3_IPCHI2_OWNPV;
              r_p1_IP = p1_IP_OWNPV;
              r_p2_IP = p2_IP_OWNPV;
              r_p3_IP = p3_IP_OWNPV;
              r_p1_PT = p1_PT;
              r_p2_PT = p2_PT;
              r_p3_PT = p3_PT;	
              r_s12_pipi_DTF = s12_pipi_DTF;
              r_s13_pipi_DTF = s13_pipi_DTF;
              r_p1_PIDK = p1_PIDK;
              r_p2_PIDK = p2_PIDK;
              r_p3_PIDK = p3_PIDK;
              r_p1_P = p1_P;
              r_p2_P = p2_P;
              r_p3_P = p3_P;
              r_nTracks = nTracks;*/
              

              data.add(vars);
           
   } // end of loop over entries

   data.Write();
   outFile->Save();
   outFile->Close();
}