rv34: Intra 16x16 handling
[ffmpeg.git] / libavcodec / rv34dsp.h
1 /*
2  * RV30/40 decoder motion compensation functions
3  * Copyright (c) 2008 Konstantin Shishkov
4  *
5  * This file is part of Libav.
6  *
7  * Libav 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  * Libav 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 Libav; 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  * RV30/40 decoder motion compensation functions
25  */
26
27 #ifndef AVCODEC_RV34DSP_H
28 #define AVCODEC_RV34DSP_H
29
30 #include "dsputil.h"
31
32 typedef void (*rv40_weight_func)(uint8_t *dst/*align width (8 or 16)*/,
33                                  uint8_t *src1/*align width (8 or 16)*/,
34                                  uint8_t *src2/*align width (8 or 16)*/,
35                                  int w1, int w2, int stride);
36
37 typedef void (*rv34_inv_transform_func)(DCTELEM *block);
38
39 typedef void (*rv34_idct_add_func)(uint8_t *dst, int stride,
40                                    const DCTELEM *block);
41 typedef void (*rv34_idct_dc_add_func)(uint8_t *dst, int stride,
42                                       int   dc);
43
44 typedef void (*rv40_weak_loop_filter_func)(uint8_t *src, int stride,
45                                            int filter_p1, int filter_q1,
46                                            int alpha, int beta,
47                                            int lims, int lim_q1, int lim_p1);
48
49 typedef void (*rv40_strong_loop_filter_func)(uint8_t *src, int stride,
50                                              int alpha, int lims,
51                                              int dmode, int chroma);
52
53 typedef int (*rv40_loop_filter_strength_func)(uint8_t *src, int stride,
54                                               int beta, int beta2, int edge,
55                                               int *p1, int *q1);
56
57 typedef struct RV34DSPContext {
58     qpel_mc_func put_pixels_tab[4][16];
59     qpel_mc_func avg_pixels_tab[4][16];
60     h264_chroma_mc_func put_chroma_pixels_tab[3];
61     h264_chroma_mc_func avg_chroma_pixels_tab[3];
62     rv40_weight_func rv40_weight_pixels_tab[2];
63     rv34_inv_transform_func rv34_inv_transform_tab[2];
64     void (*rv34_inv_transform_dc_tab[2])(DCTELEM *block);
65     rv34_idct_add_func rv34_idct_add;
66     rv34_idct_dc_add_func rv34_idct_dc_add;
67     rv40_weak_loop_filter_func rv40_weak_loop_filter[2];
68     rv40_strong_loop_filter_func rv40_strong_loop_filter[2];
69     rv40_loop_filter_strength_func rv40_loop_filter_strength[2];
70 } RV34DSPContext;
71
72 void ff_rv30dsp_init(RV34DSPContext *c, DSPContext* dsp);
73 void ff_rv34dsp_init(RV34DSPContext *c, DSPContext* dsp);
74 void ff_rv40dsp_init(RV34DSPContext *c, DSPContext* dsp);
75
76 void ff_rv34dsp_init_neon(RV34DSPContext *c, DSPContext *dsp);
77 void ff_rv34dsp_init_x86(RV34DSPContext *c, DSPContext *dsp);
78
79 void ff_rv40dsp_init_x86(RV34DSPContext *c, DSPContext *dsp);
80 void ff_rv40dsp_init_neon(RV34DSPContext *c, DSPContext *dsp);
81
82 #endif /* AVCODEC_RV34DSP_H */