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

---

Color64Extractor.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_Color64Extractor_h
#define Impala_Core_VideoSet_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/VideoSet/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 VideoSet
{

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
    HandleNewFrame(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        if (!src)
        {
            ILOG_ERROR("Not a valid source for fileId " << fileId);
            return;
        }

        Array::Array2dVec3UInt8* im = Array::ArrayCreate<Array::Array2dVec3UInt8>
            (src->FrameWidth(), src->FrameHeight(), 0, 0, src->DataPtr(), true);
        if (!im)
        {
            ILOG_ERROR("Couldn't load image for fileId " << fileId);
            return;
        }

        if (m_verbose)
        {
            std::cout << "Current Frame No. = " << src->FrameNr() << std::endl;
        }

//#ifdef __BITMAP_FUNC__
//        Core::Feature::Bitmap bitmap;
//        bitmap.SaveRgb2BitmapFile("rgb_Color64Extractor.bmp", src->DataPtr(), src->FrameWidth(), src->FrameHeight());
//#endif

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

        Quid quid = vs->GetQuidFrame(fileId, src->FrameNr());
        mHistogramTable->Add(quid, histogram);

        delete im;
        delete pExtractor;

        ILOG_DEBUG("... done keyframe " << src->FrameNr());

    }

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

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.VideoSet);

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

#endif

---