230 lines
9.0 KiB
C++
230 lines
9.0 KiB
C++
|
//Renata Kopecna
|
||
|
|
|
||
|
|
#ifndef HELPERS_HH
|
||
|
|
#define HELPERS_HH
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <vector>
|
||
|
|
#include <TF1.h>
|
||
|
|
#include <TH1D.h>
|
||
|
|
#include <TStopwatch.h>
|
||
|
|
#include <parse.hh>
|
||
|
|
|
||
|
|
//indexed auto looping over vectors
|
||
|
|
#define for_indexed(...) for_indexed_v(i, __VA_ARGS__)
|
||
|
|
#define for_indexed_v(v, ...) for(bool _i_ = true, _break_ = false; _i_;) for(size_t v = 0; _i_; _i_ = false) for(__VA_ARGS__) if(_break_) break; else for(bool _j_ = true; _j_;) for(_break_ = true; _j_; _j_ = false) for(bool _k_ = true; _k_; v++, _k_ = false, _break_ = false)
|
||
|
|
/// usage:
|
||
|
|
/// std::vector<int> v{1, 2, 3};
|
||
|
|
///for_indexed (auto const& item : v) { //the index is always i
|
||
|
|
/// if (i > 0) std::cout << ", ";
|
||
|
|
/// std::cout << i << ": " << item;
|
||
|
|
///}
|
||
|
|
/// for_indexed_v (my_counter, auto const& item : v) //the index is v here
|
||
|
|
|
||
|
|
|
||
|
|
//TODO: work on more forward declarations, it reduces the compiler time as cross-referencing is not recompiled 400 times anymore
|
||
|
|
//TODO: simirarly move your includes into .cpp in order to avoid this
|
||
|
|
|
||
|
|
namespace fcnc { //Forward declaration, fully defined in options.hh
|
||
|
|
class options;
|
||
|
|
class parameters;
|
||
|
|
class bu2kstarmumu_parameters;
|
||
|
|
class event;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//----------------------//
|
||
|
|
// Print utils //
|
||
|
|
//----------------------//
|
||
|
|
std::string boolToString(bool isTrue);
|
||
|
|
|
||
|
|
//reset the spdlog into my default
|
||
|
|
void reset_spdlog();
|
||
|
|
|
||
|
|
//set spdlog level
|
||
|
|
void set_spdlog_level(int verboseLvl);
|
||
|
|
|
||
|
|
//what lever is the spdlog?
|
||
|
|
bool spdlog_trace();
|
||
|
|
bool spdlog_debug();
|
||
|
|
bool spdlog_info();
|
||
|
|
bool spdlog_warn();
|
||
|
|
bool spdlog_error();
|
||
|
|
|
||
|
|
//Get job_id and turn it in a string
|
||
|
|
std::string get_sample_from_jobID(int job_id);
|
||
|
|
|
||
|
|
//Make pretty outputs with given precision (feault is two)
|
||
|
|
std::string format_double(double value, unsigned int digits=2);
|
||
|
|
std::string format_value(double value, double error, unsigned int digits=2);
|
||
|
|
std::string format_error(double error, unsigned int digits=2);
|
||
|
|
|
||
|
|
//----------------------//
|
||
|
|
// Year utils //
|
||
|
|
//----------------------//
|
||
|
|
|
||
|
|
//Get years according to a run
|
||
|
|
std::vector<int> get_years(int Run, bool MCsig, bool MCref);
|
||
|
|
//Check if year makes sense
|
||
|
|
void check_year(int year);
|
||
|
|
int get_yearFromRun(int year);
|
||
|
|
|
||
|
|
int MC_dataset(bool Reference, bool PHSP);
|
||
|
|
//--------------------------------------//
|
||
|
|
// General helpers //
|
||
|
|
//--------------------------------------//
|
||
|
|
|
||
|
|
//Copy file
|
||
|
|
int copyFile(std::string from, std::string to);
|
||
|
|
//Check if file/folder exists
|
||
|
|
bool existsTest (const std::string& name);
|
||
|
|
//Create new folder
|
||
|
|
void makeFolder (const std::string& name);
|
||
|
|
|
||
|
|
//Convert vector into number-space-number... string
|
||
|
|
std::string convert_vector_to_string(std::vector<int> myVector);
|
||
|
|
std::string convert_vector_to_string(std::vector<double> myVector);
|
||
|
|
std::string convert_vector_to_string(std::vector<std::string> myVector);
|
||
|
|
|
||
|
|
//Merge two vectors
|
||
|
|
std::vector<double> merge_two_vecs(std::vector<double> one, std::vector<double> two);
|
||
|
|
std::vector<int> merge_two_vecs(std::vector<int> one, std::vector<int> two);
|
||
|
|
std::vector<std::string> merge_two_vecs(std::vector<std::string> one, std::vector<std::string> two);
|
||
|
|
|
||
|
|
//Replace a part of a string
|
||
|
|
bool replace(std::string& str, const std::string& from, const std::string& to);
|
||
|
|
|
||
|
|
//Sum a vectors
|
||
|
|
int sum_vector(std::vector<int> vec);
|
||
|
|
double sum_vector(std::vector<double> vec);
|
||
|
|
|
||
|
|
//Return the center of a bin
|
||
|
|
double bin_center(double low, double high);
|
||
|
|
//Return the center of a q2 bin
|
||
|
|
double bin_center_q2(fcnc::options opts, int b);
|
||
|
|
double bin_center_q2(fcnc::options *opts, int b);
|
||
|
|
double bin_center_q2(std::vector<double> q2min, std::vector<double> q2max , int b);
|
||
|
|
|
||
|
|
//Return half of the width (useful for errors)
|
||
|
|
double bin_halfWidth(double low, double high);
|
||
|
|
double bin_halfWidth_q2(fcnc::options opts, int b);
|
||
|
|
double bin_halfWidth_q2(fcnc::options *opts, int b);
|
||
|
|
double bin_halfWidth_q2(std::vector<double> q2min, std::vector<double> q2max, int b);
|
||
|
|
|
||
|
|
//Check if the angles are in the correct range given by options
|
||
|
|
bool isEvtInAngleRange(fcnc::event *evt, fcnc::options *opts);
|
||
|
|
//Check if the angles are in the correct range given by default
|
||
|
|
bool isEvtInAngleRange(fcnc::event *evt);
|
||
|
|
//For now filter out events that are dumb dumb for folding four
|
||
|
|
bool filterFldFour(fcnc::event *evt, fcnc::options *opts);
|
||
|
|
|
||
|
|
//Check if value is in a vector
|
||
|
|
bool isInVec(int key, std::vector<int>vec);
|
||
|
|
bool isInVec(double key, std::vector<double>vec);
|
||
|
|
bool isInVec(std::string key, std::vector<std::string>vec);
|
||
|
|
|
||
|
|
|
||
|
|
//--------------------------------------//
|
||
|
|
// Helpers for converting the tuples //
|
||
|
|
//--------------------------------------//
|
||
|
|
|
||
|
|
//Return's OG tuple name
|
||
|
|
std::string get_inputTree_name(int job_id);
|
||
|
|
|
||
|
|
//--------------------------------------//
|
||
|
|
// Helpers for angular corrections //
|
||
|
|
//--------------------------------------//
|
||
|
|
|
||
|
|
TF1 *fitStraightLine(TH1D* hist, std::string lineName, double lowEdge, double highEdge);
|
||
|
|
|
||
|
|
//--------------------------------------//
|
||
|
|
// Helpers for bu2kstarmumu_pdf //
|
||
|
|
//--------------------------------------//
|
||
|
|
|
||
|
|
//Returns a vector with the {p wave observables, obervable in tex}, useful for histograms in a loop
|
||
|
|
const std::vector<std::vector<std::string>> get_angObser_withTeX_vec();
|
||
|
|
|
||
|
|
|
||
|
|
//Returns a vector with the p wave observables
|
||
|
|
std::vector<std::string> get_angObser_vec();
|
||
|
|
|
||
|
|
//Returns a vector with the p wave observables in tex
|
||
|
|
std::vector<std::string> get_angObserTeX_vec();
|
||
|
|
|
||
|
|
//--------------------------------------//
|
||
|
|
// Helpers for time measuring //
|
||
|
|
//--------------------------------------//
|
||
|
|
|
||
|
|
struct runTime{
|
||
|
|
//Measure the time for the fit:
|
||
|
|
std::vector<Double_t> real_times;
|
||
|
|
std::vector<Double_t> cpu_times;
|
||
|
|
std::vector<Double_t> cpp_times;
|
||
|
|
TStopwatch * sw;// = new TStopwatch();
|
||
|
|
runTime(){
|
||
|
|
sw = new TStopwatch();
|
||
|
|
}
|
||
|
|
void print(unsigned int nBins);
|
||
|
|
void start();
|
||
|
|
void stop(time_t startTime);
|
||
|
|
};
|
||
|
|
|
||
|
|
//--------------------------------------//
|
||
|
|
// Helpers for fit scripts //
|
||
|
|
//--------------------------------------//
|
||
|
|
|
||
|
|
//get a vector of all available angle bkg parameters
|
||
|
|
std::vector<std::string> param_string_bkg();
|
||
|
|
//get a vector of all available angle mkpi parameters
|
||
|
|
std::vector<std::string> param_string_bkg_mkpi();
|
||
|
|
//get a string vector of all effectivelly used parameters
|
||
|
|
std::vector<std::string> param_string(fcnc::options opts, bool MC);
|
||
|
|
//get a string vector of mass parameters
|
||
|
|
std::vector<std::string> params_string_mass(fcnc::options opts);
|
||
|
|
//Reads MC fit files (massFitOnly) and gets the sigma ratio of ref/MC for given bin and pdf
|
||
|
|
double get_sigmaRatio_fromMC(basic_params params, int nBins, int bin, int pdf);
|
||
|
|
|
||
|
|
//---------------------------------------------//
|
||
|
|
// Helpers for printing and saving fit results //
|
||
|
|
//---------------------------------------------//
|
||
|
|
|
||
|
|
void print_all_parameters(unsigned int nBins, fcnc::bu2kstarmumu_parameters *theParams[],
|
||
|
|
int spdlog_level, std::string savePath);
|
||
|
|
void print_all_parameters(unsigned int nBins, std::vector<UInt_t> pdf_idx,
|
||
|
|
std::vector<fcnc::parameters*> theParams[],
|
||
|
|
int spdlog_level);
|
||
|
|
|
||
|
|
void tex_all_parameters(unsigned int nBins, std::vector<UInt_t> pdf_idx,
|
||
|
|
std::vector<fcnc::parameters*> theParams[]);
|
||
|
|
|
||
|
|
void print_fit_results(unsigned int nBins, std::vector<Int_t> fitresults);
|
||
|
|
void print_fit_results(unsigned int nBins, std::vector<UInt_t> pdf_idx, std::vector<Int_t> fitresults[], bool simFit);
|
||
|
|
void print_sig_yields(unsigned int nBins, std::vector<UInt_t> pdf_idx,
|
||
|
|
std::vector<UInt_t> *evts_cntr, std::vector<double> *f_sigs, std::vector<double> *f_sigserr);
|
||
|
|
void print_bkg_yields(unsigned int nBins, std::vector<UInt_t> pdf_idx,
|
||
|
|
std::vector<UInt_t> *evts_cntr, std::vector<double> *f_sigs, std::vector<double> *f_sigserr);
|
||
|
|
void print_bkgOnly_yields(unsigned int nBins, std::vector<UInt_t> pdf_idx,
|
||
|
|
std::vector<fcnc::bu2kstarmumu_parameters*> theParams[],
|
||
|
|
std::vector<double> bkg_int_full_range[], std::vector<UInt_t> *evts_cntr);
|
||
|
|
void print_sig_yields_tex(std::string texFiletag,
|
||
|
|
unsigned int nBins, std::vector<UInt_t> pdf_idx,
|
||
|
|
fcnc::options *theOptions,
|
||
|
|
std::vector<UInt_t> *evts_cntr, std::vector<double> *f_sigs, std::vector<double> *f_sigserr);
|
||
|
|
|
||
|
|
int save_results(std::string results_file, unsigned int nBins, std::vector<UInt_t> pdf_idx, std::vector<int>*fit_results, std::vector<fcnc::parameters*> *theParams, bool simFit, fcnc::options *opts);
|
||
|
|
int save_results(std::string results_file, unsigned int nBins, int n_pdf, std::vector<int> fit_results, fcnc::bu2kstarmumu_parameters **theParams, fcnc::options *opts);
|
||
|
|
//--------------------------------------//
|
||
|
|
// Helpers for writting latex stuff //
|
||
|
|
//--------------------------------------//
|
||
|
|
|
||
|
|
int clear_Latex_noteFile(std::string tag); //Deletes the old file
|
||
|
|
int open_Latex_noteFile(std::string tag, std::ofstream &myfile); //Appends output at the end of the file
|
||
|
|
|
||
|
|
|
||
|
|
//--------------------------------------//
|
||
|
|
// Helpers for reading the parameters //
|
||
|
|
//--------------------------------------//
|
||
|
|
|
||
|
|
std::vector<double> load_param_values_into_vector(std::string paramName, std::string fileName);
|
||
|
|
#endif // HELPERS_HH
|