Keep track of macroblock partitions

[x264] / encoder / analyse.c

diff --git a/encoder/analyse.c b/encoder/analyse.c
index 8606eb84cf0ddea927e777b26333f3642ed8b15e..2cb7b2598c7f0b10e3883d2df995b5a3b6bc8124 100644 (file)
--- a/encoder/analyse.c
+++ b/encoder/analyse.c
@@ -3149,6 +3149,7 @@ static void x264_analyse_update_cache( x264_t *h, x264_mb_analysis_t *a  )
 
         case B_SKIP:
         case B_DIRECT:
+            h->mb.i_partition = h->mb.cache.direct_partition;
             x264_mb_load_mv_direct8x8( h, 0 );
             x264_mb_load_mv_direct8x8( h, 1 );
             x264_mb_load_mv_direct8x8( h, 2 );