torcmuxer.h

Go to the documentation of this file.

#ifndef TORCMUXER_H
#define TORCMUXER_H

// Qt
#include <QtGlobal>

// Torc
#include "torcsegmentedringbuffer.h"

// FFmpeg
extern "C" {
#include <libavformat/avformat.h>
}

class TorcMuxer
{
  public:
    static int AVWritePacket(void *Opaque, uint8_t *Buffer, int Size);
    static QString GetAVCCodec (const QByteArray &Packet);
    explicit TorcMuxer      (const QString &File);
    explicit TorcMuxer      (TorcSegmentedRingBuffer *Buffer);
    ~TorcMuxer();

    bool IsValid            (void);
    int  AddH264Stream      (int Width, int Height, int Profile, int Bitrate);
    int  AddDummyAudioStream(void);
    bool AddPacket          (AVPacket *Packet, bool CodecConfig);
    void FinishSegment      (bool Init);
    void Finish             (void);
    int  WriteAVPacket      (uint8_t* Buffer, int Size);

  private:
    void SetupContext       (void);
    void SetupIO            (void);
    void Start              (void);
    void WriteDummyAudio    (void);
    void CopyExtraData      (int Size, void* Source, int Stream);

  private:
    Q_DISABLE_COPY(TorcMuxer)
    // Output muxer
    AVFormatContext         *m_formatCtx;
    bool                     m_created;
    bool                     m_started;
    // Output buffers/file
    QString                  m_outputFile;
    TorcSegmentedRingBuffer *m_ringBuffer;
    AVIOContext             *m_ioContext;
    // Dummy audio generation
    AVCodecContext          *m_audioContext;
    int                      m_audioStream;
    AVFrame                 *m_audioFrame;
    AVPacket                *m_audioPacket;
    // AV Sync
    int64_t                  m_lastVideoPts;
    int64_t                  m_lastAudioPts;
};

#endif // TORCMUXER_H