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