libavcodec/svq3.c
8b82a956
 /*
406792e7
  * Copyright (c) 2003 The FFmpeg Project
8b82a956
  *
b78e7197
  * This file is part of FFmpeg.
  *
  * FFmpeg is free software; you can redistribute it and/or
8b82a956
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
b78e7197
  * version 2.1 of the License, or (at your option) any later version.
8b82a956
  *
b78e7197
  * FFmpeg is distributed in the hope that it will be useful,
8b82a956
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
b78e7197
  * License along with FFmpeg; if not, write to the Free Software
5509bffa
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
e5a389a1
  */
 
 /*
8b82a956
  * How to use this decoder:
  * SVQ3 data is transported within Apple Quicktime files. Quicktime files
89a79364
  * have stsd atoms to describe media trak properties. A stsd atom for a
  * video trak contains 1 or more ImageDescription atoms. These atoms begin
  * with the 4-byte length of the atom followed by the codec fourcc. Some
  * decoders need information in this atom to operate correctly. Such
  * is the case with SVQ3. In order to get the best use out of this decoder,
  * the calling app must make the SVQ3 ImageDescription atom available
8b82a956
  * via the AVCodecContext's extradata[_size] field:
  *
115329f1
  * AVCodecContext.extradata = pointer to ImageDescription, first characters
89a79364
  * are expected to be 'S', 'V', 'Q', and '3', NOT the 4-byte atom length
115329f1
  * AVCodecContext.extradata_size = size of ImageDescription atom memory
  * buffer (which will be the same as the ImageDescription atom size field
89a79364
  * from the QT file, minus 4 bytes since the length is missing)
  *
  * You will know you have these parameters passed correctly when the decoder
  * correctly decodes this file:
ddaf298c
  *  http://samples.mplayerhq.hu/V-codecs/SVQ3/Vertical400kbit.sorenson3.mov
8b82a956
  */
903d58f6
 #include "internal.h"
 #include "dsputil.h"
 #include "avcodec.h"
 #include "mpegvideo.h"
 #include "h264.h"
 
 #include "h264data.h" //FIXME FIXME FIXME
 
188d3c51
 #include "h264_mvpred.h"
903d58f6
 #include "golomb.h"
 #include "rectangle.h"
 #include "vdpau_internal.h"
 
b250f9c6
 #if CONFIG_ZLIB
c4864924
 #include <zlib.h>
 #endif
 
2be3fe39
 #include "svq1.h"
 
8b82a956
 /**
ba87f080
  * @file
8b82a956
  * svq3 decoder.
  */
 
8811679c
 typedef struct {
     H264Context h;
     int halfpel_flag;
     int thirdpel_flag;
     int unknown_flag;
     int next_slice_index;
     uint32_t watermark_key;
2264c110
     uint8_t *buf;
     int buf_size;
8811679c
 } SVQ3Context;
 
115329f1
 #define FULLPEL_MODE  1
 #define HALFPEL_MODE  2
94d44f45
 #define THIRDPEL_MODE 3
2e26c8d2
 #define PREDICT_MODE  4
115329f1
 
f7a8c179
 /* dual scan (from some older h264 draft)
  o-->o-->o   o
          |  /|
  o   o   o / o
  | / |   |/  |
  o   o   o   o
115329f1
    /
f7a8c179
  o-->o-->o-->o
 */
76de302d
 static const uint8_t svq3_scan[16] = {
     0+0*4, 1+0*4, 2+0*4, 2+1*4,
     2+2*4, 3+0*4, 3+1*4, 3+2*4,
     0+1*4, 0+2*4, 1+1*4, 1+2*4,
     0+3*4, 1+3*4, 2+3*4, 3+3*4,
8b82a956
 };
 
 static const uint8_t svq3_pred_0[25][2] = {
76de302d
     { 0, 0 },
     { 1, 0 }, { 0, 1 },
     { 0, 2 }, { 1, 1 }, { 2, 0 },
     { 3, 0 }, { 2, 1 }, { 1, 2 }, { 0, 3 },
     { 0, 4 }, { 1, 3 }, { 2, 2 }, { 3, 1 }, { 4, 0 },
     { 4, 1 }, { 3, 2 }, { 2, 3 }, { 1, 4 },
     { 2, 4 }, { 3, 3 }, { 4, 2 },
     { 4, 3 }, { 3, 4 },
     { 4, 4 }
8b82a956
 };
 
 static const int8_t svq3_pred_1[6][6][5] = {
76de302d
     { { 2,-1,-1,-1,-1 }, { 2, 1,-1,-1,-1 }, { 1, 2,-1,-1,-1 },
       { 2, 1,-1,-1,-1 }, { 1, 2,-1,-1,-1 }, { 1, 2,-1,-1,-1 } },
     { { 0, 2,-1,-1,-1 }, { 0, 2, 1, 4, 3 }, { 0, 1, 2, 4, 3 },
       { 0, 2, 1, 4, 3 }, { 2, 0, 1, 3, 4 }, { 0, 4, 2, 1, 3 } },
     { { 2, 0,-1,-1,-1 }, { 2, 1, 0, 4, 3 }, { 1, 2, 4, 0, 3 },
       { 2, 1, 0, 4, 3 }, { 2, 1, 4, 3, 0 }, { 1, 2, 4, 0, 3 } },
     { { 2, 0,-1,-1,-1 }, { 2, 0, 1, 4, 3 }, { 1, 2, 0, 4, 3 },
       { 2, 1, 0, 4, 3 }, { 2, 1, 3, 4, 0 }, { 2, 4, 1, 0, 3 } },
     { { 0, 2,-1,-1,-1 }, { 0, 2, 1, 3, 4 }, { 1, 2, 3, 0, 4 },
       { 2, 0, 1, 3, 4 }, { 2, 1, 3, 0, 4 }, { 2, 0, 4, 3, 1 } },
     { { 0, 2,-1,-1,-1 }, { 0, 2, 4, 1, 3 }, { 1, 4, 2, 0, 3 },
       { 4, 2, 0, 1, 3 }, { 2, 0, 1, 4, 3 }, { 4, 2, 1, 0, 3 } },
8b82a956
 };
 
 static const struct { uint8_t run; uint8_t level; } svq3_dct_tables[2][16] = {
76de302d
     { { 0, 0 }, { 0, 1 }, { 1, 1 }, { 2, 1 }, { 0, 2 }, { 3, 1 }, { 4, 1 }, { 5, 1 },
       { 0, 3 }, { 1, 2 }, { 2, 2 }, { 6, 1 }, { 7, 1 }, { 8, 1 }, { 9, 1 }, { 0, 4 } },
     { { 0, 0 }, { 0, 1 }, { 1, 1 }, { 0, 2 }, { 2, 1 }, { 0, 3 }, { 0, 4 }, { 0, 5 },
       { 3, 1 }, { 4, 1 }, { 1, 2 }, { 1, 3 }, { 0, 6 }, { 0, 7 }, { 0, 8 }, { 0, 9 } }
8b82a956
 };
 
 static const uint32_t svq3_dequant_coeff[32] = {
76de302d
      3881,  4351,  4890,  5481,  6154,  6914,  7761,  8718,
      9781, 10987, 12339, 13828, 15523, 17435, 19561, 21873,
     24552, 27656, 30847, 34870, 38807, 43747, 49103, 54683,
     61694, 68745, 77615, 89113,100253,109366,126635,141533
8b82a956
 };
 
