hist1d.hpp

Go to the documentation of this file.

#ifndef H_HIST1D
#define H_HIST1D

#include <vector>
#include <utility>
#include <memory>
#include <set>
#include <limits>

#include "TH1D.h"
#include "TLegend.h"
#include "TCanvas.h"
#include "TPad.h"
#include "TLatex.h"
#include "TLine.h"
#include "TGraphAsymmErrors.h"

#include "core/figure.hpp"
#include "core/process.hpp"
#include "core/axis.hpp"
#include "core/plot_opt.hpp"

class Hist1D final: public Figure{
public:
  class SingleHist1D final: public Figure::FigureComponent{
  public:
    SingleHist1D(const Hist1D &stack,
               const std::shared_ptr<Process> &process,
               const TH1D &hist);
    ~SingleHist1D() = default;

    TH1D raw_hist_;
    mutable TH1D scaled_hist_;

    void RecordEvent(const Baby &baby) final;

    double GetMax(double max_bound = std::numeric_limits<double>::infinity(),
                  bool include_error_bar = false,
                  bool include_overflow = false) const;
    double GetMin(double max_bound = 0.,
                  bool include_error_bar = false,
                  bool include_overflow = false) const;

  private:
    SingleHist1D() = delete;
    SingleHist1D(const SingleHist1D &) = delete;
    SingleHist1D& operator=(const SingleHist1D &) = delete;
    SingleHist1D(SingleHist1D &&) = delete;
    SingleHist1D& operator=(SingleHist1D &&) = delete;

    NamedFunc proc_and_hist_cut_;
    NamedFunc::VectorType cut_vector_, wgt_vector_, val_vector_;
  };

  Hist1D(const Axis &xaxis, const NamedFunc &cut,
             const std::vector<std::shared_ptr<Process> > &processes,
             const std::vector<PlotOpt> &plot_options = {PlotOpt()});
  Hist1D(Hist1D &&) = default;
  Hist1D& operator=(Hist1D &&) = default;
  ~Hist1D() = default;

  void Print(double luminosity,
             const std::string &subdir) final;

  std::set<const Process*> GetProcesses() const final;

  FigureComponent * GetComponent(const Process *process) final;

  std::string Name() const;
  std::string Title() const;

  Hist1D & Weight(const NamedFunc &weight);
  Hist1D & Tag(const std::string &tag);
  Hist1D & RatioTitle(const std::string &numerator,
                      const std::string &denominator);

  Axis xaxis_;
  NamedFunc cut_;
  NamedFunc weight_;
  std::string tag_;
  std::string ratio_numerator_;
  std::string ratio_denominator_;
  std::vector<PlotOpt> plot_options_;

private:
  std::vector<std::unique_ptr<SingleHist1D> > backgrounds_;
  std::vector<std::unique_ptr<SingleHist1D> > signals_;
  std::vector<std::unique_ptr<SingleHist1D> > datas_;

  mutable PlotOpt this_opt_;
  mutable double luminosity_;
  mutable double mc_scale_;
  mutable double mc_scale_error_;
  static TH1D blank_;//<!Blank histogram for creating dummy legend entries

  Hist1D(const Hist1D &) = delete;
  Hist1D& operator=(const Hist1D &) = delete;
  Hist1D() = delete;

  void RefreshScaledHistos();
  void InitializeHistos() const;
  void MergeOverflow() const;
  void ScaleHistos() const;
  void StackHistos() const;
  void NormalizeHistos() const;

  void SetRanges() const;

  void ApplyStyles() const;
  void StyleHisto(TH1D &h) const;
  void AdjustFillStyles() const;

  void GetPads(std::unique_ptr<TCanvas> &c,
               std::unique_ptr<TPad> &top,
               std::unique_ptr<TPad> &bottom) const;

  void FixYAxis(std::vector<TH1D> &bottom_plots) const;

  std::vector<std::shared_ptr<TLatex> > GetTitleTexts() const;
  TGraphAsymmErrors GetBackgroundError() const;
  std::vector<TLine> GetCutLines(double y_min, double y_max, bool adjust_bottom) const;
  std::vector<TH1D> GetBottomPlots(double &the_min, double &the_max) const;
  TLine GetBottomHorizontal() const;

  void StripTopPlotLabels() const;

  double GetMaxDraw(double max_bound = std::numeric_limits<double>::infinity()) const;
  double GetMinDraw(double min_bound = 0.) const;

  std::vector<std::shared_ptr<TLegend> > GetLegends();
  void AddEntries(std::vector<std::shared_ptr<TLegend> > &legends,
                  const std::vector<std::unique_ptr<SingleHist1D> > &hists,
                  const std::string &style,
                  std::size_t n_entries,
                  std::size_t &entries_added) const;
  double GetLegendRatio() const;

  double GetYield(std::vector<std::unique_ptr<SingleHist1D> >::const_iterator h) const;
  double GetMean(std::vector<std::unique_ptr<SingleHist1D> >::const_iterator h) const;

  void GetTitleSize(double &width, double &height, bool in_pixels) const;

  const std::vector<std::unique_ptr<SingleHist1D> >& GetComponentList(const Process *process);
};

#endif