svq1.c 43.1 KB
Newer Older
1
/*
2
 *
Nick Kurshev's avatar
Nick Kurshev committed
3
 * Copyright (C) 2002 the xine project
4
 * Copyright (C) 2002 the ffmpeg project
5
 *
6 7 8 9 10 11 12 13 14 15 16 17 18 19
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * 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
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
20
 * (SVQ1 Decoder)
21
 * Ported to mplayer by Arpi <arpi@thot.banki.hu>
Nick Kurshev's avatar
Nick Kurshev committed
22 23
 * Ported to libavcodec by Nick Kurshev <nickols_k@mail.ru>
 *
24
 * SVQ1 Encoder (c) 2004 Mike Melanson <melanson@pcisys.net>
Nick Kurshev's avatar
Nick Kurshev committed
25
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
26 27 28

/**
 * @file svq1.c
29 30 31
 * Sorenson Vector Quantizer #1 (SVQ1) video codec.
 * For more information of the SVQ1 algorithm, visit:
 *   http://www.pcisys.net/~melanson/codecs/
Michael Niedermayer's avatar
Michael Niedermayer committed
32 33
 */

34

Nick Kurshev's avatar
Nick Kurshev committed
35 36 37 38 39
//#define DEBUG_SVQ1
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
40
#include <limits.h>
Nick Kurshev's avatar
Nick Kurshev committed
41

42
#include "common.h"
Nick Kurshev's avatar
Nick Kurshev committed
43 44 45
#include "avcodec.h"
#include "dsputil.h"
#include "mpegvideo.h"
46
#include "bswap.h"
Nick Kurshev's avatar
Nick Kurshev committed
47

48 49 50
#undef NDEBUG
#include <assert.h>

51 52
extern const uint8_t mvtab[33][2];

53 54 55 56 57 58
static VLC svq1_block_type;
static VLC svq1_motion_component;
static VLC svq1_intra_multistage[6];
static VLC svq1_inter_multistage[6];
static VLC svq1_intra_mean;
static VLC svq1_inter_mean;
Nick Kurshev's avatar
Nick Kurshev committed
59

60 61 62 63
#define SVQ1_BLOCK_SKIP         0
#define SVQ1_BLOCK_INTER        1
#define SVQ1_BLOCK_INTER_4V     2
#define SVQ1_BLOCK_INTRA        3
Nick Kurshev's avatar
Nick Kurshev committed
64

65
typedef struct SVQ1Context {
66
    MpegEncContext m; // needed for motion estimation, should not be used for anything else, the idea is to make the motion estimation eventually independant of MpegEncContext, so this will be removed then (FIXME/XXX)
67 68 69
    AVCodecContext *avctx;
    DSPContext dsp;
    AVFrame picture;
70 71
    AVFrame current_picture;
    AVFrame last_picture;
72 73
    PutBitContext pb;
    GetBitContext gb;
74

75
    PutBitContext reorder_pb[6]; //why ooh why this sick breadth first order, everything is slower and more complex
76 77 78 79 80 81 82 83 84 85 86

    int frame_width;
    int frame_height;

    /* Y plane block dimensions */
    int y_block_width;
    int y_block_height;

    /* U & V plane (C planes) block dimensions */
    int c_block_width;
    int c_block_height;
87

88 89 90 91
    uint16_t *mb_type;
    uint32_t *dummy;
    int16_t (*motion_val8[3])[2];
    int16_t (*motion_val16[3])[2];
92

93
    int64_t rd_total;
94 95
} SVQ1Context;

Nick Kurshev's avatar
Nick Kurshev committed
96 97
/* motion vector (prediction) */
typedef struct svq1_pmv_s {
98 99
  int           x;
  int           y;
Nick Kurshev's avatar
Nick Kurshev committed
100 101
} svq1_pmv_t;

102
#include "svq1_cb.h"
103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
#include "svq1_vlc.h"

static const uint16_t checksum_table[256] = {
  0x0000, 0x1021, 0x2042, 0x3063, 0x4084, 0x50A5, 0x60C6, 0x70E7,
  0x8108, 0x9129, 0xA14A, 0xB16B, 0xC18C, 0xD1AD, 0xE1CE, 0xF1EF,
  0x1231, 0x0210, 0x3273, 0x2252, 0x52B5, 0x4294, 0x72F7, 0x62D6,
  0x9339, 0x8318, 0xB37B, 0xA35A, 0xD3BD, 0xC39C, 0xF3FF, 0xE3DE,
  0x2462, 0x3443, 0x0420, 0x1401, 0x64E6, 0x74C7, 0x44A4, 0x5485,
  0xA56A, 0xB54B, 0x8528, 0x9509, 0xE5EE, 0xF5CF, 0xC5AC, 0xD58D,
  0x3653, 0x2672, 0x1611, 0x0630, 0x76D7, 0x66F6, 0x5695, 0x46B4,
  0xB75B, 0xA77A, 0x9719, 0x8738, 0xF7DF, 0xE7FE, 0xD79D, 0xC7BC,
  0x48C4, 0x58E5, 0x6886, 0x78A7, 0x0840, 0x1861, 0x2802, 0x3823,
  0xC9CC, 0xD9ED, 0xE98E, 0xF9AF, 0x8948, 0x9969, 0xA90A, 0xB92B,
  0x5AF5, 0x4AD4, 0x7AB7, 0x6A96, 0x1A71, 0x0A50, 0x3A33, 0x2A12,
  0xDBFD, 0xCBDC, 0xFBBF, 0xEB9E, 0x9B79, 0x8B58, 0xBB3B, 0xAB1A,
  0x6CA6, 0x7C87, 0x4CE4, 0x5CC5, 0x2C22, 0x3C03, 0x0C60, 0x1C41,
  0xEDAE, 0xFD8F, 0xCDEC, 0xDDCD, 0xAD2A, 0xBD0B, 0x8D68, 0x9D49,
  0x7E97, 0x6EB6, 0x5ED5, 0x4EF4, 0x3E13, 0x2E32, 0x1E51, 0x0E70,
  0xFF9F, 0xEFBE, 0xDFDD, 0xCFFC, 0xBF1B, 0xAF3A, 0x9F59, 0x8F78,
  0x9188, 0x81A9, 0xB1CA, 0xA1EB, 0xD10C, 0xC12D, 0xF14E, 0xE16F,
  0x1080, 0x00A1, 0x30C2, 0x20E3, 0x5004, 0x4025, 0x7046, 0x6067,
  0x83B9, 0x9398, 0xA3FB, 0xB3DA, 0xC33D, 0xD31C, 0xE37F, 0xF35E,
  0x02B1, 0x1290, 0x22F3, 0x32D2, 0x4235, 0x5214, 0x6277, 0x7256,
  0xB5EA, 0xA5CB, 0x95A8, 0x8589, 0xF56E, 0xE54F, 0xD52C, 0xC50D,
  0x34E2, 0x24C3, 0x14A0, 0x0481, 0x7466, 0x6447, 0x5424, 0x4405,
  0xA7DB, 0xB7FA, 0x8799, 0x97B8, 0xE75F, 0xF77E, 0xC71D, 0xD73C,
  0x26D3, 0x36F2, 0x0691, 0x16B0, 0x6657, 0x7676, 0x4615, 0x5634,
  0xD94C, 0xC96D, 0xF90E, 0xE92F, 0x99C8, 0x89E9, 0xB98A, 0xA9AB,
  0x5844, 0x4865, 0x7806, 0x6827, 0x18C0, 0x08E1, 0x3882, 0x28A3,
  0xCB7D, 0xDB5C, 0xEB3F, 0xFB1E, 0x8BF9, 0x9BD8, 0xABBB, 0xBB9A,
  0x4A75, 0x5A54, 0x6A37, 0x7A16, 0x0AF1, 0x1AD0, 0x2AB3, 0x3A92,
  0xFD2E, 0xED0F, 0xDD6C, 0xCD4D, 0xBDAA, 0xAD8B, 0x9DE8, 0x8DC9,
  0x7C26, 0x6C07, 0x5C64, 0x4C45, 0x3CA2, 0x2C83, 0x1CE0, 0x0CC1,
  0xEF1F, 0xFF3E, 0xCF5D, 0xDF7C, 0xAF9B, 0xBFBA, 0x8FD9, 0x9FF8,
  0x6E17, 0x7E36, 0x4E55, 0x5E74, 0x2E93, 0x3EB2, 0x0ED1, 0x1EF0
Nick Kurshev's avatar
Nick Kurshev committed
138 139
};