290fabc6
 void ff_svq3_luma_dc_dequant_idct_c(DCTELEM *output, DCTELEM *input, int qp){
76de302d
     const int qmul = svq3_dequant_coeff[qp];
8b82a956
 #define stride 16
     int i;
     int temp[16];
290fabc6
     static const uint8_t x_offset[4]={0, 1*stride, 4*stride, 5*stride};
76de302d
 
290fabc6
     for(i=0; i<4; i++){
         const int z0 = 13*(input[4*i+0] +    input[4*i+2]);
         const int z1 = 13*(input[4*i+0] -    input[4*i+2]);
         const int z2 =  7* input[4*i+1] - 17*input[4*i+3];
         const int z3 = 17* input[4*i+1] +  7*input[4*i+3];
76de302d
 
         temp[4*i+0] = z0+z3;
         temp[4*i+1] = z1+z2;
         temp[4*i+2] = z1-z2;
         temp[4*i+3] = z0-z3;
8b82a956
     }
 
290fabc6
     for(i=0; i<4; i++){
         const int offset= x_offset[i];
         const int z0= 13*(temp[4*0+i] +    temp[4*2+i]);
         const int z1= 13*(temp[4*0+i] -    temp[4*2+i]);
         const int z2=  7* temp[4*1+i] - 17*temp[4*3+i];
         const int z3= 17* temp[4*1+i] +  7*temp[4*3+i];
 
         output[stride* 0+offset] = ((z0 + z3)*qmul + 0x80000) >> 20;
         output[stride* 2+offset] = ((z1 + z2)*qmul + 0x80000) >> 20;
         output[stride* 8+offset] = ((z1 - z2)*qmul + 0x80000) >> 20;
         output[stride*10+offset] = ((z0 - z3)*qmul + 0x80000) >> 20;
8b82a956
     }
 }
 #undef stride
 
881b5b80
 void ff_svq3_add_idct_c(uint8_t *dst, DCTELEM *block, int stride, int qp,
76de302d
                             int dc)
7f8205da
 {
76de302d
     const int qmul = svq3_dequant_coeff[qp];
8b82a956
     int i;
55fde95e
     uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
8b82a956
 
     if (dc) {
         dc = 13*13*((dc == 1) ? 1538*block[0] : ((qmul*(block[0] >> 3)) / 2));
         block[0] = 0;
     }
 
76de302d
     for (i = 0; i < 4; i++) {
         const int z0 = 13*(block[0 + 4*i] +    block[2 + 4*i]);
         const int z1 = 13*(block[0 + 4*i] -    block[2 + 4*i]);
         const int z2 =  7* block[1 + 4*i] - 17*block[3 + 4*i];
         const int z3 = 17* block[1 + 4*i] +  7*block[3 + 4*i];
8b82a956
 
76de302d
         block[0 + 4*i] = z0 + z3;
         block[1 + 4*i] = z1 + z2;
         block[2 + 4*i] = z1 - z2;
         block[3 + 4*i] = z0 - z3;
8b82a956
     }
 
76de302d
     for (i = 0; i < 4; i++) {
         const int z0 = 13*(block[i + 4*0] +    block[i + 4*2]);
         const int z1 = 13*(block[i + 4*0] -    block[i + 4*2]);
         const int z2 =  7* block[i + 4*1] - 17*block[i + 4*3];
         const int z3 = 17* block[i + 4*1] +  7*block[i + 4*3];
         const int rr = (dc + 0x80000);
 
         dst[i + stride*0] = cm[ dst[i + stride*0] + (((z0 + z3)*qmul + rr) >> 20) ];
         dst[i + stride*1] = cm[ dst[i + stride*1] + (((z1 + z2)*qmul + rr) >> 20) ];
         dst[i + stride*2] = cm[ dst[i + stride*2] + (((z1 - z2)*qmul + rr) >> 20) ];
         dst[i + stride*3] = cm[ dst[i + stride*3] + (((z0 - z3)*qmul + rr) >> 20) ];
8b82a956
     }
 }
 
7f8205da
 static inline int svq3_decode_block(GetBitContext *gb, DCTELEM *block,
                                     int index, const int type)
 {
76de302d
     static const uint8_t *const scan_patterns[4] =
     { luma_dc_zigzag_scan, zigzag_scan, svq3_scan, chroma_dc_scan };
8b82a956
 
76de302d
     int run, level, sign, vlc, limit;
     const int intra = (3 * type) >> 2;
     const uint8_t *const scan = scan_patterns[type];
8b82a956
 
76de302d
     for (limit = (16 >> intra); index < 16; index = limit, limit += 8) {
         for (; (vlc = svq3_get_ue_golomb(gb)) != 0; index++) {
8b82a956
 
764ffdd0
           if (vlc < 0)
76de302d
               return -1;
8b82a956
 
76de302d
           sign = (vlc & 0x1) - 1;
           vlc  = (vlc + 1) >> 1;
 
           if (type == 3) {
               if (vlc < 3) {
                   run   = 0;
                   level = vlc;
               } else if (vlc < 4) {
                   run   = 1;
                   level = 1;
               } else {
                   run   = (vlc & 0x3);
                   level = ((vlc + 9) >> 2) - run;
               }
           } else {
764ffdd0
               if (vlc < 16U) {
76de302d
                   run   = svq3_dct_tables[intra][vlc].run;
                   level = svq3_dct_tables[intra][vlc].level;
               } else if (intra) {
                   run   = (vlc & 0x7);
                   level = (vlc >> 3) + ((run == 0) ? 8 : ((run < 2) ? 2 : ((run < 5) ? 0 : -1)));
               } else {
                   run   = (vlc & 0xF);
                   level = (vlc >> 4) + ((run == 0) ? 4 : ((run < 3) ? 2 : ((run < 10) ? 1 : 0)));
               }
           }
8b82a956
 
76de302d
           if ((index += run) >= limit)
               return -1;
8b82a956
 
76de302d
           block[scan[index]] = (level ^ sign) - sign;
         }
8b82a956
 
76de302d
         if (type != 2) {
             break;
         }
8b82a956
     }
 
76de302d
     return 0;
8b82a956
 }
 
