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

---

TableDataSourceTrec.h

Go to the documentation of this file.

#ifndef VideoExcel_TableDataSourceTrec_h
#define VideoExcel_TableDataSourceTrec_h

#include "TrecEngine.h"
#include "TableDataSource.h"

namespace Impala {
namespace Application {
namespace VideoExcel {

class TableDataSourceTrec : public TableDataSource
{
public:
    typedef Impala::Core::VideoSet::Keyframes          Keyframes;
    typedef Impala::Core::ImageSet::ImageSet           ImageSet;

    TableDataSourceTrec():
        TableDataSource()
    {
        Init();
        SetSortColumn("id", TableDataSource::TYPE_TEXT);
    }

    ~TableDataSourceTrec()
    {
    }

    /* get individual row characteristics ************************/
    virtual String
    GetTextDataByID(String column, int row)
    {
        if (column=="id")
            return MakeString(row);
        if (column=="name")
            return mKeyframes->GetName(row);
        if (column=="type")
            return "MPEG1 352x288";
        if (column=="keyframes")
            return MakeString(mKeyframes->GetNrKeyframesShot(mKeyframes->GetShotId(row)));
        return "N/A";
    }

    virtual int
    GetIntDataByID(String column, int row)
    {
        if (column=="keyframes")
        {
            return mKeyframes->GetNrKeyframesShot(mKeyframes->GetShotId(row));
        }
        return 0;
    }

    virtual double
    GetNormalizedDataByID(String column, int row)
    {
        if (column == "isrkf")
        {
            if (mKeyframes->IsRKF(row))
                return 1.0;
            return 0.0;
        }
        return mTrecEngine->GetTableSimilarityScore(mColmap[column], row);
    }

    virtual Array2dVec3UInt8*
    GetImageDataByID(String column, int row)
    {
        Array2dVec3UInt8* ar = 0;
        ImageSet *set = mTrecEngine->GetKeyframeImageSet();
        if (set)
            return set->GetImage(row);
        return 0;
    }

    virtual Array2dVec3UInt8*
    GetVideoDataByID(String column, int row)
    {
        return mTrecEngine->GetNextStill(row);
    }

private:

    void Init()
    {
        mTrecEngine = new IDash::TrecEngine();
        mKeyframes  = mTrecEngine->GetKeyframes();

        if (!mKeyframes)
            ILOG_ERROR("Set does not have keyframes.");

        mFilteredRows = mTotalRows = mKeyframes->GetNrKeyframes();
        ILOG_DEBUG("Trec engine loaded, displaying all keyframes: " <<
                   mFilteredRows);

        AddStaticColumn("rank", TYPE_TEXT, 50);
        AddStaticColumn("image", TYPE_IMAGE, 120);
        AddStaticColumn("name", TYPE_TEXT, 120);
        AddStaticColumn("type", TYPE_TEXT, 100);
        AddStaticColumn("isrkf", TYPE_NORMALIZEDVALUE, 20, 1);
        AddStaticColumn("keyframes", TYPE_TEXT, 30);

        int max = 22;
        for (int i=0; i<mTrecEngine->GetNrTables(); i++)
        {
            AddColumn(mTrecEngine->GetTableName(i),TYPE_NORMALIZEDVALUE,20,4);
            mColmap[mTrecEngine->GetTableName(i)] = i;
            if (i > max)
                break;
        }

        /*
        AddColumn("filename", TYPE_TEXT, 300);
        AddColumn("red", TYPE_NORMALIZEDVALUE, 50, 1);
        AddColumn("green", TYPE_NORMALIZEDVALUE, 50, 2);
        AddColumn("blue", TYPE_NORMALIZEDVALUE, 50, 3);
        */
    }

    IDash::TrecEngine*          mTrecEngine;
    std::map<String, int>       mColmap;
    Keyframes*                  mKeyframes;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(TableDataSourceTrec, Application.VideoExcel);

} // namespace VideoExcel
} // namespace Application
} // namespace Impala

#endif // TableDataSourceTrec_h

---