140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172
static const uint8_t string_table[256] = {
  0x00, 0xD5, 0x7F, 0xAA, 0xFE, 0x2B, 0x81, 0x54,
  0x29, 0xFC, 0x56, 0x83, 0xD7, 0x02, 0xA8, 0x7D,
  0x52, 0x87, 0x2D, 0xF8, 0xAC, 0x79, 0xD3, 0x06,
  0x7B, 0xAE, 0x04, 0xD1, 0x85, 0x50, 0xFA, 0x2F,
  0xA4, 0x71, 0xDB, 0x0E, 0x5A, 0x8F, 0x25, 0xF0,
  0x8D, 0x58, 0xF2, 0x27, 0x73, 0xA6, 0x0C, 0xD9,
  0xF6, 0x23, 0x89, 0x5C, 0x08, 0xDD, 0x77, 0xA2,
  0xDF, 0x0A, 0xA0, 0x75, 0x21, 0xF4, 0x5E, 0x8B,
  0x9D, 0x48, 0xE2, 0x37, 0x63, 0xB6, 0x1C, 0xC9,
  0xB4, 0x61, 0xCB, 0x1E, 0x4A, 0x9F, 0x35, 0xE0,
  0xCF, 0x1A, 0xB0, 0x65, 0x31, 0xE4, 0x4E, 0x9B,
  0xE6, 0x33, 0x99, 0x4C, 0x18, 0xCD, 0x67, 0xB2,
  0x39, 0xEC, 0x46, 0x93, 0xC7, 0x12, 0xB8, 0x6D,
  0x10, 0xC5, 0x6F, 0xBA, 0xEE, 0x3B, 0x91, 0x44,
  0x6B, 0xBE, 0x14, 0xC1, 0x95, 0x40, 0xEA, 0x3F,
  0x42, 0x97, 0x3D, 0xE8, 0xBC, 0x69, 0xC3, 0x16,
  0xEF, 0x3A, 0x90, 0x45, 0x11, 0xC4, 0x6E, 0xBB,
  0xC6, 0x13, 0xB9, 0x6C, 0x38, 0xED, 0x47, 0x92,
  0xBD, 0x68, 0xC2, 0x17, 0x43, 0x96, 0x3C, 0xE9,
  0x94, 0x41, 0xEB, 0x3E, 0x6A, 0xBF, 0x15, 0xC0,
  0x4B, 0x9E, 0x34, 0xE1, 0xB5, 0x60, 0xCA, 0x1F,
  0x62, 0xB7, 0x1D, 0xC8, 0x9C, 0x49, 0xE3, 0x36,
  0x19, 0xCC, 0x66, 0xB3, 0xE7, 0x32, 0x98, 0x4D,
  0x30, 0xE5, 0x4F, 0x9A, 0xCE, 0x1B, 0xB1, 0x64,
  0x72, 0xA7, 0x0D, 0xD8, 0x8C, 0x59, 0xF3, 0x26,
  0x5B, 0x8E, 0x24, 0xF1, 0xA5, 0x70, 0xDA, 0x0F,
  0x20, 0xF5, 0x5F, 0x8A, 0xDE, 0x0B, 0xA1, 0x74,
  0x09, 0xDC, 0x76, 0xA3, 0xF7, 0x22, 0x88, 0x5D,
  0xD6, 0x03, 0xA9, 0x7C, 0x28, 0xFD, 0x57, 0x82,
  0xFF, 0x2A, 0x80, 0x55, 0x01, 0xD4, 0x7E, 0xAB,
  0x84, 0x51, 0xFB, 0x2E, 0x7A, 0xAF, 0x05, 0xD0,
  0xAD, 0x78, 0xD2, 0x07, 0x53, 0x86, 0x2C, 0xF9
Nick Kurshev's avatar
Nick Kurshev committed
173 174
};

175 176 177 178
#define SVQ1_PROCESS_VECTOR()\
    for (; level > 0; i++) {\
      /* process next depth */\
      if (i == m) {\
179 180 181
        m = n;\
        if (--level == 0)\
          break;\
182 183 184
      }\
      /* divide block if next bit set */\
      if (get_bits (bitbuf, 1) == 0)\
185
        break;\
186 187 188 189 190 191
      /* add child nodes */\
      list[n++] = list[i];\
      list[n++] = list[i] + (((level & 1) ? pitch : 1) << ((level / 2) + 1));\
    }

#define SVQ1_ADD_CODEBOOK()\
192 193 194 195 196 197
          /* add codebook entries to vector */\
          for (j=0; j < stages; j++) {\
            n3  = codebook[entries[j]] ^ 0x80808080;\
            n1 += ((n3 & 0xFF00FF00) >> 8);\
            n2 +=  (n3 & 0x00FF00FF);\
          }\
198
\
199 200 201 202 203 204 205
          /* clip to [0..255] */\
          if (n1 & 0xFF00FF00) {\
            n3  = ((( n1 >> 15) & 0x00010001) | 0x01000100) - 0x00010001;\
            n1 += 0x7F007F00;\
            n1 |= (((~n1 >> 15) & 0x00010001) | 0x01000100) - 0x00010001;\
            n1 &= (n3 & 0x00FF00FF);\
          }\
206
\
207 208 209 210 211 212
          if (n2 & 0xFF00FF00) {\
            n3  = ((( n2 >> 15) & 0x00010001) | 0x01000100) - 0x00010001;\
            n2 += 0x7F007F00;\
            n2 |= (((~n2 >> 15) & 0x00010001) | 0x01000100) - 0x00010001;\
            n2 &= (n3 & 0x00FF00FF);\
          }
Nick Kurshev's avatar
Nick Kurshev committed
213

214 215
#define SVQ1_DO_CODEBOOK_INTRA()\
      for (y=0; y < height; y++) {\
216 217 218 219 220 221 222 223
        for (x=0; x < (width / 4); x++, codebook++) {\
        n1 = n4;\
        n2 = n4;\
        SVQ1_ADD_CODEBOOK()\
        /* store result */\
        dst[x] = (n1 << 8) | n2;\
        }\
        dst += (pitch / 4);\
224 225 226 227
      }

#define SVQ1_DO_CODEBOOK_NONINTRA()\
      for (y=0; y < height; y++) {\
228 229 230 231 232 233 234 235 236 237
        for (x=0; x < (width / 4); x++, codebook++) {\
        n3 = dst[x];\
        /* add mean value to vector */\
        n1 = ((n3 & 0xFF00FF00) >> 8) + n4;\
        n2 =  (n3 & 0x00FF00FF)          + n4;\
        SVQ1_ADD_CODEBOOK()\
        /* store result */\
        dst[x] = (n1 << 8) | n2;\
        }\
        dst += (pitch / 4);\
238 239 240
      }

#define SVQ1_CALC_CODEBOOK_ENTRIES(cbook)\
241
      codebook = (const uint32_t *) cbook[level];\
242 243 244
      bit_cache = get_bits (bitbuf, 4*stages);\
      /* calculate codebook entries for this vector */\
      for (j=0; j < stages; j++) {\
245
        entries[j] = (((bit_cache >> (4*(stages - j - 1))) & 0xF) + 16*j) << (level + 1);\
246 247 248
      }\
      mean -= (stages * 128);\
      n4    = ((mean + (mean >> 31)) << 16) | (mean & 0xFFFF);
Nick Kurshev's avatar
Nick Kurshev committed
249

250
static int svq1_decode_block_intra (GetBitContext *bitbuf, uint8_t *pixels, int pitch ) {
Nick Kurshev's avatar
Nick Kurshev committed
251 252 253
  uint32_t    bit_cache;
  uint8_t    *list[63];
  uint32_t   *dst;
254
  const uint32_t *codebook;
255 256 257
  int         entries[6];
  int         i, j, m, n;
  int         mean, stages;
258
  unsigned    x, y, width, height, level;
Nick Kurshev's avatar
Nick Kurshev committed
259 260 261 262 263 264 265
  uint32_t    n1, n2, n3, n4;

  /* initialize list for breadth first processing of vectors */
  list[0] = pixels;

  /* recursively process vector */
  for (i=0, m=1, n=1, level=5; i < n; i++) {
266
    SVQ1_PROCESS_VECTOR();
Nick Kurshev's avatar
Nick Kurshev committed
267 268 269 270 271 272 273

    /* destination address and vector size */
    dst = (uint32_t *) list[i];
    width = 1 << ((4 + level) /2);
    height = 1 << ((3 + level) /2);

    /* get number of stages (-1 skips vector, 0 for mean only) */
274
    stages = get_vlc2(bitbuf, svq1_intra_multistage[level].table, 3, 3) - 1;
Nick Kurshev's avatar
Nick Kurshev committed
275 276

    if (stages == -1) {
277 278 279 280
        for (y=0; y < height; y++) {
          memset (&dst[y*(pitch / 4)], 0, width);
        }
      continue;                 /* skip vector */
Nick Kurshev's avatar
Nick Kurshev committed
281 282 283 284
    }

    if ((stages > 0) && (level >= 4)) {
#ifdef DEBUG_SVQ1
285
    av_log(s->avctx, AV_LOG_INFO, "Error (svq1_decode_block_intra): invalid vector: stages=%i level=%i\n",stages,level);
Nick Kurshev's avatar
Nick Kurshev committed
286
#endif
287
      return -1;        /* invalid vector */
Nick Kurshev's avatar
Nick Kurshev committed
288 289
    }

290
    mean = get_vlc2(bitbuf, svq1_intra_mean.table, 8, 3);
Nick Kurshev's avatar
Nick Kurshev committed
291

292
    if (stages == 0) {
Nick Kurshev's avatar
Nick Kurshev committed
293
      for (y=0; y < height; y++) {
294
        memset (&dst[y*(pitch / 4)], mean, width);
Nick Kurshev's avatar
Nick Kurshev committed
295 296
      }
    } else {
297 298 299 300
      SVQ1_CALC_CODEBOOK_ENTRIES(svq1_intra_codebooks);
      SVQ1_DO_CODEBOOK_INTRA()
    }
  }
Nick Kurshev's avatar
Nick Kurshev committed
301

302 303
  return 0;
}
Nick Kurshev's avatar
Nick Kurshev committed
304

