huffyuv.c 40.4 KB
Newer Older
Michael Niedermayer's avatar
Michael Niedermayer committed
1 2 3
/*
 * huffyuv codec for libavcodec
 *
Michael Niedermayer's avatar
Michael Niedermayer committed
4
 * Copyright (c) 2002-2003 Michael Niedermayer <michaelni@gmx.at>
Michael Niedermayer's avatar
Michael Niedermayer committed
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
 *
 * 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
 *
 * see http://www.pcisys.net/~melanson/codecs/huffyuv.txt for a description of
21
 * the algorithm used
Michael Niedermayer's avatar
Michael Niedermayer committed
22
 */
23

Michael Niedermayer's avatar
Michael Niedermayer committed
24 25 26 27
/**
 * @file huffyuv.c
 * huffyuv codec for libavcodec.
 */
Michael Niedermayer's avatar
Michael Niedermayer committed
28 29

#include "common.h"
30
#include "bitstream.h"
Michael Niedermayer's avatar
Michael Niedermayer committed
31 32 33 34
#include "avcodec.h"
#include "dsputil.h"

#define VLC_BITS 11
Michael Niedermayer's avatar
Michael Niedermayer committed
35

36 37 38 39 40 41 42 43 44 45
#ifdef WORDS_BIGENDIAN
#define B 3
#define G 2
#define R 1
#else
#define B 0
#define G 1
#define R 2
#endif

Michael Niedermayer's avatar
Michael Niedermayer committed
46 47 48 49 50
typedef enum Predictor{
    LEFT= 0,
    PLANE,
    MEDIAN,
} Predictor;
51

Michael Niedermayer's avatar
Michael Niedermayer committed
52 53 54 55 56 57 58 59 60 61 62 63 64
typedef struct HYuvContext{
    AVCodecContext *avctx;
    Predictor predictor;
    GetBitContext gb;
    PutBitContext pb;
    int interlaced;
    int decorrelate;
    int bitstream_bpp;
    int version;
    int yuy2;                               //use yuy2 instead of 422P
    int bgr32;                              //use bgr32 instead of bgr24
    int width, height;
    int flags;
65
    int context;
Michael Niedermayer's avatar
Michael Niedermayer committed
66
    int picture_number;
67
    int last_slice_end;
68
    uint8_t *temp[3];
Michael Niedermayer's avatar
Michael Niedermayer committed
69 70 71 72
    uint64_t stats[3][256];
    uint8_t len[3][256];
    uint32_t bits[3][256];
    VLC vlc[3];
73
    AVFrame picture;
74 75
    uint8_t *bitstream_buffer;
    int bitstream_buffer_size;
76
    DSPContext dsp;
Michael Niedermayer's avatar
Michael Niedermayer committed
77 78
}HYuvContext;

Zdenek Kabelac's avatar
Zdenek Kabelac committed
79
static const unsigned char classic_shift_luma[] = {
80 81 82 83 84
  34,36,35,69,135,232,9,16,10,24,11,23,12,16,13,10,14,8,15,8,
  16,8,17,20,16,10,207,206,205,236,11,8,10,21,9,23,8,8,199,70,
  69,68, 0
};

Zdenek Kabelac's avatar
Zdenek Kabelac committed
85
static const unsigned char classic_shift_chroma[] = {
86 87 88 89 90
  66,36,37,38,39,40,41,75,76,77,110,239,144,81,82,83,84,85,118,183,
  56,57,88,89,56,89,154,57,58,57,26,141,57,56,58,57,58,57,184,119,
  214,245,116,83,82,49,80,79,78,77,44,75,41,40,39,38,37,36,34, 0
};

Zdenek Kabelac's avatar
Zdenek Kabelac committed
91
static const unsigned char classic_add_luma[256] = {
92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109
    3,  9,  5, 12, 10, 35, 32, 29, 27, 50, 48, 45, 44, 41, 39, 37,
   73, 70, 68, 65, 64, 61, 58, 56, 53, 50, 49, 46, 44, 41, 38, 36,
   68, 65, 63, 61, 58, 55, 53, 51, 48, 46, 45, 43, 41, 39, 38, 36,
   35, 33, 32, 30, 29, 27, 26, 25, 48, 47, 46, 44, 43, 41, 40, 39,
   37, 36, 35, 34, 32, 31, 30, 28, 27, 26, 24, 23, 22, 20, 19, 37,
   35, 34, 33, 31, 30, 29, 27, 26, 24, 23, 21, 20, 18, 17, 15, 29,
   27, 26, 24, 22, 21, 19, 17, 16, 14, 26, 25, 23, 21, 19, 18, 16,
   15, 27, 25, 23, 21, 19, 17, 16, 14, 26, 25, 23, 21, 18, 17, 14,
   12, 17, 19, 13,  4,  9,  2, 11,  1,  7,  8,  0, 16,  3, 14,  6,
   12, 10,  5, 15, 18, 11, 10, 13, 15, 16, 19, 20, 22, 24, 27, 15,
   18, 20, 22, 24, 26, 14, 17, 20, 22, 24, 27, 15, 18, 20, 23, 25,
   28, 16, 19, 22, 25, 28, 32, 36, 21, 25, 29, 33, 38, 42, 45, 49,
   28, 31, 34, 37, 40, 42, 44, 47, 49, 50, 52, 54, 56, 57, 59, 60,
   62, 64, 66, 67, 69, 35, 37, 39, 40, 42, 43, 45, 47, 48, 51, 52,
   54, 55, 57, 59, 60, 62, 63, 66, 67, 69, 71, 72, 38, 40, 42, 43,
   46, 47, 49, 51, 26, 28, 30, 31, 33, 34, 18, 19, 11, 13,  7,  8,
};

Zdenek Kabelac's avatar
Zdenek Kabelac committed
110
static const unsigned char classic_add_chroma[256] = {
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
    3,  1,  2,  2,  2,  2,  3,  3,  7,  5,  7,  5,  8,  6, 11,  9,
    7, 13, 11, 10,  9,  8,  7,  5,  9,  7,  6,  4,  7,  5,  8,  7,
   11,  8, 13, 11, 19, 15, 22, 23, 20, 33, 32, 28, 27, 29, 51, 77,
   43, 45, 76, 81, 46, 82, 75, 55, 56,144, 58, 80, 60, 74,147, 63,
  143, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79,
   80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 27, 30, 21, 22,
   17, 14,  5,  6,100, 54, 47, 50, 51, 53,106,107,108,109,110,111,
  112,113,114,115,  4,117,118, 92, 94,121,122,  3,124,103,  2,  1,
    0,129,130,131,120,119,126,125,136,137,138,139,140,141,142,134,
  135,132,133,104, 64,101, 62, 57,102, 95, 93, 59, 61, 28, 97, 96,
   52, 49, 48, 29, 32, 25, 24, 46, 23, 98, 45, 44, 43, 20, 42, 41,
   19, 18, 99, 40, 15, 39, 38, 16, 13, 12, 11, 37, 10,  9,  8, 36,
    7,128,127,105,123,116, 35, 34, 33,145, 31, 79, 42,146, 78, 26,
   83, 48, 49, 50, 44, 47, 26, 31, 30, 18, 17, 19, 21, 24, 25, 13,
   14, 16, 17, 18, 20, 21, 12, 14, 15,  9, 10,  6,  9,  6,  5,  8,
    6, 12,  8, 10,  7,  9,  6,  4,  6,  2,  2,  3,  3,  3,  3,  2,
};

Michael Niedermayer's avatar
Michael Niedermayer committed
129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
static inline int add_left_prediction(uint8_t *dst, uint8_t *src, int w, int acc){
    int i;

    for(i=0; i<w-1; i++){
        acc+= src[i];
        dst[i]= acc;
        i++;
        acc+= src[i];
        dst[i]= acc;
    }

    for(; i<w; i++){
        acc+= src[i];
        dst[i]= acc;
    }

    return acc;
}

static inline void add_median_prediction(uint8_t *dst, uint8_t *src1, uint8_t *diff, int w, int *left, int *left_top){
    int i;
    uint8_t l, lt;

    l= *left;
    lt= *left_top;

    for(i=0; i<w; i++){
        l= mid_pred(l, src1[i], (l + src1[i] - lt)&0xFF) + diff[i];
        lt= src1[i];
        dst[i]= l;
159
    }
Michael Niedermayer's avatar
Michael Niedermayer committed
160 161 162 163

    *left= l;
    *left_top= lt;
}
164

