webmmux/webmmuxstreamaudiovorbis.h

// Copyright (c) 2010 The WebM project authors. All Rights Reserved.
//
// Use of this source code is governed by a BSD-style license
// that can be found in the LICENSE file in the root of the source
// tree. An additional intellectual property rights grant can be found
// in the file PATENTS.  All contributing project authors may
// be found in the AUTHORS file in the root of the source tree.

#include "webmmuxstreamaudio.h"
#include "webmmuxcontext.h"

class CMediaTypes;

namespace VorbisTypes
{
struct VORBISFORMAT2;
}

namespace WebmMuxLib
{

class StreamAudioVorbis : public StreamAudio
{
    StreamAudioVorbis(const StreamAudioVorbis&);
    StreamAudioVorbis& operator=(const StreamAudioVorbis&);

protected:
    StreamAudioVorbis(Context&, const AM_MEDIA_TYPE&);

    ULONG GetSamplesPerSec() const;
    BYTE GetChannels() const;

    void WriteTrackCodecID();
    void WriteTrackCodecName();
    void WriteTrackCodecPrivate();

public:
    static void GetMediaTypes(CMediaTypes&);
    static bool QueryAccept(const AM_MEDIA_TYPE&);
    //static bool OnReceiveConnection(const AM_MEDIA_TYPE&);

    //static HRESULT GetAllocatorRequirements(
    //                const AM_MEDIA_TYPE&,
    //                ALLOCATOR_PROPERTIES&);

    static StreamAudio* CreateStream(Context&, const AM_MEDIA_TYPE&);

    HRESULT Receive(IMediaSample*);
    int EndOfStream();

private:
    const GUID m_subtype;
    const VorbisTypes::VORBISFORMAT2& GetFormat() const;

    class VorbisFrame : public AudioFrame
    {
        VorbisFrame& operator=(const VorbisFrame&);
        VorbisFrame(const VorbisFrame&);

    private:
        ULONG m_timecode;
        ULONG m_duration;
        BYTE* m_data;
        ULONG m_size;
        int m_lacing;

    protected:
        int GetLacing() const;

    public:
        explicit VorbisFrame(IMediaSample*, StreamAudioVorbis*);
        ~VorbisFrame();

        ULONG GetTimecode() const;
        ULONG GetDuration() const;

        ULONG GetSize() const;
        const BYTE* GetData() const;

    };

};

}  //end namespace WebmMuxLib