enums.h 7.61 KB
Newer Older
Jingning Han's avatar
Jingning Han committed
1
2
3
4
5
6
7
8
9
10
/*
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

11
12
#ifndef VP10_COMMON_ENUMS_H_
#define VP10_COMMON_ENUMS_H_
Jingning Han's avatar
Jingning Han committed
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

#include "./vpx_config.h"
#include "vpx/vpx_integer.h"

#ifdef __cplusplus
extern "C" {
#endif

#define MI_SIZE_LOG2 3
#define MI_BLOCK_SIZE_LOG2 (6 - MI_SIZE_LOG2)  // 64 = 2^6

#define MI_SIZE (1 << MI_SIZE_LOG2)  // pixels per mi-unit
#define MI_BLOCK_SIZE (1 << MI_BLOCK_SIZE_LOG2)  // mi-units per max block

#define MI_MASK (MI_BLOCK_SIZE - 1)

// Bitstream profiles indicated by 2-3 bits in the uncompressed header.
// 00: Profile 0.  8-bit 4:2:0 only.
// 10: Profile 1.  8-bit 4:4:4, 4:2:2, and 4:4:0.
// 01: Profile 2.  10-bit and 12-bit color only, with 4:2:0 sampling.
// 110: Profile 3. 10-bit and 12-bit color only, with 4:2:2/4:4:4/4:4:0
//                 sampling.
// 111: Undefined profile.
typedef enum BITSTREAM_PROFILE {
  PROFILE_0,
  PROFILE_1,
  PROFILE_2,
  PROFILE_3,
  MAX_PROFILES
} BITSTREAM_PROFILE;

44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#define BLOCK_4X4      0
#define BLOCK_4X8      1
#define BLOCK_8X4      2
#define BLOCK_8X8      3
#define BLOCK_8X16     4
#define BLOCK_16X8     5
#define BLOCK_16X16    6
#define BLOCK_16X32    7
#define BLOCK_32X16    8
#define BLOCK_32X32    9
#define BLOCK_32X64   10
#define BLOCK_64X32   11
#define BLOCK_64X64   12

#if CONFIG_EXT_PARTITION
#define BLOCK_64X128  13
#define BLOCK_128X64  14
#define BLOCK_128X128 15
#define BLOCK_SIZES   16
#else
#define BLOCK_SIZES   13
#endif  // CONFIG_EXT_PARTITION

#define BLOCK_INVALID (BLOCK_SIZES)
#define BLOCK_LARGEST (BLOCK_SIZES - 1)

Jingning Han's avatar
Jingning Han committed
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
typedef uint8_t BLOCK_SIZE;

typedef enum PARTITION_TYPE {
  PARTITION_NONE,
  PARTITION_HORZ,
  PARTITION_VERT,
  PARTITION_SPLIT,
  PARTITION_TYPES,
  PARTITION_INVALID = PARTITION_TYPES
} PARTITION_TYPE;

typedef char PARTITION_CONTEXT;
#define PARTITION_PLOFFSET   4  // number of probability models per block size
#define PARTITION_CONTEXTS (4 * PARTITION_PLOFFSET)

// block transform size
typedef uint8_t TX_SIZE;
#define TX_4X4   ((TX_SIZE)0)   // 4x4 transform
#define TX_8X8   ((TX_SIZE)1)   // 8x8 transform
#define TX_16X16 ((TX_SIZE)2)   // 16x16 transform
#define TX_32X32 ((TX_SIZE)3)   // 32x32 transform
#define TX_SIZES ((TX_SIZE)4)

// frame transform mode
typedef enum {
  ONLY_4X4            = 0,        // only 4x4 transform used
  ALLOW_8X8           = 1,        // allow block transform size up to 8x8
  ALLOW_16X16         = 2,        // allow block transform size up to 16x16
  ALLOW_32X32         = 3,        // allow block transform size up to 32x32
  TX_MODE_SELECT      = 4,        // transform specified for each block
  TX_MODES            = 5,
} TX_MODE;

typedef enum {
  DCT_DCT   = 0,                      // DCT  in both horizontal and vertical
  ADST_DCT  = 1,                      // ADST in vertical, DCT in horizontal
  DCT_ADST  = 2,                      // DCT  in vertical, ADST in horizontal
  ADST_ADST = 3,                      // ADST in both directions
108
109
110
111
112
113
#if CONFIG_EXT_TX
  FLIPADST_DCT = 4,
  DCT_FLIPADST = 5,
  FLIPADST_FLIPADST = 6,
  ADST_FLIPADST = 7,
  FLIPADST_ADST = 8,
114
115
116
117
118
119
120
  DST_DCT = 9,
  DCT_DST = 10,
  DST_ADST = 11,
  ADST_DST = 12,
  DST_FLIPADST = 13,
  FLIPADST_DST = 14,
  DST_DST = 15,
121
  IDTX = 16,
Jingning Han's avatar
Jingning Han committed
122
123
  V_DCT = 17,
  H_DCT = 18,
124
#endif  // CONFIG_EXT_TX
125
  TX_TYPES,
Jingning Han's avatar
Jingning Han committed
126
127
} TX_TYPE;

128

129
#if CONFIG_EXT_TX
130
#define EXT_TX_SIZES       4  // number of sizes that use extended transforms
131
#define EXT_TX_SETS_INTER  4  // Sets of transform selections for INTER
132
#define EXT_TX_SETS_INTRA  3  // Sets of transform selections for INTRA
133
134
#else
#define EXT_TX_SIZES       3  // number of sizes that use extended transforms
135
136
#endif  // CONFIG_EXT_TX

Jingning Han's avatar
Jingning Han committed
137
138
typedef enum {
  VP9_LAST_FLAG = 1 << 0,
139
140
141
142
143
144
145
#if CONFIG_EXT_REFS
  VP9_LAST2_FLAG = 1 << 1,
  VP9_LAST3_FLAG = 1 << 2,
  VP9_LAST4_FLAG = 1 << 3,
  VP9_GOLD_FLAG = 1 << 4,
  VP9_ALT_FLAG = 1 << 5,
#else
Jingning Han's avatar
Jingning Han committed
146
147
  VP9_GOLD_FLAG = 1 << 1,
  VP9_ALT_FLAG = 1 << 2,
148
#endif  // CONFIG_EXT_REFS
Jingning Han's avatar
Jingning Han committed
149
150
151
152
153
154
155
156
} VP9_REFFRAME;

typedef enum {
  PLANE_TYPE_Y  = 0,
  PLANE_TYPE_UV = 1,
  PLANE_TYPES
} PLANE_TYPE;

hui su's avatar
hui su committed
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
typedef enum {
  TWO_COLORS,
  THREE_COLORS,
  FOUR_COLORS,
  FIVE_COLORS,
  SIX_COLORS,
  SEVEN_COLORS,
  EIGHT_COLORS,
  PALETTE_SIZES
} PALETTE_SIZE;

typedef enum {
  PALETTE_COLOR_ONE,
  PALETTE_COLOR_TWO,
  PALETTE_COLOR_THREE,
  PALETTE_COLOR_FOUR,
  PALETTE_COLOR_FIVE,
  PALETTE_COLOR_SIX,
  PALETTE_COLOR_SEVEN,
  PALETTE_COLOR_EIGHT,
  PALETTE_COLORS
} PALETTE_COLOR;

Jingning Han's avatar
Jingning Han committed
180
181
182
183
184
185
186
187
188
189
190
191
192
193
#define DC_PRED    0       // Average of above and left pixels
#define V_PRED     1       // Vertical
#define H_PRED     2       // Horizontal
#define D45_PRED   3       // Directional 45  deg = round(arctan(1/1) * 180/pi)
#define D135_PRED  4       // Directional 135 deg = 180 - 45
#define D117_PRED  5       // Directional 117 deg = 180 - 63
#define D153_PRED  6       // Directional 153 deg = 180 - 27
#define D207_PRED  7       // Directional 207 deg = 180 + 27
#define D63_PRED   8       // Directional 63  deg = round(arctan(2/1) * 180/pi)
#define TM_PRED    9       // True-motion
#define NEARESTMV 10
#define NEARMV    11
#define ZEROMV    12
#define NEWMV     13
Yue Chen's avatar
Yue Chen committed
194
195
#if CONFIG_EXT_INTER
#define NEWFROMNEARMV     14
196
197
198
199
200
201
202
203
204
205
#define NEAREST_NEARESTMV 15
#define NEAREST_NEARMV    16
#define NEAR_NEARESTMV    17
#define NEAREST_NEWMV     18
#define NEW_NEARESTMV     19
#define NEAR_NEWMV        20
#define NEW_NEARMV        21
#define ZERO_ZEROMV       22
#define NEW_NEWMV         23
#define MB_MODE_COUNT     24
Yue Chen's avatar
Yue Chen committed
206
#else
Jingning Han's avatar
Jingning Han committed
207
#define MB_MODE_COUNT 14
Yue Chen's avatar
Yue Chen committed
208
#endif  // CONFIG_EXT_INTER
Jingning Han's avatar
Jingning Han committed
209
210
211
212
typedef uint8_t PREDICTION_MODE;

#define INTRA_MODES (TM_PRED + 1)

213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
#if CONFIG_EXT_INTRA
typedef enum {
  FILTER_DC_PRED,
  FILTER_V_PRED,
  FILTER_H_PRED,
  FILTER_D45_PRED,
  FILTER_D135_PRED,
  FILTER_D117_PRED,
  FILTER_D153_PRED,
  FILTER_D207_PRED,
  FILTER_D63_PRED,
  FILTER_TM_PRED,
  EXT_INTRA_MODES,
} EXT_INTRA_MODE;

#define FILTER_INTRA_MODES (FILTER_TM_PRED + 1)
#define DIRECTIONAL_MODES (INTRA_MODES - 2)
#endif  // CONFIG_EXT_INTRA

Yue Chen's avatar
Yue Chen committed
232
233
234
#if CONFIG_EXT_INTER
#define INTER_MODES (1 + NEWFROMNEARMV - NEARESTMV)
#else
Jingning Han's avatar
Jingning Han committed
235
#define INTER_MODES (1 + NEWMV - NEARESTMV)
Yue Chen's avatar
Yue Chen committed
236
#endif  // CONFIG_EXT_INTER
Jingning Han's avatar
Jingning Han committed
237

238
239
240
241
#if CONFIG_EXT_INTER
#define INTER_COMPOUND_MODES (1 + NEW_NEWMV - NEAREST_NEARESTMV)
#endif  // CONFIG_EXT_INTER

Jingning Han's avatar
Jingning Han committed
242
#define SKIP_CONTEXTS 3
243
244

#if CONFIG_REF_MV
245
#define NMV_CONTEXTS 3
246

247
#define NEWMV_MODE_CONTEXTS  7
248
#define ZEROMV_MODE_CONTEXTS 2
249
#define REFMV_MODE_CONTEXTS  9
250
#define DRL_MODE_CONTEXTS    3
251
252
253
254
255
256

#define ZEROMV_OFFSET 3
#define REFMV_OFFSET  4

#define NEWMV_CTX_MASK ((1 << ZEROMV_OFFSET) - 1)
#define ZEROMV_CTX_MASK ((1 << (REFMV_OFFSET - ZEROMV_OFFSET)) - 1)
257
258
259
260
261
#define REFMV_CTX_MASK ((1 << (8 - REFMV_OFFSET)) - 1)

#define ALL_ZERO_FLAG_OFFSET   8
#define SKIP_NEARESTMV_OFFSET  9
#define SKIP_NEARMV_OFFSET    10
262
#define SKIP_NEARESTMV_SUB8X8_OFFSET 11
263
264
#endif

Jingning Han's avatar
Jingning Han committed
265
266
267
268
#define INTER_MODE_CONTEXTS 7

/* Segment Feature Masks */
#define MAX_MV_REF_CANDIDATES 2
269
270
#if CONFIG_REF_MV
#define MAX_REF_MV_STACK_SIZE 16
271
#define REF_CAT_LEVEL  160
272
#endif
Jingning Han's avatar
Jingning Han committed
273
274
275
276
277

#define INTRA_INTER_CONTEXTS 4
#define COMP_INTER_CONTEXTS 5
#define REF_CONTEXTS 5

278
279
280
281
282
#if CONFIG_VAR_TX
#define TXFM_PARTITION_CONTEXTS 9
typedef TX_SIZE TXFM_CONTEXT;
#endif

283
284
285
286
287
288
289
290
#if CONFIG_EXT_REFS
#define SINGLE_REFS 6
#define COMP_REFS 5
#else
#define SINGLE_REFS 3
#define COMP_REFS 2
#endif  // CONFIG_EXT_REFS

291
292
293
294
295
#if CONFIG_SUPERTX
#define PARTITION_SUPERTX_CONTEXTS 2
#define MAX_SUPERTX_BLOCK_SIZE BLOCK_32X32
#endif  // CONFIG_SUPERTX

Jingning Han's avatar
Jingning Han committed
296
297
298
299
#ifdef __cplusplus
}  // extern "C"
#endif

300
#endif  // VP10_COMMON_ENUMS_H_