mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
159 lines
4.7 KiB
159 lines
4.7 KiB
/* |
|
* AMD AMF support |
|
* Copyright (C) 2017 Luca Barbato |
|
* Copyright (C) 2017 Mikhail Mironov <mikhail.mironov@amd.com> |
|
* |
|
* This file is part of Libav. |
|
* |
|
* Libav 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. |
|
* |
|
* Libav 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 Libav; 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 "config.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 timestamp_last; |
|
int64_t dts_delay; |
|
|
|
// common encoder 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_send_frame(AVCodecContext *avctx, const AVFrame *frame); |
|
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; \ |
|
} |
|
|
|
#define AMF_COMMON_OPTIONS \ |
|
{ "log_to_dbg", "Enable AMF logging to debug output", OFFSET(log_to_dbg), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE } \ |
|
|
|
#endif //AVCODEC_AMFENC_H
|
|
|