7f8205da
 static inline void svq3_mc_dir_part(MpegEncContext *s,
                                     int x, int y, int width, int height,
                                     int mx, int my, int dxy,
                                     int thirdpel, int dir, int avg)
 {
76de302d
     const Picture *pic = (dir == 0) ? &s->last_picture : &s->next_picture;
     uint8_t *src, *dest;
     int i, emu = 0;
     int blocksize = 2 - (width>>3); //16->0, 8->1, 4->2
 
     mx += x;
     my += y;
8b82a956
 
76de302d
     if (mx < 0 || mx >= (s->h_edge_pos - width  - 1) ||
         my < 0 || my >= (s->v_edge_pos - height - 1)) {
115329f1
 
76de302d
         if ((s->flags & CODEC_FLAG_EMU_EDGE)) {
             emu = 1;
         }
8b82a956
 
76de302d
         mx = av_clip (mx, -16, (s->h_edge_pos - width  + 15));
         my = av_clip (my, -16, (s->v_edge_pos - height + 15));
8b82a956
     }
 
76de302d
     /* form component predictions */
     dest = s->current_picture.data[0] + x + y*s->linesize;
     src  = pic->data[0] + mx + my*s->linesize;
 
     if (emu) {
d23e3e5f
         s->dsp.emulated_edge_mc(s->edge_emu_buffer, src, s->linesize, (width + 1), (height + 1),
76de302d
                             mx, my, s->h_edge_pos, s->v_edge_pos);
a7d3e772
         src = s->edge_emu_buffer;
8b82a956
     }
76de302d
     if (thirdpel)
         (avg ? s->dsp.avg_tpel_pixels_tab : s->dsp.put_tpel_pixels_tab)[dxy](dest, src, s->linesize, width, height);
     else
         (avg ? s->dsp.avg_pixels_tab : s->dsp.put_pixels_tab)[blocksize][dxy](dest, src, s->linesize, height);
 
     if (!(s->flags & CODEC_FLAG_GRAY)) {
         mx     = (mx + (mx < (int) x)) >> 1;
         my     = (my + (my < (int) y)) >> 1;
         width  = (width  >> 1);
         height = (height >> 1);
         blocksize++;
 
         for (i = 1; i < 3; i++) {
             dest = s->current_picture.data[i] + (x >> 1) + (y >> 1)*s->uvlinesize;
             src  = pic->data[i] + mx + my*s->uvlinesize;
 
             if (emu) {
d23e3e5f
                 s->dsp.emulated_edge_mc(s->edge_emu_buffer, src, s->uvlinesize, (width + 1), (height + 1),
76de302d
                                     mx, my, (s->h_edge_pos >> 1), (s->v_edge_pos >> 1));
                 src = s->edge_emu_buffer;
             }
             if (thirdpel)
                 (avg ? s->dsp.avg_tpel_pixels_tab : s->dsp.put_tpel_pixels_tab)[dxy](dest, src, s->uvlinesize, width, height);
             else
                 (avg ? s->dsp.avg_pixels_tab : s->dsp.put_pixels_tab)[blocksize][dxy](dest, src, s->uvlinesize, height);
         }
     }
8b82a956
 }
 
7f8205da
 static inline int svq3_mc_dir(H264Context *h, int size, int mode, int dir,
                               int avg)
 {
76de302d
     int i, j, k, mx, my, dx, dy, x, y;
     MpegEncContext *const s = (MpegEncContext *) h;
     const int part_width  = ((size & 5) == 4) ? 4 : 16 >> (size & 1);
     const int part_height = 16 >> ((unsigned) (size + 1) / 3);
     const int extra_width = (mode == PREDICT_MODE) ? -16*6 : 0;
     const int h_edge_pos  = 6*(s->h_edge_pos - part_width ) - extra_width;
     const int v_edge_pos  = 6*(s->v_edge_pos - part_height) - extra_width;
 
     for (i = 0; i < 16; i += part_height) {
         for (j = 0; j < 16; j += part_width) {
             const int b_xy = (4*s->mb_x + (j >> 2)) + (4*s->mb_y + (i >> 2))*h->b_stride;
             int dxy;
             x = 16*s->mb_x + j;
             y = 16*s->mb_y + i;
             k = ((j >> 2) & 1) + ((i >> 1) & 2) + ((j >> 1) & 4) + (i & 8);
 
             if (mode != PREDICT_MODE) {
                 pred_motion(h, k, (part_width >> 2), dir, 1, &mx, &my);
             } else {
                 mx = s->next_picture.motion_val[0][b_xy][0]<<1;
                 my = s->next_picture.motion_val[0][b_xy][1]<<1;
 
                 if (dir == 0) {
                     mx = ((mx * h->frame_num_offset) / h->prev_frame_num_offset + 1) >> 1;
                     my = ((my * h->frame_num_offset) / h->prev_frame_num_offset + 1) >> 1;
                 } else {
                     mx = ((mx * (h->frame_num_offset - h->prev_frame_num_offset)) / h->prev_frame_num_offset + 1) >> 1;
                     my = ((my * (h->frame_num_offset - h->prev_frame_num_offset)) / h->prev_frame_num_offset + 1) >> 1;
                 }
             }
 
             /* clip motion vector prediction to frame border */
             mx = av_clip(mx, extra_width - 6*x, h_edge_pos - 6*x);
             my = av_clip(my, extra_width - 6*y, v_edge_pos - 6*y);
 
             /* get (optional) motion vector differential */
             if (mode == PREDICT_MODE) {
                 dx = dy = 0;
             } else {
                 dy = svq3_get_se_golomb(&s->gb);
                 dx = svq3_get_se_golomb(&s->gb);
 
                 if (dx == INVALID_VLC || dy == INVALID_VLC) {
                     av_log(h->s.avctx, AV_LOG_ERROR, "invalid MV vlc\n");
                     return -1;
                 }
             }
 
             /* compute motion vector */
             if (mode == THIRDPEL_MODE) {
                 int fx, fy;
                 mx  = ((mx + 1)>>1) + dx;
                 my  = ((my + 1)>>1) + dy;
                 fx  = ((unsigned)(mx + 0x3000))/3 - 0x1000;
                 fy  = ((unsigned)(my + 0x3000))/3 - 0x1000;
                 dxy = (mx - 3*fx) + 4*(my - 3*fy);
 
                 svq3_mc_dir_part(s, x, y, part_width, part_height, fx, fy, dxy, 1, dir, avg);
                 mx += mx;
                 my += my;
             } else if (mode == HALFPEL_MODE || mode == PREDICT_MODE) {
                 mx  = ((unsigned)(mx + 1 + 0x3000))/3 + dx - 0x1000;
                 my  = ((unsigned)(my + 1 + 0x3000))/3 + dy - 0x1000;
                 dxy = (mx&1) + 2*(my&1);
 
                 svq3_mc_dir_part(s, x, y, part_width, part_height, mx>>1, my>>1, dxy, 0, dir, avg);
                 mx *= 3;
                 my *= 3;
             } else {
                 mx = ((unsigned)(mx + 3 + 0x6000))/6 + dx - 0x1000;
                 my = ((unsigned)(my + 3 + 0x6000))/6 + dy - 0x1000;
 
                 svq3_mc_dir_part(s, x, y, part_width, part_height, mx, my, 0, 0, dir, avg);
                 mx *= 6;
                 my *= 6;
             }
 
             /* update mv_cache */
             if (mode != PREDICT_MODE) {
                 int32_t mv = pack16to32(mx,my);
 
                 if (part_height == 8 && i < 8) {
                     *(int32_t *) h->mv_cache[dir][scan8[k] + 1*8] = mv;
 
                     if (part_width == 8 && j < 8) {
                         *(int32_t *) h->mv_cache[dir][scan8[k] + 1 + 1*8] = mv;
                     }
                 }
                 if (part_width == 8 && j < 8) {
                     *(int32_t *) h->mv_cache[dir][scan8[k] + 1] = mv;
                 }
                 if (part_width == 4 || part_height == 4) {
                     *(int32_t *) h->mv_cache[dir][scan8[k]] = mv;
                 }
             }
 
             /* write back motion vectors */
             fill_rectangle(s->current_picture.motion_val[dir][b_xy], part_width>>2, part_height>>2, h->b_stride, pack16to32(mx,my), 4);
bb270c08
         }
2e26c8d2
     }
 
76de302d
     return 0;
2e26c8d2
 }
 
