itsmattkc
3a7b3ffac6
The encoder was moved to its own thread and will transcode the PCM from the audio renderer into the chosen codec while the video frames are still received. The implementation isn't perfect and could use some cleaning up, but it is functional at the moment.
68 lines
1.9 KiB
C++
68 lines
1.9 KiB
C++
#ifndef FFMPEGENCODER_H
|
|
#define FFMPEGENCODER_H
|
|
|
|
extern "C" {
|
|
#include <libavformat/avformat.h>
|
|
#include <libswscale/swscale.h>
|
|
#include <libswresample/swresample.h>
|
|
}
|
|
|
|
#include "codec/encoder.h"
|
|
|
|
class FFmpegEncoder : public Encoder
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
FFmpegEncoder(const EncodingParams ¶ms);
|
|
|
|
public slots:
|
|
virtual void WriteAudio(const AudioRenderingParams& pcm_info, const QString& pcm_filename) override;
|
|
|
|
protected:
|
|
virtual bool OpenInternal() override;
|
|
virtual void WriteInternal(FramePtr frame) override;
|
|
virtual void CloseInternal() override;
|
|
|
|
private:
|
|
/**
|
|
* @brief Handle an error
|
|
*
|
|
* Immediately closes the Decoder (freeing memory resources) and sends the string provided to the warning stream.
|
|
* As this function closes the Decoder, no further Decoder functions should be performed after this is called
|
|
* (unless the Decoder is opened again first).
|
|
*/
|
|
void Error(const QString& s);
|
|
|
|
/**
|
|
* @brief Handle an FFmpeg error code
|
|
*
|
|
* Uses the FFmpeg API to retrieve a descriptive string for this error code and sends it to Error(). As such, this
|
|
* function also automatically closes the Decoder.
|
|
*
|
|
* @param error_code
|
|
*/
|
|
void FFmpegError(const char *context, int error_code);
|
|
|
|
void WriteAVFrame(AVFrame* frame, AVCodecContext *codec_ctx, AVStream *stream);
|
|
|
|
bool InitializeStream(enum AVMediaType type, AVStream** stream, AVCodecContext** codec_ctx, const QString& codec);
|
|
bool InitializeCodecContext(AVStream** stream, AVCodecContext** codec_ctx, AVCodec* codec);
|
|
bool SetupCodecContext(AVStream *stream, AVCodecContext *codec_ctx, AVCodec *codec);
|
|
|
|
void FlushEncoders();
|
|
|
|
AVFormatContext* fmt_ctx_;
|
|
|
|
AVStream* video_stream_;
|
|
AVCodecContext* video_codec_ctx_;
|
|
SwsContext* video_scale_ctx_;
|
|
olive::PixelFormat video_conversion_fmt_;
|
|
|
|
AVStream* audio_stream_;
|
|
AVCodecContext* audio_codec_ctx_;
|
|
SwrContext* audio_resample_ctx_;
|
|
|
|
};
|
|
|
|
#endif // FFMPEGENCODER_H
|