Statistics
| Branch: | Revision:

ffmpeg / libavcodec / dv.c @ fbb89806

History | View | Annotate | Download (31.3 KB)

1 5d2231ee Fabrice Bellard
/*
2
 * DV decoder
3
 * Copyright (c) 2002 Fabrice Bellard.
4
 *
5 4fa1c4fa Roman Shaposhnik
 * DV encoder 
6
 * Copyright (c) 2003 Roman Shaposhnik.
7
 *
8
 * Many thanks to Dan Dennedy <dan@dennedy.org> for providing wealth
9
 * of DV technical info.
10
 *
11 5d2231ee Fabrice Bellard
 * This library is free software; you can redistribute it and/or
12
 * modify it under the terms of the GNU Lesser General Public
13
 * License as published by the Free Software Foundation; either
14
 * version 2 of the License, or (at your option) any later version.
15
 *
16
 * This library is distributed in the hope that it will be useful,
17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19
 * Lesser General Public License for more details.
20
 *
21
 * You should have received a copy of the GNU Lesser General Public
22
 * License along with this library; if not, write to the Free Software
23
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
24
 */
25 983e3246 Michael Niedermayer
26
/**
27
 * @file dv.c
28 4fa1c4fa Roman Shaposhnik
 * DV codec.
29 983e3246 Michael Niedermayer
 */
