kopecna/EWP-BplusToKstMuMu-AngAna
kopecna/EWP-BplusToKstMuMu-AngAna
2
1
Fork 0
Code Issues Pull Requests repo.project_board Releases Wiki Activity
6b17868719
EWP-BplusToKstMuMu-AngAna/Code/Scripts/Selection/DTFcomparison.cc

1653 lines
83 KiB
C++
Raw Normal View History

First commit, not including the most of the Codes/ folder
2022-02-10 15:49:54 +01:00
#include "../GlobalFunctions.hh"
//////////////////////////////////////////////////////////////////////
/// DTFcomparison()
/// #todo
///
string figureFormat = "eps";
std::array<string, 3> ParticlesPiZero {"B",
//"muplus",
//"muminus",
"Kstar",
"pi0",
//"gamma1",
//"gamma2"
};
std::array<string, 3> ParticlesKShort {"B",
//"muplus",
//"muminus",
"Kstar",
"Kshort"
};
std::array<string, 3> Variables{ "M",
"PT",
"ETA"
};
std::array<string,1> ExtraVariables{// "ThetaK",
// "ThetaL",
// "Phi",
"OpenAngle",
};
std::array<string, 2> DTFoptions{//"DTF_PDG",
//"PDG",
"DTF",
""
};
string particleLatex(string particle){ //one doesn't have to think much when making labels and titles
if (particle == "B") return ("B^{+}");
if (particle == "Bplus") return ("B^{+}");
if (particle == "B_plus") return ("B^{+}");
if (particle == "Kstar") return ("K^{*+}");
if (particle == "K_star") return ("K^{*+}");
if (particle == "Kplus") return ("K^{+}");
if (particle == "K_plus") return ("K^{+}");
if (particle == "Kshort") return ("K_{s}");
if (particle == "K_short") return ("K_{s}");
if (particle == "piplus") return ("#pi^{+}");
if (particle == "pi_plus") return ("#pi^{+}");
if (particle == "pizero") return ("#pi^{0}");
if (particle == "pi_zero") return ("#pi^{0}");
if (particle == "pi0") return ("#pi^{0}");
if (particle == "mu") return ("#mu");
if (particle == "mu_plus") return ("#mu^{+}");
if (particle == "muplus") return ("#mu^{+}");
if (particle == "mu_minus") return ("#mu^{-}");
if (particle == "muminus") return ("#mu^{-}");
if (particle == "gamma") return ("#gamma");
cout << "Particle "<< particle <<" not in the list!" << endl;
return "";
}
string variableLatex(string var){
if (var=="PT") return ("p_{T}");
if (var=="PX") return ("p_{X}");
if (var=="PY") return ("p_{Y}");
if (var=="PZ") return ("p_{Z}");
if (var=="ETA") return ("#eta");
if (var=="M") return ("m");
if (var=="OpenAngle") return ("Open angle (#gamma#gamma angle)");
cout << "Variable "<< var <<" is not in the list!" << endl;
return "";
}
string optionLatex(string opt){
if (opt=="DTF_PDG") return ("#splitline{info from DTF}{with masses fixed to PDG}");
if (opt=="DTF") return ("info from DTF");
if (opt=="PDG") return ("#splitline{info from TupleTool}{with masses fixed to PDG}");
if (opt=="") return ("info from TupleTool");
cout << "Option " << opt << " not in the list!" << endl;
return "";
}
string histName(string particle, string var, string opt){
return (variableLatex(var)+"("+ particleLatex(particle)+ "):" + optionLatex(opt));
}
void designCanvas(TCanvas* canvas){
canvas->SetRightMargin(0.04);
canvas->SetTopMargin(0.04);
canvas->SetLeftMargin(0.12);
canvas->SetBottomMargin(0.12);
}
void designPlots(TH1D *hist, int color){
hist->SetLineWidth(2);
hist->SetLineColor(color);
hist->GetYaxis()->SetTitleOffset(1.7);
hist->GetXaxis()->SetTitleOffset(1.0);
hist->SetTitle("");
}
void designLegend(TLegend *leg, string legTitle, TH1D *hist, string opt){
leg->SetTextSize(0.03);
leg->SetHeader(legTitle.c_str());
leg->AddEntry(hist,optionLatex(opt).c_str(),"l");
}
int DTFcomparison(std::string year = "2011", std::string magnet = "down", bool MC = false, bool ReferenceChannel = false, bool PHSP = false, bool preselected = true, bool truthMatched = false) {
//no point in keepeng the MC option, but it doesn't hurt neither
TFile* output = 0;
TChain *tree;
if (!(MC || ReferenceChannel || PHSP)) truthMatched = false;
if (!preselected){
//Kst2Kpluspi0Resolved case
if(Kst2Kpluspi0Resolved){
if(!MC){
tree=new TChain("b2KstKpi0mumuResolvedTuple/DecayTree");
if(smallSample) tree->Add(Form("%s/data/%s%s/*B2Kstmumu*13.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else{
tree->Add(Form("%s/data/%s%s/*.root",path_to_data.c_str(),year.c_str(),magnet.c_str()));
std::cout << "Adding " << Form("%s/data/%s%s/*B2Kstmumu*.root",path_to_data.c_str(),year.c_str(),magnet.c_str()) << endl;
}
}
else{
tree=new TChain("b2KstKpi0mumuResolvedTuple/DecayTree");
if(ReferenceChannel){
if(smallSample) tree->Add(Form("%s/data/MC/RefKplusPi0/%s%s/*B2KstJpsi*1.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/MC/RefKplusPi0/%s%s/*B2KstJpsi*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
else if(PHSP){
if(smallSample) tree->Add(Form("%s/data/PHSP/KplusPi0/%s%s/*B2Kstmumu*1.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/PHSP/KplusPi0/%s%s/*B2Kstmumu*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
else{
if(smallSample) tree->Add(Form("%s/data/MC/KplusPi0/%s%s/*B2Kstmumu*1.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/MC/KplusPi0/%s%s/*B2Kstmumu*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
}
}
//Kst2Kpluspi0Merged case
if(Kst2Kpluspi0Merged){
tree=new TChain("b2KstKpi0mumuMergedTuple/DecayTree");
if(!MC){
if(smallSample) tree->Add(Form("%s/data/%s%s/*B2Kstmumu*13.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/%s%s/*B2Kstmumu*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
else{
if(smallSample) tree->Add(Form("%s/data/MC/KplusPi0/%s%s/*B2Kstmumu*1.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/MC/KplusPi0/%s%s/*B2Kstmumu*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
}
//Kst2Kspiplus case
if(Kst2Kspiplus){
if(!MC){
tree=new TChain("b2KstKs0pimumu_Tuple/DecayTree");
if(smallSample) tree->Add(Form("%s/data/%s%s/*B2Kstmumu*23.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/%s%s/*B2Kstmumu*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
else{
tree=new TChain("b2KstKs0pimumu_Tuple/DecayTree");
if(ReferenceChannel){
if(smallSample) tree->Add(Form("%s/data/MC/RefKshortPiplus/%s%s/*B2KstJpsi*13.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/MC/RefKshortPiplus/%s%s/*B2KstJpsi*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
else if(PHSP){
if(smallSample) tree->Add(Form("%s/data/PHSP/KshortPiplus/%s%s/*B2Kstmumu*1.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/PHSP/KshortPiplus/%s%s/*B2Kstmumu*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
else{
if(smallSample) tree->Add(Form("%s/data/MC/KshortPiplus/%s%s/*B2Kstmumu*1.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
else tree->Add(Form("%s/data/MC/KshortPiplus/%s%s/*B2Kstmumu*.root",path_to_data.c_str(), year.c_str(),magnet.c_str()));
}
}
}
}
else{ //preselected
string name = (truthMatched ? "DecayTreeTruthMatched" : "DecayTree");
tree=new TChain(name.c_str());
// Kst2Kpluspi0Resolved case
if(Kst2Kpluspi0Resolved){
if(!MC){
tree->Add(Form("%s/data/%s%s/%s%s_pi0Resolved.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
cout << Form("%s/data/%s%s/%s%s_pi0Resolved.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()) << endl;
}
else{
if(ReferenceChannel){
tree->Add(Form("%s/data/MC/RefKplusPi0/%s%s/%s%s_pi0Resolved.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
else if(PHSP){
tree->Add(Form("%s/data/PHSP/KplusPi0/%s%s/%s%s_pi0Resolved.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
else{
tree->Add(Form("%s/data/MC/KplusPi0/%s%s/%s%s_pi0Resolved.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
}
}
// Kst2Kpluspi0Merged case
if(Kst2Kpluspi0Merged){
if(!MC){
tree->Add(Form("%s/data/%s%s/%s%s_pi0Merged.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
else{
if(ReferenceChannel){
tree->Add(Form("%s/data/MC/RefKplusPi0/%s%s/%s%s_pi0Merged.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
else if(PHSP){
tree->Add(Form("%s/data/PHSP/KplusPi0/%s%s/%s%s_pi0Merged.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
else{
tree->Add(Form("%s/data/MC/KplusPi0/%s%s/%s%s_pi0Mergedroot",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
}
}
// Kst2Kspiplus case
if(Kst2Kspiplus){
if(!MC){
tree->Add(Form("%s/data/%s%s/%s%s_piplus.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
else{
if(ReferenceChannel){
tree->Add(Form("%s/data/MC/RefKshortPiplus/%s%s/%s%s_piplus.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
else if(PHSP){
tree->Add(Form("%s/data/PHSP/KshortPiplus/%s%s/%s%s_piplus.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
else{
tree->Add(Form("%s/data/MC/KshortPiplus/%s%s/%s%s_piplus.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str()));
}
}
}
}
int N = tree->GetEntries();
if(N == 0){
std::cout << "[ERROR]\t\tNo events found!" << std::endl;
return 0;
}
// MASSES HISTOGRAMS
string tmpName = "";
//pi0
tmpName = "m_{"+ particleLatex("pi0") +"}[MeV]";
TH1 * h_pi0_M_DTF_PDG= new TH1D("h_pi0_M_DTF_PDG", histName("pi0","M", "DTF_PDG").c_str(), 70, 100, 170);
h_pi0_M_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_M_DTF_PDG->GetYaxis()->SetTitle("Counts / 1MeV");
TH1 * h_pi0_M_DTF= new TH1D("h_pi0_M_DTF" , histName("pi0","M", "DTF").c_str(), 70, 100, 170);
h_pi0_M_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_M_DTF->GetYaxis()->SetTitle("Counts / 1MeV");
tmpName = "m_{"+ particleLatex("pizero") +"}[MeV]";
TH1 * h_pi0_M_PDG= new TH1D("h_pi0_M_PDG" , histName("pi0","M", "PDG").c_str(), 70, 100, 170);
h_pi0_M_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_M_PDG->GetYaxis()->SetTitle("Counts / 1MeV");
TH1 * h_pi0_M= new TH1D("h_pi0_M" , histName("pi0","M", "").c_str(), 70, 100, 170);
h_pi0_M->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_M->GetYaxis()->SetTitle("Counts / 1MeV");
//Ks
tmpName = "m_{" + particleLatex("Kshort") + "}[MeV]";
TH1 * h_Kshort_M_DTF_PDG= new TH1D("h_Kshort_M_DTF_PDG" , histName("Kshort","M", "DTF_PDG").c_str(), 100, 400, 600);
h_Kshort_M_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_M_DTF_PDG->GetYaxis()->SetTitle("Counts / 2MeV");
TH1 * h_Kshort_M_DTF= new TH1D("h_Kshort_M_DTF" , histName("Kshort","M", "DTF").c_str(), 100, 400, 600);
h_Kshort_M_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_M_DTF->GetYaxis()->SetTitle("Counts / 2MeV");
TH1 * h_Kshort_M_PDG= new TH1D("h_Kshort_M_PDG" ,histName("Kshort","M", "PDG").c_str(), 100, 400, 600);
h_Kshort_M_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_M_PDG->GetYaxis()->SetTitle("Counts / 2MeV");
TH1 * h_Kshort_M= new TH1D("h_Kshort_M" , histName("Kshort","M", "").c_str(), 100, 400, 600);
h_Kshort_M->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_M->GetYaxis()->SetTitle("Counts / 2MeV");
//K*
if (Kst2Kpluspi0Resolved) tmpName = "m_{" + particleLatex("Kplus") + particleLatex("pi0") + "}[MeV]";
else if (Kst2Kspiplus) tmpName = "m_{" + particleLatex("Kshort") + particleLatex("piplus") + "}[MeV]";
TH1 * h_Kstar_M_DTF_PDG= new TH1D("h_Kstar_M_DTF_PDG" ,histName("Kstar","M", "DTF_PDG").c_str(), 100, 700, 1100);
h_Kstar_M_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_M_DTF_PDG->GetYaxis()->SetTitle("Counts / 4MeV");
TH1 * h_Kstar_M_DTF= new TH1D("h_Kstar_M_DTF" , histName("Kstar","M", "DTF").c_str(), 100, 700, 1100);
h_Kstar_M_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_M_DTF->GetYaxis()->SetTitle("Counts / 4MeV");
if (Kst2Kpluspi0Resolved) tmpName = "m_{" + particleLatex("Kplus") + particleLatex("pizero") + "}[MeV]";
TH1 * h_Kstar_M_PDG= new TH1D("h_Kstar_M_PDG" , histName("Kstar","M", "PDG").c_str(), 100, 700, 1100);
h_Kstar_M_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_M_PDG->GetYaxis()->SetTitle("Counts / 4MeV");
TH1 * h_Kstar_M= new TH1D("h_Kstar_M" , histName("Kstar","M", "").c_str(), 100, 700, 1100);
h_Kstar_M->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_M->GetYaxis()->SetTitle("Counts / 4MeV");
//B+
if (Kst2Kpluspi0Resolved) tmpName = "m_{" + particleLatex("Kplus") + particleLatex("pi0") + particleLatex("mu") + particleLatex("mu") + "}[MeV]";
else if (Kst2Kspiplus) tmpName = "m_{" + particleLatex("Kshort") + particleLatex("piplus") + particleLatex("mu") + particleLatex("mu") + "}[MeV]";
TH1 * h_B_M_DTF_PDG= new TH1D("h_B_M_DTF_PDG" , histName("Bplus","M", "DTF_PDG").c_str(), (7000-cut_B_plus_M_low)/20, cut_B_plus_M_low, 7000);
h_B_M_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_B_M_DTF_PDG->GetYaxis()->SetTitle("Counts / 20MeV");
TH1 * h_B_M_DTF= new TH1D("h_B_M_DTF" , histName("Bplus","M", "DTF").c_str(), (7000-cut_B_plus_M_low)/20, cut_B_plus_M_low, 7000);
h_B_M_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_B_M_DTF->GetYaxis()->SetTitle("Counts / 20MeV");
if (Kst2Kpluspi0Resolved) tmpName = "m_{" + particleLatex("Kplus") + particleLatex("pizero") + particleLatex("mu") + particleLatex("mu") + "}[MeV]";
TH1 * h_B_M_PDG= new TH1D("h_B_M_PDG", histName("Bplus","M", "PDG").c_str(), (7000-cut_B_plus_M_low)/20, cut_B_plus_M_low, 7000);
h_B_M_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_B_M_PDG->GetYaxis()->SetTitle("Counts / 20MeV");
TH1 * h_B_M= new TH1D("h_B_M" , histName("Bplus","M", "").c_str(), (7000-cut_B_plus_M_low)/20, cut_B_plus_M_low, 7000);
h_B_M->GetXaxis()->SetTitle(tmpName.c_str());
h_B_M->GetYaxis()->SetTitle("Counts / 20MeV");
//pT HISTOGRAMS
//pi0
tmpName = variableLatex("PT") + "{"+ particleLatex("pi0") +"}[MeV]";
TH1 * h_pi0_PT_DTF_PDG= new TH1D("h_pi0_PT_DTF_PDG" , histName("pi0","PT", "DTF_PDG").c_str(), 100, 0, 5000);
h_pi0_PT_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_PT_DTF_PDG->GetYaxis()->SetTitle("Counts / 50MeV");
TH1 * h_pi0_PT_DTF= new TH1D("h_pi0_PT_DTF" , histName("pizero","PT", "DTF").c_str(),100, 0, 5000);
h_pi0_PT_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_PT_DTF->GetYaxis()->SetTitle("Counts / 50MeV");
TH1 * h_pi0_PT_PDG= new TH1D("h_pi0_PT_PDG" , histName("pi0","PT", "PDG").c_str(), 100, 0, 5000);;
h_pi0_PT_PDG->GetXaxis()->SetTitle("p_{T}{#pi^{0}[MeV]");
h_pi0_PT_PDG->GetYaxis()->SetTitle("Counts / 50MeV");
TH1 * h_pi0_PT= new TH1D("h_pi0_PT" , histName("pi0","PT", "").c_str(),100, 0, 5000);
h_pi0_PT->GetXaxis()->SetTitle("p_{T}{#pi^{0}[MeV]");
h_pi0_PT->GetYaxis()->SetTitle("Counts / 50MeV");
//Ks
tmpName = variableLatex("PT") + "{" + particleLatex("Kshort") + "}[MeV]";
TH1 * h_Kshort_PT_DTF_PDG= new TH1D("h_Kshort_PT_DTF_PDG" , histName("Kshort","PT", "DTF_PDG").c_str(), 100, 0, 15000);
h_Kshort_PT_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_PT_DTF_PDG->GetYaxis()->SetTitle("Counts / 150MeV");
TH1 * h_Kshort_PT_DTF= new TH1D("h_Kshort_PT_DTF" , histName("Kshort","PT", "DTF").c_str(), 100, 0, 15000);
h_Kshort_PT_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_PT_DTF->GetYaxis()->SetTitle("Counts / 150MeV");
TH1 * h_Kshort_PT_PDG= new TH1D("h_Kshort_PT_PDG" , histName("Kshort","PT", "PDG").c_str(), 100, 0, 15000);
h_Kshort_PT_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_PT_PDG->GetYaxis()->SetTitle("Counts / 150MeV");
TH1 * h_Kshort_PT= new TH1D("h_Kshort_PT" , histName("Kshort","PT", "").c_str(), 100, 0, 15000);
h_Kshort_PT->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_PT->GetYaxis()->SetTitle("Counts / 150MeV");
//K*
if (Kst2Kpluspi0Resolved) tmpName = variableLatex("PT") + "{" + particleLatex("Kplus") + particleLatex("pi0") + "}[MeV]";
else if (Kst2Kspiplus) tmpName = variableLatex("PT") + "{" + particleLatex("Kshort") + particleLatex("piplus") + "}[MeV]";
TH1 * h_Kstar_PT_DTF_PDG= new TH1D("h_Kstar_PT_DTF_PDG" , histName("Kstar","PT", "DTF_PDG").c_str(), 100, 0, 15000);
h_Kstar_PT_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_PT_DTF_PDG->GetYaxis()->SetTitle("Counts / 150MeV");
TH1 * h_Kstar_PT_DTF= new TH1D("h_Kstar_PT_DTF" , histName("Kstar","PT", "DTF").c_str(), 100, 0, 15000);
h_Kstar_PT_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_PT_DTF->GetYaxis()->SetTitle("Counts / 150MeV");
TH1 * h_Kstar_PT_PDG= new TH1D("h_Kstar_PT_PDG" , histName("Kstar","PT", "PDG").c_str(),100, 0, 15000);
h_Kstar_PT_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_PT_PDG->GetYaxis()->SetTitle("Counts / 150MeV");
TH1 * h_Kstar_PT= new TH1D("h_Kstar_PT" , histName("Kstar","PT", "").c_str(), 100, 0, 15000);
h_Kstar_PT->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_PT->GetYaxis()->SetTitle("Counts / 150MeV");
//B+
if (Kst2Kpluspi0Resolved) tmpName = variableLatex("PT") + "{" + particleLatex("Kplus") + particleLatex("pi0") + particleLatex("mu") + particleLatex("mu") + "}[MeV]";
else if (Kst2Kspiplus) tmpName = variableLatex("PT") + "{" + particleLatex("Kshort") + particleLatex("piplus") + particleLatex("mu") + particleLatex("mu") + "}[MeV]";
TH1 * h_B_PT_DTF_PDG= new TH1D("h_B_PT_DTF_PDG" , histName("Bplus","PT", "DTF_PDG").c_str(), 100, 0, 40000);
h_B_PT_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_B_PT_DTF_PDG->GetYaxis()->SetTitle("Counts / 400MeV");
TH1 * h_B_PT_DTF= new TH1D("h_B_PT_DTF" , histName("Bplus","PT", "DTF").c_str(), 100, 0, 40000);
h_B_PT_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_B_PT_DTF->GetYaxis()->SetTitle("Counts / 20MeV");
TH1 * h_B_PT_PDG= new TH1D("h_B_PT_PDG" , histName("Bplus","PT", "PDG").c_str(), 100, 0, 40000);
h_B_PT_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_B_PT_PDG->GetYaxis()->SetTitle("Counts / 20MeV");
TH1 * h_B_PT= new TH1D("h_B_PT" , histName("Bplus","PT", "").c_str(), 100, 0, 40000);
h_B_PT->GetXaxis()->SetTitle(tmpName.c_str());
h_B_PT->GetYaxis()->SetTitle("Counts / 20MeV");
//eta HISTOGRAMS #TODO: hist titles
//pi0
tmpName = variableLatex("ETA") + "{"+ particleLatex("pi0") +"}[MeV]";
TH1 * h_pi0_ETA_DTF_PDG= new TH1D("h_pi0_ETA_DTF_PDG" , histName("pi0","ETA", "DTF_PDG").c_str(), 35, 1.5, 5);
h_pi0_ETA_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_ETA_DTF_PDG->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_pi0_ETA_DTF= new TH1D("h_pi0_ETA_DTF" , histName("pi0","ETA", "DTF").c_str(), 35, 1.5, 5);
h_pi0_ETA_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_ETA_DTF->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_pi0_ETA_PDG= new TH1D("h_pi0_ETA_PDG" , histName("pi0","ETA", "PDG").c_str(), 35, 1.5, 5);
h_pi0_ETA_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_ETA_PDG->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_pi0_ETA= new TH1D("h_pi0_ETA" , histName("pi0","ETA", "").c_str(), 35, 1.5, 5);
h_pi0_ETA->GetXaxis()->SetTitle(tmpName.c_str());
h_pi0_ETA->GetYaxis()->SetTitle("Counts / 0.1");
//Ks
tmpName = variableLatex("ETA") + "{" + particleLatex("Kshort") + "}[MeV]";
TH1 * h_Kshort_ETA_DTF_PDG= new TH1D("h_Kshort_ETA_DTF_PDG" , histName("Kshort","ETA", "DTF_PDG").c_str(), 35, 1.5, 5);
h_Kshort_ETA_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_ETA_DTF_PDG->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_Kshort_ETA_DTF= new TH1D("h_Kshort_ETA_DTF" , histName("Kshort","ETA", "DTF").c_str(), 35, 1.5, 5);
h_Kshort_ETA_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_ETA_DTF->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_Kshort_ETA_PDG= new TH1D("h_Kshort_ETA_PDG" , histName("Kshort","ETA", "PDG").c_str(), 35, 1.5, 5);
h_Kshort_ETA_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_ETA_PDG->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_Kshort_ETA= new TH1D("h_Kshort_ETA" , histName("Kshort","ETA", "").c_str(), 35, 1.5, 5);
h_Kshort_ETA->GetXaxis()->SetTitle(tmpName.c_str());
h_Kshort_ETA->GetYaxis()->SetTitle("Counts / 0.1");
//K*
if (Kst2Kpluspi0Resolved) tmpName = variableLatex("ETA") + "{" + particleLatex("Kplus") + particleLatex("pi0") + "}[MeV]";
else if (Kst2Kspiplus) tmpName = variableLatex("ETA") + "{" + particleLatex("Kshort") + particleLatex("piplus") + "}[MeV]";
TH1 * h_Kstar_ETA_DTF_PDG= new TH1D("h_Kstar_ETA_DTF_PDG" , histName("Kstar","ETA", "DTF_PDG").c_str(), 35, 1.5, 5);
h_Kstar_ETA_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_ETA_DTF_PDG->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_Kstar_ETA_DTF= new TH1D("h_Kstar_ETA_DTF" , histName("Kstar","ETA", "DTF").c_str(), 35, 1.5, 5);
h_Kstar_ETA_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_ETA_DTF->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_Kstar_ETA_PDG= new TH1D("h_Kstar_ETA_PDG" , histName("Kstar","ETA", "PDG").c_str(), 35, 1.5, 5);
h_Kstar_ETA_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_ETA_PDG->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_Kstar_ETA= new TH1D("h_Kstar_ETA" , histName("Kstar","ETA", "").c_str(), 35, 1.5, 5);
h_Kstar_ETA->GetXaxis()->SetTitle(tmpName.c_str());
h_Kstar_ETA->GetYaxis()->SetTitle("Counts / 0.1");
//B+
if (Kst2Kpluspi0Resolved) tmpName = variableLatex("ETA") + "{" + particleLatex("Kplus") + particleLatex("pi0") + particleLatex("mu") + particleLatex("mu") + "}[MeV]";
else if (Kst2Kspiplus) tmpName = variableLatex("ETA") + "{" + particleLatex("Kshort") + particleLatex("piplus") + particleLatex("mu") + particleLatex("mu") + "}[MeV]";
TH1 * h_B_ETA_DTF_PDG= new TH1D("h_B_ETA_DTF_PDG" , histName("Bplus","ETA", "DTF_PDG").c_str(), 45, 1.5, 6);
h_B_ETA_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_B_ETA_DTF_PDG->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_B_ETA_DTF= new TH1D("h_B_ETA_DTF" , histName("Bplus","ETA", "DTF").c_str(), 45, 1.5, 6);
h_B_ETA_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_B_ETA_DTF->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_B_ETA_PDG= new TH1D("h_B_ETA_PDG" , histName("Bplus","ETA", "PDG").c_str(), 45, 1.5, 6);
h_B_ETA_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_B_ETA_PDG->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_B_ETA = new TH1D("h_B_ETA" , histName("Bplus","ETA", "").c_str(), 45, 1.5, 6);
h_B_ETA->GetXaxis()->SetTitle(tmpName.c_str());
h_B_ETA->GetYaxis()->SetTitle("Counts / 0.1");
//Angular description
TH1 * h_ctk = new TH1D("h_ctk" ,"#theta_{k}", 73, -3.15, 3.15);
h_ctk->GetXaxis()->SetTitle("#theta_{k}");
h_ctk->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_ctl = new TH1D("h_ctl" , "#theta_{l}", 73, -3.15, 3.15);
h_ctl->GetXaxis()->SetTitle("#theta_{l}");
h_ctl->GetYaxis()->SetTitle("Counts / 0.1");
TH1 * h_sp = new TH1D("h_sp" ,"#phi", 73, -3.15, 3.15);
h_sp->GetXaxis()->SetTitle("#phi");
h_sp->GetYaxis()->SetTitle("Counts / 0.1");
//Open Angle
tmpName ="Open Angle";
TH1 * h_OpenAngle_DTF_PDG= new TH1D("h_OpenAngle_DTF_PDG" , ("Open Angle" + optionLatex("DTF_PDG")).c_str(), 40, 0, 0.2);
h_OpenAngle_DTF_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_OpenAngle_DTF_PDG->GetYaxis()->SetTitle("Counts / 0.005");
TH1 * h_OpenAngle_DTF= new TH1D("h_OpenAngle_DTF" , ("Open Angle" + optionLatex("DTF")).c_str(), 40, 0, 0.2);
h_OpenAngle_DTF->GetXaxis()->SetTitle(tmpName.c_str());
h_OpenAngle_DTF->GetYaxis()->SetTitle("Counts / 0.005");
TH1 * h_OpenAngle_PDG= new TH1D("h_OpenAngle_PDG" , ("Open Angle" + optionLatex("PDG")).c_str(), 40, 0, 0.2);
h_OpenAngle_PDG->GetXaxis()->SetTitle(tmpName.c_str());
h_OpenAngle_PDG->GetYaxis()->SetTitle("Counts / 0.005");
TH1 * h_OpenAngle = new TH1D("h_OpenAngle" , ("Open Angle" + optionLatex("")).c_str(), 40, 0, 0.2);
h_OpenAngle->GetXaxis()->SetTitle(tmpName.c_str());
h_OpenAngle->GetYaxis()->SetTitle("Counts / 0.05");
/////////////////////
// Detector values //
/////////////////////
TLorentzVector LorVec_B_plus;
TLorentzVector LorVec_K_star_plus;
TLorentzVector LorVec_K_plus;
TLorentzVector LorVec_pi_zero_merged;
TLorentzVector LorVec_pi_zero_resolved;
TLorentzVector LorVec_K_short;
TLorentzVector LorVec_pi_plus;
TLorentzVector LorVec_KS_pi_plus;
TLorentzVector LorVec_KS_pi_minus;
TLorentzVector LorVec_mu_minus;
TLorentzVector LorVec_mu_plus;
TLorentzVector LorVec_gamma1;
TLorentzVector LorVec_gamma2;
TLorentzVector LorVec_DiGamma;
//potentially misIDed particles:
TLorentzVector LorVec_pi_plus_MisIDed;
TLorentzVector LorVec_mu_plus_MisIDed;
TLorentzVector LorVec_B_plus_From_DoubleMisIDed;
TLorentzVector LorVec_Jpsi_From_DoubleMisIDed;
TLorentzVector LorVec_K_star_plus_From_DoubleMisIDed;
//recombined particles:
TLorentzVector LorVec_DiMuon;
TLorentzVector LorVec_KaonPion;
TLorentzVector LorVec_ReconB;
TLorentzVector LorVec_ReconBPiZero;
TLorentzVector LorVec_KshortMuMu;
TLorentzVector LorVec_KplusMuMu;
//Other variables:
Double_t OpenAngle;
/////////////////////
// DTF values //
/////////////////////
TLorentzVector LorVec_DTF_B_plus;
TLorentzVector LorVec_DTF_K_star_plus;
TLorentzVector LorVec_DTF_K_plus;
TLorentzVector LorVec_DTF_pi_zero_merged;
TLorentzVector LorVec_DTF_pi_zero_resolved;
TLorentzVector LorVec_DTF_K_short;
TLorentzVector LorVec_DTF_pi_plus;
TLorentzVector LorVec_DTF_KS_pi_plus;
TLorentzVector LorVec_DTF_KS_pi_minus;
TLorentzVector LorVec_DTF_mu_minus;
TLorentzVector LorVec_DTF_mu_plus;
TLorentzVector LorVec_DTF_gamma1;
TLorentzVector LorVec_DTF_gamma2;
TLorentzVector LorVec_DTF_DiGamma;
//potentially misIDed particles:
TLorentzVector LorVec_DTF_pi_plus_MisIDed;
TLorentzVector LorVec_DTF_mu_plus_MisIDed;
TLorentzVector LorVec_DTF_B_plus_From_DoubleMisIDed;
TLorentzVector LorVec_DTF_Jpsi_From_DoubleMisIDed;
TLorentzVector LorVec_DTF_K_star_plus_From_DoubleMisIDed;
//recombined particles:
TLorentzVector LorVec_DTF_DiMuon;
TLorentzVector LorVec_DTF_KaonPion;
TLorentzVector LorVec_DTF_ReconB;
TLorentzVector LorVec_DTF_ReconBPiZero;
TLorentzVector LorVec_DTF_KshortMuMu;
TLorentzVector LorVec_DTF_KplusMuMu;
//Other variables:
Double_t OpenAngle_DTF;
/////////////////////
// PDG+DTF values //
/////////////////////
TLorentzVector LorVec_DTF_PDG_B_plus;
TLorentzVector LorVec_DTF_PDG_K_star_plus;
TLorentzVector LorVec_DTF_PDG_K_plus;
TLorentzVector LorVec_DTF_PDG_pi_zero_merged;
TLorentzVector LorVec_DTF_PDG_pi_zero_resolved;
TLorentzVector LorVec_DTF_PDG_K_short;
TLorentzVector LorVec_DTF_PDG_pi_plus;
TLorentzVector LorVec_DTF_PDG_KS_pi_plus;
TLorentzVector LorVec_DTF_PDG_KS_pi_minus;
TLorentzVector LorVec_DTF_PDG_mu_minus;
TLorentzVector LorVec_DTF_PDG_mu_plus;
TLorentzVector LorVec_DTF_PDG_gamma1;
TLorentzVector LorVec_DTF_PDG_gamma2;
TLorentzVector LorVec_DTF_PDG_DiGamma;
//potentially misIDed particles:
TLorentzVector LorVec_DTF_PDG_pi_plus_MisIDed;
TLorentzVector LorVec_DTF_PDG_mu_plus_MisIDed;
TLorentzVector LorVec_DTF_PDG_B_plus_From_DoubleMisIDed;
TLorentzVector LorVec_DTF_PDG_Jpsi_From_DoubleMisIDed;
TLorentzVector LorVec_DTF_PDG_K_star_plus_From_DoubleMisIDed;
//recombined particles:
TLorentzVector LorVec_DTF_PDG_DiMuon;
TLorentzVector LorVec_DTF_PDG_KaonPion;
TLorentzVector LorVec_DTF_PDG_ReconB;
TLorentzVector LorVec_DTF_PDG_ReconBPiZero;
TLorentzVector LorVec_DTF_PDG_KshortMuMu;
TLorentzVector LorVec_DTF_PDG_KplusMuMu;
//Other variables:
Double_t OpenAngle_DTF_PDG;
/////////////////////
// PDG values //
/////////////////////
TLorentzVector LorVec_PDG_B_plus;
TLorentzVector LorVec_PDG_K_star_plus;
TLorentzVector LorVec_PDG_K_plus;
TLorentzVector LorVec_PDG_pi_zero_merged;
TLorentzVector LorVec_PDG_pi_zero_resolved;
TLorentzVector LorVec_PDG_K_short;
TLorentzVector LorVec_PDG_pi_plus;
TLorentzVector LorVec_PDG_KS_pi_plus;
TLorentzVector LorVec_PDG_KS_pi_minus;
TLorentzVector LorVec_PDG_mu_minus;
TLorentzVector LorVec_PDG_mu_plus;
TLorentzVector LorVec_PDG_gamma1;
TLorentzVector LorVec_PDG_gamma2;
TLorentzVector LorVec_PDG_DiGamma;
//potentially misIDed particles:
TLorentzVector LorVec_PDG_pi_plus_MisIDed;
TLorentzVector LorVec_PDG_mu_plus_MisIDed;
TLorentzVector LorVec_PDG_B_plus_From_DoubleMisIDed;
TLorentzVector LorVec_PDG_Jpsi_From_DoubleMisIDed;
TLorentzVector LorVec_PDG_K_star_plus_From_DoubleMisIDed;
//recombined particles:
TLorentzVector LorVec_PDG_DiMuon;
TLorentzVector LorVec_PDG_KaonPion;
TLorentzVector LorVec_PDG_ReconB;
TLorentzVector LorVec_PDG_ReconBPiZero;
TLorentzVector LorVec_PDG_KshortMuMu;
TLorentzVector LorVec_PDG_KplusMuMu;
//Other variables:
Double_t OpenAngle_PDG;
/////////////////////
// Branches //
/////////////////////
tree->SetBranchStatus("*",0);
tree->SetBranchStatus( "*P", 1 );
tree->SetBranchStatus( "*PX", 1 );
tree->SetBranchStatus( "*PY", 1);
tree->SetBranchStatus( "*PZ", 1);
tree->SetBranchStatus( "*PE", 1);
tree->SetBranchStatus( "*PT", 1 );
tree->SetBranchStatus( "*TAU", 1 );
tree->SetBranchStatus( "*Phi",1 );
tree->SetBranchStatus( "*ThetaL",1 );
tree->SetBranchStatus( "*ThetaK",1 );
///B+
Double_t B_plus_PX = 0.;
Double_t B_plus_PY = 0.;
Double_t B_plus_PZ = 0.;
Double_t B_plus_PE = 0.;
///K*+
Double_t K_star_plus_PX = 0.;
Double_t K_star_plus_PY = 0.;
Double_t K_star_plus_PZ = 0.;
Double_t K_star_plus_PE = 0.;
///K+
Double_t K_plus_PX = 0.;
Double_t K_plus_PY = 0.;
Double_t K_plus_PZ = 0.;
Double_t K_plus_PE = 0.;
Float_t K_plus_PX_DTF[100];
Float_t K_plus_PY_DTF[100];
Float_t K_plus_PZ_DTF[100];
Float_t K_plus_PE_DTF[100];
///pi0 (merged)
Double_t pi_zero_merged_PX = 0.;
Double_t pi_zero_merged_PY = 0.;
Double_t pi_zero_merged_PZ = 0.;
Double_t pi_zero_merged_PE = 0.;
///pi0 (resolved)
Double_t pi_zero_resolved_PX = 0.;
Double_t pi_zero_resolved_PY = 0.;
Double_t pi_zero_resolved_PZ = 0.;
Double_t pi_zero_resolved_PE = 0.;
///Kshort
Double_t K_short_PX = 0.;
Double_t K_short_PY = 0.;
Double_t K_short_PZ = 0.;
Double_t K_short_PE = 0.;
///pi+ (KS)
Double_t Ks_pi_plus_PX = 0.;
Double_t Ks_pi_plus_PY = 0.;
Double_t Ks_pi_plus_PZ = 0.;
Double_t Ks_pi_plus_PE = 0.;
Float_t Ks_pi_plus_PX_DTF[100];
Float_t Ks_pi_plus_PY_DTF[100];
Float_t Ks_pi_plus_PZ_DTF[100];
Float_t Ks_pi_plus_PE_DTF[100];
///pi- (KS)
Double_t Ks_pi_minus_PX = 0.;
Double_t Ks_pi_minus_PY = 0.;
Double_t Ks_pi_minus_PZ = 0.;
Double_t Ks_pi_minus_PE = 0.;
Float_t Ks_pi_minus_PX_DTF[100];
Float_t Ks_pi_minus_PY_DTF[100];
Float_t Ks_pi_minus_PZ_DTF[100];
Float_t Ks_pi_minus_PE_DTF[100];
///pi+
Double_t pi_plus_PX = 0.;
Double_t pi_plus_PY = 0.;
Double_t pi_plus_PZ = 0.;
Double_t pi_plus_PE = 0.;
Float_t pi_plus_PX_DTF[100];
Float_t pi_plus_PY_DTF[100];
Float_t pi_plus_PZ_DTF[100];
Float_t pi_plus_PE_DTF[100];
///mu+
Double_t mu_minus_PX = 0.;
Double_t mu_minus_PY = 0.;
Double_t mu_minus_PZ = 0.;
Double_t mu_minus_PE = 0.;
Float_t mu_minus_PX_DTF[100];
Float_t mu_minus_PY_DTF[100];
Float_t mu_minus_PZ_DTF[100];
Float_t mu_minus_PE_DTF[100];
///mu+
Double_t mu_plus_PX = 0.;
Double_t mu_plus_PY = 0.;
Double_t mu_plus_PZ = 0.;
Double_t mu_plus_PE = 0.;
Float_t mu_plus_PX_DTF[100];
Float_t mu_plus_PY_DTF[100];
Float_t mu_plus_PZ_DTF[100];
Float_t mu_plus_PE_DTF[100];
///gamma1
Double_t gamma1_PX = 0.;
Double_t gamma1_PY = 0.;
Double_t gamma1_PZ = 0.;
Double_t gamma1_PE = 0.;
Float_t gamma1_PX_DTF[100];
Float_t gamma1_PY_DTF[100];
Float_t gamma1_PZ_DTF[100];
Float_t gamma1_PE_DTF[100];
///gamma2
Double_t gamma2_PX = 0.;
Double_t gamma2_PY = 0.;
Double_t gamma2_PZ = 0.;
Double_t gamma2_PE = 0.;
Float_t gamma2_PX_DTF[100];
Float_t gamma2_PY_DTF[100];
Float_t gamma2_PZ_DTF[100];
Float_t gamma2_PE_DTF[100];
//angles
Double_t ThetaL = 0.;
Double_t ThetaK = 0.;
Double_t Phi = 0.;
//particle masses and momenta
tree -> SetBranchAddress( "B_plus_PX" , &B_plus_PX );
tree -> SetBranchAddress( "B_plus_PY" , &B_plus_PY );
tree -> SetBranchAddress( "B_plus_PZ" , &B_plus_PZ );
tree -> SetBranchAddress( "B_plus_PE" , &B_plus_PE );
tree -> SetBranchAddress( "K_star_plus_PX" , &K_star_plus_PX );
tree -> SetBranchAddress( "K_star_plus_PY" , &K_star_plus_PY );
tree -> SetBranchAddress( "K_star_plus_PZ" , &K_star_plus_PZ );
tree -> SetBranchAddress( "K_star_plus_PE" , &K_star_plus_PE );
if(Kst2Kpluspi0Merged){ //TODO: in case one really wants to do that, add DTF info
tree -> SetBranchAddress( "pi_zero_merged_PX" , &pi_zero_merged_PX );
tree -> SetBranchAddress( "pi_zero_merged_PY" , &pi_zero_merged_PY );
tree -> SetBranchAddress( "pi_zero_merged_PZ" , &pi_zero_merged_PZ );
tree -> SetBranchAddress( "pi_zero_merged_PE" , &pi_zero_merged_PE );
}
if(Kst2Kpluspi0Resolved){
tree -> SetBranchAddress( "pi_zero_resolved_PX" , &pi_zero_resolved_PX );
tree -> SetBranchAddress( "pi_zero_resolved_PY" , &pi_zero_resolved_PY );
tree -> SetBranchAddress( "pi_zero_resolved_PZ" , &pi_zero_resolved_PZ );
tree -> SetBranchAddress( "pi_zero_resolved_PE" , &pi_zero_resolved_PE );
tree -> SetBranchAddress( "gamma1_PX" , &gamma1_PX );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_pi0_gamma_0_PX" , &gamma1_PX_DTF );
tree -> SetBranchAddress( "gamma1_PY" , &gamma1_PY );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_pi0_gamma_0_PY" , &gamma1_PY_DTF );
tree -> SetBranchAddress( "gamma1_PZ" , &gamma1_PZ );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_pi0_gamma_0_PZ" , &gamma1_PZ_DTF );
tree -> SetBranchAddress( "gamma1_PE" , &gamma1_PE );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_pi0_gamma_0_PE" , &gamma1_PE_DTF );
tree -> SetBranchAddress( "gamma2_PX" , &gamma2_PX );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_pi0_gamma_PX" , &gamma2_PX_DTF );
tree -> SetBranchAddress( "gamma2_PY" , &gamma2_PY );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_pi0_gamma_PY" , &gamma2_PY_DTF );
tree -> SetBranchAddress( "gamma2_PZ" , &gamma2_PZ );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_pi0_gamma_PZ" , &gamma2_PZ_DTF );
tree -> SetBranchAddress( "gamma2_PE" , &gamma2_PE );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_pi0_gamma_PE" , &gamma2_PE_DTF );
}
if(Kst2Kpluspi0Merged || Kst2Kpluspi0Resolved){
tree -> SetBranchAddress( "K_plus_PX" , &K_plus_PX );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_Kplus_PX" , &K_plus_PX_DTF );
tree -> SetBranchAddress( "K_plus_PY" , &K_plus_PY );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_Kplus_PY" , &K_plus_PY_DTF );
tree -> SetBranchAddress( "K_plus_PZ" , &K_plus_PZ );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_Kplus_PZ" , &K_plus_PZ_DTF );
tree -> SetBranchAddress( "K_plus_PE" , &K_plus_PE );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_Kplus_PE" , &K_plus_PE_DTF );
}
if(Kst2Kspiplus){
tree -> SetBranchAddress( "K_short_PX" , &K_short_PX );
tree -> SetBranchAddress( "K_short_PY" , &K_short_PY );
tree -> SetBranchAddress( "K_short_PZ" , &K_short_PZ );
tree -> SetBranchAddress( "K_short_PE" , &K_short_PE );
tree -> SetBranchAddress( "pi_plus_PX" , &pi_plus_PX );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_piplus_PX" , &pi_plus_PX_DTF );
tree -> SetBranchAddress( "pi_plus_PY" , &pi_plus_PY );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_piplus_PY" , &pi_plus_PY_DTF );
tree -> SetBranchAddress( "pi_plus_PZ" , &pi_plus_PZ );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_piplus_PZ" , &pi_plus_PZ_DTF );
tree -> SetBranchAddress( "pi_plus_PE" , &pi_plus_PE );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_piplus_PE" , &pi_plus_PE_DTF );
tree -> SetBranchAddress( "Ks_pi_minus_PX" , &Ks_pi_minus_PX );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_KS0_piplus_PX" , &Ks_pi_minus_PX_DTF );
tree -> SetBranchAddress( "Ks_pi_minus_PY" , &Ks_pi_minus_PY );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_KS0_piplus_PY" , &Ks_pi_minus_PY_DTF );
tree -> SetBranchAddress( "Ks_pi_minus_PZ" , &Ks_pi_minus_PZ );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_KS0_piplus_PZ" , &Ks_pi_minus_PZ_DTF );
tree -> SetBranchAddress( "Ks_pi_minus_PE" , &Ks_pi_minus_PE );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_KS0_piplus_PE" , &Ks_pi_minus_PE_DTF );
tree -> SetBranchAddress( "Ks_pi_plus_PX" , &Ks_pi_plus_PX );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_KS0_piplus_0_PX" , &Ks_pi_plus_PX_DTF );
tree -> SetBranchAddress( "Ks_pi_plus_PY" , &Ks_pi_plus_PY );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_KS0_piplus_0_PY" , &Ks_pi_plus_PY_DTF );
tree -> SetBranchAddress( "Ks_pi_plus_PZ" , &Ks_pi_plus_PZ );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_KS0_piplus_0_PZ" , &Ks_pi_plus_PZ_DTF );
tree -> SetBranchAddress( "Ks_pi_plus_PE" , &Ks_pi_plus_PE );
tree -> SetBranchAddress( "B_plus_DTF_Kst_892_plus_KS0_piplus_0_PE" , &Ks_pi_plus_PE_DTF );
}
tree -> SetBranchAddress( "mu_minus_PX" , &mu_minus_PX );
tree -> SetBranchAddress( "B_plus_DTF_J_psi_1S_muminus_0_PX" , &mu_minus_PX_DTF );
tree -> SetBranchAddress( "mu_minus_PY" , &mu_minus_PY );
tree -> SetBranchAddress( "B_plus_DTF_J_psi_1S_muminus_0_PY" , &mu_minus_PY_DTF );
tree -> SetBranchAddress( "mu_minus_PZ" , &mu_minus_PZ );
tree -> SetBranchAddress( "B_plus_DTF_J_psi_1S_muminus_0_PZ" , &mu_minus_PZ_DTF );
tree -> SetBranchAddress( "mu_minus_PE" , &mu_minus_PE );
tree -> SetBranchAddress( "B_plus_DTF_J_psi_1S_muminus_0_PE" , &mu_minus_PE_DTF );
tree -> SetBranchAddress( "mu_plus_PX" , &mu_plus_PX );
tree -> SetBranchAddress( "B_plus_DTF_J_psi_1S_muminus_PX" , &mu_plus_PX_DTF );
tree -> SetBranchAddress( "mu_plus_PY" , &mu_plus_PY );
tree -> SetBranchAddress( "B_plus_DTF_J_psi_1S_muminus_PY" , &mu_plus_PY_DTF );
tree -> SetBranchAddress( "mu_plus_PZ" , &mu_plus_PZ );
tree -> SetBranchAddress( "B_plus_DTF_J_psi_1S_muminus_PZ" , &mu_plus_PZ_DTF );
tree -> SetBranchAddress( "mu_plus_PE" , &mu_plus_PE );
tree -> SetBranchAddress( "B_plus_DTF_J_psi_1S_muminus_PE" , &mu_plus_PE_DTF );
tree -> SetBranchAddress( "B_plus_ThetaL" , &ThetaL);
tree -> SetBranchAddress( "B_plus_ThetaK" , &ThetaK);
tree -> SetBranchAddress( "B_plus_Phi" , &Phi );
//////////////////////////////////
///// //////
//// Loop over events /////
///// //////
//////////////////////////////////
for (int i=0; i<N; i++){
//output the different counters!
if ((0ul == (i % 10000ul) || i + 1 == N) && i != 0) {
cout << "Read event " << i + 1 << "/" << N
<< endl;
}
tree->GetEntry(i);
/////////////////////
// PDG+DTF values //
/////////////////////
LorVec_DTF_PDG_mu_minus.SetXYZM (mu_minus_PX_DTF[0], mu_minus_PY_DTF[0], mu_minus_PZ_DTF[0], PDGMASS.MU);
LorVec_DTF_PDG_mu_plus.SetXYZM (mu_plus_PX_DTF[0], mu_plus_PY_DTF[0], mu_plus_PZ_DTF[0], PDGMASS.MU);
if(Kst2Kspiplus){
/// Ks, pi+
LorVec_DTF_PDG_pi_plus.SetXYZM (pi_plus_PX_DTF[0], pi_plus_PY_DTF[0], pi_plus_PZ_DTF[0], PDGMASS.PI_PLUS);
LorVec_DTF_PDG_KS_pi_plus.SetXYZM (Ks_pi_plus_PX_DTF[0], Ks_pi_plus_PY_DTF[0], Ks_pi_plus_PZ_DTF[0], PDGMASS.PI_PLUS);
LorVec_DTF_PDG_KS_pi_minus.SetXYZM (Ks_pi_minus_PX_DTF[0], Ks_pi_minus_PY_DTF[0], Ks_pi_minus_PZ_DTF[0], PDGMASS.PI_PLUS);
LorVec_DTF_PDG_pi_plus_MisIDed.SetXYZM (pi_plus_PX_DTF[0], pi_plus_PY_DTF[0], pi_plus_PZ_DTF[0], PDGMASS.MU);
LorVec_DTF_PDG_mu_plus_MisIDed.SetXYZM (mu_plus_PX_DTF[0], mu_plus_PY_DTF[0], mu_plus_PZ_DTF[0], PDGMASS.PI_PLUS);
}
else{
/// K+, pi0
LorVec_DTF_PDG_K_plus.SetXYZM (K_plus_PX_DTF[0], K_plus_PY_DTF[0], K_plus_PZ_DTF[0], PDGMASS.K_PLUS);
LorVec_DTF_PDG_gamma1.SetXYZM (gamma1_PX_DTF[0], gamma1_PY_DTF[0], gamma1_PZ_DTF[0], PDGMASS.GAMMA);
LorVec_DTF_PDG_gamma2.SetXYZM (gamma2_PX_DTF[0], gamma2_PY_DTF[0], gamma2_PZ_DTF[0], PDGMASS.GAMMA);
}
//directly recombine the non-stable particles for later use (K*+ not really needed/used for Kst2Kspiplus)
if(Kst2Kspiplus){
LorVec_DTF_PDG_K_short = LorVec_DTF_PDG_KS_pi_plus + LorVec_DTF_PDG_KS_pi_minus;
LorVec_DTF_PDG_KaonPion = LorVec_DTF_PDG_K_short + LorVec_DTF_PDG_pi_plus;
}
if(Kst2Kpluspi0Resolved){
LorVec_DTF_PDG_DiGamma = LorVec_DTF_PDG_gamma1 + LorVec_DTF_PDG_gamma2;
LorVec_DTF_PDG_pi_zero_resolved = LorVec_DTF_PDG_DiGamma;
LorVec_DTF_PDG_KaonPion = LorVec_DTF_PDG_K_plus + LorVec_DTF_PDG_pi_zero_resolved;
}
LorVec_DTF_PDG_K_star_plus = LorVec_DTF_PDG_KaonPion;
LorVec_DTF_PDG_DiMuon = LorVec_DTF_PDG_mu_minus + LorVec_DTF_PDG_mu_plus;
LorVec_DTF_PDG_B_plus = LorVec_DTF_PDG_K_star_plus + LorVec_DTF_PDG_DiMuon;
/////////////////////
// DTF-values //
/////////////////////
LorVec_DTF_mu_minus = TLorentzVector (mu_minus_PX_DTF[0], mu_minus_PY_DTF[0], mu_minus_PZ_DTF[0], mu_minus_PE_DTF[0]);
LorVec_DTF_mu_plus = TLorentzVector (mu_plus_PX_DTF[0], mu_plus_PY_DTF[0], mu_plus_PZ_DTF[0], mu_plus_PE_DTF[0]);
if(Kst2Kspiplus){
/// Ks, pi+
LorVec_DTF_pi_plus = TLorentzVector (pi_plus_PX_DTF[0], pi_plus_PY_DTF[0], pi_plus_PZ_DTF[0], pi_plus_PE_DTF[0]);
LorVec_DTF_KS_pi_plus = TLorentzVector (Ks_pi_plus_PX_DTF[0], Ks_pi_plus_PY_DTF[0], Ks_pi_plus_PZ_DTF[0], Ks_pi_plus_PE_DTF[0]);
LorVec_DTF_KS_pi_minus = TLorentzVector (Ks_pi_minus_PX_DTF[0], Ks_pi_minus_PY_DTF[0], Ks_pi_minus_PZ_DTF[0], Ks_pi_minus_PE_DTF[0]);
LorVec_DTF_pi_plus_MisIDed.SetXYZM (pi_plus_PX_DTF[0], pi_plus_PY_DTF[0], pi_plus_PZ_DTF[0], PDGMASS.MU);
LorVec_DTF_mu_plus_MisIDed.SetXYZM (mu_plus_PX_DTF[0], mu_plus_PY_DTF[0], mu_plus_PZ_DTF[0], PDGMASS.PI_PLUS);
}
else{
/// K+, pi0
LorVec_DTF_K_plus = TLorentzVector (K_plus_PX_DTF[0], K_plus_PY_DTF[0], K_plus_PZ_DTF[0], K_plus_PE_DTF[0]);
LorVec_DTF_gamma1 = TLorentzVector (gamma1_PX_DTF[0], gamma1_PY_DTF[0], gamma1_PZ_DTF[0], gamma1_PE_DTF[0]);
LorVec_DTF_gamma2 = TLorentzVector (gamma2_PX_DTF[0], gamma2_PY_DTF[0], gamma2_PZ_DTF[0], gamma2_PE_DTF[0]);
}
//directly recombine the non-stable particles for later use (K*+ not really needed/used for Kst2Kspiplus)
if(Kst2Kspiplus){
LorVec_DTF_K_short = LorVec_DTF_KS_pi_plus + LorVec_DTF_KS_pi_minus;
LorVec_DTF_KaonPion = LorVec_DTF_K_short + LorVec_DTF_pi_plus;
}
if(Kst2Kpluspi0Resolved){
LorVec_DTF_DiGamma = LorVec_DTF_gamma1 + LorVec_DTF_gamma2;
LorVec_DTF_pi_zero_resolved = LorVec_DTF_DiGamma;
LorVec_DTF_KaonPion = LorVec_DTF_K_plus + LorVec_DTF_pi_zero_resolved;
}
LorVec_DTF_K_star_plus = LorVec_DTF_KaonPion;
LorVec_DTF_DiMuon = LorVec_DTF_mu_minus + LorVec_DTF_mu_plus;
LorVec_DTF_B_plus = LorVec_DTF_K_star_plus + LorVec_DTF_DiMuon;
/////////////////////
// PDG values //
/////////////////////
LorVec_PDG_B_plus.SetXYZM (B_plus_PX, B_plus_PY, B_plus_PZ, PDGMASS.B_PLUS);
LorVec_PDG_K_star_plus.SetXYZM (K_star_plus_PX, K_star_plus_PY, K_star_plus_PZ, PDGMASS.K_STAR_PLUS);
LorVec_PDG_mu_minus.SetXYZM (mu_minus_PX, mu_minus_PY, mu_minus_PZ, PDGMASS.MU);
LorVec_PDG_mu_plus.SetXYZM (mu_plus_PX, mu_plus_PY, mu_plus_PZ, PDGMASS.MU);
if(Kst2Kspiplus){
/// Ks, pi+
LorVec_PDG_K_short.SetXYZM (K_short_PX, K_short_PY, K_short_PZ, PDGMASS.K_SHORT);
LorVec_PDG_pi_plus.SetXYZM (pi_plus_PX, pi_plus_PY, pi_plus_PZ, PDGMASS.PI_PLUS);
LorVec_PDG_KS_pi_plus.SetXYZM (Ks_pi_plus_PX, Ks_pi_plus_PY, Ks_pi_plus_PZ, PDGMASS.PI_PLUS);
LorVec_PDG_KS_pi_minus.SetXYZM (Ks_pi_minus_PX, Ks_pi_minus_PY, Ks_pi_minus_PZ, PDGMASS.PI_PLUS);
LorVec_PDG_pi_plus_MisIDed.SetXYZM (pi_plus_PX, pi_plus_PY, pi_plus_PZ, PDGMASS.MU);
LorVec_PDG_mu_plus_MisIDed.SetXYZM (mu_plus_PX, mu_plus_PY, mu_plus_PZ, PDGMASS.PI_PLUS);
}
else{
/// K+, pi0
LorVec_PDG_K_plus.SetXYZM (K_plus_PX, K_plus_PY, K_plus_PZ, PDGMASS.K_PLUS);
LorVec_PDG_pi_zero_merged.SetXYZM (pi_zero_merged_PX, pi_zero_merged_PY, pi_zero_merged_PZ, PDGMASS.PI_ZERO);
LorVec_PDG_pi_zero_resolved.SetXYZM (pi_zero_resolved_PX, pi_zero_resolved_PY, pi_zero_resolved_PZ, PDGMASS.PI_ZERO);
LorVec_PDG_gamma1.SetXYZM (gamma1_PX, gamma1_PY, gamma1_PZ, PDGMASS.GAMMA);
LorVec_PDG_gamma2.SetXYZM (gamma2_PX, gamma2_PY, gamma2_PZ, PDGMASS.GAMMA);
}
//Recombination of 'Parent'-Particles
LorVec_PDG_DiMuon = LorVec_PDG_mu_minus + LorVec_PDG_mu_plus;
if(Kst2Kspiplus){
LorVec_PDG_KaonPion = LorVec_PDG_K_short + LorVec_PDG_pi_plus;
}
if(Kst2Kpluspi0Merged){
LorVec_PDG_KaonPion = LorVec_PDG_K_plus + LorVec_PDG_pi_zero_merged;
}
if(Kst2Kpluspi0Resolved){ //todo: save both dtf and fixed masses
LorVec_PDG_DiGamma = LorVec_PDG_gamma1 + LorVec_PDG_gamma2;
LorVec_PDG_KaonPion = LorVec_PDG_K_plus + LorVec_PDG_DiGamma;
}
//////////////////////
// detector values //
//////////////////////
LorVec_B_plus = TLorentzVector (B_plus_PX, B_plus_PY, B_plus_PZ, B_plus_PE);
LorVec_K_star_plus = TLorentzVector (K_star_plus_PX, K_star_plus_PY, K_star_plus_PZ, K_star_plus_PE);
LorVec_mu_minus = TLorentzVector (mu_minus_PX, mu_minus_PY, mu_minus_PZ, mu_minus_PE);
LorVec_mu_plus = TLorentzVector (mu_plus_PX, mu_plus_PY, mu_plus_PZ, mu_plus_PE);
if(Kst2Kspiplus){
/// Ks, pi+
LorVec_K_short = TLorentzVector (K_short_PX, K_short_PY, K_short_PZ, K_short_PE);
LorVec_pi_plus = TLorentzVector (pi_plus_PX, pi_plus_PY, pi_plus_PZ, pi_plus_PE);
LorVec_KS_pi_plus = TLorentzVector (Ks_pi_plus_PX, Ks_pi_plus_PY, Ks_pi_plus_PZ, Ks_pi_plus_PE);
LorVec_KS_pi_minus = TLorentzVector (Ks_pi_minus_PX, Ks_pi_minus_PY, Ks_pi_minus_PZ, Ks_pi_minus_PE);
LorVec_pi_plus_MisIDed.SetXYZM (pi_plus_PX, pi_plus_PY, pi_plus_PZ, PDGMASS.MU);
LorVec_mu_plus_MisIDed.SetXYZM (mu_plus_PX, mu_plus_PY, mu_plus_PZ, PDGMASS.PI_PLUS);
}
else{
/// K+, pi0
LorVec_K_plus = TLorentzVector (K_plus_PX, K_plus_PY, K_plus_PZ, K_plus_PE);
LorVec_pi_zero_merged = TLorentzVector (pi_zero_merged_PX, pi_zero_merged_PY, pi_zero_merged_PZ, pi_zero_merged_PE);
LorVec_pi_zero_resolved = TLorentzVector (pi_zero_resolved_PX, pi_zero_resolved_PY, pi_zero_resolved_PZ, pi_zero_resolved_PE);
LorVec_gamma1 = TLorentzVector (gamma1_PX, gamma1_PY, gamma1_PZ, gamma1_PE);
LorVec_gamma2 = TLorentzVector (gamma2_PX, gamma2_PY, gamma2_PZ, gamma2_PE);
}
//Recombination of 'Parent'-Particles
LorVec_DiMuon = LorVec_mu_minus + LorVec_mu_plus;
if(Kst2Kspiplus){
LorVec_KaonPion = LorVec_K_short + LorVec_pi_plus;
}
if(Kst2Kpluspi0Merged){
LorVec_KaonPion = LorVec_K_plus + LorVec_pi_zero_merged;
}
if(Kst2Kpluspi0Resolved){
LorVec_DiGamma = LorVec_gamma1 + LorVec_gamma2;
LorVec_KaonPion = LorVec_K_plus + LorVec_DiGamma;
}
//Recombination of Particles
if(Kst2Kspiplus){
LorVec_DTF_PDG_KshortMuMu = LorVec_DTF_PDG_K_short+ LorVec_DTF_PDG_DiMuon;
LorVec_DTF_KshortMuMu = LorVec_DTF_K_short + LorVec_DTF_DiMuon;
LorVec_PDG_KshortMuMu = LorVec_PDG_K_short + LorVec_PDG_DiMuon;
LorVec_KshortMuMu = LorVec_K_short + LorVec_DiMuon;
}
if(Kst2Kpluspi0Merged){
LorVec_DTF_PDG_KplusMuMu = LorVec_DTF_PDG_K_plus + LorVec_DTF_PDG_DiMuon;
LorVec_DTF_KplusMuMu = LorVec_DTF_K_plus + LorVec_DTF_DiMuon;
LorVec_PDG_KplusMuMu = LorVec_PDG_K_plus + LorVec_PDG_DiMuon;
LorVec_KplusMuMu = LorVec_K_plus + LorVec_DiMuon;
}
if(Kst2Kpluspi0Resolved){
LorVec_DTF_PDG_KplusMuMu = LorVec_DTF_PDG_K_plus + LorVec_DTF_PDG_DiMuon;
LorVec_DTF_KplusMuMu = LorVec_DTF_K_plus + LorVec_DTF_DiMuon;
LorVec_PDG_KplusMuMu = LorVec_PDG_K_plus + LorVec_PDG_DiMuon;
LorVec_KplusMuMu = LorVec_K_plus + LorVec_DiMuon;
LorVec_DTF_PDG_ReconBPiZero = LorVec_DTF_PDG_DiMuon + LorVec_DTF_PDG_K_plus + LorVec_DTF_PDG_pi_zero_resolved;
LorVec_DTF_ReconBPiZero = LorVec_DTF_DiMuon + LorVec_DTF_K_plus + LorVec_DTF_pi_zero_resolved;
LorVec_PDG_ReconBPiZero = LorVec_PDG_DiMuon + LorVec_PDG_K_plus + LorVec_PDG_pi_zero_resolved;
LorVec_ReconBPiZero = LorVec_DiMuon + LorVec_K_plus + LorVec_pi_zero_resolved;
LorVec_DTF_PDG_pi_zero_resolved = LorVec_DTF_PDG_DiGamma;
LorVec_DTF_pi_zero_resolved = LorVec_DTF_DiGamma;
LorVec_PDG_pi_zero_resolved = LorVec_PDG_DiGamma;
LorVec_pi_zero_resolved = LorVec_DiGamma;
}
LorVec_DTF_PDG_K_star_plus = LorVec_DTF_PDG_KaonPion;
LorVec_DTF_K_star_plus = LorVec_DTF_KaonPion;
LorVec_PDG_K_star_plus = LorVec_PDG_KaonPion;
LorVec_K_star_plus = LorVec_KaonPion;
LorVec_DTF_PDG_ReconB = LorVec_DTF_PDG_DiMuon + LorVec_DTF_PDG_K_star_plus;
LorVec_DTF_ReconB = LorVec_DTF_DiMuon + LorVec_DTF_K_star_plus;
LorVec_PDG_ReconB = LorVec_PDG_DiMuon + LorVec_PDG_K_star_plus;
LorVec_ReconB = LorVec_DiMuon + LorVec_K_star_plus;
//fill MASS HISTOGRAMS
h_B_M_DTF_PDG ->Fill(LorVec_DTF_PDG_ReconB.M());
h_B_M_DTF ->Fill(LorVec_DTF_ReconB.M());
h_B_M_PDG ->Fill(LorVec_PDG_ReconB.M());
h_B_M ->Fill(LorVec_ReconB.M());
h_Kstar_M_DTF_PDG ->Fill(LorVec_DTF_PDG_K_star_plus.M());
h_Kstar_M_DTF ->Fill(LorVec_DTF_K_star_plus.M());
h_Kstar_M_PDG ->Fill(LorVec_PDG_K_star_plus.M());
h_Kstar_M ->Fill(LorVec_K_star_plus.M());
if(Kst2Kspiplus){
h_Kshort_M_DTF_PDG ->Fill(LorVec_DTF_PDG_K_short.M());
h_Kshort_M_DTF ->Fill(LorVec_DTF_K_short.M());
h_Kshort_M_PDG ->Fill(LorVec_PDG_K_short.M());
h_Kshort_M ->Fill(LorVec_K_short.M());
}
if (Kst2Kpluspi0Resolved){
h_pi0_M_DTF_PDG ->Fill(LorVec_DTF_PDG_pi_zero_resolved.M());
h_pi0_M_DTF ->Fill(LorVec_DTF_pi_zero_resolved.M());
h_pi0_M_PDG ->Fill(LorVec_PDG_pi_zero_resolved.M());
h_pi0_M ->Fill(LorVec_pi_zero_resolved.M());
}
//fill PT HISTOGRAMS
h_B_PT_DTF_PDG ->Fill(LorVec_DTF_PDG_ReconB.Pt());
h_B_PT_DTF ->Fill(LorVec_DTF_ReconB.Pt());
h_B_PT_PDG ->Fill(LorVec_PDG_ReconB.Pt());
h_B_PT ->Fill(LorVec_ReconB.Pt());
h_Kstar_PT_DTF_PDG ->Fill(LorVec_DTF_PDG_K_star_plus.Pt());
h_Kstar_PT_DTF ->Fill(LorVec_DTF_K_star_plus.Pt());
h_Kstar_PT_PDG ->Fill(LorVec_PDG_K_star_plus.Pt());
h_Kstar_PT ->Fill(LorVec_K_star_plus.Pt());
if(Kst2Kspiplus){
h_Kshort_PT_DTF_PDG ->Fill(LorVec_DTF_PDG_K_short.Pt());
h_Kshort_PT_DTF ->Fill(LorVec_DTF_K_short.Pt());
h_Kshort_PT_PDG ->Fill(LorVec_PDG_K_short.Pt());
h_Kshort_PT ->Fill(LorVec_K_short.Pt());
}
if (Kst2Kpluspi0Resolved){
h_pi0_PT_DTF_PDG ->Fill(LorVec_DTF_PDG_pi_zero_resolved.Pt());
h_pi0_PT_DTF ->Fill(LorVec_DTF_pi_zero_resolved.Pt());
h_pi0_PT_PDG ->Fill(LorVec_PDG_pi_zero_resolved.Pt());
h_pi0_PT ->Fill(LorVec_pi_zero_resolved.Pt());
}
//fill ETA HISTOGRAMS
h_B_ETA_DTF_PDG ->Fill(LorVec_DTF_PDG_ReconB.Eta() );
h_B_ETA_DTF ->Fill(LorVec_DTF_ReconB.Eta());
h_B_ETA_PDG ->Fill(LorVec_PDG_ReconB.Eta());
h_B_ETA ->Fill(LorVec_ReconB.Eta());
h_Kstar_ETA_DTF_PDG ->Fill(LorVec_DTF_PDG_K_star_plus.Eta());
h_Kstar_ETA_DTF ->Fill(LorVec_DTF_K_star_plus.Eta());
h_Kstar_ETA_PDG ->Fill(LorVec_PDG_K_star_plus.Eta());
h_Kstar_ETA ->Fill(LorVec_K_star_plus.Eta());
if(Kst2Kspiplus){
h_Kshort_ETA_DTF_PDG ->Fill(LorVec_DTF_PDG_K_short.Eta());
h_Kshort_ETA_DTF ->Fill(LorVec_DTF_K_short.Eta());
h_Kshort_ETA_PDG ->Fill(LorVec_PDG_K_short.Eta());
h_Kshort_ETA ->Fill(LorVec_K_short.Eta());
}
if (Kst2Kpluspi0Resolved){
h_pi0_ETA_DTF_PDG ->Fill(LorVec_DTF_PDG_pi_zero_resolved.Eta());
h_pi0_ETA_DTF ->Fill(LorVec_DTF_pi_zero_resolved.Eta());
h_pi0_ETA_PDG ->Fill(LorVec_PDG_pi_zero_resolved.Eta());
h_pi0_ETA ->Fill(LorVec_pi_zero_resolved.Eta());
}
//fill angles
h_ctk->Fill(ThetaK);
h_ctl->Fill(ThetaL);
h_sp->Fill(Phi);
//fill other variables
if (Kst2Kpluspi0Resolved){
OpenAngle_DTF_PDG = LorVec_DTF_PDG_gamma1.Angle(LorVec_DTF_PDG_gamma2.Vect());
OpenAngle_DTF = LorVec_DTF_gamma1.Angle(LorVec_DTF_gamma2.Vect());
OpenAngle_PDG = LorVec_PDG_gamma1.Angle(LorVec_PDG_gamma2.Vect());
OpenAngle = LorVec_gamma1.Angle(LorVec_gamma2.Vect());
h_OpenAngle_DTF_PDG ->Fill(OpenAngle_DTF_PDG);
h_OpenAngle_DTF ->Fill(OpenAngle_DTF);
h_OpenAngle_PDG ->Fill(OpenAngle_PDG);
h_OpenAngle ->Fill(OpenAngle);
}
} //end of event loop
// Kst2Kpluspi0Resolved case
string newName = "";
preselected ? newName = "_DTFcheck_preselected" : newName ="_DTFcheck";
if (truthMatched) newName = newName + "_TM";
if(Kst2Kpluspi0Resolved){
if(!MC){
output = new TFile(Form("%s/data/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else{
if(ReferenceChannel){
output = new TFile(Form("%s/data/MC/RefKplusPi0/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else if(PHSP){
output = new TFile(Form("%s/data/PHSP/KplusPi0/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else{
output = new TFile(Form("%s/data/MC/KplusPi0/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
}
}
// Kst2Kpluspi0Merged case
if(Kst2Kpluspi0Merged){
if(!MC){
output = new TFile(Form("%s/data/%s%s/%s%s_pi0Merged%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else{
if(ReferenceChannel){
output = new TFile(Form("%s/data/MC/RefKplusPi0/%s%s/%s%s_pi0Merged%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else if(PHSP){
output = new TFile(Form("%s/data/PHSP/KplusPi0/%s%s/%s%s_pi0Merged%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else{
output = new TFile(Form("%s/data/MC/KplusPi0/%s%s/%s%s_pi0Merged%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
}
}
// Kst2Kspiplus case
if(Kst2Kspiplus){
if(!MC){
output = new TFile(Form("%s/data/%s%s/%s%s_piplus%s.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else{
if(ReferenceChannel){
output = new TFile(Form("%s/data/MC/RefKshortPiplus/%s%s/%s%s_piplus%s.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else if(PHSP){
output = new TFile(Form("%s/data/PHSP/KshortPiplus/%s%s/%s%s_piplus%s.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
else{
output = new TFile(Form("%s/data/MC/KshortPiplus/%s%s/%s%s_piplus%s.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()),"RECREATE");
}
}
}
output->cd();
//save MASS HISTOGRAMS
h_B_M_DTF_PDG ->Write();
h_B_M_DTF ->Write();
h_B_M_PDG ->Write();
h_B_M ->Write();
h_Kstar_M_DTF_PDG ->Write();
h_Kstar_M_DTF ->Write();
h_Kstar_M_PDG ->Write();
h_Kstar_M ->Write();
if(Kst2Kspiplus){
h_Kshort_M_DTF_PDG ->Write();
h_Kshort_M_DTF ->Write();
h_Kshort_M_PDG ->Write();
h_Kshort_M ->Write();
}
if (Kst2Kpluspi0Resolved){
h_pi0_M_DTF_PDG ->Write();
h_pi0_M_DTF ->Write();
h_pi0_M_PDG ->Write();
h_pi0_M ->Write();
}
//save PT HISTOGRAMS
h_B_PT_DTF_PDG ->Write();
h_B_PT_DTF ->Write();
h_B_PT_PDG ->Write();
h_B_PT ->Write();
h_Kstar_PT_DTF_PDG ->Write();
h_Kstar_PT_DTF ->Write();
h_Kstar_PT_PDG ->Write();
h_Kstar_PT ->Write();
if(Kst2Kspiplus){
h_Kshort_PT_DTF_PDG ->Write();
h_Kshort_PT_DTF ->Write();
h_Kshort_PT_PDG ->Write();
h_Kshort_PT ->Write();
}
if (Kst2Kpluspi0Resolved){
h_pi0_PT_DTF_PDG ->Write();
h_pi0_PT_DTF ->Write();
h_pi0_PT_PDG ->Write();
h_pi0_PT ->Write();
}
//save ETA HISTOGRAMS
h_B_ETA_DTF_PDG ->Write();
h_B_ETA_DTF ->Write();
h_B_ETA_PDG ->Write();
h_B_ETA ->Write();
h_Kstar_ETA_DTF_PDG ->Write();
h_Kstar_ETA_DTF ->Write();
h_Kstar_ETA_PDG ->Write();
h_Kstar_ETA ->Write();
if(Kst2Kspiplus){
h_Kshort_ETA_DTF_PDG ->Write();
h_Kshort_ETA_DTF ->Write();
h_Kshort_ETA_PDG ->Write();
h_Kshort_ETA ->Write();
}
if (Kst2Kpluspi0Resolved){
h_pi0_ETA_DTF_PDG ->Write();
h_pi0_ETA_DTF ->Write();
h_pi0_ETA_PDG ->Write();
h_pi0_ETA ->Write();
//save other histograms
h_OpenAngle_DTF_PDG ->Write();
h_OpenAngle_DTF ->Write();
h_OpenAngle_PDG ->Write();
h_OpenAngle ->Write();
}
//save Angle histograms
// h_ctk ->Write();
// h_ctl ->Write();
// h_sp ->Write();
output->Close();
cout << "[INFO]\t\t Histograms created." << endl;
return 1;
}
int DTFcomparisonAllData(Int_t Run = 1, bool preselected = true) {
if(Run != 1 && Run != 2){
std::cout << "[ERROR]\t\tInvalid Run number given: " << Run << ". Exit program!" << std::endl;
return 0;
}
if ( DTFcomparison((Run == 1 ? "2011" : "2015"),"down", false, false, false, preselected, 0) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2011" : "2015"),"up" , false, false, false, preselected, 0) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2012" : "2016"),"down", false, false, false, preselected, 0) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2012" : "2016"),"up" , false, false, false, preselected, 0) == 0) return 0;
return 1;
}
int DTFcomparisonAllMC(Int_t Run = 1, bool preselected = true, bool truthMatched = false) {
if(Run != 1 && Run != 2){
std::cout << "[ERROR]\t\tInvalid Run number given: " << Run << ". Exit program!" << std::endl;
return 0;
}
if ( DTFcomparison((Run == 1 ? "2011" : "2015"),"down", true, false, false, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2011" : "2015"),"up" , true, false, false, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2012" : "2016"),"down", true, false, false, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2012" : "2016"),"up" , true, false, false, preselected, truthMatched) == 0) return 0;
return 1;
}
int DTFcomparisonAllRefChannel(Int_t Run = 1, bool preselected = true, bool truthMatched = false) {
if(Run != 1 && Run != 2){
std::cout << "[ERROR]\t\tInvalid Run number given: " << Run << ". Exit program!" << std::endl;
return 0;
}
if ( DTFcomparison((Run == 1 ? "2011" : "2015"),"down", true, true, false, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2011" : "2015"),"up" , true, true, false, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2012" : "2016"),"down", true, true, false, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2012" : "2016"),"up" , true, true, false, preselected, truthMatched) == 0) return 0;
return 1;
}
int DTFcomparisonAllPHSP(Int_t Run = 1, bool preselected = true, bool truthMatched = false) {
if(Run != 1 && Run != 2){
std::cout << "[ERROR]\t\tInvalid Run number given: " << Run << ". Exit program!" << std::endl;
return 0;
}
if ( DTFcomparison((Run == 1 ? "2011" : "2015"),"down", true, false, true, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2011" : "2015"),"up" , true, false, true, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2012" : "2016"),"down", true, false, true, preselected, truthMatched) == 0) return 0;
if ( DTFcomparison((Run == 1 ? "2012" : "2016"),"up" , true, false, true, preselected, truthMatched) == 0) return 0;
return 1;
}
int DTFcomparisonAll( Int_t Run = 1, bool preselected = true, bool truthMatched = false) {
if (DTFcomparisonAllData (Run, preselected)==0) return 0;
if (DTFcomparisonAllMC (Run, preselected, truthMatched)==0) return 0;
if (Run == 2) cout << "[INFO]\tNo reference channel for run 2, Skipping.";
else{
if (DTFcomparisonAllRefChannel (Run, preselected, truthMatched)==0) return 0;
}
if (DTFcomparisonAllPHSP (Run, preselected, truthMatched)==0) return 0;
return 1;
}
int DTFcomparisonComplete( Int_t Run = 1){
if (DTFcomparisonAllData (Run, 1)==0) return 0;
if (DTFcomparisonAllData (Run, 0)==0) return 0;
if (DTFcomparisonAllMC (Run, 0, 0)==0) return 0;
if (DTFcomparisonAllMC (Run, 1, 0)==0) return 0;
if (DTFcomparisonAllMC (Run, 1, 1)==0) return 0;
if (DTFcomparisonAllRefChannel (Run, 0, 0)==0) return 0;
if (DTFcomparisonAllRefChannel (Run, 1, 0)==0) return 0;
if (DTFcomparisonAllRefChannel (Run, 1, 1)==0) return 0;
if (DTFcomparisonAllPHSP (Run, 0, 0)==0) return 0;
if (DTFcomparisonAllPHSP (Run, 1, 0)==0) return 0;
if (DTFcomparisonAllPHSP (Run, 1, 1)==0) return 0;
return 1;
}
int plotDTFcomparison(std::string year = "2011", std::string magnet = "down", bool MC = false, bool ReferenceChannel = false, bool PHSP = false, bool preselected = true, bool truthMatched = false){
// Kst2Kpluspi0Resolved case
TFile *input = 0;
string newName = "";
string legTitle = "";
string TM = "";
preselected ? newName = "_DTFcheck_preselected" : newName ="_DTFcheck";
truthMatched ? TM = "_TM" : "";
newName = newName + TM;
Kst2Kspiplus? legTitle= particleLatex("Kstar") +" #rightarrow " +particleLatex("Kshort") + particleLatex("piplus") :legTitle= particleLatex("Kstar") +" #rightarrow " +particleLatex("Kplus") + particleLatex("pizero");
preselected ? legTitle= legTitle + ": preselected" : legTitle = legTitle+": stripped";
truthMatched ? legTitle= "#splitline{" + legTitle+":}{TruthMatched}" : legTitle = legTitle;
if (MC){
if(ReferenceChannel) legTitle = legTitle + " {MCRef}";
else if (PHSP) legTitle = legTitle + " {PHSP MC}";
else legTitle = legTitle + " {MC}";
}
if(Kst2Kpluspi0Resolved){
if(!MC){
input = TFile::Open(Form("%s/data/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
cout << Form("%s/data/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()) << endl;
}
else{
if(ReferenceChannel){
input = TFile::Open(Form("%s/data/MC/RefKplusPi0/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
else if(PHSP){
input = TFile::Open(Form("%s/data/PHSP/KplusPi0/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
else{
input = TFile::Open(Form("%s/data/MC/KplusPi0/%s%s/%s%s_pi0Resolved%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
}
}
// Kst2Kpluspi0Merged case
if(Kst2Kpluspi0Merged){
if(!MC){
input = TFile::Open(Form("%s/data/%s%s/%s%s_pi0Merged%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
else{
if(ReferenceChannel){
input = TFile::Open(Form("%s/data/MC/RefKplusPi0/%s%s/%s%s_pi0Merged%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
else if(PHSP){
input = TFile::Open(Form("%s/data/PHSP/KplusPi0/%s%s/%s%s_pi0Merged%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
else{
input = TFile::Open(Form("%s/data/MC/KplusPi0/%s%s/%s%s_pi0Merged%s.root",path_to_output_KplusPizero.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
}
}
// Kst2Kspiplus case
if(Kst2Kspiplus){
if(!MC){
input = TFile::Open(Form("%s/data/%s%s/%s%s_piplus%s.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
else{
if(ReferenceChannel){
input = TFile::Open(Form("%s/data/MC/RefKshortPiplus/%s%s/%s%s_piplus%s.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
else if(PHSP){
input = TFile::Open(Form("%s/data/PHSP/KshortPiplus/%s%s/%s%s_piplus%s.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
else{
input = TFile::Open(Form("%s/data/MC/KshortPiplus/%s%s/%s%s_piplus%s.root",path_to_output_KshortPiplus.c_str(), year.c_str(),magnet.c_str(),year.c_str(),magnet.c_str(),newName.c_str()));
}
}
}
string tmpName = "";
string tmpNameCanvas = "";
int col = 1;
if(Kst2Kpluspi0Resolved){
for (auto& var: Variables){
for (auto& particle: ParticlesPiZero){
tmpNameCanvas = "c_" + particle + "_" + var;
TCanvas *canvas = new TCanvas(tmpNameCanvas.c_str(), "", 10,10,600,600);
designCanvas(canvas);
canvas->cd();
gStyle->SetOptStat(0);
TLegend *leg = new TLegend(0.57,0.78,0.96,0.96);
col = 1;
for (auto& opt: DTFoptions){
tmpName = "h_" + particle + "_" + var + "_" + opt;
if (strcmp(&tmpName[tmpName.length()-1],"_")==0) tmpName.pop_back(); //if ending with _, remove it
TH1D* tmp = (TH1D*)input->Get(tmpName.c_str()); //check memory leaks
designPlots(tmp,col);
designLegend(leg, legTitle, tmp, opt);
//tmp->SetTitle((variableLatex(var)+"(" + particleLatex(particle)+")").c_str());
tmp->Draw("SAME");
col++;
}
leg->Draw();
tmpName = path_to_output_KplusPizero;
if (MC){
if(ReferenceChannel) tmpName = tmpName + "/data/MC/RefKplusPi0/" + year + magnet;
else if (PHSP) tmpName = tmpName + "/data/PHSP/KplusPi0/" + year + magnet;
else tmpName = tmpName + "/data/MC/KplusPi0/" + year + magnet;
}
else tmpName = tmpName + "/data/" + year + magnet;
tmpNameCanvas = tmpName + "/" + tmpNameCanvas + TM + (preselected ? "_preselected" :"") + "." +figureFormat;
canvas->Print(tmpNameCanvas.c_str(),figureFormat.c_str()); //#TODO: cancel piplus, replace by path_to accordingly
leg->Clear();
canvas->Clear();
delete(leg);
delete(canvas);
}
}
for (auto& evar: ExtraVariables){
tmpNameCanvas = "c_" + evar;
TCanvas *canvas = new TCanvas(tmpNameCanvas.c_str(), "", 10,10,600,600);
designCanvas(canvas);
canvas->cd();
gStyle->SetOptStat(0);
TLegend *leg = new TLegend(0.62,0.8,0.96,0.96);
col = 1;
for (auto& opt: DTFoptions){
tmpName = "h_" + evar + "_" + opt;
if (strcmp(&tmpName[tmpName.length()-1],"_")==0) tmpName.pop_back(); //if ending with _, remove it
TH1D* tmp = (TH1D*)input->Get(tmpName.c_str()); //check memory leaks
designPlots(tmp,col);
designLegend(leg, legTitle, tmp, opt);
// tmp->SetTitle((variableLatex(evar)).c_str());
tmp->Draw("SAME");
col++;
}
leg->Draw();
tmpName = path_to_output_KplusPizero;
if (MC){
if(ReferenceChannel) tmpName = tmpName + "/data/MC/RefKplusPi0/" + year + magnet;
else if (PHSP) tmpName = tmpName + "/data/PHSP/KplusPi0/" + year + magnet;
else tmpName = tmpName + "/data/MC/KplusPi0/" + year + magnet;
}
else tmpName = tmpName + "/data/" + year + magnet;
tmpNameCanvas = tmpName + "/" + tmpNameCanvas + TM + (preselected ? "_preselected" :"") + "." +figureFormat;
canvas->Print(tmpNameCanvas.c_str(),figureFormat.c_str()); //#TODO: cancel piplus, replace by path_to accordingly
leg->Clear();
canvas->Clear();
delete(leg);
delete(canvas);
}
}
if(Kst2Kspiplus){
for (auto& var: Variables){
for (auto& particle: ParticlesKShort){
tmpNameCanvas = "c_" + particle + "_" + var;
TCanvas *canvas = new TCanvas(tmpNameCanvas.c_str(), "", 10,10,600,600);
designCanvas(canvas);
canvas->cd();
gStyle->SetOptStat(0);
TLegend *leg = new TLegend(0.62,0.8,0.96,0.96);
col = 1;
for (auto& opt: DTFoptions){
tmpName = "h_" + particle + "_" + var + "_" + opt;
if (strcmp(&tmpName[tmpName.length()-1],"_")==0) tmpName.pop_back(); //if ending with _, remove it
TH1D* tmp = (TH1D*)input->Get(tmpName.c_str()); //check memory leaks
designPlots(tmp,col);
designLegend(leg, legTitle, tmp, opt);
// tmp->SetTitle((variableLatex(var)+"(" + particleLatex(particle)+")").c_str());
tmp->Draw("SAME");
col++;
}
leg->Draw();
tmpName = path_to_output_KshortPiplus;
if (MC){
if(ReferenceChannel) tmpName = tmpName + "/data/MC/RefKplusPi0/" + year + magnet;
else if (PHSP) tmpName = tmpName + "/data/PHSP/KplusPi0/" + year + magnet;
else tmpName = tmpName + "/data/MC/KplusPi0/" + year + magnet;
}
else tmpName = tmpName + "/data/" + year + magnet;
tmpNameCanvas = tmpName + "/piplus_" + tmpNameCanvas +TM + (preselected ? "_preselected" :"") + "." +figureFormat;
canvas->Print(tmpNameCanvas.c_str(),figureFormat.c_str()); //#TODO: cancel piplus, replace by path_to accordingly
leg->Clear();
canvas->Clear();
delete(leg);
delete(canvas);
}
}
}
return 1;
}
int plotDTFcomparisonAllData(Int_t Run = 1, bool preselected = true) {
if(Run != 1 && Run != 2){
std::cout << "[ERROR]\t\tInvalid Run number given: " << Run << ". Exit program!" << std::endl;
return 0;
}
if ( plotDTFcomparison((Run == 1 ? "2011" : "2015"),"down", false, false, false, preselected, 0) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2011" : "2015"),"up" , false, false, false, preselected, 0) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2012" : "2016"),"down", false, false, false, preselected, 0) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2012" : "2016"),"up" , false, false, false, preselected, 0) == 0) return 0;
return 1;
}
int plotDTFcomparisonAllMC(Int_t Run = 1, bool preselected = true, bool truthMatched = false) {
if(Run != 1 && Run != 2){
std::cout << "[ERROR]\t\tInvalid Run number given: " << Run << ". Exit program!" << std::endl;
return 0;
}
if ( plotDTFcomparison((Run == 1 ? "2011" : "2015"),"down", true, false, false, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2011" : "2015"),"up" , true, false, false, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2012" : "2016"),"down", true, false, false, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2012" : "2016"),"up" , true, false, false, preselected, truthMatched) == 0) return 0;
return 1;
}
int plotDTFcomparisonAllRefChannel(Int_t Run = 1, bool preselected = true, bool truthMatched = false) {
if(Run != 1 && Run != 2){
std::cout << "[ERROR]\t\tInvalid Run number given: " << Run << ". Exit program!" << std::endl;
return 0;
}
if ( plotDTFcomparison((Run == 1 ? "2011" : "2015"),"down", true, true, false, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2011" : "2015"),"up" , true, true, false, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2012" : "2016"),"down", true, true, false, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2012" : "2016"),"up" , true, true, false, preselected, truthMatched) == 0) return 0;
return 1;
}
int plotDTFcomparisonAllPHSP(Int_t Run = 1, bool preselected = true, bool truthMatched = false) {
if(Run != 1 && Run != 2){
std::cout << "[ERROR]\t\tInvalid Run number given: " << Run << ". Exit program!" << std::endl;
return 0;
}
if ( plotDTFcomparison((Run == 1 ? "2011" : "2015"),"down", true, false, true, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2011" : "2015"),"up" , true, false, true, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2012" : "2016"),"down", true, false, true, preselected, truthMatched) == 0) return 0;
if ( plotDTFcomparison((Run == 1 ? "2012" : "2016"),"up" , true, false, true, preselected, truthMatched) == 0) return 0;
return 1;
}
int plotDTFcomparisonAll( Int_t Run = 1, bool preselected = true, bool truthMatched = false) {
if (plotDTFcomparisonAllData (Run, preselected)==0) return 0;
if (plotDTFcomparisonAllMC (Run, preselected, truthMatched)==0) return 0;
if (plotDTFcomparisonAllRefChannel (Run, preselected, truthMatched)==0) return 0;
if (plotDTFcomparisonAllPHSP (Run, preselected, truthMatched)==0) return 0;
return 1;
}
int plotDTFcomparisonComplete( Int_t Run = 1){
if (plotDTFcomparisonAllData (Run, 1)==0) return 0;
if (plotDTFcomparisonAllData (Run, 0)==0) return 0;
if (plotDTFcomparisonAllMC (Run, 0, 0)==0) return 0;
if (plotDTFcomparisonAllMC (Run, 1, 0)==0) return 0;
if (plotDTFcomparisonAllMC (Run, 1, 1)==0) return 0;
if (plotDTFcomparisonAllRefChannel (Run, 0, 0)==0) return 0;
if (plotDTFcomparisonAllRefChannel (Run, 1, 0)==0) return 0;
if (plotDTFcomparisonAllRefChannel (Run, 1, 1)==0) return 0;
if (plotDTFcomparisonAllPHSP (Run, 0, 0)==0) return 0;
if (plotDTFcomparisonAllPHSP (Run, 1, 0)==0) return 0;
if (plotDTFcomparisonAllPHSP (Run, 1, 1)==0) return 0;
return 1;
}
Reference in New Issue Copy Permalink