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

---

SegmentationDocumentGuiMain.h

Go to the documentation of this file.

#ifndef Impala_Visualization_SegmentationDocumentGuiMain_h
#define Impala_Visualization_SegmentationDocumentGuiMain_h

#include "Visualization/ImageStrip.h"
#include "Visualization/ImageSet.h"
#include "OglGui/TextField.h"
#include "OglGui/TextArea.h"
#include "OglGui/Strut.h"
#include "OglGui/Button.h"
#include "Visualization/LogoWindow.h"
#include "Visualization/DataDocumentGui.h"
#include "Core/VideoSet/SegmentationDocument.h"
#include "Core/VideoSet/PlayShot.h"

namespace Impala
{
namespace Visualization
{


class SegmentationDocumentGuiMain :
        public DataDocumentGui<Core::VideoSet::SegmentationDocument>,
        public ImagesListener, public OglGui::ViewListener
{
    static const int PLAY = 100;

    static const int IM_BOOKMARKED = 1;
    static const int IM_CURRENT = 2;
    static const int IM_RETRIEVED = 3;
    static const int IM_CROSS = 4;

public:

    typedef Core::VideoSet::SegmentationDocument SegmentationDocument;
    typedef Core::Array::Array2dVec3UInt8 Array2dVec3UInt8;

    SegmentationDocumentGuiMain(OglGui::Window* parent,
                                int wndWidth, int wndHeight,
                                int nrImOnRow, int imWidth, int imHeight,
                                double viewScale, bool showAnno,
                                SegmentationDocument* segDoc,
                                Array2dVec3UInt8* logoIm, bool logoAnim,
                                int controlId) :
        DataDocumentGui<SegmentationDocument>(parent, wndWidth, wndHeight,
                                              segDoc, controlId, showAnno,
                                              showAnno, false, 1, 6, showAnno,
                                              showAnno)
    {
        mSegDoc = segDoc;

        int line = 20;
        mBookmarkedImages = new ImageStrip(this, imWidth, imHeight,
                                                viewScale, nrImOnRow);
        mBookmarkedImages->SetImagesListener(this, IM_BOOKMARKED);
        mBookmarkedImages->ActivateInfoBox(false);

        mCurrentShotStrip = 0;
        if (logoAnim || logoIm)
        {
            mCurrentShotStrip = new ImageStrip(this, imWidth, imHeight,
                                               viewScale, nrImOnRow-1);
            new LogoWindow(this, logoIm, logoAnim, 190, 170);
        }
        else
        {
            mCurrentShotStrip = new ImageStrip(this, imWidth, imHeight,
                                               viewScale, nrImOnRow);
        }
        mCurrentShotStrip->SetImagesListener(this, IM_CURRENT);
        mCurrentShotStrip->SetMakeViewWithRect(true);
        mCurrentShotStrip->SetViewListener(this);
        mCurrentShotStrip->ActivateInfoBox(false);
        //mCurrentShotText = new OglGui::TextArea(this, wndWidth-20, 3*line, "");

        // buttons
        OglGui::Button* but;
        OglGui::Strut* strut = new OglGui::Strut(this, 4000, 1);

        OglGui::TextArea* tArea = new OglGui::TextArea(this, 80, line,
                                                       "Current shot");
        tArea->SetBorderType(0);
        but = new OglGui::Button(this, 100, 22, "Play");
        but->SetButtonListener(this, PLAY);
        if (!showAnno)
        {
            but = new OglGui::Button(this, 150, 22, "Bookmark (b)");
            but->SetButtonListener(this, BUT_BOOKMARK);
            but = new OglGui::Button(this, 170, 22, "Unbookmark (B)");
            but->SetButtonListener(this, BUT_UNBOOKMARK);
        }

        strut = new OglGui::Strut(this, 4000, 1);

        mRetrievedImages = new ImageSet(this, imWidth, imHeight, viewScale,
                                        nrImOnRow, 3);
        mRetrievedImages->SetImagesListener(this, IM_RETRIEVED);
        mRetrievedImages->ActivateInfoBox(false);
    }

    virtual void
    HandleNewFile()
    {
        DataDocumentGui<SegmentationDocument>::HandleNewFile();
        HandleNewShot();
    }

    virtual void
    HandleNewShot()
    {
        mCurrentShotStrip->ReplaceImages(mSegDoc->CurKeyframes(),
                                         mSegDoc->GetImSetKeyframes(true));
        Core::Geometry::Rectangle r = mSegDoc->GetRectThumb();
        for (int i=0 ; i<mCurrentShotStrip->NrViewers() ; i++)
            mCurrentShotStrip->GetViewWithRect(i)->SetRect(r);
        //mCurrentShotText->SetText(mSegDoc->CurShotText());
    }

    virtual void
    HandleNewBookmarked()
    {
        DataDocumentGui<SegmentationDocument>::HandleNewBookmarked();
        mBookmarkedImages->ReplaceImages(mSegDoc->GetBookmarkedIds(),
                                         mSegDoc->GetImSetKeyframes(true));
    }

    virtual void
    HandleAddedBookmark()
    {
        mBookmarkedImages->AddImage(mSegDoc->CurKeyfr(),
                                    mSegDoc->GetImSetKeyframes(true));
        DataDocumentGui<SegmentationDocument>::HandleNewBookmarked(); // has no Added
    }

    virtual void
    HandleCursorBookmarked()
    {
        DataDocumentGui<SegmentationDocument>::HandleCursorBookmarked();
        mBookmarkedImages->SetImageSelected(mSegDoc->CurBookmark());
    }

    virtual void
    HandleNewRetrieved()
    {
        /*
        int firstVisY = mRetrievedImages->GetFirstVisY();
        mRetrievedImages->ReplaceImages(mSegDoc->GetRetrieved(),
                                           mSegDoc->GetImSetKeyframes(true));
        mRetrievedImages->SetFirstVisY(0, true);
        */
    }

    virtual void
    ImageSelectionEvent(ImagesWindow* src, int imIndex, void* vData)
    {
        int listenerData = (int)(long long) vData;
        switch (listenerData)
        {
        case IM_BOOKMARKED:
            mSegDoc->CursorToBookmark(imIndex);
            AllHandleNewCursor(Core::Database::LEVEL_SHOT, true);
            break;
        case IM_CURRENT:
            mSegDoc->GotoShotKeyfr(imIndex);
            break;
        case IM_RETRIEVED:
            //mSegDoc->GotoKeyfr(mSegDoc->GetRetrieved()->GetId(imIndex));
            //AllHandleNewCursor(Core::Database::LEVEL_SHOT, false);
            break;
        }
    }

    virtual void
    OnViewMouse(OglGui::View* view, int msg, int btn, int state,
                float x, float y)
    {
        ViewWithRect* viewR = (ViewWithRect*) view;
        Core::Geometry::Rectangle r = viewR->GetRect();
        mSegDoc->SetRectThumb(r);
    }

    virtual void
    ButtonSelectionEvent(OglGui::Button* src, void* vData)
    {
        int listenerData = (long long) vData;
        DataDocumentGui<SegmentationDocument>::ButtonSelectionEvent(src, vData);
        switch (listenerData)
        {
        case PLAY:
            if (! mSegDoc->HasCurShot())
                break;
            Core::VideoSet::PlayShot(mSegDoc->GetSegmentation(),
                                     mSegDoc->CurShot());
            break;
        } // end switch
    }

private:

    SegmentationDocument* mSegDoc;

    ImageStrip*       mBookmarkedImages;
    ImageStrip*       mCurrentShotStrip;
    //OglGui::TextArea* mCurrentShotText;
    ImageSet*         mRetrievedImages;

};

} // namespace Visualization
} // namespace Impala

#endif

---