diff --git a/doc/encoders.texi b/doc/encoders.texi index 6e50a90ca53fffb67c55708494c4549df801f6d5..f9fe331be0bd0e23c557a6c4e2f377e1d32d80b7 100644 --- a/doc/encoders.texi +++ b/doc/encoders.texi @@ -1357,6 +1357,15 @@ allows to store non-rgb pix_fmts. @end table +@section snow + +@subsection Options + +@table @option +@item iterative_dia_size +dia size for the iterative motion estimation +@end table + @section libtheora libtheora Theora encoder wrapper. diff --git a/libavcodec/snow.h b/libavcodec/snow.h index a16ca0cd965cffa52d4899f9d62e7319ab135ff2..a09b6221e48eee8e6647a11500a7396314ee87f9 100644 --- a/libavcodec/snow.h +++ b/libavcodec/snow.h @@ -177,6 +177,7 @@ typedef struct SnowContext{ int no_bitstream; int intra_penalty; int motion_est; + int iterative_dia_size; MpegEncContext m; // needed for motion estimation, should not be used for anything else, the idea is to eventually make the motion estimation independent of MpegEncContext, so this will be removed then (FIXME/XXX) diff --git a/libavcodec/snowenc.c b/libavcodec/snowenc.c index 9b5a488148750946cd7bdf5d2c8a3a04a6b98fa0..5e5dc35e86229e906a35d0d2546323e63a5e0889 100644 --- a/libavcodec/snowenc.c +++ b/libavcodec/snowenc.c @@ -1120,8 +1120,9 @@ static void iterative_me(SnowContext *s){ do{ int newx = block->mx; int newy = block->my; + int dia_size = s->iterative_dia_size ? s->iterative_dia_size : FFMAX(s->avctx->dia_size, 1); dia_change=0; - for(i=0; i<FFMAX(s->avctx->dia_size, 1); i++){ + for(i=0; i < dia_size; i++){ for(j=0; j<i; j++){ dia_change |= check_block_inter(s, mb_x, mb_y, newx+4*(i-j), newy+(4*j), obmc_edged, &best_rd); dia_change |= check_block_inter(s, mb_x, mb_y, newx-4*(i-j), newy-(4*j), obmc_edged, &best_rd); @@ -1898,6 +1899,7 @@ static const AVOption options[] = { { "memc_only", "Only do ME/MC (I frames -> ref, P frame -> ME+MC).", OFFSET(memc_only), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE }, { "no_bitstream", "Skip final bitstream writeout.", OFFSET(no_bitstream), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE }, { "intra_penalty", "Penalty for intra blocks in block decission", OFFSET(intra_penalty), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, VE }, + { "iterative_dia_size", "Dia size for the iterative ME", OFFSET(iterative_dia_size), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, VE }, { NULL }, };