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

---

SegmentationDocumentGuiStills.h

Go to the documentation of this file.

#ifndef Impala_Visualization_SegmentationDocumentGuiStills_h
#define Impala_Visualization_SegmentationDocumentGuiStills_h

#include "Persistency/StillsRepository.h"
#include "Core/Array/Scale.h"
#include "Core/Array/Set.h"
//#include "Core/Array/ArrayListDelete.h"
#include "Core/VideoSet/SegmentationDocument.h"
#include "Core/VideoSet/Stills.h"
#include "Core/Database/DataDocumentListener.h"
#include "Visualization/ImageSet.h"
#include "Visualization/AppControlDoc.h"
#include "OglGui/CheckBox.h"
#include "OglGui/TextField.h"
#include "OglGui/TextArea.h"
#include "OglGui/Strut.h"
#include "Basis/ILog.h"

namespace Impala {
namespace Visualization {

class SegmentationDocumentGuiStills :
        public Window,
        public AppControlDoc,
        public Core::Database::DataDocumentListener,
        public OglGui::ButtonListener,
        public ImagesListener
{
public:
    typedef Core::Array::Array2dVec3UInt8           Array2dVec3UInt8;
    typedef Core::VideoSet::SegmentationDocument    SegmentationDocument;
    typedef Core::VideoSet::Stills                  Stills;

    SegmentationDocumentGuiStills(OglGui::Window* parent, int w, int h,
                                  int nrImOnRow, int thumbW, int thumbH,
                                  double viewScale, bool doFullSize,
                                  bool imageStills,
                                  SegmentationDocument* segDoc,
                                  int controlId) :
        Window(parent, w, h, true),
        AppControlDoc(controlId, segDoc)
    {
        mSegDoc = segDoc;
        AddDocListener(this);
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        mStills = new Stills(segDoc->GetVideoSet(), "stills");
#else // REPOSITORY_USED
        Persistency::StillsLocator loc(segDoc->GetVideoSet()->GetLocator(),
                                       "stills");
        mStills = Persistency::StillsRepository().Get(loc, segDoc->GetVideoSet());
#endif // REPOSITORY_USED
        mViewScale = viewScale;
        mDoFullSize = doFullSize;
        mImageStills = imageStills;
        mForceUpdate = false;

        int line = 20;
        mCheckBox = new OglGui::CheckBox(this, 150, line, "Update stills");
        mCheckBox->SetSelected(false);
        mFrameSkipField = 0;
        if (!mImageStills)
        {
            OglGui::TextArea* tArea = new OglGui::TextArea(this, 80, 1*line,
                                                           "Frame skip");
            tArea->SetBorderType(0);
            mFrameSkipField = new OglGui::TextField(this, 60, 20, "15");
        }
        OglGui::Button* but = new OglGui::Button(this, 80, 22, "Update (u)");
        but->SetButtonListener(this, 0);
        OglGui::Strut* strut = new OglGui::Strut(this, 4000, 1);
        int imWidth = (mDoFullSize) ? thumbW/mSegDoc->GetThumbScale() : thumbW;
        int imHeight = (mDoFullSize) ? thumbH/mSegDoc->GetThumbScale(): thumbH;
        if (mDoFullSize)
        {
            nrImOnRow = 4;
            mViewScale = 0.85 * mViewScale;
        }
        int nrRow = (mDoFullSize) ? 4 : 7;
        mNrIm = nrImOnRow * nrRow;
        mImagesSet = new ImageSet(this, imWidth, imHeight, mViewScale,
                                  nrImOnRow, nrRow);
        mImagesSet->SetMakeViewWithRect(true);
        mImagesSet->ActivateInfoBox(false);
        mImagesSet->SetImagesListener(this, 0);
        SetAsBox(mImagesSet);
    }

    virtual void HandleNewFile()
    {
       // ILOG_DEBUG("HandleNewFile()");
        // RvB: Experimental only remove images when UpdateGui checkB or forced
        if ((mCheckBox->GetSelected() || mForceUpdate))
                mImagesSet->RemoveImages();
        if (mSegDoc->HasCurShot() && (mCheckBox->GetSelected() || mForceUpdate))
        {
            bool done = false;
            if (mImageStills)
                done = UpdateFromJpg();
            if (!done)
                UpdateFromMpg();
            Core::Geometry::Rectangle r = mSegDoc->GetRect();
            for (int i=0 ; i<mImagesSet->NrViewers() ; i++)
                mImagesSet->GetViewWithRect(i)->SetRect(r);
        }
        mForceUpdate = false;
        UpdateScene();
    }

    virtual void HandleNewKey(int c, int state)
    {
//        ILOG_DEBUG("HandleNewKey: " << c);
        if (c == 'u')
        {
            mForceUpdate = true;
            HandleNewFile();
        }
    }

    virtual void ButtonSelectionEvent(OglGui::Button* src, void* listenerData)
    {
        mForceUpdate = true;
        HandleNewFile();
    }

    // RvB: This OnViewMouse is never reached. So why is it here?
    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();
        if (mDoFullSize)
            mSegDoc->SetRect(r, mViewScale);
        else
            mSegDoc->SetRectThumb(r);
    }

private:

