Commit | Line | Data |
---|---|---|
2ba45a60 DM |
1 | /* |
2 | * HEVC video decoder | |
3 | * | |
4 | * Copyright (C) 2012 - 2013 Guillaume Martres | |
5 | * Copyright (C) 2013 - 2014 Pierre-Edouard Lepere | |
6 | * | |
7 | * | |
8 | * This file is part of FFmpeg. | |
9 | * | |
10 | * FFmpeg is free software; you can redistribute it and/or | |
11 | * modify it under the terms of the GNU Lesser General Public | |
12 | * License as published by the Free Software Foundation; either | |
13 | * version 2.1 of the License, or (at your option) any later version. | |
14 | * | |
15 | * FFmpeg is distributed in the hope that it will be useful, | |
16 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
17 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
18 | * Lesser General Public License for more details. | |
19 | * | |
20 | * You should have received a copy of the GNU Lesser General Public | |
21 | * License along with FFmpeg; if not, write to the Free Software | |
22 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
23 | */ | |
24 | ||
25 | #ifndef AVCODEC_HEVCDSP_H | |
26 | #define AVCODEC_HEVCDSP_H | |
27 | ||
28 | #include "get_bits.h" | |
29 | ||
30 | #define MAX_PB_SIZE 64 | |
31 | ||
32 | typedef struct SAOParams { | |
33 | int offset_abs[3][4]; ///< sao_offset_abs | |
34 | int offset_sign[3][4]; ///< sao_offset_sign | |
35 | ||
36 | uint8_t band_position[3]; ///< sao_band_position | |
37 | ||
38 | int eo_class[3]; ///< sao_eo_class | |
39 | ||
40 | int16_t offset_val[3][5]; ///<SaoOffsetVal | |
41 | ||
42 | uint8_t type_idx[3]; ///< sao_type_idx | |
43 | } SAOParams; | |
44 | ||
45 | typedef struct HEVCDSPContext { | |
46 | void (*put_pcm)(uint8_t *_dst, ptrdiff_t _stride, int width, int height, | |
47 | struct GetBitContext *gb, int pcm_bit_depth); | |
48 | ||
49 | void (*transform_add[4])(uint8_t *_dst, int16_t *coeffs, ptrdiff_t _stride); | |
50 | ||
51 | void (*transform_skip)(int16_t *coeffs, int16_t log2_size); | |
52 | ||
53 | void (*transform_rdpcm)(int16_t *coeffs, int16_t log2_size, int mode); | |
54 | ||
55 | void (*idct_4x4_luma)(int16_t *coeffs); | |
56 | ||
57 | void (*idct[4])(int16_t *coeffs, int col_limit); | |
58 | ||
59 | void (*idct_dc[4])(int16_t *coeffs); | |
60 | ||
61 | void (*sao_band_filter)(uint8_t *_dst, uint8_t *_src, ptrdiff_t _stride_dst, ptrdiff_t _stride_src, | |
62 | struct SAOParams *sao, int *borders, | |
63 | int width, int height, int c_idx); | |
64 | ||
65 | void (*sao_edge_filter[2])(uint8_t *_dst, uint8_t *_src, ptrdiff_t _stride_dst, ptrdiff_t _stride_src, | |
66 | struct SAOParams *sao, int *borders, int _width, | |
67 | int _height, int c_idx, uint8_t *vert_edge, | |
68 | uint8_t *horiz_edge, uint8_t *diag_edge); | |
69 | ||
70 | void (*put_hevc_qpel[10][2][2])(int16_t *dst, uint8_t *src, ptrdiff_t srcstride, | |
71 | int height, intptr_t mx, intptr_t my, int width); | |
72 | void (*put_hevc_qpel_uni[10][2][2])(uint8_t *dst, ptrdiff_t dststride, uint8_t *src, ptrdiff_t srcstride, | |
73 | int height, intptr_t mx, intptr_t my, int width); | |
74 | void (*put_hevc_qpel_uni_w[10][2][2])(uint8_t *_dst, ptrdiff_t _dststride, uint8_t *_src, ptrdiff_t _srcstride, | |
75 | int height, int denom, int wx, int ox, intptr_t mx, intptr_t my, int width); | |
76 | ||
77 | void (*put_hevc_qpel_bi[10][2][2])(uint8_t *dst, ptrdiff_t dststride, uint8_t *_src, ptrdiff_t _srcstride, | |
78 | int16_t *src2, | |
79 | int height, intptr_t mx, intptr_t my, int width); | |
80 | void (*put_hevc_qpel_bi_w[10][2][2])(uint8_t *dst, ptrdiff_t dststride, uint8_t *_src, ptrdiff_t _srcstride, | |
81 | int16_t *src2, | |
82 | int height, int denom, int wx0, int wx1, | |
83 | int ox0, int ox1, intptr_t mx, intptr_t my, int width); | |
84 | void (*put_hevc_epel[10][2][2])(int16_t *dst, uint8_t *src, ptrdiff_t srcstride, | |
85 | int height, intptr_t mx, intptr_t my, int width); | |
86 | ||
87 | void (*put_hevc_epel_uni[10][2][2])(uint8_t *dst, ptrdiff_t dststride, uint8_t *_src, ptrdiff_t _srcstride, | |
88 | int height, intptr_t mx, intptr_t my, int width); | |
89 | void (*put_hevc_epel_uni_w[10][2][2])(uint8_t *_dst, ptrdiff_t _dststride, uint8_t *_src, ptrdiff_t _srcstride, | |
90 | int height, int denom, int wx, int ox, intptr_t mx, intptr_t my, int width); | |
91 | void (*put_hevc_epel_bi[10][2][2])(uint8_t *dst, ptrdiff_t dststride, uint8_t *_src, ptrdiff_t _srcstride, | |
92 | int16_t *src2, | |
93 | int height, intptr_t mx, intptr_t my, int width); | |
94 | void (*put_hevc_epel_bi_w[10][2][2])(uint8_t *dst, ptrdiff_t dststride, uint8_t *_src, ptrdiff_t _srcstride, | |
95 | int16_t *src2, | |
96 | int height, int denom, int wx0, int ox0, int wx1, | |
97 | int ox1, intptr_t mx, intptr_t my, int width); | |
98 | ||
99 | void (*hevc_h_loop_filter_luma)(uint8_t *pix, ptrdiff_t stride, | |
100 | int beta, int32_t *tc, | |
101 | uint8_t *no_p, uint8_t *no_q); | |
102 | void (*hevc_v_loop_filter_luma)(uint8_t *pix, ptrdiff_t stride, | |
103 | int beta, int32_t *tc, | |
104 | uint8_t *no_p, uint8_t *no_q); | |
105 | void (*hevc_h_loop_filter_chroma)(uint8_t *pix, ptrdiff_t stride, | |
106 | int32_t *tc, uint8_t *no_p, uint8_t *no_q); | |
107 | void (*hevc_v_loop_filter_chroma)(uint8_t *pix, ptrdiff_t stride, | |
108 | int32_t *tc, uint8_t *no_p, uint8_t *no_q); | |
109 | void (*hevc_h_loop_filter_luma_c)(uint8_t *pix, ptrdiff_t stride, | |
110 | int beta, int32_t *tc, | |
111 | uint8_t *no_p, uint8_t *no_q); | |
112 | void (*hevc_v_loop_filter_luma_c)(uint8_t *pix, ptrdiff_t stride, | |
113 | int beta, int32_t *tc, | |
114 | uint8_t *no_p, uint8_t *no_q); | |
115 | void (*hevc_h_loop_filter_chroma_c)(uint8_t *pix, ptrdiff_t stride, | |
116 | int32_t *tc, uint8_t *no_p, | |
117 | uint8_t *no_q); | |
118 | void (*hevc_v_loop_filter_chroma_c)(uint8_t *pix, ptrdiff_t stride, | |
119 | int32_t *tc, uint8_t *no_p, | |
120 | uint8_t *no_q); | |
121 | } HEVCDSPContext; | |
122 | ||
123 | void ff_hevc_dsp_init(HEVCDSPContext *hpc, int bit_depth); | |
124 | ||
125 | extern const int8_t ff_hevc_epel_filters[7][4]; | |
126 | extern const int8_t ff_hevc_qpel_filters[3][16]; | |
127 | ||
128 | void ff_hevc_dsp_init_x86(HEVCDSPContext *c, const int bit_depth); | |
129 | ||
130 | #endif /* AVCODEC_HEVCDSP_H */ |