30 5d2231ee Fabrice Bellard
#include "avcodec.h"
31
#include "dsputil.h"
32
#include "mpegvideo.h"
33
#include "simple_idct.h"
34 7458ccbb Roman Shaposhnik
#include "dvdata.h"
35 5d2231ee Fabrice Bellard
36
typedef struct DVVideoDecodeContext {
37 7458ccbb Roman Shaposhnik
    const DVprofile* sys;
38 492cd3a9 Michael Niedermayer
    AVFrame picture;
39 7458ccbb Roman Shaposhnik
    
40 0c1a9eda Zdenek Kabelac
    uint8_t dv_zigzag[2][64];
41 10acc479 Roman Shaposhnik
    uint8_t dv_idct_shift[2][22][64];
42
  
43 7458ccbb Roman Shaposhnik
    void (*get_pixels)(DCTELEM *block, const uint8_t *pixels, int line_size);
44 10acc479 Roman Shaposhnik
    void (*fdct[2])(DCTELEM *block);
45 0c1a9eda Zdenek Kabelac
    void (*idct_put[2])(uint8_t *dest, int line_size, DCTELEM *block);
46 10acc479 Roman Shaposhnik
    
47
    GetBitContext gb;
48
    DCTELEM block[5*6][64] __align8;
49 5d2231ee Fabrice Bellard
} DVVideoDecodeContext;
50
51 7458ccbb Roman Shaposhnik
#define TEX_VLC_BITS 9
52 e15c3302 Roman Shaposhnik
53
#ifdef DV_CODEC_TINY_TARGET
54
#define DV_VLC_MAP_RUN_SIZE 15
55
#define DV_VLC_MAP_LEV_SIZE 23
56
#else
57
#define DV_VLC_MAP_RUN_SIZE  64 
58
#define DV_VLC_MAP_LEV_SIZE 512
59
#endif
60
61 5d2231ee Fabrice Bellard
/* XXX: also include quantization */
62
static RL_VLC_ELEM *dv_rl_vlc[1];
63 e15c3302 Roman Shaposhnik
/* VLC encoding lookup table */
64
static struct dv_vlc_pair {
65
   uint32_t vlc;
66
   uint8_t  size;
67
} (*dv_vlc_map)[DV_VLC_MAP_LEV_SIZE] = NULL;
68 5d2231ee Fabrice Bellard
69 10acc479 Roman Shaposhnik
static void dv_build_unquantize_tables(DVVideoDecodeContext *s, uint8_t* perm)
70 5d2231ee Fabrice Bellard
{
71 7be166e4 Fabrice Bellard
    int i, q, j;
72 5d2231ee Fabrice Bellard
73
    /* NOTE: max left shift is 6 */
74 7be166e4 Fabrice Bellard
    for(q = 0; q < 22; q++) {
75 10acc479 Roman Shaposhnik
        /* 88DCT */
76 7be166e4 Fabrice Bellard
        for(i = 1; i < 64; i++) {
77
            /* 88 table */
78 10acc479 Roman Shaposhnik
            j = perm[i];
79
            s->dv_idct_shift[0][q][j] =
80 7be166e4 Fabrice Bellard
                dv_quant_shifts[q][dv_88_areas[i]] + 1;
81
        }
82
        
83 10acc479 Roman Shaposhnik
        /* 248DCT */
84 7be166e4 Fabrice Bellard
        for(i = 1; i < 64; i++) {
85
            /* 248 table */
86 10acc479 Roman Shaposhnik
            s->dv_idct_shift[1][q][i] =  
87
                dv_quant_shifts[q][dv_248_areas[i]] + 1;
88 5d2231ee Fabrice Bellard
        }
89
    }
90
}
91
92 4fa1c4fa Roman Shaposhnik
static int dvvideo_init(AVCodecContext *avctx)
93 5d2231ee Fabrice Bellard
{
94
    DVVideoDecodeContext *s = avctx->priv_data;
95 10acc479 Roman Shaposhnik
    DSPContext dsp;
96 7604246d Wolfgang Hesseler
    static int done=0;
97 e15c3302 Roman Shaposhnik
    int i, j;
98 5d2231ee Fabrice Bellard
99
    if (!done) {
100
        int i;
101 7458ccbb Roman Shaposhnik
        VLC dv_vlc;
102 5d2231ee Fabrice Bellard
103
        done = 1;
104
105 e15c3302 Roman Shaposhnik
        dv_vlc_map = av_mallocz(DV_VLC_MAP_LEV_SIZE*DV_VLC_MAP_RUN_SIZE*sizeof(struct dv_vlc_pair));
106
        if (!dv_vlc_map)
107
            return -ENOMEM;
108
109 5d2231ee Fabrice Bellard
        /* NOTE: as a trick, we use the fact the no codes are unused
110
           to accelerate the parsing of partial codes */
111
        init_vlc(&dv_vlc, TEX_VLC_BITS, NB_DV_VLC, 
112
                 dv_vlc_len, 1, 1, dv_vlc_bits, 2, 2);
113
114
        dv_rl_vlc[0] = av_malloc(dv_vlc.table_size * sizeof(RL_VLC_ELEM));
115 e15c3302 Roman Shaposhnik
        if (!dv_rl_vlc[0]) {
116
            av_free(dv_vlc_map);
117
            return -ENOMEM;
118
        }
119 5d2231ee Fabrice Bellard
        for(i = 0; i < dv_vlc.table_size; i++){
120
            int code= dv_vlc.table[i][0];
121
            int len = dv_vlc.table[i][1];
122
            int level, run;
123
        
124
            if(len<0){ //more bits needed
125
                run= 0;
126
                level= code;
127
            } else if (code == (NB_DV_VLC - 1)) {
128
                /* EOB */
129
                run = 0;
130
                level = 256;
131
            } else {
132
                run=   dv_vlc_run[code] + 1;
133
                level= dv_vlc_level[code];
134
            }
135
            dv_rl_vlc[0][i].len = len;
136
            dv_rl_vlc[0][i].level = level;
137
            dv_rl_vlc[0][i].run = run;
138
        }
139 7458ccbb Roman Shaposhnik
140
        for (i = 0; i < NB_DV_VLC - 1; i++) {
141 e15c3302 Roman Shaposhnik
           if (dv_vlc_run[i] >= DV_VLC_MAP_RUN_SIZE || dv_vlc_level[i] >= DV_VLC_MAP_LEV_SIZE)
142
               continue;
143
           
144
           if (dv_vlc_map[dv_vlc_run[i]][dv_vlc_level[i]].size != 0)
145
               continue;
146
               
147
           dv_vlc_map[dv_vlc_run[i]][dv_vlc_level[i]].vlc = dv_vlc_bits[i] << 
148
                                                            (!!dv_vlc_level[i]);
149
           dv_vlc_map[dv_vlc_run[i]][dv_vlc_level[i]].size = dv_vlc_len[i] + 
150
                                                             (!!dv_vlc_level[i]);
151
        }
152
        for (i = 0; i < DV_VLC_MAP_RUN_SIZE; i++) {
153
#ifdef DV_CODEC_TINY_TARGET
154
           for (j = 1; j < DV_VLC_MAP_LEV_SIZE; j++) {
155
              if (dv_vlc_map[i][j].size == 0) {
156
                  dv_vlc_map[i][j].vlc = dv_vlc_map[0][j].vlc |
157
                            (dv_vlc_map[i-1][0].vlc << (dv_vlc_map[0][j].size));
158
                  dv_vlc_map[i][j].size = dv_vlc_map[i-1][0].size + 
159
                                          dv_vlc_map[0][j].size;
160
              }
161
           }
162
#else
163
           for (j = 1; j < DV_VLC_MAP_LEV_SIZE/2; j++) {
164
              if (dv_vlc_map[i][j].size == 0) {
165
                  dv_vlc_map[i][j].vlc = dv_vlc_map[0][j].vlc |
166
                            (dv_vlc_map[i-1][0].vlc << (dv_vlc_map[0][j].size));
167
                  dv_vlc_map[i][j].size = dv_vlc_map[i-1][0].size + 
168
                                          dv_vlc_map[0][j].size;
169
              }
170
              dv_vlc_map[i][((uint16_t)(-j))&0x1ff].vlc = 
171
                                            dv_vlc_map[i][j].vlc | 1;
172
              dv_vlc_map[i][((uint16_t)(-j))&0x1ff].size = 
173
                                            dv_vlc_map[i][j].size;
174
           }
175
#endif
176 7458ccbb Roman Shaposhnik
        }
177 5d2231ee Fabrice Bellard
    }
178 7be166e4 Fabrice Bellard
179 10acc479 Roman Shaposhnik
    /* Generic DSP setup */
180
    dsputil_init(&dsp, avctx);
181
    s->get_pixels = dsp.get_pixels;
182 7be166e4 Fabrice Bellard
183 10acc479 Roman Shaposhnik
    /* 88DCT setup */
184
    s->fdct[0] = dsp.fdct;
185
    s->idct_put[0] = dsp.idct_put;
186
    for (i=0; i<64; i++)
187
       s->dv_zigzag[0][i] = dsp.idct_permutation[ff_zigzag_direct[i]];
188 7be166e4 Fabrice Bellard
189 10acc479 Roman Shaposhnik
    /* 248DCT setup */
190
    s->fdct[1] = dsp.fdct248;
191
    s->idct_put[1] = simple_idct248_put;  // FIXME: need to add it to DSP
192
    memcpy(s->dv_zigzag[1], ff_zigzag248_direct, 64);
193 7be166e4 Fabrice Bellard
194 5d2231ee Fabrice Bellard
    /* XXX: do it only for constant case */
195 10acc479 Roman Shaposhnik
    dv_build_unquantize_tables(s, dsp.idct_permutation);
196 7458ccbb Roman Shaposhnik
197 4fa1c4fa Roman Shaposhnik
    /* FIXME: I really don't think this should be here */
198
    if (dv_codec_profile(avctx))
199
        avctx->pix_fmt = dv_codec_profile(avctx)->pix_fmt; 
200 2744ca9a Roman Shaposhnik
    avctx->coded_frame = &s->picture;
201 4fa1c4fa Roman Shaposhnik
    
202 5d2231ee Fabrice Bellard
    return 0;
203
}
204
205 7458ccbb Roman Shaposhnik
// #define VLC_DEBUG
206 5d2231ee Fabrice Bellard
207 7be166e4 Fabrice Bellard
typedef struct BlockInfo {
208 0c1a9eda Zdenek Kabelac
    const uint8_t *shift_table;
209
    const uint8_t *scan_table;
210
    uint8_t pos; /* position in block */
211
    uint8_t eob_reached; /* true if EOB has been reached */
212
    uint8_t dct_mode;
213
    uint8_t partial_bit_count;
214
    uint16_t partial_bit_buffer;
215 7be166e4 Fabrice Bellard
    int shift_offset;
216
} BlockInfo;
217 5d2231ee Fabrice Bellard
218
/* block size in bits */
219 0c1a9eda Zdenek Kabelac
static const uint16_t block_sizes[6] = {
220 5d2231ee Fabrice Bellard
    112, 112, 112, 112, 80, 80
221
};
222 e15c3302 Roman Shaposhnik
/* bit budget for AC only in 5 MBs */
223
static const int vs_total_ac_bits = (100 * 4 + 68*2) * 5;
224
/* see dv_88_areas and dv_248_areas for details */
225
static const int mb_area_start[5] = { 1, 6, 21, 43, 64 }; 
226 5d2231ee Fabrice Bellard
227
#ifndef ALT_BITSTREAM_READER
228 924311cd Michael Niedermayer
#warning only works with ALT_BITSTREAM_READER
229 5d2231ee Fabrice Bellard
#endif
230
231
/* decode ac coefs */
232
static void dv_decode_ac(DVVideoDecodeContext *s, 
233 0e15384d Michael Niedermayer
                         BlockInfo *mb, DCTELEM *block, int last_index)