Michael Niedermayer's avatar
Michael Niedermayer committed
165 166 167 168 169 170 171 172
static inline void add_left_prediction_bgr32(uint8_t *dst, uint8_t *src, int w, int *red, int *green, int *blue){
    int i;
    int r,g,b;
    r= *red;
    g= *green;
    b= *blue;

    for(i=0; i<w; i++){
173 174 175
        b+= src[4*i+B];
        g+= src[4*i+G];
        r+= src[4*i+R];
176

177 178 179
        dst[4*i+B]= b;
        dst[4*i+G]= g;
        dst[4*i+R]= r;
Michael Niedermayer's avatar
Michael Niedermayer committed
180 181 182 183 184 185 186
    }

    *red= r;
    *green= g;
    *blue= b;
}

187
static inline int sub_left_prediction(HYuvContext *s, uint8_t *dst, uint8_t *src, int w, int left){
Michael Niedermayer's avatar
Michael Niedermayer committed
188
    int i;
189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
    if(w<32){
        for(i=0; i<w; i++){
            const int temp= src[i];
            dst[i]= temp - left;
            left= temp;
        }
        return left;
    }else{
        for(i=0; i<16; i++){
            const int temp= src[i];
            dst[i]= temp - left;
            left= temp;
        }
        s->dsp.diff_bytes(dst+16, src+16, src+15, w-16);
        return src[w-1];
Michael Niedermayer's avatar
Michael Niedermayer committed
204 205
    }
}
206

Michael Niedermayer's avatar
Michael Niedermayer committed
207 208
static void read_len_table(uint8_t *dst, GetBitContext *gb){
    int i, val, repeat;
209

Michael Niedermayer's avatar
Michael Niedermayer committed
210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226
    for(i=0; i<256;){
        repeat= get_bits(gb, 3);
        val   = get_bits(gb, 5);
        if(repeat==0)
            repeat= get_bits(gb, 8);
//printf("%d %d\n", val, repeat);
        while (repeat--)
            dst[i++] = val;
    }
}

