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

---

mainTrecResult.cpp

Go to the documentation of this file.

#include "Core/Trec/SearchResult.h"
#include "Core/Trec/SearchJudge.h"
#include "Basis/CmdOptions.h"
#include "Basis/Timer.h"
#include "Visualization/ImageStrip.h"
#include "OglGui/StringSelector.h"
#include "OglGui/TextArea.h"
#include "OglGui/Button.h"
#include "Core/ImageSet/MakeImageSet.h"
#include "Core/VideoSet/MakeVideoSet.h"
#include "Core/VideoSet/Segmentation.h"
#include "Core/VideoSet/Keyframes.h"
#include "Core/VideoSet/PlayShot.h"


// since we are not using libraries:
#include "OglGui/OglLib.cpp"
#include "Link/ImpalaLib.cpp"

namespace Impala
{
namespace Application
{


using namespace Visualization;

class WindowTrecResult : public Window, public OglGui::StringListener,
                         public OglGui::ButtonListener
{
public:

    WindowTrecResult(int width, int height) :
        Window(0, 0, width, height, true)
    {
        CmdOptions& options = CmdOptions::GetInstance();
        mViewScale = 1.0;

        ReadData();
        mSearchResult.Load(options.GetArg(0));
        mSearchResult.Judge(mJudge);
        BuildGUI(width, height);
        SetActiveShot(0);

        AppController::Instance().MainLoop();
    }

    void
    SetActiveShot(int idx)
    {
        mResultWnd->SetCurrentString(idx);
        std::string shotName = mSearchResult.GetShot(idx);
        mCurShotId = mShots->GetShotId(shotName);
        mCurVidId = mShots->GetVideoId(mCurShotId);
        mImageStrip->RemoveImages();
        int nrKey = mKeyframes->GetNrKeyframesShot(mCurShotId);
        int firstKey = mKeyframes->GetFirstKeyframeShot(mCurShotId);
        for (int i=0 ; i<nrKey ; i++)
        {
            std::string imName = mKeyframes->GetName(firstKey + i);
            Core::Array::Array2dVec3UInt8* im = mImSetThumb->GetImage(firstKey + i);
            mImageStrip->AddImage(im, std::string("Direct"), imName, false);
        }
    }

    // specialization of base classes

    virtual void
    StringSelectionEvent(OglGui::StringSelector* src, int strIndex,
                         void* listenerData)
    {
        SetActiveShot(strIndex);
    }

    virtual void
    ButtonSelectionEvent(OglGui::Button* src, void* listenerData)
    {
        if (mCurShotId != -1)
            Core::VideoSet::PlayShot(mShots, mCurShotId);
    }

private:

    void
    ReadData()
    {
        CmdOptions& options = CmdOptions::GetInstance();
        std::string videoSetName = options.GetString("videoSet");
        Timer timer(1);

        std::cout << "Reading thumbnails" << std::endl;
        mImSetThumb =
            Core::ImageSet::MakeImageSet(options.GetString("imageSetThumbnails"));
        if (mImSetThumb)
        {
            mImSetThumb->SetImageSrc(! options.GetBool("noArchive"),
                                     options.GetBool("imFileArchive"),
                                     options.GetBool("imSplitArchive"));
            Core::Array::Array2dVec3UInt8* im = mImSetThumb->GetImage(0);
            mThumbWidth = im->CW();
            mThumbHeight = im->CH();
            delete im;
        }
        std::cout << "time: " << timer.SplitTime() << std::endl;

        std::cout << "Reading video set" << std::endl;
        timer.Start();
        mVidSet = Core::VideoSet::MakeVideoSet(videoSetName);
        std::cout << "time: " << timer.SplitTime() << std::endl;

        std::cout << "Reading shots" << std::endl;
        timer.Start();
        mShots = new Core::VideoSet::Segmentation(mVidSet, "segmentation");;
        std::cout << "time: " << timer.SplitTime() << std::endl;

        std::cout << "Reading keyframe set" << std::endl;
        timer.Start();
        mKeyframes = new Core::VideoSet::Keyframes(mVidSet, "keyframes");
        std::cout << "time: " << timer.SplitTime() << std::endl;

        std::string judgeFile = options.GetString("judgeFile");
        mJudge = new Core::Trec::SearchJudge(judgeFile);
        mJudge->Stats();
    }

    void
    BuildGUI(int wndWidth, int wndHeight)
    {
        SetBackground(0xffd2c799);
        mImageStrip = new ImageStrip(this, mThumbWidth, mThumbHeight,
                                     mViewScale, 5);
        mResultWnd = new OglGui::StringSelector(this, 400, 20*20);
        mResultWnd->SetStringListener(this);
        double ap = mSearchResult.GetAveragePrecision();
        std::string apString = std::string("ap = ") + MakeString(ap);
        mApText = new OglGui::TextArea(this, 100, 1*20, apString);
        mPlayBut = new OglGui::Button(this, 100, 20, "play");
        mPlayBut->SetButtonListener(this);

        for (int i=0 ; i<mSearchResult.NrElements() ; i++)
            mResultWnd->AddString(mSearchResult.Describe(i));
    }

    double                   mViewScale;
    Core::Trec::SearchResult mSearchResult;
    int                      mCurShotId;
    int                      mCurVidId;

    Core::ImageSet::ImageSet*     mImSetThumb;
    Core::VideoSet::VideoSet*     mVidSet;
    Core::VideoSet::Segmentation* mShots;
    Core::VideoSet::Keyframes*    mKeyframes;
    Core::Trec::SearchJudge*      mJudge;

    int mThumbWidth;
    int mThumbHeight;

    ImageStrip*             mImageStrip;
    OglGui::StringSelector* mResultWnd;
    OglGui::TextArea*       mApText;
    OglGui::Button*         mPlayBut;

};

int
mainTrecResult(int argc, char* argv[])
{
    CmdOptions& options = CmdOptions::GetInstance();
    options.Initialise(true, false, true);
    options.SetDefault("wndWidth", "950");
    options.SetDefault("wndHeight", "580");
    if (! options.ParseArgs(argc, argv, "topic_result.xml", 1))
        return 1;

    int reqWndWidth = options.GetInt("wndWidth");
    int reqWndHeight = options.GetInt("wndHeight");
    WindowTrecResult* oglWnd = new WindowTrecResult(reqWndWidth, reqWndHeight);
    if (! oglWnd->Valid())
    {
        std::cout << "WindowTrecResult failed" << std::endl;
        return 0;
    }
    return 1;
}

} // namespace Application
} // namespace Impala

int
main(int argc, char* argv[])
{
    return Impala::Application::mainTrecResult(argc, argv);
}

---