FFmpeg: libavcodec/h264dsp.h Source File

00001 /*
00002  * Copyright (c) 2003-2010 Michael Niedermayer <michaelni@gmx.at>
00003  *
00004  * This file is part of FFmpeg.
00005  *
00006  * FFmpeg is free software; you can redistribute it and/or
00007  * modify it under the terms of the GNU Lesser General Public
00008  * License as published by the Free Software Foundation; either
00009  * version 2.1 of the License, or (at your option) any later version.
00010  *
00011  * FFmpeg is distributed in the hope that it will be useful,
00012  * but WITHOUT ANY WARRANTY; without even the implied warranty of
00013  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00014  * Lesser General Public License for more details.
00015  *
00016  * You should have received a copy of the GNU Lesser General Public
00017  * License along with FFmpeg; if not, write to the Free Software
00018  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
00019  */
00020 
00027 #ifndef AVCODEC_H264DSP_H
00028 #define AVCODEC_H264DSP_H
00029 
00030 #include <stdint.h>
00031 
00032 #include "dsputil.h"
00033 
00034 typedef void (*h264_weight_func)(uint8_t *block, int stride, int height,
00035                                  int log2_denom, int weight, int offset);
00036 typedef void (*h264_biweight_func)(uint8_t *dst, uint8_t *src,
00037                                    int stride, int height, int log2_denom,
00038                                    int weightd, int weights, int offset);
00039 
00043 typedef struct H264DSPContext {
00044     /* weighted MC */
00045     h264_weight_func weight_h264_pixels_tab[4];
00046     h264_biweight_func biweight_h264_pixels_tab[4];
00047 
00048     /* loop filter */
00049     void (*h264_v_loop_filter_luma)(uint8_t *pix /*align 16*/, int stride,
00050                                     int alpha, int beta, int8_t *tc0);
00051     void (*h264_h_loop_filter_luma)(uint8_t *pix /*align 4 */, int stride,
00052                                     int alpha, int beta, int8_t *tc0);
00053     void (*h264_h_loop_filter_luma_mbaff)(uint8_t *pix /*align 16*/, int stride,
00054                                           int alpha, int beta, int8_t *tc0);
00055     /* v/h_loop_filter_luma_intra: align 16 */
00056     void (*h264_v_loop_filter_luma_intra)(uint8_t *pix, int stride,
00057                                           int alpha, int beta);
00058     void (*h264_h_loop_filter_luma_intra)(uint8_t *pix, int stride,
00059                                           int alpha, int beta);
00060     void (*h264_h_loop_filter_luma_mbaff_intra)(uint8_t *pix /*align 16*/,
00061                                                 int stride, int alpha, int beta);
00062     void (*h264_v_loop_filter_chroma)(uint8_t *pix /*align 8*/, int stride,
00063                                       int alpha, int beta, int8_t *tc0);
00064     void (*h264_h_loop_filter_chroma)(uint8_t *pix /*align 4*/, int stride,
00065                                       int alpha, int beta, int8_t *tc0);
00066     void (*h264_h_loop_filter_chroma_mbaff)(uint8_t *pix /*align 8*/,
00067                                             int stride, int alpha, int beta,
00068                                             int8_t *tc0);
00069     void (*h264_v_loop_filter_chroma_intra)(uint8_t *pix /*align 8*/,
00070                                             int stride, int alpha, int beta);
00071     void (*h264_h_loop_filter_chroma_intra)(uint8_t *pix /*align 8*/,
00072                                             int stride, int alpha, int beta);
00073     void (*h264_h_loop_filter_chroma_mbaff_intra)(uint8_t *pix /*align 8*/,
00074                                                   int stride, int alpha, int beta);
00075     // h264_loop_filter_strength: simd only. the C version is inlined in h264.c
00076     void (*h264_loop_filter_strength)(int16_t bS[2][4][4], uint8_t nnz[40],
00077                                       int8_t ref[2][40], int16_t mv[2][40][2],
00078                                       int bidir, int edges, int step,
00079                                       int mask_mv0, int mask_mv1, int field);
00080 
00081     /* IDCT */
00082     void (*h264_idct_add)(uint8_t *dst /*align 4*/,
00083                           DCTELEM *block /*align 16*/, int stride);
00084     void (*h264_idct8_add)(uint8_t *dst /*align 8*/,
00085                            DCTELEM *block /*align 16*/, int stride);
00086     void (*h264_idct_dc_add)(uint8_t *dst /*align 4*/,
00087                              DCTELEM *block /*align 16*/, int stride);
00088     void (*h264_idct8_dc_add)(uint8_t *dst /*align 8*/,
00089                               DCTELEM *block /*align 16*/, int stride);
00090 
00091     void (*h264_idct_add16)(uint8_t *dst /*align 16*/, const int *blockoffset,
00092                             DCTELEM *block /*align 16*/, int stride,
00093                             const uint8_t nnzc[15 * 8]);
00094     void (*h264_idct8_add4)(uint8_t *dst /*align 16*/, const int *blockoffset,
00095                             DCTELEM *block /*align 16*/, int stride,
00096                             const uint8_t nnzc[15 * 8]);
00097     void (*h264_idct_add8)(uint8_t **dst /*align 16*/, const int *blockoffset,
00098                            DCTELEM *block /*align 16*/, int stride,
00099                            const uint8_t nnzc[15 * 8]);
00100     void (*h264_idct_add16intra)(uint8_t *dst /*align 16*/, const int *blockoffset,
00101                                  DCTELEM *block /*align 16*/,
00102                                  int stride, const uint8_t nnzc[15 * 8]);
00103     void (*h264_luma_dc_dequant_idct)(DCTELEM *output,
00104                                       DCTELEM *input /*align 16*/, int qmul);
00105     void (*h264_chroma_dc_dequant_idct)(DCTELEM *block, int qmul);
00106 } H264DSPContext;
00107 
00108 void ff_h264dsp_init(H264DSPContext *c, const int bit_depth,
00109                      const int chroma_format_idc);
00110 void ff_h264dsp_init_arm(H264DSPContext *c, const int bit_depth,
00111                          const int chroma_format_idc);
00112 void ff_h264dsp_init_ppc(H264DSPContext *c, const int bit_depth,
00113                          const int chroma_format_idc);
00114 void ff_h264dsp_init_x86(H264DSPContext *c, const int bit_depth,
00115                          const int chroma_format_idc);
00116 
00117 #endif /* AVCODEC_H264DSP_H */