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

---

Color64Extractor.h

Go to the documentation of this file.

#ifndef Impala_Core_ImageSet_Color64Extractor_h
#define Impala_Core_ImageSet_Color64Extractor_h

// extract color 64 bin features, provided by xirong.
// color64: 44-d color correlgoram, 14-d color texture moment, and 6-d RGB color moment

#include "Core/ImageSet/Reporter.h"
#include "Core/Vector/VectorTem.h"
#include "Core/Feature/FeatureTable.h"
#include "Core/Feature/Color64.h"
//#include "Core/Stream/SeqConvKernel.h"
//#include "Core/Array/RGB2Gray.h"

namespace Impala
{
namespace Core
{
namespace ImageSet
{

class Color64Extractor : public Listener
{

protected:

    typedef Vector::VectorTem<Real64> VectorReal64;

public:

    Color64Extractor(Reporter* reporter, 
        CmdOptions& options, CString featureName, int binCount)
    {
        mReporter = reporter;
        mFeatureName = featureName;
        mBinCount = binCount;
        Feature::FeatureDefinition def(GetFeatureName());
        mHistogramTable = new Feature::FeatureTable(def, 1000, GetBinCount());

        m_verbose = FALSE;
        m_verbose2 = FALSE;
    }

    Color64Extractor(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        mFeatureName = "color64"; // default feature name
        mBinCount = 64;
        Feature::FeatureDefinition def(GetFeatureName());
        mHistogramTable = new Feature::FeatureTable(def, 1000, GetBinCount());

        m_verbose = FALSE;
        m_verbose2 = FALSE;
        //m_verbose = TRUE;
        //m_verbose2 = TRUE;
        
    }

    virtual 
    ~Color64Extractor()
    {
        delete mHistogramTable;
    }

    String
    GetFeatureName() const
    {
        return mFeatureName;
    }

    virtual void
    //HandleNewFile(ImageSet* is, int fileId, Stream::RgbDataSrc* src)
    HandleNewFile(ImageSet* is, int fileId, Array::Array2dVec3UInt8* im)
    {
        if (!im)
        {
            ILOG_ERROR("Couldn't load image for fileId " << fileId);
            return;
        }

        if (m_verbose)
        {
            std::cout << "Current File No. = " << fileId << std::endl;
        }

#ifdef __BITMAP_FUNC__
        Core::Feature::Bitmap bitmap;
        bitmap.SaveRgb2BitmapFile("rgb_Color64Extractor.bmp", im->CPB(), im->CW(), im->CH());

        Impala::String fName = "rgb_Color64Extractor.png";
        Array::WritePng(im, fName, is->GetDatabase());
#endif

        VectorReal64 histogram(mBinCount);
                
        Core::Feature::Color64* pExtractor = new Core::Feature::Color64(mFeatureName);
        pExtractor->Extract(im, histogram);

        Quid quid = is->GetQuidImage(fileId);
        mHistogramTable->Add(quid, histogram);

        delete pExtractor;

        ILOG_DEBUG("... done file " << fileId);

    }

    virtual void
    //HandleDoneDir(ImageSet* is, int dirId, Stream::RgbDataSrc* src)
    HandleDoneDir(ImageSet* is, int dirId)
    {
        bool binary = true;
        Feature::FeatureDefinition def = mHistogramTable->GetFeatureDefinition();
        //std::string fName = is->GetFilePathFeatureData("Keyframes", def, dirId, false, -1, true, false);
        std::string fName = is->GetFilePathFeatureData(def, dirId, false, -1, true, false);
        
        if (!fName.empty())
            Table::Write(mHistogramTable, fName, is->GetDatabase(), binary);
        mHistogramTable->SetSize(0);

        ILOG_DEBUG("... done dir " << dirId);
    }

protected:

    int
    GetBinCount() const
    {
        return mBinCount;
    }

private:

    //static const int BINS_PER_CHANNEL = 32;

    Reporter*              mReporter;
    String                 mFeatureName;
    int                    mBinCount;
    Feature::FeatureTable* mHistogramTable;

    BOOL                   m_verbose;
    BOOL                   m_verbose2; // for dedug only; more detailed info

    ILOG_VAR_DECL;
};

ILOG_VAR_INIT(Color64Extractor, Impala.Core.ImageSet);

} // namespace ImageSet
} // namespace Core
} // namespace Impala

#endif

---