Función 1.encode_frame
En la función principal de aomenc.c del final de la codificación, después de ingresar al bucle del proceso de codificación, lea cada cuadro del video en un bucle y luego codifique cada cuadro a través de la función encode_frame .
La función encode_frame es principalmente para realizar algún trabajo de escalado del cuadro actual, iniciar el temporizador y luego llamar a la función aom_codec_encode para codificar.
static void encode_frame(struct stream_state *stream,
struct AvxEncoderConfig *global, struct aom_image *img,
unsigned int frames_in) {
aom_codec_pts_t frame_start, next_frame_start; //起始时间戳
struct aom_codec_enc_cfg *cfg = &stream->config.cfg;
struct aom_usec_timer timer;
frame_start =
(cfg->g_timebase.den * (int64_t)(frames_in - 1) * global->framerate.den) /
cfg->g_timebase.num / global->framerate.num;
next_frame_start =
(cfg->g_timebase.den * (int64_t)(frames_in)*global->framerate.den) /
cfg->g_timebase.num / global->framerate.num;
/* Scale if necessary */
if (img) {
if ((img->fmt & AOM_IMG_FMT_HIGHBITDEPTH) &&
(img->d_w != cfg->g_w || img->d_h != cfg->g_h)) {
if (img->fmt != AOM_IMG_FMT_I42016) {
fprintf(stderr, "%s can only scale 4:2:0 inputs\n", exec_name);
exit(EXIT_FAILURE);
}
#if CONFIG_LIBYUV
if (!stream->img) {
stream->img =
aom_img_alloc(NULL, AOM_IMG_FMT_I42016, cfg->g_w, cfg->g_h, 16);
}
I420Scale_16(
(uint16_t *)img->planes[AOM_PLANE_Y], img->stride[AOM_PLANE_Y] / 2,
(uint16_t *)img->planes[AOM_PLANE_U], img->stride[AOM_PLANE_U] / 2,
(uint16_t *)img->planes[AOM_PLANE_V], img->stride[AOM_PLANE_V] / 2,
img->d_w, img->d_h, (uint16_t *)stream->img->planes[AOM_PLANE_Y],
stream->img->stride[AOM_PLANE_Y] / 2,
(uint16_t *)stream->img->planes[AOM_PLANE_U],
stream->img->stride[AOM_PLANE_U] / 2,
(uint16_t *)stream->img->planes[AOM_PLANE_V],
stream->img->stride[AOM_PLANE_V] / 2, stream->img->d_w,
stream->img->d_h, kFilterBox);
img = stream->img;
#else
stream->encoder.err = 1;
ctx_exit_on_error(&stream->encoder,
"Stream %d: Failed to encode frame.\n"
"libyuv is required for scaling but is currently "
"disabled.\n"
"Be sure to specify -DCONFIG_LIBYUV=1 when running "
"cmake.\n",
stream->index);
#endif
}
}
if (img && (img->d_w != cfg->g_w || img->d_h != cfg->g_h)) {
if (img->fmt != AOM_IMG_FMT_I420 && img->fmt != AOM_IMG_FMT_YV12) {
fprintf(stderr, "%s can only scale 4:2:0 8bpp inputs\n", exec_name);
exit(EXIT_FAILURE);
}
#if CONFIG_LIBYUV
if (!stream->img)
stream->img =
aom_img_alloc(NULL, AOM_IMG_FMT_I420, cfg->g_w, cfg->g_h, 16);
I420Scale(
img->planes[AOM_PLANE_Y], img->stride[AOM_PLANE_Y],
img->planes[AOM_PLANE_U], img->stride[AOM_PLANE_U],
img->planes[AOM_PLANE_V], img->stride[AOM_PLANE_V], img->d_w, img->d_h,
stream->img->planes[AOM_PLANE_Y], stream->img->stride[AOM_PLANE_Y],
stream->img->planes[AOM_PLANE_U], stream->img->stride[AOM_PLANE_U],
stream->img->planes[AOM_PLANE_V], stream->img->stride[AOM_PLANE_V],
stream->img->d_w, stream->img->d_h, kFilterBox);
img = stream->img;
#else
stream->encoder.err = 1;
ctx_exit_on_error(&stream->encoder,
"Stream %d: Failed to encode frame.\n"
"Scaling disabled in this configuration. \n"
"To enable, configure with --enable-libyuv\n",
stream->index);
#endif
}
aom_usec_timer_start(&timer);
aom_codec_encode(&stream->encoder, img, frame_start,
(uint32_t)(next_frame_start - frame_start), 0);
aom_usec_timer_mark(&timer);
stream->cx_time += aom_usec_timer_elapsed(&timer);
ctx_exit_on_error(&stream->encoder, "Stream %d: Failed to encode frame",
stream->index);
}Función 2.aom_codec_encode
La función principal de esta función es codificar un marco. Codifique el fotograma de vídeo en un "tiempo de presentación" determinado. El sello de tiempo de presentación (PTS) debe aumentarse estrictamente.
aom_codec_err_t aom_codec_encode(aom_codec_ctx_t *ctx, const aom_image_t *img,
aom_codec_pts_t pts, unsigned long duration,
aom_enc_frame_flags_t flags);estructura de contexto del códec aom_codec_ctx_t , la interfaz real entre el código de usuario y el códec
Todos los códecs deben admitir completamente esta estructura de contexto. En términos generales, estos datos deben considerarse como datos privados del algoritmo del códec y no manipulados ni verificados por la aplicación que realiza la llamada. La aplicación puede hacer referencia al miembro "nombre" para obtener una descripción imprimible del algoritmo.
Almacena el nombre del códec, un puntero al códec aom que lo inicializó, indicadores de inicialización, la configuración del codificador o decodificador y un puntero a los datos internos.
typedef struct aom_codec_ctx {
const char *name; /**< Printable interface name 可打印接口名称 */
aom_codec_iface_t *iface; /**< Interface pointers 接口指针*/
aom_codec_err_t err; /**< Last returned error 上次返回的错误*/
const char *err_detail; /**< Detailed info, if available 详细信息（如果有） */
aom_codec_flags_t init_flags; /**< Flags passed at init time 初始化时传递的标志*/
union {
/**< Decoder Configuration Pointer 解码器配置指针 */
const struct aom_codec_dec_cfg *dec;
/**< Encoder Configuration Pointer */
const struct aom_codec_enc_cfg *enc;
const void *raw;
} config; /**< Configuration pointer aliasing union */
aom_codec_priv_t *priv; /**< Algorithm private storage 算法专用存储 */
} aom_codec_ctx_t;
parámetro:
- ctx apunta al puntero ctx de este contexto de instancia
- img Los datos de la imagen img que se codificarán, NULL significa actualizar el búfer.
- pts muestra la marca de tiempo, expresada en unidades de base de tiempo.
- Duración muestra la duración del fotograma, en unidades de base de tiempo.
- Las banderas se utilizan para codificar las banderas de este marco.
valor de retorno:
- AOM_CODEC_OK: La operación se ha completado sin errores.
- AOM_CODEC_INCAPABLE: El algoritmo no tiene funciones requeridas.
- AOM_CODEC_INVALID_PARAM: Los parámetros proporcionados por la aplicación no son válidos, el formato de imagen no es compatible, etc.
Cuando se haya pasado el último fotograma al codificador, debe continuar llamando a esta función y establecer el parámetro img en NULL. Esto señalará el final de la secuencia al codificador y le permitirá codificar cualquier búfer reservado. Cuando se llama aom_codec_encode () y aom_codec_get_cx_data () no devuelve ningún dato, la codificación está completa.
Esta función llama principalmente encoder_encode para la codificación.
aom_codec_err_t aom_codec_encode(aom_codec_ctx_t *ctx, const aom_image_t *img,
aom_codec_pts_t pts, unsigned long duration,
aom_enc_frame_flags_t flags) {
aom_codec_err_t res = AOM_CODEC_OK;
if (!ctx || (img && !duration))
res = AOM_CODEC_INVALID_PARAM;
else if (!ctx->iface || !ctx->priv)
res = AOM_CODEC_ERROR;
else if (!(ctx->iface->caps & AOM_CODEC_CAP_ENCODER))
res = AOM_CODEC_INCAPABLE;
else {
/* Execute in a normalized floating point environment, if the platform
* requires it.
*/
FLOATING_POINT_INIT
res = ctx->iface->enc.encode(get_alg_priv(ctx), img, pts, duration, flags);
FLOATING_POINT_RESTORE
}La función de get_alg_priv () función para obtener aom_codec_ctx_t cuerpo estructural aom_codec_priv_t puntero
static aom_codec_alg_priv_t *get_alg_priv(aom_codec_ctx_t *ctx) {
return (aom_codec_alg_priv_t *)ctx->priv;
}aom_codec_priv_t
Estructura de datos privados del códec.
Contiene datos específicos de la implementación del códec. Esta estructura no es transparente para la aplicación.