i_color_histogram.h

Go to the documentation of this file.

/*
 * $Id: i_color_histogram.h 17393 2014-08-23 20:19:14Z predoehl $
 */

#ifndef KJB_CPP_COLOR_HISTOGRAM_H
#define KJB_CPP_COLOR_HISTOGRAM_H

#define DEFAULT_COL_HIST_NUM_BINS 8

#include "i_cpp/i_image.h"
#include <m_cpp/m_vector.h>
#include <l_cpp/l_readable.h>
#include <l_cpp/l_writeable.h>

#include <numeric>

namespace kjb 
{

class Color_histogram : public Readable, public Writeable
{
public:

    Color_histogram(unsigned int inum_bins = DEFAULT_COL_HIST_NUM_BINS);

    Color_histogram(unsigned int inum_bins, const kjb::Image & img);

    Color_histogram(const char * fname)
    {
        Readable::read(fname);
    }

    Color_histogram(std::istream &is)
    {
        read(is);
    }

    Color_histogram
    (
        unsigned int inum_bins,
        const Image & img,
        int top_left_x,
        int top_left_y,
        int bottom_right_x,
        int bottom_right_y
    );

    Color_histogram(const Color_histogram & src);

    Color_histogram & operator =(const Color_histogram & src);

    void compute_histogram
    (
        const kjb::Image & img,
        unsigned int inum_bins,
        int top_left_x,
        int top_left_y,
        int bottom_right_x,
        int bottom_right_y
    );

    void compute_histogram(const kjb::Image & img, unsigned int inum_bins);

    void compute_histogram(const kjb::Image & img);

    void compute_histogram
    (
        const kjb::Image & img,
        int top_left_x,
        int top_left_y,
        int bottom_right_x,
        int bottom_right_y
    );

    unsigned int get_num_bins() const
    {
        return num_bins;
    }

    inline double get_value(float r, float g, float b) const
    {
        return m_histogram[compute_rgb_bin(r,g,b)];
    }

    inline double get_value(int ibin_number) const
    {
        return m_histogram[ibin_number];
    }

    void print_human_readable(std::ostream & out) const;

    double compare(const Color_histogram & ch) const;

    double sum() const
    {
        /*
        double dsum = 0.0;
        for(int i = 0; i < m_histogram.size(); ++i)
        {
            dsum+= m_histogram[i];
        }
        return dsum;
        */
        return std::accumulate( m_histogram.begin(), m_histogram.end(), 0.0 );
    }

    virtual void read(std::istream& in);

    virtual void read(const char * fname)
    {
        Readable::read(fname);
    }


    virtual void write(std::ostream& out) const
      ;

    virtual void write(const char * fname) const
    {
        Writeable::write(fname);
    }
private:
    Vector m_histogram;

    unsigned int compute_rgb_bin(float r, float g, float b) const;

    unsigned int find_bin(float ivalue) const;

    unsigned int num_bins;

    float bin_size;
};

void write_histograms(const std::vector<Color_histogram> & chs, const std::string & file_name);

void read_histograms(std::vector<Color_histogram> & chs, const std::string & file_name);


}

#endif /*KJB_CPP_IMAGE_H */