234 5d2231ee Fabrice Bellard
{
235
    int last_re_index;
236 7be166e4 Fabrice Bellard
    int shift_offset = mb->shift_offset;
237 0c1a9eda Zdenek Kabelac
    const uint8_t *scan_table = mb->scan_table;
238
    const uint8_t *shift_table = mb->shift_table;
239 7be166e4 Fabrice Bellard
    int pos = mb->pos;
240 5d2231ee Fabrice Bellard
    int level, pos1, sign, run;
241
    int partial_bit_count;
242 924311cd Michael Niedermayer
#ifndef ALT_BITSTREAM_READER //FIXME
243
    int re_index=0; 
244
    int re1_index=0;
245
#endif
246 5d2231ee Fabrice Bellard
    OPEN_READER(re, &s->gb);
247
    
248
#ifdef VLC_DEBUG
249 7be166e4 Fabrice Bellard
    printf("start\n");
250 5d2231ee Fabrice Bellard
#endif
251
252
    /* if we must parse a partial vlc, we do it here */
253 7be166e4 Fabrice Bellard
    partial_bit_count = mb->partial_bit_count;
254 5d2231ee Fabrice Bellard
    if (partial_bit_count > 0) {
255 0c1a9eda Zdenek Kabelac
        uint8_t buf[4];
256
        uint32_t v;
257 5d2231ee Fabrice Bellard
        int l, l1;
258
        GetBitContext gb1;
259
260
        /* build the dummy bit buffer */
261
        l = 16 - partial_bit_count;
262
        UPDATE_CACHE(re, &s->gb);
263
#ifdef VLC_DEBUG
264
        printf("show=%04x\n", SHOW_UBITS(re, &s->gb, 16));
265
#endif
266 7be166e4 Fabrice Bellard
        v = (mb->partial_bit_buffer << l) | SHOW_UBITS(re, &s->gb, l);
267 5d2231ee Fabrice Bellard
        buf[0] = v >> 8;
268
        buf[1] = v;
269
#ifdef VLC_DEBUG
270
        printf("v=%04x cnt=%d %04x\n", 
271 7be166e4 Fabrice Bellard
               v, partial_bit_count, (mb->partial_bit_buffer << l));
272 5d2231ee Fabrice Bellard
#endif
273
        /* try to read the codeword */
274 68f593b4 Michael Niedermayer
        init_get_bits(&gb1, buf, 4*8);
275 5d2231ee Fabrice Bellard
        {
276
            OPEN_READER(re1, &gb1);
277
            UPDATE_CACHE(re1, &gb1);
278
            GET_RL_VLC(level, run, re1, &gb1, dv_rl_vlc[0], 
279
                       TEX_VLC_BITS, 2);
280
            l = re1_index;
281
            CLOSE_READER(re1, &gb1);
282
        }
283
#ifdef VLC_DEBUG
284
        printf("****run=%d level=%d size=%d\n", run, level, l);
285
#endif
286
        /* compute codeword length */
287
        l1 = (level != 256 && level != 0);
288
        /* if too long, we cannot parse */
289
        l -= partial_bit_count;
290
        if ((re_index + l + l1) > last_index)
291
            return;
292
        /* skip read bits */
293
        last_re_index = 0; /* avoid warning */
294
        re_index += l;
295 7be166e4 Fabrice Bellard
        /* by definition, if we can read the vlc, all partial bits
296 5d2231ee Fabrice Bellard
           will be read (otherwise we could have read the vlc before) */
297 7be166e4 Fabrice Bellard
        mb->partial_bit_count = 0;
298 5d2231ee Fabrice Bellard
        UPDATE_CACHE(re, &s->gb);
299
        goto handle_vlc;
300
    }
301
302
    /* get the AC coefficients until last_index is reached */
303
    for(;;) {
304
        UPDATE_CACHE(re, &s->gb);
305
#ifdef VLC_DEBUG
306
        printf("%2d: bits=%04x index=%d\n", 
307
               pos, SHOW_UBITS(re, &s->gb, 16), re_index);
308
#endif
309
        last_re_index = re_index;
310
        GET_RL_VLC(level, run, re, &s->gb, dv_rl_vlc[0], 
311
                   TEX_VLC_BITS, 2);
312
    handle_vlc:
313
#ifdef VLC_DEBUG
314
        printf("run=%d level=%d\n", run, level);
315
#endif
316
        if (level == 256) {
317
            if (re_index > last_index) {
318
            cannot_read:
319
                /* put position before read code */
320
                re_index = last_re_index;
321 7be166e4 Fabrice Bellard
                mb->eob_reached = 0;
322 5d2231ee Fabrice Bellard
                break;
323
            }
324
            /* EOB */
325 7be166e4 Fabrice Bellard
            mb->eob_reached = 1;
326 5d2231ee Fabrice Bellard
            break;
327
        } else if (level != 0) {
328
            if ((re_index + 1) > last_index)
329
                goto cannot_read;
330
            sign = SHOW_SBITS(re, &s->gb, 1);
331
            level = (level ^ sign) - sign;
332
            LAST_SKIP_BITS(re, &s->gb, 1);
333
            pos += run;
334
            /* error */
335
            if (pos >= 64) {
336
                goto read_error;
337
            }
338
            pos1 = scan_table[pos];
339 7be166e4 Fabrice Bellard
            level = level << (shift_table[pos1] + shift_offset);
340 5d2231ee Fabrice Bellard
            block[pos1] = level;
341
            //            printf("run=%d level=%d shift=%d\n", run, level, shift_table[pos1]);
342
        } else {
343
            if (re_index > last_index)
344
                goto cannot_read;
345
            /* level is zero: means run without coding. No
346
               sign is coded */
347
            pos += run;
348
            /* error */
349
            if (pos >= 64) {
350
            read_error:
351
#if defined(VLC_DEBUG) || 1
352 9b879566 Michel Bardiaux
                av_log(NULL, AV_LOG_ERROR, "error pos=%d\n", pos);
353 5d2231ee Fabrice Bellard
#endif
354
                /* for errors, we consider the eob is reached */
355 7be166e4 Fabrice Bellard
                mb->eob_reached = 1;
356 5d2231ee Fabrice Bellard
                break;
357
            }
358
        }
359
    }
360
    CLOSE_READER(re, &s->gb);
361 7be166e4 Fabrice Bellard
    mb->pos = pos;
362 5d2231ee Fabrice Bellard
}
363
364
static inline void bit_copy(PutBitContext *pb, GetBitContext *gb, int bits_left)
365
{
366
    while (bits_left >= 16) {
367
        put_bits(pb, 16, get_bits(gb, 16));
368
        bits_left -= 16;
369
    }
370
    if (bits_left > 0) {
371
        put_bits(pb, bits_left, get_bits(gb, bits_left));
372
    }
373
}
374
375
/* mb_x and mb_y are in units of 8 pixels */
376
static inline void dv_decode_video_segment(DVVideoDecodeContext *s, 
377 0c1a9eda Zdenek Kabelac
                                           uint8_t *buf_ptr1, 
378
                                           const uint16_t *mb_pos_ptr)
