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

---

FrameRegionTable.h

Go to the documentation of this file.

#ifndef Impala_Core_Table_FrameRegionTable_h
#define Impala_Core_Table_FrameRegionTable_h

#include "Util/DatabaseReadString.h"
#include "Core/Table/TableTem.h"
#include "Core/Table/Read.h"

namespace Impala
{
namespace Core
{
namespace Tracking
{

class FrameRegion
{
public:
    FrameRegion(int videoId, int frame, int left, int top, int right, int bottom) :
        mRect(left, top, right, bottom)
    {
        mVideoId = videoId;
        mFrame = frame;
    }

    int GetFrameNumber()
    {
        return mFrame;
    }

    Geometry::Rectangle GetRectangle()
    {
        return mRect;
    }

    int mVideoId;
    int mFrame;
    Geometry::Rectangle mRect;
};


typedef int VideoId;

typedef Table::TableTem<Column::ColumnTem<VideoId>,
                        Column::ColumnTem<int>,
                        Column::ColumnTem<int>,
                        Column::ColumnTem<int>,
                        Column::ColumnTem<int>,
                        Column::ColumnTem<int> >
        FrameRegionTableBaseType;


class FrameRegionTable : public FrameRegionTableBaseType
{
public:

    typedef Column::ColumnTem<int> ColumnInt;

    FrameRegionTable() :
        FrameRegionTableBaseType(ColumnInt(0), ColumnInt(0), ColumnInt(0),
                                 ColumnInt(0), ColumnInt(0), ColumnInt(0))
    {
        SetColName(1, "videoid");
        SetColName(2, "frame number");
        SetColName(3, "x");
        SetColName(4, "y");
        SetColName(5, "w");
        SetColName(6, "h");

    }

    FrameRegion GetRegion(int i)
    {
        return FrameRegion(Get1(i), Get2(i), Get3(i), Get4(i), Get5(i), Get6(i));
    }

    void Add(Geometry::Rectangle r)
    {
        // assume video id and framenum are same as last entry.
        int vid=0, frame=0;
        if(mLast != 0)
        {
            vid = Get1(mLast - 1);
            frame = Get2(mLast - 1) + 1;
        }
        FrameRegionTableBaseType::Add(vid, frame, r.mLeft, r.mTop,
                                      r.mRight, r.mBottom);
    }

    void Add(FrameRegion fr)
    {
        FrameRegionTableBaseType::Add(fr.mVideoId, fr.mFrame, fr.mRect.mLeft,
                                      fr.mRect.mTop, fr.mRect.mRight, fr.mRect.mBottom);
    }

    void WriteTo(std::ostream& os)
    {
        // does not write vid id, this is the file format for PhD student use,
        // not in a 'system'
        int i;
        for(i=0 ; i<mLast ; ++i)
        {
            os << Get2(i) << " "
               << Get3(i) << " " << Get4(i) << " "
               << Get5(i) << " " << Get6(i) << "\n";
        }
    }

    void ReadFrom(std::istream& is)
    {
        // does not read vid id, this is the file format for PhD student use,
        // not in a 'system'
        while(!is.eof())
        {
            int frame = -1;
            int left, top, right, bottom;
            is >> frame >> left >> top >> right >> bottom;
            if(frame != -1)
            FrameRegionTableBaseType::Add(0, frame, left, top, right, bottom);
        }
    }
};

} // namespace Tracking
} // namespace Core
} // namespace Impala

#endif

---