From b4dbd87c0411ddfafd645d65e1214bcb2c376ddd Mon Sep 17 00:00:00 2001
From: Juanjo <pulento@users.sourceforge.net>
Date: Fri, 15 Feb 2002 20:46:37 +0000
Subject: [PATCH] - Minor changes on bitrate control

Originally committed as revision 300 to svn://svn.ffmpeg.org/ffmpeg/trunk
---
 libavcodec/motion_est.c |  2 +-
 libavcodec/mpegvideo.c  | 22 +++++++++++++++++-----
 libavcodec/mpegvideo.h  |  1 +
 3 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/libavcodec/motion_est.c b/libavcodec/motion_est.c
index 8e215f9..0cdb370 100644
--- a/libavcodec/motion_est.c
+++ b/libavcodec/motion_est.c
@@ -458,7 +458,7 @@ int estimate_motion(MpegEncContext * s,
     vard = vard >> 8;
     sum = sum >> 8;
     varc = (varc >> 8) - (sum * sum);
-    
+    s->mb_var[s->mb_width * mb_y + mb_x] = varc;
     s->avg_mb_var += varc;
      
 #if 0
diff --git a/libavcodec/mpegvideo.c b/libavcodec/mpegvideo.c
index 261a847..07c9041 100644
--- a/libavcodec/mpegvideo.c
+++ b/libavcodec/mpegvideo.c
@@ -150,16 +150,21 @@ int MPV_common_init(MpegEncContext *s)
     
     if (s->encoding) {
         /* Allocate MB type table */
-        s->mb_type = malloc(s->mb_num * sizeof(char));
+        s->mb_type = av_mallocz(s->mb_num * sizeof(char));
         if (s->mb_type == NULL) {
             perror("malloc");
             goto fail;
         }
-    
+        
+        s->mb_var = av_mallocz(s->mb_num * sizeof(INT16));
+        if (s->mb_var == NULL) {
+            perror("malloc");
+            goto fail;
+        }
         /* Allocate MV table */
         /* By now we just have one MV per MB */
-        s->mv_table[0] = malloc(s->mb_num * sizeof(INT16));
-        s->mv_table[1] = malloc(s->mb_num * sizeof(INT16));
+        s->mv_table[0] = av_mallocz(s->mb_num * sizeof(INT16));
+        s->mv_table[1] = av_mallocz(s->mb_num * sizeof(INT16));
         if (s->mv_table[1] == NULL || s->mv_table[0] == NULL) {
             perror("malloc");
             goto fail;
@@ -225,6 +230,8 @@ int MPV_common_init(MpegEncContext *s)
  fail:
     if (s->mb_type)
         free(s->mb_type);
+    if (s->mb_var)
+        free(s->mb_var);
     if (s->mv_table[0])
         free(s->mv_table[0]);
     if (s->mv_table[1])
@@ -259,6 +266,8 @@ void MPV_common_end(MpegEncContext *s)
 
     if (s->mb_type)
         free(s->mb_type);
+    if (s->mb_var)
+        free(s->mb_var);
     if (s->mv_table[0])
         free(s->mv_table[0]);
     if (s->mv_table[1])
@@ -1079,7 +1088,10 @@ static void encode_picture(MpegEncContext *s, int picture_number)
                 sub_pixels_2(s->block[5], ptr, s->linesize >> 1, dxy);
             }
             emms_c();
-
+            //if (s->avg_mb_var)
+            //    printf("\nqscale=%2d dquant=%2d var=%4d avgvar=%4d", s->qscale,
+            //        s->qscale*(s->mb_var[s->mb_width*mb_y+mb_x]/s->avg_mb_var),
+            //        s->mb_var[s->mb_width*mb_y+mb_x], s->avg_mb_var);
             /* DCT & quantize */
             if (s->h263_msmpeg4) {
                 msmpeg4_dc_scale(s);
diff --git a/libavcodec/mpegvideo.h b/libavcodec/mpegvideo.h
index 626bf8f..b797837 100644
--- a/libavcodec/mpegvideo.h
+++ b/libavcodec/mpegvideo.h
@@ -117,6 +117,7 @@ typedef struct MpegEncContext {
     int mb_x, mb_y;
     int mb_incr;
     int mb_intra;
+    INT16 *mb_var;      /* Table for MB variances */
     char *mb_type;    /* Table for MB type */
     
     /* matrix transmitted in the bitstream */
-- 
2.7.4