1/* 2* This file is part of FFmpeg. 3* 4* FFmpeg is free software; you can redistribute it and/or 5* modify it under the terms of the GNU Lesser General Public 6* License as published by the Free Software Foundation; either 7* version 2.1 of the License, or (at your option) any later version. 8* 9* FFmpeg is distributed in the hope that it will be useful, 10* but WITHOUT ANY WARRANTY; without even the implied warranty of 11* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 12* Lesser General Public License for more details. 13* 14* You should have received a copy of the GNU Lesser General Public 15* License along with FFmpeg; if not, write to the Free Software 16* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 17*/ 18 19#ifndef AVCODEC_AMFENC_H 20#define AVCODEC_AMFENC_H 21 22#include <AMF/core/Factory.h> 23 24#include <AMF/components/VideoEncoderVCE.h> 25#include <AMF/components/VideoEncoderHEVC.h> 26 27#include "libavutil/fifo.h" 28 29#include "avcodec.h" 30#include "hwconfig.h" 31 32 33/** 34* AMF trace writer callback class 35* Used to capture all AMF logging 36*/ 37 38typedef struct AmfTraceWriter { 39 AMFTraceWriterVtbl *vtbl; 40 AVCodecContext *avctx; 41} AmfTraceWriter; 42 43/** 44* AMF encoder context 45*/ 46 47typedef struct AmfContext { 48 AVClass *avclass; 49 // access to AMF runtime 50 amf_handle library; ///< handle to DLL library 51 AMFFactory *factory; ///< pointer to AMF factory 52 AMFDebug *debug; ///< pointer to AMF debug interface 53 AMFTrace *trace; ///< pointer to AMF trace interface 54 55 amf_uint64 version; ///< version of AMF runtime 56 AmfTraceWriter tracer; ///< AMF writer registered with AMF 57 AMFContext *context; ///< AMF context 58 //encoder 59 AMFComponent *encoder; ///< AMF encoder object 60 amf_bool eof; ///< flag indicating EOF happened 61 AMF_SURFACE_FORMAT format; ///< AMF surface format 62 63 AVBufferRef *hw_device_ctx; ///< pointer to HW accelerator (decoder) 64 AVBufferRef *hw_frames_ctx; ///< pointer to HW accelerator (frame allocator) 65 66 int hwsurfaces_in_queue; 67 int hwsurfaces_in_queue_max; 68 69 // helpers to handle async calls 70 int delayed_drain; 71 AMFSurface *delayed_surface; 72 AVFrame *delayed_frame; 73 74 // shift dts back by max_b_frames in timing 75 AVFifo *timestamp_list; 76 int64_t dts_delay; 77 78 // common encoder option options 79 80 int log_to_dbg; 81 82 // Static options, have to be set before Init() call 83 int usage; 84 int profile; 85 int level; 86 int preanalysis; 87 int quality; 88 int b_frame_delta_qp; 89 int ref_b_frame_delta_qp; 90 91 // Dynamic options, can be set after Init() call 92 93 int rate_control_mode; 94 int enforce_hrd; 95 int filler_data; 96 int enable_vbaq; 97 int skip_frame; 98 int qp_i; 99 int qp_p; 100 int qp_b; 101 int max_au_size; 102 int header_spacing; 103 int b_frame_ref; 104 int intra_refresh_mb; 105 int coding_mode; 106 int me_half_pel; 107 int me_quarter_pel; 108 int aud; 109 110 // HEVC - specific options 111 112 int gops_per_idr; 113 int header_insertion_mode; 114 int min_qp_i; 115 int max_qp_i; 116 int min_qp_p; 117 int max_qp_p; 118 int tier; 119} AmfContext; 120 121extern const AVCodecHWConfigInternal *const ff_amfenc_hw_configs[]; 122 123/** 124* Common encoder initization function 125*/ 126int ff_amf_encode_init(AVCodecContext *avctx); 127/** 128* Common encoder termination function 129*/ 130int ff_amf_encode_close(AVCodecContext *avctx); 131 132/** 133* Ecoding one frame - common function for all AMF encoders 134*/ 135int ff_amf_receive_packet(AVCodecContext *avctx, AVPacket *avpkt); 136 137/** 138* Supported formats 139*/ 140extern const enum AVPixelFormat ff_amf_pix_fmts[]; 141 142/** 143* Error handling helper 144*/ 145#define AMF_RETURN_IF_FALSE(avctx, exp, ret_value, /*message,*/ ...) \ 146 if (!(exp)) { \ 147 av_log(avctx, AV_LOG_ERROR, __VA_ARGS__); \ 148 return ret_value; \ 149 } 150 151#endif //AVCODEC_AMFENC_H 152