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

---

StrategyUsingNativeIndex.h

Go to the documentation of this file.

#ifndef Impala_Core_Stream_Lavc_StrategyUsingNativeIndex_h
#define Impala_Core_Stream_Lavc_StrategyUsingNativeIndex_h

#include "Core/Stream/Lavc/VideoAccessStrategy.h"
#include "Core/Stream/Lavc/VideoIndex.h"

namespace Impala
{
namespace Core
{
namespace Stream
{
namespace Lavc
{

class StrategyUsingNativeIndex : public VideoAccessStrategy
{

public:

    StrategyUsingNativeIndex(const Lavc::VideoAccessObject* const vao) 
        : VideoAccessStrategy(vao)
    {
        if (!vao->NativeIndexValid())
        {
            ILOG_ERROR("Native index not valid; PTS based video access not possible");
            return;
        }

        SetNrOfFrames(vao->NativeIndexSize());

        ILOG_DEBUG("Applying video access strategy using libavcodec's " <<
            "native frame index, measuring in PTS");
    }

    virtual bool
    FrameCanBeRead(int frameNr) const
    {
        // frames recognized by lavc to be bad, can not be read
        return !FrameIsBadInNativeIndex(frameNr);
    }


protected:

    virtual int
    GetJumpFlags() const
    {
        //return AVSEEK_FLAG_ANY | AVSEEK_FLAG_BACKWARD;
        //return AVSEEK_FLAG_FRAME | AVSEEK_FLAG_ANY;
        return AVSEEK_FLAG_ANY;
    }

    virtual bool
    ScanProtected()
    {
        const AVIndexEntry* const nativeIndex = mVao->NativeIndexEntries();
        const int nativeIndexSize = mVao->NativeIndexSize();
        ILOG_INFO("Native index has size " << nativeIndexSize);


        //int offset = 0;
        //for (int e = offset; e <= offset + 10 && e <= nativeIndexSize; e++)
        //    ILOG_DEBUG("index entry " << e << ": size=" << 
        //    nativeIndex[e].size << ", flags=" << nativeIndex[e].flags << 
        //    ", timestamp=" << nativeIndex[e].timestamp << ", pos=" << 
        //    nativeIndex[e].pos << ", min_distance=" << 
        //    nativeIndex[e].min_distance << ")"); 


        int nrOfBadFrames = 0;
        int nrOfLeadingBadFrames = 0;
        for (int indexPos = 0; indexPos < nativeIndexSize; indexPos++)
        //int debugMax = 50;
        //for (int indexPos = 0; indexPos < debugMax; indexPos++)
        //for (int indexPos = nativeIndexSize - 1; indexPos >= 0; indexPos--)
        {
            if (indexPos % 250 == 0)
                ILOG_DEBUG("Scanning using native index entry " << indexPos);

            const int indexFlags = nativeIndex[indexPos].flags;
            const bool isKeyFrameAccToIndex = (indexFlags == AVINDEX_KEYFRAME);

            bool isValidFrame = !FrameIsBadInNativeIndex(indexPos);
            //if (isValidFrame)
            //{
            //    // don't attempt reading the frame since the codec ignores it
            //    ProcessFrame(false, false, indexPos);
            //    nrOfBadFrames++;
            //    if (indexPos < nrOfBadFrames)
            //        nrOfLeadingBadFrames++;
            //    continue;
            //}

            if (isValidFrame && isKeyFrameAccToIndex)
            {
                const UInt64& pts = nativeIndex[indexPos].timestamp;
                if (!mVao->Seek(pts, GetJumpFlags()))
                {
                    ILOG_ERROR("Failed on PTS based seek for frame " << 
                        indexPos << "; native index flags: " << indexFlags << ")");
                    return false;
                }
            }

            if (isValidFrame && !DecodeNextValidFrame(isKeyFrameAccToIndex, 0))
            {
                //ILOG_ERROR("Failed to read valid frame " << indexPos);
                ILOG_WARN("Failed to read valid frame according to native index: " << 
                    indexPos);
                isValidFrame = false;
                //return false;
            }

            bool treatAsKeyFrame = isKeyFrameAccToIndex;

            if (isValidFrame)
            {
                if (isKeyFrameAccToIndex && !mVao->FrameIsKey())
                {
                    ILOG_WARN("Key frame " << indexPos << 
                        " according to native index, appears not to be " << 
                        "a key frame upon decoding");
                    //return false;
                    treatAsKeyFrame = false;
                }
                else if (!isKeyFrameAccToIndex && mVao->FrameIsKey())
                {
                    ILOG_WARN("Non-key frame " << indexPos << 
                        " according to native index, appears to be " << 
                        "a key frame upon decoding");
                    //return false;
                    treatAsKeyFrame = false;
                }
            }
            else
            {
                nrOfBadFrames++;
                if (indexPos < nrOfBadFrames)
                    nrOfLeadingBadFrames++;
            }

            ProcessFrame(isValidFrame, treatAsKeyFrame, indexPos);
        }

        ILOG_DEBUG("Bad frames according to native index: " << 
            nrOfBadFrames << " (of which are " <<
            "leading: " << nrOfLeadingBadFrames << ")");
        mBadFrameCount = nrOfBadFrames;
        mLeadingBadFrameCount = nrOfLeadingBadFrames;
        return true;
    }