static int generate_bits_table(uint32_t *dst, uint8_t *len_table){
    int len, index;
    uint32_t bits=0;

    for(len=32; len>0; len--){
        for(index=0; index<256; index++){
Michael Niedermayer's avatar
Michael Niedermayer committed
227 228
            if(len_table[index]==len)
                dst[index]= bits++;
Michael Niedermayer's avatar
Michael Niedermayer committed
229
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
230
        if(bits & 1){
231
            av_log(NULL, AV_LOG_ERROR, "Error generating huffman table\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
232 233 234
            return -1;
        }
        bits >>= 1;
Michael Niedermayer's avatar
Michael Niedermayer committed
235 236 237 238 239 240 241 242
    }
    return 0;
}

static void generate_len_table(uint8_t *dst, uint64_t *stats, int size){
    uint64_t counts[2*size];
    int up[2*size];
    int offset, i, next;
243

Michael Niedermayer's avatar
Michael Niedermayer committed
244 245 246 247
    for(offset=1; ; offset<<=1){
        for(i=0; i<size; i++){
            counts[i]= stats[i] + offset - 1;
        }
248

Michael Niedermayer's avatar
Michael Niedermayer committed
249 250 251
        for(next=size; next<size*2; next++){
            uint64_t min1, min2;
            int min1_i, min2_i;
252

Michael Niedermayer's avatar
Michael Niedermayer committed
253 254
            min1=min2= INT64_MAX;
            min1_i= min2_i=-1;
255

Michael Niedermayer's avatar
Michael Niedermayer committed
256 257 258 259 260 261 262 263 264 265 266 267 268
            for(i=0; i<next; i++){
                if(min2 > counts[i]){
                    if(min1 > counts[i]){
                        min2= min1;
                        min2_i= min1_i;
                        min1= counts[i];
                        min1_i= i;
                    }else{
                        min2= counts[i];
                        min2_i= i;
                    }
                }
            }
269

Michael Niedermayer's avatar
Michael Niedermayer committed
270
            if(min2==INT64_MAX) break;
271

Michael Niedermayer's avatar
Michael Niedermayer committed
272 273
            counts[next]= min1 + min2;
            counts[min1_i]=
Michael Niedermayer's avatar
Michael Niedermayer committed
274
            counts[min2_i]= INT64_MAX;
Michael Niedermayer's avatar
Michael Niedermayer committed
275 276 277 278
            up[min1_i]=
            up[min2_i]= next;
            up[next]= -1;
        }
279

Michael Niedermayer's avatar
Michael Niedermayer committed
280 281 282
        for(i=0; i<size; i++){
            int len;
            int index=i;
283

Michael Niedermayer's avatar
Michael Niedermayer committed
284 285
            for(len=0; up[index] != -1; len++)
                index= up[index];
286

Michael Niedermayer's avatar
Michael Niedermayer committed
287
            if(len >= 32) break;
288

Michael Niedermayer's avatar
Michael Niedermayer committed
289 290 291 292 293 294 295 296 297
            dst[i]= len;
        }
        if(i==size) break;
    }
}

static int read_huffman_tables(HYuvContext *s, uint8_t *src, int length){
    GetBitContext gb;
    int i;
298

299
    init_get_bits(&gb, src, length*8);
300

Michael Niedermayer's avatar
Michael Niedermayer committed
301 302
    for(i=0; i<3; i++){
        read_len_table(s->len[i], &gb);
303

Michael Niedermayer's avatar
Michael Niedermayer committed
304 305 306 307 308 309 310 311
        if(generate_bits_table(s->bits[i], s->len[i])<0){
            return -1;
        }
#if 0
for(j=0; j<256; j++){
printf("%6X, %2d,  %3d\n", s->bits[i][j], s->len[i][j], j);
}
#endif
312
        free_vlc(&s->vlc[i]);
313
        init_vlc(&s->vlc[i], VLC_BITS, 256, s->len[i], 1, 1, s->bits[i], 4, 4, 0);
Michael Niedermayer's avatar
Michael Niedermayer committed
314
    }
315

316
    return (get_bits_count(&gb)+7)/8;
Michael Niedermayer's avatar
Michael Niedermayer committed
317 318 319
}

static int read_old_huffman_tables(HYuvContext *s){
320
#if 1
Michael Niedermayer's avatar
Michael Niedermayer committed
321 322 323
    GetBitContext gb;
    int i;

324
    init_get_bits(&gb, classic_shift_luma, sizeof(classic_shift_luma)*8);
Michael Niedermayer's avatar
Michael Niedermayer committed
325
    read_len_table(s->len[0], &gb);
326
    init_get_bits(&gb, classic_shift_chroma, sizeof(classic_shift_chroma)*8);
Michael Niedermayer's avatar
Michael Niedermayer committed
327
    read_len_table(s->len[1], &gb);
328

Michael Niedermayer's avatar
Michael Niedermayer committed
329 330 331 332 333 334 335 336 337
    for(i=0; i<256; i++) s->bits[0][i] = classic_add_luma  [i];
    for(i=0; i<256; i++) s->bits[1][i] = classic_add_chroma[i];

    if(s->bitstream_bpp >= 24){
        memcpy(s->bits[1], s->bits[0], 256*sizeof(uint32_t));
        memcpy(s->len[1] , s->len [0], 256*sizeof(uint8_t));
    }
    memcpy(s->bits[2], s->bits[1], 256*sizeof(uint32_t));
    memcpy(s->len[2] , s->len [1], 256*sizeof(uint8_t));
338

339 340
    for(i=0; i<3; i++){
        free_vlc(&s->vlc[i]);
341
        init_vlc(&s->vlc[i], VLC_BITS, 256, s->len[i], 1, 1, s->bits[i], 4, 4, 0);
342
    }
343

Michael Niedermayer's avatar
Michael Niedermayer committed
344 345 346 347 348 349 350
    return 0;
#else
    fprintf(stderr, "v1 huffyuv is not supported \n");
    return -1;
#endif
}

Michael Niedermayer's avatar
Michael Niedermayer committed
351 352
static void alloc_temp(HYuvContext *s){
    int i;
353

Michael Niedermayer's avatar
Michael Niedermayer committed
354 355 356 357 358 359 360 361 362
    if(s->bitstream_bpp<24){
        for(i=0; i<3; i++){
            s->temp[i]= av_malloc(s->width + 16);
        }
    }else{
        s->temp[0]= av_malloc(4*s->width + 16);
    }
}

363
static int common_init(AVCodecContext *avctx){
Michael Niedermayer's avatar
Michael Niedermayer committed
364 365 366 367
    HYuvContext *s = avctx->priv_data;

    s->avctx= avctx;
    s->flags= avctx->flags;
368

Michael Niedermayer's avatar
Michael Niedermayer committed
369
    dsputil_init(&s->dsp, avctx);
370

371 372 373
    s->width= avctx->width;
    s->height= avctx->height;
    assert(s->width>0 && s->height>0);
374

375 376 377 378 379 380 381 382
    return 0;
}

static int decode_init(AVCodecContext *avctx)
{
    HYuvContext *s = avctx->priv_data;

    common_init(avctx);
383
    memset(s->vlc, 0, 3*sizeof(VLC));
384

385
    avctx->coded_frame= &s->picture;
386
    s->interlaced= s->height > 288;
Michael Niedermayer's avatar
Michael Niedermayer committed
387

Michael Niedermayer's avatar
Michael Niedermayer committed
388 389 390 391
s->bgr32=1;
//if(avctx->extradata)
//  printf("extradata:%X, extradata_size:%d\n", *(uint32_t*)avctx->extradata, avctx->extradata_size);
    if(avctx->extradata_size){
392
        if((avctx->bits_per_sample&7) && avctx->bits_per_sample != 12)
Michael Niedermayer's avatar
Michael Niedermayer committed
393 394 395 396 397
            s->version=1; // do such files exist at all?
        else
            s->version=2;
    }else
        s->version=0;
398

Michael Niedermayer's avatar
Michael Niedermayer committed
399
    if(s->version==2){
400
        int method, interlace;
Michael Niedermayer's avatar
Michael Niedermayer committed
401 402 403 404 405

        method= ((uint8_t*)avctx->extradata)[0];
        s->decorrelate= method&64 ? 1 : 0;
        s->predictor= method&63;
        s->bitstream_bpp= ((uint8_t*)avctx->extradata)[1];
406
        if(s->bitstream_bpp==0)
Michael Niedermayer's avatar
Michael Niedermayer committed
407
            s->bitstream_bpp= avctx->bits_per_sample&~7;
408 409
        interlace= (((uint8_t*)avctx->extradata)[2] & 0x30) >> 4;
        s->interlaced= (interlace==1) ? 1 : (interlace==2) ? 0 : s->interlaced;
410
        s->context= ((uint8_t*)avctx->extradata)[2] & 0x40 ? 1 : 0;
411

Michael Niedermayer's avatar
Michael Niedermayer committed
412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437
        if(read_huffman_tables(s, ((uint8_t*)avctx->extradata)+4, avctx->extradata_size) < 0)
            return -1;
    }else{
        switch(avctx->bits_per_sample&7){
        case 1:
            s->predictor= LEFT;
            s->decorrelate= 0;
            break;
        case 2:
            s->predictor= LEFT;
            s->decorrelate= 1;
            break;
        case 3:
            s->predictor= PLANE;
            s->decorrelate= avctx->bits_per_sample >= 24;
            break;
        case 4:
            s->predictor= MEDIAN;
            s->decorrelate= 0;
            break;
        default:
            s->predictor= LEFT; //OLD
            s->decorrelate= 0;
            break;
        }
        s->bitstream_bpp= avctx->bits_per_sample & ~7;
438
        s->context= 0;
439

Michael Niedermayer's avatar
Michael Niedermayer committed
440 441 442
        if(read_old_huffman_tables(s) < 0)
            return -1;
    }
443

Michael Niedermayer's avatar
Michael Niedermayer committed
444 445 446 447 448 449 450 451 452 453 454 455 456 457
    switch(s->bitstream_bpp){
    case 12:
        avctx->pix_fmt = PIX_FMT_YUV420P;
        break;
    case 16:
        if(s->yuy2){
            avctx->pix_fmt = PIX_FMT_YUV422;
        }else{
            avctx->pix_fmt = PIX_FMT_YUV422P;
        }
        break;
    case 24:
    case 32:
        if(s->bgr32){
458
            avctx->pix_fmt = PIX_FMT_RGBA32;
Michael Niedermayer's avatar
Michael Niedermayer committed
459 460 461 462 463 464 465
        }else{
            avctx->pix_fmt = PIX_FMT_BGR24;
        }
        break;
    default:
        assert(0);
    }
466

Michael Niedermayer's avatar
Michael Niedermayer committed
467
    alloc_temp(s);
468

469 470
//    av_log(NULL, AV_LOG_DEBUG, "pred:%d bpp:%d hbpp:%d il:%d\n", s->predictor, s->bitstream_bpp, avctx->bits_per_sample, s->interlaced);

Michael Niedermayer's avatar
Michael Niedermayer committed
471 472 473
    return 0;
}

474
static int store_table(HYuvContext *s, uint8_t *len, uint8_t *buf){
Michael Niedermayer's avatar
Michael Niedermayer committed
475
    int i;
476
    int index= 0;
Michael Niedermayer's avatar
Michael Niedermayer committed
477 478 479

    for(i=0; i<256;){
        int val= len[i];
480
        int repeat=0;
481

482 483
        for(; i<256 && len[i]==val && repeat<255; i++)
            repeat++;
484

485
        assert(val < 32 && val >0 && repeat<256 && repeat>0);
Michael Niedermayer's avatar
Michael Niedermayer committed
486
        if(repeat>7){
487 488
            buf[index++]= val;
            buf[index++]= repeat;
Michael Niedermayer's avatar
Michael Niedermayer committed
489
        }else{
490
            buf[index++]= val | (repeat<<5);
Michael Niedermayer's avatar
Michael Niedermayer committed
491 492
        }
    }
493

494
    return index;
Michael Niedermayer's avatar
Michael Niedermayer committed
495 496 497 498 499
}

static int encode_init(AVCodecContext *avctx)
{
    HYuvContext *s = avctx->priv_data;
500
    int i, j;
Michael Niedermayer's avatar
Michael Niedermayer committed
501

502
    common_init(avctx);
503

504 505
    avctx->extradata= av_mallocz(1024*30); // 256*3+4 == 772
    avctx->stats_out= av_mallocz(1024*30); // 21*256*3(%llu ) + 3(\n) + 1(0) = 16132
Michael Niedermayer's avatar
Michael Niedermayer committed
506
    s->version=2;
507

508
    avctx->coded_frame= &s->picture;
509

Michael Niedermayer's avatar
Michael Niedermayer committed
510
    switch(avctx->pix_fmt){
511 512 513
    case PIX_FMT_YUV420P:
        s->bitstream_bpp= 12;
        break;
Michael Niedermayer's avatar
Michael Niedermayer committed
514 515 516 517
    case PIX_FMT_YUV422P:
        s->bitstream_bpp= 16;
        break;
    default:
518
        av_log(avctx, AV_LOG_ERROR, "format not supported\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
519 520 521 522 523
        return -1;
    }
    avctx->bits_per_sample= s->bitstream_bpp;
    s->decorrelate= s->bitstream_bpp >= 24;
    s->predictor= avctx->prediction_method;
524
    s->interlaced= avctx->flags&CODEC_FLAG_INTERLACED_ME ? 1 : 0;
525 526 527 528 529 530 531
    if(avctx->context_model==1){
        s->context= avctx->context_model;
        if(s->flags & (CODEC_FLAG_PASS1|CODEC_FLAG_PASS2)){
            av_log(avctx, AV_LOG_ERROR, "context=1 is not compatible with 2 pass huffyuv encoding\n");
            return -1;
        }
    }else s->context= 0;
532

533 534 535 536 537 538 539 540 541
    if(avctx->codec->id==CODEC_ID_HUFFYUV){
        if(avctx->pix_fmt==PIX_FMT_YUV420P){
            av_log(avctx, AV_LOG_ERROR, "Error: YV12 is not supported by huffyuv; use vcodec=ffvhuff or format=422p\n");
            return -1;
        }
        if(avctx->context_model){
            av_log(avctx, AV_LOG_ERROR, "Error: per-frame huffman tables are not supported by huffyuv; use vcodec=ffvhuff\n");
            return -1;
        }
542
        if(s->interlaced != ( s->height > 288 ))
543
            av_log(avctx, AV_LOG_INFO, "using huffyuv 2.2.0 or newer interlacing flag\n");
544 545
    }else if(avctx->strict_std_compliance>FF_COMPLIANCE_EXPERIMENTAL){
        av_log(avctx, AV_LOG_ERROR, "This codec is under development; files encoded with it may not be decodable with future versions!!! Set vstrict=-2 / -strict -2 to use it anyway.\n");
546 547
        return -1;
    }
548

Michael Niedermayer's avatar
Michael Niedermayer committed
549 550
    ((uint8_t*)avctx->extradata)[0]= s->predictor;
    ((uint8_t*)avctx->extradata)[1]= s->bitstream_bpp;
551
    ((uint8_t*)avctx->extradata)[2]= s->interlaced ? 0x10 : 0x20;
552 553
    if(s->context)
        ((uint8_t*)avctx->extradata)[2]|= 0x40;
Michael Niedermayer's avatar
Michael Niedermayer committed
554 555
    ((uint8_t*)avctx->extradata)[3]= 0;
    s->avctx->extradata_size= 4;
556

Michael Niedermayer's avatar
Michael Niedermayer committed
557 558
    if(avctx->stats_in){
        char *p= avctx->stats_in;
559

Michael Niedermayer's avatar
Michael Niedermayer committed
560 561 562 563 564 565 566 567 568 569 570 571
        for(i=0; i<3; i++)
            for(j=0; j<256; j++)
                s->stats[i][j]= 1;

        for(;;){
            for(i=0; i<3; i++){
                char *next;

                for(j=0; j<256; j++){
                    s->stats[i][j]+= strtol(p, &next, 0);
                    if(next==p) return -1;
                    p=next;
572
                }
Michael Niedermayer's avatar
Michael Niedermayer committed
573 574 575 576 577 578 579
            }
            if(p[0]==0 || p[1]==0 || p[2]==0) break;
        }
    }else{
        for(i=0; i<3; i++)
            for(j=0; j<256; j++){
                int d= FFMIN(j, 256-j);
580

Michael Niedermayer's avatar
Michael Niedermayer committed
581 582 583
                s->stats[i][j]= 100000000/(d+1);
            }
    }
584

Michael Niedermayer's avatar
Michael Niedermayer committed
585 586 587 588 589 590
    for(i=0; i<3; i++){
        generate_len_table(s->len[i], s->stats[i], 256);

        if(generate_bits_table(s->bits[i], s->len[i])<0){
            return -1;
        }
591

592 593
        s->avctx->extradata_size+=
        store_table(s, s->len[i], &((uint8_t*)s->avctx->extradata)[s->avctx->extradata_size]);
Michael Niedermayer's avatar
Michael Niedermayer committed
594 595
    }

596 597
    if(s->context){
        for(i=0; i<3; i++){
598
            int pels = s->width*s->height / (i?40:10);
599 600 601 602 603 604 605 606 607 608
            for(j=0; j<256; j++){
                int d= FFMIN(j, 256-j);
                s->stats[i][j]= pels/(d+1);
            }
        }
    }else{
        for(i=0; i<3; i++)
            for(j=0; j<256; j++)
                s->stats[i][j]= 0;
    }
609

Michael Niedermayer's avatar
Michael Niedermayer committed
610
//    printf("pred:%d bpp:%d hbpp:%d il:%d\n", s->predictor, s->bitstream_bpp, avctx->bits_per_sample, s->interlaced);
611

Michael Niedermayer's avatar
Michael Niedermayer committed
612 613
    alloc_temp(s);

Michael Niedermayer's avatar
Michael Niedermayer committed
614
    s->picture_number=0;
615

Michael Niedermayer's avatar
Michael Niedermayer committed
616 617 618 619 620
    return 0;
}

static void decode_422_bitstream(HYuvContext *s, int count){
    int i;
621

Michael Niedermayer's avatar
Michael Niedermayer committed
622
    count/=2;
623

Michael Niedermayer's avatar
Michael Niedermayer committed
624
    for(i=0; i<count; i++){
625 626 627 628
        s->temp[0][2*i  ]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3);
        s->temp[1][  i  ]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3);
        s->temp[0][2*i+1]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3);
        s->temp[2][  i  ]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3);
Michael Niedermayer's avatar
Michael Niedermayer committed
629 630 631
    }
}

