axis.hpp

Go to the documentation of this file.

#ifndef H_AXIS
#define H_AXIS

#include <cstddef>

#include <vector>
#include <set>
#include <string>

#include "core/named_func.hpp"

class Axis{
public:
  Axis(const std::vector<double> &bins,
       const NamedFunc &var,
       const std::string &title = "",
       const std::set<double> &cut_vals = {});
  Axis(std::size_t nbins,
       double xmin, double xmax,
       const NamedFunc &var,
       const std::string &title = "",
       const std::set<double> &cut_vals = {});

  Axis(const Axis &) = default;
  Axis& operator=(const Axis &) = default;
  Axis(Axis &&) = default;
  Axis& operator=(Axis &&) = default;
  ~Axis() = default;

  std::size_t Nbins() const;
  Axis & Bins(const std::vector<double> &bins);
  Axis & Bins(std::size_t nbins, double xmin, double xmax);
  const std::vector<double> & Bins() const;
  double AvgBinWidth() const;

  std::string Title() const;

  NamedFunc var_;
  std::string title_;
  std::string units_;
  std::set<double> cut_vals_;

private:
  std::vector<double> bins_;
  
  static std::vector<double> GetEdges(std::size_t nbins, double xmin, double xmax);
  void ParseUnits();
};

#endif