Result.hh

Go to the documentation of this file.

#pragma once

#include <map>

#include "TMVA/Types.h"

#include "boca/generic/Mutable.hh"
#include "boca/plotting/Significance.hh"
#include "boca/branch/Info.hh"

namespace boca
{

class Result
{
public:
    Result() {};
    Result(boca::branch::Info const& info_branch, std::vector<double> const& bdts, TMVA::Types::EMVA mva);
    Result(boca::branch::Info const& info_branch, std::pair<boca::branch::Info, int> const& trainer_info_branch, std::vector<double> const& bdts, TMVA::Types::EMVA mva);
    Result(boca::branch::Info const& info_branch, std::vector<std::vector<bool>> const& passed, TMVA::Types::EMVA mva);
    Result(boca::branch::Info const& info_branch, std::pair<boca::branch::Info, int> const& trainer_info_branch, std::vector<std::vector<bool>> const& passed, TMVA::Types::EMVA mva);
    boca::branch::Info const& Info() const;
    boca::branch::Info const& TrainerInfo() const;
    std::vector<double> const& Bdts() const;
    std::vector<double> const& Events() const;
    std::vector<double> const& PreCutEfficiencies() const;
    std::vector<double> const& PureEfficiencies() const;
    std::vector<Crosssection> const& Crosssections() const;
    std::vector<int> const& PartialSum() const;
    std::vector<int> const& Bins() const;
    static int Steps();
    static double XValue(int value);
    double BestMDValue(Significance significance) const;
    double BestMIValue(Significance significance) const;
    static TMVA::Types::EMVA const& Mva();
    std::vector<Crosssection>& MI(Significance significance);
    std::vector<Crosssection> const& MI(Significance significance)const;
    std::vector<double>& MD(Significance significance);
    std::vector<double> const& MD(Significance significance) const;
    int& BestMDBin(Significance significance);
    int BestMDBin(Significance significance) const;
    int& BestMIBin(Significance significance);
    int BestMIBin(Significance significance)const;
    std::vector<double> const& SelectedEfficiencies() const;
    void AddSelectedEfficiency(double selected_efficiency);
    void AddSelectedEfficiency(int selected_efficiency);
    int TrainerSize() const;
    void SetScalingFactor(double scaling_factor);
    double ScalingFactor() const;
private:
    void Inititialize();
    int XBin(double value) const;
    int trainer_size_;
    static TMVA::Types::EMVA mva_;
    boca::branch::Info info_branch_;
    boca::branch::Info trainer_info_branch_;
    Mutable<std::vector<int>> bins_;
    Mutable<std::vector<int>> event_sums_;
    std::vector<double> bdts_;
    std::vector<double> selected_efficiencies_;
    Mutable<std::vector<double>> events_;
    Mutable<std::vector<double>> efficiencies_;
    Mutable<std::vector<double>> pure_efficiencies_;
    Mutable<std::vector<Crosssection>> crosssections_;
    static std::vector<std::vector<bool>> passed_;
    std::map<Significance, int> best_model_dependent_bin_;
    std::map<Significance, int> best_model_independent_bin_;
    std::map<Significance, std::vector<double>> model_dependent_;
    std::map<Significance, std::vector<Crosssection>> model_independent_;
    double scaling_factor_ = 1;
};

}