| /* |
| * 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_VAAPI_ENCODE_H |
| #define AVCODEC_VAAPI_ENCODE_H |
| |
| #include <stdint.h> |
| |
| #include <va/va.h> |
| |
| #include "libavutil/hwcontext.h" |
| #include "libavutil/hwcontext_vaapi.h" |
| |
| #include "avcodec.h" |
| |
| struct VAAPIEncodeType; |
| struct VAAPIEncodePicture; |
| |
| enum { |
| MAX_CONFIG_ATTRIBUTES = 4, |
| MAX_GLOBAL_PARAMS = 4, |
| MAX_PICTURE_REFERENCES = 2, |
| MAX_PICTURE_SLICES = 1, |
| MAX_PARAM_BUFFERS = 16, |
| MAX_REORDER_DELAY = 16, |
| MAX_PARAM_BUFFER_SIZE = 1024, |
| MAX_OUTPUT_BUFFER_SIZE = 1024 * 1024, |
| }; |
| |
| enum { |
| PICTURE_TYPE_IDR = 0, |
| PICTURE_TYPE_I = 1, |
| PICTURE_TYPE_P = 2, |
| PICTURE_TYPE_B = 3, |
| }; |
| |
| enum { |
| // All encode operations are done independently. |
| ISSUE_MODE_SERIALISE_EVERYTHING = 0, |
| // Overlap as many operations as possible. |
| ISSUE_MODE_MAXIMISE_THROUGHPUT, |
| // Overlap operations only when satisfying parallel dependencies. |
| ISSUE_MODE_MINIMISE_LATENCY, |
| }; |
| |
| typedef struct VAAPIEncodeSlice { |
| void *priv_data; |
| void *codec_slice_params; |
| } VAAPIEncodeSlice; |
| |
| typedef struct VAAPIEncodePicture { |
| struct VAAPIEncodePicture *next; |
| |
| int64_t display_order; |
| int64_t encode_order; |
| int64_t pts; |
| |
| int type; |
| int input_available; |
| int encode_issued; |
| int encode_complete; |
| |
| AVFrame *input_image; |
| VASurfaceID input_surface; |
| |
| AVFrame *recon_image; |
| VASurfaceID recon_surface; |
| |
| int nb_param_buffers; |
| VABufferID param_buffers[MAX_PARAM_BUFFERS]; |
| |
| VABufferID output_buffer; |
| |
| void *priv_data; |
| void *codec_picture_params; |
| |
| int nb_refs; |
| struct VAAPIEncodePicture *refs[MAX_PICTURE_REFERENCES]; |
| |
| int nb_slices; |
| VAAPIEncodeSlice *slices[MAX_PICTURE_SLICES]; |
| } VAAPIEncodePicture; |
| |
| typedef struct VAAPIEncodeContext { |
| const AVClass *class; |
| |
| // Codec-specific hooks. |
| const struct VAAPIEncodeType *codec; |
| |
| // Codec-specific state. |
| void *priv_data; |
| |
| VAProfile va_profile; |
| VAEntrypoint va_entrypoint; |
| VAConfigID va_config; |
| VAContextID va_context; |
| |
| int va_rc_mode; |
| |
| AVBufferRef *device_ref; |
| AVHWDeviceContext *device; |
| AVVAAPIDeviceContext *hwctx; |
| |
| AVBufferRef *input_frames_ref; |
| AVHWFramesContext *input_frames; |
| |
| // Input size, set from input frames. |
| int input_width; |
| int input_height; |
| // Aligned size, set by codec init, becomes hwframe size. |
| int aligned_width; |
| int aligned_height; |
| |
| int nb_recon_frames; |
| AVBufferRef *recon_frames_ref; |
| AVHWFramesContext *recon_frames; |
| |
| VAConfigAttrib config_attributes[MAX_CONFIG_ATTRIBUTES]; |
| int nb_config_attributes; |
| |
| VAEncMiscParameterBuffer *global_params[MAX_GLOBAL_PARAMS]; |
| size_t global_params_size[MAX_GLOBAL_PARAMS]; |
| int nb_global_params; |
| |
| // Per-sequence parameter structure (VAEncSequenceParameterBuffer*). |
| void *codec_sequence_params; |
| |
| // Per-sequence parameters found in the per-picture parameter |
| // structure (VAEncPictureParameterBuffer*). |
| void *codec_picture_params; |
| |
| // Current encoding window, in display (input) order. |
| VAAPIEncodePicture *pic_start, *pic_end; |
| |
| // Next input order index (display order). |
| int64_t input_order; |
| // Number of frames that output is behind input. |
| int64_t output_delay; |
| // Number of frames decode output will need to be delayed. |
| int64_t decode_delay; |
| // Next output order index (encode order). |
| int64_t output_order; |
| |
| int issue_mode; |
| |
| // Timestamp handling. |
| int64_t first_pts; |
| int64_t dts_pts_diff; |
| int64_t ts_ring[MAX_REORDER_DELAY * 3]; |
| |
| // Frame type decision. |
| int i_per_idr; |
| int p_per_i; |
| int b_per_p; |
| int idr_counter; |
| int i_counter; |
| int p_counter; |
| int end_of_stream; |
| |
| // Codec-local options are allocated to follow this structure in |
| // memory (in the AVCodec definition, set priv_data_size to |
| // sizeof(VAAPIEncodeContext) + sizeof(VAAPIEncodeFooOptions)). |
| void *codec_options; |
| char codec_options_data[0]; |
| } VAAPIEncodeContext; |
| |
| |
| typedef struct VAAPIEncodeType { |
| size_t priv_data_size; |
| |
| int (*init)(AVCodecContext *avctx); |
| int (*close)(AVCodecContext *avctx); |
| |
| size_t sequence_params_size; |
| size_t picture_params_size; |
| size_t slice_params_size; |
| |
| int (*init_sequence_params)(AVCodecContext *avctx); |
| int (*init_picture_params)(AVCodecContext *avctx, |
| VAAPIEncodePicture *pic); |
| int (*init_slice_params)(AVCodecContext *avctx, |
| VAAPIEncodePicture *pic, |
| VAAPIEncodeSlice *slice); |
| |
| int sequence_header_type; |
| int picture_header_type; |
| int slice_header_type; |
| |
| int (*write_sequence_header)(AVCodecContext *avctx, |
| char *data, size_t *data_len); |
| int (*write_picture_header)(AVCodecContext *avctx, |
| VAAPIEncodePicture *pic, |
| char *data, size_t *data_len); |
| int (*write_slice_header)(AVCodecContext *avctx, |
| VAAPIEncodePicture *pic, |
| VAAPIEncodeSlice *slice, |
| char *data, size_t *data_len); |
| |
| int (*write_extra_buffer)(AVCodecContext *avctx, |
| VAAPIEncodePicture *pic, |
| int index, int *type, |
| char *data, size_t *data_len); |
| int (*write_extra_header)(AVCodecContext *avctx, |
| VAAPIEncodePicture *pic, |
| int index, int *type, |
| char *data, size_t *data_len); |
| } VAAPIEncodeType; |
| |
| |
| int ff_vaapi_encode2(AVCodecContext *avctx, AVPacket *pkt, |
| const AVFrame *input_image, int *got_packet); |
| |
| int ff_vaapi_encode_init(AVCodecContext *avctx, |
| const VAAPIEncodeType *type); |
| int ff_vaapi_encode_close(AVCodecContext *avctx); |
| |
| #endif /* AVCODEC_VAAPI_ENCODE_H */ |