Histogram1D.hpp

Go to the documentation of this file.

 
#ifndef H_HISTOGRAM1D
#define H_HISTOGRAM1D
 
#include <vector>
#include <string>
#include <iostream>
 
#include <nlohmann/json.hpp>
 
#include <zlib.h>
 
#include "binners.hpp"
 
 
namespace pottu {
 
    template <class Binner, class Cell>
    class Histogram1D {
    public:
 
    Histogram1D( const Binner &binner )
        : _binner(binner)
    {
        _data.resize( _binner.binCount(), Cell() );
    }
 
    Histogram1D( typename Binner::value_type low, typename Binner::value_type high, unsigned int binc )
        : _binner( low, high, binc )
    {
        _data.resize( _binner.binCount(), Cell() );
    }
 
    const std::string &title() const { return( _title ); }
    void setTitle( const std::string &newtitle ) { _title = newtitle; }
 
    const std::string &xlabel() const { return( _xlabel ); }
    void setXlabel( const std::string &newxlabel ) { _xlabel = newxlabel; }
 
    const std::string &ylabel() const { return( _ylabel ); }
    void setYlabel( const std::string &newylabel ) { _ylabel = newylabel; }
    
    void fill( typename Binner::value_type value ) noexcept {
        if( _binner.isUnderflow(value) )
        ++_underflows;
        else if( _binner.isOverflow(value) )
        ++_overflows;
        else
        ++_data[_binner(value)];
    }
 
    void fill( typename Binner::value_type value, Cell amount ) noexcept {
        if( _binner.isUnderflow(value) )
        ++_underflows;
        else if( _binner.isOverflow(value) )
        ++_overflows;
        else
        _data[_binner(value)] += amount;
    }
 
    void fillNoRangeTesting( typename Binner::value_type value ) noexcept {
        ++_data[_binner(value)];
    }
 
 
 
    void exportAscii( std::ostream &os ) const {
        os << "# histogram1D\n"
           << "# title      = \"" << _title << "\"\n"
           << "# xlabel     = \"" << _xlabel << "\"\n"
           << "# ylabel     = \"" << _ylabel << "\"\n"
           << "# low        = " << _binner.low() << "\n"
           << "# high       = " << _binner.high() << "\n"
           << "# binCount   = " << _binner.binCount() << "\n"
           << "# binWidth   = " << _binner.binWidth() << "\n"
           << "# underflows = " << _underflows << "\n"
           << "# overflows  = " << _overflows << "\n"
           << "# Data as (Bin_center,counts): \n";
        for( unsigned int i = 0; i < _binner.binCount(); ++i )
        os << _binner.binCenter(i) << "\t" << _data[i] << "\n";
    }
 
    nlohmann::json getAsJSON() const {
        nlohmann::json j;
        j["type"]   = "Histogram1D";
        j["title"]  = _title;
        j["xlabel"] = _xlabel;
        j["ylabel"] = _ylabel;
        {
        nlohmann::json b;
        b["low"] = _binner.low();
        b["high"] = _binner.high();
        b["bincount"] = _binner.binCount();
        b["binwidth"] = _binner.binWidth();
        j["binner"] = std::move(b);
        }
        j["underflows"] = _underflows;
        j["overflows"]  = _overflows;
        j["data"]       = _data;
        return j;
    }
 
    
    Binner              _binner;
    std::vector<Cell>   _data;
    Cell                _underflows = 0;
    Cell                _overflows = 0;
 
    std::string         _title;
    std::string         _xlabel;
    std::string         _ylabel;
    
 
    };
 
}
 
#endif