/*

Alice Morris

MPhys Project 2015/16

Script to select the sideband and signal regions

in order to subtract the background events using

sidebandSubtraction.C

*/

void sidebandSelection(){

// -- define tuple file name, tuple name and cuts to apply

// -- and also the name of the output file

const std::string filename = "~/cern/ntuples/Lb2chicpK_2011_2012_signal_fixed.root";

const std::string treename("MyTuple");

// SIDEBAND

/* const std::string cuts = "m_b_DTF_jpsi_chic1_constr > 5.66 && m_b_DTF_jpsi_chic1_constr < 5.70 && dtf_b < 7 && dtf_b > 0 && c2ip_b < 25 && dls_b > 10 && c2ip_kminus > 4 && c2ip_pplus > 4 && minCl_gamma > 0.03 && m_chic > 3.4 && m_chic < 3.7 && trig_b_l0tos_tos == 1 && trig_b_l1tos_tos == 1 && trig_b_l2tos_tos == 1 && trgh_track[0] < 0.3 && trgh_track[1] < 0.3 && trgh_track[2] < 0.3 && trgh_track[3] < 0.3 && K_ProbNNk > 0.15 && minann_mu > 0.1" ;

const std::string outFilename("~/cern/ntuples/sideband.root");

*/

// SIGNAL

const std::string cuts = "m_b_DTF_jpsi_chic1_constr > 5.61 && m_b_DTF_jpsi_chic1_constr < 5.63 && dtf_b < 7 && dtf_b > 0 && c2ip_b < 25 && dls_b > 10 && c2ip_kminus > 4 && c2ip_pplus > 4 && minCl_gamma > 0.03 && m_chic > 3.4 && m_chic < 3.7 && trig_b_l0tos_tos == 1 && trig_b_l1tos_tos == 1 && trig_b_l2tos_tos == 1 && trgh_track[0] < 0.3 && trgh_track[1] < 0.3 && trgh_track[2] < 0.3 && trgh_track[3] < 0.3 && K_ProbNNk > 0.15 && minann_mu > 0.1" ;

const std::string outFilename("~/cern/ntuples/signal.root");

TFile* file = TFile::Open( filename.c_str() );

if( !file ) std::cout << "file " << filename << " does not exist" << std::endl;

TTree* tree = (TTree*)file->Get( treename.c_str() );

if( !tree ) std::cout << "tree " << treename << " does not exist" << std::endl;

// -- activate the branches you need

tree->SetBranchStatus("*", 0);

tree->SetBranchStatus("m_b_DTF_jpsi_chic1_constr",1);

tree->SetBranchStatus("dtf_b",1);

tree->SetBranchStatus("c2ip_b",1);

tree->SetBranchStatus("dls_b",1);

tree->SetBranchStatus("c2ip_kminus",1);

tree->SetBranchStatus("c2ip_pplus",1);

tree->SetBranchStatus("minCl_gamma",1);

tree->SetBranchStatus("m_chic",1);

tree->SetBranchStatus("trig_b_l0tos_tos",1);

tree->SetBranchStatus("trig_b_l1tos_tos",1);

tree->SetBranchStatus("trig_b_l2tos_tos",1);

tree->SetBranchStatus("trgh_track",1);

tree->SetBranchStatus("ann_kaon",1);

tree->SetBranchStatus("K_ProbNNk",1);

tree->SetBranchStatus("ann_proton",1);

tree->SetBranchStatus("P_ProbNNp",1);

tree->SetBranchStatus("ann_mu",1);

tree->SetBranchStatus("minann_mu",1);

tree->SetBranchStatus("m_chicp",1);

tree->SetBranchStatus("m_pK",1);

tree->SetBranchStatus("m_jpsi",1);

// -- this file is just here to make the 'CopyTree' happy

TFile* dummyFile = new TFile("~/cern/ntuples/dummy.root","RECREATE");

TTree* rTree1 = tree->CopyTree( cuts.c_str() );

float mass, mass_chicp, mass_pK, mass_jpsi, mass_chic, dtf_b, c2ip_b, dls_b, c2ip_kminus, c2ip_pplus, minCl_gamma, trgh_track, ann_kaon, K_ProbNNk, ann_proton, P_ProbNNp, ann_mu, minann_mu;

float trig_b_l0tos_tos, trig_b_l1tos_tos, trig_b_l2tos_tos;

rTree1->SetBranchAddress("m_b_DTF_jpsi_chic1_constr", &mass);

rTree1->SetBranchAddress("m_chicp", &mass_chicp);

rTree1->SetBranchAddress("m_pK", &mass_pK);

rTree1->SetBranchAddress("m_jpsi", &mass_jpsi);

rTree1->SetBranchAddress("m_chic", &mass_chic);

rTree1->SetBranchAddress("dtf_b", &dtf_b);

rTree1->SetBranchAddress("c2ip_b", &c2ip_b);

rTree1->SetBranchAddress("dls_b", &dls_b);

rTree1->SetBranchAddress("c2ip_kminus", &c2ip_kminus);

rTree1->SetBranchAddress("c2ip_pplus", &c2ip_pplus);

rTree1->SetBranchAddress("minCl_gamma", &minCl_gamma);

rTree1->SetBranchAddress("trig_b_l0tos_tos", &trig_b_l0tos_tos);

rTree1->SetBranchAddress("trig_b_l1tos_tos", &trig_b_l1tos_tos);

rTree1->SetBranchAddress("trig_b_l2tos_tos", &trig_b_l2tos_tos);

rTree1->SetBranchAddress("trgh_track", &trgh_track);

rTree1->SetBranchAddress("ann_kaon", &ann_kaon);

rTree1->SetBranchAddress("K_ProbNNk", &K_ProbNNk);

rTree1->SetBranchAddress("ann_proton", &ann_proton);

rTree1->SetBranchAddress("P_ProbNNp", &P_ProbNNp);

rTree1->SetBranchAddress("ann_mu", &ann_mu);

rTree1->SetBranchAddress("minann_mu", &minann_mu);

TFile* rFile = new TFile( outFilename.c_str() ,"RECREATE");

TTree* rTree2 = new TTree();

//rTree2->SetName("sideband");

rTree2->SetName("signal");

rTree2->Branch("mass", &mass, "mass");

rTree2->Branch("mass_chicp", &mass_chicp, "mass_chicp/F");

rTree2->Branch("mass_pK", &mass_pK, "mass_pK/F");

rTree2->Branch("mass_jpsi", &mass_jpsi, "mass_jpsi/F");

rTree2->Branch("mass_chic", &mass_chic, "mass_chic/F");

rTree2->Branch("dtf_b", &dtf_b, "dtf_b/F");

rTree2->Branch("c2ip_b", &c2ip_b, "c2ip_b/F");

rTree2->Branch("dls_b", &dls_b, "dls_b/F");

rTree2->Branch("c2ip_kminus", &c2ip_kminus, "c2ip_kminus/F");

rTree2->Branch("c2ip_pplus", &c2ip_pplus, "c2ip_pplus/F");

rTree2->Branch("minCl_gamma", &minCl_gamma, "minCl_gamma/F");

rTree2->Branch("trig_b_l0tos_tos", &trig_b_l0tos_tos, "trig_b_l0tos_tos/F");

rTree2->Branch("trig_b_l1tos_tos", &trig_b_l1tos_tos, "trig_b_l1tos_tos/F");

rTree2->Branch("trig_b_l2tos_tos", &trig_b_l2tos_tos, "trig_b_l2tos_tos/F");

rTree2->Branch("trgh_track", &trgh_track, "trgh_track/F");

rTree2->Branch("ann_kaon", &ann_kaon, "ann_kaon/F");

rTree2->Branch("K_ProbNNk", &K_ProbNNk, "K_ProbNNk/F");

rTree2->Branch("ann_proton", &ann_proton, "ann_proton/F");

rTree2->Branch("P_ProbNNp", &P_ProbNNp, "P_ProbNNp/F");

rTree2->Branch("ann_mu", &ann_mu, "ann_mu/F");

rTree2->Branch("minann_mu", &minann_mu, "minann_mu/F");

int percentCounter = 1;

for(int i = 0; i < rTree1->GetEntries(); ++i){

const int percent = (int)(rTree1->GetEntries()/100.0);

if( i == percent*percentCounter ){

//std::cout << percentCounter << " %" << std::endl;

percentCounter++;

}

rTree1->GetEntry(i);

rTree2->Fill();

//std::cout << mass << std::endl;

}

rTree2->Print();

rTree2->Write();

rFile->Save();

}