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

---

CheckNrFrames.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_CheckNrFrames_h
#define Impala_Core_VideoSet_CheckNrFrames_h

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

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class CheckNrFrames : public Listener
{
public:

    CheckNrFrames(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
        mSegmentation = 0;
        mMargin = 0;
        if (options.GetNrArg() < 3)
        {
            ILOG_ERROR("Need at least 3 arguments");
        }
        else
        {
            mMargin = atol(options.GetArg(2));
        }
        mSbDir = "";
        if (options.GetNrArg() > 3)
            mSbDir = options.GetArg(3);
    }

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

    virtual void
    HandleNewFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        if (!mSegmentation)
        {
            ILOG_ERROR("No segmentation given");
            return;
        }
        int srcNr = src->LastFrame() + 1;
        int segNr = mSegmentation->GetNrFramesVideo(fileId);
        int d = abs(srcNr - segNr);
        if (d > mMargin)
        {
            ILOG_ERROR("Frame difference is " << d << " for file " <<
                       vs->GetFile(fileId) << " (src says " << srcNr <<
                       ", segmentation says " << segNr << ")");
        }
        if ((d > 0) && (mMargin != 0))
            ILOG_INFO("Frame difference is " << d << " for file " <<
                       vs->GetFile(fileId) << " (src says " << srcNr <<
                       ", segmentation says " << segNr << ")");

        if (mSbDir.empty())
            return;

        String fName = FileNameBase(vs->GetFile(fileId)) + ".sb";
        fName = FileNameConcat(mSbDir, fName);
        Persistency::FileLocator fLoc(vs->GetLocator(), fName);
        typedef Persistency::RepositoryInFileSystem FS;
        Persistency::File file = FS::GetInstance().GetFile(fLoc, false, false);
        std::vector<String> sb;
        file.ReadStrings(sb);
        int nrShots = mSegmentation->GetNrShotsVideo(fileId);
        // First two lines in .sb file are "comments"
        if (nrShots != sb.size() - 2)
        {
            ILOG_ERROR("Different nrShots: segmentation=" << nrShots <<
                       ", sb=" << sb.size()-2);
            return;
        }
        int firstShot = mSegmentation->GetFirstShotVideo(fileId);
        for (int i=0 ; i<nrShots ; i++)
        {
            Util::StringParser p(sb[i+2]);
            int sbStart = p.GetInt(' ');
            int sbEnd = p.GetInt(' ');

            int start = mSegmentation->GetStart(firstShot + i);
            int d = abs(start - sbStart);
            if (d > mMargin)
                ILOG_ERROR("Start differs: segmentation=" << start <<
                           ", sb=" << sbStart);
            if ((d > 0) && (mMargin != 0))
                ILOG_INFO("Start differs: segmentation=" << start <<
                          ", sb=" << sbStart);

            int end = mSegmentation->GetEnd(firstShot + i);
            d = abs(end - sbEnd);
            if (d > mMargin)
                ILOG_ERROR("End differs: segmentation=" << end <<
                           ", sb=" << sbEnd);
            if ((d > 0) && (mMargin != 0))
                ILOG_INFO("End differs: segmentation=" << end <<
                          ", sb=" << sbEnd);
        }
    }

private:

    Reporter*     mReporter;
    Segmentation* mSegmentation;
    int           mMargin;
    String        mSbDir;

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(CheckNrFrames, Impala.Core.VideoSet);

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

#endif

---