mediacodecenc.c File Reference

#include "config_components.h"
#include "libavutil/avassert.h"
#include "libavutil/hwcontext_mediacodec.h"
#include "libavutil/imgutils.h"
#include "libavutil/mem.h"
#include "libavutil/opt.h"
#include "avcodec.h"
#include "bsf.h"
#include "codec_internal.h"
#include "encode.h"
#include "hwconfig.h"
#include "jni.h"
#include "mediacodec.h"
#include "mediacodec_wrapper.h"
#include "mediacodecdec_common.h"
#include "profiles.h"

Go to the source code of this file.

Data Structures
struct	MediaCodecEncContext

Macros
#define	INPUT_DEQUEUE_TIMEOUT_US   8000
#define	OUTPUT_DEQUEUE_TIMEOUT_US   8000
#define	OFFSET(x)   offsetof(MediaCodecEncContext, x)
#define	VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
#define	COMMON_OPTION
#define	MEDIACODEC_ENCODER_CLASS(name)
#define	DECLARE_MEDIACODEC_ENCODER(short_name, long_name, codec_id)

Enumerations
enum	BitrateMode { BITRATE_MODE_CQ = 0, BITRATE_MODE_VBR = 1, BITRATE_MODE_CBR = 2, BITRATE_MODE_CBR_FD = 3 }
enum	{ COLOR_FormatYUV420Planar = 0x13, COLOR_FormatYUV420SemiPlanar = 0x15, COLOR_FormatSurface = 0x7F000789 }

Functions
static void	mediacodec_output_format (AVCodecContext *avctx)
static int	mediacodec_init_bsf (AVCodecContext *avctx)
static av_cold int	mediacodec_init (AVCodecContext *avctx)
static int	mediacodec_receive (AVCodecContext *avctx, AVPacket *pkt, int *got_packet)
static void	copy_frame_to_buffer (AVCodecContext *avctx, const AVFrame *frame, uint8_t *dst, size_t size)
static int	mediacodec_send (AVCodecContext *avctx, const AVFrame *frame)
static int	mediacodec_encode (AVCodecContext *avctx, AVPacket *pkt)
static av_cold int	mediacodec_close (AVCodecContext *avctx)

Variables
struct {
int   color_format
enum AVPixelFormat   pix_fmt
}	color_formats []
static enum AVPixelFormat	avc_pix_fmts []
static const AVCodecHWConfigInternal *const	mediacodec_hw_configs []

Macro Definition Documentation

INPUT_DEQUEUE_TIMEOUT_US

#define INPUT_DEQUEUE_TIMEOUT_US   8000

Definition at line 42 of file mediacodecenc.c.

OUTPUT_DEQUEUE_TIMEOUT_US

#define OUTPUT_DEQUEUE_TIMEOUT_US   8000

Definition at line 43 of file mediacodecenc.c.

OFFSET

#define OFFSET

(

x

)

offsetof(MediaCodecEncContext, x)

Definition at line 589 of file mediacodecenc.c.

VE

#define VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 590 of file mediacodecenc.c.

COMMON_OPTION

#define COMMON_OPTION

Value:

    { "ndk_codec", "Use MediaCodec from NDK",                                                               \
                    OFFSET(use_ndk_codec), AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1, VE },                      \
    { "codec_name", "Select codec by name",                                                                 \
                    OFFSET(name), AV_OPT_TYPE_STRING, {0}, 0, 0, VE },                                      \
    { "bitrate_mode", "Bitrate control method",                                                             \
                    OFFSET(bitrate_mode), AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX, VE, .unit = "bitrate_mode" },  \
    { "cq", "Constant quality mode",                                                                                \
                    0, AV_OPT_TYPE_CONST, {.i64 = BITRATE_MODE_CQ}, 0, 0, VE, .unit = "bitrate_mode" },             \
    { "vbr", "Variable bitrate mode",                                                                               \
                    0, AV_OPT_TYPE_CONST, {.i64 = BITRATE_MODE_VBR}, 0, 0, VE, .unit = "bitrate_mode" },            \
    { "cbr", "Constant bitrate mode",                                                                               \
                    0, AV_OPT_TYPE_CONST, {.i64 = BITRATE_MODE_CBR}, 0, 0, VE, .unit = "bitrate_mode" },            \
    { "cbr_fd", "Constant bitrate mode with frame drops",                                                           \
                    0, AV_OPT_TYPE_CONST, {.i64 = BITRATE_MODE_CBR_FD}, 0, 0, VE, .unit = "bitrate_mode" },         \
    { "pts_as_dts", "Use PTS as DTS. It is enabled automatically if avctx max_b_frames <= 0, "              \
                    "since most of Android devices don't output B frames by default.",                      \
                    OFFSET(pts_as_dts), AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1, VE },                         \

Definition at line 591 of file mediacodecenc.c.

MEDIACODEC_ENCODER_CLASS

#define MEDIACODEC_ENCODER_CLASS

(

name

)

Value:

static const AVClass name ## _mediacodec_class = {  \
    .class_name = #name "_mediacodec",              \
    .item_name  = av_default_item_name,             \
    .option     = name ## _options,                 \
    .version    = LIBAVUTIL_VERSION_INT,            \
};                                                  \

