295 lines
11 KiB
C++
295 lines
11 KiB
C++
//Renata Kopecna
|
|
|
|
|
|
#include "ReferencePlots.hh"
|
|
|
|
#include <vector>
|
|
|
|
#include "ScriptHelpers.hh"
|
|
#include <bu2kstarmumu_parameters.hh>
|
|
#include <paths.hh>
|
|
#include <helpers.hh>
|
|
|
|
#include <TColor.h>
|
|
#include <TH2D.h>
|
|
#include <TCanvas.h>
|
|
#include <TROOT.h>
|
|
#include <TFile.h>
|
|
#include <TTree.h>
|
|
|
|
#include <spdlog.h>
|
|
|
|
const int oneSigma_RGB[3] = {34,136,51};
|
|
const int twoSigma_RGB[3] = {170,170,0};
|
|
const int threeSigma_RGB[3] = {238,22,136};
|
|
const int moreSigma_RGB[3] = {204,51,17};
|
|
|
|
//TODO: make it into a texfile instead of a cout!
|
|
|
|
const std::vector<valErr> Davids = {
|
|
valErr(+0.572,0.005),//valErr(+0.321,0.004),
|
|
valErr(-0.002,0.007),
|
|
valErr(-0.246,0.008),
|
|
valErr(-0.003,0.008),
|
|
valErr(-0.002,0.005),//valErr(-0.003,0.006),
|
|
valErr(-0.001,0.008),
|
|
valErr(-0.063,0.008),
|
|
valErr(-0.084,0.007)
|
|
};
|
|
const std::vector<valErr> Belles = {
|
|
valErr(+0.604,0.015),//valErr(+0.297,0.011),
|
|
valErr(-0.018,0.017),
|
|
valErr(-0.255,0.010),
|
|
valErr(0.0,0.0),
|
|
valErr(0.0,0.0),
|
|
valErr(0.0,0.0),
|
|
valErr(-0.037,0.018),
|
|
valErr(-0.041,0.016)
|
|
};
|
|
const std::vector<valErr> BaBars = {
|
|
valErr(+0.556,0.009),//valErr(+0.333,0.007),
|
|
valErr(+0.011,0.011),
|
|
valErr(-0.237,0.007),
|
|
valErr(0.0,0.0),
|
|
valErr(0.0,0.0),
|
|
valErr(0.0,0.0),
|
|
valErr(-0.058,0.015),
|
|
valErr(-0.095,0.014)
|
|
};
|
|
const std::vector<valErr> B0s = {
|
|
valErr(+0.572,0.008),//valErr(+0.321,0.006),
|
|
valErr(-0.013,0.010),
|
|
valErr(-0.250,0.006),
|
|
valErr(0.0,0.0),
|
|
valErr(0.0,0.0),
|
|
valErr(0.0,0.0),
|
|
valErr(-0.048,0.007),
|
|
valErr(-0.084,0.006)
|
|
};
|
|
|
|
std::vector<double>getSigma(std::vector<valErr> vec_a, std::vector<valErr> vec_b){
|
|
if (vec_a.size() != vec_b.size()){
|
|
spdlog::error("Two valErr vectors have different lenghts! Abort.");
|
|
assert(0);
|
|
}
|
|
std::vector<double>sigs;
|
|
for_indexed(auto a: vec_a){
|
|
sigs.push_back(sigmaAway(a,vec_b[i]));
|
|
}
|
|
return sigs;
|
|
}
|
|
|
|
std::vector<const int*>getSigmaColor(std::vector<double>vec_sigma){
|
|
std::vector<const int*> tmp;
|
|
for(auto sigma: vec_sigma){
|
|
if (fabs(sigma) < 1.0) tmp.push_back(oneSigma_RGB);
|
|
else if (fabs(sigma) < 2.0) tmp.push_back(twoSigma_RGB);
|
|
else if (fabs(sigma) < 3.0) tmp.push_back(threeSigma_RGB);
|
|
else tmp.push_back(moreSigma_RGB);
|
|
}
|
|
return tmp;
|
|
}
|
|
|
|
std::string getSigmaColor_str(double sigma){
|
|
if (fabs(sigma) < 1.0) return "\\cellcolor{oneSigma}";
|
|
else if (fabs(sigma) < 2.0) return "\\cellcolor{twoSigmas}";
|
|
else if (fabs(sigma) < 3.0) return"\\cellcolor{threeSigmas}";
|
|
else return "\\cellcolor{moreSigmas}";
|
|
}
|
|
|
|
std::vector<std::string>getSigmaColor_strVec(std::vector<double>vec_sigma){
|
|
std::vector<std::string> tmp;
|
|
for(auto sigma: vec_sigma) tmp.push_back(getSigmaColor_str(sigma));
|
|
return tmp;
|
|
}
|
|
double getCorrectedError(std::string name, double err, int jobID){ //For now, no bins, cause it is used anyhow only for reference //TODO
|
|
|
|
basic_params tmpParams; //This is dumb, but no time
|
|
tmpParams.jobID = jobID;
|
|
|
|
std::string fileName = get_ToyPullPlot_folder(tmpParams)+"pullResults.root";
|
|
spdlog::info("Opening " + fileName);
|
|
TFile* file = new TFile(fileName.c_str(), "READ");
|
|
|
|
if (!file->GetListOfKeys()->Contains(name.c_str())){
|
|
spdlog::critical("Wrong tree name " + name + "! Abort.");
|
|
assert(0);
|
|
}
|
|
//Get the tree and set the branches to active
|
|
TTree* tree = (TTree*)file->Get(name.c_str());
|
|
tree->SetBranchStatus("*",1); //Activate all branches
|
|
double width = DEFAULT_TREE_ERR;
|
|
double mean = DEFAULT_TREE_VAL;
|
|
|
|
tree->SetBranchAddress("width", &width);
|
|
tree->SetBranchAddress("mean", &mean);
|
|
|
|
tree->GetEntry(0);
|
|
file->Close();
|
|
|
|
//get_param_valueError_from_rootfile()
|
|
|
|
return err*width;
|
|
}
|
|
|
|
int plotComparisonFull(){
|
|
spdlog::debug("Starting with printing the values for RefChan measurement in a nice table.");
|
|
double fractionOfStats = 1.0;
|
|
|
|
basic_params params_Fit= basic_params();
|
|
params_Fit.Run = 12;
|
|
params_Fit.nBins = 1;
|
|
params_Fit.reference =true;
|
|
std::string fitFile = final_result_name(true, false, params_Fit,
|
|
true, 1, params_Fit.Run,
|
|
false, fractionOfStats,
|
|
false, -1);
|
|
|
|
std::vector<valErr> myValues;
|
|
spdlog::debug("Filling my values into a vector");
|
|
for (auto obs: ANG_OBS){
|
|
//The easiest rounding in c++
|
|
double err = get_param_error_from_rootfile(fitFile, obs, 1,0);
|
|
//The easiest rounding in c++
|
|
err = round(getCorrectedError(obs,err,649)*1000.0)/1000.0; //TODO
|
|
double val = round(get_param_value_from_rootfile(fitFile, obs, 1,0)*1000.0)/1000.0;
|
|
myValues.push_back(valErr(val,err));
|
|
}
|
|
std::vector<std::string>col_Davids = getSigmaColor_strVec(getSigma(myValues,Davids));
|
|
std::vector<std::string>col_Belles = getSigmaColor_strVec(getSigma(myValues,Belles));
|
|
std::vector<std::string>col_BaBars = getSigmaColor_strVec(getSigma(myValues,BaBars));
|
|
std::vector<std::string>col_B0s = getSigmaColor_strVec( getSigma(myValues,B0s));
|
|
|
|
std::cout << "\\begin{table}[]" << std::endl;
|
|
std::cout << "\\begin{tabular}{llllll}" << std::endl;
|
|
std::cout << " & this fit & \\lhcb \\Bu & Belle (\\Bu) & BaBar (\\Bu + \\Bd) & \\lhcb \\Bd \\\\\\hline" << std::endl;
|
|
for_indexed(auto myVal: myValues){
|
|
std::cout << std::setprecision(3) << "$" << ANG_OBS_TLaTeX[i] << "$";
|
|
std::cout << " & " << myVal.val << "$\\pm$" <<myVal.err;
|
|
std::cout << " & " << col_Davids[i] << " " << Davids[i].val << "$\\pm$" <<Davids[i].err;
|
|
std::cout << " & " << col_Belles[i] << " " << Belles[i].val << "$\\pm$" <<Belles[i].err;
|
|
std::cout << " & " << col_BaBars[i] << " " << BaBars[i].val << "$\\pm$" <<BaBars[i].err;
|
|
std::cout << " & " << col_B0s[i] << " " << B0s[i].val << "$\\pm$" <<B0s[i].err;
|
|
std::cout << "\\\\" << std::endl;
|
|
}
|
|
|
|
std::cout << " \\end{tabular}" << std::endl;
|
|
std::cout << " \\end{table}" << std::endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
|
|
int foldingTable(){
|
|
spdlog::debug("Starting with printing the values for different foldings RefChan measurement in a nice table.");
|
|
double fractionOfStats = 1.0;
|
|
|
|
basic_params params_Fit= basic_params();
|
|
params_Fit.Run = 12;
|
|
params_Fit.nBins = 1;
|
|
params_Fit.reference =true;
|
|
|
|
std::cout << "\\begin{table}[]" << std::endl;
|
|
std::cout << "\\begin{tabular}{l|"; //First cell is for the obs name
|
|
for (int f = -1; f<MAX_FOLDING; f++) std::cout << "l";
|
|
std::cout << "}" << std::endl;
|
|
std::cout << std::setprecision(1) << " & No folding" ;
|
|
for (int f = 0; f<MAX_FOLDING; f++) std::cout << " & Folding " << f;
|
|
std::cout <<"\\\\\\hline" << std::endl;
|
|
|
|
for_indexed (auto obs: ANG_OBS){ //TODO: if it would be for me, this would be a struct of name and latexname
|
|
std::cout << std::setprecision(3) << "$" << ANG_OBS_TLaTeX[i] << "$";
|
|
params_Fit.folding = -1; //Needed extra as it has to be saved to myVec to get the sigmas away
|
|
std::string fitFile = final_result_name(true, false, params_Fit,
|
|
true, 1, params_Fit.Run,
|
|
false, fractionOfStats,
|
|
false, -1);
|
|
//The easiest rounding in c++
|
|
double err = get_param_error_from_rootfile(fitFile, obs, 1,0);
|
|
//The easiest rounding in c++
|
|
err = round(getCorrectedError(obs,err,649)*1000.0)/1000.0; //TODO
|
|
|
|
double val = round(get_param_value_from_rootfile(fitFile, obs, 1,0)*1000.0)/1000.0;
|
|
valErr full(err,val);
|
|
std::cout << " & " << val << "$\\pm$" << err;
|
|
|
|
for (int f = 0; f<MAX_FOLDING; f++){
|
|
if (plotObsv(f,obs)){
|
|
params_Fit.folding = f;
|
|
int jobID = 626+params_Fit.folding;
|
|
if (params_Fit.folding==4) jobID = 650;
|
|
|
|
std::string fitFile = final_result_name(true, false, params_Fit,
|
|
true, 1, params_Fit.Run,
|
|
false, fractionOfStats,
|
|
false, -1);
|
|
|
|
//The easiest rounding in c++
|
|
double err = get_param_error_from_rootfile(fitFile, obs, 1,0);
|
|
//The easiest rounding in c++
|
|
err = round(getCorrectedError(obs,err,jobID)*1000.0)/1000.0;
|
|
double val = round(get_param_value_from_rootfile(fitFile, obs, 1,0)*1000.0)/1000.0;
|
|
|
|
std::cout << " & " << getSigmaColor_str(sigmaAway(full,valErr(err,val))) << val << "$\\pm$" << err;
|
|
}
|
|
else std::cout << " & ";
|
|
}
|
|
std::cout << "\\\\" << std::endl;
|
|
}
|
|
|
|
std::cout << " \\end{tabular}" << std::endl;
|
|
std::cout << " \\end{table}" << std::endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int plotComparisonSigmas(){
|
|
spdlog::debug("Starting with printing the values for RefChan measurement in a nice table.");
|
|
double fractionOfStats = 1.0;
|
|
|
|
basic_params params_Fit= basic_params();
|
|
params_Fit.Run = 12;
|
|
params_Fit.nBins = 1;
|
|
params_Fit.reference =true;
|
|
std::string fitFile = final_result_name(true, false, params_Fit,
|
|
true, 1, params_Fit.Run,
|
|
false, fractionOfStats,
|
|
false, -1);
|
|
|
|
std::vector<valErr> myValues;
|
|
spdlog::debug("Filling my values into a vector");
|
|
for (auto obs: ANG_OBS){
|
|
double err = get_param_error_from_rootfile(fitFile, obs, 1,0);
|
|
//The easiest rounding in c++
|
|
err = round(getCorrectedError(obs,err,649)*1000.0)/1000.0; //TODO
|
|
double val = round(get_param_value_from_rootfile(fitFile, obs, 1,0)*1000.0)/1000.0;
|
|
myValues.push_back(valErr(val,err));
|
|
}
|
|
std::vector<double>sig_Davids = getSigma(myValues,Davids);
|
|
std::vector<double>sig_Belles = getSigma(myValues,Belles);
|
|
std::vector<double>sig_BaBars = getSigma(myValues,BaBars);
|
|
std::vector<double>sig_B0s = getSigma(myValues,B0s);
|
|
|
|
std::cout << "\\begin{table}[]" << std::endl;
|
|
std::cout << "\\begin{tabular}{l|llll}" << std::endl;
|
|
std::cout << " & \\lhcb \\Bu & Belle (\\Bu) & BaBar (\\Bu + \\Bd) & \\lhcb \\Bd \\\\\\hline" << std::endl;
|
|
//for_indexed(auto myVal: myValues){ //TODO remove: yields warning in compilation due to unused 'myVals'
|
|
for(UInt_t i = 0; i < sig_Davids.size(); i++){
|
|
std::cout << std::setprecision(2) << "$" << ANG_OBS_TLaTeX[i] << "$";
|
|
std::cout << " & " << sig_Davids[i];
|
|
std::cout << " & " << sig_Belles[i];
|
|
std::cout << " & " << sig_BaBars[i];
|
|
std::cout << " & " << sig_B0s[i];
|
|
std::cout << "\\\\" << std::endl;
|
|
}
|
|
|
|
std::cout << " \\end{tabular}" << std::endl;
|
|
std::cout << " \\end{table}" << std::endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|