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

---

TestBlocksFile.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_TestBlocksFile_h
#define Impala_Core_VideoSet_TestBlocksFile_h

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

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class TestBlocksFile : public Listener
{
public:

    TestBlocksFile(Reporter* reporter)
    {
        mReporter = reporter;
    }

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

    virtual void
    HandleNewFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        String fName = vs->GetFilePathMetaData(fileId, "", true, ".blocks",
                                               false, false);
        ILOG_INFO("Checking " << fName);
        File shots(fName, "r");
        if (!shots.Valid())
        {
            ILOG_ERROR("couldn't find " << fName);
            return;
        }
        int firstShot = mSegmentation->GetFirstShotVideo(fileId);
        int nrShots = mSegmentation->GetNrShotsVideo(fileId);
        //std::cout << "should have " << nrShots << " shots" << std::endl;
        int i = 0;
        while (! shots.Eof())
        {
            // start with shots file
            String line = shots.ReadLine(true);
            if (! line[0])
                continue;
            Util::StringParser p(line);
            if (p.GetInt() != mSegmentation->GetStart(firstShot + i))
            {
                ILOG_ERROR("start didn't match in shot " << i+1);
            }
            if (p.GetInt() != mSegmentation->GetEnd(firstShot + i))
            {
                ILOG_ERROR("end didn't match in shot " << i+1);
            }
            if (p.GetString() != mSegmentation->GetName(firstShot + i))
            {
                ILOG_ERROR("name didn't match in shot " << i+1);
            }
            i++;
        }
        if (i != nrShots)
        {
            ILOG_ERROR("nr shots didn't match in " << fName);
        }
    }

private:

    Reporter*     mReporter;
    Segmentation* mSegmentation;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(TestBlocksFile, Impala.Core.VideoSet);

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

#endif

---