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

---

TableShots.h

Go to the documentation of this file.

#ifndef Impala_Core_Table_TableShots_h
#define Impala_Core_Table_TableShots_h

#include "Basis/NativeTypes.h"
#include "Core/Table/TableTem.h"
#include "Core/Geometry/FileReadVxRectangle.h"
#include "Core/Geometry/FileWriteVxRectangle.h"
#include "Core/VideoSet/Segmentation.h"
#include "Core/VideoSet/Keyframes.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


typedef Table::TableTem<Column::ColumnTem<Int32>,
                        Column::ColumnTem<Int32>,
                        Column::ColumnTem<std::string>,
                        Column::ColumnTem<Geometry::Rectangle> > TableShotsBaseType;

class TableShots : public TableShotsBaseType
{
public:

    TableShots() : TableShotsBaseType(0)
    {
        SetColName(1, "shotId");
        SetColName(2, "keyframeId");
        SetColName(3, "anno");
        SetColName(4, "rect");
    }

    void
    ReadVxsFile(std::string fileName, Keyframes* keyframes)
    {
        std::vector<Geometry::VxRectangle> vrList;
        Geometry::FileReadVxRectangle(std::back_inserter(vrList), fileName);
        for (int i=0 ; i<vrList.size() ; i++)
        {
            Geometry::VxRectangle vr = vrList[i];
            int keyfr = keyframes->GetFrameId(vr.mExtra);
            int shot = keyframes->GetShotId(keyfr);
            Add(shot, keyfr, vr.mAnno, vr.mRect);
        }
    }

    // conceptName is used only when bookmark does not have its own annotation
    void
    WriteVxsFile(std::string fileName, std::string conceptName,
                 Segmentation* shots, Keyframes* keyframes)
    {
        std::vector<Geometry::VxRectangle> vrList;
        for (int i=0 ; i<Size() ; i++)
        {
            int shot = Get1(i);
            Geometry::VxRectangle vr;
            vr.mStartFrame = shots->GetStart(shot);
            vr.mEndFrame = shots->GetEnd(shot);
            vr.mAnno = Get3(i);
            if (vr.mAnno == std::string(""))
                vr.mAnno = conceptName;
            vr.mRect = Get4(i);
            int vid = shots->GetVideoId(shot);
            vr.mVideoName = shots->GetVideoSet()->GetFile(vid);
            vr.mSection = shots->GetVideoSet()->GetSection(vid);
            vr.mDir = shots->GetVideoSet()->GetDir(vid);
            vr.mExtra = keyframes->GetName(Get2(i));
            vrList.push_back(vr);
        }
        Geometry::FileWriteVxRectangle(fileName, vrList.begin(), vrList.end());
    }

};

} // namespace VideoSet
} // namespace Core
} // namespace Impala

#endif

---