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

---

SegmentationDocumentGuiKeyframes.h

Go to the documentation of this file.

//345678901234567890123456789012345678901234567890123456789012345678901234567890
#ifndef Impala_Visualization_SegmentationDocumentGuiKeyframes_h
#define Impala_Visualization_SegmentationDocumentGuiKeyframes_h

#include "Core/VideoSet/SegmentationDocument.h"
#include "Visualization/ImageSet.h"
#include "Visualization/AppControlDoc.h"
#include "Visualization/ImageSetIdxGridScroller.h"

namespace Impala
{
namespace Visualization
{


class SegmentationDocumentGuiKeyframes :
        public Window,
        public AppControlDoc,
        public Core::Database::DataDocumentListener,
        public ImageSetIdxGridListener,
        public ImagesListener
{
public:
    typedef Core::VideoSet::SegmentationDocument SegmentationDocument;

    SegmentationDocumentGuiKeyframes(OglGui::Window* parent,
                                     int wndWidth, int wndHeight, int nrImOnRow,
                                     int thumbWidth, int thumbHeight,
                                     double viewScale,
                                     SegmentationDocument* segDoc,
                                     int controlId) :
        Window(parent, wndWidth, wndHeight, true),
        AppControlDoc(controlId, segDoc)
    {
        Init(parent,nrImOnRow,thumbWidth,thumbHeight,viewScale,segDoc);
    }

    SegmentationDocumentGuiKeyframes(OglGui::Window* parent, int x, int y,
                                     int wndWidth, int wndHeight, int nrImOnRow,
                                     int thumbWidth, int thumbHeight,
                                     double viewScale,
                                     SegmentationDocument* segDoc,
                                     int controlId) :
        Window(parent, x, y, wndWidth, wndHeight, true),
        AppControlDoc(controlId, segDoc)
    {
        Init(parent,nrImOnRow,thumbWidth,thumbHeight,viewScale,segDoc);
    }

    virtual void
    HandleNewFile()
    {
        Core::VideoSet::Keyframes* keys = mSegDoc->GetKeyframes();
        Core::ImageSet::ImageSet* imSet = mSegDoc->GetImSetKeyframes(true);
        if (!(mSegDoc->HasCursor() && keys && imSet))
        {
            mCurFile = -1;
            if (mImagesSet)
                mImagesSet->RemoveImages();
            if (mImSetIdxGrid)
                mImSetIdxGrid->Clear();
            return;
        }
        int fileId = mSegDoc->CurFileId();
        if (fileId == mCurFile)
            return;
        mCurFile = fileId;
        if (mImagesSet)
        {
            mImagesSet->RemoveImages();
            mImagesSet->SetFirstVisY(0, true);
        }
        if (mImSetIdxGrid)
            mImSetIdxGrid->Clear();
        mStartKey = keys->GetFirstKeyframeVideo(fileId);
        int nrKey = keys->GetNrKeyframesVideo(fileId);
        int end = mStartKey + (mImagesSet ? Min(nrKey, 3000) : nrKey);
        for (int i=mStartKey ; i<end ; i++)
        {
            if (mImagesSet)
                mImagesSet->AddImage(i, imSet);
            if (mImSetIdxGrid)
                mImSetIdxGrid->AddImageSetIdx(imSet,i);
        }
        if (mImSetIdxGrid)
            mImSetIdxGrid->UpdateLayout(true);
    }

    virtual void
    ImageSelectionEvent(ImagesWindow* src, int imIndex, void* listenerData)
    {
        mSegDoc->GotoKeyfr(mStartKey + imIndex);
        AllHandleNewCursor(Core::Database::LEVEL_SHOT, false);
    }

    virtual void
    ImageSelectionEvent(ImageSetIdxGrid* src, int imIndex, void* listenerData)
    {
        mSegDoc->GotoKeyfr(mStartKey + imIndex);
        AllHandleNewCursor(Core::Database::LEVEL_SHOT, false);
    }

private:

    void Init(OglGui::Window* parent, int nrImOnRow, int thumbWidth, int thumbHeight,
              double viewScale, SegmentationDocument* segDoc)
    {
        mSegDoc = segDoc;
        mCurFile = -1;
        mImagesSet = 0;
        mImSetIdxGrid = 0;
        AddDocListener(this);

        CmdOptions& options = CmdOptions::GetInstance();

        if (!options.GetBool("keyframesIdxGrid",false))
        {
            int nrRow = H() / (thumbHeight + 3);
            mImagesSet = new ImageSet(this,0,0,thumbWidth, thumbHeight, viewScale,
                                      nrImOnRow, nrRow);
            mImagesSet->SetDimensions(2,2,parent->W()-4,parent->H()-4);
            mImagesSet->ActivateInfoBox(false);
            mImagesSet->SetImagesListener(this, 0);
        }
        else
        {
            mImSetIdxGridScroller =
                new ImageSetIdxGridScroller(this,2,2,parent->W()-4,
                                            parent->H()-4,nrImOnRow);
            mImSetIdxGrid = mImSetIdxGridScroller->ImSetIdxGrid();
            mImSetIdxGrid->SetImageSetIdxGridListener(this);
            //mImSetIdxGrid->ShowNames(false);
            mImSetIdxGrid->ShowDims(false);
        }
    }

    SegmentationDocument*       mSegDoc;
    int                         mCurFile;
    int                         mStartKey;
    ImageSet*                   mImagesSet;
    ImageSetIdxGridScroller*    mImSetIdxGridScroller;
    ImageSetIdxGrid*            mImSetIdxGrid;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(SegmentationDocumentGuiKeyframes, Visualization);

} // namespace Visualization
} // namespace Impala

#endif

---