305
static int svq1_decode_block_non_intra (GetBitContext *bitbuf, uint8_t *pixels, int pitch ) {
306 307 308
  uint32_t    bit_cache;
  uint8_t    *list[63];
  uint32_t   *dst;
309
  const uint32_t *codebook;
310 311 312 313
  int         entries[6];
  int         i, j, m, n;
  int         mean, stages;
  int         x, y, width, height, level;
314
  uint32_t    n1, n2, n3, n4;
Nick Kurshev's avatar
Nick Kurshev committed
315

316 317
  /* initialize list for breadth first processing of vectors */
  list[0] = pixels;
Nick Kurshev's avatar
Nick Kurshev committed
318

319 320 321
  /* recursively process vector */
  for (i=0, m=1, n=1, level=5; i < n; i++) {
    SVQ1_PROCESS_VECTOR();
Nick Kurshev's avatar
Nick Kurshev committed
322

323 324 325 326
    /* destination address and vector size */
    dst = (uint32_t *) list[i];
    width = 1 << ((4 + level) /2);
    height = 1 << ((3 + level) /2);
Nick Kurshev's avatar
Nick Kurshev committed
327

328
    /* get number of stages (-1 skips vector, 0 for mean only) */
329
    stages = get_vlc2(bitbuf, svq1_inter_multistage[level].table, 3, 2) - 1;
330

331
    if (stages == -1) continue; /* skip vector */
332 333 334

    if ((stages > 0) && (level >= 4)) {
#ifdef DEBUG_SVQ1
335
    av_log(s->avctx, AV_LOG_INFO, "Error (svq1_decode_block_non_intra): invalid vector: stages=%i level=%i\n",stages,level);
336
#endif
337
      return -1;        /* invalid vector */
Nick Kurshev's avatar
Nick Kurshev committed
338 339
    }

340
    mean = get_vlc2(bitbuf, svq1_inter_mean.table, 9, 3) - 256;
341 342 343 344

    SVQ1_CALC_CODEBOOK_ENTRIES(svq1_inter_codebooks);
    SVQ1_DO_CODEBOOK_NONINTRA()
  }
Nick Kurshev's avatar
Nick Kurshev committed
345 346 347
  return 0;
}

348
static int svq1_decode_motion_vector (GetBitContext *bitbuf, svq1_pmv_t *mv, svq1_pmv_t **pmv) {
349 350
  int        diff;
  int        i;
Nick Kurshev's avatar
Nick Kurshev committed
351 352 353 354

  for (i=0; i < 2; i++) {

    /* get motion code */
355
    diff = get_vlc2(bitbuf, svq1_motion_component.table, 7, 2);
356
    if(diff<0)
357 358 359 360
        return -1;
    else if(diff){
        if(get_bits1(bitbuf)) diff= -diff;
    }
Nick Kurshev's avatar
Nick Kurshev committed
361 362 363

    /* add median of motion vector predictors and clip result */
    if (i == 1)
364
      mv->y = ((diff + mid_pred(pmv[0]->y, pmv[1]->y, pmv[2]->y)) << 26) >> 26;
Nick Kurshev's avatar
Nick Kurshev committed
365
    else
366
      mv->x = ((diff + mid_pred(pmv[0]->x, pmv[1]->x, pmv[2]->x)) << 26) >> 26;
Nick Kurshev's avatar
Nick Kurshev committed
367 368 369 370 371
  }

  return 0;
}

372
static void svq1_skip_block (uint8_t *current, uint8_t *previous, int pitch, int x, int y) {
Nick Kurshev's avatar
Nick Kurshev committed
373 374
  uint8_t *src;
  uint8_t *dst;
375
  int      i;
Nick Kurshev's avatar
Nick Kurshev committed
376 377 378 379 380 381 382 383 384 385 386

  src = &previous[x + y*pitch];
  dst = current;

  for (i=0; i < 16; i++) {
    memcpy (dst, src, 16);
    src += pitch;
    dst += pitch;
  }
}

387
static int svq1_motion_inter_block (MpegEncContext *s, GetBitContext *bitbuf,
388 389
                               uint8_t *current, uint8_t *previous, int pitch,
                               svq1_pmv_t *motion, int x, int y) {
Nick Kurshev's avatar
Nick Kurshev committed
390 391 392 393
  uint8_t    *src;
  uint8_t    *dst;
  svq1_pmv_t  mv;
  svq1_pmv_t *pmv[3];
394
  int         result;
Nick Kurshev's avatar
Nick Kurshev committed
395 396 397 398

  /* predict and decode motion vector */
  pmv[0] = &motion[0];
  if (y == 0) {
399
    pmv[1] =
Nick Kurshev's avatar
Nick Kurshev committed
400 401
    pmv[2] = pmv[0];
  }
402 403 404 405
  else {
    pmv[1] = &motion[(x / 8) + 2];
    pmv[2] = &motion[(x / 8) + 4];
  }
Nick Kurshev's avatar
Nick Kurshev committed
406

407
  result = svq1_decode_motion_vector (bitbuf, &mv, pmv);
Nick Kurshev's avatar
Nick Kurshev committed
408 409 410 411

  if (result != 0)
    return result;

412 413 414 415 416 417
  motion[0].x                =
  motion[(x / 8) + 2].x      =
  motion[(x / 8) + 3].x      = mv.x;
  motion[0].y                =
  motion[(x / 8) + 2].y      =
  motion[(x / 8) + 3].y      = mv.y;
418

419 420 421 422 423 424 425 426 427
  if(y + (mv.y >> 1)<0)
     mv.y= 0;
  if(x + (mv.x >> 1)<0)
     mv.x= 0;

#if 0
  int w= (s->width+15)&~15;
  int h= (s->height+15)&~15;
  if(x + (mv.x >> 1)<0 || y + (mv.y >> 1)<0 || x + (mv.x >> 1) + 16 > w || y + (mv.y >> 1) + 16> h)
428
      av_log(s->avctx, AV_LOG_INFO, "%d %d %d %d\n", x, y, x + (mv.x >> 1), y + (mv.y >> 1));
429
#endif
430

Nick Kurshev's avatar
Nick Kurshev committed
431 432 433
  src = &previous[(x + (mv.x >> 1)) + (y + (mv.y >> 1))*pitch];
  dst = current;

434
  s->dsp.put_pixels_tab[0][((mv.y & 1) << 1) | (mv.x & 1)](dst,src,pitch,16);
Nick Kurshev's avatar
Nick Kurshev committed
435 436 437 438

  return 0;
}

