source: trunk/src/Histogram.h @ 231

// $Id: Histogram.h 231 2005-02-21 14:50:17Z peter $

#ifndef _theplu_cpptools_histogram_
#define _theplu_cpptools_histogram_

#include <string>
#include <vector>

#include "WeightedAverager.h"


namespace theplu {
namespace cpptools {

  ///
  /// Histograms provide a convenient way of presenting the
  /// distribution of a set of data. A histogram consists of a set of
  /// bins which count the number of events falling into these
  /// bins. Currently only one dimensional histograms with uniformly
  /// spaced bins are supported.
  ///
  class Histogram
  {
  public:

    ///
    /// The default constructor.
    ///
    Histogram(void);

    ///
    /// The copy constructor.
    ///
    Histogram(const Histogram&);

    ///
    /// Construct a histogram object that covers \f$(xmin,xmax]\f$
    /// with the bin spacing \f$(xmax-xmin)/n\f$.
    ///
    Histogram(const double xmin, const double xmax, const size_t n);

    virtual ~Histogram(void);

    ///
    /// Update the histogram by adding \a weight to the bin whose
    /// range contains the observation \a x. No bins are updated when
    /// \a x lies outside the range of the histogram but the value is
    /// added to the overall integral of the histogram.
    ///
    /// @short Add an data point to the histogram.
    ///
    /// @return 0 if \a x lies within the range of the histogram, -1
    /// if \a x is smaller than the lower limit of the histogram, and
    /// similarly, 1 is returned if \a x is greater than or equal to
    /// the upper limit.
    ///
    int add(const double x,const double weight=1.0);

    ///
    /// Gives access to the WeightedAverager object that keeps track of
    /// average of all events presented to the histogram.
    ///
    /// @short Average of all events presented to the histogram.
    ///
    /// @return A const reference to an WeightedAverager object.
    ///
    inline const statistics::WeightedAverager& averager_all(void) const
      { return sum_all_; }

    ///
    /// Gives access to the WeightedAverager object that keeps track of
    /// average of events that fits within the histogram lower and
    /// upper limits. This function is equivalent to averager().
    ///
    /// @short Average of events fitting within histogram.
    ///
    /// @return A const reference to an WeightedAverager object.
    ///
    inline const statistics::WeightedAverager& averager_histogram(void) const
      { return sum_histogram_; }

    ///
    /// @return The number of bins in the histogram
    ///
    inline size_t nof_bins(void) const { return histogram_.size(); }

    ///
    /// Rescale the histogram using the scale factor \a
    /// scale_factor. You may need to averager_all or
    /// averager_histogram to set the proper scale factor.
    ///
    void normalize(double scale_factor=1);

    ///
    /// @return The value of the histogram corresponding to bin \a
    /// bin.
    ///
    /// @note No check is done that \a bin is within the size of the
    /// histogram.
    ///
    inline double observation_value(const size_t bin) const
      { return xmin_+spacing()*(bin+0.5); }

    ///
    /// Set everyting to default values, here it means that everything
    /// is set to zero except the boundary values that are kept.
    ///
    void reset(void);

    ///
    /// @return The size of the bins in the histogram.
    ///
    inline double spacing(void) const { return (xmax_-xmin_)/nof_bins(); }

    ///
    /// @return The histogram upper boundary.
    ///
    /// @note The upper boundary value is inside the histogram.
    ///
    inline double xmax(void) const { return xmax_; }

    ///
    /// @return The histogram lower boundary.
    ///
    /// @note The lower boundary value is outside the histogram.
    ///
    inline double xmin(void) const { return xmin_; }

    ///
    /// @return The count of bin \a bin.in the histogram.
    inline double operator[](size_t bin) const { return histogram_[bin]; } 

    ///
    /// The assignment operator
    const Histogram& operator=(const Histogram&);

  private:
    // Returns zero if outside boundaries
    inline size_t bin(double d)
    { return (((d<xmin_) || (d>xmax_)) ? 0 :
              static_cast<size_t>(floor((d-xmin_)/spacing() ))); }

    std::vector<double> histogram_;
    double xmax_;
    double xmin_;
    statistics::WeightedAverager sum_all_;      // average of all data
    statistics::WeightedAverager sum_histogram_;// average of data in histogram
  };

///
/// The Histogram output operator
///
std::ostream& operator<<(std::ostream& s,const Histogram&);

}} // of namespace cpptools and namespace theplu

#endif