/* * mpeg2_internal.h * Copyright (C) 2000-2003 Michel Lespinasse * Copyright (C) 1999-2000 Aaron Holtzman * * 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 */ #ifndef MPEG2_INTERNAL_H #define MPEG2_INTERNAL_H #include "config.h" /* for Rockbox CPU_ #defines */ /* macroblock modes */ #define MACROBLOCK_INTRA 1 #define MACROBLOCK_PATTERN 2 #define MACROBLOCK_MOTION_BACKWARD 4 #define MACROBLOCK_MOTION_FORWARD 8 #define MACROBLOCK_QUANT 16 #define DCT_TYPE_INTERLACED 32 /* motion_type */ #define MOTION_TYPE_SHIFT 6 #define MC_FIELD 1 #define MC_FRAME 2 #define MC_16X8 2 #define MC_DMV 3 /* picture structure */ #define TOP_FIELD 1 #define BOTTOM_FIELD 2 #define FRAME_PICTURE 3 /* picture coding type */ #define I_TYPE 1 #define P_TYPE 2 #define B_TYPE 3 #define D_TYPE 4 typedef void mpeg2_mc_fct (uint8_t *, const uint8_t *, int, int); typedef struct { uint8_t * ref[2][MPEG2_COMPONENTS]; uint8_t ** ref2[2]; int pmv[2][2]; int f_code[2]; } motion_t; typedef void motion_parser_t(mpeg2_decoder_t * decoder, motion_t * motion, mpeg2_mc_fct * const * table); struct mpeg2_decoder_s { /* first, state that carries information from one macroblock to the */ /* next inside a slice, and is never used outside of mpeg2_slice() */ /* bit parsing stuff */ uint32_t bitstream_buf; /* current 32 bit working set */ int bitstream_bits; /* used bits in working set */ const uint8_t * bitstream_ptr; /* buffer with stream data */ uint8_t * dest[MPEG2_COMPONENTS]; int offset; int stride; int uv_stride; int slice_stride; int slice_uv_stride; int stride_frame; unsigned int limit_x; unsigned int limit_y_16; unsigned int limit_y_8; unsigned int limit_y; /* Motion vectors */ /* The f_ and b_ correspond to the forward and backward motion */ /* predictors */ motion_t b_motion; motion_t f_motion; motion_parser_t * motion_parser[5]; /* predictor for DC coefficients in intra blocks */ int16_t dc_dct_pred[MPEG2_COMPONENTS]; /* DCT coefficients */ int16_t * ATTR_ALIGN(16) DCTblock; /* put buffer separately to have it in IRAM */ uint8_t * picture_dest[MPEG2_COMPONENTS]; void (* convert) (void * convert_id, uint8_t * const * src, unsigned int v_offset); void * convert_id; int dmv_offset; unsigned int v_offset; /* now non-slice-specific information */ /* sequence header stuff */ uint16_t * quantizer_matrix[4]; uint16_t (* chroma_quantizer[2])[64]; uint16_t quantizer_prescale[4][32][64]; /* The width and height of the picture snapped to macroblock units */ int width; int height; int vertical_position_extension; int chroma_format; /* picture header stuff */ /* what type of picture this is (I, P, B, D) */ int coding_type; /* picture coding extension stuff */ /* quantization factor for intra dc coefficients */ int intra_dc_precision; /* top/bottom/both fields */ int picture_structure; /* bool to indicate all predictions are frame based */ int frame_pred_frame_dct; /* bool to indicate whether intra blocks have motion vectors */ /* (for concealment) */ int concealment_motion_vectors; /* bool to use different vlc tables */ int intra_vlc_format; /* used for DMV MC */ int top_field_first; /* stuff derived from bitstream */ /* pointer to the zigzag scan we're supposed to be using */ const uint8_t * scan; int second_field; int mpeg1; }; typedef struct { mpeg2_fbuf_t fbuf; } fbuf_alloc_t; struct mpeg2dec_s { mpeg2_decoder_t decoder; mpeg2_info_t info; uint32_t shift; int is_display_initialized; mpeg2_state_t (* action) (struct mpeg2dec_s * mpeg2dec); mpeg2_state_t state; uint32_t ext_state; /* allocated in init - gcc has problems allocating such big structures */ uint8_t * ATTR_ALIGN(4) chunk_buffer; /* pointer to start of the current chunk */ uint8_t * chunk_start; /* pointer to current position in chunk_buffer */ uint8_t * chunk_ptr; /* last start code ? */ uint8_t code; /* picture tags */ uint32_t tag_current, tag2_current, tag_previous, tag2_previous; int num_tags; int bytes_since_tag; int first; int alloc_index_user; int alloc_index; uint8_t first_decode_slice; uint8_t nb_decode_slices; unsigned int user_data_len; mpeg2_sequence_t new_sequence; mpeg2_sequence_t sequence; mpeg2_gop_t new_gop; mpeg2_gop_t gop; mpeg2_picture_t new_picture; mpeg2_picture_t pictures[4]; mpeg2_picture_t * picture; /*const*/ mpeg2_fbuf_t * fbuf[3]; /* 0: current fbuf, 1-2: prediction fbufs */ fbuf_alloc_t fbuf_alloc[3]; int custom_fbuf; uint8_t * yuv_buf[3][MPEG2_COMPONENTS]; int yuv_index; mpeg2_convert_t * convert; void * convert_arg; unsigned int convert_id_size; int convert_stride; void (* convert_start) (void * id, const mpeg2_fbuf_t * fbuf, const mpeg2_picture_t * picture, const mpeg2_gop_t * gop); uint8_t * buf_start; uint8_t * buf_end; int16_t display_offset_x, display_offset_y; int copy_matrix; int8_t q_scale_type, scaled[4]; uint8_t quantizer_matrix[4][64]; uint8_t new_quantizer_matrix[4][64]; }; /* decode.c */ mpeg2_state_t mpeg2_seek_header (mpeg2dec_t * mpeg2dec); mpeg2_state_t mpeg2_parse_header (mpeg2dec_t * mpeg2dec); /* header.c */ void mpeg2_header_state_init (mpeg2dec_t * mpeg2dec); void mpeg2_reset_info (mpeg2_info_t * info); int mpeg2_header_sequence (mpeg2dec_t * mpeg2dec); int mpeg2_header_gop (mpeg2dec_t * mpeg2dec); mpeg2_state_t mpeg2_header_picture_start (mpeg2dec_t * mpeg2dec); int mpeg2_header_picture (mpeg2dec_t * mpeg2dec); int mpeg2_header_extension (mpeg2dec_t * mpeg2dec); int mpeg2_header_user_data (mpeg2dec_t * mpeg2dec); void mpeg2_header_sequence_finalize (mpeg2dec_t * mpeg2dec); void mpeg2_header_gop_finalize (mpeg2dec_t * mpeg2dec); void mpeg2_header_picture_finalize (mpeg2dec_t * mpeg2dec); mpeg2_state_t mpeg2_header_slice_start (mpeg2dec_t * mpeg2dec); mpeg2_state_t mpeg2_header_end (mpeg2dec_t * mpeg2dec); void mpeg2_set_fbuf (mpeg2dec_t * mpeg2dec, int b_type); /* idct.c */ void mpeg2_idct_init (void); void mpeg2_idct_copy(int16_t * block, uint8_t * dest, const int stride); void mpeg2_idct_add(const int last, int16_t * block, uint8_t * dest, const int stride); extern const uint8_t default_mpeg2_scan_norm[64]; extern const uint8_t default_mpeg2_scan_alt[64]; extern uint8_t mpeg2_scan_norm[64]; extern uint8_t mpeg2_scan_alt[64]; /* motion_comp.c */ void mpeg2_mc_init (void); typedef struct { mpeg2_mc_fct * put [8]; mpeg2_mc_fct * avg [8]; } mpeg2_mc_t; extern const mpeg2_mc_t mpeg2_mc; #endif /* MPEG2_INTERNAL_H */