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 38 typedef struct AmfTraceWriter { 39 AMFTraceWriterVtbl *vtbl; 40 AVCodecContext *avctx; 41 } AmfTraceWriter; 42 43 /** 44 * AMF encoder context 45 */ 46 47 typedef 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 121 extern const AVCodecHWConfigInternal *const ff_amfenc_hw_configs[]; 122 123 /** 124 * Common encoder initization function 125 */ 126 int ff_amf_encode_init(AVCodecContext *avctx); 127 /** 128 * Common encoder termination function 129 */ 130 int ff_amf_encode_close(AVCodecContext *avctx); 131 132 /** 133 * Ecoding one frame - common function for all AMF encoders 134 */ 135 int ff_amf_receive_packet(AVCodecContext *avctx, AVPacket *avpkt); 136 137 /** 138 * Supported formats 139 */ 140 extern 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