1/* 2 * Copyright (c) 2003-2010 Michael Niedermayer <michaelni@gmx.at> 3 * 4 * This file is part of FFmpeg. 5 * 6 * FFmpeg is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU Lesser General Public 8 * License as published by the Free Software Foundation; either 9 * version 2.1 of the License, or (at your option) any later version. 10 * 11 * FFmpeg is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with FFmpeg; if not, write to the Free Software 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 19 */ 20 21/** 22 * @file 23 * H.264 DSP functions. 24 * @author Michael Niedermayer <michaelni@gmx.at> 25 */ 26 27#ifndef AVCODEC_H264DSP_H 28#define AVCODEC_H264DSP_H 29 30#include <stdint.h> 31#include "dsputil.h" 32 33//typedef void (*h264_chroma_mc_func)(uint8_t *dst/*align 8*/, uint8_t *src/*align 1*/, int srcStride, int h, int x, int y); 34typedef void (*h264_weight_func)(uint8_t *block, int stride, int log2_denom, int weight, int offset); 35typedef void (*h264_biweight_func)(uint8_t *dst, uint8_t *src, int stride, int log2_denom, int weightd, int weights, int offset); 36 37/** 38 * Context for storing H.264 DSP functions 39 */ 40typedef struct H264DSPContext{ 41 /* weighted MC */ 42 h264_weight_func weight_h264_pixels_tab[10]; 43 h264_biweight_func biweight_h264_pixels_tab[10]; 44 45 /* loop filter */ 46 void (*h264_v_loop_filter_luma)(uint8_t *pix/*align 16*/, int stride, int alpha, int beta, int8_t *tc0); 47 void (*h264_h_loop_filter_luma)(uint8_t *pix/*align 4 */, int stride, int alpha, int beta, int8_t *tc0); 48 /* v/h_loop_filter_luma_intra: align 16 */ 49 void (*h264_v_loop_filter_luma_intra)(uint8_t *pix, int stride, int alpha, int beta); 50 void (*h264_h_loop_filter_luma_intra)(uint8_t *pix, int stride, int alpha, int beta); 51 void (*h264_v_loop_filter_chroma)(uint8_t *pix/*align 8*/, int stride, int alpha, int beta, int8_t *tc0); 52 void (*h264_h_loop_filter_chroma)(uint8_t *pix/*align 4*/, int stride, int alpha, int beta, int8_t *tc0); 53 void (*h264_v_loop_filter_chroma_intra)(uint8_t *pix/*align 8*/, int stride, int alpha, int beta); 54 void (*h264_h_loop_filter_chroma_intra)(uint8_t *pix/*align 8*/, int stride, int alpha, int beta); 55 // h264_loop_filter_strength: simd only. the C version is inlined in h264.c 56 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], 57 int bidir, int edges, int step, int mask_mv0, int mask_mv1, int field); 58 59 /* IDCT */ 60 /* NOTE!!! if you implement any of h264_idct8_add, h264_idct8_add4 then you must implement all of them 61 NOTE!!! if you implement any of h264_idct_add, h264_idct_add16, h264_idct_add16intra, h264_idct_add8 then you must implement all of them 62 The reason for above, is that no 2 out of one list may use a different permutation. 63 */ 64 void (*h264_idct_add)(uint8_t *dst/*align 4*/, DCTELEM *block/*align 16*/, int stride); 65 void (*h264_idct8_add)(uint8_t *dst/*align 8*/, DCTELEM *block/*align 16*/, int stride); 66 void (*h264_idct_dc_add)(uint8_t *dst/*align 4*/, DCTELEM *block/*align 16*/, int stride); 67 void (*h264_idct8_dc_add)(uint8_t *dst/*align 8*/, DCTELEM *block/*align 16*/, int stride); 68 void (*h264_dct)(DCTELEM block[4][4]); 69 void (*h264_idct_add16)(uint8_t *dst/*align 16*/, const int *blockoffset, DCTELEM *block/*align 16*/, int stride, const uint8_t nnzc[6*8]); 70 void (*h264_idct8_add4)(uint8_t *dst/*align 16*/, const int *blockoffset, DCTELEM *block/*align 16*/, int stride, const uint8_t nnzc[6*8]); 71 void (*h264_idct_add8)(uint8_t **dst/*align 16*/, const int *blockoffset, DCTELEM *block/*align 16*/, int stride, const uint8_t nnzc[6*8]); 72 void (*h264_idct_add16intra)(uint8_t *dst/*align 16*/, const int *blockoffset, DCTELEM *block/*align 16*/, int stride, const uint8_t nnzc[6*8]); 73}H264DSPContext; 74 75void ff_h264dsp_init(H264DSPContext *c); 76void ff_h264dsp_init_arm(H264DSPContext *c); 77void ff_h264dsp_init_ppc(H264DSPContext *c); 78void ff_h264dsp_init_x86(H264DSPContext *c); 79 80#endif /* AVCODEC_H264DSP_H */ 81