1/**
2 * @file
3 * VP5 and VP6 compatible video decoder (common data)
4 *
5 * Copyright (C) 2006  Aurelien Jacobs <aurel@gnuage.org>
6 *
7 * This file is part of FFmpeg.
8 *
9 * FFmpeg is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU Lesser General Public
11 * License as published by the Free Software Foundation; either
12 * version 2.1 of the License, or (at your option) any later version.
13 *
14 * FFmpeg is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17 * Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public
20 * License along with FFmpeg; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 */
23
24#ifndef AVCODEC_VP56DATA_H
25#define AVCODEC_VP56DATA_H
26
27#include "libavutil/common.h"
28
29typedef enum {
30    VP56_FRAME_NONE     =-1,
31    VP56_FRAME_CURRENT  = 0,
32    VP56_FRAME_PREVIOUS = 1,
33    VP56_FRAME_GOLDEN   = 2,
34    VP56_FRAME_GOLDEN2  = 3,
35    VP56_FRAME_UNUSED   = 4,
36    VP56_FRAME_UNUSED2  = 5,
37} VP56Frame;
38
39typedef enum {
40    VP56_MB_INTER_NOVEC_PF = 0,  /**< Inter MB, no vector, from previous frame */
41    VP56_MB_INTRA          = 1,  /**< Intra MB */
42    VP56_MB_INTER_DELTA_PF = 2,  /**< Inter MB, above/left vector + delta, from previous frame */
43    VP56_MB_INTER_V1_PF    = 3,  /**< Inter MB, first vector, from previous frame */
44    VP56_MB_INTER_V2_PF    = 4,  /**< Inter MB, second vector, from previous frame */
45    VP56_MB_INTER_NOVEC_GF = 5,  /**< Inter MB, no vector, from golden frame */
46    VP56_MB_INTER_DELTA_GF = 6,  /**< Inter MB, above/left vector + delta, from golden frame */
47    VP56_MB_INTER_4V       = 7,  /**< Inter MB, 4 vectors, from previous frame */
48    VP56_MB_INTER_V1_GF    = 8,  /**< Inter MB, first vector, from golden frame */
49    VP56_MB_INTER_V2_GF    = 9,  /**< Inter MB, second vector, from golden frame */
50} VP56mb;
51
52typedef struct {
53  int8_t val;
54  int8_t prob_idx;
55} VP56Tree;
56
57extern const uint8_t vp56_b2p[];
58extern const uint8_t vp56_b6to4[];
59extern const uint8_t vp56_coeff_parse_table[6][11];
60extern const uint8_t vp56_def_mb_types_stats[3][10][2];
61extern const VP56Tree vp56_pva_tree[];
62extern const VP56Tree vp56_pc_tree[];
63extern const uint8_t vp56_coeff_bias[];
64extern const uint8_t vp56_coeff_bit_length[];
65
66static const VP56Frame vp56_reference_frame[] = {
67    VP56_FRAME_PREVIOUS,  /* VP56_MB_INTER_NOVEC_PF */
68    VP56_FRAME_CURRENT,   /* VP56_MB_INTRA */
69    VP56_FRAME_PREVIOUS,  /* VP56_MB_INTER_DELTA_PF */
70    VP56_FRAME_PREVIOUS,  /* VP56_MB_INTER_V1_PF */
71    VP56_FRAME_PREVIOUS,  /* VP56_MB_INTER_V2_PF */
72    VP56_FRAME_GOLDEN,    /* VP56_MB_INTER_NOVEC_GF */
73    VP56_FRAME_GOLDEN,    /* VP56_MB_INTER_DELTA_GF */
74    VP56_FRAME_PREVIOUS,  /* VP56_MB_INTER_4V */
75    VP56_FRAME_GOLDEN,    /* VP56_MB_INTER_V1_GF */
76    VP56_FRAME_GOLDEN,    /* VP56_MB_INTER_V2_GF */
77};
78
79static const uint8_t vp56_ac_dequant[64] = {
80    94, 92, 90, 88, 86, 82, 78, 74,
81    70, 66, 62, 58, 54, 53, 52, 51,
82    50, 49, 48, 47, 46, 45, 44, 43,
83    42, 40, 39, 37, 36, 35, 34, 33,
84    32, 31, 30, 29, 28, 27, 26, 25,
85    24, 23, 22, 21, 20, 19, 18, 17,
86    16, 15, 14, 13, 12, 11, 10,  9,
87     8,  7,  6,  5,  4,  3,  2,  1,
88};
89
90static const uint8_t vp56_dc_dequant[64] = {
91    47, 47, 47, 47, 45, 43, 43, 43,
92    43, 43, 42, 41, 41, 40, 40, 40,
93    40, 35, 35, 35, 35, 33, 33, 33,
94    33, 32, 32, 32, 27, 27, 26, 26,
95    25, 25, 24, 24, 23, 23, 19, 19,
96    19, 19, 18, 18, 17, 16, 16, 16,
97    16, 16, 15, 11, 11, 11, 10, 10,
98     9,  8,  7,  5,  3,  3,  2,  2,
99};
100
101static const uint8_t vp56_pre_def_mb_type_stats[16][3][10][2] = {
102  { { {   9, 15 }, {  32, 25 }, {  7,  19 }, {   9, 21 }, {  1, 12 },
103      {  14, 12 }, {   3, 18 }, { 14,  23 }, {   3, 10 }, {  0,  4 }, },
104    { {  41, 22 }, {   1,  0 }, {  1,  31 }, {   0,  0 }, {  0,  0 },
105      {   0,  1 }, {   1,  7 }, {  0,   1 }, {  98, 25 }, {  4, 10 }, },
106    { {   2,  3 }, {   2,  3 }, {  0,   2 }, {   0,  2 }, {  0,  0 },
107      {  11,  4 }, {   1,  4 }, {  0,   2 }, {   3,  2 }, {  0,  4 }, }, },
108  { { {  48, 39 }, {   1,  2 }, { 11,  27 }, {  29, 44 }, {  7, 27 },
109      {   1,  4 }, {   0,  3 }, {  1,   6 }, {   1,  2 }, {  0,  0 }, },
110    { { 123, 37 }, {   6,  4 }, {  1,  27 }, {   0,  0 }, {  0,  0 },
111      {   5,  8 }, {   1,  7 }, {  0,   1 }, {  12, 10 }, {  0,  2 }, },
112    { {  49, 46 }, {   3,  4 }, {  7,  31 }, {  42, 41 }, {  0,  0 },
113      {   2,  6 }, {   1,  7 }, {  1,   4 }, {   2,  4 }, {  0,  1 }, }, },
114  { { {  21, 32 }, {   1,  2 }, {  4,  10 }, {  32, 43 }, {  6, 23 },
115      {   2,  3 }, {   1, 19 }, {  1,   6 }, {  12, 21 }, {  0,  7 }, },
116    { {  26, 14 }, {  14, 12 }, {  0,  24 }, {   0,  0 }, {  0,  0 },
117      {  55, 17 }, {   1,  9 }, {  0,  36 }, {   5,  7 }, {  1,  3 }, },
118    { {  26, 25 }, {   1,  1 }, {  2,  10 }, {  67, 39 }, {  0,  0 },
119      {   1,  1 }, {   0, 14 }, {  0,   2 }, {  31, 26 }, {  1,  6 }, }, },
120  { { {  69, 83 }, {   0,  0 }, {  0,   2 }, {  10, 29 }, {  3, 12 },
121      {   0,  1 }, {   0,  3 }, {  0,   3 }, {   2,  2 }, {  0,  0 }, },
122    { { 209,  5 }, {   0,  0 }, {  0,  27 }, {   0,  0 }, {  0,  0 },
123      {   0,  1 }, {   0,  1 }, {  0,   1 }, {   0,  0 }, {  0,  0 }, },
124    { { 103, 46 }, {   1,  2 }, {  2,  10 }, {  33, 42 }, {  0,  0 },
125      {   1,  4 }, {   0,  3 }, {  0,   1 }, {   1,  3 }, {  0,  0 }, }, },
126  { { {  11, 20 }, {   1,  4 }, { 18,  36 }, {  43, 48 }, { 13, 35 },
127      {   0,  2 }, {   0,  5 }, {  3,  12 }, {   1,  2 }, {  0,  0 }, },
128    { {   2,  5 }, {   4,  5 }, {  0, 121 }, {   0,  0 }, {  0,  0 },
129      {   0,  3 }, {   2,  4 }, {  1,   4 }, {   2,  2 }, {  0,  1 }, },
130    { {  14, 31 }, {   9, 13 }, { 14,  54 }, {  22, 29 }, {  0,  0 },
131      {   2,  6 }, {   4, 18 }, {  6,  13 }, {   1,  5 }, {  0,  1 }, }, },
132  { { {  70, 44 }, {   0,  1 }, {  2,  10 }, {  37, 46 }, {  8, 26 },
133      {   0,  2 }, {   0,  2 }, {  0,   2 }, {   0,  1 }, {  0,  0 }, },
134    { { 175,  5 }, {   0,  1 }, {  0,  48 }, {   0,  0 }, {  0,  0 },
135      {   0,  2 }, {   0,  1 }, {  0,   2 }, {   0,  1 }, {  0,  0 }, },
136    { {  85, 39 }, {   0,  0 }, {  1,   9 }, {  69, 40 }, {  0,  0 },
137      {   0,  1 }, {   0,  3 }, {  0,   1 }, {   2,  3 }, {  0,  0 }, }, },
138  { { {   8, 15 }, {   0,  1 }, {  8,  21 }, {  74, 53 }, { 22, 42 },
139      {   0,  1 }, {   0,  2 }, {  0,   3 }, {   1,  2 }, {  0,  0 }, },
140    { {  83,  5 }, {   2,  3 }, {  0, 102 }, {   0,  0 }, {  0,  0 },
141      {   1,  3 }, {   0,  2 }, {  0,   1 }, {   0,  0 }, {  0,  0 }, },
142    { {  31, 28 }, {   0,  0 }, {  3,  14 }, { 130, 34 }, {  0,  0 },
143      {   0,  1 }, {   0,  3 }, {  0,   1 }, {   3,  3 }, {  0,  1 }, }, },
144  { { { 141, 42 }, {   0,  0 }, {  1,   4 }, {  11, 24 }, {  1, 11 },
145      {   0,  1 }, {   0,  1 }, {  0,   2 }, {   0,  0 }, {  0,  0 }, },
146    { { 233,  6 }, {   0,  0 }, {  0,   8 }, {   0,  0 }, {  0,  0 },
147      {   0,  1 }, {   0,  1 }, {  0,   0 }, {   0,  1 }, {  0,  0 }, },
148    { { 171, 25 }, {   0,  0 }, {  1,   5 }, {  25, 21 }, {  0,  0 },
149      {   0,  1 }, {   0,  1 }, {  0,   0 }, {   0,  0 }, {  0,  0 }, }, },
150  { { {   8, 19 }, {   4, 10 }, { 24,  45 }, {  21, 37 }, {  9, 29 },
151      {   0,  3 }, {   1,  7 }, { 11,  25 }, {   0,  2 }, {  0,  1 }, },
152    { {  34, 16 }, { 112, 21 }, {  1,  28 }, {   0,  0 }, {  0,  0 },
153      {   6,  8 }, {   1,  7 }, {  0,   3 }, {   2,  5 }, {  0,  2 }, },
154    { {  17, 21 }, {  68, 29 }, {  6,  15 }, {  13, 22 }, {  0,  0 },
155      {   6, 12 }, {   3, 14 }, {  4,  10 }, {   1,  7 }, {  0,  3 }, }, },
156  { { {  46, 42 }, {   0,  1 }, {  2,  10 }, {  54, 51 }, { 10, 30 },
157      {   0,  2 }, {   0,  2 }, {  0,   1 }, {   0,  1 }, {  0,  0 }, },
158    { { 159, 35 }, {   2,  2 }, {  0,  25 }, {   0,  0 }, {  0,  0 },
159      {   3,  6 }, {   0,  5 }, {  0,   1 }, {   4,  4 }, {  0,  1 }, },
160    { {  51, 39 }, {   0,  1 }, {  2,  12 }, {  91, 44 }, {  0,  0 },
161      {   0,  2 }, {   0,  3 }, {  0,   1 }, {   2,  3 }, {  0,  1 }, }, },
162  { { {  28, 32 }, {   0,  0 }, {  3,  10 }, {  75, 51 }, { 14, 33 },
163      {   0,  1 }, {   0,  2 }, {  0,   1 }, {   1,  2 }, {  0,  0 }, },
164    { {  75, 39 }, {   5,  7 }, {  2,  48 }, {   0,  0 }, {  0,  0 },
165      {   3, 11 }, {   2, 16 }, {  1,   4 }, {   7, 10 }, {  0,  2 }, },
166    { {  81, 25 }, {   0,  0 }, {  2,   9 }, { 106, 26 }, {  0,  0 },
167      {   0,  1 }, {   0,  1 }, {  0,   1 }, {   1,  1 }, {  0,  0 }, }, },
168  { { { 100, 46 }, {   0,  1 }, {  3,   9 }, {  21, 37 }, {  5, 20 },
169      {   0,  1 }, {   0,  2 }, {  1,   2 }, {   0,  1 }, {  0,  0 }, },
170    { { 212, 21 }, {   0,  1 }, {  0,   9 }, {   0,  0 }, {  0,  0 },
171      {   1,  2 }, {   0,  2 }, {  0,   0 }, {   2,  2 }, {  0,  0 }, },
172    { { 140, 37 }, {   0,  1 }, {  1,   8 }, {  24, 33 }, {  0,  0 },
173      {   1,  2 }, {   0,  2 }, {  0,   1 }, {   1,  2 }, {  0,  0 }, }, },
174  { { {  27, 29 }, {   0,  1 }, {  9,  25 }, {  53, 51 }, { 12, 34 },
175      {   0,  1 }, {   0,  3 }, {  1,   5 }, {   0,  2 }, {  0,  0 }, },
176    { {   4,  2 }, {   0,  0 }, {  0, 172 }, {   0,  0 }, {  0,  0 },
177      {   0,  1 }, {   0,  2 }, {  0,   0 }, {   2,  0 }, {  0,  0 }, },
178    { {  14, 23 }, {   1,  3 }, { 11,  53 }, {  90, 31 }, {  0,  0 },
179      {   0,  3 }, {   1,  5 }, {  2,   6 }, {   1,  2 }, {  0,  0 }, }, },
180  { { {  80, 38 }, {   0,  0 }, {  1,   4 }, {  69, 33 }, {  5, 16 },
181      {   0,  1 }, {   0,  1 }, {  0,   0 }, {   0,  1 }, {  0,  0 }, },
182    { { 187, 22 }, {   1,  1 }, {  0,  17 }, {   0,  0 }, {  0,  0 },
183      {   3,  6 }, {   0,  4 }, {  0,   1 }, {   4,  4 }, {  0,  1 }, },
184    { { 123, 29 }, {   0,  0 }, {  1,   7 }, {  57, 30 }, {  0,  0 },
185      {   0,  1 }, {   0,  1 }, {  0,   1 }, {   0,  1 }, {  0,  0 }, }, },
186  { { {  16, 20 }, {   0,  0 }, {  2,   8 }, { 104, 49 }, { 15, 33 },
187      {   0,  1 }, {   0,  1 }, {  0,   1 }, {   1,  1 }, {  0,  0 }, },
188    { { 133,  6 }, {   1,  2 }, {  1,  70 }, {   0,  0 }, {  0,  0 },
189      {   0,  2 }, {   0,  4 }, {  0,   3 }, {   1,  1 }, {  0,  0 }, },
190    { {  13, 14 }, {   0,  0 }, {  4,  20 }, { 175, 20 }, {  0,  0 },
191      {   0,  1 }, {   0,  1 }, {  0,   1 }, {   1,  1 }, {  0,  0 }, }, },
192  { { { 194, 16 }, {   0,  0 }, {  1,   1 }, {   1,  9 }, {  1,  3 },
193      {   0,  0 }, {   0,  1 }, {  0,   1 }, {   0,  0 }, {  0,  0 }, },
194    { { 251,  1 }, {   0,  0 }, {  0,   2 }, {   0,  0 }, {  0,  0 },
195      {   0,  0 }, {   0,  0 }, {  0,   0 }, {   0,  0 }, {  0,  0 }, },
196    { { 202, 23 }, {   0,  0 }, {  1,   3 }, {   2,  9 }, {  0,  0 },
197      {   0,  1 }, {   0,  1 }, {  0,   1 }, {   0,  0 }, {  0,  0 }, }, },
198};
199
200static const uint8_t vp56_filter_threshold[] = {
201    14, 14, 13, 13, 12, 12, 10, 10,
202    10, 10,  8,  8,  8,  8,  8,  8,
203     8,  8,  8,  8,  8,  8,  8,  8,
204     8,  8,  8,  8,  8,  8,  8,  8,
205     8,  8,  8,  8,  7,  7,  7,  7,
206     7,  7,  6,  6,  6,  6,  6,  6,
207     5,  5,  5,  5,  4,  4,  4,  4,
208     4,  4,  4,  3,  3,  3,  3,  2,
209};
210
211static const uint8_t vp56_mb_type_model_model[] = {
212    171, 83, 199, 140, 125, 104,
213};
214
215static const VP56Tree vp56_pmbtm_tree[] = {
216    { 4, 0},
217    { 2, 1}, {-8}, {-4},
218    { 8, 2},
219    { 6, 3},
220    { 4, 4},
221    { 2, 5}, {-24}, {-20}, {-16}, {-12}, {-0},
222};
223
224static const VP56Tree vp56_pmbt_tree[] = {
225    { 8, 1},
226    { 4, 2},
227    { 2, 4}, {-VP56_MB_INTER_NOVEC_PF}, {-VP56_MB_INTER_DELTA_PF},
228    { 2, 5}, {-VP56_MB_INTER_V1_PF},    {-VP56_MB_INTER_V2_PF},
229    { 4, 3},
230    { 2, 6}, {-VP56_MB_INTRA},          {-VP56_MB_INTER_4V},
231    { 4, 7},
232    { 2, 8}, {-VP56_MB_INTER_NOVEC_GF}, {-VP56_MB_INTER_DELTA_GF},
233    { 2, 9}, {-VP56_MB_INTER_V1_GF},    {-VP56_MB_INTER_V2_GF},
234};
235
236/* relative pos of surrounding blocks, from closest to farthest */
237static const int8_t vp56_candidate_predictor_pos[12][2] = {
238    {  0, -1 },
239    { -1,  0 },
240    { -1, -1 },
241    {  1, -1 },
242    {  0, -2 },
243    { -2,  0 },
244    { -2, -1 },
245    { -1, -2 },
246    {  1, -2 },
247    {  2, -1 },
248    { -2, -2 },
249    {  2, -2 },
250};
251
252#endif /* AVCODEC_VP56DATA_H */
253