632 633
static void decode_gray_bitstream(HYuvContext *s, int count){
    int i;
634

635
    count/=2;
636

637
    for(i=0; i<count; i++){
638 639
        s->temp[0][2*i  ]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3);
        s->temp[0][2*i+1]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3);
640 641 642
    }
}

643
static int encode_422_bitstream(HYuvContext *s, int count){
Michael Niedermayer's avatar
Michael Niedermayer committed
644
    int i;
645

646 647 648 649
    if(s->pb.buf_end - s->pb.buf - (put_bits_count(&s->pb)>>3) < 2*4*count){
        av_log(s->avctx, AV_LOG_ERROR, "encoded frame too large\n");
        return -1;
    }
650

Michael Niedermayer's avatar
Michael Niedermayer committed
651 652 653 654 655 656 657 658
    count/=2;
    if(s->flags&CODEC_FLAG_PASS1){
        for(i=0; i<count; i++){
            s->stats[0][ s->temp[0][2*i  ] ]++;
            s->stats[1][ s->temp[1][  i  ] ]++;
            s->stats[0][ s->temp[0][2*i+1] ]++;
            s->stats[2][ s->temp[2][  i  ] ]++;
        }
659 660 661 662
    }
    if(s->avctx->flags2&CODEC_FLAG2_NO_OUTPUT)
        return 0;
    if(s->context){
663 664 665 666 667 668 669 670 671 672
        for(i=0; i<count; i++){
            s->stats[0][ s->temp[0][2*i  ] ]++;
            put_bits(&s->pb, s->len[0][ s->temp[0][2*i  ] ], s->bits[0][ s->temp[0][2*i  ] ]);
            s->stats[1][ s->temp[1][  i  ] ]++;
            put_bits(&s->pb, s->len[1][ s->temp[1][  i  ] ], s->bits[1][ s->temp[1][  i  ] ]);
            s->stats[0][ s->temp[0][2*i+1] ]++;
            put_bits(&s->pb, s->len[0][ s->temp[0][2*i+1] ], s->bits[0][ s->temp[0][2*i+1] ]);
            s->stats[2][ s->temp[2][  i  ] ]++;
            put_bits(&s->pb, s->len[2][ s->temp[2][  i  ] ], s->bits[2][ s->temp[2][  i  ] ]);
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
673 674 675 676 677 678 679 680
    }else{
        for(i=0; i<count; i++){
            put_bits(&s->pb, s->len[0][ s->temp[0][2*i  ] ], s->bits[0][ s->temp[0][2*i  ] ]);
            put_bits(&s->pb, s->len[1][ s->temp[1][  i  ] ], s->bits[1][ s->temp[1][  i  ] ]);
            put_bits(&s->pb, s->len[0][ s->temp[0][2*i+1] ], s->bits[0][ s->temp[0][2*i+1] ]);
            put_bits(&s->pb, s->len[2][ s->temp[2][  i  ] ], s->bits[2][ s->temp[2][  i  ] ]);
        }
    }
681
    return 0;
Michael Niedermayer's avatar
Michael Niedermayer committed
682 683
}

684
static int encode_gray_bitstream(HYuvContext *s, int count){
685
    int i;
686

687 688 689 690 691
    if(s->pb.buf_end - s->pb.buf - (put_bits_count(&s->pb)>>3) < 4*count){
        av_log(s->avctx, AV_LOG_ERROR, "encoded frame too large\n");
        return -1;
    }

692 693 694 695 696 697
    count/=2;
    if(s->flags&CODEC_FLAG_PASS1){
        for(i=0; i<count; i++){
            s->stats[0][ s->temp[0][2*i  ] ]++;
            s->stats[0][ s->temp[0][2*i+1] ]++;
        }
698 699 700
    }
    if(s->avctx->flags2&CODEC_FLAG2_NO_OUTPUT)
        return 0;
701

702
    if(s->context){
703 704 705 706 707 708
        for(i=0; i<count; i++){
            s->stats[0][ s->temp[0][2*i  ] ]++;
            put_bits(&s->pb, s->len[0][ s->temp[0][2*i  ] ], s->bits[0][ s->temp[0][2*i  ] ]);
            s->stats[0][ s->temp[0][2*i+1] ]++;
            put_bits(&s->pb, s->len[0][ s->temp[0][2*i+1] ], s->bits[0][ s->temp[0][2*i+1] ]);
        }
709 710 711 712 713 714
    }else{
        for(i=0; i<count; i++){
            put_bits(&s->pb, s->len[0][ s->temp[0][2*i  ] ], s->bits[0][ s->temp[0][2*i  ] ]);
            put_bits(&s->pb, s->len[0][ s->temp[0][2*i+1] ], s->bits[0][ s->temp[0][2*i+1] ]);
        }
    }
715
    return 0;
716 717
}

Michael Niedermayer's avatar
Michael Niedermayer committed
718 719
static void decode_bgr_bitstream(HYuvContext *s, int count){
    int i;
720

Michael Niedermayer's avatar
Michael Niedermayer committed
721 722 723
    if(s->decorrelate){
        if(s->bitstream_bpp==24){
            for(i=0; i<count; i++){
724
                s->temp[0][4*i+G]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3);
725 726
                s->temp[0][4*i+B]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3) + s->temp[0][4*i+G];
                s->temp[0][4*i+R]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3) + s->temp[0][4*i+G];
Michael Niedermayer's avatar
Michael Niedermayer committed
727 728 729
            }
        }else{
            for(i=0; i<count; i++){
730
                s->temp[0][4*i+G]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3);
731
                s->temp[0][4*i+B]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3) + s->temp[0][4*i+G];