    void UpdateFromMpg()
    {
        int shot = mSegDoc->CurShot();
        int vidId = mSegDoc->CurFileId();
        Core::VideoSet::Segmentation* shots = mSegDoc->GetSegmentation();
        int startF = shots->GetStart(shot) + 1;
        int endF = shots->GetEnd(shot);
        Core::Stream::RgbDataSrc* src = mSegDoc->GetVideoSet()->GetVideo(vidId);
        if (!src)
            return;
        int skipFr = 15;
        if (mFrameSkipField)
            skipFr = atol(mFrameSkipField->GetText());
        int interval = (endF - startF) / (mNrIm - 1);
        if (interval < skipFr)
            interval = skipFr;
#ifdef unix
#define NO_PLAYER
#endif
#ifdef NO_PLAYER
/* Ork: WARNING: RgbDataSrcDX not yet ready for unixes */
/* TODO: verify and fix this. */
#else
        src->SetInterval(interval);
        while (startF < endF)
        {
            src->GotoFrame(startF);
            Array2dVec3UInt8* im = Core::Array::ArrayCreate<Array2dVec3UInt8>
                (src->FrameWidth(), src->FrameHeight(), 0, 0,
                 src->DataPtr(), true);
            Array2dVec3UInt8* sc = 0;
            if (mDoFullSize)
                Core::Array::Set(sc, im);
            else
                Core::Array::Scale(sc, im, mSegDoc->GetThumbScale(),
                                   mSegDoc->GetThumbScale(),
                                   Core::Geometry::NEAREST, true);
            std::string cap = std::string("frame ") + MakeString(startF);
            mImagesSet->AddImage(sc, std::string("Direct"), cap, true);
            delete im;
            startF += interval;
        }
#endif
    }

    bool UpdateFromJpg()
    {
        Core::ImageSet::ImageSet* imSetStills = mSegDoc->GetImSetStills();
        if (!imSetStills)
            return false;
        int shot = mSegDoc->CurShot();
        int start = mStills->GetFirstStillShot(shot);
        int end = start + mStills->GetNrStillsShot(shot);
        for (int i=start ; i<end ; i++)
        {
            Array2dVec3UInt8* im = imSetStills->GetImage(i);
            Array2dVec3UInt8* sc = 0;
            if (mDoFullSize)
                Core::Array::Set(sc, im);
            else
                Core::Array::Scale(sc, im, mSegDoc->GetThumbScale(),
                                   mSegDoc->GetThumbScale(),
                                   Core::Geometry::NEAREST, true);
            mImagesSet->AddImage(sc, std::string("Direct"), "", true);
            delete im;
        }
        return true;
    }

    void SetAsBox(OglGui::Window *w)
    {
        //w->SetRoundness(2,10,10,2);
        w->SetBorderType(4);
        w->SetBorderFillShaded(2);
    }

    bool                    mDoFullSize;
    bool                    mImageStills;
    bool                    mForceUpdate;

    double                  mViewScale;

    int                     mNrIm;

    SegmentationDocument*   mSegDoc;
    Stills*                 mStills;
    
    OglGui::CheckBox*       mCheckBox;
    OglGui::TextField*      mFrameSkipField;
    ImageSet*               mImagesSet;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(SegmentationDocumentGuiStills, Visualization);

} // namespace Visualization
} // namespace Impala

#endif

---