105 lines
2.2 KiB
C++
105 lines
2.2 KiB
C++
#ifndef EXPORTTHREAD_H
|
|
#define EXPORTTHREAD_H
|
|
|
|
#include <QThread>
|
|
#include <QOffscreenSurface>
|
|
#include <QMutex>
|
|
#include <QWaitCondition>
|
|
|
|
class ExportDialog;
|
|
struct AVFormatContext;
|
|
struct AVCodecContext;
|
|
struct AVFrame;
|
|
struct AVPacket;
|
|
struct AVStream;
|
|
struct AVCodec;
|
|
struct SwsContext;
|
|
struct SwrContext;
|
|
|
|
extern "C" {
|
|
#include <libavcodec/avcodec.h>
|
|
}
|
|
|
|
#define COMPRESSION_TYPE_CBR 0
|
|
#define COMPRESSION_TYPE_CFR 1
|
|
#define COMPRESSION_TYPE_TARGETSIZE 2
|
|
#define COMPRESSION_TYPE_TARGETBR 3
|
|
|
|
// structs that store parameters passed from the export dialogs to this thread
|
|
|
|
struct ExportParams {
|
|
// export parameters
|
|
QString filename;
|
|
bool video_enabled;
|
|
int video_codec;
|
|
int video_width;
|
|
int video_height;
|
|
double video_frame_rate;
|
|
int video_compression_type;
|
|
double video_bitrate;
|
|
bool audio_enabled;
|
|
int audio_codec;
|
|
int audio_sampling_rate;
|
|
int audio_bitrate;
|
|
long start_frame;
|
|
long end_frame;
|
|
};
|
|
|
|
struct VideoCodecParams {
|
|
int pix_fmt;
|
|
};
|
|
|
|
class ExportThread : public QThread {
|
|
Q_OBJECT
|
|
public:
|
|
ExportThread(const ExportParams& iparams, const VideoCodecParams& ivparams, QObject* parent = nullptr);
|
|
void run();
|
|
|
|
QOffscreenSurface surface;
|
|
|
|
ExportDialog* ed;
|
|
|
|
bool continueEncode;
|
|
signals:
|
|
void progress_changed(int value, qint64 remaining_ms);
|
|
public slots:
|
|
void wake();
|
|
private:
|
|
bool encode(AVFormatContext* ofmt_ctx, AVCodecContext* codec_ctx, AVFrame* frame, AVPacket* packet, AVStream* stream, bool rescale);
|
|
bool setupVideo();
|
|
bool setupAudio();
|
|
bool setupContainer();
|
|
|
|
// params imported from dialogs
|
|
ExportParams params;
|
|
VideoCodecParams vcodec_params;
|
|
|
|
AVFormatContext* fmt_ctx;
|
|
AVStream* video_stream;
|
|
AVCodec* vcodec;
|
|
AVCodecContext* vcodec_ctx;
|
|
AVFrame* video_frame;
|
|
AVFrame* sws_frame;
|
|
SwsContext* sws_ctx;
|
|
AVStream* audio_stream;
|
|
AVCodec* acodec;
|
|
AVFrame* audio_frame;
|
|
AVFrame* swr_frame;
|
|
AVCodecContext* acodec_ctx;
|
|
AVPacket video_pkt;
|
|
AVPacket audio_pkt;
|
|
SwrContext* swr_ctx;
|
|
|
|
bool vpkt_alloc;
|
|
bool apkt_alloc;
|
|
|
|
int aframe_bytes;
|
|
int ret;
|
|
char* c_filename;
|
|
|
|
QMutex mutex;
|
|
QWaitCondition waitCond;
|
|
};
|
|
|
|
#endif // EXPORTTHREAD_H
|