827d6fe73d
This commit follows the same logic as 061a0c14bb, but for the encode API: The new public encoding API will no longer be a wrapper around the old deprecated one, and the internal API used by the encoders now consists of a single receive_packet() callback that pulls frames as required. amf encoders adapted by James Almer librav1e encoder adapted by James Almer nvidia encoders adapted by James Almer MediaFoundation encoders adapted by James Almer vaapi encoders adapted by Linjie Fu v4l2_m2m encoders adapted by Andriy Gelman Signed-off-by: James Almer <jamrial@gmail.com>
149 lines
4.3 KiB
C
149 lines
4.3 KiB
C
/*
|
|
* This file is part of FFmpeg.
|
|
*
|
|
* FFmpeg is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* FFmpeg 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
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with FFmpeg; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#ifndef AVCODEC_AMFENC_H
|
|
#define AVCODEC_AMFENC_H
|
|
|
|
#include <AMF/core/Factory.h>
|
|
|
|
#include <AMF/components/VideoEncoderVCE.h>
|
|
#include <AMF/components/VideoEncoderHEVC.h>
|
|
|
|
#include "libavutil/fifo.h"
|
|
|
|
#include "avcodec.h"
|
|
|
|
|
|
/**
|
|
* AMF trace writer callback class
|
|
* Used to capture all AMF logging
|
|
*/
|
|
|
|
typedef struct AmfTraceWriter {
|
|
AMFTraceWriterVtbl *vtbl;
|
|
AVCodecContext *avctx;
|
|
} AmfTraceWriter;
|
|
|
|
/**
|
|
* AMF encoder context
|
|
*/
|
|
|
|
typedef struct AmfContext {
|
|
AVClass *avclass;
|
|
// access to AMF runtime
|
|
amf_handle library; ///< handle to DLL library
|
|
AMFFactory *factory; ///< pointer to AMF factory
|
|
AMFDebug *debug; ///< pointer to AMF debug interface
|
|
AMFTrace *trace; ///< pointer to AMF trace interface
|
|
|
|
amf_uint64 version; ///< version of AMF runtime
|
|
AmfTraceWriter tracer; ///< AMF writer registered with AMF
|
|
AMFContext *context; ///< AMF context
|
|
//encoder
|
|
AMFComponent *encoder; ///< AMF encoder object
|
|
amf_bool eof; ///< flag indicating EOF happened
|
|
AMF_SURFACE_FORMAT format; ///< AMF surface format
|
|
|
|
AVBufferRef *hw_device_ctx; ///< pointer to HW accelerator (decoder)
|
|
AVBufferRef *hw_frames_ctx; ///< pointer to HW accelerator (frame allocator)
|
|
|
|
int hwsurfaces_in_queue;
|
|
int hwsurfaces_in_queue_max;
|
|
|
|
// helpers to handle async calls
|
|
int delayed_drain;
|
|
AMFSurface *delayed_surface;
|
|
AVFrame *delayed_frame;
|
|
|
|
// shift dts back by max_b_frames in timing
|
|
AVFifoBuffer *timestamp_list;
|
|
int64_t dts_delay;
|
|
|
|
// common encoder option options
|
|
|
|
int log_to_dbg;
|
|
|
|
// Static options, have to be set before Init() call
|
|
int usage;
|
|
int profile;
|
|
int level;
|
|
int preanalysis;
|
|
int quality;
|
|
int b_frame_delta_qp;
|
|
int ref_b_frame_delta_qp;
|
|
|
|
// Dynamic options, can be set after Init() call
|
|
|
|
int rate_control_mode;
|
|
int enforce_hrd;
|
|
int filler_data;
|
|
int enable_vbaq;
|
|
int skip_frame;
|
|
int qp_i;
|
|
int qp_p;
|
|
int qp_b;
|
|
int max_au_size;
|
|
int header_spacing;
|
|
int b_frame_ref;
|
|
int intra_refresh_mb;
|
|
int coding_mode;
|
|
int me_half_pel;
|
|
int me_quarter_pel;
|
|
int aud;
|
|
|
|
// HEVC - specific options
|
|
|
|
int gops_per_idr;
|
|
int header_insertion_mode;
|
|
int min_qp_i;
|
|
int max_qp_i;
|
|
int min_qp_p;
|
|
int max_qp_p;
|
|
int tier;
|
|
} AmfContext;
|
|
|
|
/**
|
|
* Common encoder initization function
|
|
*/
|
|
int ff_amf_encode_init(AVCodecContext *avctx);
|
|
/**
|
|
* Common encoder termination function
|
|
*/
|
|
int ff_amf_encode_close(AVCodecContext *avctx);
|
|
|
|
/**
|
|
* Ecoding one frame - common function for all AMF encoders
|
|
*/
|
|
int ff_amf_receive_packet(AVCodecContext *avctx, AVPacket *avpkt);
|
|
|
|
/**
|
|
* Supported formats
|
|
*/
|
|
extern const enum AVPixelFormat ff_amf_pix_fmts[];
|
|
|
|
/**
|
|
* Error handling helper
|
|
*/
|
|
#define AMF_RETURN_IF_FALSE(avctx, exp, ret_value, /*message,*/ ...) \
|
|
if (!(exp)) { \
|
|
av_log(avctx, AV_LOG_ERROR, __VA_ARGS__); \
|
|
return ret_value; \
|
|
}
|
|
|
|
#endif //AVCODEC_AMFENC_H
|