Definition at line 611 of file mediacodecenc.c.

DECLARE_MEDIACODEC_ENCODER

#define DECLARE_MEDIACODEC_ENCODER	(		short_name,
			long_name,
			codec_id
	)

Value:

MEDIACODEC_ENCODER_CLASS(short_name)                                    \
const FFCodec ff_ ## short_name ## _mediacodec_encoder = {              \
    .p.name           = #short_name "_mediacodec",                      \
    CODEC_LONG_NAME(long_name " Android MediaCodec encoder"),           \
    .p.type           = AVMEDIA_TYPE_VIDEO,                             \
    .p.id             = codec_id,                                       \
    .p.capabilities   = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_DELAY           \
                        | AV_CODEC_CAP_HARDWARE,                        \
    .priv_data_size   = sizeof(MediaCodecEncContext),                   \
    .p.pix_fmts       = avc_pix_fmts,                                   \
    .init             = mediacodec_init,                                \
    FF_CODEC_RECEIVE_PACKET_CB(mediacodec_encode),                      \
    .close            = mediacodec_close,                               \
    .p.priv_class     = &short_name ## _mediacodec_class,               \
    .caps_internal    = FF_CODEC_CAP_INIT_CLEANUP,                      \
    .p.wrapper_name = "mediacodec",                                     \
    .hw_configs     = mediacodec_hw_configs,                            \
};                                                                      \

Definition at line 619 of file mediacodecenc.c.

Enumeration Type Documentation

BitrateMode

enum BitrateMode

Enumerator
BITRATE_MODE_CQ
BITRATE_MODE_VBR
BITRATE_MODE_CBR
BITRATE_MODE_CBR_FD

Definition at line 45 of file mediacodecenc.c.

anonymous enum

anonymous enum

Enumerator
COLOR_FormatYUV420Planar
COLOR_FormatYUV420SemiPlanar
COLOR_FormatSurface

Definition at line 79 of file mediacodecenc.c.

Function Documentation

mediacodec_output_format()

static void mediacodec_output_format ( AVCodecContext *  avctx )

static

Definition at line 101 of file mediacodecenc.c.

Referenced by mediacodec_init(), and mediacodec_receive().

mediacodec_init_bsf()

static int mediacodec_init_bsf ( AVCodecContext *  avctx )

static

Definition at line 115 of file mediacodecenc.c.

Referenced by mediacodec_init().

mediacodec_init()

static av_cold int mediacodec_init ( AVCodecContext *  avctx )

static

Definition at line 151 of file mediacodecenc.c.

mediacodec_receive()

static int mediacodec_receive ( AVCodecContext *  avctx, AVPacket *  pkt, int *  got_packet  )

static

Definition at line 355 of file mediacodecenc.c.

Referenced by mediacodec_encode().

copy_frame_to_buffer()

static void copy_frame_to_buffer ( AVCodecContext *  avctx, const AVFrame *  frame, uint8_t *  dst, size_t  size  )

static

Definition at line 433 of file mediacodecenc.c.

Referenced by mediacodec_send().

mediacodec_send()

static int mediacodec_send ( AVCodecContext *  avctx, const AVFrame *  frame  )

static

Definition at line 460 of file mediacodecenc.c.

Referenced by mediacodec_encode().

mediacodec_encode()

static int mediacodec_encode ( AVCodecContext *  avctx, AVPacket *  pkt  )

static

Definition at line 509 of file mediacodecenc.c.

mediacodec_close()

static av_cold int mediacodec_close ( AVCodecContext *  avctx )

static

Definition at line 556 of file mediacodecenc.c.

Variable Documentation

color_format

int color_format

Definition at line 86 of file mediacodecenc.c.

Referenced by mediacodec_init().

pix_fmt

enum AVPixelFormat pix_fmt

Definition at line 87 of file mediacodecenc.c.

color_formats

const { ... } color_formats[]

Initial value:

= {
    { COLOR_FormatYUV420Planar,         AV_PIX_FMT_YUV420P },
    { COLOR_FormatYUV420SemiPlanar,     AV_PIX_FMT_NV12    },
    { COLOR_FormatSurface,              AV_PIX_FMT_MEDIACODEC },
}

Referenced by mediacodec_init().

avc_pix_fmts

enum AVPixelFormat avc_pix_fmts[]

static

Initial value:

= {
    AV_PIX_FMT_MEDIACODEC,
    AV_PIX_FMT_YUV420P,
    AV_PIX_FMT_NV12,
    AV_PIX_FMT_NONE
}

Definition at line 94 of file mediacodecenc.c.

mediacodec_hw_configs

const AVCodecHWConfigInternal* const mediacodec_hw_configs[]

static

Initial value:

= {
    &(const AVCodecHWConfigInternal) {
        .public          = {
            .pix_fmt     = AV_PIX_FMT_MEDIACODEC,
            .methods     = AV_CODEC_HW_CONFIG_METHOD_AD_HOC |
                           AV_CODEC_HW_CONFIG_METHOD_HW_DEVICE_CTX,
            .device_type = AV_HWDEVICE_TYPE_MEDIACODEC,
        },
        .hwaccel         = NULL,
    },
    NULL
}

Definition at line 576 of file mediacodecenc.c.