439
static int svq1_motion_inter_4v_block (MpegEncContext *s, GetBitContext *bitbuf,
440 441
                                  uint8_t *current, uint8_t *previous, int pitch,
                                  svq1_pmv_t *motion,int x, int y) {
Nick Kurshev's avatar
Nick Kurshev committed
442 443 444 445
  uint8_t    *src;
  uint8_t    *dst;
  svq1_pmv_t  mv;
  svq1_pmv_t *pmv[4];
446
  int         i, result;
Nick Kurshev's avatar
Nick Kurshev committed
447 448 449 450

  /* predict and decode motion vector (0) */
  pmv[0] = &motion[0];
  if (y == 0) {
451
    pmv[1] =
Nick Kurshev's avatar
Nick Kurshev committed
452 453
    pmv[2] = pmv[0];
  }
454 455 456 457
  else {
    pmv[1] = &motion[(x / 8) + 2];
    pmv[2] = &motion[(x / 8) + 4];
  }
Nick Kurshev's avatar
Nick Kurshev committed
458

459
  result = svq1_decode_motion_vector (bitbuf, &mv, pmv);
Nick Kurshev's avatar
Nick Kurshev committed
460 461 462 463 464 465 466

  if (result != 0)
    return result;

  /* predict and decode motion vector (1) */
  pmv[0] = &mv;
  if (y == 0) {
467
    pmv[1] =
Nick Kurshev's avatar
Nick Kurshev committed
468 469
    pmv[2] = pmv[0];
  }
470 471 472 473
  else {
    pmv[1] = &motion[(x / 8) + 3];
  }
  result = svq1_decode_motion_vector (bitbuf, &motion[0], pmv);
Nick Kurshev's avatar
Nick Kurshev committed
474 475 476 477 478 479 480 481

  if (result != 0)
    return result;

  /* predict and decode motion vector (2) */
  pmv[1] = &motion[0];
  pmv[2] = &motion[(x / 8) + 1];

482
  result = svq1_decode_motion_vector (bitbuf, &motion[(x / 8) + 2], pmv);
Nick Kurshev's avatar
Nick Kurshev committed
483 484 485 486 487 488 489 490

  if (result != 0)
    return result;

  /* predict and decode motion vector (3) */
  pmv[2] = &motion[(x / 8) + 2];
  pmv[3] = &motion[(x / 8) + 3];

491
  result = svq1_decode_motion_vector (bitbuf, pmv[3], pmv);
Nick Kurshev's avatar
Nick Kurshev committed
492 493 494 495 496 497

  if (result != 0)
    return result;

  /* form predictions */
  for (i=0; i < 4; i++) {
498 499
    int mvx= pmv[i]->x + (i&1)*16;
    int mvy= pmv[i]->y + (i>>1)*16;
500

501 502 503 504 505 506 507 508 509 510
    ///XXX /FIXME cliping or padding?
    if(y + (mvy >> 1)<0)
       mvy= 0;
    if(x + (mvx >> 1)<0)
       mvx= 0;

#if 0
  int w= (s->width+15)&~15;
  int h= (s->height+15)&~15;
  if(x + (mvx >> 1)<0 || y + (mvy >> 1)<0 || x + (mvx >> 1) + 8 > w || y + (mvy >> 1) + 8> h)
511
      av_log(s->avctx, AV_LOG_INFO, "%d %d %d %d\n", x, y, x + (mvx >> 1), y + (mvy >> 1));
512 513
#endif
    src = &previous[(x + (mvx >> 1)) + (y + (mvy >> 1))*pitch];
Nick Kurshev's avatar
Nick Kurshev committed
514
    dst = current;
515

516
    s->dsp.put_pixels_tab[1][((mvy & 1) << 1) | (mvx & 1)](dst,src,pitch,8);
Nick Kurshev's avatar
Nick Kurshev committed
517 518 519 520 521 522 523 524 525 526 527 528

    /* select next block */
    if (i & 1) {
      current  += 8*(pitch - 1);
    } else {
      current  += 8;
    }
  }

  return 0;
}

529
static int svq1_decode_delta_block (MpegEncContext *s, GetBitContext *bitbuf,
530 531
                        uint8_t *current, uint8_t *previous, int pitch,
                        svq1_pmv_t *motion, int x, int y) {
Nick Kurshev's avatar
Nick Kurshev committed
532
  uint32_t block_type;
533
  int      result = 0;
Nick Kurshev's avatar
Nick Kurshev committed
534 535

  /* get block type */
536
  block_type = get_vlc2(bitbuf, svq1_block_type.table, 2, 2);
Nick Kurshev's avatar
Nick Kurshev committed
537 538 539

  /* reset motion vectors */
  if (block_type == SVQ1_BLOCK_SKIP || block_type == SVQ1_BLOCK_INTRA) {
540 541
    motion[0].x                 =
    motion[0].y                 =
542 543 544
    motion[(x / 8) + 2].x =
    motion[(x / 8) + 2].y =
    motion[(x / 8) + 3].x =
Nick Kurshev's avatar
Nick Kurshev committed
545 546 547 548 549
    motion[(x / 8) + 3].y = 0;
  }

  switch (block_type) {
  case SVQ1_BLOCK_SKIP:
550
    svq1_skip_block (current, previous, pitch, x, y);
Nick Kurshev's avatar
Nick Kurshev committed
551 552 553
    break;

  case SVQ1_BLOCK_INTER:
554
    result = svq1_motion_inter_block (s, bitbuf, current, previous, pitch, motion, x, y);
Nick Kurshev's avatar
Nick Kurshev committed
555 556 557 558

    if (result != 0)
    {
#ifdef DEBUG_SVQ1
559
    av_log(s->avctx, AV_LOG_INFO, "Error in svq1_motion_inter_block %i\n",result);
Nick Kurshev's avatar
Nick Kurshev committed
560 561 562
#endif
      break;
    }
563
    result = svq1_decode_block_non_intra (bitbuf, current, pitch);
Nick Kurshev's avatar
Nick Kurshev committed
564 565 566
    break;

  case SVQ1_BLOCK_INTER_4V:
567
    result = svq1_motion_inter_4v_block (s, bitbuf, current, previous, pitch, motion, x, y);
Nick Kurshev's avatar
Nick Kurshev committed
568 569 570 571

    if (result != 0)
    {
#ifdef DEBUG_SVQ1
572
    av_log(s->avctx, AV_LOG_INFO, "Error in svq1_motion_inter_4v_block %i\n",result);
Nick Kurshev's avatar
Nick Kurshev committed
573 574 575
#endif
      break;
    }
576
    result = svq1_decode_block_non_intra (bitbuf, current, pitch);
Nick Kurshev's avatar
Nick Kurshev committed
577 578 579
    break;

  case SVQ1_BLOCK_INTRA:
580
    result = svq1_decode_block_intra (bitbuf, current, pitch);
Nick Kurshev's avatar
Nick Kurshev committed
581 582 583 584 585 586 587
    break;
  }

  return result;
}

/* standard video sizes */
588
static struct { int width; int height; } svq1_frame_size_table[8] = {
Nick Kurshev's avatar
Nick Kurshev committed
589 590 591 592
  { 160, 120 }, { 128,  96 }, { 176, 144 }, { 352, 288 },
  { 704, 576 }, { 240, 180 }, { 320, 240 }, {  -1,  -1 }
};

593 594 595 596 597 598 599 600 601 602
static uint16_t svq1_packet_checksum (uint8_t *data, int length, int value) {
  int i;

  for (i=0; i < length; i++) {
    value = checksum_table[data[i] ^ (value >> 8)] ^ ((value & 0xFF) << 8);
  }

  return value;
}

603
#if 0 /* unused, remove? */
604 605 606 607 608 609 610 611 612 613 614 615 616 617
static uint16_t svq1_component_checksum (uint16_t *pixels, int pitch,
                                         int width, int height, int value) {
  int x, y;

  for (y=0; y < height; y++) {
    for (x=0; x < width; x++) {
      value = checksum_table[pixels[x] ^ (value >> 8)] ^ ((value & 0xFF) << 8);
    }

    pixels += pitch;
  }

  return value;
}
618
#endif
619 620 621 622 623 624 625 626 627

static void svq1_parse_string (GetBitContext *bitbuf, uint8_t *out) {
  uint8_t seed;
  int     i;

  out[0] = get_bits (bitbuf, 8);

  seed = string_table[out[0]];

628
  for (i=1; i <= out[0]; i++) {
629 630 631 632 633 634
    out[i] = get_bits (bitbuf, 8) ^ seed;
    seed   = string_table[out[i] ^ seed];
  }
}