732
                s->temp[0][4*i+R]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3) + s->temp[0][4*i+G];
Michael Niedermayer's avatar
Michael Niedermayer committed
733 734 735 736 737 738
                                   get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3); //?!
            }
        }
    }else{
        if(s->bitstream_bpp==24){
            for(i=0; i<count; i++){
739
                s->temp[0][4*i+B]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3);
740 741
                s->temp[0][4*i+G]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3);
                s->temp[0][4*i+R]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3);
Michael Niedermayer's avatar
Michael Niedermayer committed
742 743 744
            }
        }else{
            for(i=0; i<count; i++){
745
                s->temp[0][4*i+B]= get_vlc2(&s->gb, s->vlc[0].table, VLC_BITS, 3);
746 747
                s->temp[0][4*i+G]= get_vlc2(&s->gb, s->vlc[1].table, VLC_BITS, 3);
                s->temp[0][4*i+R]= get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3);
Michael Niedermayer's avatar
Michael Niedermayer committed
748 749 750 751 752 753
                                   get_vlc2(&s->gb, s->vlc[2].table, VLC_BITS, 3); //?!
            }
        }
    }
}

754 755
static void draw_slice(HYuvContext *s, int y){
    int h, cy;
Michael Niedermayer's avatar
Michael Niedermayer committed
756
    int offset[4];
757 758

    if(s->avctx->draw_horiz_band==NULL)
759
        return;
760

761 762
    h= y - s->last_slice_end;
    y -= h;
763

764 765 766 767 768
    if(s->bitstream_bpp==12){
        cy= y>>1;
    }else{
        cy= y;
    }
Michael Niedermayer's avatar
Michael Niedermayer committed
769 770 771 772 773

    offset[0] = s->picture.linesize[0]*y;
    offset[1] = s->picture.linesize[1]*cy;
    offset[2] = s->picture.linesize[2]*cy;
    offset[3] = 0;
774 775
    emms_c();

776
    s->avctx->draw_horiz_band(s->avctx, &s->picture, offset, y, 3, h);
777

778 779 780
    s->last_slice_end= y + h;
}

