itsmattkc
15556c64ae
I'll be honest, this probably makes no noticeable difference at all. It should be faster and more efficient, but whether that translates into any tangible improvement is yet to be seen.
100 lines
2.7 KiB
C++
100 lines
2.7 KiB
C++
/***
|
|
|
|
Olive - Non-Linear Video Editor
|
|
Copyright (C) 2021 Olive Team
|
|
|
|
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, see <http://www.gnu.org/licenses/>.
|
|
|
|
***/
|
|
|
|
#ifndef FFMPEGENCODER_H
|
|
#define FFMPEGENCODER_H
|
|
|
|
extern "C" {
|
|
#include <libavformat/avformat.h>
|
|
#include <libswscale/swscale.h>
|
|
#include <libswresample/swresample.h>
|
|
#include <libavutil/opt.h>
|
|
}
|
|
|
|
#include "codec/encoder.h"
|
|
|
|
namespace olive {
|
|
|
|
class FFmpegEncoder : public Encoder
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
FFmpegEncoder(const EncodingParams ¶ms);
|
|
|
|
virtual QStringList GetPixelFormatsForCodec(ExportCodec::Codec c) const override;
|
|
|
|
virtual bool Open() override;
|
|
|
|
virtual bool WriteFrame(olive::FramePtr frame, olive::rational time) override;
|
|
|
|
virtual bool WriteAudio(olive::SampleBufferPtr audio) override;
|
|
|
|
virtual void Close() override;
|
|
|
|
virtual VideoParams::Format GetDesiredPixelFormat() const override
|
|
{
|
|
return video_conversion_fmt_;
|
|
}
|
|
|
|
private:
|
|
/**
|
|
* @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 QString &context, int error_code);
|
|
|
|
bool WriteAVFrame(AVFrame* frame, AVCodecContext *codec_ctx, AVStream *stream);
|
|
|
|
bool InitializeStream(enum AVMediaType type, AVStream** stream, AVCodecContext** codec_ctx, const ExportCodec::Codec &codec);
|
|
bool InitializeCodecContext(AVStream** stream, AVCodecContext** codec_ctx, AVCodec* codec);
|
|
bool SetupCodecContext(AVStream *stream, AVCodecContext *codec_ctx, AVCodec *codec);
|
|
|
|
void FlushEncoders();
|
|
void FlushCodecCtx(AVCodecContext* codec_ctx, AVStream *stream);
|
|
|
|
bool InitializeResampleContext(SampleBufferPtr audio);
|
|
|
|
AVFormatContext* fmt_ctx_;
|
|
|
|
AVStream* video_stream_;
|
|
AVCodecContext* video_codec_ctx_;
|
|
SwsContext* video_alpha_scale_ctx_;
|
|
SwsContext* video_noalpha_scale_ctx_;
|
|
VideoParams::Format video_conversion_fmt_;
|
|
|
|
AVStream* audio_stream_;
|
|
AVCodecContext* audio_codec_ctx_;
|
|
SwrContext* audio_resample_ctx_;
|
|
AVFrame* audio_frame_;
|
|
int audio_frame_offset_;
|
|
int audio_write_count_;
|
|
|
|
bool open_;
|
|
|
|
};
|
|
|
|
}
|
|
|
|
#endif // FFMPEGENCODER_H
|