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

---

IndexSegmentation.h

Go to the documentation of this file.

#ifndef Impala_Core_VideoSet_IndexSegmentation_h
#define Impala_Core_VideoSet_IndexSegmentation_h

#include "Persistency/SegmentationRepository.h"
#include "Persistency/KeyframesRepository.h"
#include "Persistency/Mpeg7DocRepository.h"
#include "Core/VideoSet/Reporter.h"
#include "Core/VideoSet/Segmentation.h"
#include "Core/VideoSet/Keyframes.h"
#include "Core/VideoSet/Mpeg7Doc.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class IndexSegmentation : public Listener
{
public:

    IndexSegmentation(Reporter* reporter, CmdOptions& options)
    {
        mReporter = reporter;
    }

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

    virtual void
    HandleNewFile(VideoSet* vs, int fileId, Stream::RgbDataSrc* src)
    {
        if ((!src) || (src->FrameRateNum() < 0) || (src->FrameRateDen() < 0))
        {
            ILOG_ERROR("Need src with known frame rate");
            return;
        }

        String container = vs->GetContainerFile(fileId);
        String name = vs->GetFileBase(fileId) + ".xml";
        const int MPEG7_SHOT = Persistency::Mpeg7DocLocator::MPEG7_SHOT;
        Persistency::Mpeg7DocLocator loc(vs->GetLocator(), MPEG7_SHOT,
                                         container, name);
        loc.SetFrameRateNum(src->FrameRateNum());
        loc.SetFrameRateDen(src->FrameRateDen());
        Mpeg7Doc* mp7 = Persistency::Mpeg7DocRepository().Get(loc);
        for (int s=0 ; s<mp7->NrShots() ; s++)
        {
            mSegmentation->Add(fileId, mp7->StartFrame(s), mp7->EndFrame(s),
                               mp7->ShotName(s));
            int shotId = mSegmentation->Size() - 1;
            for (int k=0 ; k<mp7->NrKeyframes(s) ; k++)
            {
                String name = mp7->KeyframeName(s, k) + ".jpg";
                mKeyframes->Add(fileId, shotId, mp7->KeyframeFrame(s, k), name);
            }
        }
        delete mp7;
    }

    virtual void
    HandleDoneWalk(VideoSet* vs)
    {
        bool binary = true;
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        mSegmentation->Save("segmentation", binary);
        mKeyframes->Save("keyframes", binary);
#else // REPOSITORY_USED
        Persistency::SegmentationLocator sLoc(vs->GetLocator(), "segmentation");
        Persistency::SegmentationRepository().Add(sLoc, mSegmentation);
        Persistency::KeyframesLocator kLoc(vs->GetLocator(), "keyframes");
        Persistency::KeyframesRepository().Add(kLoc, mKeyframes);
#endif // REPOSITORY_USED
        mKeyframes->WriteImageSets();
    }

private:

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

    ILOG_VAR_DEC;

};

ILOG_VAR_INIT(IndexSegmentation, Impala.Core.VideoSet);

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

#endif

---