379 5d2231ee Fabrice Bellard
{
380
    int quant, dc, dct_mode, class1, j;
381
    int mb_index, mb_x, mb_y, v, last_index;
382
    DCTELEM *block, *block1;
383
    int c_offset, bits_left;
384 0c1a9eda Zdenek Kabelac
    uint8_t *y_ptr;
385 7be166e4 Fabrice Bellard
    BlockInfo mb_data[5 * 6], *mb, *mb1;
386 0c1a9eda Zdenek Kabelac
    void (*idct_put)(uint8_t *dest, int line_size, DCTELEM *block);
387
    uint8_t *buf_ptr;
388 5d2231ee Fabrice Bellard
    PutBitContext pb, vs_pb;
389 0c1a9eda Zdenek Kabelac
    uint8_t mb_bit_buffer[80 + 4]; /* allow some slack */
390 7be166e4 Fabrice Bellard
    int mb_bit_count;
391 0c1a9eda Zdenek Kabelac
    uint8_t vs_bit_buffer[5 * 80 + 4]; /* allow some slack */
392 5d2231ee Fabrice Bellard
    int vs_bit_count;
393
    
394
    memset(s->block, 0, sizeof(s->block));
395
396
    /* pass 1 : read DC and AC coefficients in blocks */
397
    buf_ptr = buf_ptr1;
398
    block1 = &s->block[0][0];
399 7be166e4 Fabrice Bellard
    mb1 = mb_data;
400 ed7debda Alex Beregszaszi
    init_put_bits(&vs_pb, vs_bit_buffer, 5 * 80);
401 5d2231ee Fabrice Bellard
    vs_bit_count = 0;
402
    for(mb_index = 0; mb_index < 5; mb_index++) {
403
        /* skip header */
404
        quant = buf_ptr[3] & 0x0f;
405
        buf_ptr += 4;
406 ed7debda Alex Beregszaszi
        init_put_bits(&pb, mb_bit_buffer, 80);
407 7be166e4 Fabrice Bellard
        mb_bit_count = 0;
408
        mb = mb1;
409 5d2231ee Fabrice Bellard
        block = block1;
410
        for(j = 0;j < 6; j++) {
411
            /* NOTE: size is not important here */
412 68f593b4 Michael Niedermayer
            init_get_bits(&s->gb, buf_ptr, 14*8);
413 5d2231ee Fabrice Bellard
            
414
            /* get the dc */
415
            dc = get_bits(&s->gb, 9);
416
            dc = (dc << (32 - 9)) >> (32 - 9);
417
            dct_mode = get_bits1(&s->gb);
418 7be166e4 Fabrice Bellard
            mb->dct_mode = dct_mode;
419
            mb->scan_table = s->dv_zigzag[dct_mode];
420 5d2231ee Fabrice Bellard
            class1 = get_bits(&s->gb, 2);
421 7be166e4 Fabrice Bellard
            mb->shift_offset = (class1 == 3);
422 10acc479 Roman Shaposhnik
            mb->shift_table = s->dv_idct_shift[dct_mode]
423 5d2231ee Fabrice Bellard
                [quant + dv_quant_offset[class1]];
424
            dc = dc << 2;
425
            /* convert to unsigned because 128 is not added in the
426
               standard IDCT */
427
            dc += 1024;
428
            block[0] = dc;
429
            last_index = block_sizes[j];
430
            buf_ptr += last_index >> 3;
431 7be166e4 Fabrice Bellard
            mb->pos = 0;
432
            mb->partial_bit_count = 0;
433 5d2231ee Fabrice Bellard
434 4fa1c4fa Roman Shaposhnik
#ifdef VLC_DEBUG
435
            printf("MB block: %d, %d ", mb_index, j);
436
#endif
437 7be166e4 Fabrice Bellard
            dv_decode_ac(s, mb, block, last_index);
438 5d2231ee Fabrice Bellard
439
            /* write the remaining bits  in a new buffer only if the
440
               block is finished */
441 924311cd Michael Niedermayer
            bits_left = last_index - get_bits_count(&s->gb);
442 7be166e4 Fabrice Bellard
            if (mb->eob_reached) {
443
                mb->partial_bit_count = 0;
444
                mb_bit_count += bits_left;
445 5d2231ee Fabrice Bellard
                bit_copy(&pb, &s->gb, bits_left);
446
            } else {
447
                /* should be < 16 bits otherwise a codeword could have
448
                   been parsed */
449 7be166e4 Fabrice Bellard
                mb->partial_bit_count = bits_left;
450
                mb->partial_bit_buffer = get_bits(&s->gb, bits_left);
451 5d2231ee Fabrice Bellard
            }
452
            block += 64;
453 7be166e4 Fabrice Bellard
            mb++;
454 5d2231ee Fabrice Bellard
        }
455
        
456
        flush_put_bits(&pb);
457
458
        /* pass 2 : we can do it just after */
459
#ifdef VLC_DEBUG
460 4fa1c4fa Roman Shaposhnik
        printf("***pass 2 size=%d MB#=%d\n", mb_bit_count, mb_index);
461 5d2231ee Fabrice Bellard
#endif
462
        block = block1;
463 7be166e4 Fabrice Bellard
        mb = mb1;
464 68f593b4 Michael Niedermayer
        init_get_bits(&s->gb, mb_bit_buffer, 80*8);
465 5d2231ee Fabrice Bellard
        for(j = 0;j < 6; j++) {
466 924311cd Michael Niedermayer
            if (!mb->eob_reached && get_bits_count(&s->gb) < mb_bit_count) {
467 7be166e4 Fabrice Bellard
                dv_decode_ac(s, mb, block, mb_bit_count);
468 5d2231ee Fabrice Bellard
                /* if still not finished, no need to parse other blocks */
469 7be166e4 Fabrice Bellard
                if (!mb->eob_reached) {
470 5d2231ee Fabrice Bellard
                    /* we could not parse the current AC coefficient,
471
                       so we add the remaining bytes */
472 924311cd Michael Niedermayer
                    bits_left = mb_bit_count - get_bits_count(&s->gb);
473 5d2231ee Fabrice Bellard
                    if (bits_left > 0) {
474 7be166e4 Fabrice Bellard
                        mb->partial_bit_count += bits_left;
475
                        mb->partial_bit_buffer = 
476
                            (mb->partial_bit_buffer << bits_left) | 
477 5d2231ee Fabrice Bellard
                            get_bits(&s->gb, bits_left);
478
                    }
479
                    goto next_mb;
480
                }
481
            }
482
            block += 64;
483 7be166e4 Fabrice Bellard
            mb++;
484 5d2231ee Fabrice Bellard
        }
485
        /* all blocks are finished, so the extra bytes can be used at
486
           the video segment level */
487 924311cd Michael Niedermayer
        bits_left = mb_bit_count - get_bits_count(&s->gb);
488 5d2231ee Fabrice Bellard
        vs_bit_count += bits_left;
489
        bit_copy(&vs_pb, &s->gb, bits_left);
490
    next_mb:
491 7be166e4 Fabrice Bellard
        mb1 += 6;
492 5d2231ee Fabrice Bellard
        block1 += 6 * 64;
493
    }
494
495
    /* we need a pass other the whole video segment */
496
    flush_put_bits(&vs_pb);
497
        
498
#ifdef VLC_DEBUG
499
    printf("***pass 3 size=%d\n", vs_bit_count);
500
#endif
501
    block = &s->block[0][0];
502
    mb = mb_data;
503 68f593b4 Michael Niedermayer
    init_get_bits(&s->gb, vs_bit_buffer, 5 * 80*8);
504 5d2231ee Fabrice Bellard
    for(mb_index = 0; mb_index < 5; mb_index++) {
505
        for(j = 0;j < 6; j++) {
506 7be166e4 Fabrice Bellard
            if (!mb->eob_reached) {
507 5d2231ee Fabrice Bellard
#ifdef VLC_DEBUG
508
                printf("start %d:%d\n", mb_index, j);
509
#endif
510 7be166e4 Fabrice Bellard
                dv_decode_ac(s, mb, block, vs_bit_count);
511 5d2231ee Fabrice Bellard
            }
512
            block += 64;
513 7be166e4 Fabrice Bellard
            mb++;
514 5d2231ee Fabrice Bellard
        }
515
    }
516
    
517
    /* compute idct and place blocks */
518
    block = &s->block[0][0];
519
    mb = mb_data;
520
    for(mb_index = 0; mb_index < 5; mb_index++) {
521
        v = *mb_pos_ptr++;
522
        mb_x = v & 0xff;
523
        mb_y = v >> 8;
524 7458ccbb Roman Shaposhnik
        y_ptr = s->picture.data[0] + (mb_y * s->picture.linesize[0] * 8) + (mb_x * 8);
525
        if (s->sys->pix_fmt == PIX_FMT_YUV411P)
526
            c_offset = (mb_y * s->picture.linesize[1] * 8) + ((mb_x >> 2) * 8);
527 5d2231ee Fabrice Bellard
        else
528 7458ccbb Roman Shaposhnik
            c_offset = ((mb_y >> 1) * s->picture.linesize[1] * 8) + ((mb_x >> 1) * 8);
529 5d2231ee Fabrice Bellard
        for(j = 0;j < 6; j++) {
530 7be166e4 Fabrice Bellard
            idct_put = s->idct_put[mb->dct_mode];
531 5d2231ee Fabrice Bellard
            if (j < 4) {
532 7458ccbb Roman Shaposhnik
                if (s->sys->pix_fmt == PIX_FMT_YUV411P && mb_x < (704 / 8)) {
533 1c05e11d Fabrice Bellard
                    /* NOTE: at end of line, the macroblock is handled as 420 */
534 7458ccbb Roman Shaposhnik
                    idct_put(y_ptr + (j * 8), s->picture.linesize[0], block);
535 5d2231ee Fabrice Bellard
                } else {
536 7458ccbb Roman Shaposhnik
                    idct_put(y_ptr + ((j & 1) * 8) + ((j >> 1) * 8 * s->picture.linesize[0]),
537
                             s->picture.linesize[0], block);
538 5d2231ee Fabrice Bellard
                }
539
            } else {
540 7458ccbb Roman Shaposhnik
                if (s->sys->pix_fmt == PIX_FMT_YUV411P && mb_x >= (704 / 8)) {
541 00c28046 Roman Shaposhnik
                    uint64_t aligned_pixels[64/8];
542 f539eb52 Michael Niedermayer
                    uint8_t *pixels= (uint8_t*)aligned_pixels;
543 00c28046 Roman Shaposhnik
                    uint8_t *c_ptr, *c_ptr1, *ptr;
544 1c05e11d Fabrice Bellard
                    int y, linesize;
545
                    /* NOTE: at end of line, the macroblock is handled as 420 */
546
                    idct_put(pixels, 8, block);
547 7458ccbb Roman Shaposhnik
                    linesize = s->picture.linesize[6 - j];
548
                    c_ptr = s->picture.data[6 - j] + c_offset;
549 1c05e11d Fabrice Bellard
                    ptr = pixels;
550
                    for(y = 0;y < 8; y++) {
551
                        /* convert to 411P */
552 62cf114b Michael Niedermayer
                        c_ptr1 = c_ptr + 8*linesize;
553
                        c_ptr[0]= ptr[0]; c_ptr1[0]= ptr[4];
554
                        c_ptr[1]= ptr[1]; c_ptr1[1]= ptr[5];
555
                        c_ptr[2]= ptr[2]; c_ptr1[2]= ptr[6];
556
                        c_ptr[3]= ptr[3]; c_ptr1[3]= ptr[7];
557
                        c_ptr += linesize;
558 1c05e11d Fabrice Bellard
                        ptr += 8;
559
                    }
560
                } else {
561
                    /* don't ask me why they inverted Cb and Cr ! */
562 7458ccbb Roman Shaposhnik
                    idct_put(s->picture.data[6 - j] + c_offset, 
563
                             s->picture.linesize[6 - j], block);
564 1c05e11d Fabrice Bellard
                }
565 5d2231ee Fabrice Bellard
            }
566
            block += 64;
567 7be166e4 Fabrice Bellard
            mb++;
568 5d2231ee Fabrice Bellard
        }
569
    }
570
}
571
572 e15c3302 Roman Shaposhnik
#ifdef DV_CODEC_TINY_TARGET
573 4fa1c4fa Roman Shaposhnik
/* Converts run and level (where level != 0) pair into vlc, returning bit size */
574 e15c3302 Roman Shaposhnik
static always_inline int dv_rl2vlc(int run, int l, uint32_t* vlc)
575 4fa1c4fa Roman Shaposhnik
{
576
    int sign = l >> 8;
577
    int level = (l ^ sign) - sign;
578
    int size;
579
    
580
    sign = (sign & 1);
581
582 e15c3302 Roman Shaposhnik
    if (run < DV_VLC_MAP_RUN_SIZE && level < DV_VLC_MAP_LEV_SIZE) {
583
        *vlc = dv_vlc_map[run][level].vlc | sign;
584
        size = dv_vlc_map[run][level].size;
585 4fa1c4fa Roman Shaposhnik
    }
586
    else { 
587 e15c3302 Roman Shaposhnik
        if (level < DV_VLC_MAP_LEV_SIZE) {
588
            *vlc = dv_vlc_map[0][level].vlc | sign;
589
            size = dv_vlc_map[0][level].size;
590 4fa1c4fa Roman Shaposhnik
        } else {
591 e15c3302 Roman Shaposhnik
            *vlc = 0xfe00 | (level << 1) | sign;
592 4fa1c4fa Roman Shaposhnik
            size = 16;
593
        }
594 e15c3302 Roman Shaposhnik
        if (run) {
595
            *vlc |= ((run < 16) ? dv_vlc_map[run-1][0].vlc : 
596
                                  (0x1f80 | (run - 1))) << size;
597
            size += (run < 16) ? dv_vlc_map[run-1][0].size : 13;
598 4fa1c4fa Roman Shaposhnik
        }
599
    }
600
    
601
    return size;
602
}
603
604 e15c3302 Roman Shaposhnik
static always_inline int dv_rl2vlc_size(int run, int l)
605
{
606
    int level = (l ^ (l >> 8)) - (l >> 8);
607
    int size;
608
    
609
    if (run < DV_VLC_MAP_RUN_SIZE && level < DV_VLC_MAP_LEV_SIZE) {
610
        size = dv_vlc_map[run][level].size; 
611
    }
612
    else { 
613
        size = (level < DV_VLC_MAP_LEV_SIZE) ? dv_vlc_map[0][level].size : 16;
614
        if (run) {
615
            size += (run < 16) ? dv_vlc_map[run-1][0].size : 13;
616
        }
617
    }
618
    return size;
619
}
620
#else
621
static always_inline int dv_rl2vlc(int run, int l, uint32_t* vlc)
622
{
623
    *vlc = dv_vlc_map[run][((uint16_t)l)&0x1ff].vlc;
624
    return dv_vlc_map[run][((uint16_t)l)&0x1ff].size;
625
}
626
627
static always_inline int dv_rl2vlc_size(int run, int l)
628
{
629
    return dv_vlc_map[run][((uint16_t)l)&0x1ff].size;
630
}
631
#endif
632
633 4fa1c4fa Roman Shaposhnik
typedef struct EncBlockInfo {
634 e15c3302 Roman Shaposhnik
    int area_q[4];
635
    int bit_size[4];
636
    int prev_run[4];
637
    int cur_ac;
638 4fa1c4fa Roman Shaposhnik
    int cno;
639
    int dct_mode;
640
    DCTELEM *mb;
641 e15c3302 Roman Shaposhnik
    uint8_t partial_bit_count;
642
    uint32_t partial_bit_buffer; /* we can't use uint16_t here */
643 4fa1c4fa Roman Shaposhnik
} EncBlockInfo;
644
645 e15c3302 Roman Shaposhnik
static always_inline int dv_bits_left(PutBitContext* s)
646 4fa1c4fa Roman Shaposhnik
{
647 e15c3302 Roman Shaposhnik
    return (s->buf_end - s->buf) * 8 - 
648 a8380f44 Michael Niedermayer
           ((s->buf_ptr - s->buf) * 8 + 32 - (int64_t)s->bit_left);
649 4fa1c4fa Roman Shaposhnik
}
650
651 e15c3302 Roman Shaposhnik
static always_inline void dv_encode_ac(EncBlockInfo* bi, PutBitContext* pb_pool, 
652
                                       int pb_size)
