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

---

ReadBlocksFile.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_ReadBlocksFile_h
#define Impala_Core_VideoSet_ReadBlocksFile_h

#include "Core/VideoSet/Reporter.h"
#include "Core/VideoSet/Segmentation.h"
#include "Core/VideoSet/Keyframes.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class ReadBlocksFile : public Listener
{
public:

    ReadBlocksFile(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        mCurShot = 0;
    }

    virtual void
    HandleNewWalk(VideoSet* vs)
    {
        mSegmentation = new Segmentation(vs, "");
        mKeyframes = new Keyframes(vs, "");
    }

    virtual void
    HandleNewFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        // todo : this doesn't work anymore, but is also not needed anymore
        //std::string fName = vs->GetFilePathId("VideoMetaData", "", fileId, true,
        //                                      ".blocks", false);
        std::string fName("dummy");
        ILOG_DEBUG("Checking " << fName);
        File shots(fName, "r");
        if (!shots.Valid())
        {
            ILOG_ERROR("couldn't find " << fName);
            return;
        }
        int i = 1;
        while (! shots.Eof())
        {
            // start with shots file
            std::string line = shots.ReadLine(true);
            if (! line[0])
                continue;
            Util::StringParser p(line);
            int startFrame = p.GetInt();
            int endFrame = p.GetInt();
            std::string name = p.GetString();
            mSegmentation->Add(fileId, startFrame, endFrame, name);
            mKeyframes->Add(fileId, mCurShot++, (startFrame+endFrame)/2,
                            name + "_RKF.jpg");
            i++;
        }
    }

    virtual void
    HandleDoneWalk(VideoSet* vs)
    {
        bool binary = true;
        mSegmentation->Save("segmentation", binary);
        mKeyframes->Save("keyframes", binary);
        mKeyframes->WriteImageSets();
    }

private:

    Reporter*     mReporter;
    int           mCurShot;
    Segmentation* mSegmentation;
    Keyframes*    mKeyframes;

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(ReadBlocksFile, Core.VideoSet);

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

#endif

---