|| || || || || || || ||

---

RgbHistExtractor.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_RgbHistExtractor_h
#define Impala_Core_VideoSet_RgbHistExtractor_h

#include "Core/VideoSet/HistogramExtractor.h"
#include "Core/Vector/DivAssign.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{

class RgbHistExtractor : public HistogramExtractor
{

public:

    RgbHistExtractor(Reporter* reporter, CmdOptions& options) :
      HistogramExtractor(reporter, options, "rgbhistogram", 3 * BINS_PER_CHANNEL)
    {
    }

    virtual 
    ~RgbHistExtractor()
    {
    }

protected:

    virtual VectorReal64
    ComputeHistogram(Array::Array2dVec3UInt8* image)
    {
        UInt8* data = image->CPB(0, 0);
        const int nPix = image->CW() * image->CH();

        VectorReal64 histogram(GetBinCount());
        Real64* histR = histogram.GetData();
        Real64* histG = histR + BINS_PER_CHANNEL;
        Real64* histB = histG + BINS_PER_CHANNEL;

        for (int i = 0; i < GetBinCount(); i++)
            histR[i] = 0;

        for (int j = 0; j < nPix; j++) 
        {
            const int r = ComputeBin((int) data[0], 0, 255, BINS_PER_CHANNEL);
            histR[r] += 1;

            const int g = ComputeBin((int) data[1], 0, 255, BINS_PER_CHANNEL);
            histG[g] += 1;

            const int b = ComputeBin((int) data[2], 0, 255, BINS_PER_CHANNEL);
            histB[b] += 1;

            data += 3;
        }

        const Real64 factor = nPix;
        Vector::DivAssign(histogram, factor);

        return histogram;
    }


private:

    static const int BINS_PER_CHANNEL = 32;

    ILOG_VAR_DECL;
};

ILOG_VAR_INIT(RgbHistExtractor, Impala.Core.VideoSet);

} // namespace VideoSet
} // namespace Core
} // namespace Impala

#endif

---