653 4fa1c4fa Roman Shaposhnik
{
654 e15c3302 Roman Shaposhnik
    int run;
655
    int bits_left;
656
    PutBitContext* pb = pb_pool;
657
    int size = bi->partial_bit_count;
658
    uint32_t vlc = bi->partial_bit_buffer;
659 4fa1c4fa Roman Shaposhnik
    
660 e15c3302 Roman Shaposhnik
    bi->partial_bit_count = bi->partial_bit_buffer = 0;
661
vlc_loop:
662
       /* Find suitable storage space */
663
       for (; size > (bits_left = dv_bits_left(pb)); pb++) {
664
          if (bits_left) {
665
              size -= bits_left;
666
              put_bits(pb, bits_left, vlc >> size);
667
              vlc = vlc & ((1<<size)-1);
668
          }
669
          if (pb_size == 1) {
670
              bi->partial_bit_count = size;
671
              bi->partial_bit_buffer = vlc;
672
              return;
673
          }
674
          --pb_size;
675
       }
676
       
677
       /* Store VLC */
678
       put_bits(pb, size, vlc);
679
       
680
       /* Construct the next VLC */
681
       run = 0;
682
       for (; bi->cur_ac < 64; bi->cur_ac++, run++) {
683
           if (bi->mb[bi->cur_ac]) {
684
               size = dv_rl2vlc(run, bi->mb[bi->cur_ac], &vlc);
685
               bi->cur_ac++;
686
               goto vlc_loop;
687 4fa1c4fa Roman Shaposhnik
           }
688 e15c3302 Roman Shaposhnik
       }
689 4fa1c4fa Roman Shaposhnik
   
690 e15c3302 Roman Shaposhnik
       if (bi->cur_ac == 64) {
691
           size = 4; vlc = 6; /* End Of Block stamp */
692
           bi->cur_ac++;
693
           goto vlc_loop;
694
       }
695 4fa1c4fa Roman Shaposhnik
}
696
697 e15c3302 Roman Shaposhnik
static always_inline void dv_set_class_number(DCTELEM* blk, EncBlockInfo* bi, 
698
                                              const uint8_t* zigzag_scan, int bias)
