winamp/Src/h264dec/dec_api.c

393 lines
11 KiB
C

#include "dec_api.h"
#include "global.h"
#include "nalu.h"
#include "image.h"
#include "meminput.h"
#include "output.h"
#include "fmo.h"
#include "erc_api.h"
#include "parset.h"
#include "memcache.h"
#include "block.h"
#include "optim.h"
#include "mc_prediction.h"
#include "vlc.h"
#include <stddef.h> // for offsetof
#define BUILD_BUG_ON(condition) ((void)sizeof(char[1 - 2*!!(condition)]))
OptimizedFunctions opt;
DecoderParams *alloc_decoder();
void Configure(VideoParameters *p_Vid, InputParameters *p_Inp);
void malloc_slice(InputParameters *p_Inp, VideoParameters *p_Vid);
void init (VideoParameters *p_Vid);
void free_slice (Slice *currSlice);
void free_img( VideoParameters *p_Vid);
int sse2_flag = 0, mmx_flag=0, sse_flag=0, sse3_flag=0, sse4_1_flag=0;
int H264_Init()
{
int flags_edx, flags_ecx;
#ifdef H264_IPP
ippStaticInit();
#endif
#ifdef _M_IX86
_asm {
mov eax, 1
cpuid
mov flags_edx, edx
mov flags_ecx, ecx
}
mmx_flag = flags_edx & 0x00800000;
sse_flag = flags_edx & 0x02000000;
sse2_flag = flags_edx & 0x04000000;
sse3_flag = flags_ecx & 0x00000001;
sse4_1_flag= flags_ecx & (1 << 19);
#elif defined(_M_X64)
sse2_flag = 1;
#endif
#ifdef _M_IX86
/* if you get any compile errors here, you need to change biari.asm */
BUILD_BUG_ON(offsetof(TextureInfoContexts, map_contexts) != 436);
BUILD_BUG_ON(offsetof(TextureInfoContexts, last_contexts) != 3252);
BUILD_BUG_ON(offsetof(TextureInfoContexts, one_contexts) != 6068);
BUILD_BUG_ON(offsetof(TextureInfoContexts, abs_contexts) != 6508);
BUILD_BUG_ON(offsetof(Macroblock, p_Slice) != 0);
BUILD_BUG_ON(offsetof(Macroblock, p_Vid) != 4);
BUILD_BUG_ON(offsetof(Macroblock, qp) != 60);
BUILD_BUG_ON(offsetof(Macroblock, qpc) != 64);
BUILD_BUG_ON(offsetof(Macroblock, qp_scaled) != 72);
BUILD_BUG_ON(offsetof(Macroblock, cbp_blk) != 248);
BUILD_BUG_ON(offsetof(Macroblock, mb_field) != 344);
BUILD_BUG_ON(offsetof(Macroblock, read_and_store_CBP_block_bit) != 400);
BUILD_BUG_ON(offsetof(Slice, tex_ctx) != 100);
BUILD_BUG_ON(offsetof(Slice, mb_rec) != 1696);
BUILD_BUG_ON(offsetof(Slice, mb_pred) != 928);
BUILD_BUG_ON(offsetof(Slice, coeff) != 15632);
BUILD_BUG_ON(offsetof(Slice, coeff_ctr) != 15760);
BUILD_BUG_ON(offsetof(Slice, pos) != 15764);
BUILD_BUG_ON(offsetof(Slice, cof) != 2464);
BUILD_BUG_ON(offsetof(Slice, last_dquant) != 88);
BUILD_BUG_ON(offsetof(Slice, mot_ctx) != 96);
BUILD_BUG_ON(offsetof(Slice, slice_type) != 64);
BUILD_BUG_ON(offsetof(StorablePicture, structure) != 0);
BUILD_BUG_ON(offsetof(StorablePicture, chroma_qp_offset) != 158688);
BUILD_BUG_ON(offsetof(StorablePicture, motion) != 158524);
BUILD_BUG_ON(offsetof(StorablePicture, plane_images) != 158512);
BUILD_BUG_ON(offsetof(StorablePicture, imgY) != 158512);
BUILD_BUG_ON(offsetof(VideoParameters, structure) != 697200);
BUILD_BUG_ON(offsetof(VideoParameters, bitdepth_chroma_qp_scale) != 697456);
BUILD_BUG_ON(offsetof(VideoParameters, dec_picture) != 698192);
BUILD_BUG_ON(offsetof(DecodingEnvironment, Dcodestrm_len) != 16);
BUILD_BUG_ON(offsetof(DecodingEnvironment, Dcodestrm) != 12);
BUILD_BUG_ON(offsetof(DecodingEnvironment, DbitsLeft) != 8);
BUILD_BUG_ON(offsetof(DecodingEnvironment, Dvalue) != 4);
BUILD_BUG_ON(offsetof(DecodingEnvironment, Drange) != 0);
BUILD_BUG_ON(sizeof(BiContextType) != 4);
BUILD_BUG_ON(offsetof(BiContextType, state) != 0);
BUILD_BUG_ON(offsetof(BiContextType, MPS) != 2);
BUILD_BUG_ON(offsetof(OptimizedFunctions, copy_image_data_16x16_stride) != 32);
#endif
if (sse2_flag)
{
//opt.itrans4x4 = itrans4x4_mmx;
opt.itrans8x8 = itrans8x8_sse2;
opt.weighted_mc_prediction16x16 = weighted_mc_prediction16x16_sse2;
opt.weighted_mc_prediction16x8 = weighted_mc_prediction16x8_sse2;
opt.weighted_mc_prediction8x8 = weighted_mc_prediction8x8_sse2;
opt.weighted_bi_prediction16x16 = weighted_bi_prediction16x16_sse2;
opt.weighted_bi_prediction16x8 = weighted_bi_prediction16x8_sse2;
opt.weighted_bi_prediction8x8 = weighted_bi_prediction8x8_sse2;
opt.bi_prediction8x8 = bi_prediction8x8_sse2;
opt.copy_image_data_16x16_stride = copy_image_data_16x16_stride_sse;
opt.code_from_bitstream_2d_5_4 = code_from_bitstream_2d_5_4_sse2;
opt.code_from_bitstream_2d_17_4 = code_from_bitstream_2d_17_4_sse2;
opt.code_from_bitstream_2d_16_1 = code_from_bitstream_2d_16_1_sse2;
}
else if (sse_flag && mmx_flag)
{
//opt.itrans4x4 = itrans4x4_mmx;
opt.itrans8x8 = itrans8x8_c;//itrans8x8_mmx;
opt.weighted_mc_prediction16x16 = weighted_mc_prediction16x16_ipp;
opt.weighted_mc_prediction16x8 = weighted_mc_prediction16x8_ipp;
opt.weighted_mc_prediction8x8 = weighted_mc_prediction8x8_ipp;
opt.weighted_bi_prediction16x16 = weighted_bi_prediction16x16_ipp;
opt.weighted_bi_prediction16x8 = weighted_bi_prediction16x8_ipp;
opt.weighted_bi_prediction8x8 = weighted_bi_prediction8x8_ipp;
opt.bi_prediction8x8 = bi_prediction8x8_ipp;
opt.copy_image_data_16x16_stride = copy_image_data_16x16_stride_sse;
opt.code_from_bitstream_2d_5_4 = code_from_bitstream_2d_5_4_c;
opt.code_from_bitstream_2d_17_4 = code_from_bitstream_2d_17_4_c;
opt.code_from_bitstream_2d_16_1 = code_from_bitstream_2d_16_1_c;
}
else
return 0;
return 1;
}
h264_decoder_t H264_CreateDecoder()
{
DecoderParams *decoder=alloc_decoder();
if (decoder)
{
InputParameters *p_Inp = decoder->p_Inp;
Configure(decoder->p_Vid, p_Inp);
p_Inp->intra_profile_deblocking = 1;
initBitsFile(decoder->p_Vid);
malloc_slice(decoder->p_Inp, decoder->p_Vid);
init_old_slice(decoder->p_Vid->old_slice);
init(decoder->p_Vid);
init_out_buffer(decoder->p_Vid);
decoder->p_Vid->current_mb_nr = -4711; // initialized to an impossible value for debugging -- correct value is taken from slice header
}
return decoder;
}
void H264_DestroyDecoder(h264_decoder_t d)
{
DecoderParams *decoder = (DecoderParams *)d;
if (decoder)
{
free_slice(decoder->p_Vid->currentSlice);
FmoFinit(decoder->p_Vid);
free_global_buffers(decoder->p_Vid);
flush_dpb(decoder->p_Vid);
#if (PAIR_FIELDS_IN_OUTPUT)
flush_pending_output(decoder->p_Vid);
#endif
out_storable_pictures_destroy(decoder->p_Vid);
ercClose(decoder->p_Vid, decoder->p_Vid->erc_errorVar);
CleanUpPPS(decoder->p_Vid);
free_dpb(decoder->p_Vid);
uninit_out_buffer(decoder->p_Vid);
image_cache_flush(&decoder->p_Vid->image_cache[0]);
image_cache_flush(&decoder->p_Vid->image_cache[1]);
motion_cache_flush(&decoder->p_Vid->motion_cache);
FreeNALU(decoder->p_Vid->nalu);
free (decoder->p_Inp);
free_img (decoder->p_Vid);
free(decoder);
}
}
void H264_DecodeFrame(h264_decoder_t d, const void *buffer, size_t bufferlen, uint64_t time_code)
{
DecoderParams *decoder = (DecoderParams *)d;
int ret;
memory_input_t *mem_input = decoder->p_Vid->mem_input;
mem_input->user_buffer=buffer;
mem_input->user_buffer_size=bufferlen;
mem_input->user_buffer_read=0;
__try
{
ret = decode_one_frame(decoder->p_Vid, time_code);
}
__except(EXCEPTION_EXECUTE_HANDLER)
{
int x;
x=0;
}
#ifdef _M_IX86
_mm_empty();
#endif
}
void H264_GetPicture(h264_decoder_t d, StorablePicture **pic)
{
DecoderParams *decoder = (DecoderParams *)d;
if (pic)
{
out_storable_picture_get(decoder->p_Vid, pic);
}
}
static double GetAspectRatio(const vui_seq_parameters_t *vui)
{
int aspect_ratio_width=1, aspect_ratio_height=1;
if (vui->aspect_ratio_info_present_flag)
{
switch(vui->aspect_ratio_idc)
{
case VUI_AR_UNDEFINED:
case VUI_AR_SQUARE:
aspect_ratio_width = 1;
aspect_ratio_height = 1;
break;
case VUI_AR_12_11:
aspect_ratio_width = 12;
aspect_ratio_height = 11;
break;
case VUI_AR_10_11:
aspect_ratio_width = 10;
aspect_ratio_height = 11;
break;
case VUI_AR_16_11:
aspect_ratio_width = 16;
aspect_ratio_height = 11;
break;
case VUI_AR_40_33:
aspect_ratio_width = 40;
aspect_ratio_height = 33;
break;
case VUI_AR_24_11:
aspect_ratio_width = 24;
aspect_ratio_height = 11;
break;
case VUI_AR_20_11:
aspect_ratio_width = 20;
aspect_ratio_height = 11;
break;
case VUI_AR_32_11:
aspect_ratio_width = 32;
aspect_ratio_height = 11;
break;
case VUI_AR_80_33:
aspect_ratio_width = 80;
aspect_ratio_height = 33;
break;
case VUI_AR_18_11:
aspect_ratio_width = 18;
aspect_ratio_height = 11;
break;
case VUI_AR_15_11:
aspect_ratio_width = 15;
aspect_ratio_height = 11;
break;
case VUI_AR_64_33:
aspect_ratio_width = 64;
aspect_ratio_height = 33;
break;
case VUI_AR_160_99:
aspect_ratio_width = 160;
aspect_ratio_height = 99;
break;
case VUI_AR_4_3:
aspect_ratio_width = 4;
aspect_ratio_height = 3;
break;
case VUI_AR_3_2:
aspect_ratio_width = 3;
aspect_ratio_height = 2;
break;;
case VUI_AR_2_1:
aspect_ratio_width = 2;
aspect_ratio_height = 1;
break;;
case VUI_EXTENDED_SAR:
default:
aspect_ratio_width = vui->sar_width;
aspect_ratio_height = vui->sar_height;
break;
}
}
return (double)aspect_ratio_width / (double)aspect_ratio_height;
}
const FrameFormat *H264_GetOutputFormat(h264_decoder_t d, double *aspect_ratio)
{
DecoderParams *decoder = (DecoderParams *)d;
if (decoder && decoder->p_Inp)
{
if (decoder->p_Vid->active_sps)
*aspect_ratio = GetAspectRatio(&decoder->p_Vid->active_sps->vui_seq_parameters);
return &decoder->p_Inp->output;
}
else
return 0;
}
void H264_Flush(h264_decoder_t d)
{
DecoderParams *decoder = (DecoderParams *)d;
if (decoder && decoder->p_Vid)
{
StorablePicture *pic=0;
exit_picture(decoder->p_Vid, &decoder->p_Vid->dec_picture);
if (pic)
free_storable_picture(decoder->p_Vid, pic);
pic=0;
decoder->p_Vid->frame_num = 0;
decoder->p_Vid->pre_frame_num = INT_MIN;
decoder->p_Vid->PreviousFrameNum=0;
decoder->p_Vid->PreviousFrameNumOffset = 0;
decoder->p_Vid->PrevPicOrderCntLsb = 0;
decoder->p_Vid->PrevPicOrderCntMsb = 0;
flush_dpb(decoder->p_Vid);
do
{
pic=0;
out_storable_picture_get(decoder->p_Vid, &pic);
if (pic)
free_storable_picture(decoder->p_Vid, pic);
} while (pic);
decoder->p_Vid->mem_input->resetting = 1;
}
}
void H264_FreePicture(h264_decoder_t d, StorablePicture *p)
{
DecoderParams *decoder = (DecoderParams *)d;
if (decoder && decoder->p_Vid && p)
{
free_storable_picture(decoder->p_Vid, p);
}
}
void H264_EndOfStream(h264_decoder_t d)
{
DecoderParams *decoder = (DecoderParams *)d;
if (decoder && decoder->p_Vid)
{
if (decoder->p_Vid->dec_picture)
exit_picture(decoder->p_Vid, &decoder->p_Vid->dec_picture);
else
flush_dpb(decoder->p_Vid);
}
}
void H264_HurryUp(h264_decoder_t d, int state)
{
DecoderParams *decoder = (DecoderParams *)d;
if (decoder && decoder->p_Vid)
{
memory_input_t *mem_input = decoder->p_Vid->mem_input;
if (mem_input)
mem_input->skip_b_frames = state;
}
}