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

---

KeyframeResultBar.h

Go to the documentation of this file.

#ifndef Impala_Visualization_GUI_KeyframeResultBar_h
#define Impala_Visualization_GUI_KeyframeResultBar_h

#include "OglGui/Window.h"
#include "Visualization/ImageSet.h"
#include "Core/Trec/KeyframeResult.h"
#include "Core/ImageSet/ImageSet.h"

namespace Impala {
namespace Visualization {
namespace GUI {

using namespace OglGui;

class KeyframeResultBarListener
{
public:
    virtual void
    KeyframeSelectedEvent(String name, int keyframe)
    {
    }
};


class KeyframeResultBar :   public OglGui::Window,
                            public Visualization::ImagesListener
{
public:
    typedef std::list<Core::Trec::KeyframeResult>   KeyframeResultList;
    typedef Core::VideoSet::Keyframes               Keyframes;

    KeyframeResultBar(OglGui::Window *parent, int w, int h, String name,
                      KeyframeResultList results, Keyframes* keyframes,
                      Core::ImageSet::ImageSet* imSet, int thumbW, int thumbH,
                      double viewscale) :
        OglGui::Window(parent, w, h)
    {
        Init(w, h, name, results, keyframes, imSet, thumbW, thumbH, viewscale);
    }

    virtual void ImageSelectionEvent(Visualization::ImagesWindow *src,
                                     int imIndex, void* listenerData)
    {
        ILOG_DEBUG("ImSelEvent: selected rank=" << imIndex << " keyframe=" <<
                   GetImageKeyframeID(imIndex));
        if (!mBarListener) return;
        mBarListener->KeyframeSelectedEvent(mName, GetImageKeyframeID(imIndex));
    }

    void SetListener(KeyframeResultBarListener *listener)
    {
        mBarListener = listener;
    }

    String NiceName()
    {
        if (mName == "visual_vissem")
            return "Weibull";
        if (mName == "visual_vissemgabor")
            return "Gabor";
        if (mName == "visual_fusionvissemgabor")
            return "Weibull/Gabor fusion";
        if (mName == "visual_labhistogram")
            return "LAB color histogram";
        return mName;
    }

protected:
    // add a simple plot of similarity scores for first n ranked similars
    void DoRankPlot(int wndW, int wndH)
    {
        int plotW   = mKeyResults.size() * 2;
        int plotH   = 16;

        OglGui::Window* simPlotWin = new Window(this,wndW-plotW-4, wndH-plotH-3,
                                                plotW, plotH, true);
        simPlotWin->SetBackground(oglLIGHTGREEN);
        simPlotWin->ConnectTo(this, R2R|T2T|B2T);

        OGLWND *oglWnd = simPlotWin->GetOGLWND();
        mSimPlotView = new Visualization::View(oglWnd, 0, 0, plotW, plotH, 1.0,
                                               false, false, true, NULL);
        KeyframeResultList::reverse_iterator lastResult = mKeyResults.rbegin();
        // RvB: Next line caused crash
        // lastResult++; // SK: Shouldn't need this, but MSVC++ appears to do...
        double maxScore = lastResult->score;

        static Core::Array::Element::Vec3Int32 colorCanvas(40, 40, 40);
        static Core::Array::Element::Vec3Int32 colorPlot(255, 255, 0);

        Core::Array::Array2dVec3UInt8* im =
            new Core::Array::Array2dVec3UInt8(plotW, plotH, 0, 0);

        int xCoord = 0;
        KeyframeResultList::iterator res = mKeyResults.begin();
        while (res != mKeyResults.end())
        {
            int valueToPlot = (int)((res->score / maxScore) * (plotH-1.0)+0.5);
            for (int yCoord = 0; yCoord < plotH; yCoord++)
            {
                Core::Array::Element::Vec3Int32& pixel =
                    (plotH-1-yCoord == valueToPlot ? colorPlot : colorCanvas);
                im->SetValue(pixel, xCoord, yCoord);
                im->SetValue(pixel, xCoord + 1, yCoord);
            }
            xCoord += 2;
            res++;
        }
        mSimPlotView->UpdateImage(im, "Direct");
        // RvB: 2-2-2009 I think this needs deleting
        delete im;
    }

    void Init(int wndW, int wndH, String name, KeyframeResultList results,
              Keyframes *keyframes, Core::ImageSet::ImageSet* imSet,
              int thumbW, int thumbH, double viewscale)
    {
        mName           = name;
        mKeyResults     = results;
        mKeyframes      = keyframes;
        mBarListener    = NULL;
        mPreviewImages  = NULL;
        mSimPlotView    = NULL;

        SetRoundness(4,4,4,4);
        SetBorderType(4);
        SetBorderFillShaded(2);

        StaticText *t = new StaticText(this,2,wndH-17,wndW-100,17,NiceName());
        t->SetAlign(oglLeftAlign, oglCenterAlign);
        t->ConnectTo(this, L2L|T2T|B2T);

        // images:
        if (results.size() == 0)
            return;

        int nrImOnRow   = (int)(wndW      / ((double)thumbW * viewscale));
        int nrRows      = (int)((wndH-18) / ((double)thumbH * viewscale));
        
        if (nrRows == 0) nrRows = 1;

        Window *wnd = new Window(this, 3, 2, wndW-6, wndH-19);

        mPreviewImages = new Visualization::ImageSet(wnd, 1, 1, thumbW, thumbH,
                                                    viewscale,nrImOnRow,nrRows);
        mPreviewImages->ActivateInfoBox(false);
        mPreviewImages->SetImagesListener(this, 0);
        // RvB: Added because views inside seem to be corrupt
        // and cause crash when doing standard viewSys.view2d stuff
        mPreviewImages->SetDisableOGLViewKeys( true );
        mPreviewImages->SetDisableOGLViewMouse( true );

        int i=0;
        KeyframeResultList::iterator shot;
        for (shot = results.begin(); shot != results.end(); shot++)
        {
            int frameID = shot->keyframeid;
            if (frameID >= 0)
                mPreviewImages->AddImage(imSet->GetImage(frameID), "Direct",
                                         "", true);
            if (++i >= nrImOnRow * nrRows)
                break;
        }
        DoRankPlot(wndW, wndH);
    }

    virtual int GetImageKeyframeID(int rank)
    {
        int foundrank = 0;
        KeyframeResultList::iterator shot;
        for (shot = mKeyResults.begin(); shot != mKeyResults.end(); shot++)
        {
            int frameID = shot->keyframeid;
            if (frameID >= 0)
            {
                if (rank == foundrank)
                    return frameID;
                foundrank ++;
            }
        }
        return -1;
    }

    String                      mName;
    Visualization::ImageSet*    mPreviewImages;
    KeyframeResultList          mKeyResults;
    Keyframes*                  mKeyframes;
    KeyframeResultBarListener*  mBarListener;
    Visualization::View*        mSimPlotView;
    
ILOG_VAR_DEC;
};

ILOG_VAR_INIT(KeyframeResultBar, Impala.Visualization.GUI);

} // namespace GUI
} // namespace Visualization
} // namespace Impala
#endif

---