/*

 * 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 / MPEG-4 part10 DSP functions.

 * @author Michael Niedermayer <michaelni@gmx.at>

 */

#include <stdint.h>

#include "libavutil/attributes.h"

#include "libavutil/avassert.h"

#include "h264dsp.h"

#include "h264idct.h"

#include "startcode.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

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 SET_PIXSIZE_FUNCS(depth) \

    c->luma_dc_dequant_idct = FUNC(ff_h264_luma_dc_dequant_idct, depth);\

    if (chroma_format_idc <= 1)\

        c->chroma_dc_dequant_idct = FUNC(ff_h264_chroma_dc_dequant_idct, depth);\

    else\

        c->chroma_dc_dequant_idct = FUNC(ff_h264_chroma422_dc_dequant_idct, depth);\

    c->add_pixels4_clear = FUNC(ff_h264_add_pixels4, depth);\

    c->add_pixels8_clear = FUNC(ff_h264_add_pixels8, depth)

    if (bit_depth > 8 && bit_depth <= 16) {

        SET_PIXSIZE_FUNCS(16);

    } else {

        SET_PIXSIZE_FUNCS(8);

    }

#define H264_DSP(depth) \

    c->idct_add     = FUNC(ff_h264_idct_add, depth);\

    c->idct8_add    = FUNC(ff_h264_idct8_add, depth);\

    c->idct_dc_add  = FUNC(ff_h264_idct_dc_add, depth);\

    c->idct8_dc_add = FUNC(ff_h264_idct8_dc_add, depth);\

    c->idct_add16   = FUNC(ff_h264_idct_add16, depth);\

    c->idct8_add4   = FUNC(ff_h264_idct8_add4, depth);\

    if (chroma_format_idc <= 1)\

        c->idct_add8  = FUNC(ff_h264_idct_add8, depth);\

    else\

        c->idct_add8  = FUNC(ff_h264_idct_add8_422, depth);\

    c->idct_add16intra = FUNC(ff_h264_idct_add16intra, depth);\

\

    c->weight_pixels_tab[0]   = FUNC(weight_h264_pixels16, depth);\

    c->weight_pixels_tab[1]   = FUNC(weight_h264_pixels8, depth);\

    c->weight_pixels_tab[2]   = FUNC(weight_h264_pixels4, depth);\

    c->weight_pixels_tab[3]   = FUNC(weight_h264_pixels2, depth);\

    c->biweight_pixels_tab[0] = FUNC(biweight_h264_pixels16, depth);\

    c->biweight_pixels_tab[1] = FUNC(biweight_h264_pixels8, depth);\

    c->biweight_pixels_tab[2] = FUNC(biweight_h264_pixels4, depth);\

    c->biweight_pixels_tab[3] = FUNC(biweight_h264_pixels2, depth);\

\

    c->v_loop_filter_luma       = FUNC(h264_v_loop_filter_luma, depth);\

    c->h_loop_filter_luma       = FUNC(h264_h_loop_filter_luma, depth);\

    c->h_loop_filter_luma_mbaff = FUNC(h264_h_loop_filter_luma_mbaff, depth);\

    c->v_loop_filter_luma_intra = FUNC(h264_v_loop_filter_luma_intra, depth);\

    c->h_loop_filter_luma_intra = FUNC(h264_h_loop_filter_luma_intra, depth);\

    c->h_loop_filter_luma_mbaff_intra = FUNC(h264_h_loop_filter_luma_mbaff_intra, depth);\

    c->v_loop_filter_chroma = FUNC(h264_v_loop_filter_chroma, depth);\

    if (chroma_format_idc <= 1)\

        c->h_loop_filter_chroma = FUNC(h264_h_loop_filter_chroma, depth);\

    else\

        c->h_loop_filter_chroma = FUNC(h264_h_loop_filter_chroma422, depth);\

    if (chroma_format_idc <= 1)\

        c->h_loop_filter_chroma_mbaff = FUNC(h264_h_loop_filter_chroma_mbaff, depth);\

    else\

        c->h_loop_filter_chroma_mbaff = FUNC(h264_h_loop_filter_chroma422_mbaff, depth);\

    c->v_loop_filter_chroma_intra= FUNC(h264_v_loop_filter_chroma_intra, depth);\

    if (chroma_format_idc <= 1)\

        c->h_loop_filter_chroma_intra = FUNC(h264_h_loop_filter_chroma_intra, depth);\

    else\

        c->h_loop_filter_chroma_intra = FUNC(h264_h_loop_filter_chroma422_intra, depth);\

    if (chroma_format_idc <= 1)\

        c->h_loop_filter_chroma_mbaff_intra = FUNC(h264_h_loop_filter_chroma_mbaff_intra, depth);\

    else\

        c->h_loop_filter_chroma_mbaff_intra = FUNC(h264_h_loop_filter_chroma422_mbaff_intra, depth);\

    c->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->startcode_find_candidate = ff_startcode_find_candidate_c;

#if ARCH_AARCH64

    ff_h264dsp_init_aarch64(c, bit_depth, chroma_format_idc);

#elif ARCH_ARM

    ff_h264dsp_init_arm(c, bit_depth, chroma_format_idc);

#elif ARCH_PPC

    ff_h264dsp_init_ppc(c, bit_depth, chroma_format_idc);

#elif ARCH_RISCV

    ff_h264dsp_init_riscv(c, bit_depth, chroma_format_idc);

#elif ARCH_X86 && HAVE_X86ASM

    ff_h264dsp_init_x86(c, bit_depth, chroma_format_idc);

#elif ARCH_MIPS

    ff_h264dsp_init_mips(c, bit_depth, chroma_format_idc);

#elif ARCH_LOONGARCH

    ff_h264dsp_init_loongarch(c, bit_depth, chroma_format_idc);

#endif

}