8811679c
 static int svq3_decode_mb(SVQ3Context *svq3, unsigned int mb_type)
7f8205da
 {
8811679c
     H264Context *h = &svq3->h;
76de302d
     int i, j, k, m, dir, mode;
     int cbp = 0;
     uint32_t vlc;
     int8_t *top, *left;
     MpegEncContext *const s = (MpegEncContext *) h;
     const int mb_xy = h->mb_xy;
     const int b_xy  = 4*s->mb_x + 4*s->mb_y*h->b_stride;
 
     h->top_samples_available      = (s->mb_y == 0) ? 0x33FF : 0xFFFF;
     h->left_samples_available     = (s->mb_x == 0) ? 0x5F5F : 0xFFFF;
     h->topright_samples_available = 0xFFFF;
 
     if (mb_type == 0) {           /* SKIP */
ce5e49b0
         if (s->pict_type == AV_PICTURE_TYPE_P || s->next_picture.mb_type[mb_xy] == -1) {
76de302d
             svq3_mc_dir_part(s, 16*s->mb_x, 16*s->mb_y, 16, 16, 0, 0, 0, 0, 0, 0);
 
ce5e49b0
             if (s->pict_type == AV_PICTURE_TYPE_B) {
76de302d
                 svq3_mc_dir_part(s, 16*s->mb_x, 16*s->mb_y, 16, 16, 0, 0, 0, 0, 1, 1);
             }
 
             mb_type = MB_TYPE_SKIP;
         } else {
             mb_type = FFMIN(s->next_picture.mb_type[mb_xy], 6);
             if (svq3_mc_dir(h, mb_type, PREDICT_MODE, 0, 0) < 0)
                 return -1;
             if (svq3_mc_dir(h, mb_type, PREDICT_MODE, 1, 1) < 0)
                 return -1;
8b82a956
 
76de302d
             mb_type = MB_TYPE_16x16;
bb270c08
         }
76de302d
     } else if (mb_type < 8) {     /* INTER */
8811679c
         if (svq3->thirdpel_flag && svq3->halfpel_flag == !get_bits1 (&s->gb)) {
76de302d
             mode = THIRDPEL_MODE;
8811679c
         } else if (svq3->halfpel_flag && svq3->thirdpel_flag == !get_bits1 (&s->gb)) {
76de302d
             mode = HALFPEL_MODE;
         } else {
             mode = FULLPEL_MODE;
bb270c08
         }
8b82a956
 
76de302d
         /* fill caches */
         /* note ref_cache should contain here:
             ????????
             ???11111
             N??11111
             N??11111
             N??11111
         */
 
         for (m = 0; m < 2; m++) {
3b606e71
             if (s->mb_x > 0 && h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - 1]+6] != -1) {
76de302d
                 for (i = 0; i < 4; i++) {
                     *(uint32_t *) h->mv_cache[m][scan8[0] - 1 + i*8] = *(uint32_t *) s->current_picture.motion_val[m][b_xy - 1 + i*h->b_stride];
                 }
             } else {
                 for (i = 0; i < 4; i++) {
                     *(uint32_t *) h->mv_cache[m][scan8[0] - 1 + i*8] = 0;
                 }
             }
             if (s->mb_y > 0) {
                 memcpy(h->mv_cache[m][scan8[0] - 1*8], s->current_picture.motion_val[m][b_xy - h->b_stride], 4*2*sizeof(int16_t));
3b606e71
                 memset(&h->ref_cache[m][scan8[0] - 1*8], (h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride]] == -1) ? PART_NOT_AVAILABLE : 1, 4);
76de302d
 
                 if (s->mb_x < (s->mb_width - 1)) {
                     *(uint32_t *) h->mv_cache[m][scan8[0] + 4 - 1*8] = *(uint32_t *) s->current_picture.motion_val[m][b_xy - h->b_stride + 4];
                     h->ref_cache[m][scan8[0] + 4 - 1*8] =
3b606e71
                         (h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride + 1]+6] == -1 ||
                          h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride    ]  ] == -1) ? PART_NOT_AVAILABLE : 1;
76de302d
                 }else
                     h->ref_cache[m][scan8[0] + 4 - 1*8] = PART_NOT_AVAILABLE;
                 if (s->mb_x > 0) {
                     *(uint32_t *) h->mv_cache[m][scan8[0] - 1 - 1*8] = *(uint32_t *) s->current_picture.motion_val[m][b_xy - h->b_stride - 1];
3b606e71
                     h->ref_cache[m][scan8[0] - 1 - 1*8] = (h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride - 1]+3] == -1) ? PART_NOT_AVAILABLE : 1;
76de302d
                 }else
                     h->ref_cache[m][scan8[0] - 1 - 1*8] = PART_NOT_AVAILABLE;
             }else
                 memset(&h->ref_cache[m][scan8[0] - 1*8 - 1], PART_NOT_AVAILABLE, 8);
 
ce5e49b0
             if (s->pict_type != AV_PICTURE_TYPE_B)
76de302d
                 break;
bb270c08
         }
8b82a956
 
76de302d
         /* decode motion vector(s) and form prediction(s) */
ce5e49b0
         if (s->pict_type == AV_PICTURE_TYPE_P) {
76de302d
             if (svq3_mc_dir(h, (mb_type - 1), mode, 0, 0) < 0)
                 return -1;
ce5e49b0
         } else {        /* AV_PICTURE_TYPE_B */
76de302d
             if (mb_type != 2) {
                 if (svq3_mc_dir(h, 0, mode, 0, 0) < 0)
                     return -1;
             } else {
                 for (i = 0; i < 4; i++) {
                     memset(s->current_picture.motion_val[0][b_xy + i*h->b_stride], 0, 4*2*sizeof(int16_t));
                 }
             }
             if (mb_type != 1) {
                 if (svq3_mc_dir(h, 0, mode, 1, (mb_type == 3)) < 0)
                     return -1;
             } else {
                 for (i = 0; i < 4; i++) {
                     memset(s->current_picture.motion_val[1][b_xy + i*h->b_stride], 0, 4*2*sizeof(int16_t));
                 }
             }
bb270c08
         }
8b82a956
 
