* DV decoder simplifications. Now it looks to be 6% faster. At least
authorRoman Shaposhnik <roman@shaposhnik.org>
Sun, 14 Mar 2004 22:09:58 +0000 (22:09 +0000)
committerRoman Shaposhnik <roman@shaposhnik.org>
Sun, 14 Mar 2004 22:09:58 +0000 (22:09 +0000)
       on my desktop.
     * Misc. DV cleanups and fixes.

Originally committed as revision 2892 to svn://svn.ffmpeg.org/ffmpeg/trunk

libavcodec/common.h
libavcodec/dv.c
libavcodec/dvdata.h

index cdbe0f0..169e15e 100644 (file)
@@ -919,6 +919,11 @@ static inline int get_bits_left(GetBitContext *s)
     return s->size_in_bits - get_bits_count(s);
 }
 
+static inline int get_bits_size(GetBitContext *s)
+{
+    return s->size_in_bits;
+}
+
 int check_marker(GetBitContext *s, const char *msg);
 void align_get_bits(GetBitContext *s);
 int init_vlc(VLC *vlc, int nb_bits, int nb_codes,
index 08be11d..1529efc 100644 (file)
@@ -59,7 +59,7 @@ typedef struct DVVideoDecodeContext {
 #endif
 
 /* XXX: also include quantization */
-static RL_VLC_ELEM *dv_rl_vlc[1];
+static RL_VLC_ELEM *dv_rl_vlc;
 /* VLC encoding lookup table */
 static struct dv_vlc_pair {
    uint32_t vlc;
@@ -111,8 +111,8 @@ static int dvvideo_init(AVCodecContext *avctx)
         init_vlc(&dv_vlc, TEX_VLC_BITS, NB_DV_VLC, 
                  dv_vlc_len, 1, 1, dv_vlc_bits, 2, 2);
 
-        dv_rl_vlc[0] = av_malloc(dv_vlc.table_size * sizeof(RL_VLC_ELEM));
-       if (!dv_rl_vlc[0]) {
+        dv_rl_vlc = av_malloc(dv_vlc.table_size * sizeof(RL_VLC_ELEM));
+       if (!dv_rl_vlc) {
            av_free(dv_vlc_map);
            return -ENOMEM;
        }
@@ -124,18 +124,15 @@ static int dvvideo_init(AVCodecContext *avctx)
             if(len<0){ //more bits needed
                 run= 0;
                 level= code;
-            } else if (code == (NB_DV_VLC - 1)) {
-                /* EOB */
-                run = 0;
-                level = 256;
             } else {
                 run=   dv_vlc_run[code] + 1;
                 level= dv_vlc_level[code];
             }
-            dv_rl_vlc[0][i].len = len;
-            dv_rl_vlc[0][i].level = level;
-            dv_rl_vlc[0][i].run = run;
+            dv_rl_vlc[i].len = len;
+            dv_rl_vlc[i].level = level;
+            dv_rl_vlc[i].run = run;
         }
+       free_vlc(&dv_vlc);
 
        for (i = 0; i < NB_DV_VLC - 1; i++) {
            if (dv_vlc_run[i] >= DV_VLC_MAP_RUN_SIZE || dv_vlc_level[i] >= DV_VLC_MAP_LEV_SIZE)
@@ -208,7 +205,6 @@ typedef struct BlockInfo {
     const uint8_t *shift_table;
     const uint8_t *scan_table;
     uint8_t pos; /* position in block */
-    uint8_t eob_reached; /* true if EOB has been reached */
     uint8_t dct_mode;
     uint8_t partial_bit_count;
     uint16_t partial_bit_buffer;
@@ -229,16 +225,17 @@ static const int mb_area_start[5] = { 1, 6, 21, 43, 64 };
 #endif
 
 /* decode ac coefs */
-static void dv_decode_ac(DVVideoDecodeContext *s, 
-                         BlockInfo *mb, DCTELEM *block, int last_index)
+static void dv_decode_ac(DVVideoDecodeContext *s, BlockInfo *mb, DCTELEM *block)
 {
+    int last_index = get_bits_size(&s->gb);
     int last_re_index;
     int shift_offset = mb->shift_offset;
     const uint8_t *scan_table = mb->scan_table;
     const uint8_t *shift_table = mb->shift_table;
     int pos = mb->pos;
-    int level, pos1, sign, run;
+    int level, pos1, run;
     int partial_bit_count;
+    int sign = 0;
 #ifndef ALT_BITSTREAM_READER //FIXME
     int re_index=0; 
     int re1_index=0;
@@ -254,7 +251,7 @@ static void dv_decode_ac(DVVideoDecodeContext *s,
     if (partial_bit_count > 0) {
         uint8_t buf[4];
         uint32_t v;
-        int l, l1;
+        int l;
         GetBitContext gb1;
 
         /* build the dummy bit buffer */
@@ -275,7 +272,7 @@ static void dv_decode_ac(DVVideoDecodeContext *s,
         {
             OPEN_READER(re1, &gb1);
             UPDATE_CACHE(re1, &gb1);
-            GET_RL_VLC(level, run, re1, &gb1, dv_rl_vlc[0]
+            GET_RL_VLC(level, run, re1, &gb1, dv_rl_vlc, 
                        TEX_VLC_BITS, 2);
             l = re1_index;
             CLOSE_READER(re1, &gb1);
@@ -283,12 +280,14 @@ static void dv_decode_ac(DVVideoDecodeContext *s,
 #ifdef VLC_DEBUG
         printf("****run=%d level=%d size=%d\n", run, level, l);
 #endif
-        /* compute codeword length */
-        l1 = (level != 256 && level != 0);
-        /* if too long, we cannot parse */
+        /* compute codeword length -- if too long, we cannot parse */
         l -= partial_bit_count;
-        if ((re_index + l + l1) > last_index)
+        if ((re_index + l + (level != 0)) > last_index) {
+           mb->partial_bit_count += (last_index - re_index);
+           mb->partial_bit_buffer = v >> (16 - mb->partial_bit_count);
             return;
+       }
+       
         /* skip read bits */
         last_re_index = 0; /* avoid warning */
         re_index += l;
@@ -307,62 +306,45 @@ static void dv_decode_ac(DVVideoDecodeContext *s,
                pos, SHOW_UBITS(re, &s->gb, 16), re_index);
 #endif
         last_re_index = re_index;
-        GET_RL_VLC(level, run, re, &s->gb, dv_rl_vlc[0]
+        GET_RL_VLC(level, run, re, &s->gb, dv_rl_vlc, 
                    TEX_VLC_BITS, 2);
     handle_vlc:
 #ifdef VLC_DEBUG
         printf("run=%d level=%d\n", run, level);
 #endif
-        if (level == 256) {
-            if (re_index > last_index) {
-            cannot_read:
-                /* put position before read code */
-                re_index = last_re_index;
-                mb->eob_reached = 0;
-                break;
-            }
-            /* EOB */
-            mb->eob_reached = 1;
-            break;
-        } else if (level != 0) {
-            if ((re_index + 1) > last_index)
-                goto cannot_read;
-            sign = SHOW_SBITS(re, &s->gb, 1);
+       if (level) {
+           sign = SHOW_SBITS(re, &s->gb, 1);
+           LAST_SKIP_BITS(re, &s->gb, 1);
+       }
+       if (re_index > last_index) {
+           /* should be < 16 bits otherwise a codeword could have been parsed */
+           re_index = last_re_index;
+           UPDATE_CACHE(re, &s->gb);
+           mb->partial_bit_count = last_index - re_index;
+           mb->partial_bit_buffer = SHOW_UBITS(re, &s->gb, mb->partial_bit_count);
+           re_index = last_index;
+           break;
+       }
+        
+       pos += run;     
+       if (pos >= 64)
+           break;
+        
+       if (level) {
             level = (level ^ sign) - sign;
-            LAST_SKIP_BITS(re, &s->gb, 1);
-            pos += run;
-            /* error */
-            if (pos >= 64) {
-                goto read_error;
-            }
             pos1 = scan_table[pos];
             level = level << (shift_table[pos1] + shift_offset);
             block[pos1] = level;
             //            printf("run=%d level=%d shift=%d\n", run, level, shift_table[pos1]);
-        } else {
-            if (re_index > last_index)
-                goto cannot_read;
-            /* level is zero: means run without coding. No
-               sign is coded */
-            pos += run;
-            /* error */
-            if (pos >= 64) {
-            read_error:
-#if defined(VLC_DEBUG) || 1
-                av_log(NULL, AV_LOG_ERROR, "error pos=%d\n", pos);
-#endif
-                /* for errors, we consider the eob is reached */
-                mb->eob_reached = 1;
-                break;
-            }
-        }
+        } 
     }
     CLOSE_READER(re, &s->gb);
     mb->pos = pos;
 }
 
-static inline void bit_copy(PutBitContext *pb, GetBitContext *gb, int bits_left)
+static inline void bit_copy(PutBitContext *pb, GetBitContext *gb)
 {
+    int bits_left = get_bits_left(gb);
     while (bits_left >= 16) {
         put_bits(pb, 16, get_bits(gb, 16));
         bits_left -= 16;
@@ -380,16 +362,14 @@ static inline void dv_decode_video_segment(DVVideoDecodeContext *s,
     int quant, dc, dct_mode, class1, j;
     int mb_index, mb_x, mb_y, v, last_index;
     DCTELEM *block, *block1;
-    int c_offset, bits_left;
+    int c_offset;
     uint8_t *y_ptr;
     BlockInfo mb_data[5 * 6], *mb, *mb1;
     void (*idct_put)(uint8_t *dest, int line_size, DCTELEM *block);
     uint8_t *buf_ptr;
     PutBitContext pb, vs_pb;
     uint8_t mb_bit_buffer[80 + 4]; /* allow some slack */
-    int mb_bit_count;
     uint8_t vs_bit_buffer[5 * 80 + 4]; /* allow some slack */
-    int vs_bit_count;
     
     memset(s->block, 0, sizeof(s->block));
 
@@ -398,18 +378,16 @@ static inline void dv_decode_video_segment(DVVideoDecodeContext *s,
     block1 = &s->block[0][0];
     mb1 = mb_data;
     init_put_bits(&vs_pb, vs_bit_buffer, 5 * 80);
-    vs_bit_count = 0;
-    for(mb_index = 0; mb_index < 5; mb_index++) {
+    for(mb_index = 0; mb_index < 5; mb_index++, mb1 += 6, block1 += 6 * 64) {
         /* skip header */
         quant = buf_ptr[3] & 0x0f;
         buf_ptr += 4;
         init_put_bits(&pb, mb_bit_buffer, 80);
-        mb_bit_count = 0;
         mb = mb1;
         block = block1;
         for(j = 0;j < 6; j++) {
-            /* NOTE: size is not important here */
-            init_get_bits(&s->gb, buf_ptr, 14*8);
+            last_index = block_sizes[j];
+           init_get_bits(&s->gb, buf_ptr, last_index);
             
             /* get the dc */
             dc = get_bits(&s->gb, 9);
@@ -426,7 +404,6 @@ static inline void dv_decode_video_segment(DVVideoDecodeContext *s,
                standard IDCT */
             dc += 1024;
             block[0] = dc;
-            last_index = block_sizes[j];
             buf_ptr += last_index >> 3;
             mb->pos = 0;
             mb->partial_bit_count = 0;
@@ -434,81 +411,57 @@ static inline void dv_decode_video_segment(DVVideoDecodeContext *s,
 #ifdef VLC_DEBUG
             printf("MB block: %d, %d ", mb_index, j);
 #endif
-            dv_decode_ac(s, mb, block, last_index);
+            dv_decode_ac(s, mb, block);
 
             /* write the remaining bits  in a new buffer only if the
                block is finished */
-            bits_left = last_index - get_bits_count(&s->gb);
-            if (mb->eob_reached) {
-                mb->partial_bit_count = 0;
-                mb_bit_count += bits_left;
-                bit_copy(&pb, &s->gb, bits_left);
-            } else {
-                /* should be < 16 bits otherwise a codeword could have
-                   been parsed */
-                mb->partial_bit_count = bits_left;
-                mb->partial_bit_buffer = get_bits(&s->gb, bits_left);
-            }
+            if (mb->pos >= 64)
+                bit_copy(&pb, &s->gb);
+            
             block += 64;
             mb++;
         }
         
-        flush_put_bits(&pb);
-
         /* pass 2 : we can do it just after */
 #ifdef VLC_DEBUG
-        printf("***pass 2 size=%d MB#=%d\n", mb_bit_count, mb_index);
+        printf("***pass 2 size=%d MB#=%d\n", put_bits_count(&pb), mb_index);
 #endif
         block = block1;
         mb = mb1;
-        init_get_bits(&s->gb, mb_bit_buffer, 80*8);
-        for(j = 0;j < 6; j++) {
-            if (!mb->eob_reached && get_bits_count(&s->gb) < mb_bit_count) {
-                dv_decode_ac(s, mb, block, mb_bit_count);
+        init_get_bits(&s->gb, mb_bit_buffer, put_bits_count(&pb));
+       flush_put_bits(&pb);
+        for(j = 0;j < 6; j++, block += 64, mb++) {
+            if (mb->pos < 64 && get_bits_left(&s->gb) > 0) {
+                dv_decode_ac(s, mb, block);
                 /* if still not finished, no need to parse other blocks */
-                if (!mb->eob_reached) {
-                    /* we could not parse the current AC coefficient,
-                       so we add the remaining bytes */
-                    bits_left = mb_bit_count - get_bits_count(&s->gb);
-                    if (bits_left > 0) {
-                        mb->partial_bit_count += bits_left;
-                        mb->partial_bit_buffer = 
-                            (mb->partial_bit_buffer << bits_left) | 
-                            get_bits(&s->gb, bits_left);
-                    }
-                    goto next_mb;
-                }
+                if (mb->pos < 64)
+                    break;
             }
-            block += 64;
-            mb++;
         }
         /* all blocks are finished, so the extra bytes can be used at
            the video segment level */
-        bits_left = mb_bit_count - get_bits_count(&s->gb);
-        vs_bit_count += bits_left;
-        bit_copy(&vs_pb, &s->gb, bits_left);
-    next_mb:
-        mb1 += 6;
-        block1 += 6 * 64;
+        if (j >= 6)
+           bit_copy(&vs_pb, &s->gb);
     }
 
     /* we need a pass other the whole video segment */
-    flush_put_bits(&vs_pb);
-        
 #ifdef VLC_DEBUG
-    printf("***pass 3 size=%d\n", vs_bit_count);
+    printf("***pass 3 size=%d\n", put_bits_count(&vs_pb));
 #endif
     block = &s->block[0][0];
     mb = mb_data;
-    init_get_bits(&s->gb, vs_bit_buffer, 5 * 80*8);
+    init_get_bits(&s->gb, vs_bit_buffer, put_bits_count(&vs_pb));
+    flush_put_bits(&vs_pb);
     for(mb_index = 0; mb_index < 5; mb_index++) {
         for(j = 0;j < 6; j++) {
-            if (!mb->eob_reached) {
+            if (mb->pos < 64) {
 #ifdef VLC_DEBUG
                 printf("start %d:%d\n", mb_index, j);
 #endif
-                dv_decode_ac(s, mb, block, vs_bit_count);
+                dv_decode_ac(s, mb, block);
             }
+           if (mb->pos >= 64 && mb->pos < 127)
+               av_log(NULL, AV_LOG_ERROR, "AC EOB marker is absent pos=%d\n", mb->pos);
             block += 64;
             mb++;
         }
index e6e0986..e60d994 100644 (file)
@@ -218,7 +218,7 @@ static const uint8_t dv_vlc_run[409] = {
   0,  0,  0,  0,  0,  0,  0,  0,
   0,  0,  0,  0,  0,  0,  0,  0,
   0,  0,  0,  0,  0,  0,  0,  0,
-  0,
+127,
 };
 
 static const uint8_t dv_vlc_level[409] = {