void PIDStudy::InitializeFits()
{
for (int multBin = 0; multBin < nMultiplicityBins; multBin++)
for (int PBin=0; PBin < nPBins_lin; PBin++)
{
TF1 pion("pion", "gaus(0)", 2.0, 4.0);
pion.SetParameter(0,10000);
pion.SetParameter(1,2.8);
pion.SetParameter(2,0.2);
TF1 kaon("kaon", "gaus(0)", 3.0, 6.0);
kaon.SetParameter(0,2000);
kaon.SetParameter(1,4);
kaon.SetParameter(2,0.2);
TF1 prot("prot", "gaus(0)", 5.0, 8.0);
prot.SetParameter(0,1000);
prot.SetParameter(1,7);
prot.SetParameter(2,0.4);
// "R" means only using fit range.
//dEdxDistr_log[multBin][PBin]->Fit(&pion, "R");
//dEdxDistr_log[multBin][PBin]->Fit(&kaon, "R");
//dEdxDistr_log[multBin][PBin]->Fit(&prot, "R");
double pion_amp = pion.GetParameter(0);
double pion_pos = pion.GetParameter(1);
double pion_sig = pion.GetParameter(2);
double kaon_amp = kaon.GetParameter(0);
double kaon_pos = kaon.GetParameter(1);
double kaon_sig = kaon.GetParameter(2);
double prot_amp = prot.GetParameter(0);
double prot_pos = prot.GetParameter(1);
double prot_sig = prot.GetParameter(2);
// Pion
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(0,pion_amp);
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(1,pion_pos);
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(2,pion_sig);
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(3,kaon_amp);
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(4,kaon_pos);
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(5,kaon_sig);
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(6,prot_amp);
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(7,prot_pos);
dEdxDistr_log_Fit[multBin][PBin]->SetParameter(8,prot_sig);
}
}
void newReduceTree(){
// -- define tuple file name, tuple name and cuts to apply
// -- and also the name of the output file
const std::string filename = "/afs/cern.ch/work/a/apmorris/public/Lb2chicpK/signal_samples/Lb2chicpK_MC_2011_2012_signal_cut.root";//change
const std::string treename = "DecayTree";
const std::string cuts = "";
const std::string outFile = "/afs/cern.ch/work/a/apmorris/public/Lb2chicpK/signal_samples/reduced_Lb2chicpK_MC_2011_2012_signal.root");//change
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("chi_c_M",1);
tree->SetBranchStatus("chi_c_P",1);
tree->SetBranchStatus("chi_c_PE",1);
tree->SetBranchStatus("chi_c_PT",1);
tree->SetBranchStatus("chi_c_PX",1);
tree->SetBranchStatus("chi_c_PY",1);
tree->SetBranchStatus("chi_c_PZ",1);
tree->SetBranchStatus("chi_c_ETA",1);
tree->SetBranchStatus("kaon_M",1);
tree->SetBranchStatus("kaon_P",1);
tree->SetBranchStatus("kaon_PE",1);
tree->SetBranchStatus("kaon_PT",1);
tree->SetBranchStatus("kaon_PX",1);
tree->SetBranchStatus("kaon_PY",1);
tree->SetBranchStatus("kaon_PZ",1);
tree->SetBranchStatus("kaon_ETA",1);
tree->SetBranchStatus("kaon_IPCHI2_OWNPV",1);
tree->SetBranchStatus("kaon_TRACK_GhostProb",1);
tree->SetBranchStatus("kaon_ProbNNp",1);
tree->SetBranchStatus("kaon_ProbNNk",1);
tree->SetBranchStatus("proton_M",1);
tree->SetBranchStatus("proton_P",1);
tree->SetBranchStatus("proton_PE",1);
tree->SetBranchStatus("proton_PT",1);
tree->SetBranchStatus("proton_PX",1);
tree->SetBranchStatus("proton_PY",1);
tree->SetBranchStatus("proton_PZ",1);
tree->SetBranchStatus("proton_ETA",1);
tree->SetBranchStatus("proton_IPCHI2_OWNPV",1);
tree->SetBranchStatus("proton_TRACK_GhostProb",1);
tree->SetBranchStatus("proton_ProbNNp",1);
tree->SetBranchStatus("proton_ProbNNk",1);
tree->SetBranchStatus("Jpsi_M",1);
tree->SetBranchStatus("Jpsi_P",1);
tree->SetBranchStatus("Jpsi_PE",1);
tree->SetBranchStatus("Jpsi_PT",1);
tree->SetBranchStatus("Jpsi_PX",1);
tree->SetBranchStatus("Jpsi_PY",1);
tree->SetBranchStatus("Jpsi_PZ",1);
tree->SetBranchStatus("Jpsi_ETA",1);
tree->SetBranchStatus("gamma_M",1);
tree->SetBranchStatus("gamma_P",1);
tree->SetBranchStatus("gamma_PE",1);
tree->SetBranchStatus("gamma_PT",1);
tree->SetBranchStatus("gamma_PX",1);
tree->SetBranchStatus("gamma_PY",1);
tree->SetBranchStatus("gamma_PZ",1);
tree->SetBranchStatus("gamma_CL",1);
tree->SetBranchStatus("gamma_ETA",1);
tree->SetBranchStatus("muminus_M",1);
tree->SetBranchStatus("muminus_P",1);
tree->SetBranchStatus("muminus_PE",1);
tree->SetBranchStatus("muminus_PT",1);
tree->SetBranchStatus("muminus_PX",1);
tree->SetBranchStatus("muminus_PY",1);
tree->SetBranchStatus("muminus_PZ",1);
tree->SetBranchStatus("muminus_ETA",1);
tree->SetBranchStatus("muminus_ProbNNmu",1);
tree->SetBranchStatus("muminus_TRACK_GhostProb",1);
tree->SetBranchStatus("muplus_M",1);
tree->SetBranchStatus("muplus_P",1);
tree->SetBranchStatus("muplus_PE",1);
tree->SetBranchStatus("muplus_PT",1);
tree->SetBranchStatus("muplus_PX",1);
tree->SetBranchStatus("muplus_PY",1);
tree->SetBranchStatus("muplus_PZ",1);
tree->SetBranchStatus("muplus_ETA",1);
tree->SetBranchStatus("muplus_ProbNNmu",1);
tree->SetBranchStatus("muplus_TRACK_GhostProb",1);
tree->SetBranchStatus("Lambda_b0_DTF_MASS_constr1",1);
tree->SetBranchStatus("Lambda_b0_DTF_MASS_constr2",1);
tree->SetBranchStatus("Lambda_b0_DTF_CHI2NDOF",1);
tree->SetBranchStatus("Lambda_b0_IPCHI2_OWNPV",1);
tree->SetBranchStatus("Lambda_b0_FDS",1);
tree->SetBranchStatus("Lambda_b0_L0MuonDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_L0DiMuonDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1DiMuonHighMassDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1DiMuonLowMassDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1TrackMuonDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1TrackAllL0Decision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt2DiMuonDetachedDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS",1);
tree->SetBranchStatus("Lambda_b0_pi0veto",1);
tree->SetBranchStatus("Lambda_b0_PT",1);
//for MC only //
tree->SetBranchStatus("chi_c_BKGCAT",1); //
tree->SetBranchStatus("Jpsi_BKGCAT",1); //
tree->SetBranchStatus("Lambda_b0_BKGCAT",1); //
// -- this file is just here to make the 'CopyTree' happy
TFile* dummyFile = new TFile("/afs/cern.ch/work/a/apmorris/private/cern/ntuples/dummy.root","RECREATE");
TTree* rTree1 = tree->CopyTree(cuts.c_str());
double chi_c_M, chi_c_P, chi_c_PE, chi_c_PT, chi_c_PX, chi_c_PY, chi_c_PZ, chi_c_ETA;
double kaon_M, kaon_P, kaon_PE, kaon_PX, kaon_PT, kaon_PY, kaon_PZ, kaon_ETA;
double proton_M, proton_P, proton_PE, proton_PT, proton_PX, proton_PY, proton_PZ, proton_ETA;
double Jpsi_M, Jpsi_P, Jpsi_PE, Jpsi_PT, Jpsi_PX, Jpsi_PY, Jpsi_PZ, Jpsi_ETA;
double gamma_M, gamma_P, gamma_PE, gamma_PT, gamma_PX, gamma_PY, gamma_PZ, gamma_ETA, gamma_CL;
double muminus_M, muminus_P, muminus_PE, muminus_PT, muminus_PX, muminus_PY, muminus_PZ, muminus_ETA, muminus_ProbNNmu, muminus_TRACK_GhostProb;
double muplus_M, muplus_P, muplus_PE, muplus_PT, muplus_PX, muplus_PY, muplus_PZ, muplus_ETA, muplus_ProbNNmu, muplus_TRACK_GhostProb;
double kaon_IPCHI2_OWNPV, kaon_TRACK_GhostProb, kaon_ProbNNp, kaon_ProbNNk;
double proton_IPCHI2_OWNPV, proton_TRACK_GhostProb, proton_ProbNNp, proton_ProbNNk;
double Lambda_b0_DTF_MASS_constr1, Lambda_b0_DTF_MASS_constr2, Lambda_b0_DTF_CHI2NDOF, Lambda_b0_IPCHI2_OWNPV;
double Lambda_b0_FDS, Lambda_b0_pi0veto, Lambda_b0_PT;
bool Lambda_b0_L0MuonDecision_TOS, Lambda_b0_L0DiMuonDecision_TOS;
bool Lambda_b0_Hlt1DiMuonHighMassDecision_TOS, Lambda_b0_Hlt1DiMuonLowMassDecision_TOS;
bool Lambda_b0_Hlt1TrackMuonDecision_TOS, Lambda_b0_Hlt1TrackAllL0Decision_TOS;
bool Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS, Lambda_b0_Hlt2DiMuonDetachedDecision_TOS;
bool Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS, Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS;
rTree1->SetBranchAddress("chi_c_M", &chi_c_M);
rTree1->SetBranchAddress("chi_c_P", &chi_c_P);
rTree1->SetBranchAddress("chi_c_PE", &chi_c_PE);
rTree1->SetBranchAddress("chi_c_PT", &chi_c_PT);
rTree1->SetBranchAddress("chi_c_PX", &chi_c_PX);
rTree1->SetBranchAddress("chi_c_PY", &chi_c_PY);
rTree1->SetBranchAddress("chi_c_PZ", &chi_c_PZ);
rTree1->SetBranchAddress("chi_c_ETA", &chi_c_ETA);
rTree1->SetBranchAddress("kaon_M", &kaon_M);
rTree1->SetBranchAddress("kaon_P", &kaon_P);
rTree1->SetBranchAddress("kaon_PE", &kaon_PE);
rTree1->SetBranchAddress("kaon_PX", &kaon_PX);
rTree1->SetBranchAddress("kaon_PT", &kaon_PT);
rTree1->SetBranchAddress("kaon_PY", &kaon_PY);
rTree1->SetBranchAddress("kaon_PZ", &kaon_PZ);
rTree1->SetBranchAddress("kaon_ETA", &kaon_ETA);
rTree1->SetBranchAddress("kaon_IPCHI2_OWNPV", &kaon_IPCHI2_OWNPV);
rTree1->SetBranchAddress("kaon_TRACK_GhostProb", &kaon_TRACK_GhostProb);
rTree1->SetBranchAddress("kaon_ProbNNp", &kaon_ProbNNp);
rTree1->SetBranchAddress("kaon_ProbNNk", &kaon_ProbNNk);
rTree1->SetBranchAddress("proton_M", &proton_M);
rTree1->SetBranchAddress("proton_P", &proton_P);
rTree1->SetBranchAddress("proton_PE", &proton_PE);
rTree1->SetBranchAddress("proton_PT", &proton_PT);
rTree1->SetBranchAddress("proton_PX", &proton_PX);
rTree1->SetBranchAddress("proton_PY", &proton_PY);
rTree1->SetBranchAddress("proton_PZ", &proton_PZ);
rTree1->SetBranchAddress("proton_ETA", &proton_ETA);
rTree1->SetBranchAddress("proton_IPCHI2_OWNPV", &proton_IPCHI2_OWNPV);
rTree1->SetBranchAddress("proton_TRACK_GhostProb", &proton_TRACK_GhostProb);
rTree1->SetBranchAddress("proton_ProbNNp", &proton_ProbNNp);
rTree1->SetBranchAddress("proton_ProbNNk", &proton_ProbNNk);
rTree1->SetBranchAddress("Jpsi_M", &Jpsi_M);
rTree1->SetBranchAddress("Jpsi_P", &Jpsi_P);
rTree1->SetBranchAddress("Jpsi_PE", &Jpsi_PE);
rTree1->SetBranchAddress("Jpsi_PT", &Jpsi_PT);
rTree1->SetBranchAddress("Jpsi_PX", &Jpsi_PX);
rTree1->SetBranchAddress("Jpsi_PY", &Jpsi_PY);
rTree1->SetBranchAddress("Jpsi_PZ", &Jpsi_PZ);
rTree1->SetBranchAddress("Jpsi_ETA", &Jpsi_ETA);
rTree1->SetBranchAddress("gamma_M", &gamma_M);
rTree1->SetBranchAddress("gamma_P", &gamma_P);
rTree1->SetBranchAddress("gamma_PE", &gamma_PE);
rTree1->SetBranchAddress("gamma_PT", &gamma_PT);
rTree1->SetBranchAddress("gamma_PX", &gamma_PX);
rTree1->SetBranchAddress("gamma_PY", &gamma_PY);
rTree1->SetBranchAddress("gamma_PZ", &gamma_PZ);
rTree1->SetBranchAddress("gamma_ETA", &gamma_ETA);
rTree1->SetBranchAddress("gamma_CL", &gamma_CL);
rTree1->SetBranchAddress("muminus_M", &muminus_M);
rTree1->SetBranchAddress("muminus_P", &muminus_P);
rTree1->SetBranchAddress("muminus_PE", &muminus_PE);
rTree1->SetBranchAddress("muminus_PT", &muminus_PT);
rTree1->SetBranchAddress("muminus_PX", &muminus_PX);
rTree1->SetBranchAddress("muminus_PY", &muminus_PY);
rTree1->SetBranchAddress("muminus_PZ", &muminus_PZ);
rTree1->SetBranchAddress("muminus_ETA", &muminus_ETA);
rTree1->SetBranchAddress("muminus_ProbNNmu", &muminus_ProbNNmu);
rTree1->SetBranchAddress("muminus_TRACK_GhostProb", &muminus_TRACK_GhostProb);
rTree1->SetBranchAddress("muplus_M", &muplus_M);
rTree1->SetBranchAddress("muplus_P", &muplus_P);
rTree1->SetBranchAddress("muplus_PE", &muplus_PE);
rTree1->SetBranchAddress("muplus_PT", &muplus_PT);
rTree1->SetBranchAddress("muplus_PX", &muplus_PX);
rTree1->SetBranchAddress("muplus_PY", &muplus_PY);
rTree1->SetBranchAddress("muplus_PZ", &muplus_PZ);
rTree1->SetBranchAddress("muplus_ETA", &muplus_ETA);
rTree1->SetBranchAddress("muplus_ProbNNmu", &muplus_ProbNNmu);
rTree1->SetBranchAddress("muplus_TRACK_GhostProb", &muplus_TRACK_GhostProb);
rTree1->SetBranchAddress("Lambda_b0_DTF_MASS_constr1", &Lambda_b0_DTF_MASS_constr1);
rTree1->SetBranchAddress("Lambda_b0_DTF_MASS_constr2", &Lambda_b0_DTF_MASS_constr2);
rTree1->SetBranchAddress("Lambda_b0_DTF_CHI2NDOF", &Lambda_b0_DTF_CHI2NDOF);
rTree1->SetBranchAddress("Lambda_b0_IPCHI2_OWNPV", &Lambda_b0_IPCHI2_OWNPV);
rTree1->SetBranchAddress("Lambda_b0_L0DiMuonDecision_TOS", &Lambda_b0_L0DiMuonDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_L0MuonDecision_TOS", &Lambda_b0_L0MuonDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_FDS", &Lambda_b0_FDS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1DiMuonHighMassDecision_TOS", &Lambda_b0_Hlt1DiMuonHighMassDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1DiMuonLowMassDecision_TOS", &Lambda_b0_Hlt1DiMuonLowMassDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1TrackMuonDecision_TOS", &Lambda_b0_Hlt1TrackMuonDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1TrackAllL0Decision_TOS", &Lambda_b0_Hlt1TrackAllL0Decision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS", &Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt2DiMuonDetachedDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS);
rTree1->SetBranchAddress("Lambda_b0_pi0veto", &Lambda_b0_pi0veto);
rTree1->SetBranchAddress("Lambda_b0_PT", &Lambda_b0_PT);
//for MC only //
int chi_c_BKGCAT, Jpsi_BKGCAT, Lambda_b0_BKGCAT; //
rTree1->SetBranchAddress("chi_c_BKGCAT", &chi_c_BKGCAT); //
rTree1->SetBranchAddress("Jpsi_BKGCAT", &Jpsi_BKGCAT); //
rTree1->SetBranchAddress("Lambda_b0_BKGCAT", &Lambda_b0_BKGCAT); //
TFile* rFile = new TFile( outFile.c_str() ,"RECREATE");
TTree* rTree2 = new TTree();
rTree2->SetName("DecayTree");
rTree2->Branch("chi_c_M", &chi_c_M, "chi_c_M/D");
rTree2->Branch("chi_c_P", &chi_c_P, "chi_c_P/D");
rTree2->Branch("chi_c_PE", &chi_c_PE, "chi_c_PE/D");
rTree2->Branch("chi_c_PT", &chi_c_PT, "chi_c_PT/D");
rTree2->Branch("chi_c_PX", &chi_c_PX, "chi_c_PX/D");
rTree2->Branch("chi_c_PY", &chi_c_PY, "chi_c_PY/D");
rTree2->Branch("chi_c_PZ", &chi_c_PZ, "chi_c_PZ/D");
rTree2->Branch("chi_c_ETA", &chi_c_ETA, "chi_c_ETA/D");
rTree2->Branch("kaon_M", &kaon_M, "kaon_M/D");
rTree2->Branch("kaon_P", &kaon_P, "kaon_P/D");
rTree2->Branch("kaon_PE", &kaon_PE, "kaon_PE/D");
rTree2->Branch("kaon_PX", &kaon_PX, "kaon_PX/D");
rTree2->Branch("kaon_PT", &kaon_PT, "kaon_PT/D");
rTree2->Branch("kaon_PY", &kaon_PY, "kaon_PY/D");
rTree2->Branch("kaon_PZ", &kaon_PZ, "kaon_PZ/D");
rTree2->Branch("kaon_ETA", &kaon_ETA, "kaon_ETA/D");
rTree2->Branch("kaon_IPCHI2_OWNPV", &kaon_IPCHI2_OWNPV, "kaon_IPCHI2_OWNPV/D");
rTree2->Branch("kaon_TRACK_GhostProb", &kaon_TRACK_GhostProb, "kaon_TRACK_GhostProb/D");
rTree2->Branch("kaon_ProbNNp", &kaon_ProbNNp, "kaon_ProbNNp/D");
rTree2->Branch("kaon_ProbNNk", &kaon_ProbNNk, "kaon_ProbNNk/D");
rTree2->Branch("proton_M", &proton_M, "proton_M/D");
rTree2->Branch("proton_P", &proton_P, "proton_P/D");
rTree2->Branch("proton_PE", &proton_PE, "proton_PE/D");
rTree2->Branch("proton_PT", &proton_PT, "proton_PT/D");
rTree2->Branch("proton_PX", &proton_PX, "proton_PX/D");
rTree2->Branch("proton_PY", &proton_PY, "proton_PY/D");
rTree2->Branch("proton_PZ", &proton_PZ, "proton_PZ/D");
rTree2->Branch("proton_ETA", &proton_ETA, "proton_ETA/D");
rTree2->Branch("proton_IPCHI2_OWNPV", &proton_IPCHI2_OWNPV, "proton_IPCHI2_OWNPV/D");
rTree2->Branch("proton_TRACK_GhostProb", &proton_TRACK_GhostProb, "proton_TRACK_GhostProb/D");
rTree2->Branch("proton_ProbNNp", &proton_ProbNNp, "proton_ProbNNp/D");
rTree2->Branch("proton_ProbNNk", &proton_ProbNNk, "proton_ProbNNk/D");
rTree2->Branch("Jpsi_M", &Jpsi_M, "Jpsi_M/D");
rTree2->Branch("Jpsi_P", &Jpsi_P, "Jpsi_P/D");
rTree2->Branch("Jpsi_PE", &Jpsi_PE, "Jpsi_PE/D");
rTree2->Branch("Jpsi_PT", &Jpsi_PT, "Jpsi_PT/D");
rTree2->Branch("Jpsi_PX", &Jpsi_PX, "Jpsi_PX/D");
rTree2->Branch("Jpsi_PY", &Jpsi_PY, "Jpsi_PY/D");
rTree2->Branch("Jpsi_PZ", &Jpsi_PZ, "Jpsi_PZ/D");
rTree2->Branch("Jpsi_ETA", &Jpsi_ETA, "Jpsi_ETA/D");
rTree2->Branch("gamma_M", &gamma_M, "gamma_M/D");
rTree2->Branch("gamma_P", &gamma_P, "gamma_P/D");
rTree2->Branch("gamma_PE", &gamma_PE, "gamma_PE/D");
rTree2->Branch("gamma_PT", &gamma_PT, "gamma_PT/D");
rTree2->Branch("gamma_PX", &gamma_PX, "gamma_PX/D");
rTree2->Branch("gamma_PY", &gamma_PY, "gamma_PY/D");
rTree2->Branch("gamma_PZ", &gamma_PZ, "gamma_PZ/D");
rTree2->Branch("gamma_ETA", &gamma_ETA, "gamma_ETA/D");
rTree2->Branch("gamma_CL", &gamma_CL, "gamma_CL/D");
rTree2->Branch("muminus_M", &muminus_M, "muminus_M/D");
rTree2->Branch("muminus_P", &muminus_P, "muminus_P/D");
rTree2->Branch("muminus_PE", &muminus_PE, "muminus_PE/D");
rTree2->Branch("muminus_PT", &muminus_PT, "muminus_PT/D");
rTree2->Branch("muminus_PX", &muminus_PX, "muminus_PX/D");
rTree2->Branch("muminus_PY", &muminus_PY, "muminus_PY/D");
rTree2->Branch("muminus_PZ", &muminus_PZ, "muminus_PZ/D");
rTree2->Branch("muminus_ETA", &muminus_ETA, "muminus_ETA/D");
rTree2->Branch("muminus_ProbNNmu", &muminus_ProbNNmu, "muminus_ProbNNmu/D");
rTree2->Branch("muminus_TRACK_GhostProb", &muminus_TRACK_GhostProb, "muminus_TRACK_GhostProb/D");
rTree2->Branch("muplus_M", &muplus_M, "muplus_M/D");
rTree2->Branch("muplus_P", &muplus_P, "muplus_P/D");
rTree2->Branch("muplus_PE", &muplus_PE, "muplus_PE/D");
rTree2->Branch("muplus_PT", &muplus_PT, "muplus_PT/D");
rTree2->Branch("muplus_PX", &muplus_PX, "muplus_PX/D");
rTree2->Branch("muplus_PY", &muplus_PY, "muplus_PY/D");
rTree2->Branch("muplus_PZ", &muplus_PZ, "muplus_PZ/D");
rTree2->Branch("muplus_ETA", &muplus_ETA, "muplus_ETA/D");
rTree2->Branch("muplus_ProbNNmu", &muplus_ProbNNmu, "muplus_ProbNNmu/D");
rTree2->Branch("muplus_TRACK_GhostProb", &muplus_TRACK_GhostProb, "muplus_TRACK_GhostProb/D");
rTree2->Branch("Lambda_b0_DTF_MASS_constr1", &Lambda_b0_DTF_MASS_constr1, "Lambda_b0_DTF_MASS_constr1/D");
rTree2->Branch("Lambda_b0_DTF_MASS_constr2", &Lambda_b0_DTF_MASS_constr2, "Lambda_b0_DTF_MASS_constr2/D");
rTree2->Branch("Lambda_b0_DTF_CHI2NDOF", &Lambda_b0_DTF_CHI2NDOF, "Lambda_b0_DTF_CHI2NDOF/D");
rTree2->Branch("Lambda_b0_IPCHI2_OWNPV", &Lambda_b0_IPCHI2_OWNPV, "Lambda_b0_IPCHI2_OWNPV/D");
rTree2->Branch("Lambda_b0_L0DiMuonDecision_TOS",&Lambda_b0_L0DiMuonDecision_TOS,"Lambda_b0_L0DiMuonDecision_TOS/B");
rTree2->Branch("Lambda_b0_L0MuonDecision_TOS", &Lambda_b0_L0MuonDecision_TOS, "Lambda_b0_L0MuonDecision_TOS/B");
rTree2->Branch("Lambda_b0_FDS", &Lambda_b0_FDS, "Lambda_b0_FDS/D");
rTree2->Branch("Lambda_b0_Hlt1DiMuonHighMassDecision_TOS", &Lambda_b0_Hlt1DiMuonHighMassDecision_TOS, "Lambda_b0_Hlt1DiMuonHighMassDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt1DiMuonLowMassDecision_TOS", &Lambda_b0_Hlt1DiMuonLowMassDecision_TOS, "Lambda_b0_Hlt1DiMuonLowMassDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt1TrackMuonDecision_TOS", &Lambda_b0_Hlt1TrackMuonDecision_TOS, "Lambda_b0_Hlt1TrackMuonDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt1TrackAllL0Decision_TOS", &Lambda_b0_Hlt1TrackAllL0Decision_TOS, "Lambda_b0_Hlt1TrackAllL0Decision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS", &Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS, "Lambda_b0_Hlt1SingleMuonHighPTDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt2DiMuonDetachedDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedDecision_TOS, "Lambda_b0_Hlt2DiMuonDetachedDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS, "Lambda_b0_Hlt2DiMuonDetachedJPsiDecision_TOS/B");
rTree2->Branch("Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS", &Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS, "Lambda_b0_Hlt2DiMuonDetachedHeavyDecision_TOS/B");
rTree2->Branch("Lambda_b0_pi0veto", &Lambda_b0_pi0veto, "Lambda_b0_pi0veto/D");
rTree2->Branch("Lambda_b0_PT", &Lambda_b0_PT, "Lambda_b0_PT/D");
//for MC only //
rTree2->Branch("chi_c_BKGCAT", &chi_c_BKGCAT, "chi_c_BKGCAT/I"); //
rTree2->Branch("Jpsi_BKGCAT", &Jpsi_BKGCAT, "Jpsi_BKGCAT/I"); //
rTree2->Branch("Lambda_b0_BKGCAT", &Lambda_b0_BKGCAT, "Lambda_b0_BKGCAT/I"); //
//-----------------------------------------------------------------------------------
const double mK(493.677);
const double mpi(139.57);
const double mp(938.27);
double m_pK, m_chicp;
double proton_as_kaon_M, proton_as_pion_M, proton_as_proton_M;
double m_chicpK_proton_as_kaon, m_chicpK_proton_as_pion, m_chicpK_proton_as_proton;
double m_JpsipK;
rTree2->Branch("m_JpsipK", &m_JpsipK, "m_JpsipK/D");
rTree2->Branch("m_chicp", &m_chicp, "m_chicp/D");
rTree2->Branch("m_pK", &m_pK, "m_pK/D");
rTree2->Branch("proton_as_kaon_M", &proton_as_kaon_M, "proton_as_kaon_M/D");
rTree2->Branch("proton_as_pion_M", &proton_as_pion_M, "proton_as_pion_M/D");
rTree2->Branch("proton_as_proton_M", &proton_as_proton_M, "proton_as_proton_M/D");
rTree2->Branch("m_chicpK_proton_as_kaon", &m_chicpK_proton_as_kaon, "m_chicpK_proton_as_kaon/D");
rTree2->Branch("m_chicpK_proton_as_pion", &m_chicpK_proton_as_pion, "m_chicpK_proton_as_pion/D");
rTree2->Branch("m_chicpK_proton_as_proton", &m_chicpK_proton_as_proton, "m_chicpK_proton_as_proton/D");
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);
double proton_P = sqrt(proton_PX*proton_PX + proton_PY*proton_PY + proton_PZ*proton_PZ) ;
TLorentzVector proton_as_kaon( proton_PX, proton_PY, proton_PZ, sqrt(proton_P*proton_P + mK*mK));
TLorentzVector proton_as_pion( proton_PX, proton_PY, proton_PZ, sqrt(proton_P*proton_P + mpi*mpi));
TLorentzVector proton_as_proton( proton_PX, proton_PY, proton_PZ, sqrt(proton_P*proton_P + proton_M*proton_M));
TLorentzVector kaon( kaon_PX, kaon_PY, kaon_PZ, kaon_PE);
TLorentzVector chic( chi_c_PX, chi_c_PY, chi_c_PZ, chi_c_PE);
TLorentzVector proton(proton_PX, proton_PY, proton_PZ, proton_PE);
TLorentzVector Jpsi( Jpsi_PX, Jpsi_PY, Jpsi_PZ, Jpsi_PE);
proton_as_kaon_M = proton_as_kaon.M();
proton_as_pion_M = proton_as_pion.M();
proton_as_proton_M = proton_as_proton.M();
TLorentzVector chic_PasK_K = chic + kaon + proton_as_kaon;
TLorentzVector chic_PasPi_K = chic + kaon + proton_as_pion;
TLorentzVector chic_PasP_K = chic + kaon + proton_as_proton;
m_chicpK_proton_as_kaon = chic_PasK_K.M();
m_chicpK_proton_as_pion = chic_PasPi_K.M();
m_chicpK_proton_as_proton = chic_PasP_K.M();
TLorentzVector JpsipK = Jpsi + proton + kaon;
TLorentzVector chicp = chic + proton;
TLorentzVector pK = proton + kaon;
m_JpsipK = JpsipK.M();
m_chicp = chicp.M();
m_pK = pK.M();
rTree2->Fill();
}
rTree2->Print();
rTree2->Write();
rFile->Save();
}
