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

---

SearchTopic.h

Go to the documentation of this file.

#ifndef Impala_Core_Trec_SearchTopic_h
#define Impala_Core_Trec_SearchTopic_h

#include <vector>
#include "Basis/File.h"
#include "Basis/Timer.h"
#include "Basis/FileName.h"
#include "Core/ImageSet/ImageSet.h"
#include "Core/VideoSet/VideoSet.h"
#include "Core/VideoSet/Keyframes.h"

namespace Impala
{
namespace Core
{
namespace Trec
{


class SearchTopic
{
public:

    SearchTopic() : mTimer(1)
    {
        mTopics = 0;
        mImages = 0;
        mThumbnails = 0;
        mVideos = 0;
        mKeyframes = 0;
        mCurTopic = 0;
    }

    SearchTopic(Core::ImageSet::ImageSet* images,
                Core::ImageSet::ImageSet* thumbnails,
                Core::VideoSet::VideoSet* videos) : mTimer(1)
    {
        mImages = images;
        mThumbnails = thumbnails;
        mVideos = videos;
        mKeyframes = 0;

        if (mImages)
            mTopics = mImages;
        else if (mVideos)
            mTopics = mVideos;
        else if (mThumbnails)
            mTopics = mThumbnails;
        mCurTopic = 0;
    }

    // inquiry

    int
    NrTopic()
    {
        if (mTopics)
            return mTopics->NrDirs();
        return mNum.size();
    }

    String
    GetTopicNum(int id)
    {
        if ((id >= 0) && (id < mNum.size()))
            return mNum[id];
        return MakeString(id);
    }

    String
    GetTopicText(int id)
    {
        if ((id >= 0) && (id < mText.size()))
            return mText[id];
        if (mTopics)
            return mImages->GetDir(id);
        return String("");
    }

    bool
    HasVideos()
    {
        return mVideos != 0;
    }

    int
    GetNrExamples(int id)
    {
        if (mTopics)
            return mTopics->GetNrFiles(id);
        return 0;
    }

    String
    GetExampleName(int topic, int example)
    {
        int fileId = mTopics->GetFirstFileId(topic) + example;
        return FileNameBase(mTopics->GetFile(fileId));
    }

    String
    GetVideoExampleName(int topic, int example, bool toWrite)
    {
        int fileId = mVideos->GetFirstFileId(topic) + example;
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        return mVideos->GetFilePathVideoData(fileId, toWrite, false);
#else // REPOSITORY_USED
        ILOG_ERROR("Not implemented yet");
        return "";
#endif // REPOSITORY_USED
    }

    Quid
    GetVideoExampleQuid(int topic, int example)
    {
        if (!mKeyframes)
        {
            mKeyframes = new Core::VideoSet::Keyframes(mVideos, "");
        }

        int fileId = mTopics->GetFirstFileId(topic) + example;
        
        int keyframeId = mKeyframes->GetFrameId(mTopics->GetFile(fileId));
        Quid q = mKeyframes->GetQuidFrame(keyframeId);
        return q;
    }

    Core::Array::Array2dVec3UInt8*
    GetTopicImageExample(int topic, int example, bool thumbnail)
    {
        int fileId = mTopics->GetFirstFileId(topic) + example;
        if (thumbnail && mThumbnails)
            return mThumbnails->GetImage(fileId);
        return mImages->GetImage(fileId);
    }

    String
    GetImageExampleName(int topic, int example, bool thumbnail,
                        bool toWrite = false)
    {
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        if (thumbnail)
        {
            int fileId = mThumbnails->GetFirstFileId(topic) + example;
            return mThumbnails->GetFilePathImageData(fileId, toWrite, false);
        }
        int fileId = mImages->GetFirstFileId(topic) + example;
        return mImages->GetFilePathImageData(fileId, toWrite, false);
#else // REPOSITORY_USED
        ILOG_ERROR("Not implemented yet");
        return "";
#endif // REPOSITORY_USED
    }

    // current topic part

    virtual void
    TopicSelect(int topic)
    {
        mCurTopic = topic;
        mTimer.Start();
    }

    int
    CurTopic()
    {
        return mCurTopic;
    }

    String
    GetCurTopicNum()
    {
        return GetTopicNum(mCurTopic);
    }

    String
    GetCurTopicText()
    {
        return GetTopicText(mCurTopic);
    }

    // number of video examples is equal to number of image examples
    int
    GetCurTopicNrExamples()
    {
        return GetNrExamples(mCurTopic);
    }

    String
    GetCurTopicExampleName(int nr)
    {
        return GetExampleName(mCurTopic, nr);
    }

    Quid
    GetCurTopicExampleQuid(int example)
    {
        return GetVideoExampleQuid(mCurTopic, example);
    }

    String
    GetCurTopicVideoExampleName(int nr)
    {
        return GetVideoExampleName(mCurTopic, nr, false);
    }

    Core::Array::Array2dVec3UInt8*
    GetCurTopicImageExample(int nr, bool thumbnail)
    {
        int fileId = mTopics->GetFirstFileId(mCurTopic) + nr;
        if (thumbnail && mThumbnails)
            return mThumbnails->GetImage(fileId);
        return mImages->GetImage(fileId);
    }

    // time in minutes since last topicSelect
    double
    GetElapsedTime()
    {
        return mTimer.SplitTime() / 60;
    }

protected:

    std::vector<String> mNum;
    std::vector<String> mText;

    Core::Database::RawDataSet* mTopics; // "master" definition topics, is mImages or mVideos
    Core::ImageSet::ImageSet*   mImages;
    Core::ImageSet::ImageSet*   mThumbnails;
    Core::VideoSet::VideoSet*   mVideos;
    Core::VideoSet::Keyframes*  mKeyframes;

    int    mCurTopic;
    Timer  mTimer;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(SearchTopic, Impala.Core.Trec);

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

#endif

---