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

---

MapFunction.h

Go to the documentation of this file.

#ifndef Impala_Visualization_MapFunction_h
#define Impala_Visualization_MapFunction_h

#include "Basis/ILog.h"
#include "Core/Array/Arrays.h"
#include "Core/VideoSet/Segmentation.h"

namespace Impala {
namespace Visualization {

class MapFunction
{
public:
    MapFunction(int width, int height)
    {
        mWidth = width;
        mHeight = height;
    }

    virtual void ShotToXY(int shot, int* x, int* y) = 0;
    virtual int  XYToShot(int x, int y) = 0;
    virtual int  GetNrShots() = 0;

    int GetWidth()
    {
        return mWidth;
    }

    int GetHeight()
    {
        return mHeight;
    }

protected:
    int mWidth, mHeight;
};


class VideoSquareMapFunction : public MapFunction
{
public:
    typedef Core::VideoSet::Segmentation            Segmentation;

    VideoSquareMapFunction(Segmentation *segmentation, int width, int height) :
        MapFunction(width, height)
    {
        mSegmentation = segmentation;
    }

    virtual void ShotToXY(int shot, int* x, int* y)
    {
        int videos = mSegmentation->GetNrVideos();
        int video = mSegmentation->GetVideoId(shot);
        int shotsinvideo = mSegmentation->GetNrShotsVideo(video);
        int videoshot = shot - mSegmentation->GetFirstShotVideo(video);
        *x = videoshot * mWidth / shotsinvideo;
        *y = video * mHeight / videos;
    }

    virtual int XYToShot(int x, int y)
    {
        int videos = mSegmentation->GetNrVideos();
        int video = y * videos / mHeight;
        if (video < 0) video = 0;
        if (video > videos-1) video = videos-1;
        
        int shotsinvideo = mSegmentation->GetNrShotsVideo(video);
        
        int videoshot = x * shotsinvideo / mWidth;
        if (videoshot < 0) videoshot = 0;
        if (videoshot > shotsinvideo-1) videoshot = shotsinvideo-1;

        return videoshot + mSegmentation->GetFirstShotVideo(video);
    }

    virtual int GetNrShots()
    {
        return mSegmentation->GetNrShots();
    }

private:
    Segmentation *mSegmentation;
};

} // namespace Visialization
} // namespace Impala
#endif

---