699 4fa1c4fa Roman Shaposhnik
{
700 e15c3302 Roman Shaposhnik
    int i, area;
701
    int run;
702
    int classes[] = {12, 24, 36, 0xffff};
703
704
    run = 0;
705
    bi->mb[0] = blk[0]; 
706
    bi->cno = 0;
707
    for (area = 0; area < 4; area++) {
708
       bi->prev_run[area] = run;
709
       bi->bit_size[area] = 0;
710
       for (i=mb_area_start[area]; i<mb_area_start[area+1]; i++) {
711
          bi->mb[i] = (blk[zigzag_scan[i]] / 16);
712
          while ((bi->mb[i] ^ (bi->mb[i] >> 8)) > classes[bi->cno])
713
              bi->cno++;
714
       
715
          if (bi->mb[i]) {
716
              bi->bit_size[area] += dv_rl2vlc_size(run, bi->mb[i]);
717
              run = 0;
718
          } else
719
              ++run;
720
       }
721 4fa1c4fa Roman Shaposhnik
    }
722 e15c3302 Roman Shaposhnik
    bi->bit_size[3] += 4; /* EOB marker */
723
    bi->cno += bias;
724 4fa1c4fa Roman Shaposhnik
    
725 e15c3302 Roman Shaposhnik
    if (bi->cno >= 3) { /* FIXME: we have to recreate bit_size[], prev_run[] */
726 4fa1c4fa Roman Shaposhnik
        bi->cno = 3;
727 e15c3302 Roman Shaposhnik
        for (i=1; i<64; i++)
728
           bi->mb[i] /= 2;
729
    }
730 4fa1c4fa Roman Shaposhnik
}
731
732 e15c3302 Roman Shaposhnik
#define SC(x, y) ((s[x] - s[y]) ^ ((s[x] - s[y]) >> 7))
733
static always_inline int dv_guess_dct_mode(DCTELEM *blk) {
734
    DCTELEM *s;
735
    int score88 = 0;
736
    int score248 = 0;
737
    int i;
738 10acc479 Roman Shaposhnik
    
739 e15c3302 Roman Shaposhnik
    /* Compute 8-8 score (small values give a better chance for 8-8 DCT) */
740
    s = blk;
741
    for(i=0; i<7; i++) {
742
        score88 += SC(0,  8) + SC(1, 9) + SC(2, 10) + SC(3, 11) + 
743
                   SC(4, 12) + SC(5,13) + SC(6, 14) + SC(7, 15);
744
        s += 8;
745
    }
746
    /* Compute 2-4-8 score (small values give a better chance for 2-4-8 DCT) */
747
    s = blk;
748
    for(i=0; i<6; i++) {
749
        score248 += SC(0, 16) + SC(1,17) + SC(2, 18) + SC(3, 19) +
750
                    SC(4, 20) + SC(5,21) + SC(6, 22) + SC(7, 23);
751
        s += 8;
752 10acc479 Roman Shaposhnik
    }
753 e15c3302 Roman Shaposhnik
754
    return (score88 - score248 > -10);
755
}
756
757
static inline void dv_guess_qnos(EncBlockInfo* blks, int* qnos)
758
{
759
    int size[5];
760
    int i, j, k, a, run;
761
    EncBlockInfo* b;
762 10acc479 Roman Shaposhnik
    
763 e15c3302 Roman Shaposhnik
    do {
764
       b = blks;
765
       for (i=0; i<5; i++) {
766
          if (!qnos[i])
767
              continue;
768
          
769
          qnos[i]--;
770
          size[i] = 0;
771
          for (j=0; j<6; j++, b++) {
772
             for (a=0; a<4; a++) {
773
                if (b->area_q[a] != dv_quant_shifts[qnos[i] + dv_quant_offset[b->cno]][a]) {
774
                    b->bit_size[a] = (a==3)?4:0;
775
                    b->area_q[a]++;
776
                    run = b->prev_run[a];
777
                    for (k=mb_area_start[a]; k<mb_area_start[a+1]; k++) {
778
                       b->mb[k] /= 2;
779
                       if (b->mb[k]) {
780
                           b->bit_size[a] += dv_rl2vlc_size(run, b->mb[k]);
781
                           run = 0;
782
                       } else
783
                           ++run;
784
                    }
785
                }
786
                size[i] += b->bit_size[a];
787
             }
788
          }
789
       }
790
    } while ((vs_total_ac_bits < size[0] + size[1] + size[2] + size[3] + size[4]) && 
791
             (qnos[0]|qnos[1]|qnos[2]|qnos[3]|qnos[4]));
792 10acc479 Roman Shaposhnik
}
793
794 4fa1c4fa Roman Shaposhnik
/*
795
 * This is a very rough initial implementaion. The performance is
796 10acc479 Roman Shaposhnik
 * horrible and the weighting is missing. But it's missing from the 
797
 * decoding step also -- so at least we're on the same page with decoder ;-)
798 4fa1c4fa Roman Shaposhnik
 */