76de302d
         mb_type = MB_TYPE_16x16;
     } else if (mb_type == 8 || mb_type == 33) {   /* INTRA4x4 */
         memset(h->intra4x4_pred_mode_cache, -1, 8*5*sizeof(int8_t));
 
         if (mb_type == 8) {
             if (s->mb_x > 0) {
                 for (i = 0; i < 4; i++) {
3b606e71
                     h->intra4x4_pred_mode_cache[scan8[0] - 1 + i*8] = h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - 1]+6-i];
76de302d
                 }
                 if (h->intra4x4_pred_mode_cache[scan8[0] - 1] == -1) {
                     h->left_samples_available = 0x5F5F;
                 }
             }
             if (s->mb_y > 0) {
3b606e71
                 h->intra4x4_pred_mode_cache[4+8*0] = h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride]+0];
                 h->intra4x4_pred_mode_cache[5+8*0] = h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride]+1];
                 h->intra4x4_pred_mode_cache[6+8*0] = h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride]+2];
                 h->intra4x4_pred_mode_cache[7+8*0] = h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride]+3];
76de302d
 
                 if (h->intra4x4_pred_mode_cache[4+8*0] == -1) {
                     h->top_samples_available = 0x33FF;
                 }
             }
 
             /* decode prediction codes for luma blocks */
             for (i = 0; i < 16; i+=2) {
                 vlc = svq3_get_ue_golomb(&s->gb);
 
764ffdd0
                 if (vlc >= 25U){
76de302d
                     av_log(h->s.avctx, AV_LOG_ERROR, "luma prediction:%d\n", vlc);
                     return -1;
                 }
 
                 left    = &h->intra4x4_pred_mode_cache[scan8[i] - 1];
                 top     = &h->intra4x4_pred_mode_cache[scan8[i] - 8];
 
                 left[1] = svq3_pred_1[top[0] + 1][left[0] + 1][svq3_pred_0[vlc][0]];
                 left[2] = svq3_pred_1[top[1] + 1][left[1] + 1][svq3_pred_0[vlc][1]];
 
                 if (left[1] == -1 || left[2] == -1){
                     av_log(h->s.avctx, AV_LOG_ERROR, "weird prediction\n");
                     return -1;
                 }
             }
         } else {    /* mb_type == 33, DC_128_PRED block type */
             for (i = 0; i < 4; i++) {
                 memset(&h->intra4x4_pred_mode_cache[scan8[0] + 8*i], DC_PRED, 4);
             }
884182b3
         }
8b82a956
 
903d58f6
         ff_h264_write_back_intra_pred_mode(h);
8b82a956
 
76de302d
         if (mb_type == 8) {
2bedc0e8
             ff_h264_check_intra4x4_pred_mode(h);
da3b9756
 
76de302d
             h->top_samples_available  = (s->mb_y == 0) ? 0x33FF : 0xFFFF;
             h->left_samples_available = (s->mb_x == 0) ? 0x5F5F : 0xFFFF;
         } else {
             for (i = 0; i < 4; i++) {
                 memset(&h->intra4x4_pred_mode_cache[scan8[0] + 8*i], DC_128_PRED, 4);
             }
da3b9756
 
76de302d
             h->top_samples_available  = 0x33FF;
             h->left_samples_available = 0x5F5F;
         }
da3b9756
 
76de302d
         mb_type = MB_TYPE_INTRA4x4;
     } else {                      /* INTRA16x16 */
         dir = i_mb_type_info[mb_type - 8].pred_mode;
         dir = (dir >> 1) ^ 3*(dir & 1) ^ 1;
8b82a956
 
b89a0c9d
         if ((h->intra16x16_pred_mode = ff_h264_check_intra16x16_pred_mode(h, dir)) == -1){
76de302d
             av_log(h->s.avctx, AV_LOG_ERROR, "check_intra_pred_mode = -1\n");
             return -1;
         }
8b82a956
 
76de302d
         cbp = i_mb_type_info[mb_type - 8].cbp;
         mb_type = MB_TYPE_INTRA16x16;
884182b3
     }
8b82a956
 
ce5e49b0
     if (!IS_INTER(mb_type) && s->pict_type != AV_PICTURE_TYPE_I) {
76de302d
         for (i = 0; i < 4; i++) {
             memset(s->current_picture.motion_val[0][b_xy + i*h->b_stride], 0, 4*2*sizeof(int16_t));
         }
ce5e49b0
         if (s->pict_type == AV_PICTURE_TYPE_B) {
76de302d
             for (i = 0; i < 4; i++) {
                 memset(s->current_picture.motion_val[1][b_xy + i*h->b_stride], 0, 4*2*sizeof(int16_t));
             }
         }
8b82a956
     }
76de302d
     if (!IS_INTRA4x4(mb_type)) {
3b606e71
         memset(h->intra4x4_pred_mode+h->mb2br_xy[mb_xy], DC_PRED, 8);
da3b9756
     }
ce5e49b0
     if (!IS_SKIP(mb_type) || s->pict_type == AV_PICTURE_TYPE_B) {
11177a4d
         memset(h->non_zero_count_cache + 8, 0, 14*8*sizeof(uint8_t));
         s->dsp.clear_blocks(h->mb+  0);
         s->dsp.clear_blocks(h->mb+384);
884182b3
     }
2e26c8d2
 
ce5e49b0
     if (!IS_INTRA16x16(mb_type) && (!IS_SKIP(mb_type) || s->pict_type == AV_PICTURE_TYPE_B)) {
764ffdd0
         if ((vlc = svq3_get_ue_golomb(&s->gb)) >= 48U){
76de302d
             av_log(h->s.avctx, AV_LOG_ERROR, "cbp_vlc=%d\n", vlc);
             return -1;
         }
8b82a956
 
76de302d
         cbp = IS_INTRA(mb_type) ? golomb_to_intra4x4_cbp[vlc] : golomb_to_inter_cbp[vlc];
884182b3
     }
ce5e49b0
     if (IS_INTRA16x16(mb_type) || (s->pict_type != AV_PICTURE_TYPE_I && s->adaptive_quant && cbp)) {
76de302d
         s->qscale += svq3_get_se_golomb(&s->gb);
8b82a956
 
764ffdd0
         if (s->qscale > 31U){
76de302d
             av_log(h->s.avctx, AV_LOG_ERROR, "qscale:%d\n", s->qscale);
bb270c08
             return -1;
         }
8b82a956
     }
76de302d
     if (IS_INTRA16x16(mb_type)) {
11177a4d
         AV_ZERO128(h->mb_luma_dc[0]+0);
         AV_ZERO128(h->mb_luma_dc[0]+8);
290fabc6
         if (svq3_decode_block(&s->gb, h->mb_luma_dc, 0, 1)){
76de302d
             av_log(h->s.avctx, AV_LOG_ERROR, "error while decoding intra luma dc\n");
             return -1;
884182b3
         }
76de302d
     }
8b82a956
 
