/*
* slice.c
* Copyright (C) 2000-2002 Michel Lespinasse <walken@zoy.org>
* Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
*
* This file is part of mpeg2dec, a free MPEG-2 video stream decoder.
* See http://libmpeg2.sourceforge.net/ for updates.
*
* mpeg2dec is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* mpeg2dec 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include "config.h"
#include <inttypes.h>
#include "xine_internal.h"
#include "video_out.h"
#include "mpeg2_internal.h"
#include "attributes.h"
extern mpeg2_mc_t mpeg2_mc;
extern void (* mpeg2_idct_copy) (int16_t * block, uint8_t * dest, int stride);
extern void (* mpeg2_idct_add) (int16_t * block, uint8_t * dest, int stride);
extern void (* mpeg2_cpu_state_save) (cpu_state_t * state);
extern void (* mpeg2_cpu_state_restore) (cpu_state_t * state);
#include "vlc.h"
static int non_linear_quantizer_scale [] = {
0, 1, 2, 3, 4, 5, 6, 7,
8, 10, 12, 14, 16, 18, 20, 22,
24, 28, 32, 36, 40, 44, 48, 52,
56, 64, 72, 80, 88, 96, 104, 112
};
static inline int get_macroblock_modes (picture_t * picture)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int macroblock_modes;
MBtab * tab;
switch (picture->picture_coding_type) {
case I_TYPE:
tab = MB_I + UBITS (bit_buf, 1);
DUMPBITS (bit_buf, bits, tab->len);
macroblock_modes = tab->modes;
if ((! (picture->frame_pred_frame_dct)) &&
(picture->picture_structure == FRAME_PICTURE)) {
macroblock_modes |= UBITS (bit_buf, 1) * DCT_TYPE_INTERLACED;
DUMPBITS (bit_buf, bits, 1);
}
return macroblock_modes;
case P_TYPE:
tab = MB_P + UBITS (bit_buf, 5);
DUMPBITS (bit_buf, bits, tab->len);
macroblock_modes = tab->modes;
if (picture->picture_structure != FRAME_PICTURE) {
if (macroblock_modes & MACROBLOCK_MOTION_FORWARD) {
macroblock_modes |= UBITS (bit_buf, 2) * MOTION_TYPE_BASE;
DUMPBITS (bit_buf, bits, 2);
}
return macroblock_modes;
} else if (picture->frame_pred_frame_dct) {
if (macroblock_modes & MACROBLOCK_MOTION_FORWARD)
macroblock_modes |= MC_FRAME;
return macroblock_modes;
} else {
if (macroblock_modes & MACROBLOCK_MOTION_FORWARD) {
macroblock_modes |= UBITS (bit_buf, 2) * MOTION_TYPE_BASE;
DUMPBITS (bit_buf, bits, 2);
}
if (macroblock_modes & (MACROBLOCK_INTRA | MACROBLOCK_PATTERN)) {
macroblock_modes |= UBITS (bit_buf, 1) * DCT_TYPE_INTERLACED;
DUMPBITS (bit_buf, bits, 1);
}
return macroblock_modes;
}
case B_TYPE:
tab = MB_B + UBITS (bit_buf, 6);
DUMPBITS (bit_buf, bits, tab->len);
macroblock_modes = tab->modes;
if (picture->picture_structure != FRAME_PICTURE) {
if (! (macroblock_modes & MACROBLOCK_INTRA)) {
macroblock_modes |= UBITS (bit_buf, 2) * MOTION_TYPE_BASE;
DUMPBITS (bit_buf, bits, 2);
}
return macroblock_modes;
} else if (picture->frame_pred_frame_dct) {
/* if (! (macroblock_modes & MACROBLOCK_INTRA)) */
macroblock_modes |= MC_FRAME;
return macroblock_modes;
} else {
if (macroblock_modes & MACROBLOCK_INTRA)
goto intra;
macroblock_modes |= UBITS (bit_buf, 2) * MOTION_TYPE_BASE;
DUMPBITS (bit_buf, bits, 2);
if (macroblock_modes & (MACROBLOCK_INTRA | MACROBLOCK_PATTERN)) {
intra:
macroblock_modes |= UBITS (bit_buf, 1) * DCT_TYPE_INTERLACED;
DUMPBITS (bit_buf, bits, 1);
}
return macroblock_modes;
}
case D_TYPE:
DUMPBITS (bit_buf, bits, 1);
return MACROBLOCK_INTRA;
default:
return 0;
}
#undef bit_buf
#undef bits
#undef bit_ptr
}
static inline int get_quantizer_scale (picture_t * picture)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int quantizer_scale_code;
quantizer_scale_code = UBITS (bit_buf, 5);
DUMPBITS (bit_buf, bits, 5);
if (picture->q_scale_type)
return non_linear_quantizer_scale [quantizer_scale_code];
else
return quantizer_scale_code << 1;
#undef bit_buf
#undef bits
#undef bit_ptr
}
static inline int get_motion_delta (picture_t * picture, int f_code)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int delta;
int sign;
MVtab * tab;
if (bit_buf & 0x80000000) {
DUMPBITS (bit_buf, bits, 1);
return 0;
} else if (bit_buf >= 0x0c000000) {
tab = MV_4 + UBITS (bit_buf, 4);
delta = (tab->delta << f_code) + 1;
bits += tab->len + f_code + 1;
bit_buf <<= tab->len;
sign = SBITS (bit_buf, 1);
bit_buf <<= 1;
if (f_code)
delta += UBITS (bit_buf, f_code);
bit_buf <<= f_code;
return (delta ^ sign) - sign;
} else {
tab = MV_10 + UBITS (bit_buf, 10);
delta = (tab->delta << f_code) + 1;
bits += tab->len + 1;
bit_buf <<= tab->len;
sign = SBITS (bit_buf, 1);
bit_buf <<= 1;
if (f_code) {
NEEDBITS (bit_buf, bits, bit_ptr);
delta += UBITS (bit_buf, f_code);
DUMPBITS (bit_buf, bits, f_code);
}
return (delta ^ sign) - sign;
}
#undef bit_buf
#undef bits
#undef bit_ptr
}
static inline int bound_motion_vector (int vector, int f_code)
{
#if 1
unsigned int limit;
int sign;
limit = 16 << f_code;
if ((unsigned int)(vector + limit) < 2 * limit)
return vector;
else {
sign = ((int32_t)vector) >> 31;
return vector - ((2 * limit) ^ sign) + sign;
}
#else
return ((int32_t)vector << (27 - f_code)) >> (27 - f_code);
#endif
}
static inline int get_dmv (picture_t * picture)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
DMVtab * tab;
tab = DMV_2 + UBITS (bit_buf, 2);
DUMPBITS (bit_buf, bits, tab->len);
return tab->dmv;
#undef bit_buf
#undef bits
#undef bit_ptr
}
static inline int get_coded_block_pattern (picture_t * picture)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
CBPtab * tab;
NEEDBITS (bit_buf, bits, bit_ptr);
if (bit_buf >= 0x20000000) {
tab = CBP_7 + (UBITS (bit_buf, 7) - 16);
DUMPBITS (bit_buf, bits, tab->len);
return tab->cbp;
} else {
tab = CBP_9 + UBITS (bit_buf, 9);
DUMPBITS (bit_buf, bits, tab->len);
return tab->cbp;
}
#undef bit_buf
#undef bits
#undef bit_ptr
}
static inline int get_luma_dc_dct_diff (picture_t * picture)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
DCtab * tab;
int size;
int dc_diff;
if (bit_buf < 0xf8000000) {
tab = DC_lum_5 + UBITS (bit_buf, 5);
size = tab->size;
if (size) {
bits += tab->len + size;
bit_buf <<= tab->len;
dc_diff =
UBITS (bit_buf, size) - UBITS (SBITS (~bit_buf, 1), size);
bit_buf <<= size;
return dc_diff;
} else {
DUMPBITS (bit_buf, bits, 3);
return 0;
}
} else {
tab = DC_long + (UBITS (bit_buf, 9) - 0x1e0);
size = tab->size;
DUMPBITS (bit_buf, bits, tab->len);
NEEDBITS (bit_buf, bits, bit_ptr);
dc_diff = UBITS (bit_buf, size) - UBITS (SBITS (~bit_buf, 1), size);
DUMPBITS (bit_buf, bits, size);
return dc_diff;
}
#undef bit_buf
#undef bits
#undef bit_ptr
}
static inline int get_chroma_dc_dct_diff (picture_t * picture)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
DCtab * tab;
int size;
int dc_diff;
if (bit_buf < 0xf8000000) {
tab = DC_chrom_5 + UBITS (bit_buf, 5);
size = tab->size;
if (size) {
bits += tab->len + size;
bit_buf <<= tab->len;
dc_diff =
UBITS (bit_buf, size) - UBITS (SBITS (~bit_buf, 1), size);
bit_buf <<= size;
return dc_diff;
} else {
DUMPBITS (bit_buf, bits, 2);
return 0;
}
} else {
tab = DC_long + (UBITS (bit_buf, 10) - 0x3e0);
size = tab->size;
DUMPBITS (bit_buf, bits, tab->len + 1);
NEEDBITS (bit_buf, bits, bit_ptr);
dc_diff = UBITS (bit_buf, size) - UBITS (SBITS (~bit_buf, 1), size);
DUMPBITS (bit_buf, bits, size);
return dc_diff;
}
#undef bit_buf
#undef bits
#undef bit_ptr
}
#define SATURATE(val) \
do { \
if ((uint32_t)(val + 2048) > 4095) \
val = (val > 0) ? 2047 : -2048; \
} while (0)
static void get_intra_block_B14 (picture_t * picture)
{
int i;
int j;
int val;
uint8_t * scan = picture->scan;
uint8_t * quant_matrix = picture->intra_quantizer_matrix;
int quantizer_scale = picture->quantizer_scale;
int mismatch;
DCTtab * tab;
uint32_t bit_buf;
int bits;
uint8_t * bit_ptr;
int16_t * dest;
dest = picture->DCTblock;
i = 0;
mismatch = ~dest[0];
bit_buf = picture->bitstream_buf;
bits = picture->bitstream_bits;
bit_ptr = picture->bitstream_ptr;
NEEDBITS (bit_buf, bits, bit_ptr);
while (1) {
if (bit_buf >= 0x28000000) {
tab = DCT_B14AC_5 + (UBITS (bit_buf, 5) - 5);
i += tab->run;
if (i >= 64)
break; /* end of block */
normal_code:
j = scan[i];
bit_buf <<= tab->len;
bits += tab->len + 1;
val = (tab->level * quantizer_scale * quant_matrix[j]) >> 4;
/* if (bitstream_get (1)) val = -val; */
val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
SATURATE (val);
dest[j] = val;
mismatch ^= val;
bit_buf <<= 1;
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
} else if (bit_buf >= 0x04000000) {
tab = DCT_B14_8 + (UBITS (bit_buf, 8) - 4);
i += tab->run;
if (i < 64)
goto normal_code;
/* escape code */
i += UBITS (bit_buf << 6, 6) - 64;
if (i >= 64)
break; /* illegal, check needed to avoid buffer overflow */
j = scan[i];
DUMPBITS (bit_buf, bits, 12);
NEEDBITS (bit_buf, bits, bit_ptr);
val = (SBITS (bit_buf, 12) *
quantizer_scale * quant_matrix[j]) / 16;
SATURATE (val);
dest[j] = val;
mismatch ^= val;
DUMPBITS (bit_buf, bits, 12);
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
} else if (bit_buf >= 0x02000000) {
tab = DCT_B14_10 + (UBITS (bit_buf, 10) - 8);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00800000) {
tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00200000) {
tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else {
tab = DCT_16 + UBITS (bit_buf, 16);
bit_buf <<= 16;
GETWORD (bit_buf, bits + 16, bit_ptr);
i += tab->run;
if (i < 64)
goto normal_code;
}
break; /* illegal, check needed to avoid buffer overflow */
}
dest[63] ^= mismatch & 1;
DUMPBITS (bit_buf, bits, 2); /* dump end of block code */
picture->bitstream_buf = bit_buf;
picture->bitstream_bits = bits;
picture->bitstream_ptr = bit_ptr;
}
static void get_intra_block_B15 (picture_t * picture)
{
int i;
int j;
int val;
uint8_t * scan = picture->scan;
uint8_t * quant_matrix = picture->intra_quantizer_matrix;
int quantizer_scale = picture->quantizer_scale;
int mismatch;
DCTtab * tab;
uint32_t bit_buf;
int bits;
uint8_t * bit_ptr;
int16_t * dest;
dest = picture->DCTblock;
i = 0;
mismatch = ~dest[0];
bit_buf = picture->bitstream_buf;
bits = picture->bitstream_bits;
bit_ptr = picture->bitstream_ptr;
NEEDBITS (bit_buf, bits, bit_ptr);
while (1) {
if (bit_buf >= 0x04000000) {
tab = DCT_B15_8 + (UBITS (bit_buf, 8) - 4);
i += tab->run;
if (i < 64) {
normal_code:
j = scan[i];
bit_buf <<= tab->len;
bits += tab->len + 1;
val = (tab->level * quantizer_scale * quant_matrix[j]) >> 4;
/* if (bitstream_get (1)) val = -val; */
val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
SATURATE (val);
dest[j] = val;
mismatch ^= val;
bit_buf <<= 1;
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
} else {
/* end of block. I commented out this code because if we */
/* dont exit here we will still exit at the later test :) */
/* if (i >= 128) break; */ /* end of block */
/* escape code */
i += UBITS (bit_buf << 6, 6) - 64;
if (i >= 64)
break; /* illegal, check against buffer overflow */
j = scan[i];
DUMPBITS (bit_buf, bits, 12);
NEEDBITS (bit_buf, bits, bit_ptr);
val = (SBITS (bit_buf, 12) *
quantizer_scale * quant_matrix[j]) / 16;
SATURATE (val);
dest[j] = val;
mismatch ^= val;
DUMPBITS (bit_buf, bits, 12);
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
}
} else if (bit_buf >= 0x02000000) {
tab = DCT_B15_10 + (UBITS (bit_buf, 10) - 8);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00800000) {
tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00200000) {
tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else {
tab = DCT_16 + UBITS (bit_buf, 16);
bit_buf <<= 16;
GETWORD (bit_buf, bits + 16, bit_ptr);
i += tab->run;
if (i < 64)
goto normal_code;
}
break; /* illegal, check needed to avoid buffer overflow */
}
dest[63] ^= mismatch & 1;
DUMPBITS (bit_buf, bits, 4); /* dump end of block code */
picture->bitstream_buf = bit_buf;
picture->bitstream_bits = bits;
picture->bitstream_ptr = bit_ptr;
}
static void get_non_intra_block (picture_t * picture)
{
int i;
int j;
int val;
uint8_t * scan = picture->scan;
uint8_t * quant_matrix = picture->non_intra_quantizer_matrix;
int quantizer_scale = picture->quantizer_scale;
int mismatch;
DCTtab * tab;
uint32_t bit_buf;
int bits;
uint8_t * bit_ptr;
int16_t * dest;
i = -1;
mismatch = 1;
dest = picture->DCTblock;
bit_buf = picture->bitstream_buf;
bits = picture->bitstream_bits;
bit_ptr = picture->bitstream_ptr;
NEEDBITS (bit_buf, bits, bit_ptr);
if (bit_buf >= 0x28000000) {
tab = DCT_B14DC_5 + (UBITS (bit_buf, 5) - 5);
goto entry_1;
} else
goto entry_2;
while (1) {
if (bit_buf >= 0x28000000) {
tab = DCT_B14AC_5 + (UBITS (bit_buf, 5) - 5);
entry_1:
i += tab->run;
if (i >= 64)
break; /* end of block */
normal_code:
j = scan[i];
bit_buf <<= tab->len;
bits += tab->len + 1;
val = ((2*tab->level+1) * quantizer_scale * quant_matrix[j]) >> 5;
/* if (bitstream_get (1)) val = -val; */
val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
SATURATE (val);
dest[j] = val;
mismatch ^= val;
bit_buf <<= 1;
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
}
entry_2:
if (bit_buf >= 0x04000000) {
tab = DCT_B14_8 + (UBITS (bit_buf, 8) - 4);
i += tab->run;
if (i < 64)
goto normal_code;
/* escape code */
i += UBITS (bit_buf << 6, 6) - 64;
if (i >= 64)
break; /* illegal, check needed to avoid buffer overflow */
j = scan[i];
DUMPBITS (bit_buf, bits, 12);
NEEDBITS (bit_buf, bits, bit_ptr);
val = 2 * (SBITS (bit_buf, 12) + SBITS (bit_buf, 1)) + 1;
val = (val * quantizer_scale * quant_matrix[j]) / 32;
SATURATE (val);
dest[j] = val;
mismatch ^= val;
DUMPBITS (bit_buf, bits, 12);
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
} else if (bit_buf >= 0x02000000) {
tab = DCT_B14_10 + (UBITS (bit_buf, 10) - 8);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00800000) {
tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00200000) {
tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else {
tab = DCT_16 + UBITS (bit_buf, 16);
bit_buf <<= 16;
GETWORD (bit_buf, bits + 16, bit_ptr);
i += tab->run;
if (i < 64)
goto normal_code;
}
break; /* illegal, check needed to avoid buffer overflow */
}
dest[63] ^= mismatch & 1;
DUMPBITS (bit_buf, bits, 2); /* dump end of block code */
picture->bitstream_buf = bit_buf;
picture->bitstream_bits = bits;
picture->bitstream_ptr = bit_ptr;
}
static void get_mpeg1_intra_block (picture_t * picture)
{
int i;
int j;
int val;
uint8_t * scan = picture->scan;
uint8_t * quant_matrix = picture->intra_quantizer_matrix;
int quantizer_scale = picture->quantizer_scale;
DCTtab * tab;
uint32_t bit_buf;
int bits;
uint8_t * bit_ptr;
int16_t * dest;
i = 0;
dest = picture->DCTblock;
bit_buf = picture->bitstream_buf;
bits = picture->bitstream_bits;
bit_ptr = picture->bitstream_ptr;
NEEDBITS (bit_buf, bits, bit_ptr);
while (1) {
if (bit_buf >= 0x28000000) {
tab = DCT_B14AC_5 + (UBITS (bit_buf, 5) - 5);
i += tab->run;
if (i >= 64)
break; /* end of block */
normal_code:
j = scan[i];
bit_buf <<= tab->len;
bits += tab->len + 1;
val = (tab->level * quantizer_scale * quant_matrix[j]) >> 4;
/* oddification */
val = (val - 1) | 1;
/* if (bitstream_get (1)) val = -val; */
val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
SATURATE (val);
dest[j] = val;
bit_buf <<= 1;
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
} else if (bit_buf >= 0x04000000) {
tab = DCT_B14_8 + (UBITS (bit_buf, 8) - 4);
i += tab->run;
if (i < 64)
goto normal_code;
/* escape code */
i += UBITS (bit_buf << 6, 6) - 64;
if (i >= 64)
break; /* illegal, check needed to avoid buffer overflow */
j = scan[i];
DUMPBITS (bit_buf, bits, 12);
NEEDBITS (bit_buf, bits, bit_ptr);
val = SBITS (bit_buf, 8);
if (! (val & 0x7f)) {
DUMPBITS (bit_buf, bits, 8);
val = UBITS (bit_buf, 8) + 2 * val;
}
val = (val * quantizer_scale * quant_matrix[j]) / 16;
/* oddification */
val = (val + ~SBITS (val, 1)) | 1;
SATURATE (val);
dest[j] = val;
DUMPBITS (bit_buf, bits, 8);
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
} else if (bit_buf >= 0x02000000) {
tab = DCT_B14_10 + (UBITS (bit_buf, 10) - 8);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00800000) {
tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00200000) {
tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else {
tab = DCT_16 + UBITS (bit_buf, 16);
bit_buf <<= 16;
GETWORD (bit_buf, bits + 16, bit_ptr);
i += tab->run;
if (i < 64)
goto normal_code;
}
break; /* illegal, check needed to avoid buffer overflow */
}
DUMPBITS (bit_buf, bits, 2); /* dump end of block code */
picture->bitstream_buf = bit_buf;
picture->bitstream_bits = bits;
picture->bitstream_ptr = bit_ptr;
}
static void get_mpeg1_non_intra_block (picture_t * picture)
{
int i;
int j;
int val;
uint8_t * scan = picture->scan;
uint8_t * quant_matrix = picture->non_intra_quantizer_matrix;
int quantizer_scale = picture->quantizer_scale;
DCTtab * tab;
uint32_t bit_buf;
int bits;
uint8_t * bit_ptr;
int16_t * dest;
i = -1;
dest = picture->DCTblock;
bit_buf = picture->bitstream_buf;
bits = picture->bitstream_bits;
bit_ptr = picture->bitstream_ptr;
NEEDBITS (bit_buf, bits, bit_ptr);
if (bit_buf >= 0x28000000) {
tab = DCT_B14DC_5 + (UBITS (bit_buf, 5) - 5);
goto entry_1;
} else
goto entry_2;
while (1) {
if (bit_buf >= 0x28000000) {
tab = DCT_B14AC_5 + (UBITS (bit_buf, 5) - 5);
entry_1:
i += tab->run;
if (i >= 64)
break; /* end of block */
normal_code:
j = scan[i];
bit_buf <<= tab->len;
bits += tab->len + 1;
val = ((2*tab->level+1) * quantizer_scale * quant_matrix[j]) >> 5;
/* oddification */
val = (val - 1) | 1;
/* if (bitstream_get (1)) val = -val; */
val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
SATURATE (val);
dest[j] = val;
bit_buf <<= 1;
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
}
entry_2:
if (bit_buf >= 0x04000000) {
tab = DCT_B14_8 + (UBITS (bit_buf, 8) - 4);
i += tab->run;
if (i < 64)
goto normal_code;
/* escape code */
i += UBITS (bit_buf << 6, 6) - 64;
if (i >= 64)
break; /* illegal, check needed to avoid buffer overflow */
j = scan[i];
DUMPBITS (bit_buf, bits, 12);
NEEDBITS (bit_buf, bits, bit_ptr);
val = SBITS (bit_buf, 8);
if (! (val & 0x7f)) {
DUMPBITS (bit_buf, bits, 8);
val = UBITS (bit_buf, 8) + 2 * val;
}
val = 2 * (val + SBITS (val, 1)) + 1;
val = (val * quantizer_scale * quant_matrix[j]) / 32;
/* oddification */
val = (val + ~SBITS (val, 1)) | 1;
SATURATE (val);
dest[j] = val;
DUMPBITS (bit_buf, bits, 8);
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
} else if (bit_buf >= 0x02000000) {
tab = DCT_B14_10 + (UBITS (bit_buf, 10) - 8);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00800000) {
tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else if (bit_buf >= 0x00200000) {
tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
i += tab->run;
if (i < 64)
goto normal_code;
} else {
tab = DCT_16 + UBITS (bit_buf, 16);
bit_buf <<= 16;
GETWORD (bit_buf, bits + 16, bit_ptr);
i += tab->run;
if (i < 64)
goto normal_code;
}
break; /* illegal, check needed to avoid buffer overflow */
}
DUMPBITS (bit_buf, bits, 2); /* dump end of block code */
picture->bitstream_buf = bit_buf;
picture->bitstream_bits = bits;
picture->bitstream_ptr = bit_ptr;
}
static inline void slice_intra_DCT (picture_t * picture, int cc,
uint8_t * dest, int stride)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
NEEDBITS (bit_buf, bits, bit_ptr);
/* Get the intra DC coefficient and inverse quantize it */
if (cc == 0)
picture->dc_dct_pred[0] += get_luma_dc_dct_diff (picture);
else
picture->dc_dct_pred[cc] += get_chroma_dc_dct_diff (picture);
picture->DCTblock[0] =
picture->dc_dct_pred[cc] << (3 - picture->intra_dc_precision);
if (picture->mpeg1) {
if (picture->picture_coding_type != D_TYPE)
get_mpeg1_intra_block (picture);
} else if (picture->intra_vlc_format)
get_intra_block_B15 (picture);
else
get_intra_block_B14 (picture);
mpeg2_idct_copy (picture->DCTblock, dest, stride);
#undef bit_buf
#undef bits
#undef bit_ptr
}
static inline void slice_non_intra_DCT (picture_t * picture, uint8_t * dest,
int stride)
{
if (picture->mpeg1)
get_mpeg1_non_intra_block (picture);
else
get_non_intra_block (picture);
mpeg2_idct_add (picture->DCTblock, dest, stride);
}
#define MOTION(table,ref,motion_x,motion_y,size,y) \
pos_x = 2 * picture->offset + motion_x; \
pos_y = 2 * picture->v_offset + motion_y + 2 * y; \
if (pos_x > picture->limit_x) { \
pos_x = ((int)pos_x < 0) ? 0 : picture->limit_x; \
motion_x = pos_x - 2 * picture->offset; \
} \
if (pos_y > picture->limit_y_ ## size){ \
pos_y = ((int)pos_y < 0) ? 0 : picture->limit_y_ ## size; \
motion_y = pos_y - 2 * picture->v_offset - 2 * y; \
} \
xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
table[xy_half] (picture->dest[0] + y * picture->pitches[0] + \
picture->offset, ref[0] + (pos_x >> 1) + \
(pos_y >> 1) * picture->pitches[0], picture->pitches[0], \
size); \
motion_x /= 2; motion_y /= 2; \
xy_half = ((motion_y & 1) << 1) | (motion_x & 1); \
table[4+xy_half] (picture->dest[1] + y/2 * picture->pitches[1] + \
(picture->offset >> 1), ref[1] + \
(((picture->offset + motion_x) >> 1) + \
((((picture->v_offset + motion_y) >> 1) + y/2) * \
picture->pitches[1])), picture->pitches[1], size/2); \
table[4+xy_half] (picture->dest[2] + y/2 * picture->pitches[2] + \
(picture->offset >> 1), ref[2] + \
(((picture->offset + motion_x) >> 1) + \
((((picture->v_offset + motion_y) >> 1) + y/2) * \
picture->pitches[2])), picture->pitches[2], size/2) \
#define MOTION_FIELD(table,ref,motion_x,motion_y,dest_field,op,src_field) \
pos_x = 2 * picture->offset + motion_x; \
pos_y = picture->v_offset + motion_y; \
if (pos_x > picture->limit_x) { \
pos_x = ((int)pos_x < 0) ? 0 : picture->limit_x; \
motion_x = pos_x - 2 * picture->offset; \
} \
if (pos_y > picture->limit_y){ \
pos_y = ((int)pos_y < 0) ? 0 : picture->limit_y; \
motion_y = pos_y - picture->v_offset; \
} \
xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
table[xy_half] (picture->dest[0] + dest_field * picture->pitches[0] + \
picture->offset, \
(ref[0] + (pos_x >> 1) + \
((pos_y op) + src_field) * picture->pitches[0]), \
2 * picture->pitches[0], 8); \
motion_x /= 2; motion_y /= 2; \
xy_half = ((motion_y & 1) << 1) | (motion_x & 1); \
table[4+xy_half] (picture->dest[1] + dest_field * picture->pitches[1] + \
(picture->offset >> 1), ref[1] + \
(((picture->offset + motion_x) >> 1) + \
(((picture->v_offset >> 1) + \
(motion_y op) + src_field) * picture->pitches[1])), \
2 * picture->pitches[1], 4); \
table[4+xy_half] (picture->dest[2] + dest_field * picture->pitches[2] + \
(picture->offset >> 1), ref[2] + \
(((picture->offset + motion_x) >> 1) + \
(((picture->v_offset >> 1) + \
(motion_y op) + src_field) * picture->pitches[2])), \
2 * picture->pitches[2], 4)
static void motion_mp1 (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int motion_x, motion_y;
unsigned int pos_x, pos_y, xy_half;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_x = (motion->pmv[0][0] +
(get_motion_delta (picture,
motion->f_code[0]) << motion->f_code[1]));
motion_x = bound_motion_vector (motion_x,
motion->f_code[0] + motion->f_code[1]);
motion->pmv[0][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_y = (motion->pmv[0][1] +
(get_motion_delta (picture,
motion->f_code[0]) << motion->f_code[1]));
motion_y = bound_motion_vector (motion_y,
motion->f_code[0] + motion->f_code[1]);
motion->pmv[0][1] = motion_y;
MOTION (table, motion->ref[0], motion_x, motion_y, 16, 0);
#undef bit_buf
#undef bits
#undef bit_ptr
}
static void motion_fr_frame (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int motion_x, motion_y;
unsigned int pos_x, pos_y, xy_half;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_x = motion->pmv[0][0] + get_motion_delta (picture,
motion->f_code[0]);
motion_x = bound_motion_vector (motion_x, motion->f_code[0]);
motion->pmv[1][0] = motion->pmv[0][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_y = motion->pmv[0][1] + get_motion_delta (picture,
motion->f_code[1]);
motion_y = bound_motion_vector (motion_y, motion->f_code[1]);
motion->pmv[1][1] = motion->pmv[0][1] = motion_y;
MOTION (table, motion->ref[0], motion_x, motion_y, 16, 0);
#undef bit_buf
#undef bits
#undef bit_ptr
}
static void motion_fr_field (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int motion_x, motion_y, field;
unsigned int pos_x, pos_y, xy_half;
NEEDBITS (bit_buf, bits, bit_ptr);
field = UBITS (bit_buf, 1);
DUMPBITS (bit_buf, bits, 1);
motion_x = motion->pmv[0][0] + get_motion_delta (picture,
motion->f_code[0]);
motion_x = bound_motion_vector (motion_x, motion->f_code[0]);
motion->pmv[0][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_y = (motion->pmv[0][1] >> 1) + get_motion_delta (picture,
motion->f_code[1]);
/* motion_y = bound_motion_vector (motion_y, motion->f_code[1]); */
motion->pmv[0][1] = motion_y << 1;
MOTION_FIELD (table, motion->ref[0], motion_x, motion_y, 0, & ~1, field);
NEEDBITS (bit_buf, bits, bit_ptr);
field = UBITS (bit_buf, 1);
DUMPBITS (bit_buf, bits, 1);
motion_x = motion->pmv[1][0] + get_motion_delta (picture,
motion->f_code[0]);
motion_x = bound_motion_vector (motion_x, motion->f_code[0]);
motion->pmv[1][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_y = (motion->pmv[1][1] >> 1) + get_motion_delta (picture,
motion->f_code[1]);
/* motion_y = bound_motion_vector (motion_y, motion->f_code[1]); */
motion->pmv[1][1] = motion_y << 1;
MOTION_FIELD (table, motion->ref[0], motion_x, motion_y, 1, & ~1, field);
#undef bit_buf
#undef bits
#undef bit_ptr
}
static void motion_fr_dmv (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int motion_x, motion_y, dmv_x, dmv_y, m, other_x, other_y;
unsigned int pos_x, pos_y, xy_half, offset;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_x = motion->pmv[0][0] + get_motion_delta (picture,
motion->f_code[0]);
motion_x = bound_motion_vector (motion_x, motion->f_code[0]);
motion->pmv[1][0] = motion->pmv[0][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
dmv_x = get_dmv (picture);
motion_y = (motion->pmv[0][1] >> 1) + get_motion_delta (picture,
motion->f_code[1]);
/* motion_y = bound_motion_vector (motion_y, motion->f_code[1]); */
motion->pmv[1][1] = motion->pmv[0][1] = motion_y << 1;
dmv_y = get_dmv (picture);
m = picture->top_field_first ? 1 : 3;
other_x = ((motion_x * m + (motion_x > 0)) >> 1) + dmv_x;
other_y = ((motion_y * m + (motion_y > 0)) >> 1) + dmv_y - 1;
MOTION_FIELD (mpeg2_mc.put, motion->ref[0], other_x, other_y, 0, | 1, 0);
m = picture->top_field_first ? 3 : 1;
other_x = ((motion_x * m + (motion_x > 0)) >> 1) + dmv_x;
other_y = ((motion_y * m + (motion_y > 0)) >> 1) + dmv_y + 1;
MOTION_FIELD (mpeg2_mc.put, motion->ref[0], other_x, other_y, 1, & ~1, 0);
pos_x = 2 * picture->offset + motion_x;
pos_y = picture->v_offset + motion_y;
if(pos_x > picture->limit_x){
pos_x = ((int)pos_x < 0) ? 0 : picture->limit_x;
motion_x = pos_x - 2 * picture->offset;
}
if(pos_y > picture->limit_y){
pos_y = ((int)pos_y < 0) ? 0 : picture->limit_y;
motion_y = pos_y - picture->v_offset;
}
xy_half = ((pos_y & 1) << 1) | (pos_x & 1);
offset = (pos_x >> 1) + (pos_y & ~1) * picture->pitches[0];
mpeg2_mc.avg[xy_half]
(picture->dest[0] + picture->offset,
motion->ref[0][0] + offset, 2 * picture->pitches[0], 8);
mpeg2_mc.avg[xy_half]
(picture->dest[0] + picture->pitches[0] + picture->offset,
motion->ref[0][0] + picture->pitches[0] + offset,
2 * picture->pitches[0], 8);
motion_x /= 2; motion_y /= 2;
xy_half = ((motion_y & 1) << 1) | (motion_x & 1);
offset = (((picture->offset + motion_x) >> 1) +
(((picture->v_offset >> 1) + (motion_y & ~1)) *
picture->pitches[1]));
mpeg2_mc.avg[4+xy_half]
(picture->dest[1] + (picture->offset >> 1),
motion->ref[0][1] + offset, 2 * picture->pitches[1], 4);
mpeg2_mc.avg[4+xy_half]
(picture->dest[1] + picture->pitches[1] + (picture->offset >> 1),
motion->ref[0][1] + picture->pitches[1] + offset,
2 * picture->pitches[1], 4);
offset = (((picture->offset + motion_x) >> 1) +
(((picture->v_offset >> 1) + (motion_y & ~1)) *
picture->pitches[2]));
mpeg2_mc.avg[4+xy_half]
(picture->dest[2] + (picture->offset >> 1),
motion->ref[0][2] + offset, 2 * picture->pitches[2], 4);
mpeg2_mc.avg[4+xy_half]
(picture->dest[2] + picture->pitches[2] + (picture->offset >> 1),
motion->ref[0][2] + picture->pitches[2] + offset,
2 * picture->pitches[2], 4);
#undef bit_buf
#undef bits
#undef bit_ptr
}
static void motion_reuse (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
int motion_x, motion_y;
unsigned int pos_x, pos_y, xy_half;
motion_x = motion->pmv[0][0];
motion_y = motion->pmv[0][1];
MOTION (table, motion->ref[0], motion_x, motion_y, 16, 0);
}
static void motion_zero (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
table[0] (picture->dest[0] + picture->offset,
(motion->ref[0][0] + picture->offset +
picture->v_offset * picture->pitches[0]),
picture->pitches[0], 16);
table[4] (picture->dest[1] + (picture->offset >> 1),
motion->ref[0][1] + (picture->offset >> 1) +
(picture->v_offset >> 1) * picture->pitches[1],
picture->pitches[1], 8);
table[4] (picture->dest[2] + (picture->offset >> 1),
motion->ref[0][2] + (picture->offset >> 1) +
(picture->v_offset >> 1) * picture->pitches[2],
picture->pitches[2], 8);
}
/* like motion_frame, but parsing without actual motion compensation */
static void motion_fr_conceal (picture_t * picture)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int tmp;
NEEDBITS (bit_buf, bits, bit_ptr);
tmp = (picture->f_motion.pmv[0][0] +
get_motion_delta (picture, picture->f_motion.f_code[0]));
tmp = bound_motion_vector (tmp, picture->f_motion.f_code[0]);
picture->f_motion.pmv[1][0] = picture->f_motion.pmv[0][0] = tmp;
NEEDBITS (bit_buf, bits, bit_ptr);
tmp = (picture->f_motion.pmv[0][1] +
get_motion_delta (picture, picture->f_motion.f_code[1]));
tmp = bound_motion_vector (tmp, picture->f_motion.f_code[1]);
picture->f_motion.pmv[1][1] = picture->f_motion.pmv[0][1] = tmp;
DUMPBITS (bit_buf, bits, 1); /* remove marker_bit */
#undef bit_buf
#undef bits
#undef bit_ptr
}
static void motion_fi_field (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int motion_x, motion_y;
uint8_t ** ref_field;
unsigned int pos_x, pos_y, xy_half;
NEEDBITS (bit_buf, bits, bit_ptr);
ref_field = motion->ref2[UBITS (bit_buf, 1)];
DUMPBITS (bit_buf, bits, 1);
motion_x = motion->pmv[0][0] + get_motion_delta (picture,
motion->f_code[0]);
motion_x = bound_motion_vector (motion_x, motion->f_code[0]);
motion->pmv[1][0] = motion->pmv[0][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_y = motion->pmv[0][1] + get_motion_delta (picture,
motion->f_code[1]);
motion_y = bound_motion_vector (motion_y, motion->f_code[1]);
motion->pmv[1][1] = motion->pmv[0][1] = motion_y;
MOTION (table, ref_field, motion_x, motion_y, 16, 0);
#undef bit_buf
#undef bits
#undef bit_ptr
}
static void motion_fi_16x8 (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int motion_x, motion_y;
uint8_t ** ref_field;
unsigned int pos_x, pos_y, xy_half;
NEEDBITS (bit_buf, bits, bit_ptr);
ref_field = motion->ref2[UBITS (bit_buf, 1)];
DUMPBITS (bit_buf, bits, 1);
motion_x = motion->pmv[0][0] + get_motion_delta (picture,
motion->f_code[0]);
motion_x = bound_motion_vector (motion_x, motion->f_code[0]);
motion->pmv[0][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_y = motion->pmv[0][1] + get_motion_delta (picture,
motion->f_code[1]);
motion_y = bound_motion_vector (motion_y, motion->f_code[1]);
motion->pmv[0][1] = motion_y;
MOTION (table, ref_field, motion_x, motion_y, 8, 0);
NEEDBITS (bit_buf, bits, bit_ptr);
ref_field = motion->ref2[UBITS (bit_buf, 1)];
DUMPBITS (bit_buf, bits, 1);
motion_x = motion->pmv[1][0] + get_motion_delta (picture,
motion->f_code[0]);
motion_x = bound_motion_vector (motion_x, motion->f_code[0]);
motion->pmv[1][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_y = motion->pmv[1][1] + get_motion_delta (picture,
motion->f_code[1]);
motion_y = bound_motion_vector (motion_y, motion->f_code[1]);
motion->pmv[1][1] = motion_y;
MOTION (table, ref_field, motion_x, motion_y, 8, 8);
#undef bit_buf
#undef bits
#undef bit_ptr
}
static void motion_fi_dmv (picture_t * picture, motion_t * motion,
void (** table) (uint8_t *, uint8_t *, int, int))
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int motion_x, motion_y, other_x, other_y;
unsigned int pos_x, pos_y, xy_half;
NEEDBITS (bit_buf, bits, bit_ptr);
motion_x = motion->pmv[0][0] + get_motion_delta (picture,
motion->f_code[0]);
motion_x = bound_motion_vector (motion_x, motion->f_code[0]);
motion->pmv[1][0] = motion->pmv[0][0] = motion_x;
NEEDBITS (bit_buf, bits, bit_ptr);
other_x = ((motion_x + (motion_x > 0)) >> 1) + get_dmv (picture);
motion_y = motion->pmv[0][1] + get_motion_delta (picture,
motion->f_code[1]);
motion_y = bound_motion_vector (motion_y, motion->f_code[1]);
motion->pmv[1][1] = motion->pmv[0][1] = motion_y;
other_y = (((motion_y + (motion_y > 0)) >> 1) + get_dmv (picture) +
picture->dmv_offset);
MOTION (mpeg2_mc.put, motion->ref[0], motion_x, motion_y, 16, 0);
MOTION (mpeg2_mc.avg, motion->ref[1], other_x, other_y, 16, 0);
#undef bit_buf
#undef bits
#undef bit_ptr
}
static void motion_fi_conceal (picture_t * picture)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int tmp;
NEEDBITS (bit_buf, bits, bit_ptr);
DUMPBITS (bit_buf, bits, 1); /* remove field_select */
tmp = (picture->f_motion.pmv[0][0] +
get_motion_delta (picture, picture->f_motion.f_code[0]));
tmp = bound_motion_vector (tmp, picture->f_motion.f_code[0]);
picture->f_motion.pmv[1][0] = picture->f_motion.pmv[0][0] = tmp;
NEEDBITS (bit_buf, bits, bit_ptr);
tmp = (picture->f_motion.pmv[0][1] +
get_motion_delta (picture, picture->f_motion.f_code[1]));
tmp = bound_motion_vector (tmp, picture->f_motion.f_code[1]);
picture->f_motion.pmv[1][1] = picture->f_motion.pmv[0][1] = tmp;
DUMPBITS (bit_buf, bits, 1); /* remove marker_bit */
#undef bit_buf
#undef bits
#undef bit_ptr
}
#define MOTION_CALL(routine,direction) \
do { \
if ((direction) & MACROBLOCK_MOTION_FORWARD) \
routine (picture, &(picture->f_motion), mpeg2_mc.put); \
if ((direction) & MACROBLOCK_MOTION_BACKWARD) \
routine (picture, &(picture->b_motion), \
((direction) & MACROBLOCK_MOTION_FORWARD ? \
mpeg2_mc.avg : mpeg2_mc.put)); \
} while (0)
#define NEXT_MACROBLOCK \
do { \
picture->offset += 16; \
if (picture->offset == picture->coded_picture_width) { \
do { /* just so we can use the break statement */ \
if (picture->current_frame->copy) { \
picture->current_frame->copy (picture->current_frame, \
picture->dest); \
} \
picture->dest[0] += 16 * picture->pitches[0]; \
picture->dest[1] += 8 * picture->pitches[1]; \
picture->dest[2] += 8 * picture->pitches[2]; \
} while (0); \
picture->v_offset += 16; \
if (picture->v_offset > picture->limit_y) { \
if (mpeg2_cpu_state_restore) \
mpeg2_cpu_state_restore (&cpu_state); \
return; \
} \
picture->offset = 0; \
} \
} while (0)
static inline int slice_init (picture_t * picture, int code)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
int offset, height;
struct vo_frame_s * forward_reference_frame;
struct vo_frame_s * backward_reference_frame;
MBAtab * mba;
offset = picture->picture_structure == BOTTOM_FIELD;
picture->pitches[0] = picture->current_frame->pitches[0];
picture->pitches[1] = picture->current_frame->pitches[1];
picture->pitches[2] = picture->current_frame->pitches[2];
if( picture->forward_reference_frame ) {
forward_reference_frame = picture->forward_reference_frame;
}
else {
/* return 1; */
forward_reference_frame = picture->current_frame;
}
if( picture->backward_reference_frame ) {
backward_reference_frame = picture->backward_reference_frame;
}
else {
/* return 1; */
backward_reference_frame = picture->current_frame;
}
picture->f_motion.ref[0][0] =
forward_reference_frame->base[0] + (offset ? picture->pitches[0] : 0);
picture->f_motion.ref[0][1] =
forward_reference_frame->base[1] + (offset ? picture->pitches[1] : 0);
picture->f_motion.ref[0][2] =
forward_reference_frame->base[2] + (offset ? picture->pitches[2] : 0);
picture->b_motion.ref[0][0] =
backward_reference_frame->base[0] + (offset ? picture->pitches[0] : 0);
picture->b_motion.ref[0][1] =
backward_reference_frame->base[1] + (offset ? picture->pitches[1] : 0);
picture->b_motion.ref[0][2] =
backward_reference_frame->base[2] + (offset ? picture->pitches[2] : 0);
if (picture->picture_structure != FRAME_PICTURE) {
uint8_t ** forward_ref;
int bottom_field;
bottom_field = (picture->picture_structure == BOTTOM_FIELD);
picture->dmv_offset = bottom_field ? 1 : -1;
picture->f_motion.ref2[0] = picture->f_motion.ref[bottom_field];
picture->f_motion.ref2[1] = picture->f_motion.ref[!bottom_field];
picture->b_motion.ref2[0] = picture->b_motion.ref[bottom_field];
picture->b_motion.ref2[1] = picture->b_motion.ref[!bottom_field];
forward_ref = forward_reference_frame->base;
if (picture->second_field && (picture->picture_coding_type != B_TYPE))
forward_ref = picture->current_frame->base;
picture->f_motion.ref[1][0] = forward_ref[0] + (bottom_field ? 0 : picture->pitches[0]);
picture->f_motion.ref[1][1] = forward_ref[1] + (bottom_field ? 0 : picture->pitches[1]);
picture->f_motion.ref[1][2] = forward_ref[2] + (bottom_field ? 0 : picture->pitches[2]);
picture->b_motion.ref[1][0] =
backward_reference_frame->base[0] + (bottom_field ? 0 : picture->pitches[0]);
picture->b_motion.ref[1][1] =
backward_reference_frame->base[1] + (bottom_field ? 0 : picture->pitches[1]);
picture->b_motion.ref[1][2] =
backward_reference_frame->base[2] + (bottom_field ? 0 : picture->pitches[2]);
}
picture->f_motion.pmv[0][0] = picture->f_motion.pmv[0][1] = 0;
picture->f_motion.pmv[1][0] = picture->f_motion.pmv[1][1] = 0;
picture->b_motion.pmv[0][0] = picture->b_motion.pmv[0][1] = 0;
picture->b_motion.pmv[1][0] = picture->b_motion.pmv[1][1] = 0;
picture->v_offset = (code - 1) * 16;
offset = (code - 1);
if (picture->picture_structure != FRAME_PICTURE)
offset = 2 * offset;
picture->dest[0] = picture->current_frame->base[0] + picture->pitches[0] * offset * 16;
picture->dest[1] = picture->current_frame->base[1] + picture->pitches[1] * offset * 8;
picture->dest[2] = picture->current_frame->base[2] + picture->pitches[2] * offset * 8;
height = picture->coded_picture_height;
switch (picture->picture_structure) {
case BOTTOM_FIELD:
picture->dest[0] += picture->pitches[0];
picture->dest[1] += picture->pitches[1];
picture->dest[2] += picture->pitches[2];
/* follow thru */
case TOP_FIELD:
picture->pitches[0] <<= 1;
picture->pitches[1] <<= 1;
picture->pitches[2] <<= 1;
height >>= 1;
}
picture->limit_x = 2 * picture->coded_picture_width - 32;
picture->limit_y_16 = 2 * height - 32;
picture->limit_y_8 = 2 * height - 16;
picture->limit_y = height - 16;
picture->dc_dct_pred[0] = picture->dc_dct_pred[1] =
picture->dc_dct_pred[2] = 1 << (picture->intra_dc_precision + 7);
picture->quantizer_scale = get_quantizer_scale (picture);
/* ignore intra_slice and all the extra data */
while (bit_buf & 0x80000000) {
DUMPBITS (bit_buf, bits, 9);
NEEDBITS (bit_buf, bits, bit_ptr);
}
/* decode initial macroblock address increment */
offset = 0;
while (1) {
if (bit_buf >= 0x08000000) {
mba = MBA_5 + (UBITS (bit_buf, 6) - 2);
break;
} else if (bit_buf >= 0x01800000) {
mba = MBA_11 + (UBITS (bit_buf, 12) - 24);
break;
} else switch (UBITS (bit_buf, 12)) {
case 8: /* macroblock_escape */
offset += 33;
DUMPBITS (bit_buf, bits, 11);
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
case 15: /* macroblock_stuffing (MPEG1 only) */
bit_buf &= 0xfffff;
DUMPBITS (bit_buf, bits, 11);
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
default: /* error */
return 1;
}
}
DUMPBITS (bit_buf, bits, mba->len + 1);
picture->offset = (offset + mba->mba) << 4;
while (picture->offset - picture->coded_picture_width >= 0) {
picture->offset -= picture->coded_picture_width;
if ((picture->current_frame->copy == NULL) ||
(picture->picture_coding_type != B_TYPE)) {
picture->dest[0] += 16 * picture->pitches[0];
picture->dest[1] += 8 * picture->pitches[1];
picture->dest[2] += 8 * picture->pitches[2];
}
picture->v_offset += 16;
}
if (picture->v_offset > picture->limit_y)
return 1;
return 0;
#undef bit_buf
#undef bits
#undef bit_ptr
}
void mpeg2_slice (picture_t * picture, int code, uint8_t * buffer)
{
#define bit_buf (picture->bitstream_buf)
#define bits (picture->bitstream_bits)
#define bit_ptr (picture->bitstream_ptr)
cpu_state_t cpu_state;
bitstream_init (picture, buffer);
if (slice_init (picture, code))
return;
if (mpeg2_cpu_state_save)
mpeg2_cpu_state_save (&cpu_state);
while (1) {
int macroblock_modes;
int mba_inc;
MBAtab * mba;
NEEDBITS (bit_buf, bits, bit_ptr);
macroblock_modes = get_macroblock_modes (picture);
/* maybe integrate MACROBLOCK_QUANT test into get_macroblock_modes ? */
if (macroblock_modes & MACROBLOCK_QUANT)
picture->quantizer_scale = get_quantizer_scale (picture);
if (macroblock_modes & MACROBLOCK_INTRA) {
int DCT_offset, DCT_stride;
int offset;
uint8_t * dest_y;
if (picture->concealment_motion_vectors) {
if (picture->picture_structure == FRAME_PICTURE)
motion_fr_conceal (picture);
else
motion_fi_conceal (picture);
} else {
picture->f_motion.pmv[0][0] = picture->f_motion.pmv[0][1] = 0;
picture->f_motion.pmv[1][0] = picture->f_motion.pmv[1][1] = 0;
picture->b_motion.pmv[0][0] = picture->b_motion.pmv[0][1] = 0;
picture->b_motion.pmv[1][0] = picture->b_motion.pmv[1][1] = 0;
}
if (macroblock_modes & DCT_TYPE_INTERLACED) {
DCT_offset = picture->pitches[0];
DCT_stride = picture->pitches[0] * 2;
} else {
DCT_offset = picture->pitches[0] * 8;
DCT_stride = picture->pitches[0];
}
offset = picture->offset;
dest_y = picture->dest[0] + offset;
slice_intra_DCT (picture, 0, dest_y, DCT_stride);
slice_intra_DCT (picture, 0, dest_y + 8, DCT_stride);
slice_intra_DCT (picture, 0, dest_y + DCT_offset, DCT_stride);
slice_intra_DCT (picture, 0, dest_y + DCT_offset + 8, DCT_stride);
slice_intra_DCT (picture, 1, picture->dest[1] + (offset >> 1),
picture->pitches[1]);
slice_intra_DCT (picture, 2, picture->dest[2] + (offset >> 1),
picture->pitches[2]);
if (picture->picture_coding_type == D_TYPE) {
NEEDBITS (bit_buf, bits, bit_ptr);
DUMPBITS (bit_buf, bits, 1);
}
} else {
if (picture->picture_structure == FRAME_PICTURE)
switch (macroblock_modes & MOTION_TYPE_MASK) {
case MC_FRAME:
if (picture->mpeg1)
MOTION_CALL (motion_mp1, macroblock_modes);
else
MOTION_CALL (motion_fr_frame, macroblock_modes);
break;
case MC_FIELD:
MOTION_CALL (motion_fr_field, macroblock_modes);
break;
case MC_DMV:
MOTION_CALL (motion_fr_dmv, MACROBLOCK_MOTION_FORWARD);
break;
case 0:
/* non-intra mb without forward mv in a P picture */
picture->f_motion.pmv[0][0] = 0;
picture->f_motion.pmv[0][1] = 0;
picture->f_motion.pmv[1][0] = 0;
picture->f_motion.pmv[1][1] = 0;
MOTION_CALL (motion_zero, MACROBLOCK_MOTION_FORWARD);
break;
}
else
switch (macroblock_modes & MOTION_TYPE_MASK) {
case MC_FIELD:
MOTION_CALL (motion_fi_field, macroblock_modes);
break;
case MC_16X8:
MOTION_CALL (motion_fi_16x8, macroblock_modes);
break;
case MC_DMV:
MOTION_CALL (motion_fi_dmv, MACROBLOCK_MOTION_FORWARD);
break;
case 0:
/* non-intra mb without forward mv in a P picture */
picture->f_motion.pmv[0][0] = 0;
picture->f_motion.pmv[0][1] = 0;
picture->f_motion.pmv[1][0] = 0;
picture->f_motion.pmv[1][1] = 0;
MOTION_CALL (motion_zero, MACROBLOCK_MOTION_FORWARD);
break;
}
if (macroblock_modes & MACROBLOCK_PATTERN) {
int coded_block_pattern;
int DCT_offset, DCT_stride;
int offset;
uint8_t * dest_y;
if (macroblock_modes & DCT_TYPE_INTERLACED) {
DCT_offset = picture->pitches[0];
DCT_stride = picture->pitches[0] * 2;
} else {
DCT_offset = picture->pitches[0] * 8;
DCT_stride = picture->pitches[0];
}
coded_block_pattern = get_coded_block_pattern (picture);
offset = picture->offset;
dest_y = picture->dest[0] + offset;
if (coded_block_pattern & 0x20)
slice_non_intra_DCT (picture, dest_y, DCT_stride);
if (coded_block_pattern & 0x10)
slice_non_intra_DCT (picture, dest_y + 8, DCT_stride);
if (coded_block_pattern & 0x08)
slice_non_intra_DCT (picture, dest_y + DCT_offset,
DCT_stride);
if (coded_block_pattern & 0x04)
slice_non_intra_DCT (picture, dest_y + DCT_offset + 8,
DCT_stride);
if (coded_block_pattern & 0x2)
slice_non_intra_DCT (picture,
picture->dest[1] + (offset >> 1),
picture->pitches[1]);
if (coded_block_pattern & 0x1)
slice_non_intra_DCT (picture,
picture->dest[2] + (offset >> 1),
picture->pitches[2]);
}
picture->dc_dct_pred[0] = picture->dc_dct_pred[1] =
picture->dc_dct_pred[2] = 128 << picture->intra_dc_precision;
}
NEXT_MACROBLOCK;
NEEDBITS (bit_buf, bits, bit_ptr);
mba_inc = 0;
while (1) {
if (bit_buf >= 0x10000000) {
mba = MBA_5 + (UBITS (bit_buf, 5) - 2);
break;
} else if (bit_buf >= 0x03000000) {
mba = MBA_11 + (UBITS (bit_buf, 11) - 24);
break;
} else switch (UBITS (bit_buf, 11)) {
case 8: /* macroblock_escape */
mba_inc += 33;
/* pass through */
case 15: /* macroblock_stuffing (MPEG1 only) */
DUMPBITS (bit_buf, bits, 11);
NEEDBITS (bit_buf, bits, bit_ptr);
continue;
default: /* end of slice, or error */
if (mpeg2_cpu_state_restore)
mpeg2_cpu_state_restore (&cpu_state);
return;
}
}
DUMPBITS (bit_buf, bits, mba->len);
mba_inc += mba->mba;
if (mba_inc) {
picture->dc_dct_pred[0] = picture->dc_dct_pred[1] =
picture->dc_dct_pred[2] = 128 << picture->intra_dc_precision;
if (picture->picture_coding_type == P_TYPE) {
picture->f_motion.pmv[0][0] = picture->f_motion.pmv[0][1] = 0;
picture->f_motion.pmv[1][0] = picture->f_motion.pmv[1][1] = 0;
do {
MOTION_CALL (motion_zero, MACROBLOCK_MOTION_FORWARD);
NEXT_MACROBLOCK;
} while (--mba_inc);
} else {
do {
MOTION_CALL (motion_reuse, macroblock_modes);
NEXT_MACROBLOCK;
} while (--mba_inc);
}
}
}
#undef bit_buf
#undef bits
#undef bit_ptr
}