799
static inline void dv_encode_video_segment(DVVideoDecodeContext *s, 
800
                                           uint8_t *dif, 
801
                                           const uint16_t *mb_pos_ptr)
802
{
803
    int mb_index, i, j, v;
804
    int mb_x, mb_y, c_offset, linesize; 
805
    uint8_t*  y_ptr;
806
    uint8_t*  data;
807 e15c3302 Roman Shaposhnik
    uint8_t*  ptr;
808 4fa1c4fa Roman Shaposhnik
    int       do_edge_wrap;
809 e15c3302 Roman Shaposhnik
    DCTELEM   block[64] __align8;
810 4fa1c4fa Roman Shaposhnik
    EncBlockInfo  enc_blks[5*6];
811 e15c3302 Roman Shaposhnik
    PutBitContext pbs[5*6];
812
    PutBitContext* pb; 
813 4fa1c4fa Roman Shaposhnik
    EncBlockInfo* enc_blk;
814 e15c3302 Roman Shaposhnik
    int       vs_bit_size = 0;
815
    int       qnos[5];
816 4fa1c4fa Roman Shaposhnik
   
817 10acc479 Roman Shaposhnik
    enc_blk = &enc_blks[0];
818 e15c3302 Roman Shaposhnik
    pb = &pbs[0];
819 4fa1c4fa Roman Shaposhnik
    for(mb_index = 0; mb_index < 5; mb_index++) {
820
        v = *mb_pos_ptr++;
821
        mb_x = v & 0xff;
822
        mb_y = v >> 8;
823
        y_ptr = s->picture.data[0] + (mb_y * s->picture.linesize[0] * 8) + (mb_x * 8);
824
        c_offset = (s->sys->pix_fmt == PIX_FMT_YUV411P) ?
825
                   ((mb_y * s->picture.linesize[1] * 8) + ((mb_x >> 2) * 8)) :
826
                   (((mb_y >> 1) * s->picture.linesize[1] * 8) + ((mb_x >> 1) * 8));
827
        do_edge_wrap = 0;
828 e15c3302 Roman Shaposhnik
        qnos[mb_index] = 15; /* No quantization */
829
        ptr = dif + mb_index*80 + 4;
830 4fa1c4fa Roman Shaposhnik
        for(j = 0;j < 6; j++) {
831
            if (j < 4) {  /* Four Y blocks */
832
                /* NOTE: at end of line, the macroblock is handled as 420 */
833
                if (s->sys->pix_fmt == PIX_FMT_YUV411P && mb_x < (704 / 8)) {
834
                    data = y_ptr + (j * 8);
835
                } else {
836
                    data = y_ptr + ((j & 1) * 8) + ((j >> 1) * 8 * s->picture.linesize[0]);
837
                }
838
                linesize = s->picture.linesize[0];
839
            } else {      /* Cr and Cb blocks */
840
                /* don't ask Fabrice why they inverted Cb and Cr ! */
841
                data = s->picture.data[6 - j] + c_offset;
842
                linesize = s->picture.linesize[6 - j];
843
                if (s->sys->pix_fmt == PIX_FMT_YUV411P && mb_x >= (704 / 8))
844
                    do_edge_wrap = 1;
845
            }        
846
            
847
            /* Everything is set up -- now just copy data -> DCT block */
848
            if (do_edge_wrap) {  /* Edge wrap copy: 4x16 -> 8x8 */
849
                uint8_t* d;
850
                DCTELEM *b = block;
851
                for (i=0;i<8;i++) {
852
                   d = data + 8 * linesize;
853
                   b[0] = data[0]; b[1] = data[1]; b[2] = data[2]; b[3] = data[3];
854
                   b[4] =    d[0]; b[5] =    d[1]; b[6] =    d[2]; b[7] =    d[3];
855
                   data += linesize;
856
                   b += 8;
857
                }
858
            } else {             /* Simple copy: 8x8 -> 8x8 */
859
                s->get_pixels(block, data, linesize);
860
            }
861 10acc479 Roman Shaposhnik
          
862 e15c3302 Roman Shaposhnik
            enc_blk->dct_mode = dv_guess_dct_mode(block);
863
            enc_blk->mb = &s->block[mb_index*6+j][0];
864
            enc_blk->area_q[0] = enc_blk->area_q[1] = enc_blk->area_q[2] = enc_blk->area_q[3] = 0;
865
            enc_blk->partial_bit_count = 0;
866
            enc_blk->partial_bit_buffer = 0;
867
            enc_blk->cur_ac = 1;
868 10acc479 Roman Shaposhnik
            
869
            s->fdct[enc_blk->dct_mode](block);
870 4fa1c4fa Roman Shaposhnik
            
871 e15c3302 Roman Shaposhnik
            dv_set_class_number(block, enc_blk, 
872
                                enc_blk->dct_mode ? ff_zigzag248_direct : ff_zigzag_direct,
873
                                j/4*(j%2));
874
           
875
            init_put_bits(pb, ptr, block_sizes[j]/8);
876
            put_bits(pb, 9, (uint16_t)(((enc_blk->mb[0] >> 3) - 1024) >> 2));
877
            put_bits(pb, 1, enc_blk->dct_mode);
878
            put_bits(pb, 2, enc_blk->cno);
879
            
880
            vs_bit_size += enc_blk->bit_size[0] + enc_blk->bit_size[1] +
881
                           enc_blk->bit_size[2] + enc_blk->bit_size[3];
882
            ++enc_blk;
883
            ++pb;
884
            ptr += block_sizes[j]/8;
885 4fa1c4fa Roman Shaposhnik
        }
886
    }
887
888 e15c3302 Roman Shaposhnik
    if (vs_total_ac_bits < vs_bit_size)
889
        dv_guess_qnos(&enc_blks[0], &qnos[0]);
890 4fa1c4fa Roman Shaposhnik
891
    for (i=0; i<5; i++) {
892 e15c3302 Roman Shaposhnik
       dif[i*80 + 3] = qnos[i];
893
    }
894 4fa1c4fa Roman Shaposhnik
895 e15c3302 Roman Shaposhnik
    /* First pass over individual cells only */
896
    for (j=0; j<5*6; j++)
897
       dv_encode_ac(&enc_blks[j], &pbs[j], 1);
898 4fa1c4fa Roman Shaposhnik
899 e15c3302 Roman Shaposhnik
    /* Second pass over each MB space */
900
    for (j=0; j<5*6; j++) {
901
       if (enc_blks[j].cur_ac < 65 || enc_blks[j].partial_bit_count)
902
           dv_encode_ac(&enc_blks[j], &pbs[(j/6)*6], 6);
903 4fa1c4fa Roman Shaposhnik
    }
904
905 e15c3302 Roman Shaposhnik
    /* Third and final pass over the whole vides segment space */
906
    for (j=0; j<5*6; j++) {
907
       if (enc_blks[j].cur_ac < 65 || enc_blks[j].partial_bit_count)
908
           dv_encode_ac(&enc_blks[j], &pbs[0], 6*5);
909 4fa1c4fa Roman Shaposhnik
    }
910 e15c3302 Roman Shaposhnik
911
    for (j=0; j<5*6; j++)
912
       flush_put_bits(&pbs[j]);
913 4fa1c4fa Roman Shaposhnik
}
914
915 5d2231ee Fabrice Bellard
/* NOTE: exactly one frame must be given (120000 bytes for NTSC,
916
   144000 bytes for PAL) */
