CorrelationRepo.h

Go to the documentation of this file.

#ifndef CORRELATIONREPO_H
#define CORRELATIONREPO_H
 
#include "Math.h"
#include "Include.h"
#include "ExperimentObs.h"
 
template<typename T>
struct CorrelationMatrixPair {
    SparseMatrix<T> stat;   
    SparseMatrix<T> syst;   
 
    std::pair<double, double> at(const std::pair<T, T>& key) const {
        double statv = stat.contains(key) ? stat.at(key) : 0.0;
        double systv = syst.contains(key) ? syst.at(key) : 0.0;
        return std::make_pair(statv, systv);
    };
 
    void emplace(std::pair<T, T>&& key, double stat_val, double syst_val) {
        stat.emplace(key, stat_val);
        syst.emplace(key, syst_val);
        stat.emplace(std::make_pair(key.second, key.first), stat_val);
        syst.emplace(std::make_pair(key.second, key.first), syst_val);
    };
 
    template<typename U>
    friend std::ostream& operator<<(std::ostream&, const CorrelationMatrixPair<U>&);
};
 
class CorrelationRepository {
public:
    std::pair<double, double> get_correlation(ParamId p1, ParamId p2) const;
 
    std::pair<double, double> get_correlation(const ExperimentObs& o1,
                                              const ExperimentObs& o2) const;
 
    std::pair<double, double> get_correlation(const std::string& experiment,
                                              Observables o1,
                                              Observables o2) const;
 
    std::pair<double, double> get_correlation(const std::string& experiment,
                                              ObservableId o1,
                                              ObservableId o2) const;
 
    std::pair<double, double> get_correlation(const std::string& experiment,
                                              const BinnedObservableId& o1,
                                              const BinnedObservableId& o2) const;
 
    std::pair<double, double> get_correlation(const std::string& exp1,
                                              const BinnedObservableId& o1,
                                              const std::string& exp2,
                                              const BinnedObservableId& o2) const;
 
    template<typename T>
    double get_combined_correlation(T p1, T p2) const {
        auto corr = get_correlation(p1, p2);
        return std::hypot(corr.first, corr.second);
    }
 
    void set_correlation_matrix(std::shared_ptr<CorrelationMatrixPair<ParamId>> correlation_matrices);
 
    void set_correlation_matrix(std::shared_ptr<CorrelationMatrixPair<ExperimentObs>> correlation_matrix);
 
    void merge_correlation_matrix(std::shared_ptr<CorrelationMatrixPair<ParamId>> correlation_matrix);
 
    void merge_correlation_matrix(std::shared_ptr<CorrelationMatrixPair<ExperimentObs>> correlation_matrix);
 
    void print_content() const;
 
private:
    std::shared_ptr<CorrelationMatrixPair<ParamId>> parameter_correlations;   
    std::shared_ptr<CorrelationMatrixPair<ExperimentObs>> observable_correlations; 
};
 
#endif // CORRELATIONREPO_H