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

---

TrackVisualiser.h

Go to the documentation of this file.

#ifndef TrackVisualiser_h
#define TrackVisualiser_h

#include "OglGui/Window.h"
#include "Visualization/ViewListenerWithRectangles.h"
#include "Visualization/Plot/Plot.h"
#include "Visualization/Plot/Line.h"
#include "Visualization/Plot/TimeLine.h"
#include "Visualization/Window.h"
#include "Domain/FrameRegionTrack.h"
#include "Core/Geometry/Rectangle.h"
#include "Core/Array/Sort.h"

namespace Impala
{


class TrackVisualiser
{
public:
    TrackVisualiser(OglGui::Window* parent, 
                    Domain::ImageRegionTrack* track,
                    Domain::DataArray* similarity,
                    Domain::ImageSequence* video)
    {
        using Visualization::Plot::Plot;
        using Visualization::Plot::Line;
        using Visualization::Plot::TimeLine;

        mTrack = track;
        mSimilarity = similarity;
        mVideoViewer = new Visualization::Window(parent, video->GetWidth(), video->GetHeight());
        mVideoViewer->SetBackground(0xffff0000);
        mVideoPlayer = new Domain::ImageSequenceReader(video);
        mFeatureViewer = new Visualization::Window(parent, 23, 23);
        mSortedFeature = 0;

        mPlot1 = new Plot(parent, 300, 300);
        mPlot1->SetPerspective(false);
        mPlot1->SetDrawAxes(true);
        mPlotDistribution = new Line(0);
        mPlotDistribution->SetColor(1, 1, 0);
        mPlot1->Add(mPlotDistribution);

        mPlot2 = new Plot(parent, 700, 300);
        mPlot2->SetPerspective(false);
        mPlot2->SetDrawAxes(true);
        mPlotHighest = new TimeLine(0, 200);
        mPlotHighest->SetColor(0, 1, 0);
        mPlot2->Add(mPlotHighest);
    }

    void operator()()
    {
        using Core::Array::Array2dScalarReal64;

        Visualization::ViewListenerWithRectangles* rectDrawer = new Visualization::ViewListenerWithRectangles;
        for(int i=0 ; i<mTrack->size() && mVideoPlayer->HasAvailable() ; i++)
        {
            if(i<mTrack->size())
            {
                Core::Geometry::Rectangle& rect = mTrack->at(i)->GetRectangle();
                rectDrawer->SetRectangle(rect.mLeft, rect.mTop, rect.mRight, rect.mBottom);
            }
            Domain::Image* im = mVideoPlayer->GetImage();
            Array2dVec3UInt8* a = im->GetData()->mDataVec3UInt8;
            mVideoViewer->UpdateView(0, a, rectDrawer);

            if(i<mTrack->size() && i>0)
            {
                Array2dScalarReal64* feature = reinterpret_cast<Array2dScalarReal64*>(mSimilarity->at(i-1));
                mFeatureViewer->UpdateView(0, feature, "Stretch");
                Core::Array::Sort(mSortedFeature, feature);

                mPlotDistribution->SetData(mSortedFeature->CPB(), mSortedFeature->W() * mSortedFeature->H());
                mPlot1->AutoScale();
                mPlotHighest->Add(*mSortedFeature->CPE());
                mPlot2->AutoScale();
            }
            else
            {
                //this is the first frame: no data yet
            }
            if(OglGui::Sys::Instance().WindowManage())
                break;
            Sleep(50);
        }
    }


private:
    Visualization::Window* mVideoViewer;
    Visualization::Window* mFeatureViewer;
    Visualization::Plot::Plot* mPlot1;
    Visualization::Plot::Plot* mPlot2;
    Visualization::Plot::TimeLine* mPlotHighest;
    Visualization::Plot::Line* mPlotDistribution;
    Core::Array::Array2dScalarReal64* mSortedFeature;
    Domain::ImageRegionTrack* mTrack;
    Domain::DataArray* mSimilarity;
    Domain::ImageSequenceReader* mVideoPlayer;
};

}//namespace Impala

#endif TrackVisualiser_h

---