134 lines
6.8 KiB
C
134 lines
6.8 KiB
C
/*
|
|
* 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 DSP functions.
|
|
* @author Michael Niedermayer <michaelni@gmx.at>
|
|
*/
|
|
|
|
#ifndef AVCODEC_H264DSP_H
|
|
#define AVCODEC_H264DSP_H
|
|
|
|
#include <stdint.h>
|
|
#include <stddef.h>
|
|
|
|
typedef void (*h264_weight_func)(uint8_t *block, ptrdiff_t stride, int height,
|
|
int log2_denom, int weight, int offset);
|
|
typedef void (*h264_biweight_func)(uint8_t *dst, uint8_t *src,
|
|
ptrdiff_t stride, int height, int log2_denom,
|
|
int weightd, int weights, int offset);
|
|
|
|
/**
|
|
* Context for storing H.264 DSP functions
|
|
*/
|
|
typedef struct H264DSPContext {
|
|
/* weighted MC */
|
|
h264_weight_func weight_h264_pixels_tab[4];
|
|
h264_biweight_func biweight_h264_pixels_tab[4];
|
|
|
|
/* loop filter */
|
|
void (*h264_v_loop_filter_luma)(uint8_t *pix /*align 16*/, ptrdiff_t stride,
|
|
int alpha, int beta, int8_t *tc0);
|
|
void (*h264_h_loop_filter_luma)(uint8_t *pix /*align 4 */, ptrdiff_t stride,
|
|
int alpha, int beta, int8_t *tc0);
|
|
void (*h264_h_loop_filter_luma_mbaff)(uint8_t *pix /*align 16*/, ptrdiff_t stride,
|
|
int alpha, int beta, int8_t *tc0);
|
|
/* v/h_loop_filter_luma_intra: align 16 */
|
|
void (*h264_v_loop_filter_luma_intra)(uint8_t *pix, ptrdiff_t stride,
|
|
int alpha, int beta);
|
|
void (*h264_h_loop_filter_luma_intra)(uint8_t *pix, ptrdiff_t stride,
|
|
int alpha, int beta);
|
|
void (*h264_h_loop_filter_luma_mbaff_intra)(uint8_t *pix /*align 16*/,
|
|
ptrdiff_t stride, int alpha, int beta);
|
|
void (*h264_v_loop_filter_chroma)(uint8_t *pix /*align 8*/, ptrdiff_t stride,
|
|
int alpha, int beta, int8_t *tc0);
|
|
void (*h264_h_loop_filter_chroma)(uint8_t *pix /*align 4*/, ptrdiff_t stride,
|
|
int alpha, int beta, int8_t *tc0);
|
|
void (*h264_h_loop_filter_chroma_mbaff)(uint8_t *pix /*align 8*/,
|
|
ptrdiff_t stride, int alpha, int beta,
|
|
int8_t *tc0);
|
|
void (*h264_v_loop_filter_chroma_intra)(uint8_t *pix /*align 8*/,
|
|
ptrdiff_t stride, int alpha, int beta);
|
|
void (*h264_h_loop_filter_chroma_intra)(uint8_t *pix /*align 8*/,
|
|
ptrdiff_t stride, int alpha, int beta);
|
|
void (*h264_h_loop_filter_chroma_mbaff_intra)(uint8_t *pix /*align 8*/,
|
|
ptrdiff_t stride, int alpha, int beta);
|
|
// h264_loop_filter_strength: simd only. the C version is inlined in h264_loopfilter.c
|
|
void (*h264_loop_filter_strength)(int16_t bS[2][4][4], uint8_t nnz[40],
|
|
int8_t ref[2][40], int16_t mv[2][40][2],
|
|
int bidir, int edges, int step,
|
|
int mask_mv0, int mask_mv1, int field);
|
|
|
|
/* IDCT */
|
|
void (*h264_idct_add)(uint8_t *dst /*align 4*/,
|
|
int16_t *block /*align 16*/, int stride);
|
|
void (*h264_idct8_add)(uint8_t *dst /*align 8*/,
|
|
int16_t *block /*align 16*/, int stride);
|
|
void (*h264_idct_dc_add)(uint8_t *dst /*align 4*/,
|
|
int16_t *block /*align 16*/, int stride);
|
|
void (*h264_idct8_dc_add)(uint8_t *dst /*align 8*/,
|
|
int16_t *block /*align 16*/, int stride);
|
|
|
|
void (*h264_idct_add16)(uint8_t *dst /*align 16*/, const int *blockoffset,
|
|
int16_t *block /*align 16*/, int stride,
|
|
const uint8_t nnzc[15 * 8]);
|
|
void (*h264_idct8_add4)(uint8_t *dst /*align 16*/, const int *blockoffset,
|
|
int16_t *block /*align 16*/, int stride,
|
|
const uint8_t nnzc[15 * 8]);
|
|
void (*h264_idct_add8)(uint8_t **dst /*align 16*/, const int *blockoffset,
|
|
int16_t *block /*align 16*/, int stride,
|
|
const uint8_t nnzc[15 * 8]);
|
|
void (*h264_idct_add16intra)(uint8_t *dst /*align 16*/, const int *blockoffset,
|
|
int16_t *block /*align 16*/,
|
|
int stride, const uint8_t nnzc[15 * 8]);
|
|
void (*h264_luma_dc_dequant_idct)(int16_t *output,
|
|
int16_t *input /*align 16*/, int qmul);
|
|
void (*h264_chroma_dc_dequant_idct)(int16_t *block, int qmul);
|
|
|
|
/* bypass-transform */
|
|
void (*h264_add_pixels8_clear)(uint8_t *dst, int16_t *block, int stride);
|
|
void (*h264_add_pixels4_clear)(uint8_t *dst, int16_t *block, int stride);
|
|
|
|
/**
|
|
* Search buf from the start for up to size bytes. Return the index
|
|
* of a zero byte, or >= size if not found. Ideally, use lookahead
|
|
* to filter out any zero bytes that are known to not be followed by
|
|
* one or more further zero bytes and a one byte. Better still, filter
|
|
* out any bytes that form the trailing_zero_8bits syntax element too.
|
|
*/
|
|
int (*startcode_find_candidate)(const uint8_t *buf, int size);
|
|
} H264DSPContext;
|
|
|
|
void ff_h264dsp_init(H264DSPContext *c, const int bit_depth,
|
|
const int chroma_format_idc);
|
|
void ff_h264dsp_init_aarch64(H264DSPContext *c, const int bit_depth,
|
|
const int chroma_format_idc);
|
|
void ff_h264dsp_init_arm(H264DSPContext *c, const int bit_depth,
|
|
const int chroma_format_idc);
|
|
void ff_h264dsp_init_ppc(H264DSPContext *c, const int bit_depth,
|
|
const int chroma_format_idc);
|
|
void ff_h264dsp_init_x86(H264DSPContext *c, const int bit_depth,
|
|
const int chroma_format_idc);
|
|
void ff_h264dsp_init_mips(H264DSPContext *c, const int bit_depth,
|
|
const int chroma_format_idc);
|
|
|
|
#endif /* AVCODEC_H264DSP_H */
|