static int svq1_decode_frame_header (GetBitContext *bitbuf,MpegEncContext *s) {
Nick Kurshev's avatar
Nick Kurshev committed
635
  int frame_size_code;
636
  int temporal_reference;
Nick Kurshev's avatar
Nick Kurshev committed
637

638
  temporal_reference = get_bits (bitbuf, 8);
Nick Kurshev's avatar
Nick Kurshev committed
639 640

  /* frame type */
641
  s->pict_type= get_bits (bitbuf, 2)+1;
642
  if(s->pict_type==4)
643
      return -1;
644

645
  if (s->pict_type == I_TYPE) {
Nick Kurshev's avatar
Nick Kurshev committed
646 647 648

    /* unknown fields */
    if (s->f_code == 0x50 || s->f_code == 0x60) {
649 650 651 652
      int csum = get_bits (bitbuf, 16);

      csum = svq1_packet_checksum ((uint8_t *)bitbuf->buffer, bitbuf->size_in_bits>>3, csum);

653
//      av_log(s->avctx, AV_LOG_INFO, "%s checksum (%02x) for packet data\n",
654
//              (csum == 0) ? "correct" : "incorrect", csum);
Nick Kurshev's avatar
Nick Kurshev committed
655 656 657
    }

    if ((s->f_code ^ 0x10) >= 0x50) {
658 659 660 661
      char msg[256];

      svq1_parse_string (bitbuf, (char *) msg);

662
      av_log(s->avctx, AV_LOG_INFO, "embedded message: \"%s\"\n", (char *) msg);
Nick Kurshev's avatar
Nick Kurshev committed
663 664
    }

Alex Beregszaszi's avatar
Alex Beregszaszi committed
665 666 667
    skip_bits (bitbuf, 2);
    skip_bits (bitbuf, 2);
    skip_bits1 (bitbuf);
Nick Kurshev's avatar
Nick Kurshev committed
668 669 670 671 672 673 674 675 676 677 678 679 680

    /* load frame size */
    frame_size_code = get_bits (bitbuf, 3);

    if (frame_size_code == 7) {
      /* load width, height (12 bits each) */
      s->width = get_bits (bitbuf, 12);
      s->height = get_bits (bitbuf, 12);

      if (!s->width || !s->height)
        return -1;
    } else {
      /* get width, height from table */
681 682
      s->width = svq1_frame_size_table[frame_size_code].width;
      s->height = svq1_frame_size_table[frame_size_code].height;
Nick Kurshev's avatar
Nick Kurshev committed
683 684 685 686 687
    }
  }

  /* unknown fields */
  if (get_bits (bitbuf, 1) == 1) {
Alex Beregszaszi's avatar
Alex Beregszaszi committed
688 689
    skip_bits1 (bitbuf);       /* use packet checksum if (1) */
    skip_bits1 (bitbuf);       /* component checksums after image data if (1) */
Nick Kurshev's avatar
Nick Kurshev committed
690 691 692 693 694 695

    if (get_bits (bitbuf, 2) != 0)
      return -1;
  }

  if (get_bits (bitbuf, 1) == 1) {
Alex Beregszaszi's avatar
Alex Beregszaszi committed
696 697 698 699
    skip_bits1 (bitbuf);
    skip_bits (bitbuf, 4);
    skip_bits1 (bitbuf);
    skip_bits (bitbuf, 2);
Nick Kurshev's avatar
Nick Kurshev committed
700 701

    while (get_bits (bitbuf, 1) == 1) {
Alex Beregszaszi's avatar
Alex Beregszaszi committed
702
      skip_bits (bitbuf, 8);
Nick Kurshev's avatar
Nick Kurshev committed
703 704
    }
  }
705

Nick Kurshev's avatar
Nick Kurshev committed
706 707 708
  return 0;
}

709
static int svq1_decode_frame(AVCodecContext *avctx,
Nick Kurshev's avatar
Nick Kurshev committed
710
                             void *data, int *data_size,
711
                             uint8_t *buf, int buf_size)
Nick Kurshev's avatar
Nick Kurshev committed
712 713
{
  MpegEncContext *s=avctx->priv_data;
714 715
  uint8_t        *current, *previous;
  int             result, i, x, y, width, height;
716
  AVFrame *pict = data;
Nick Kurshev's avatar
Nick Kurshev committed
717 718

  /* initialize bit buffer */
719
  init_get_bits(&s->gb,buf,buf_size*8);
Nick Kurshev's avatar
Nick Kurshev committed
720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735

  /* decode frame header */
  s->f_code = get_bits (&s->gb, 22);

  if ((s->f_code & ~0x70) || !(s->f_code & 0x60))
    return -1;

  /* swap some header bytes (why?) */
  if (s->f_code != 0x20) {
    uint32_t *src = (uint32_t *) (buf + 4);

    for (i=0; i < 4; i++) {
      src[i] = ((src[i] << 16) | (src[i] >> 16)) ^ src[7 - i];
    }
  }

736
  result = svq1_decode_frame_header (&s->gb, s);
Nick Kurshev's avatar
Nick Kurshev committed
737 738 739 740

  if (result != 0)
  {
#ifdef DEBUG_SVQ1
741
    av_log(s->avctx, AV_LOG_INFO, "Error in svq1_decode_frame_header %i\n",result);
Nick Kurshev's avatar
Nick Kurshev committed
742 743 744
#endif
    return result;
  }
745

746
  //FIXME this avoids some confusion for "B frames" without 2 references
747
  //this should be removed after libavcodec can handle more flexible picture types & ordering
748
  if(s->pict_type==B_TYPE && s->last_picture_ptr==NULL) return buf_size;
749

Michael Niedermayer's avatar
Michael Niedermayer committed
750
  if(avctx->hurry_up && s->pict_type==B_TYPE) return buf_size;
Michael Niedermayer's avatar
Michael Niedermayer committed
751 752 753
  if(  (avctx->skip_frame >= AVDISCARD_NONREF && s->pict_type==B_TYPE)
     ||(avctx->skip_frame >= AVDISCARD_NONKEY && s->pict_type!=I_TYPE)
     || avctx->skip_frame >= AVDISCARD_ALL)
754
      return buf_size;
Nick Kurshev's avatar
Nick Kurshev committed
755

Michael Niedermayer's avatar
Michael Niedermayer committed
756 757 758
  if(MPV_frame_start(s, avctx) < 0)
      return -1;

Nick Kurshev's avatar
Nick Kurshev committed
759 760
  /* decode y, u and v components */
  for (i=0; i < 3; i++) {
761
    int linesize;
Nick Kurshev's avatar
Nick Kurshev committed
762
    if (i == 0) {
763 764
      width  = (s->width+15)&~15;
      height = (s->height+15)&~15;
765
      linesize= s->linesize;
Nick Kurshev's avatar
Nick Kurshev committed
766
    } else {
767
      if(s->flags&CODEC_FLAG_GRAY) break;
768 769
      width  = (s->width/4+15)&~15;
      height = (s->height/4+15)&~15;
770
      linesize= s->uvlinesize;
Nick Kurshev's avatar
Nick Kurshev committed
771 772
    }

Michael Niedermayer's avatar
Michael Niedermayer committed
773
    current  = s->current_picture.data[i];
Nick Kurshev's avatar
Nick Kurshev committed
774

775
    if(s->pict_type==B_TYPE){
Michael Niedermayer's avatar
Michael Niedermayer committed
776
        previous = s->next_picture.data[i];
777
    }else{
Michael Niedermayer's avatar
Michael Niedermayer committed
778
        previous = s->last_picture.data[i];
779 780 781
    }

    if (s->pict_type == I_TYPE) {
Nick Kurshev's avatar
Nick Kurshev committed
782 783
      /* keyframe */
      for (y=0; y < height; y+=16) {
784 785 786 787
        for (x=0; x < width; x+=16) {
          result = svq1_decode_block_intra (&s->gb, &current[x], linesize);
          if (result != 0)
          {
788
//#ifdef DEBUG_SVQ1
789
            av_log(s->avctx, AV_LOG_INFO, "Error in svq1_decode_block %i (keyframe)\n",result);
790
//#endif
791 792 793 794
            return result;
          }
        }
        current += 16*linesize;
Nick Kurshev's avatar
Nick Kurshev committed
795 796
      }
    } else {
797
      svq1_pmv_t pmv[width/8+3];
Nick Kurshev's avatar
Nick Kurshev committed
798
      /* delta frame */
799
      memset (pmv, 0, ((width / 8) + 3) * sizeof(svq1_pmv_t));
Nick Kurshev's avatar
Nick Kurshev committed
800 801

      for (y=0; y < height; y+=16) {
802 803 804 805 806
        for (x=0; x < width; x+=16) {
          result = svq1_decode_delta_block (s, &s->gb, &current[x], previous,
                                            linesize, pmv, x, y);
          if (result != 0)
          {
Nick Kurshev's avatar
Nick Kurshev committed
807
#ifdef DEBUG_SVQ1
808
    av_log(s->avctx, AV_LOG_INFO, "Error in svq1_decode_delta_block %i\n",result);
Nick Kurshev's avatar
Nick Kurshev committed
809
#endif
810 811 812
            return result;
          }
        }
Nick Kurshev's avatar
Nick Kurshev committed
813

814 815
        pmv[0].x =
        pmv[0].y = 0;
Nick Kurshev's avatar
Nick Kurshev committed
816

817
        current += 16*linesize;
Nick Kurshev's avatar
Nick Kurshev committed
818 819 820
      }
    }
  }
821

822
  *pict = *(AVFrame*)&s->current_picture;
823

Michael Niedermayer's avatar
Michael Niedermayer committed
824 825

  MPV_frame_end(s);
826

827
  *data_size=sizeof(AVFrame);
Michael Niedermayer's avatar
Michael Niedermayer committed
828
  return buf_size;
Nick Kurshev's avatar
Nick Kurshev committed
829 830 831 832 833
}

