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

---

MakeQuidLookup.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_MakeQuidLookup_h
#define Impala_Core_VideoSet_MakeQuidLookup_h

#include "Core/VideoSet/Reporter.h"
#include "Core/Table/Write.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{

class MakeQuidLookup : public Listener
{
    typedef Column::ColumnTem<Quid> ColumnQuid;
    typedef Column::ColumnTem<std::string> ColumnString;
    typedef Table::TableTem<ColumnQuid,
                            ColumnString,
                            ColumnString,
                            ColumnQuid >
            QuidLookupTableBaseType;

    class QuidLookupTable : public QuidLookupTableBaseType
    {
    public:
        QuidLookupTable() :
            QuidLookupTableBaseType(ColumnQuid(0), ColumnString(0), 
                                    ColumnString(0), ColumnQuid(0))
        {
            SetInfo("QuidLookupTable");
            SetColName(1, "quid");
            SetColName(2, "string");
            SetColName(3, "friendlystring");
            SetColName(4, "shotquid");
        }
    };

public:

    MakeQuidLookup(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        mTable = 0;
        mKeyframes = 0;
        mSegmentation = 0;
    }

    virtual
    ~MakeQuidLookup()
    {
    }

    virtual void
    HandleNewWalk(VideoSet* vs)
    {
        // I don't know what the right way is to construct the Keyframes instance...
        mTable = new QuidLookupTable();
        mKeyframes = new Keyframes(vs, "keyframes");
        mSegmentation = new Segmentation(vs, "segmentation");
    }

    virtual void
    HandleNewFrame(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        String videoName = src->GetName();
        String::size_type p = videoName.rfind("/");
        if (p != std::string::npos)
            videoName = videoName.substr(p+1, videoName.size());

        String imName = FileNameCtor(videoName, "frame", src->FrameNr(), "");
        imName = StringReplaceAll(imName, ".", "_");
        
        Quid quid = vs->GetQuidFrame(fileId, src->FrameNr());

        String friendlyName = "";
        if(mKeyframes)
        {
            int shotId = mKeyframes->GetFrameId(quid);
            if(shotId != -1)
            {
                friendlyName = mKeyframes->GetName(shotId);
            }
        }

        Quid shotQuid = 0;
        if(mSegmentation)
        {
            shotQuid = mSegmentation->GetQuidShot(fileId, src->FrameNr());
        }

        mTable->Add(quid, imName, friendlyName, shotQuid);
    }

    virtual void
    HandleDoneWalk(VideoSet* vs)
    {
        String filename = vs->GetDatabase()->GetFilePath(".", "quidlookup.tab", true, false);
        Write(mTable, filename, vs->GetDatabase(), true);
        delete mTable;
    }

private:

    Reporter*        mReporter;
    QuidLookupTable* mTable;
    Keyframes*       mKeyframes;
    Segmentation*    mSegmentation;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(MakeQuidLookup, Core.VideoSet);

} // namespace ImageSet
} // namespace Core
} // namespace Impala

#endif

---