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

---

TableDataSourceTrec.h

Go to the documentation of this file.

#ifndef MediaTable_TableDataSourceTrec_h
#define MediaTable_TableDataSourceTrec_h

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

namespace Impala {
namespace Application {
namespace MediaTable {

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

    TableDataSourceTrec():
        TableDataSource()
    {
        Init();
    }

    ~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=="shotname")
                        return mTrecEngine->GetShots()->GetName(mKeyframes->GetShotId(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 double
    GetDoubleDataByID(String column, int row)
    {
        return GetNormalizedDataByID(column, row);
    }

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

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

private:

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

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

        ILOG_DEBUG("Trec engine loaded, displaying all keyframes: " <<
                   GetTotalRows());

        AddStaticColumn("shotname", TYPE_TEXT, 90);

        //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;
        }
    }

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

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(TableDataSourceTrec, Application.MediaTable);

} // namespace MediaTable
} // namespace Application
} // namespace Impala

#endif // TableDataSourceTrec_h

---