static int svq1_decode_init(AVCodecContext *avctx)
{
    MpegEncContext *s = avctx->priv_data;
834
    int i;
835

Michael Niedermayer's avatar
Michael Niedermayer committed
836 837
    MPV_decode_defaults(s);

Nick Kurshev's avatar
Nick Kurshev committed
838 839 840 841 842
    s->avctx = avctx;
    s->width = (avctx->width+3)&~3;
    s->height = (avctx->height+3)&~3;
    s->codec_id= avctx->codec->id;
    avctx->pix_fmt = PIX_FMT_YUV410P;
Michael Niedermayer's avatar
Michael Niedermayer committed
843
    avctx->has_b_frames= 1; // not true, but DP frames and these behave like unidirectional b frames
844
    s->flags= avctx->flags;
845
    if (MPV_common_init(s) < 0) return -1;
846 847 848

    init_vlc(&svq1_block_type, 2, 4,
        &svq1_block_type_vlc[0][1], 2, 1,
849
        &svq1_block_type_vlc[0][0], 2, 1, 1);
850

851 852
    init_vlc(&svq1_motion_component, 7, 33,
        &mvtab[0][1], 2, 1,
853
        &mvtab[0][0], 2, 1, 1);
854 855 856 857

    for (i = 0; i < 6; i++) {
        init_vlc(&svq1_intra_multistage[i], 3, 8,
            &svq1_intra_multistage_vlc[i][0][1], 2, 1,
858
            &svq1_intra_multistage_vlc[i][0][0], 2, 1, 1);
859 860
        init_vlc(&svq1_inter_multistage[i], 3, 8,
            &svq1_inter_multistage_vlc[i][0][1], 2, 1,
861
            &svq1_inter_multistage_vlc[i][0][0], 2, 1, 1);
862 863 864 865
    }

    init_vlc(&svq1_intra_mean, 8, 256,
        &svq1_intra_mean_vlc[0][1], 4, 2,
866
        &svq1_intra_mean_vlc[0][0], 4, 2, 1);
867 868 869

    init_vlc(&svq1_inter_mean, 9, 512,
        &svq1_inter_mean_vlc[0][1], 4, 2,
870
        &svq1_inter_mean_vlc[0][0], 4, 2, 1);
871

Nick Kurshev's avatar
Nick Kurshev committed
872 873 874 875 876 877 878 879 880 881 882
    return 0;
}

static int svq1_decode_end(AVCodecContext *avctx)
{
    MpegEncContext *s = avctx->priv_data;

    MPV_common_end(s);
    return 0;
}

883 884
static void svq1_write_header(SVQ1Context *s, int frame_type)
{
885 886
    int i;

887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904
    /* frame code */
    put_bits(&s->pb, 22, 0x20);

    /* temporal reference (sure hope this is a "don't care") */
    put_bits(&s->pb, 8, 0x00);

    /* frame type */
    put_bits(&s->pb, 2, frame_type - 1);

    if (frame_type == I_TYPE) {

        /* no checksum since frame code is 0x20 */

        /* no embedded string either */

        /* output 5 unknown bits (2 + 2 + 1) */
        put_bits(&s->pb, 5, 0);

905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920
        for (i = 0; i < 7; i++)
        {
            if ((svq1_frame_size_table[i].width == s->frame_width) &&
                (svq1_frame_size_table[i].height == s->frame_height))
            {
                put_bits(&s->pb, 3, i);
                break;
            }
        }

        if (i == 7)
        {
            put_bits(&s->pb, 3, 7);
                put_bits(&s->pb, 12, s->frame_width);
                put_bits(&s->pb, 12, s->frame_height);
        }
921 922 923 924 925 926 927 928 929 930
    }

    /* no checksum or extra data (next 2 bits get 0) */
    put_bits(&s->pb, 2, 0);
}


#define QUALITY_THRESHOLD 100
#define THRESHOLD_MULTIPLIER 0.6

931 932 933
#if defined(HAVE_ALTIVEC)
#undef vector
#endif
934

935
static int encode_block(SVQ1Context *s, uint8_t *src, uint8_t *ref, uint8_t *decoded, int stride, int level, int threshold, int lambda, int intra){
936 937 938 939 940 941 942
    int count, y, x, i, j, split, best_mean, best_score, best_count;
    int best_vector[6];
    int block_sum[7]= {0, 0, 0, 0, 0, 0};
    int w= 2<<((level+2)>>1);
    int h= 2<<((level+1)>>1);
    int size=w*h;
    int16_t block[7][256];
943 944 945 946
    const int8_t *codebook_sum, *codebook;
    const uint16_t (*mean_vlc)[2];
    const uint8_t (*multistage_vlc)[2];

947 948
    best_score=0;
    //FIXME optimize, this doenst need to be done multiple times
949 950 951 952 953
    if(intra){
        codebook_sum= svq1_intra_codebook_sum[level];
        codebook= svq1_intra_codebooks[level];
        mean_vlc= svq1_intra_mean_vlc;
        multistage_vlc= svq1_intra_multistage_vlc[level];
954 955 956 957 958 959 960 961
        for(y=0; y<h; y++){
            for(x=0; x<w; x++){
                int v= src[x + y*stride];
                block[0][x + w*y]= v;
                best_score += v*v;
                block_sum[0] += v;
            }
        }
962 963 964
    }else{
        codebook_sum= svq1_inter_codebook_sum[level];
        codebook= svq1_inter_codebooks[level];
965
        mean_vlc= svq1_inter_mean_vlc + 256;
966
        multistage_vlc= svq1_inter_multistage_vlc[level];
967 968 969 970 971 972 973
        for(y=0; y<h; y++){
            for(x=0; x<w; x++){
                int v= src[x + y*stride] - ref[x + y*stride];
                block[0][x + w*y]= v;
                best_score += v*v;
                block_sum[0] += v;
            }
974 975 976 977 978 979 980 981 982 983
        }
    }

    best_count=0;
    best_score -= ((block_sum[0]*block_sum[0])>>(level+3));
    best_mean= (block_sum[0] + (size>>1)) >> (level+3);

    if(level<4){
        for(count=1; count<7; count++){
            int best_vector_score= INT_MAX;
984
            int best_vector_sum=-999, best_vector_mean=-999;
985
            const int stage= count-1;
986
            const int8_t *vector;
987

988
            for(i=0; i<16; i++){
989
                int sum= codebook_sum[stage*16 + i];
990 991
                int sqr=0;
                int diff, mean, score;
992

993
                vector = codebook + stage*size*16 + i*size;
994

995 996 997 998 999 1000
                for(j=0; j<size; j++){
                    int v= vector[j];
                    sqr += (v - block[stage][j])*(v - block[stage][j]);
                }
                diff= block_sum[stage] - sum;
                mean= (diff + (size>>1)) >> (level+3);
1001 1002 1003
                assert(mean >-300 && mean<300);
                if(intra) mean= clip(mean, 0, 255);
                else      mean= clip(mean, -256, 255);
1004 1005 1006 1007 1008 1009 1010 1011
                score= sqr - ((diff*(int64_t)diff)>>(level+3)); //FIXME 64bit slooow
                if(score < best_vector_score){
                    best_vector_score= score;
                    best_vector[stage]= i;
                    best_vector_sum= sum;
                    best_vector_mean= mean;
                }
            }
1012
            assert(best_vector_mean != -999);
1013
            vector= codebook + stage*size*16 + best_vector[stage]*size;
1014 1015 1016 1017
            for(j=0; j<size; j++){
                block[stage+1][j] = block[stage][j] - vector[j];
            }
            block_sum[stage+1]= block_sum[stage] - best_vector_sum;
1018
            best_vector_score +=
1019
                lambda*(+ 1 + 4*count
1020 1021
                        + multistage_vlc[1+count][1]
                        + mean_vlc[best_vector_mean][1]);
1022

1023 1024 1025 1026 1027 1028 1029
            if(best_vector_score < best_score){
                best_score= best_vector_score;
                best_count= count;
                best_mean= best_vector_mean;
            }
        }
    }
1030

1031 1032 1033 1034 1035 1036 1037 1038 1039
    split=0;
    if(best_score > threshold && level){
        int score=0;
        int offset= (level&1) ? stride*h/2 : w/2;
        PutBitContext backup[6];

        for(i=level-1; i>=0; i--){
            backup[i]= s->reorder_pb[i];
        }
1040 1041
        score += encode_block(s, src         , ref         , decoded         , stride, level-1, threshold>>1, lambda, intra);
        score += encode_block(s, src + offset, ref + offset, decoded + offset, stride, level-1, threshold>>1, lambda, intra);
1042
        score += lambda;
1043

1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056
        if(score < best_score){
            best_score= score;
            split=1;
        }else{
            for(i=level-1; i>=0; i--){
                s->reorder_pb[i]= backup[i];
            }
        }
    }
    if (level > 0)
        put_bits(&s->reorder_pb[level], 1, split);

    if(!split){
1057 1058
        assert((best_mean >= 0 && best_mean<256) || !intra);
        assert(best_mean >= -256 && best_mean<256);
1059 1060
        assert(best_count >=0 && best_count<7);
        assert(level<4 || best_count==0);
1061

1062
        /* output the encoding */
1063
        put_bits(&s->reorder_pb[level],
1064 1065 1066 1067
            multistage_vlc[1 + best_count][1],
            multistage_vlc[1 + best_count][0]);
        put_bits(&s->reorder_pb[level], mean_vlc[best_mean][1],
            mean_vlc[best_mean][0]);
1068 1069 1070 1071 1072

        for (i = 0; i < best_count; i++){
            assert(best_vector[i]>=0 && best_vector[i]<16);
            put_bits(&s->reorder_pb[level], 4, best_vector[i]);
        }
1073

1074 1075 1076 1077 1078
        for(y=0; y<h; y++){
            for(x=0; x<w; x++){
                decoded[x + y*stride]= src[x + y*stride] - block[best_count][x + w*y] + best_mean;
            }
        }
1079 1080 1081 1082 1083
    }

    return best_score;
}