Michael Niedermayer's avatar
Michael Niedermayer committed
781 782 783 784 785
static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, uint8_t *buf, int buf_size){
    HYuvContext *s = avctx->priv_data;
    const int width= s->width;
    const int width2= s->width>>1;
    const int height= s->height;
Michael Niedermayer's avatar
Michael Niedermayer committed
786
    int fake_ystride, fake_ustride, fake_vstride;
787
    AVFrame * const p= &s->picture;
788
    int table_size= 0;
Michael Niedermayer's avatar
Michael Niedermayer committed
789

790
    AVFrame *picture = data;
Michael Niedermayer's avatar
Michael Niedermayer committed
791

792
    s->bitstream_buffer= av_fast_realloc(s->bitstream_buffer, &s->bitstream_buffer_size, buf_size + FF_INPUT_BUFFER_PADDING_SIZE);
Michael Niedermayer's avatar
Michael Niedermayer committed
793

Michael Niedermayer's avatar
Michael Niedermayer committed
794
    s->dsp.bswap_buf((uint32_t*)s->bitstream_buffer, (uint32_t*)buf, buf_size/4);
795

796 797 798
    if(p->data[0])
        avctx->release_buffer(avctx, p);

Michael Niedermayer's avatar
Michael Niedermayer committed
799 800
    p->reference= 0;
    if(avctx->get_buffer(avctx, p) < 0){
801
        av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
802
        return -1;
Michael Niedermayer's avatar
Michael Niedermayer committed
803
    }
804

805 806 807 808 809 810 811
    if(s->context){
        table_size = read_huffman_tables(s, s->bitstream_buffer, buf_size);
        if(table_size < 0)
            return -1;
    }

    init_get_bits(&s->gb, s->bitstream_buffer+table_size, (buf_size-table_size)*8);
Michael Niedermayer's avatar
Michael Niedermayer committed
812 813 814 815

    fake_ystride= s->interlaced ? p->linesize[0]*2  : p->linesize[0];
    fake_ustride= s->interlaced ? p->linesize[1]*2  : p->linesize[1];
    fake_vstride= s->interlaced ? p->linesize[2]*2  : p->linesize[2];
816

817
    s->last_slice_end= 0;
818

Michael Niedermayer's avatar
Michael Niedermayer committed
819
    if(s->bitstream_bpp<24){
820
        int y, cy;
Michael Niedermayer's avatar
Michael Niedermayer committed
821 822
        int lefty, leftu, leftv;
        int lefttopy, lefttopu, lefttopv;
823

Michael Niedermayer's avatar
Michael Niedermayer committed
824
        if(s->yuy2){
Michael Niedermayer's avatar
Michael Niedermayer committed
825 826 827 828
            p->data[0][3]= get_bits(&s->gb, 8);
            p->data[0][2]= get_bits(&s->gb, 8);
            p->data[0][1]= get_bits(&s->gb, 8);
            p->data[0][0]= get_bits(&s->gb, 8);
829

830
            av_log(avctx, AV_LOG_ERROR, "YUY2 output is not implemented yet\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
831 832
            return -1;
        }else{
833

Michael Niedermayer's avatar
Michael Niedermayer committed
834 835 836 837
            leftv= p->data[2][0]= get_bits(&s->gb, 8);
            lefty= p->data[0][1]= get_bits(&s->gb, 8);
            leftu= p->data[1][0]= get_bits(&s->gb, 8);
                   p->data[0][0]= get_bits(&s->gb, 8);
838

Michael Niedermayer's avatar
Michael Niedermayer committed
839 840 841 842
            switch(s->predictor){
            case LEFT:
            case PLANE:
                decode_422_bitstream(s, width-2);
Michael Niedermayer's avatar
Michael Niedermayer committed
843
                lefty= add_left_prediction(p->data[0] + 2, s->temp[0], width-2, lefty);
Michael Niedermayer's avatar
Michael Niedermayer committed
844
                if(!(s->flags&CODEC_FLAG_GRAY)){
Michael Niedermayer's avatar
Michael Niedermayer committed
845 846
                    leftu= add_left_prediction(p->data[1] + 1, s->temp[1], width2-1, leftu);
                    leftv= add_left_prediction(p->data[2] + 1, s->temp[2], width2-1, leftv);
Michael Niedermayer's avatar
Michael Niedermayer committed
847 848
                }

849
                for(cy=y=1; y<s->height; y++,cy++){
Michael Niedermayer's avatar
Michael Niedermayer committed
850
                    uint8_t *ydst, *udst, *vdst;
851

852 853
                    if(s->bitstream_bpp==12){
                        decode_gray_bitstream(s, width);
854

Michael Niedermayer's avatar
Michael Niedermayer committed
855
                        ydst= p->data[0] + p->linesize[0]*y;
856 857 858 859 860 861 862 863 864

                        lefty= add_left_prediction(ydst, s->temp[0], width, lefty);
                        if(s->predictor == PLANE){
                            if(y>s->interlaced)
                                s->dsp.add_bytes(ydst, ydst - fake_ystride, width);
                        }
                        y++;
                        if(y>=s->height) break;
                    }
865

866
                    draw_slice(s, y);
867

Michael Niedermayer's avatar
Michael Niedermayer committed
868 869 870
                    ydst= p->data[0] + p->linesize[0]*y;
                    udst= p->data[1] + p->linesize[1]*cy;
                    vdst= p->data[2] + p->linesize[2]*cy;
871

872
                    decode_422_bitstream(s, width);
Michael Niedermayer's avatar
Michael Niedermayer committed
873 874 875 876 877 878
                    lefty= add_left_prediction(ydst, s->temp[0], width, lefty);
                    if(!(s->flags&CODEC_FLAG_GRAY)){
                        leftu= add_left_prediction(udst, s->temp[1], width2, leftu);
                        leftv= add_left_prediction(vdst, s->temp[2], width2, leftv);
                    }
                    if(s->predictor == PLANE){
879
                        if(cy>s->interlaced){
Michael Niedermayer's avatar
Michael Niedermayer committed
880 881 882 883 884 885 886 887
                            s->dsp.add_bytes(ydst, ydst - fake_ystride, width);
                            if(!(s->flags&CODEC_FLAG_GRAY)){
                                s->dsp.add_bytes(udst, udst - fake_ustride, width2);
                                s->dsp.add_bytes(vdst, vdst - fake_vstride, width2);
                            }
                        }
                    }
                }
888
                draw_slice(s, height);
889

Michael Niedermayer's avatar
Michael Niedermayer committed
890 891 892 893
                break;
            case MEDIAN:
                /* first line except first 2 pixels is left predicted */
                decode_422_bitstream(s, width-2);
Michael Niedermayer's avatar
Michael Niedermayer committed
894
                lefty= add_left_prediction(p->data[0] + 2, s->temp[0], width-2, lefty);
Michael Niedermayer's avatar
Michael Niedermayer committed
895
                if(!(s->flags&CODEC_FLAG_GRAY)){
Michael Niedermayer's avatar
Michael Niedermayer committed
896 897
                    leftu= add_left_prediction(p->data[1] + 1, s->temp[1], width2-1, leftu);
                    leftv= add_left_prediction(p->data[2] + 1, s->temp[2], width2-1, leftv);
Michael Niedermayer's avatar
Michael Niedermayer committed
898
                }
899

900
                cy=y=1;
901

Michael Niedermayer's avatar
Michael Niedermayer committed
902 903 904
                /* second line is left predicted for interlaced case */
                if(s->interlaced){
                    decode_422_bitstream(s, width);
Michael Niedermayer's avatar
Michael Niedermayer committed
905
                    lefty= add_left_prediction(p->data[0] + p->linesize[0], s->temp[0], width, lefty);
Michael Niedermayer's avatar
Michael Niedermayer committed
906
                    if(!(s->flags&CODEC_FLAG_GRAY)){
Michael Niedermayer's avatar
Michael Niedermayer committed
907 908
                        leftu= add_left_prediction(p->data[1] + p->linesize[2], s->temp[1], width2, leftu);
                        leftv= add_left_prediction(p->data[2] + p->linesize[1], s->temp[2], width2, leftv);
Michael Niedermayer's avatar
Michael Niedermayer committed
909
                    }
910
                    y++; cy++;
Michael Niedermayer's avatar
Michael Niedermayer committed
911 912 913 914
                }

                /* next 4 pixels are left predicted too */
                decode_422_bitstream(s, 4);
Michael Niedermayer's avatar
Michael Niedermayer committed
915
                lefty= add_left_prediction(p->data[0] + fake_ystride, s->temp[0], 4, lefty);
Michael Niedermayer's avatar
Michael Niedermayer committed
916
                if(!(s->flags&CODEC_FLAG_GRAY)){
Michael Niedermayer's avatar
Michael Niedermayer committed
917 918
                    leftu= add_left_prediction(p->data[1] + fake_ustride, s->temp[1], 2, leftu);
                    leftv= add_left_prediction(p->data[2] + fake_vstride, s->temp[2], 2, leftv);
Michael Niedermayer's avatar
Michael Niedermayer committed
919 920 921
                }

                /* next line except the first 4 pixels is median predicted */
Michael Niedermayer's avatar
Michael Niedermayer committed
922
                lefttopy= p->data[0][3];
Michael Niedermayer's avatar
Michael Niedermayer committed
923
                decode_422_bitstream(s, width-4);
Michael Niedermayer's avatar
Michael Niedermayer committed
924
                add_median_prediction(p->data[0] + fake_ystride+4, p->data[0]+4, s->temp[0], width-4, &lefty, &lefttopy);
Michael Niedermayer's avatar
Michael Niedermayer committed
925
                if(!(s->flags&CODEC_FLAG_GRAY)){
Michael Niedermayer's avatar
Michael Niedermayer committed
926 927 928 929
                    lefttopu= p->data[1][1];
                    lefttopv= p->data[2][1];
                    add_median_prediction(p->data[1] + fake_ustride+2, p->data[1]+2, s->temp[1], width2-2, &leftu, &lefttopu);
                    add_median_prediction(p->data[2] + fake_vstride+2, p->data[2]+2, s->temp[2], width2-2, &leftv, &lefttopv);
Michael Niedermayer's avatar
Michael Niedermayer committed
930
                }
931
                y++; cy++;
932

933
                for(; y<height; y++,cy++){
Michael Niedermayer's avatar
Michael Niedermayer committed
934
                    uint8_t *ydst, *udst, *vdst;
935 936 937 938

                    if(s->bitstream_bpp==12){
                        while(2*cy > y){
                            decode_gray_bitstream(s, width);
Michael Niedermayer's avatar
Michael Niedermayer committed
939
                            ydst= p->data[0] + p->linesize[0]*y;
940 941 942 943 944
                            add_median_prediction(ydst, ydst - fake_ystride, s->temp[0], width, &lefty, &lefttopy);
                            y++;
                        }
                        if(y>=height) break;
                    }
945
                    draw_slice(s, y);
946

Michael Niedermayer's avatar
Michael Niedermayer committed
947
                    decode_422_bitstream(s, width);
948

Michael Niedermayer's avatar
Michael Niedermayer committed
949 950 951
                    ydst= p->data[0] + p->linesize[0]*y;
                    udst= p->data[1] + p->linesize[1]*cy;
                    vdst= p->data[2] + p->linesize[2]*cy;
Michael Niedermayer's avatar
Michael Niedermayer committed
952 953 954 955 956 957 958

                    add_median_prediction(ydst, ydst - fake_ystride, s->temp[0], width, &lefty, &lefttopy);
                    if(!(s->flags&CODEC_FLAG_GRAY)){
                        add_median_prediction(udst, udst - fake_ustride, s->temp[1], width2, &leftu, &lefttopu);
                        add_median_prediction(vdst, vdst - fake_vstride, s->temp[2], width2, &leftv, &lefttopv);
                    }
                }
959 960

                draw_slice(s, height);
Michael Niedermayer's avatar
Michael Niedermayer committed
961 962 963 964 965 966
                break;
            }
        }
    }else{
        int y;
        int leftr, leftg, leftb;
Michael Niedermayer's avatar
Michael Niedermayer committed
967
        const int last_line= (height-1)*p->linesize[0];
968

Michael Niedermayer's avatar
Michael Niedermayer committed
969
        if(s->bitstream_bpp==32){
970 971 972 973
            skip_bits(&s->gb, 8);
            leftr= p->data[0][last_line+R]= get_bits(&s->gb, 8);
            leftg= p->data[0][last_line+G]= get_bits(&s->gb, 8);
            leftb= p->data[0][last_line+B]= get_bits(&s->gb, 8);
Michael Niedermayer's avatar
Michael Niedermayer committed
974
        }else{
975 976 977
            leftr= p->data[0][last_line+R]= get_bits(&s->gb, 8);
            leftg= p->data[0][last_line+G]= get_bits(&s->gb, 8);
            leftb= p->data[0][last_line+B]= get_bits(&s->gb, 8);
Michael Niedermayer's avatar
Michael Niedermayer committed
978 979
            skip_bits(&s->gb, 8);
        }
980

Michael Niedermayer's avatar
Michael Niedermayer committed
981 982 983 984 985
        if(s->bgr32){
            switch(s->predictor){
            case LEFT:
            case PLANE:
                decode_bgr_bitstream(s, width-1);
Michael Niedermayer's avatar
Michael Niedermayer committed
986
                add_left_prediction_bgr32(p->data[0] + last_line+4, s->temp[0], width-1, &leftr, &leftg, &leftb);
Michael Niedermayer's avatar
Michael Niedermayer committed
987 988 989

                for(y=s->height-2; y>=0; y--){ //yes its stored upside down
                    decode_bgr_bitstream(s, width);
990

Michael Niedermayer's avatar
Michael Niedermayer committed
991
                    add_left_prediction_bgr32(p->data[0] + p->linesize[0]*y, s->temp[0], width, &leftr, &leftg, &leftb);
Michael Niedermayer's avatar
Michael Niedermayer committed
992
                    if(s->predictor == PLANE){
993
                        if((y&s->interlaced)==0 && y<s->height-1-s->interlaced){
994
                            s->dsp.add_bytes(p->data[0] + p->linesize[0]*y,
Michael Niedermayer's avatar
Michael Niedermayer committed
995
                                             p->data[0] + p->linesize[0]*y + fake_ystride, fake_ystride);
Michael Niedermayer's avatar
Michael Niedermayer committed
996 997 998
                        }
                    }
                }
999
                draw_slice(s, height); // just 1 large slice as this is not possible in reverse order
Michael Niedermayer's avatar
Michael Niedermayer committed
1000 1001
                break;
            default:
1002
                av_log(avctx, AV_LOG_ERROR, "prediction type not supported!\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
1003 1004 1005
            }
        }else{

1006
            av_log(avctx, AV_LOG_ERROR, "BGR24 output is not implemented yet\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
1007 1008 1009 1010
            return -1;
        }
    }
    emms_c();
1011

Michael Niedermayer's avatar
Michael Niedermayer committed
1012
    *picture= *p;
1013
    *data_size = sizeof(AVFrame);
1014

1015
    return (get_bits_count(&s->gb)+31)/32*4;
Michael Niedermayer's avatar
Michael Niedermayer committed
1016 1017
}

1018 1019
static int common_end(HYuvContext *s){
    int i;
1020

1021 1022 1023 1024 1025 1026
    for(i=0; i<3; i++){
        av_freep(&s->temp[i]);
    }
    return 0;
}

Michael Niedermayer's avatar
Michael Niedermayer committed
1027 1028 1029 1030
static int decode_end(AVCodecContext *avctx)
{
    HYuvContext *s = avctx->priv_data;
    int i;
1031

1032 1033
    common_end(s);
    av_freep(&s->bitstream_buffer);
1034

Michael Niedermayer's avatar
Michael Niedermayer committed
1035 1036 1037
    for(i=0; i<3; i++){
        free_vlc(&s->vlc[i]);
    }
Michael Niedermayer's avatar
Michael Niedermayer committed
1038

Michael Niedermayer's avatar
Michael Niedermayer committed
1039 1040 1041 1042 1043
    return 0;
}

static int encode_frame(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){
    HYuvContext *s = avctx->priv_data;
1044
    AVFrame *pict = data;
Michael Niedermayer's avatar
Michael Niedermayer committed
1045 1046 1047 1048 1049 1050
    const int width= s->width;
    const int width2= s->width>>1;
    const int height= s->height;
    const int fake_ystride= s->interlaced ? pict->linesize[0]*2  : pict->linesize[0];
    const int fake_ustride= s->interlaced ? pict->linesize[1]*2  : pict->linesize[1];
    const int fake_vstride= s->interlaced ? pict->linesize[2]*2  : pict->linesize[2];
1051
    AVFrame * const p= &s->picture;
1052
    int i, j, size=0;
Michael Niedermayer's avatar
Michael Niedermayer committed
1053

Michael Niedermayer's avatar
Michael Niedermayer committed
1054
    *p = *pict;
Michael Niedermayer's avatar
Michael Niedermayer committed
1055 1056
    p->pict_type= FF_I_TYPE;
    p->key_frame= 1;
1057

1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072
    if(s->context){
        for(i=0; i<3; i++){
            generate_len_table(s->len[i], s->stats[i], 256);
            if(generate_bits_table(s->bits[i], s->len[i])<0)
                return -1;
            size+= store_table(s, s->len[i], &buf[size]);
        }

        for(i=0; i<3; i++)
            for(j=0; j<256; j++)
                s->stats[i][j] >>= 1;
    }

    init_put_bits(&s->pb, buf+size, buf_size-size);

1073 1074
    if(avctx->pix_fmt == PIX_FMT_YUV422P || avctx->pix_fmt == PIX_FMT_YUV420P){
        int lefty, leftu, leftv, y, cy;
Michael Niedermayer's avatar
Michael Niedermayer committed
1075

Michael Niedermayer's avatar
Michael Niedermayer committed
1076 1077 1078 1079
        put_bits(&s->pb, 8, leftv= p->data[2][0]);
        put_bits(&s->pb, 8, lefty= p->data[0][1]);
        put_bits(&s->pb, 8, leftu= p->data[1][0]);
        put_bits(&s->pb, 8,        p->data[0][0]);
1080

Michael Niedermayer's avatar
Michael Niedermayer committed
1081 1082 1083
        lefty= sub_left_prediction(s, s->temp[0], p->data[0]+2, width-2 , lefty);
        leftu= sub_left_prediction(s, s->temp[1], p->data[1]+1, width2-1, leftu);
        leftv= sub_left_prediction(s, s->temp[2], p->data[2]+1, width2-1, leftv);
1084

Michael Niedermayer's avatar
Michael Niedermayer committed
1085
        encode_422_bitstream(s, width-2);
1086

Michael Niedermayer's avatar
Michael Niedermayer committed
1087 1088
        if(s->predictor==MEDIAN){
            int lefttopy, lefttopu, lefttopv;
1089
            cy=y=1;
Michael Niedermayer's avatar
Michael Niedermayer committed
1090
            if(s->interlaced){
Michael Niedermayer's avatar
Michael Niedermayer committed
1091 1092 1093
                lefty= sub_left_prediction(s, s->temp[0], p->data[0]+p->linesize[0], width , lefty);
                leftu= sub_left_prediction(s, s->temp[1], p->data[1]+p->linesize[1], width2, leftu);
                leftv= sub_left_prediction(s, s->temp[2], p->data[2]+p->linesize[2], width2, leftv);
1094

Michael Niedermayer's avatar
Michael Niedermayer committed
1095
                encode_422_bitstream(s, width);
1096
                y++; cy++;
Michael Niedermayer's avatar
Michael Niedermayer committed
1097
            }
1098

Michael Niedermayer's avatar
Michael Niedermayer committed
1099
            lefty= sub_left_prediction(s, s->temp[0], p->data[0]+fake_ystride, 4, lefty);
1100 1101
            leftu= sub_left_prediction(s, s->temp[1], p->data[1]+fake_ustride, 2, leftu);
            leftv= sub_left_prediction(s, s->temp[2], p->data[2]+fake_vstride, 2, leftv);
1102

Michael Niedermayer's avatar
Michael Niedermayer committed
1103
            encode_422_bitstream(s, 4);
1104

Michael Niedermayer's avatar
Michael Niedermayer committed
1105 1106 1107
            lefttopy= p->data[0][3];
            lefttopu= p->data[1][1];
            lefttopv= p->data[2][1];
1108 1109 1110
            s->dsp.sub_hfyu_median_prediction(s->temp[0], p->data[0]+4, p->data[0] + fake_ystride+4, width-4 , &lefty, &lefttopy);
            s->dsp.sub_hfyu_median_prediction(s->temp[1], p->data[1]+2, p->data[1] + fake_ustride+2, width2-2, &leftu, &lefttopu);
            s->dsp.sub_hfyu_median_prediction(s->temp[2], p->data[2]+2, p->data[2] + fake_vstride+2, width2-2, &leftv, &lefttopv);
Michael Niedermayer's avatar
Michael Niedermayer committed
1111
            encode_422_bitstream(s, width-4);
1112
            y++; cy++;
Michael Niedermayer's avatar
Michael Niedermayer committed
1113

1114
            for(; y<height; y++,cy++){
Michael Niedermayer's avatar
Michael Niedermayer committed
1115
                uint8_t *ydst, *udst, *vdst;
1116

1117 1118
                if(s->bitstream_bpp==12){
                    while(2*cy > y){
Michael Niedermayer's avatar
Michael Niedermayer committed
1119
                        ydst= p->data[0] + p->linesize[0]*y;
1120
                        s->dsp.sub_hfyu_median_prediction(s->temp[0], ydst - fake_ystride, ydst, width , &lefty, &lefttopy);
1121 1122 1123 1124 1125
                        encode_gray_bitstream(s, width);
                        y++;
                    }
                    if(y>=height) break;
                }
Michael Niedermayer's avatar
Michael Niedermayer committed
1126 1127 1128
                ydst= p->data[0] + p->linesize[0]*y;
                udst= p->data[1] + p->linesize[1]*cy;
                vdst= p->data[2] + p->linesize[2]*cy;
Michael Niedermayer's avatar
Michael Niedermayer committed
1129

1130 1131 1132
                s->dsp.sub_hfyu_median_prediction(s->temp[0], ydst - fake_ystride, ydst, width , &lefty, &lefttopy);
                s->dsp.sub_hfyu_median_prediction(s->temp[1], udst - fake_ustride, udst, width2, &leftu, &lefttopu);
                s->dsp.sub_hfyu_median_prediction(s->temp[2], vdst - fake_vstride, vdst, width2, &leftv, &lefttopv);
Michael Niedermayer's avatar
Michael Niedermayer committed
1133 1134 1135 1136

                encode_422_bitstream(s, width);
            }
        }else{
1137
            for(cy=y=1; y<height; y++,cy++){
Michael Niedermayer's avatar
Michael Niedermayer committed
1138
                uint8_t *ydst, *udst, *vdst;
1139

1140 1141
                /* encode a luma only line & y++ */
                if(s->bitstream_bpp==12){
Michael Niedermayer's avatar
Michael Niedermayer committed
1142
                    ydst= p->data[0] + p->linesize[0]*y;
1143 1144

                    if(s->predictor == PLANE && s->interlaced < y){
1145
                        s->dsp.diff_bytes(s->temp[1], ydst, ydst - fake_ystride, width);
1146

1147
                        lefty= sub_left_prediction(s, s->temp[0], s->temp[1], width , lefty);
1148
                    }else{
1149
                        lefty= sub_left_prediction(s, s->temp[0], ydst, width , lefty);
1150 1151 1152 1153 1154
                    }
                    encode_gray_bitstream(s, width);
                    y++;
                    if(y>=height) break;
                }
1155

Michael Niedermayer's avatar
Michael Niedermayer committed
1156 1157 1158
                ydst= p->data[0] + p->linesize[0]*y;
                udst= p->data[1] + p->linesize[1]*cy;
                vdst= p->data[2] + p->linesize[2]*cy;
Michael Niedermayer's avatar
Michael Niedermayer committed
1159

1160
                if(s->predictor == PLANE && s->interlaced < cy){
1161 1162
                    s->dsp.diff_bytes(s->temp[1], ydst, ydst - fake_ystride, width);
                    s->dsp.diff_bytes(s->temp[2], udst, udst - fake_ustride, width2);
Michael Niedermayer's avatar
Michael Niedermayer committed
1163
                    s->dsp.diff_bytes(s->temp[2] + width2, vdst, vdst - fake_vstride, width2);
Michael Niedermayer's avatar
Michael Niedermayer committed
1164

1165 1166
                    lefty= sub_left_prediction(s, s->temp[0], s->temp[1], width , lefty);
                    leftu= sub_left_prediction(s, s->temp[1], s->temp[2], width2, leftu);
Michael Niedermayer's avatar
Michael Niedermayer committed
1167
                    leftv= sub_left_prediction(s, s->temp[2], s->temp[2] + width2, width2, leftv);
Michael Niedermayer's avatar
Michael Niedermayer committed
1168
                }else{
1169 1170 1171
                    lefty= sub_left_prediction(s, s->temp[0], ydst, width , lefty);
                    leftu= sub_left_prediction(s, s->temp[1], udst, width2, leftu);
                    leftv= sub_left_prediction(s, s->temp[2], vdst, width2, leftv);
Michael Niedermayer's avatar
Michael Niedermayer committed
1172 1173 1174 1175
                }

                encode_422_bitstream(s, width);
            }
1176
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
1177
    }else{
1178
        av_log(avctx, AV_LOG_ERROR, "Format not supported!\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
1179 1180
    }
    emms_c();
1181

1182 1183
    size+= (put_bits_count(&s->pb)+31)/8;
    size/= 4;
1184

Michael Niedermayer's avatar
Michael Niedermayer committed
1185 1186 1187
    if((s->flags&CODEC_FLAG_PASS1) && (s->picture_number&31)==0){
        int j;
        char *p= avctx->stats_out;
Michael Niedermayer's avatar
Michael Niedermayer committed
1188
        char *end= p + 1024*30;
Michael Niedermayer's avatar
Michael Niedermayer committed
1189 1190
        for(i=0; i<3; i++){
            for(j=0; j<256; j++){
1191
                snprintf(p, end-p, "%"PRIu64" ", s->stats[i][j]);
Michael Niedermayer's avatar
Michael Niedermayer committed
1192 1193 1194
                p+= strlen(p);
                s->stats[i][j]= 0;
            }
Michael Niedermayer's avatar
Michael Niedermayer committed
1195
            snprintf(p, end-p, "\n");
Michael Niedermayer's avatar
Michael Niedermayer committed
1196 1197
            p++;
        }
1198 1199
    }
    if(!(s->avctx->flags2 & CODEC_FLAG2_NO_OUTPUT)){
1200
        flush_put_bits(&s->pb);
Michael Niedermayer's avatar
Michael Niedermayer committed
1201
        s->dsp.bswap_buf((uint32_t*)buf, (uint32_t*)buf, size);
1202
        avctx->stats_out[0] = '\0';
Michael Niedermayer's avatar
Michael Niedermayer committed
1203
    }
1204

Michael Niedermayer's avatar
Michael Niedermayer committed
1205
    s->picture_number++;
Michael Niedermayer's avatar
Michael Niedermayer committed
1206

Michael Niedermayer's avatar
Michael Niedermayer committed
1207 1208 1209 1210 1211
    return size*4;
}

static int encode_end(AVCodecContext *avctx)
{
1212
    HYuvContext *s = avctx->priv_data;
1213

1214
    common_end(s);
Michael Niedermayer's avatar
Michael Niedermayer committed
1215 1216 1217

    av_freep(&avctx->extradata);
    av_freep(&avctx->stats_out);
1218

Michael Niedermayer's avatar
Michael Niedermayer committed
1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
    return 0;
}

AVCodec huffyuv_decoder = {
    "huffyuv",
    CODEC_TYPE_VIDEO,
    CODEC_ID_HUFFYUV,
    sizeof(HYuvContext),
    decode_init,
    NULL,
    decode_end,
    decode_frame,
1231
    CODEC_CAP_DR1 | CODEC_CAP_DRAW_HORIZ_BAND,
Michael Niedermayer's avatar
Michael Niedermayer committed
1232 1233 1234
    NULL
};

1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247
AVCodec ffvhuff_decoder = {
    "ffvhuff",
    CODEC_TYPE_VIDEO,
    CODEC_ID_FFVHUFF,
    sizeof(HYuvContext),
    decode_init,
    NULL,
    decode_end,
    decode_frame,
    CODEC_CAP_DR1 | CODEC_CAP_DRAW_HORIZ_BAND,
    NULL
};

1248 1249
#ifdef CONFIG_ENCODERS

Michael Niedermayer's avatar
Michael Niedermayer committed
1250 1251 1252 1253 1254 1255 1256 1257 1258
AVCodec huffyuv_encoder = {
    "huffyuv",
    CODEC_TYPE_VIDEO,
    CODEC_ID_HUFFYUV,
    sizeof(HYuvContext),
    encode_init,
    encode_frame,
    encode_end,
};
1259

1260 1261 1262 1263 1264 1265 1266 1267 1268 1269
AVCodec ffvhuff_encoder = {
    "ffvhuff",
    CODEC_TYPE_VIDEO,
    CODEC_ID_FFVHUFF,
    sizeof(HYuvContext),
    encode_init,
    encode_frame,
    encode_end,
};

1270
#endif //CONFIG_ENCODERS