76de302d
     if (cbp) {
         const int index = IS_INTRA16x16(mb_type) ? 1 : 0;
         const int type = ((s->qscale < 24 && IS_INTRA4x4(mb_type)) ? 2 : 1);
 
         for (i = 0; i < 4; i++) {
             if ((cbp & (1 << i))) {
                 for (j = 0; j < 4; j++) {
                     k = index ? ((j&1) + 2*(i&1) + 2*(j&2) + 4*(i&2)) : (4*i + j);
                     h->non_zero_count_cache[ scan8[k] ] = 1;
 
                     if (svq3_decode_block(&s->gb, &h->mb[16*k], index, type)){
                         av_log(h->s.avctx, AV_LOG_ERROR, "error while decoding block\n");
                         return -1;
                     }
                 }
             }
         }
8b82a956
 
76de302d
         if ((cbp & 0x30)) {
11177a4d
             for (i = 1; i < 3; ++i) {
               if (svq3_decode_block(&s->gb, &h->mb[16*16*i], 0, 3)){
772225c0
                 av_log(h->s.avctx, AV_LOG_ERROR, "error while decoding chroma dc block\n");
                 return -1;
               }
76de302d
             }
 
             if ((cbp & 0x20)) {
11177a4d
                 for (i = 1; i < 3; i++) {
                     for (j = 0; j < 4; j++) {
                         k = 16*i + j;
                         h->non_zero_count_cache[ scan8[k] ] = 1;
 
                         if (svq3_decode_block(&s->gb, &h->mb[16*k], 1, 1)){
                             av_log(h->s.avctx, AV_LOG_ERROR, "error while decoding chroma ac block\n");
                             return -1;
                         }
76de302d
                     }
                 }
             }
bb270c08
         }
8b82a956
     }
 
2a5a9c28
     h->cbp= cbp;
76de302d
     s->current_picture.mb_type[mb_xy] = mb_type;
8b82a956
 
76de302d
     if (IS_INTRA(mb_type)) {
b89a0c9d
         h->chroma_pred_mode = ff_h264_check_intra_chroma_pred_mode(h, DC_PRED8x8);
76de302d
     }
8b82a956
 
76de302d
     return 0;
8b82a956
 }
 
8811679c
 static int svq3_decode_slice_header(AVCodecContext *avctx)
7f8205da
 {
8811679c
     SVQ3Context *svq3 = avctx->priv_data;
     H264Context *h = &svq3->h;
     MpegEncContext *s = &h->s;
76de302d
     const int mb_xy = h->mb_xy;
     int i, header;
da3b9756
 
76de302d
     header = get_bits(&s->gb, 8);
da3b9756
 
76de302d
     if (((header & 0x9F) != 1 && (header & 0x9F) != 2) || (header & 0x60) == 0) {
         /* TODO: what? */
8811679c
         av_log(avctx, AV_LOG_ERROR, "unsupported slice header (%02X)\n", header);
76de302d
         return -1;
     } else {
         int length = (header >> 5) & 3;
da3b9756
 
8811679c
         svq3->next_slice_index = get_bits_count(&s->gb) + 8*show_bits(&s->gb, 8*length) + 8*length;
da3b9756
 
8811679c
         if (svq3->next_slice_index > s->gb.size_in_bits) {
             av_log(avctx, AV_LOG_ERROR, "slice after bitstream end\n");
76de302d
             return -1;
303e50e6
     }
da3b9756
 
8811679c
         s->gb.size_in_bits = svq3->next_slice_index - 8*(length - 1);
76de302d
         skip_bits(&s->gb, 8);
da3b9756
 
8811679c
         if (svq3->watermark_key) {
76de302d
             uint32_t header = AV_RL32(&s->gb.buffer[(get_bits_count(&s->gb)>>3)+1]);
8811679c
             AV_WL32(&s->gb.buffer[(get_bits_count(&s->gb)>>3)+1], header ^ svq3->watermark_key);
76de302d
         }
         if (length > 0) {
             memcpy((uint8_t *) &s->gb.buffer[get_bits_count(&s->gb) >> 3],
                    &s->gb.buffer[s->gb.size_in_bits >> 3], (length - 1));
         }
db794291
         skip_bits_long(&s->gb, 0);
da3b9756
     }
 
764ffdd0
     if ((i = svq3_get_ue_golomb(&s->gb)) >= 3U){
76de302d
         av_log(h->s.avctx, AV_LOG_ERROR, "illegal slice type %d \n", i);
         return -1;
     }
da3b9756
 
76de302d
     h->slice_type = golomb_to_pict_type[i];
da3b9756
 
76de302d
     if ((header & 0x9F) == 2) {
         i = (s->mb_num < 64) ? 6 : (1 + av_log2 (s->mb_num - 1));
         s->mb_skip_run = get_bits(&s->gb, i) - (s->mb_x + (s->mb_y * s->mb_width));
     } else {
         skip_bits1(&s->gb);
         s->mb_skip_run = 0;
     }
da3b9756
 
76de302d
     h->slice_num = get_bits(&s->gb, 8);
     s->qscale = get_bits(&s->gb, 5);
     s->adaptive_quant = get_bits1(&s->gb);
da3b9756
 
76de302d
     /* unknown fields */
     skip_bits1(&s->gb);
da3b9756
 
8811679c
     if (svq3->unknown_flag) {
76de302d
         skip_bits1(&s->gb);
     }
da3b9756
 
76de302d
     skip_bits1(&s->gb);
     skip_bits(&s->gb, 2);
da3b9756
 
76de302d
     while (get_bits1(&s->gb)) {
         skip_bits(&s->gb, 8);
     }
da3b9756
 
76de302d
     /* reset intra predictors and invalidate motion vector references */
da3b9756
     if (s->mb_x > 0) {
3b606e71
         memset(h->intra4x4_pred_mode+h->mb2br_xy[mb_xy - 1      ]+3, -1, 4*sizeof(int8_t));
         memset(h->intra4x4_pred_mode+h->mb2br_xy[mb_xy - s->mb_x]  , -1, 8*sizeof(int8_t)*s->mb_x);
76de302d
     }
     if (s->mb_y > 0) {
3b606e71
         memset(h->intra4x4_pred_mode+h->mb2br_xy[mb_xy - s->mb_stride], -1, 8*sizeof(int8_t)*(s->mb_width - s->mb_x));
76de302d
 
         if (s->mb_x > 0) {
3b606e71
             h->intra4x4_pred_mode[h->mb2br_xy[mb_xy - s->mb_stride - 1]+3] = -1;
76de302d
         }
da3b9756
     }
 
76de302d
     return 0;
da3b9756
 }
 
5ef251e5
 static av_cold int svq3_decode_init(AVCodecContext *avctx)
