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

---

RgbDataSrcKeyframes.h

Go to the documentation of this file.

#ifndef Impala_Core_Stream_RgbDataSrcKeyframes_h
#define Impala_Core_Stream_RgbDataSrcKeyframes_h

#include "Persistency/ImageArchiveRepository.h"
#include "Persistency/FrameRepository.h"
#include "Util/ChannelPool.h"
#include "Core/Stream/RgbDataSrc.h"
#include "Core/VideoSet/Keyframes.h"
#include "Core/Array/ImageArchiveFile.h"

namespace Impala
{
namespace Core
{
namespace VideoSet
{


class RgbDataSrcKeyframes : public Stream::RgbDataSrc
{
public:

    RgbDataSrcKeyframes(VideoSet* videoSet, Keyframes* keyframes, int videoId)
        : RgbDataSrc(-1, videoSet->GetFile(videoId))
    {
        mVideoSet = videoSet;
        mKeyframes = keyframes;
        mVideoId = videoId;
        mUseFileArchive = false;
        mCurIm = 0;
        mArchive = 0;

#ifndef REPOSITORY_USED // Here comes the deprecated stuff
        String fName = mVideoSet->GetFilePathFrames(mVideoId, "images.raw",
                                                    false, true);
        if (!fName.empty())
        {
            mUseFileArchive = true;
            mArchive = new Array::ImageArchiveFile(fName, false,
                                                   mVideoSet->GetDatabase());
        }
#else // REPOSITORY_USED
        String container = mVideoSet->GetContainerFile(mVideoId);
        Persistency::ImageArchiveLocator loc(mVideoSet->GetLocator(), true,
                                             container, "images.raw", 0);
        mArchive = Persistency::ImageArchiveRepository().Get(loc);
        if (mArchive)
            mUseFileArchive = true;
#endif // REPOSITORY_USED

        int firstK = mKeyframes->GetFirstKeyframeVideo(videoId);
        Array::Array2dVec3UInt8* im = ReadImage(mKeyframes->GetFrameNr(firstK));
        if (im == 0)
        {
            ILOG_ERROR("Unable to read image");
            return;
        }
        // this only works in case all images have the same size
        mFrameWidth = im->CW();
        mFrameHeight = im->CH();
        int lastKey = firstK + mKeyframes->GetNrKeyframesVideo(videoId) - 1;
        mLastFrame = mKeyframes->GetFrameNr(lastKey);
        delete im;
    }

    virtual
    ~RgbDataSrcKeyframes()
    {
        if (mCurIm)
            delete mCurIm;
        if (mArchive)
            delete mArchive;
    }

    virtual bool
    Valid() const
    {
        return (mKeyframes != 0) && mArchive->Valid();
    }

    String
    GetKeyframeName()
    {
        int key = mKeyframes->GetFrameIdVideo(mVideoId, mFrameNr);
        return mKeyframes->GetName(key);
    }

protected:

    bool
    ReadFrameData()
    {
        if (mCurIm)
            delete mCurIm;
        mCurIm = ReadImage(FrameNr());
        if (!mCurIm)
        {
            ILOG_ERROR("unable to read frame " << FrameNr());
            return false;
        }
        mData = mCurIm->CPB();
        return true;
    }

    Array::Array2dVec3UInt8*
    ReadImage(int frameNr)
    {
        Array::Array2dVec3UInt8* im = 0;
        if (mUseFileArchive)
        {
            int firstK = mKeyframes->GetFirstKeyframeVideo(mVideoId);
            int key = mKeyframes->GetFrameIdVideo(mVideoId, frameNr);
            im = mArchive->ReadImage(key - firstK);
        }
        else
        {
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
            String fName = mVideoSet->GetFilePathFrames(mVideoId, frameNr,
                                                        ".png", false, false);
            Array::ReadPng(im, fName, mVideoSet->GetDatabase());
#else // REPOSITORY_USED
            Persistency::FrameLocator loc(mVideoSet->GetLocator(),
                                          mVideoSet->GetContainerFile(mVideoId),
                                          frameNr, ".png");
            im = Persistency::FrameRepository().Get(loc);
#endif // REPOSITORY_USED
        }
        return im;
    }

private:

    VideoSet*                mVideoSet;
    Keyframes*               mKeyframes;
    int                      mVideoId;
    bool                     mUseFileArchive;
    Array::Array2dVec3UInt8* mCurIm;
#ifndef REPOSITORY_USED // Here comes the deprecated stuff
    Array::ImageArchiveFile* mArchive;
#else // REPOSITORY_USED
    Array::ImageArchive*     mArchive;
#endif // REPOSITORY_USED

    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(RgbDataSrcKeyframes, Impala.Core.VideoSet);

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

#endif

---