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

---

ThreadRank.h

Go to the documentation of this file.

#ifndef Impala_Core_Trec_ThreadRank_h
#define Impala_Core_Trec_ThreadRank_h

#include "Core/Trec/Thread.h"
#include "Core/Table/SimilarityTableSet.h"
#include "Core/Column/Find.h"
#include "Core/VideoSet/Segmentation.h"

namespace Impala
{
namespace Core
{
namespace Trec
{


class ThreadRank : public Thread
{
public:

    ThreadRank(VideoSet::Segmentation* segmentation, VideoSet::Keyframes* keyframes, Table::SimilarityTableSet* simSet,
               int simId) :
        Thread(std::string("rank_") + simSet->GetName(simId))
    {
        mSimSet = simSet;
        mSimId = simId;
        mKeyframes = keyframes;
        mSegmentation = segmentation;
        mRank = mSimSet->GetRankTable(mSimId);
        mKeyEqualShot = false;
        mUseShotQuids = false;
        Quid t = mRank->Get1(0);
        if (QuidClass(t) == QUID_CLASS_SHOT)
        {
            mUseShotQuids = true;
        }
    }

    virtual int
    GetShot(int position)
    {
        if ((position < 0) || (position >= mRank->Size())) {
            //ILOG_DEBUG("GetShot(" << position << "): fail, position " <<
            //           position << " out of bounds 0, " << mRank->Size());
            return -1;
        }
        Quid quid = mRank->Get1(position);
        if (QuidClass(quid) == QUID_CLASS_SHOT)
        {
            return mSegmentation->GetShotId(quid);
        } else {
            return Key2Shot(mKeyframes->GetFrameId(quid));
        }
    }

    /* retrieve the similarity score for a position within this thread */
    Real64
    GetShotScore(int position)
    {
        Quid q = mRank->Get1(position);
        int pos = mSimSet->FindQuid(q);
        return mSimSet->GetSimTable(mSimId)->Get1(pos);
    }

    virtual int
    GetLength()
    {
        //ILOG_DEBUG("GetLength(): " << mRank->Size());
        return mRank->Size();
    }

    virtual int
    GetShotAt(int shot, int offset)
    {
        Quid quidf;
        if (mUseShotQuids)
        {
            quidf = mSegmentation->GetQuidShot(shot);
        } else {
            quidf = mKeyframes->GetQuidFrame(Shot2Key(shot));
        }
        int r = Column::Find(mRank->GetColumn1(), quidf, 0, mRank->Size());
        int dst = r + offset;
        if ((dst < 0) || (dst >= mRank->Size())) {
            //ILOG_DEBUG("GetShotAt(" << shot << " key=" << key << " offset=" <<
            //           offset << " quidf = " <<
            //           QuidObj(quidf) << "): failed, dst=" << dst);
            return -1;
        }
        Quid quid = mRank->Get1(dst);
        if (QuidClass(quid) == QUID_CLASS_SHOT)
        {
            return mSegmentation->GetShotId(quid);
        } else {
            return Key2Shot(mKeyframes->GetFrameId(quid));
        }
    }

    virtual int
    GetShotPosition(int shot)
    {
        Quid quidf;
        if (mUseShotQuids)
        {
            quidf = mSegmentation->GetQuidShot(shot);
        } else {
            quidf = mKeyframes->GetQuidFrame(Shot2Key(shot));
        }
        int r = Column::Find(mRank->GetColumn1(), quidf, 0, mRank->Size());
        if (r == mRank->Size()) {
            //ILOG_DEBUG("GetShotPosition(" << shot << " quid=" <<
            //           quidf <<"): failed, r = " << r);
            return -1;
        }
        //ILOG_DEBUG("GetShotPosition(" << shot <<
        //           " quid=" << quidf <<"): r = " << r);
        return r;
    }

    virtual int
    GetType()
    {
        return RANK;
    }

private:

    int
    Shot2Key(int shot)
    {
        return (mKeyEqualShot) ? shot : mKeyframes->GetShotRKF(shot);
    }

    int
    Key2Shot(int key)
    {
        return (mKeyEqualShot) ? key : mKeyframes->GetShotId(key);
    }

    Table::SimilarityTableSet*                mSimSet;
    int                                       mSimId;
    Table::SimilarityTableSet::RankTableType* mRank;
    VideoSet::Keyframes*                      mKeyframes;
    VideoSet::Segmentation*                   mSegmentation;
    bool                                      mKeyEqualShot;
    bool                                      mUseShotQuids;

    ILOG_VAR_DEC;
};
ILOG_VAR_INIT(ThreadRank, Core.Trec);


} // namespace Trec
} // namespace Core
} // namespace Impala

#endif

---