917
static int dvvideo_decode_frame(AVCodecContext *avctx, 
918
                                 void *data, int *data_size,
919 0c1a9eda Zdenek Kabelac
                                 uint8_t *buf, int buf_size)
920 5d2231ee Fabrice Bellard
{
921
    DVVideoDecodeContext *s = avctx->priv_data;
922 7458ccbb Roman Shaposhnik
    int ds, vs;
923 0c1a9eda Zdenek Kabelac
    const uint16_t *mb_pos_ptr;
924 5d2231ee Fabrice Bellard
    
925 7458ccbb Roman Shaposhnik
    s->sys = dv_frame_profile(buf);
926
    if (!s->sys || buf_size < s->sys->frame_size)
927
        return -1; /* NOTE: we only accept several full frames */
928 e9feea59 Arpi
929 7458ccbb Roman Shaposhnik
        
930 e20c4069 Michael Niedermayer
    if(s->picture.data[0])
931
        avctx->release_buffer(avctx, &s->picture);
932
    
933 7458ccbb Roman Shaposhnik
    s->picture.reference = 0;
934
    avctx->pix_fmt = s->sys->pix_fmt;
935 86a7e115 Roman Shaposhnik
    avctx->width = s->sys->width;
936
    avctx->height = s->sys->height;
937 1e491e29 Michael Niedermayer
    if(avctx->get_buffer(avctx, &s->picture) < 0) {
938 9b879566 Michel Bardiaux
        av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
939 1e491e29 Michael Niedermayer
        return -1;
940 b0397e0e Zdenek Kabelac
    }
941 2744ca9a Roman Shaposhnik
    s->picture.interlaced_frame = 1;
942 9dad924e Michael Niedermayer
    s->picture.top_field_first = 0;
943 b0397e0e Zdenek Kabelac
944 5d2231ee Fabrice Bellard
    /* for each DIF segment */
945 7458ccbb Roman Shaposhnik
    mb_pos_ptr = s->sys->video_place;
946
    for (ds = 0; ds < s->sys->difseg_size; ds++) {
947
        buf += 6 * 80; /* skip DIF segment header */
948 5d2231ee Fabrice Bellard
        
949
        for(vs = 0; vs < 27; vs++) {
950 7458ccbb Roman Shaposhnik
            if ((vs % 3) == 0)
951
                buf += 80; /* skip audio block */
952
            
953 4fa1c4fa Roman Shaposhnik
#ifdef VLC_DEBUG
954
            printf("********************* %d, %d **********************\n", ds, vs);
955
#endif
956 7458ccbb Roman Shaposhnik
            dv_decode_video_segment(s, buf, mb_pos_ptr);
957
            buf += 5 * 80;
958 5d2231ee Fabrice Bellard
            mb_pos_ptr += 5;
959
        }
960
    }
961
962 9e398782 Arpi
    emms_c();
963
964 5d2231ee Fabrice Bellard
    /* return image */
965 492cd3a9 Michael Niedermayer
    *data_size = sizeof(AVFrame);
966
    *(AVFrame*)data= s->picture;
967 1e491e29 Michael Niedermayer
    
968 7458ccbb Roman Shaposhnik
    return s->sys->frame_size;
969 5d2231ee Fabrice Bellard
}
970
971 4fa1c4fa Roman Shaposhnik
static int dvvideo_encode_frame(AVCodecContext *c, uint8_t *buf, int buf_size, 
972
                                void *data)
973 5d2231ee Fabrice Bellard
{
974 4fa1c4fa Roman Shaposhnik
    DVVideoDecodeContext *s = c->priv_data;
975
    const uint16_t *mb_pos_ptr;
976
    int ds, vs;
977
978
    s->sys = dv_codec_profile(c);
979
    if (!s->sys)
980
        return -1;
981
    
982
    c->pix_fmt = s->sys->pix_fmt;
983
    s->picture = *((AVFrame *)data);
984
985
    /* for each DIF segment */
986
    mb_pos_ptr = s->sys->video_place;
987
    for (ds = 0; ds < s->sys->difseg_size; ds++) {
988
        buf += 6 * 80; /* skip DIF segment header */
989
        
990
        for(vs = 0; vs < 27; vs++) {
991
            if ((vs % 3) == 0)
992
                buf += 80; /* skip audio block */
993
994
#ifdef VLC_DEBUG
995
            printf("********************* %d, %d **********************\n", ds, vs);
996
#endif
997
            dv_encode_video_segment(s, buf, mb_pos_ptr);
998
            buf += 5 * 80;
999
            mb_pos_ptr += 5;
1000
        }
1001
    }
1002 5d2231ee Fabrice Bellard
1003 4fa1c4fa Roman Shaposhnik
    emms_c();
1004
    return s->sys->frame_size;
1005
}
1006
1007
static int dvvideo_end(AVCodecContext *avctx)
1008
{
1009
    avcodec_default_free_buffers(avctx);    
1010 5d2231ee Fabrice Bellard
    return 0;
1011
}
1012
1013
AVCodec dvvideo_decoder = {
1014
    "dvvideo",
1015
    CODEC_TYPE_VIDEO,
1016
    CODEC_ID_DVVIDEO,
1017
    sizeof(DVVideoDecodeContext),
1018 4fa1c4fa Roman Shaposhnik
    dvvideo_init,
1019
    dvvideo_encode_frame,
1020
    dvvideo_end,
1021 5d2231ee Fabrice Bellard
    dvvideo_decode_frame,
1022 b0397e0e Zdenek Kabelac
    CODEC_CAP_DR1,
1023 5d2231ee Fabrice Bellard
    NULL
1024
};