1084 1085
#ifdef CONFIG_ENCODERS

1086
static int svq1_encode_plane(SVQ1Context *s, int plane, unsigned char *src_plane, unsigned char *ref_plane, unsigned char *decoded_plane,
1087
    int width, int height, int src_stride, int stride)
1088 1089
{
    int x, y;
1090
    int i;
1091 1092 1093
    int block_width, block_height;
    int level;
    int threshold[6];
1094
    const int lambda= (s->picture.quality*s->picture.quality) >> (2*FF_LAMBDA_SHIFT);
1095 1096 1097 1098 1099 1100 1101 1102 1103

    /* figure out the acceptable level thresholds in advance */
    threshold[5] = QUALITY_THRESHOLD;
    for (level = 4; level >= 0; level--)
        threshold[level] = threshold[level + 1] * THRESHOLD_MULTIPLIER;

    block_width = (width + 15) / 16;
    block_height = (height + 15) / 16;

1104 1105 1106 1107 1108 1109
    if(s->picture.pict_type == P_TYPE){
        s->m.avctx= s->avctx;
        s->m.current_picture_ptr= &s->m.current_picture;
        s->m.last_picture_ptr   = &s->m.last_picture;
        s->m.last_picture.data[0]= ref_plane;
        s->m.linesize=
1110 1111
        s->m.last_picture.linesize[0]=
        s->m.new_picture.linesize[0]=
1112 1113 1114 1115 1116 1117 1118 1119 1120 1121
        s->m.current_picture.linesize[0]= stride;
        s->m.width= width;
        s->m.height= height;
        s->m.mb_width= block_width;
        s->m.mb_height= block_height;
        s->m.mb_stride= s->m.mb_width+1;
        s->m.b8_stride= 2*s->m.mb_width+1;
        s->m.f_code=1;
        s->m.pict_type= s->picture.pict_type;
        s->m.me_method= s->avctx->me_method;
1122 1123 1124 1125
        s->m.me.scene_change_score=0;
        s->m.flags= s->avctx->flags;
//        s->m.out_format = FMT_H263;
//        s->m.unrestricted_mv= 1;
1126

1127 1128 1129
        s->m.lambda= s->picture.quality;
        s->m.qscale= (s->m.lambda*139 + FF_LAMBDA_SCALE*64) >> (FF_LAMBDA_SHIFT + 7);
        s->m.lambda2= (s->m.lambda*s->m.lambda + FF_LAMBDA_SCALE/2) >> FF_LAMBDA_SHIFT;
1130

1131
        if(!s->motion_val8[plane]){
1132 1133
            s->motion_val8 [plane]= av_mallocz((s->m.b8_stride*block_height*2 + 2)*2*sizeof(int16_t));
            s->motion_val16[plane]= av_mallocz((s->m.mb_stride*(block_height + 2) + 1)*2*sizeof(int16_t));
1134
        }
1135

1136
        s->m.mb_type= s->mb_type;
1137

1138
        //dummies, to avoid segfaults
1139 1140 1141
        s->m.current_picture.mb_mean=   (uint8_t *)s->dummy;
        s->m.current_picture.mb_var=    (uint16_t*)s->dummy;
        s->m.current_picture.mc_mb_var= (uint16_t*)s->dummy;
1142
        s->m.current_picture.mb_type= s->dummy;
1143

1144 1145
        s->m.current_picture.motion_val[0]= s->motion_val8[plane] + 2;
        s->m.p_mv_table= s->motion_val16[plane] + s->m.mb_stride + 1;
1146 1147
        s->m.dsp= s->dsp; //move
        ff_init_me(&s->m);
1148

1149 1150 1151 1152
        s->m.me.dia_size= s->avctx->dia_size;
        s->m.first_slice_line=1;
        for (y = 0; y < block_height; y++) {
            uint8_t src[stride*16];
1153

1154 1155
            s->m.new_picture.data[0]= src - y*16*stride; //ugly
            s->m.mb_y= y;
1156

1157 1158 1159 1160 1161 1162 1163
            for(i=0; i<16 && i + 16*y<height; i++){
                memcpy(&src[i*stride], &src_plane[(i+16*y)*src_stride], width);
                for(x=width; x<16*block_width; x++)
                    src[i*stride+x]= src[i*stride+x-1];
            }
            for(; i<16 && i + 16*y<16*block_height; i++)
                memcpy(&src[i*stride], &src[(i-1)*stride], 16*block_width);
1164

1165 1166 1167 1168
            for (x = 0; x < block_width; x++) {
                s->m.mb_x= x;
                ff_init_block_index(&s->m);
                ff_update_block_index(&s->m);
1169

1170 1171 1172 1173
                ff_estimate_p_frame_motion(&s->m, x, y);
            }
            s->m.first_slice_line=0;
        }
1174

1175 1176 1177
        ff_fix_long_p_mvs(&s->m);
        ff_fix_long_mvs(&s->m, NULL, 0, s->m.p_mv_table, s->m.f_code, CANDIDATE_MB_TYPE_INTER, 0);
    }
1178

1179
    s->m.first_slice_line=1;
1180
    for (y = 0; y < block_height; y++) {
1181
        uint8_t src[stride*16];
1182

1183 1184 1185 1186 1187 1188 1189
        for(i=0; i<16 && i + 16*y<height; i++){
            memcpy(&src[i*stride], &src_plane[(i+16*y)*src_stride], width);
            for(x=width; x<16*block_width; x++)
                src[i*stride+x]= src[i*stride+x-1];
        }
        for(; i<16 && i + 16*y<16*block_height; i++)
            memcpy(&src[i*stride], &src[(i-1)*stride], 16*block_width);
1190

1191
        s->m.mb_y= y;
1192
        for (x = 0; x < block_width; x++) {
1193 1194
            uint8_t reorder_buffer[3][6][7*32];
            int count[3][6];
1195 1196 1197
            int offset = y * 16 * stride + x * 16;
            uint8_t *decoded= decoded_plane + offset;
            uint8_t *ref= ref_plane + offset;
1198
            int score[4]={0,0,0,0}, best;
1199
            uint8_t temp[16*stride];
1200

1201 1202 1203 1204
            if(s->pb.buf_end - s->pb.buf - (put_bits_count(&s->pb)>>3) < 3000){ //FIXME check size
                av_log(s->avctx, AV_LOG_ERROR, "encoded frame too large\n");
                return -1;
            }
1205

1206 1207 1208
            s->m.mb_x= x;
            ff_init_block_index(&s->m);
            ff_update_block_index(&s->m);
1209

1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225
            if(s->picture.pict_type == I_TYPE || (s->m.mb_type[x + y*s->m.mb_stride]&CANDIDATE_MB_TYPE_INTRA)){
                for(i=0; i<6; i++){
                    init_put_bits(&s->reorder_pb[i], reorder_buffer[0][i], 7*32);
                }
                if(s->picture.pict_type == P_TYPE){
                    const uint8_t *vlc= svq1_block_type_vlc[SVQ1_BLOCK_INTRA];
                    put_bits(&s->reorder_pb[5], vlc[1], vlc[0]);
                    score[0]= vlc[1]*lambda;
                }
                score[0]+= encode_block(s, src+16*x, NULL, temp, stride, 5, 64, lambda, 1);
                for(i=0; i<6; i++){
                    count[0][i]= put_bits_count(&s->reorder_pb[i]);
                    flush_put_bits(&s->reorder_pb[i]);
                }
            }else
                score[0]= INT_MAX;
1226

1227
            best=0;
1228

1229 1230
            if(s->picture.pict_type == P_TYPE){
                const uint8_t *vlc= svq1_block_type_vlc[SVQ1_BLOCK_INTER];
1231 1232 1233 1234 1235 1236 1237 1238
                int mx, my, pred_x, pred_y, dxy;
                int16_t *motion_ptr;

                motion_ptr= h263_pred_motion(&s->m, 0, 0, &pred_x, &pred_y);
                if(s->m.mb_type[x + y*s->m.mb_stride]&CANDIDATE_MB_TYPE_INTER){
                    for(i=0; i<6; i++)
                        init_put_bits(&s->reorder_pb[i], reorder_buffer[1][i], 7*32);

1239
                    put_bits(&s->reorder_pb[5], vlc[1], vlc[0]);
1240 1241

                    s->m.pb= s->reorder_pb[5];
1242 1243 1244 1245 1246 1247 1248 1249 1250 1251
                    mx= motion_ptr[0];
                    my= motion_ptr[1];
                    assert(mx>=-32 && mx<=31);
                    assert(my>=-32 && my<=31);
                    assert(pred_x>=-32 && pred_x<=31);
                    assert(pred_y>=-32 && pred_y<=31);
                    ff_h263_encode_motion(&s->m, mx - pred_x, 1);
                    ff_h263_encode_motion(&s->m, my - pred_y, 1);
                    s->reorder_pb[5]= s->m.pb;
                    score[1] += lambda*put_bits_count(&s->reorder_pb[5]);
1252

1253
                    dxy= (mx&1) + 2*(my&1);
1254

1255
                    s->dsp.put_pixels_tab[0][dxy](temp+16, ref + (mx>>1) + stride*(my>>1), stride, 16);
1256

1257 1258
                    score[1]+= encode_block(s, src+16*x, temp+16, decoded, stride, 5, 64, lambda, 0);
                    best= score[1] <= score[0];
1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270

                    vlc= svq1_block_type_vlc[SVQ1_BLOCK_SKIP];
                    score[2]= s->dsp.sse[0](NULL, src+16*x, ref, stride, 16);
                    score[2]+= vlc[1]*lambda;
                    if(score[2] < score[best] && mx==0 && my==0){
                        best=2;
                        s->dsp.put_pixels_tab[0][0](decoded, ref, stride, 16);
                        for(i=0; i<6; i++){
                            count[2][i]=0;
                        }
                        put_bits(&s->pb, vlc[1], vlc[0]);
                    }
1271
                }
1272

1273 1274 1275 1276 1277
                if(best==1){
                    for(i=0; i<6; i++){
                        count[1][i]= put_bits_count(&s->reorder_pb[i]);
                        flush_put_bits(&s->reorder_pb[i]);
                    }
1278 1279 1280 1281 1282
                }else{
                    motion_ptr[0                 ] = motion_ptr[1                 ]=
                    motion_ptr[2                 ] = motion_ptr[3                 ]=
                    motion_ptr[0+2*s->m.b8_stride] = motion_ptr[1+2*s->m.b8_stride]=
                    motion_ptr[2+2*s->m.b8_stride] = motion_ptr[3+2*s->m.b8_stride]=0;
1283
                }
1284
            }
1285

1286 1287 1288 1289 1290 1291 1292
            s->rd_total += score[best];

            for(i=5; i>=0; i--){
                ff_copy_bits(&s->pb, reorder_buffer[best][i], count[best][i]);
            }
            if(best==0){
                s->dsp.put_pixels_tab[0][0](decoded, temp, stride, 16);
1293
            }
1294
        }
1295
        s->m.first_slice_line=0;
1296
    }
1297
    return 0;
1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315
}