7f8205da
 {
8811679c
     SVQ3Context *svq3 = avctx->priv_data;
     H264Context *h = &svq3->h;
     MpegEncContext *s = &h->s;
f4cca718
     int m;
76de302d
     unsigned char *extradata;
     unsigned int size;
 
903d58f6
     if (ff_h264_decode_init(avctx) < 0)
f4cca718
         return -1;
 
76de302d
     s->flags  = avctx->flags;
     s->flags2 = avctx->flags2;
     s->unrestricted_mv = 1;
54dab661
     h->is_complex=1;
94621d32
     avctx->pix_fmt = avctx->codec->pix_fmts[0];
76de302d
 
     if (!s->context_initialized) {
8811679c
         h->chroma_qp[0] = h->chroma_qp[1] = 4;
 
8dfc6d1f
         svq3->halfpel_flag  = 1;
8811679c
         svq3->thirdpel_flag = 1;
8dfc6d1f
         svq3->unknown_flag  = 0;
76de302d
 
 
         /* prowl for the "SEQH" marker in the extradata */
         extradata = (unsigned char *)avctx->extradata;
         for (m = 0; m < avctx->extradata_size; m++) {
             if (!memcmp(extradata, "SEQH", 4))
                 break;
             extradata++;
         }
8b82a956
 
76de302d
         /* if a match was found, parse the extra data */
         if (extradata && !memcmp(extradata, "SEQH", 4)) {
8b82a956
 
76de302d
             GetBitContext gb;
faccfeec
             int frame_size_code;
8b82a956
 
76de302d
             size = AV_RB32(&extradata[4]);
             init_get_bits(&gb, extradata + 8, size*8);
8b82a956
 
76de302d
             /* 'frame size code' and optional 'width, height' */
faccfeec
             frame_size_code = get_bits(&gb, 3);
             switch (frame_size_code) {
                 case 0: avctx->width = 160; avctx->height = 120; break;
                 case 1: avctx->width = 128; avctx->height =  96; break;
                 case 2: avctx->width = 176; avctx->height = 144; break;
                 case 3: avctx->width = 352; avctx->height = 288; break;
                 case 4: avctx->width = 704; avctx->height = 576; break;
                 case 5: avctx->width = 240; avctx->height = 180; break;
                 case 6: avctx->width = 320; avctx->height = 240; break;
                 case 7:
                     avctx->width  = get_bits(&gb, 12);
                     avctx->height = get_bits(&gb, 12);
                     break;
76de302d
             }
8b82a956
 
8811679c
             svq3->halfpel_flag  = get_bits1(&gb);
             svq3->thirdpel_flag = get_bits1(&gb);
8b82a956
 
76de302d
             /* unknown fields */
             skip_bits1(&gb);
             skip_bits1(&gb);
             skip_bits1(&gb);
             skip_bits1(&gb);
8b82a956
 
76de302d
             s->low_delay = get_bits1(&gb);
1e002b60
 
76de302d
             /* unknown field */
             skip_bits1(&gb);
 
             while (get_bits1(&gb)) {
                 skip_bits(&gb, 8);
             }
 
8811679c
             svq3->unknown_flag = get_bits1(&gb);
76de302d
             avctx->has_b_frames = !s->low_delay;
8811679c
             if (svq3->unknown_flag) {
b250f9c6
 #if CONFIG_ZLIB
76de302d
                 unsigned watermark_width  = svq3_get_ue_golomb(&gb);
                 unsigned watermark_height = svq3_get_ue_golomb(&gb);
                 int u1 = svq3_get_ue_golomb(&gb);
                 int u2 = get_bits(&gb, 8);
                 int u3 = get_bits(&gb, 2);
                 int u4 = svq3_get_ue_golomb(&gb);
891263ef
                 unsigned long buf_len = watermark_width*watermark_height*4;
76de302d
                 int offset = (get_bits_count(&gb)+7)>>3;
                 uint8_t *buf;
 
                 if ((uint64_t)watermark_width*4 > UINT_MAX/watermark_height)
                     return -1;
 
                 buf = av_malloc(buf_len);
                 av_log(avctx, AV_LOG_DEBUG, "watermark size: %dx%d\n", watermark_width, watermark_height);
                 av_log(avctx, AV_LOG_DEBUG, "u1: %x u2: %x u3: %x compressed data size: %d offset: %d\n", u1, u2, u3, u4, offset);
891263ef
                 if (uncompress(buf, &buf_len, extradata + 8 + offset, size - offset) != Z_OK) {
76de302d
                     av_log(avctx, AV_LOG_ERROR, "could not uncompress watermark logo\n");
                     av_free(buf);
                     return -1;
                 }
8811679c
                 svq3->watermark_key = ff_svq1_packet_checksum(buf, buf_len, 0);
                 svq3->watermark_key = svq3->watermark_key << 16 | svq3->watermark_key;
                 av_log(avctx, AV_LOG_DEBUG, "watermark key %#x\n", svq3->watermark_key);
76de302d
                 av_free(buf);
1e002b60
 #else
76de302d
                 av_log(avctx, AV_LOG_ERROR, "this svq3 file contains watermark which need zlib support compiled in\n");
                 return -1;
1e002b60
 #endif
76de302d
             }
         }
cdca7c37
 
         s->width  = avctx->width;
         s->height = avctx->height;
 
         if (MPV_common_init(s) < 0)
             return -1;
 
         h->b_stride = 4*s->mb_width;
 
         ff_h264_alloc_tables(h);
da3b9756
     }
76de302d
 
f4cca718
     return 0;
 }
 
 static int svq3_decode_frame(AVCodecContext *avctx,
                              void *data, int *data_size,
7a00bbad
                              AVPacket *avpkt)
