]> git.sesse.net Git - ffmpeg/blob - libavcodec/hevcpred.c
avcodec: Fix reference data type for nvdec vc1 hwaccel
[ffmpeg] / libavcodec / hevcpred.c
1 /*
2  * HEVC video Decoder
3  *
4  * Copyright (C) 2012 - 2013 Guillaume Martres
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #include "hevcdec.h"
24
25 #include "hevcpred.h"
26
27 #define BIT_DEPTH 8
28 #include "hevcpred_template.c"
29 #undef BIT_DEPTH
30
31 #define BIT_DEPTH 9
32 #include "hevcpred_template.c"
33 #undef BIT_DEPTH
34
35 #define BIT_DEPTH 10
36 #include "hevcpred_template.c"
37 #undef BIT_DEPTH
38
39 #define BIT_DEPTH 12
40 #include "hevcpred_template.c"
41 #undef BIT_DEPTH
42
43 void ff_hevc_pred_init(HEVCPredContext *hpc, int bit_depth)
44 {
45 #undef FUNC
46 #define FUNC(a, depth) a ## _ ## depth
47
48 #define HEVC_PRED(depth)                                \
49     hpc->intra_pred[0]   = FUNC(intra_pred_2, depth);   \
50     hpc->intra_pred[1]   = FUNC(intra_pred_3, depth);   \
51     hpc->intra_pred[2]   = FUNC(intra_pred_4, depth);   \
52     hpc->intra_pred[3]   = FUNC(intra_pred_5, depth);   \
53     hpc->pred_planar[0]  = FUNC(pred_planar_0, depth);  \
54     hpc->pred_planar[1]  = FUNC(pred_planar_1, depth);  \
55     hpc->pred_planar[2]  = FUNC(pred_planar_2, depth);  \
56     hpc->pred_planar[3]  = FUNC(pred_planar_3, depth);  \
57     hpc->pred_dc         = FUNC(pred_dc, depth);        \
58     hpc->pred_angular[0] = FUNC(pred_angular_0, depth); \
59     hpc->pred_angular[1] = FUNC(pred_angular_1, depth); \
60     hpc->pred_angular[2] = FUNC(pred_angular_2, depth); \
61     hpc->pred_angular[3] = FUNC(pred_angular_3, depth);
62
63     switch (bit_depth) {
64     case 9:
65         HEVC_PRED(9);
66         break;
67     case 10:
68         HEVC_PRED(10);
69         break;
70     case 12:
71         HEVC_PRED(12);
72         break;
73     default:
74         HEVC_PRED(8);
75         break;
76     }
77
78     if (ARCH_MIPS)
79         ff_hevc_pred_init_mips(hpc, bit_depth);
80 }