static int svq1_encode_init(AVCodecContext *avctx)
{
    SVQ1Context * const s = avctx->priv_data;

    dsputil_init(&s->dsp, avctx);
    avctx->coded_frame= (AVFrame*)&s->picture;

    s->frame_width = avctx->width;
    s->frame_height = avctx->height;

    s->y_block_width = (s->frame_width + 15) / 16;
    s->y_block_height = (s->frame_height + 15) / 16;

    s->c_block_width = (s->frame_width / 4 + 15) / 16;
    s->c_block_height = (s->frame_height / 4 + 15) / 16;

1316
    s->avctx= avctx;
Michael Niedermayer's avatar
Michael Niedermayer committed
1317
    s->m.avctx= avctx;
1318
    s->m.me.scratchpad= av_mallocz((avctx->width+64)*2*16*2*sizeof(uint8_t));
1319 1320 1321 1322 1323
    s->m.me.map       = av_mallocz(ME_MAP_SIZE*sizeof(uint32_t));
    s->m.me.score_map = av_mallocz(ME_MAP_SIZE*sizeof(uint32_t));
    s->mb_type        = av_mallocz((s->y_block_width+1)*s->y_block_height*sizeof(int16_t));
    s->dummy          = av_mallocz((s->y_block_width+1)*s->y_block_height*sizeof(int32_t));
    h263_encode_init(&s->m); //mv_penalty
1324

1325 1326 1327
    return 0;
}

1328
static int svq1_encode_frame(AVCodecContext *avctx, unsigned char *buf,
1329 1330 1331 1332 1333
    int buf_size, void *data)
{
    SVQ1Context * const s = avctx->priv_data;
    AVFrame *pict = data;
    AVFrame * const p= (AVFrame*)&s->picture;
1334 1335
    AVFrame temp;
    int i;
1336

1337 1338 1339 1340
    if(avctx->pix_fmt != PIX_FMT_YUV410P){
        av_log(avctx, AV_LOG_ERROR, "unsupported pixel format\n");
        return -1;
    }
1341

1342 1343 1344 1345
    if(!s->current_picture.data[0]){
        avctx->get_buffer(avctx, &s->current_picture);
        avctx->get_buffer(avctx, &s->last_picture);
    }
1346

1347 1348 1349
    temp= s->current_picture;
    s->current_picture= s->last_picture;
    s->last_picture= temp;
1350

1351 1352 1353
    init_put_bits(&s->pb, buf, buf_size);

    *p = *pict;
1354
    p->pict_type = avctx->frame_number % avctx->gop_size ? P_TYPE : I_TYPE;
1355
    p->key_frame = p->pict_type == I_TYPE;
1356 1357

    svq1_write_header(s, p->pict_type);
1358
    for(i=0; i<3; i++){
1359
        if(svq1_encode_plane(s, i,
1360
            s->picture.data[i], s->last_picture.data[i], s->current_picture.data[i],
1361
            s->frame_width / (i?4:1), s->frame_height / (i?4:1),
1362 1363
            s->picture.linesize[i], s->current_picture.linesize[i]) < 0)
                return -1;
1364 1365 1366 1367 1368
    }

//    align_put_bits(&s->pb);
    while(put_bits_count(&s->pb) & 31)
        put_bits(&s->pb, 1, 0);
1369

Michael Niedermayer's avatar
Michael Niedermayer committed
1370
    flush_put_bits(&s->pb);
1371 1372 1373 1374 1375 1376 1377

    return (put_bits_count(&s->pb) / 8);
}

static int svq1_encode_end(AVCodecContext *avctx)
{
    SVQ1Context * const s = avctx->priv_data;
1378
    int i;
1379

1380
    av_log(avctx, AV_LOG_DEBUG, "RD: %f\n", s->rd_total/(double)(avctx->width*avctx->height*avctx->frame_number));
1381 1382

    av_freep(&s->m.me.scratchpad);
1383 1384 1385 1386 1387 1388 1389 1390 1391
    av_freep(&s->m.me.map);
    av_freep(&s->m.me.score_map);
    av_freep(&s->mb_type);
    av_freep(&s->dummy);

    for(i=0; i<3; i++){
        av_freep(&s->motion_val8[i]);
        av_freep(&s->motion_val16[i]);
    }
1392 1393 1394 1395

    return 0;
}

1396 1397
#endif //CONFIG_ENCODERS

Nick Kurshev's avatar
Nick Kurshev committed
1398 1399 1400 1401 1402 1403 1404 1405
AVCodec svq1_decoder = {
    "svq1",
    CODEC_TYPE_VIDEO,
    CODEC_ID_SVQ1,
    sizeof(MpegEncContext),
    svq1_decode_init,
    NULL,
    svq1_decode_end,
1406 1407
    svq1_decode_frame,
    CODEC_CAP_DR1,
Michael Niedermayer's avatar
Michael Niedermayer committed
1408
    .flush= ff_mpeg_flush,
1409
    .pix_fmts= (enum PixelFormat[]){PIX_FMT_YUV410P, -1},
Nick Kurshev's avatar
Nick Kurshev committed
1410
};
1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421

#ifdef CONFIG_ENCODERS

AVCodec svq1_encoder = {
    "svq1",
    CODEC_TYPE_VIDEO,
    CODEC_ID_SVQ1,
    sizeof(SVQ1Context),
    svq1_encode_init,
    svq1_encode_frame,
    svq1_encode_end,
1422
    .pix_fmts= (enum PixelFormat[]){PIX_FMT_YUV410P, -1},
1423 1424 1425
};

#endif //CONFIG_ENCODERS