f4cca718
 {
8811679c
     SVQ3Context *svq3 = avctx->priv_data;
     H264Context *h = &svq3->h;
     MpegEncContext *s = &h->s;
7a00bbad
     int buf_size = avpkt->size;
b0e7a932
     int m, mb_type, left;
2264c110
     uint8_t *buf;
f4cca718
 
76de302d
     /* special case for last picture */
     if (buf_size == 0) {
         if (s->next_picture_ptr && !s->low_delay) {
             *(AVFrame *) data = *(AVFrame *) &s->next_picture;
             s->next_picture_ptr = NULL;
             *data_size = sizeof(AVFrame);
         }
         return 0;
da3b9756
     }
8b82a956
 
76de302d
     s->mb_x = s->mb_y = h->mb_xy = 0;
8b82a956
 
2264c110
     if (svq3->watermark_key) {
f44967b1
         av_fast_malloc(&svq3->buf, &svq3->buf_size,
                        buf_size+FF_INPUT_BUFFER_PADDING_SIZE);
2264c110
         if (!svq3->buf)
             return AVERROR(ENOMEM);
         memcpy(svq3->buf, avpkt->data, buf_size);
         buf = svq3->buf;
     } else {
         buf = avpkt->data;
     }
 
     init_get_bits(&s->gb, buf, 8*buf_size);
 
8811679c
     if (svq3_decode_slice_header(avctx))
76de302d
         return -1;
8b82a956
 
76de302d
     s->pict_type = h->slice_type;
     s->picture_number = h->slice_num;
8b82a956
 
76de302d
     if (avctx->debug&FF_DEBUG_PICT_INFO){
         av_log(h->s.avctx, AV_LOG_DEBUG, "%c hpel:%d, tpel:%d aqp:%d qp:%d, slice_num:%02X\n",
8811679c
                av_get_picture_type_char(s->pict_type), svq3->halfpel_flag, svq3->thirdpel_flag,
76de302d
                s->adaptive_quant, s->qscale, h->slice_num);
     }
8b82a956
 
8ed2ae09
     /* for skipping the frame */
76de302d
     s->current_picture.pict_type = s->pict_type;
ce5e49b0
     s->current_picture.key_frame = (s->pict_type == AV_PICTURE_TYPE_I);
76de302d
 
     /* Skip B-frames if we do not have reference frames. */
ce5e49b0
     if (s->last_picture_ptr == NULL && s->pict_type == AV_PICTURE_TYPE_B)
76de302d
         return 0;
ce5e49b0
     if (  (avctx->skip_frame >= AVDISCARD_NONREF && s->pict_type == AV_PICTURE_TYPE_B)
         ||(avctx->skip_frame >= AVDISCARD_NONKEY && s->pict_type != AV_PICTURE_TYPE_I)
76de302d
         || avctx->skip_frame >= AVDISCARD_ALL)
         return 0;
 
     if (s->next_p_frame_damaged) {
ce5e49b0
         if (s->pict_type == AV_PICTURE_TYPE_B)
76de302d
             return 0;
         else
             s->next_p_frame_damaged = 0;
     }
da3b9756
 
903d58f6
     if (ff_h264_frame_start(h) < 0)
76de302d
         return -1;
8b82a956
 
ce5e49b0
     if (s->pict_type == AV_PICTURE_TYPE_B) {
76de302d
         h->frame_num_offset = (h->slice_num - h->prev_frame_num);
8b82a956
 
76de302d
         if (h->frame_num_offset < 0) {
             h->frame_num_offset += 256;
         }
         if (h->frame_num_offset == 0 || h->frame_num_offset >= h->prev_frame_num_offset) {
             av_log(h->s.avctx, AV_LOG_ERROR, "error in B-frame picture id\n");
             return -1;
         }
     } else {
         h->prev_frame_num = h->frame_num;
         h->frame_num = h->slice_num;
         h->prev_frame_num_offset = (h->frame_num - h->prev_frame_num);
da3b9756
 
76de302d
         if (h->prev_frame_num_offset < 0) {
             h->prev_frame_num_offset += 256;
         }
da3b9756
     }
 
76de302d
     for (m = 0; m < 2; m++){
         int i;
         for (i = 0; i < 4; i++){
             int j;
             for (j = -1; j < 4; j++)
                 h->ref_cache[m][scan8[0] + 8*i + j]= 1;
             if (i < 3)
                 h->ref_cache[m][scan8[0] + 8*i + j]= PART_NOT_AVAILABLE;
         }
da3b9756
     }
 
76de302d
     for (s->mb_y = 0; s->mb_y < s->mb_height; s->mb_y++) {
         for (s->mb_x = 0; s->mb_x < s->mb_width; s->mb_x++) {
             h->mb_xy = s->mb_x + s->mb_y*s->mb_stride;
115329f1
 
76de302d
             if ( (get_bits_count(&s->gb) + 7) >= s->gb.size_in_bits &&
                 ((get_bits_count(&s->gb) & 7) == 0 || show_bits(&s->gb, (-get_bits_count(&s->gb) & 7)) == 0)) {
da3b9756
 
8811679c
                 skip_bits(&s->gb, svq3->next_slice_index - get_bits_count(&s->gb));
76de302d
                 s->gb.size_in_bits = 8*buf_size;
da3b9756
 
8811679c
                 if (svq3_decode_slice_header(avctx))
76de302d
                     return -1;
da3b9756
 
76de302d
                 /* TODO: support s->mb_skip_run */
             }
da3b9756
 
76de302d
             mb_type = svq3_get_ue_golomb(&s->gb);
da3b9756
 
ce5e49b0
             if (s->pict_type == AV_PICTURE_TYPE_I) {
76de302d
                 mb_type += 8;
ce5e49b0
             } else if (s->pict_type == AV_PICTURE_TYPE_B && mb_type >= 4) {
76de302d
                 mb_type += 4;
             }
0fa8d199
             if ((unsigned)mb_type > 33 || svq3_decode_mb(svq3, mb_type)) {
76de302d
                 av_log(h->s.avctx, AV_LOG_ERROR, "error while decoding MB %d %d\n", s->mb_x, s->mb_y);
                 return -1;
             }
8b82a956
 
76de302d
             if (mb_type != 0) {
903d58f6
                 ff_h264_hl_decode_mb (h);
76de302d
             }
8b82a956
 
ce5e49b0
             if (s->pict_type != AV_PICTURE_TYPE_B && !s->low_delay) {
76de302d
                 s->current_picture.mb_type[s->mb_x + s->mb_y*s->mb_stride] =
ce5e49b0
                     (s->pict_type == AV_PICTURE_TYPE_P && mb_type < 8) ? (mb_type - 1) : -1;
76de302d
             }
         }
2e26c8d2
 
76de302d
         ff_draw_horiz_band(s, 16*s->mb_y, 16);
8b82a956
     }
4c701ac8
 
b0e7a932
     left = buf_size*8 - get_bits_count(&s->gb);
 
     if (s->mb_y != s->mb_height || s->mb_x != s->mb_width) {
         av_log(avctx, AV_LOG_INFO, "frame num %d incomplete pic x %d y %d left %d\n", avctx->frame_number, s->mb_y, s->mb_x, left);
         //av_hex_dump(stderr, buf+buf_size-8, 8);
     }
 
     if (left < 0) {
         av_log(avctx, AV_LOG_ERROR, "frame num %d left %d\n", avctx->frame_number, left);
         return -1;
     }
 
76de302d
     MPV_frame_end(s);
8b82a956
 
ce5e49b0
     if (s->pict_type == AV_PICTURE_TYPE_B || s->low_delay) {
76de302d
         *(AVFrame *) data = *(AVFrame *) &s->current_picture;
     } else {
         *(AVFrame *) data = *(AVFrame *) &s->last_picture;
     }
da3b9756
 
76de302d
     /* Do not output the last pic after seeking. */
     if (s->last_picture_ptr || s->low_delay) {
         *data_size = sizeof(AVFrame);
     }
da3b9756
 
76de302d
     return buf_size;
8b82a956
 }
 
8811679c
 static int svq3_decode_end(AVCodecContext *avctx)
 {
     SVQ3Context *svq3 = avctx->priv_data;
     H264Context *h = &svq3->h;
     MpegEncContext *s = &h->s;
 
     ff_h264_free_context(h);
 
     MPV_common_end(s);
 
2264c110
     av_freep(&svq3->buf);
     svq3->buf_size = 0;
 
8811679c
     return 0;
 }
8b82a956
 
e7e2df27
 AVCodec ff_svq3_decoder = {
8b82a956
     "svq3",
72415b2a
     AVMEDIA_TYPE_VIDEO,
8b82a956
     CODEC_ID_SVQ3,
8811679c
     sizeof(SVQ3Context),
f4cca718
     svq3_decode_init,
8b82a956
     NULL,
8811679c
     svq3_decode_end,
8b82a956
     svq3_decode_frame,
934982c4
     CODEC_CAP_DRAW_HORIZ_BAND | CODEC_CAP_DR1 | CODEC_CAP_DELAY,
2917367c
     .long_name = NULL_IF_CONFIG_SMALL("Sorenson Vector Quantizer 3 / Sorenson Video 3 / SVQ3"),
2ba83017
     .pix_fmts= (const enum PixelFormat[]){PIX_FMT_YUVJ420P, PIX_FMT_NONE},
8b82a956
 };