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

---

LabHist3dExtractor.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_LabHist3dExtractor_h
#define Impala_Core_VideoSet_LabHist3dExtractor_h

#include "Core/VideoSet/HistogramExtractor.h"

#include "Core/Array/Element/Vec3Real64.h"
#include "Core/Array/Element/Color.h"
#include "Core/Vector/DivAssign.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{

class LabHist3dExtractor : public HistogramExtractor
{

protected:

    typedef Vector::VectorTem<Array::Array2dScalarReal64> Array3dReal64;

public:

    LabHist3dExtractor(Reporter* reporter, CmdOptions& options) :
      HistogramExtractor(reporter, options, "labhistogram3d", 
          BINS_FOR_L * BINS_FOR_A_B * BINS_FOR_A_B)
    {
    }

    virtual 
    ~LabHist3dExtractor()
    {
    }

protected:

    // Lab histogram computation - maintaining L-a-b relation information 
    // using a 3D array.
    virtual VectorReal64
    ComputeHistogram(Array::Array2dVec3UInt8* image)
    {
        const UInt8* data = image->CPB(0, 0);
        const int nPix = image->CW() * image->CH();

        VectorReal64 histogram(GetBinCount());
        Real64* const histData = histogram.GetData();
        for (int i = 0; i < GetBinCount(); i++)
            histData[i] = 0;

        for (int j = 0; j < nPix; j++) 
        {
            const Array::Element::Vec3Real64 pixel = 
                Array::Element::Vec3Real64(data[0], data[1], data[2]);
            const Array::Element::Color crgb(pixel, Array::Element::RGB);

            Array::Element::Color clab = crgb.ToLab();
            Array::Element::Vec3Real64 lab = clab.Value();

            int l = ComputeBin(lab.X(), 0.0, 100.0, BINS_FOR_L);
            int a = ComputeBin(lab.Y(), -90.0, 90.0, BINS_FOR_A_B);
            int b = ComputeBin(lab.Z(), -90.0, 90.0, BINS_FOR_A_B);

            int offset = 
                (l * BINS_FOR_A_B * BINS_FOR_A_B) + (a * BINS_FOR_A_B) + b;
            histData[offset] += 1;

            data += 3;
        }

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

private:

    static const int BINS_FOR_L = 6;
    static const int BINS_FOR_A_B = 5;

    ILOG_VAR_DECL;
};

ILOG_VAR_INIT(LabHist3dExtractor, Impala.Core.VideoSet);

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

#endif

---