]> git.sesse.net Git - ffmpeg/blob - libavcodec/vp9shared.h
Merge commit 'e23190269fb6e8217d080918893641ba3e0e3556'
[ffmpeg] / libavcodec / vp9shared.h
1 /*
2  * VP9 compatible video decoder
3  *
4  * Copyright (C) 2013 Ronald S. Bultje <rsbultje gmail com>
5  * Copyright (C) 2013 Clément Bœsch <u pkh me>
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_VP9SHARED_H
25 #define AVCODEC_VP9SHARED_H
26
27 #include <stddef.h>
28 #include <stdint.h>
29
30 #include "vp9.h"
31 #include "thread.h"
32 #include "vp56.h"
33
34 enum BlockPartition {
35     PARTITION_NONE,    // [ ] <-.
36     PARTITION_H,       // [-]   |
37     PARTITION_V,       // [|]   |
38     PARTITION_SPLIT,   // [+] --'
39 };
40
41 enum InterPredMode {
42     NEARESTMV = 10,
43     NEARMV    = 11,
44     ZEROMV    = 12,
45     NEWMV     = 13,
46 };
47
48 enum CompPredMode {
49     PRED_SINGLEREF,
50     PRED_COMPREF,
51     PRED_SWITCHABLE,
52 };
53
54 typedef struct VP9mvrefPair {
55     VP56mv mv[2];
56     int8_t ref[2];
57 } VP9mvrefPair;
58
59 typedef struct VP9Frame {
60     ThreadFrame tf;
61     AVBufferRef *extradata;
62     uint8_t *segmentation_map;
63     VP9mvrefPair *mv;
64     int uses_2pass;
65
66     AVBufferRef *hwaccel_priv_buf;
67     void *hwaccel_picture_private;
68 } VP9Frame;
69
70 enum BlockLevel {
71     BL_64X64,
72     BL_32X32,
73     BL_16X16,
74     BL_8X8,
75 };
76
77 enum BlockSize {
78     BS_64x64,
79     BS_64x32,
80     BS_32x64,
81     BS_32x32,
82     BS_32x16,
83     BS_16x32,
84     BS_16x16,
85     BS_16x8,
86     BS_8x16,
87     BS_8x8,
88     BS_8x4,
89     BS_4x8,
90     BS_4x4,
91     N_BS_SIZES,
92 };
93
94 typedef struct VP9BitstreamHeader {
95     // bitstream header
96     uint8_t profile;
97     uint8_t bpp;
98     uint8_t keyframe;
99     uint8_t invisible;
100     uint8_t errorres;
101     uint8_t intraonly;
102     uint8_t resetctx;
103     uint8_t refreshrefmask;
104     uint8_t highprecisionmvs;
105     enum FilterMode filtermode;
106     uint8_t allowcompinter;
107     uint8_t refreshctx;
108     uint8_t parallelmode;
109     uint8_t framectxid;
110     uint8_t use_last_frame_mvs;
111     uint8_t refidx[3];
112     uint8_t signbias[3];
113     uint8_t fixcompref;
114     uint8_t varcompref[2];
115     struct {
116         uint8_t level;
117         int8_t sharpness;
118     } filter;
119     struct {
120         uint8_t enabled;
121         uint8_t updated;
122         int8_t mode[2];
123         int8_t ref[4];
124     } lf_delta;
125     uint8_t yac_qi;
126     int8_t ydc_qdelta, uvdc_qdelta, uvac_qdelta;
127     uint8_t lossless;
128 #define MAX_SEGMENT 8
129     struct {
130         uint8_t enabled;
131         uint8_t temporal;
132         uint8_t absolute_vals;
133         uint8_t update_map;
134         uint8_t prob[7];
135         uint8_t pred_prob[3];
136         struct {
137             uint8_t q_enabled;
138             uint8_t lf_enabled;
139             uint8_t ref_enabled;
140             uint8_t skip_enabled;
141             uint8_t ref_val;
142             int16_t q_val;
143             int8_t lf_val;
144             int16_t qmul[2][2];
145             uint8_t lflvl[4][2];
146         } feat[MAX_SEGMENT];
147     } segmentation;
148     enum TxfmMode txfmmode;
149     enum CompPredMode comppredmode;
150     struct {
151         unsigned log2_tile_cols, log2_tile_rows;
152         unsigned tile_cols, tile_rows;
153     } tiling;
154
155     int uncompressed_header_size;
156     int compressed_header_size;
157 } VP9BitstreamHeader;
158
159 typedef struct VP9SharedContext {
160     VP9BitstreamHeader h;
161
162     ThreadFrame refs[8];
163 #define CUR_FRAME 0
164 #define REF_FRAME_MVPAIR 1
165 #define REF_FRAME_SEGMAP 2
166     VP9Frame frames[3];
167 } VP9SharedContext;
168
169 #endif /* AVCODEC_VP9SHARED_H */