    virtual VideoIndex*
    ConstructIndexProtected() const
    {
        const AVIndexEntry* const idx = mVao->NativeIndexEntries();
        VideoIndex* videoIndex = new VideoIndex();

        int prevSeekableFrame = -1;
        for (int frameNr = 0; frameNr < NrOfFrames(); frameNr++)
        {
            // note that currently the external index may be bypassed when
            // navigating to a frame; but for PTS this may not work because
            // certain invalid frames will not be 'seen' by sequential reads,
            // while certain valid frames will not be recognized (matched) when
            // reading sequentially. And so I have to introduce restrictions for
            // reading sequentially i.e. in ~Lavc.


            if (!FrameIsValid(frameNr))
            {
                // if invalidity followed from flags or size in the index,
                // I assume that the frame will not be 'seen' during sequential
                // reads?!

                // if invalidity followed from reading a bad frame, the frame 
                // *will* be 'seen' by sequential reads.

                if (idx[frameNr].size > 0 &&
                    ((idx[frameNr].flags | AVINDEX_KEYFRAME) == AVINDEX_KEYFRAME))
                {
                    bool lookForKeyFrame = (idx[frameNr].flags == AVINDEX_KEYFRAME);
                    //bool foundFrame = 
                        //mVao->DecodeNextValidFrame(lookForKeyFrame, 0);
                    bool isValidFrame;
                    bool isKeyFrame;
                    int res = DecodeNextFrame(&isValidFrame, &isKeyFrame, 0);
                    //if (foundFrame)
                    if (res >= 0 && isValidFrame)
                    {
                        String hash = mVao->CurrentHash();
                        ILOG_WARN("Unexpectedly able to successfully read a (" << 
                            mVao->FrameType() << ") frame for (invalid) frame " << 
                            frameNr << ", having hash value " << hash <<
                            "; frame data will be ignored");
                    }

                    //currentDistance++;
                }

                continue;
            }

            const UInt64& pts = idx[frameNr].timestamp;
            bool isKeyFrame = FrameIsKey(frameNr);
            if (isKeyFrame)
            {
                if (!mVao->Seek(pts, GetJumpFlags()))
                {
                    ILOG_ERROR("Failed on PTS based seek for frame: " << 
                        frameNr << (isKeyFrame ? " (is key frame)" : ""));
                    return 0;
                }
            }

            // read next frame
            //bool lookForKeyFrame = false;
            //PacketTrace trace(10);
            //bool foundFrame = 
            //    mVao->DecodeNextValidFrame(lookForKeyFrame, 0);
            //bool isValidFrame;
            //int res = mVao->DecodeNextFrame(&isValidFrame, &isKeyFrame);
            //if (res < 0 || !isValidFrame)
            //{
            //    ILOG_ERROR("Unexpectedly unable to successfully read frame: " << 
            //        frameNr);
            //    return 0;
            //}
            //mVao->CurrentFrameToRgb();
            if (!DecodeNextValidFrame(isKeyFrame, 0))
            {
                ILOG_ERROR("Failed to read valid frame " << frameNr);
                return 0;
            }

            mVao->CurrentFrameToRgb();

            // if not a key frame, check whether the frame is reproducable
            // using sequential reading from previous seekable frame
            //if (!mScanner.FrameIsKey(frameNr))
            if (!isKeyFrame)
            {
                if (prevSeekableFrame >= 0)
                {
                    // kijk of dit frame via seq read vanaf laatste seek reprodbr is
                    //if (foundFrame)
                    
                    if (CurrentFrameMatchesFrame(frameNr, 0))
                    {
                        //currentDistance++;
                        continue;
                    }
                    else
                    {
                        ILOG_ERROR("Non-reproducable frame: " << frameNr <<
                            ", current hash value is " << mVao->CurrentHash());
                        return 0;
                    }
                }
            }

            if (!CurrentFrameMatchesFrame(frameNr, 0))
            {
                ILOG_ERROR("Non-reproducable key frame: " << frameNr <<
                    ", current hash value is " << mVao->CurrentHash());
                return 0;
            }

            // add an entry to the external index for this key frame
            videoIndex->AddFrame(frameNr, true, pts);
            ILOG_DEBUG("Indexed frame: " << frameNr << " " << 
                frameNr << " " << true << " " << pts);
            prevSeekableFrame = frameNr;
            //currentDistance = 0;
        }

        return videoIndex;
    }

    virtual bool
    NextFrameMatchesFrame(int frameNr, const VideoIndex& videoIndex) const
    {
        return true; // already tested at index creation
    }


private:

    // Returns 'true' if the specified frame is bad according to the
    // native index
    bool
    FrameIsBadInNativeIndex(int frameNr) const
    {
        const AVIndexEntry* const nativeIndex = mVao->NativeIndexEntries();
        const int indexFlags = nativeIndex[frameNr].flags;
        const bool isBad = (nativeIndex[frameNr].size <= 0) ||
            (indexFlags != 0 && indexFlags != AVINDEX_KEYFRAME);
        return isBad;
    }


    ILOG_VAR_DECL;

}; //class

ILOG_VAR_INIT(StrategyUsingNativeIndex, Impala.Core.Stream.Lavc);

}}}} // namespace Impala::Core::Stream::Lavc

#endif

---