1/* 2 * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder 3 * Copyright (c) 2003-2010 Michael Niedermayer <michaelni@gmx.at> 4 * 5 * This file is part of FFmpeg. 6 * 7 * FFmpeg is free software; you can redistribute it and/or 8 * modify it under the terms of the GNU Lesser General Public 9 * License as published by the Free Software Foundation; either 10 * version 2.1 of the License, or (at your option) any later version. 11 * 12 * FFmpeg is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15 * Lesser General Public License for more details. 16 * 17 * You should have received a copy of the GNU Lesser General Public 18 * License along with FFmpeg; if not, write to the Free Software 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20 */ 21 22/** 23 * @file 24 * H.264 / AVC / MPEG-4 part10 DSP functions. 25 * @author Michael Niedermayer <michaelni@gmx.at> 26 */ 27 28#include <stdint.h> 29 30#include "libavutil/attributes.h" 31#include "libavutil/avassert.h" 32 33#include "h264dsp.h" 34#include "h264idct.h" 35#include "startcode.h" 36#include "libavutil/common.h" 37 38#define BIT_DEPTH 8 39#include "h264dsp_template.c" 40#undef BIT_DEPTH 41 42#define BIT_DEPTH 9 43#include "h264dsp_template.c" 44#undef BIT_DEPTH 45 46#define BIT_DEPTH 10 47#include "h264dsp_template.c" 48#undef BIT_DEPTH 49 50#define BIT_DEPTH 12 51#include "h264dsp_template.c" 52#undef BIT_DEPTH 53 54#define BIT_DEPTH 14 55#include "h264dsp_template.c" 56#undef BIT_DEPTH 57 58#define BIT_DEPTH 8 59#include "h264addpx_template.c" 60#undef BIT_DEPTH 61 62#define BIT_DEPTH 16 63#include "h264addpx_template.c" 64#undef BIT_DEPTH 65 66av_cold void ff_h264dsp_init(H264DSPContext *c, const int bit_depth, 67 const int chroma_format_idc) 68{ 69#undef FUNC 70#define FUNC(a, depth) a ## _ ## depth ## _c 71 72#define ADDPX_DSP(depth) \ 73 c->h264_add_pixels4_clear = FUNC(ff_h264_add_pixels4, depth);\ 74 c->h264_add_pixels8_clear = FUNC(ff_h264_add_pixels8, depth) 75 76 if (bit_depth > 8 && bit_depth <= 16) { 77 ADDPX_DSP(16); 78 } else { 79 ADDPX_DSP(8); 80 } 81 82#define H264_DSP(depth) \ 83 c->h264_idct_add= FUNC(ff_h264_idct_add, depth);\ 84 c->h264_idct8_add= FUNC(ff_h264_idct8_add, depth);\ 85 c->h264_idct_dc_add= FUNC(ff_h264_idct_dc_add, depth);\ 86 c->h264_idct8_dc_add= FUNC(ff_h264_idct8_dc_add, depth);\ 87 c->h264_idct_add16 = FUNC(ff_h264_idct_add16, depth);\ 88 c->h264_idct8_add4 = FUNC(ff_h264_idct8_add4, depth);\ 89 if (chroma_format_idc <= 1)\ 90 c->h264_idct_add8 = FUNC(ff_h264_idct_add8, depth);\ 91 else\ 92 c->h264_idct_add8 = FUNC(ff_h264_idct_add8_422, depth);\ 93 c->h264_idct_add16intra= FUNC(ff_h264_idct_add16intra, depth);\ 94 c->h264_luma_dc_dequant_idct= FUNC(ff_h264_luma_dc_dequant_idct, depth);\ 95 if (chroma_format_idc <= 1)\ 96 c->h264_chroma_dc_dequant_idct= FUNC(ff_h264_chroma_dc_dequant_idct, depth);\ 97 else\ 98 c->h264_chroma_dc_dequant_idct= FUNC(ff_h264_chroma422_dc_dequant_idct, depth);\ 99\ 100 c->weight_h264_pixels_tab[0]= FUNC(weight_h264_pixels16, depth);\ 101 c->weight_h264_pixels_tab[1]= FUNC(weight_h264_pixels8, depth);\ 102 c->weight_h264_pixels_tab[2]= FUNC(weight_h264_pixels4, depth);\ 103 c->weight_h264_pixels_tab[3]= FUNC(weight_h264_pixels2, depth);\ 104 c->biweight_h264_pixels_tab[0]= FUNC(biweight_h264_pixels16, depth);\ 105 c->biweight_h264_pixels_tab[1]= FUNC(biweight_h264_pixels8, depth);\ 106 c->biweight_h264_pixels_tab[2]= FUNC(biweight_h264_pixels4, depth);\ 107 c->biweight_h264_pixels_tab[3]= FUNC(biweight_h264_pixels2, depth);\ 108\ 109 c->h264_v_loop_filter_luma= FUNC(h264_v_loop_filter_luma, depth);\ 110 c->h264_h_loop_filter_luma= FUNC(h264_h_loop_filter_luma, depth);\ 111 c->h264_h_loop_filter_luma_mbaff= FUNC(h264_h_loop_filter_luma_mbaff, depth);\ 112 c->h264_v_loop_filter_luma_intra= FUNC(h264_v_loop_filter_luma_intra, depth);\ 113 c->h264_h_loop_filter_luma_intra= FUNC(h264_h_loop_filter_luma_intra, depth);\ 114 c->h264_h_loop_filter_luma_mbaff_intra= FUNC(h264_h_loop_filter_luma_mbaff_intra, depth);\ 115 c->h264_v_loop_filter_chroma= FUNC(h264_v_loop_filter_chroma, depth);\ 116 if (chroma_format_idc <= 1)\ 117 c->h264_h_loop_filter_chroma= FUNC(h264_h_loop_filter_chroma, depth);\ 118 else\ 119 c->h264_h_loop_filter_chroma= FUNC(h264_h_loop_filter_chroma422, depth);\ 120 if (chroma_format_idc <= 1)\ 121 c->h264_h_loop_filter_chroma_mbaff= FUNC(h264_h_loop_filter_chroma_mbaff, depth);\ 122 else\ 123 c->h264_h_loop_filter_chroma_mbaff= FUNC(h264_h_loop_filter_chroma422_mbaff, depth);\ 124 c->h264_v_loop_filter_chroma_intra= FUNC(h264_v_loop_filter_chroma_intra, depth);\ 125 if (chroma_format_idc <= 1)\ 126 c->h264_h_loop_filter_chroma_intra= FUNC(h264_h_loop_filter_chroma_intra, depth);\ 127 else\ 128 c->h264_h_loop_filter_chroma_intra= FUNC(h264_h_loop_filter_chroma422_intra, depth);\ 129 if (chroma_format_idc <= 1)\ 130 c->h264_h_loop_filter_chroma_mbaff_intra= FUNC(h264_h_loop_filter_chroma_mbaff_intra, depth);\ 131 else\ 132 c->h264_h_loop_filter_chroma_mbaff_intra= FUNC(h264_h_loop_filter_chroma422_mbaff_intra, depth);\ 133 c->h264_loop_filter_strength= NULL; 134 135 switch (bit_depth) { 136 case 9: 137 H264_DSP(9); 138 break; 139 case 10: 140 H264_DSP(10); 141 break; 142 case 12: 143 H264_DSP(12); 144 break; 145 case 14: 146 H264_DSP(14); 147 break; 148 default: 149 av_assert0(bit_depth<=8); 150 H264_DSP(8); 151 break; 152 } 153 c->startcode_find_candidate = ff_startcode_find_candidate_c; 154 155#if ARCH_AARCH64 156 ff_h264dsp_init_aarch64(c, bit_depth, chroma_format_idc); 157#elif ARCH_ARM 158 ff_h264dsp_init_arm(c, bit_depth, chroma_format_idc); 159#elif ARCH_PPC 160 ff_h264dsp_init_ppc(c, bit_depth, chroma_format_idc); 161#elif ARCH_X86 162 ff_h264dsp_init_x86(c, bit_depth, chroma_format_idc); 163#elif ARCH_MIPS 164 ff_h264dsp_init_mips(c, bit_depth, chroma_format_idc); 165#elif ARCH_LOONGARCH 166 ff_h264dsp_init_loongarch(c, bit_depth, chroma_format_idc); 167#endif 168} 169