|
|
//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; }
|