mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
186 lines
6.5 KiB
186 lines
6.5 KiB
/* |
|
* H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder |
|
* Copyright (c) 2003-2010 Michael Niedermayer <michaelni@gmx.at> |
|
* |
|
* This file is part of FFmpeg. |
|
* |
|
* FFmpeg is free software; you can redistribute it and/or |
|
* modify it under the terms of the GNU Lesser General Public |
|
* License as published by the Free Software Foundation; either |
|
* version 2.1 of the License, or (at your option) any later version. |
|
* |
|
* FFmpeg is distributed in the hope that it will be useful, |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
* Lesser General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU Lesser General Public |
|
* License along with FFmpeg; if not, write to the Free Software |
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
*/ |
|
|
|
/** |
|
* @file |
|
* H.264 / AVC / MPEG4 part10 DSP functions. |
|
* @author Michael Niedermayer <michaelni@gmx.at> |
|
*/ |
|
|
|
#include <stdint.h> |
|
|
|
#include "libavutil/attributes.h" |
|
#include "libavutil/avassert.h" |
|
|
|
#include "avcodec.h" |
|
#include "h264dsp.h" |
|
#include "h264idct.h" |
|
#include "libavutil/common.h" |
|
|
|
#define BIT_DEPTH 8 |
|
#include "h264dsp_template.c" |
|
#undef BIT_DEPTH |
|
|
|
#define BIT_DEPTH 9 |
|
#include "h264dsp_template.c" |
|
#undef BIT_DEPTH |
|
|
|
#define BIT_DEPTH 10 |
|
#include "h264dsp_template.c" |
|
#undef BIT_DEPTH |
|
|
|
#define BIT_DEPTH 12 |
|
#include "h264dsp_template.c" |
|
#undef BIT_DEPTH |
|
|
|
#define BIT_DEPTH 14 |
|
#include "h264dsp_template.c" |
|
#undef BIT_DEPTH |
|
|
|
#define BIT_DEPTH 8 |
|
#include "h264addpx_template.c" |
|
#undef BIT_DEPTH |
|
|
|
#define BIT_DEPTH 16 |
|
#include "h264addpx_template.c" |
|
#undef BIT_DEPTH |
|
|
|
static int h264_find_start_code_candidate_c(const uint8_t *buf, int size) |
|
{ |
|
int i = 0; |
|
#if HAVE_FAST_UNALIGNED |
|
/* we check i < size instead of i + 3 / 7 because it is |
|
* simpler and there must be FF_INPUT_BUFFER_PADDING_SIZE |
|
* bytes at the end. |
|
*/ |
|
# if HAVE_FAST_64BIT |
|
while (i < size && |
|
!((~*(const uint64_t *)(buf + i) & |
|
(*(const uint64_t *)(buf + i) - 0x0101010101010101ULL)) & |
|
0x8080808080808080ULL)) |
|
i += 8; |
|
# else |
|
while (i < size && |
|
!((~*(const uint32_t *)(buf + i) & |
|
(*(const uint32_t *)(buf + i) - 0x01010101U)) & |
|
0x80808080U)) |
|
i += 4; |
|
# endif |
|
#endif |
|
for (; i < size; i++) |
|
if (!buf[i]) |
|
break; |
|
return i; |
|
} |
|
|
|
av_cold void ff_h264dsp_init(H264DSPContext *c, const int bit_depth, |
|
const int chroma_format_idc) |
|
{ |
|
#undef FUNC |
|
#define FUNC(a, depth) a ## _ ## depth ## _c |
|
|
|
#define ADDPX_DSP(depth) \ |
|
c->h264_add_pixels4_clear = FUNC(ff_h264_add_pixels4, depth);\ |
|
c->h264_add_pixels8_clear = FUNC(ff_h264_add_pixels8, depth) |
|
|
|
if (bit_depth > 8 && bit_depth <= 16) { |
|
ADDPX_DSP(16); |
|
} else { |
|
ADDPX_DSP(8); |
|
} |
|
|
|
#define H264_DSP(depth) \ |
|
c->h264_idct_add= FUNC(ff_h264_idct_add, depth);\ |
|
c->h264_idct8_add= FUNC(ff_h264_idct8_add, depth);\ |
|
c->h264_idct_dc_add= FUNC(ff_h264_idct_dc_add, depth);\ |
|
c->h264_idct8_dc_add= FUNC(ff_h264_idct8_dc_add, depth);\ |
|
c->h264_idct_add16 = FUNC(ff_h264_idct_add16, depth);\ |
|
c->h264_idct8_add4 = FUNC(ff_h264_idct8_add4, depth);\ |
|
if (chroma_format_idc == 1)\ |
|
c->h264_idct_add8 = FUNC(ff_h264_idct_add8, depth);\ |
|
else\ |
|
c->h264_idct_add8 = FUNC(ff_h264_idct_add8_422, depth);\ |
|
c->h264_idct_add16intra= FUNC(ff_h264_idct_add16intra, depth);\ |
|
c->h264_luma_dc_dequant_idct= FUNC(ff_h264_luma_dc_dequant_idct, depth);\ |
|
if (chroma_format_idc == 1)\ |
|
c->h264_chroma_dc_dequant_idct= FUNC(ff_h264_chroma_dc_dequant_idct, depth);\ |
|
else\ |
|
c->h264_chroma_dc_dequant_idct= FUNC(ff_h264_chroma422_dc_dequant_idct, depth);\ |
|
\ |
|
c->weight_h264_pixels_tab[0]= FUNC(weight_h264_pixels16, depth);\ |
|
c->weight_h264_pixels_tab[1]= FUNC(weight_h264_pixels8, depth);\ |
|
c->weight_h264_pixels_tab[2]= FUNC(weight_h264_pixels4, depth);\ |
|
c->weight_h264_pixels_tab[3]= FUNC(weight_h264_pixels2, depth);\ |
|
c->biweight_h264_pixels_tab[0]= FUNC(biweight_h264_pixels16, depth);\ |
|
c->biweight_h264_pixels_tab[1]= FUNC(biweight_h264_pixels8, depth);\ |
|
c->biweight_h264_pixels_tab[2]= FUNC(biweight_h264_pixels4, depth);\ |
|
c->biweight_h264_pixels_tab[3]= FUNC(biweight_h264_pixels2, depth);\ |
|
\ |
|
c->h264_v_loop_filter_luma= FUNC(h264_v_loop_filter_luma, depth);\ |
|
c->h264_h_loop_filter_luma= FUNC(h264_h_loop_filter_luma, depth);\ |
|
c->h264_h_loop_filter_luma_mbaff= FUNC(h264_h_loop_filter_luma_mbaff, depth);\ |
|
c->h264_v_loop_filter_luma_intra= FUNC(h264_v_loop_filter_luma_intra, depth);\ |
|
c->h264_h_loop_filter_luma_intra= FUNC(h264_h_loop_filter_luma_intra, depth);\ |
|
c->h264_h_loop_filter_luma_mbaff_intra= FUNC(h264_h_loop_filter_luma_mbaff_intra, depth);\ |
|
c->h264_v_loop_filter_chroma= FUNC(h264_v_loop_filter_chroma, depth);\ |
|
if (chroma_format_idc == 1)\ |
|
c->h264_h_loop_filter_chroma= FUNC(h264_h_loop_filter_chroma, depth);\ |
|
else\ |
|
c->h264_h_loop_filter_chroma= FUNC(h264_h_loop_filter_chroma422, depth);\ |
|
if (chroma_format_idc == 1)\ |
|
c->h264_h_loop_filter_chroma_mbaff= FUNC(h264_h_loop_filter_chroma_mbaff, depth);\ |
|
else\ |
|
c->h264_h_loop_filter_chroma_mbaff= FUNC(h264_h_loop_filter_chroma422_mbaff, depth);\ |
|
c->h264_v_loop_filter_chroma_intra= FUNC(h264_v_loop_filter_chroma_intra, depth);\ |
|
if (chroma_format_idc == 1)\ |
|
c->h264_h_loop_filter_chroma_intra= FUNC(h264_h_loop_filter_chroma_intra, depth);\ |
|
else\ |
|
c->h264_h_loop_filter_chroma_intra= FUNC(h264_h_loop_filter_chroma422_intra, depth);\ |
|
if (chroma_format_idc == 1)\ |
|
c->h264_h_loop_filter_chroma_mbaff_intra= FUNC(h264_h_loop_filter_chroma_mbaff_intra, depth);\ |
|
else\ |
|
c->h264_h_loop_filter_chroma_mbaff_intra= FUNC(h264_h_loop_filter_chroma422_mbaff_intra, depth);\ |
|
c->h264_loop_filter_strength= NULL; |
|
|
|
switch (bit_depth) { |
|
case 9: |
|
H264_DSP(9); |
|
break; |
|
case 10: |
|
H264_DSP(10); |
|
break; |
|
case 12: |
|
H264_DSP(12); |
|
break; |
|
case 14: |
|
H264_DSP(14); |
|
break; |
|
default: |
|
av_assert0(bit_depth<=8); |
|
H264_DSP(8); |
|
break; |
|
} |
|
c->h264_find_start_code_candidate = h264_find_start_code_candidate_c; |
|
|
|
if (ARCH_ARM) ff_h264dsp_init_arm(c, bit_depth, chroma_format_idc); |
|
if (ARCH_PPC) ff_h264dsp_init_ppc(c, bit_depth, chroma_format_idc); |
|
if (ARCH_X86) ff_h264dsp_init_x86(c, bit_depth, chroma_format_idc); |
|
}
|
|
|