FLVParser.h

Go to the documentation of this file.

// FLVParser.h:  Flash Video file format parser, for Gnash.
//
//   Copyright (C) 2007, 2008, 2009, 2010, 2011, 2012
//   Free Software Foundation, Inc.
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
//


// Information about the FLV format can be found at http://osflash.org/flv

#ifndef GNASH_FLVPARSER_H
#define GNASH_FLVPARSER_H

#include "dsodefs.h"
#include "MediaParser.h" // for inheritance
#include "SimpleBuffer.h" 

#include <set>
#include <memory>
#include <map>

#include <boost/thread/mutex.hpp>

namespace gnash {
namespace media {

//
class ExtraVideoInfoFlv : public VideoInfo::ExtraInfo
{
public:

    //
    ExtraVideoInfoFlv(boost::uint8_t* extradata, size_t datasize)
        :
        data(extradata),
        size(datasize)
    {
    }

    boost::scoped_array<boost::uint8_t> data;

    size_t size;
};

//
class ExtraAudioInfoFlv : public AudioInfo::ExtraInfo
{
public:

    //
    ExtraAudioInfoFlv(boost::uint8_t* extradata, size_t datasize)
        :
        data(extradata),
        size(datasize)
    {
    }

    boost::scoped_array<boost::uint8_t> data;

    size_t size;
};

class DSOEXPORT FLVParser : public MediaParser
{

public:

    //
    static const size_t paddingBytes = 8;

        //
        FLVParser(std::auto_ptr<IOChannel> lt);

        ~FLVParser();

        // see dox in MediaParser.h
        virtual bool seek(boost::uint32_t&);

        // see dox in MediaParser.h
        virtual bool parseNextChunk();

        // see dox in MediaParser.h
        boost::uint64_t getBytesLoaded() const;

        // see dox in MediaParser.h
        bool indexingCompleted() const
        {
                return _indexingCompleted;
        }

    //
    //
    //
    virtual void fetchMetaTags(OrderedMetaTags& tags, boost::uint64_t ts);

private:

        enum tagType
        {
                FLV_AUDIO_TAG = 0x08,
                FLV_VIDEO_TAG = 0x09,
                FLV_META_TAG = 0x12
        };

        struct FLVTag : public boost::noncopyable
        {
                FLVTag(boost::uint8_t* stream)
                    :
            type(stream[0]),
            body_size(getUInt24(stream+1)),
            timestamp(getUInt24(stream+4) | (stream[7] << 24) )
                {}

                boost::uint8_t type;
                boost::uint32_t body_size;
                boost::uint32_t timestamp;
        };

        struct FLVAudioTag : public boost::noncopyable
        {
                FLVAudioTag(const boost::uint8_t& byte)
                    :
            codec( (byte & 0xf0) >> 4 ),
                    samplerate( flv_audio_rates[(byte & 0x0C) >> 2] ),
                    samplesize( 1 + ((byte & 0x02) >> 1)),
                    stereo( (byte & 0x01) )
                {
                }

                boost::uint8_t codec;

                boost::uint16_t samplerate;

                boost::uint8_t samplesize;

                bool stereo;

    private:
        
        static const boost::uint16_t flv_audio_rates[];
        
    };

        enum frameType
        {
                FLV_VIDEO_KEYFRAME = 1,
                FLV_VIDEO_INTERLACED = 2,
                FLV_VIDEO_DISPOSABLE = 3
        };

        struct FLVVideoTag : public boost::noncopyable
        {
                FLVVideoTag(const boost::uint8_t& byte)
            :
            frametype( (byte & 0xf0) >> 4 ),
                    codec( byte & 0x0f )
                {}

                boost::uint8_t frametype;
                boost::uint8_t codec;
        };

        //
        bool parseNextTag(bool index_only);

        std::auto_ptr<EncodedAudioFrame> parseAudioTag(const FLVTag& flvtag,
            const FLVAudioTag& audiotag, boost::uint32_t thisTagPos);
        
    std::auto_ptr<EncodedVideoFrame> parseVideoTag(const FLVTag& flvtag,
            const FLVVideoTag& videotag, boost::uint32_t thisTagPos);

        void indexAudioTag(const FLVTag& tag, boost::uint32_t thisTagPos);
        
    void indexVideoTag(const FLVTag& tag, const FLVVideoTag& videotag,
            boost::uint32_t thisTagPos);

        bool parseHeader();

        static boost::uint32_t getUInt24(boost::uint8_t* in);

        boost::uint64_t _lastParsedPosition;

        boost::uint64_t _nextPosToIndex;

        //
        size_t _nextAudioFrame;

        //
        size_t _nextVideoFrame;

        bool _audio;

        bool _video;

        std::auto_ptr<EncodedAudioFrame>
        readAudioFrame(boost::uint32_t dataSize, boost::uint32_t timestamp);

        std::auto_ptr<EncodedVideoFrame>
        readVideoFrame(boost::uint32_t dataSize, boost::uint32_t timestamp);

        typedef std::map<boost::uint64_t, long> CuePointsMap;
        CuePointsMap _cuePoints;

        bool _indexingCompleted;

    MetaTags _metaTags;

    boost::mutex _metaTagsMutex;
};

} // end of gnash::media namespace
} // end of gnash namespace

#endif