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

---

ThreadSetGui.h

Go to the documentation of this file.

#ifndef Impala_Visualization_ThreadSetGui_h
#define Impala_Visualization_ThreadSetGui_h

#include "Core/Trec/ThreadSet.h"
#include "Core/VideoSet/SegmentationDocument.h"
#include "OglGui/StringSelector.h"
#include "OglGui/TextArea.h"
#include "OglGui/Strut.h"
#include "OglGui/Button.h"
#include "Visualization/ImageSet.h"

namespace Impala {
namespace Visualization {

class ThreadSetGui :    public Window,
                        public ImagesListener,
                        public OglGui::StringListener,
                        public OglGui::ButtonListener
{
    static const int IM_SET     = 1;
    static const int IM_CURRENT = 2;
    static const int BUT_PREV   = 1;
    static const int BUT_NEXT   = 2;

public:
    typedef Core::VideoSet::SegmentationDocument SegmentationDocument;

    ThreadSetGui(OglGui::Window* parent, int wndWidth, int wndHeight,
                 int nrImOnRow, int imWidth, int imHeight, double viewScale,
                 Core::Trec::ThreadSet* threadSet, SegmentationDocument* segDoc,
                 int controlId) :
        Window(parent, wndWidth, wndHeight, true)
    {
        mThreadSet = threadSet;
        mSegDoc = segDoc;
        mCurThread = 0;
        SetControlId(controlId);

        int line = 20;
        mThreadNameList = new OglGui::StringSelector(this, wndWidth - 20,
                                                     6*line);
        mThreadNameList->SetStringListener(this, 0);

        UpdateThreadList();

        mTextArea = new OglGui::TextArea(this, wndWidth - 20, 3*line, "");

        mImageSet = new ImageSet(this, imWidth, imHeight, viewScale,
                                 nrImOnRow, 5);
        mImageSet->SetImagesListener(this, IM_SET);
        mImageSet->ActivateInfoBox(false);

        mCurShotWnd = new ImagesWindow(this, wndWidth/4, wndHeight/4);
        mCurShotWnd->SetImagesListener(this, IM_CURRENT);
        mCurShotWnd->ActivateInfoBox(false);

        OglGui::Button* but;
        but = new OglGui::Button(this, 60, 22, "Prev");
        but->SetRepeatMode(true);
        but->SetButtonListener(this, BUT_PREV);
        but = new OglGui::Button(this, 60, 22, "Next");
        but->SetRepeatMode(true);
        but->SetButtonListener(this, BUT_NEXT);
    }

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

    virtual void ButtonSelectionEvent(OglGui::Button* src, void* vData)
    {
        int curPos = mCurThread->GetShotPosition(mSegDoc->CurShot());
        if (curPos == -1)
            return;
        int listenerData = (int)(long long) vData;
        switch (listenerData)
        {
        case BUT_PREV:
            ThreadStep(curPos - 1);
            break;
        case BUT_NEXT:
            ThreadStep(curPos + 1);
            break;
        } // end switch
    }

    virtual void
    ImageSelectionEvent(ImagesWindow* src, int imIndex, void* listenerData)
    {
        if (listenerData == (void*) IM_SET)
            ThreadStep(imIndex);
    }

    void ShowThread(int idx)
    {
        mThreadNameList->SetCurrentString(idx);
        mCurThread = mThreadSet->GetThread(idx);
        int curShot = mSegDoc->CurShot();
        mCurThread->SetOrigin(curShot);
        mTextArea->SetText(mCurThread->GetName());

        mImageSet->RemoveImages();
        mCurShotWnd->RemoveImages();
        if (curShot == -1)
            return;

        mCurShotWnd->AddImage(mThreadSet->GetImageByShotID(curShot), "Direct",
                              mThreadSet->GetShotName(curShot), true);
        bool useArchive = mThreadSet->GetImageSetThumbnails()->GetUseArchive();
        int  maxNrIm    = useArchive ? 300 : 100;
        int  nrIm       = Min(mCurThread->GetLength(), maxNrIm);
        int firstShot = mCurThread->GetFirstShot();
        for (int i=0 ; i<nrIm ; i++)
        {
            int shot = mCurThread->GetShotAt(firstShot, i);
            if (shot == -1)
                break;
            mImageSet->AddImage(mThreadSet->GetImageByShotID(shot), "Direct",
                                mThreadSet->GetShotName(shot), true);
        }
        mImageSet->SetFirstVisY(0, true);
    }

    void ThreadStep(int position)
    {
        if (!mCurThread || (!mSegDoc->HasCurShot()))
            return;
        int firstShot = mCurThread->GetFirstShot();
        int newShot = mCurThread->GetShotAt(firstShot, position);
        if (newShot == -1)
            return;
        mCurShotWnd->RemoveImages();
        mSegDoc->GotoShot(newShot);
        int curShot = mSegDoc->CurShot();
        mCurShotWnd->AddImage(mThreadSet->GetImageByShotID(curShot), "Direct",
                              mThreadSet->GetShotName(curShot), true);
    }

    void UpdateThreadList()
    {
        mThreadNameList->RemoveStrings();
        for (int i=0 ; i<mThreadSet->GetNrThreads() ; i++)
            mThreadNameList->AddString(mThreadSet->GetThread(i)->GetName());
    }

    virtual void HandleActivate()
    {
        UpdateThreadList();
        ShowThread(0);
    }

private:

    Core::Trec::ThreadSet*  mThreadSet;
    Core::Trec::Thread*     mCurThread;
    SegmentationDocument*   mSegDoc;

    OglGui::StringSelector* mThreadNameList;
    OglGui::TextArea*       mTextArea;
    ImageSet*               mImageSet;
    ImagesWindow*           mCurShotWnd;

};

} // namespace Visualization
} // namespace Impala

#endif

---