vp8.h 9.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351
  1. /*
  2. * VP8 compatible video decoder
  3. *
  4. * Copyright (C) 2010 David Conrad
  5. * Copyright (C) 2010 Ronald S. Bultje
  6. * Copyright (C) 2010 Fiona Glaser
  7. * Copyright (C) 2012 Daniel Kang
  8. *
  9. * This file is part of FFmpeg.
  10. *
  11. * FFmpeg is free software; you can redistribute it and/or
  12. * modify it under the terms of the GNU Lesser General Public
  13. * License as published by the Free Software Foundation; either
  14. * version 2.1 of the License, or (at your option) any later version.
  15. *
  16. * FFmpeg is distributed in the hope that it will be useful,
  17. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19. * Lesser General Public License for more details.
  20. *
  21. * You should have received a copy of the GNU Lesser General Public
  22. * License along with FFmpeg; if not, write to the Free Software
  23. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  24. */
  25. #ifndef AVCODEC_VP8_H
  26. #define AVCODEC_VP8_H
  27. #include <stdatomic.h>
  28. #include "libavutil/buffer.h"
  29. #include "libavutil/thread.h"
  30. #include "h264pred.h"
  31. #include "thread.h"
  32. #include "vp56.h"
  33. #include "vp8dsp.h"
  34. #define VP8_MAX_QUANT 127
  35. enum dct_token {
  36. DCT_0,
  37. DCT_1,
  38. DCT_2,
  39. DCT_3,
  40. DCT_4,
  41. DCT_CAT1,
  42. DCT_CAT2,
  43. DCT_CAT3,
  44. DCT_CAT4,
  45. DCT_CAT5,
  46. DCT_CAT6,
  47. DCT_EOB,
  48. NUM_DCT_TOKENS
  49. };
  50. // used to signal 4x4 intra pred in luma MBs
  51. #define MODE_I4x4 4
  52. enum inter_mvmode {
  53. VP8_MVMODE_ZERO = MODE_I4x4 + 1,
  54. VP8_MVMODE_MV,
  55. VP8_MVMODE_SPLIT
  56. };
  57. enum inter_splitmvmode {
  58. VP8_SPLITMVMODE_16x8 = 0, ///< 2 16x8 blocks (vertical)
  59. VP8_SPLITMVMODE_8x16, ///< 2 8x16 blocks (horizontal)
  60. VP8_SPLITMVMODE_8x8, ///< 2x2 blocks of 8x8px each
  61. VP8_SPLITMVMODE_4x4, ///< 4x4 blocks of 4x4px each
  62. VP8_SPLITMVMODE_NONE, ///< (only used in prediction) no split MVs
  63. };
  64. typedef struct VP8FilterStrength {
  65. uint8_t filter_level;
  66. uint8_t inner_limit;
  67. uint8_t inner_filter;
  68. } VP8FilterStrength;
  69. typedef struct VP8Macroblock {
  70. uint8_t skip;
  71. // TODO: make it possible to check for at least (i4x4 or split_mv)
  72. // in one op. are others needed?
  73. uint8_t mode;
  74. uint8_t ref_frame;
  75. uint8_t partitioning;
  76. uint8_t chroma_pred_mode;
  77. uint8_t segment;
  78. uint8_t intra4x4_pred_mode_mb[16];
  79. DECLARE_ALIGNED(4, uint8_t, intra4x4_pred_mode_top)[4];
  80. VP56mv mv;
  81. VP56mv bmv[16];
  82. } VP8Macroblock;
  83. typedef struct VP8intmv {
  84. int x;
  85. int y;
  86. } VP8intmv;
  87. typedef struct VP8mvbounds {
  88. VP8intmv mv_min;
  89. VP8intmv mv_max;
  90. } VP8mvbounds;
  91. typedef struct VP8ThreadData {
  92. DECLARE_ALIGNED(16, int16_t, block)[6][4][16];
  93. DECLARE_ALIGNED(16, int16_t, block_dc)[16];
  94. /**
  95. * This is the index plus one of the last non-zero coeff
  96. * for each of the blocks in the current macroblock.
  97. * So, 0 -> no coeffs
  98. * 1 -> dc-only (special transform)
  99. * 2+-> full transform
  100. */
  101. DECLARE_ALIGNED(16, uint8_t, non_zero_count_cache)[6][4];
  102. /**
  103. * For coeff decode, we need to know whether the above block had non-zero
  104. * coefficients. This means for each macroblock, we need data for 4 luma
  105. * blocks, 2 u blocks, 2 v blocks, and the luma dc block, for a total of 9
  106. * per macroblock. We keep the last row in top_nnz.
  107. */
  108. DECLARE_ALIGNED(8, uint8_t, left_nnz)[9];
  109. int thread_nr;
  110. #if HAVE_THREADS
  111. pthread_mutex_t lock;
  112. pthread_cond_t cond;
  113. #endif
  114. atomic_int thread_mb_pos; // (mb_y << 16) | (mb_x & 0xFFFF)
  115. atomic_int wait_mb_pos; // What the current thread is waiting on.
  116. #define EDGE_EMU_LINESIZE 32
  117. DECLARE_ALIGNED(16, uint8_t, edge_emu_buffer)[21 * EDGE_EMU_LINESIZE];
  118. VP8FilterStrength *filter_strength;
  119. VP8mvbounds mv_bounds;
  120. } VP8ThreadData;
  121. typedef struct VP8Frame {
  122. ThreadFrame tf;
  123. AVBufferRef *seg_map;
  124. AVBufferRef *hwaccel_priv_buf;
  125. void *hwaccel_picture_private;
  126. } VP8Frame;
  127. #define MAX_THREADS 8
  128. typedef struct VP8Context {
  129. VP8ThreadData *thread_data;
  130. AVCodecContext *avctx;
  131. enum AVPixelFormat pix_fmt;
  132. int actually_webp;
  133. VP8Frame *framep[4];
  134. VP8Frame *next_framep[4];
  135. VP8Frame *curframe;
  136. VP8Frame *prev_frame;
  137. uint16_t mb_width; /* number of horizontal MB */
  138. uint16_t mb_height; /* number of vertical MB */
  139. ptrdiff_t linesize;
  140. ptrdiff_t uvlinesize;
  141. uint8_t keyframe;
  142. uint8_t deblock_filter;
  143. uint8_t mbskip_enabled;
  144. uint8_t profile;
  145. VP8mvbounds mv_bounds;
  146. int8_t sign_bias[4]; ///< one state [0, 1] per ref frame type
  147. int ref_count[3];
  148. /**
  149. * Base parameters for segmentation, i.e. per-macroblock parameters.
  150. * These must be kept unchanged even if segmentation is not used for
  151. * a frame, since the values persist between interframes.
  152. */
  153. struct {
  154. uint8_t enabled;
  155. uint8_t absolute_vals;
  156. uint8_t update_map;
  157. uint8_t update_feature_data;
  158. int8_t base_quant[4];
  159. int8_t filter_level[4]; ///< base loop filter level
  160. } segmentation;
  161. struct {
  162. uint8_t simple;
  163. uint8_t level;
  164. uint8_t sharpness;
  165. } filter;
  166. VP8Macroblock *macroblocks;
  167. uint8_t *intra4x4_pred_mode_top;
  168. uint8_t intra4x4_pred_mode_left[4];
  169. /**
  170. * Macroblocks can have one of 4 different quants in a frame when
  171. * segmentation is enabled.
  172. * If segmentation is disabled, only the first segment's values are used.
  173. */
  174. struct {
  175. // [0] - DC qmul [1] - AC qmul
  176. int16_t luma_qmul[2];
  177. int16_t luma_dc_qmul[2]; ///< luma dc-only block quant
  178. int16_t chroma_qmul[2];
  179. } qmat[4];
  180. // Raw quantisation values, which may be needed by hwaccel decode.
  181. struct {
  182. int yac_qi;
  183. int ydc_delta;
  184. int y2dc_delta;
  185. int y2ac_delta;
  186. int uvdc_delta;
  187. int uvac_delta;
  188. } quant;
  189. struct {
  190. uint8_t enabled; ///< whether each mb can have a different strength based on mode/ref
  191. uint8_t update;
  192. /**
  193. * filter strength adjustment for the following macroblock modes:
  194. * [0-3] - i16x16 (always zero)
  195. * [4] - i4x4
  196. * [5] - zero mv
  197. * [6] - inter modes except for zero or split mv
  198. * [7] - split mv
  199. * i16x16 modes never have any adjustment
  200. */
  201. int8_t mode[VP8_MVMODE_SPLIT + 1];
  202. /**
  203. * filter strength adjustment for macroblocks that reference:
  204. * [0] - intra / VP56_FRAME_CURRENT
  205. * [1] - VP56_FRAME_PREVIOUS
  206. * [2] - VP56_FRAME_GOLDEN
  207. * [3] - altref / VP56_FRAME_GOLDEN2
  208. */
  209. int8_t ref[4];
  210. } lf_delta;
  211. uint8_t (*top_border)[16 + 8 + 8];
  212. uint8_t (*top_nnz)[9];
  213. VP56RangeCoder c; ///< header context, includes mb modes and motion vectors
  214. /* This contains the entropy coder state at the end of the header
  215. * block, in the form specified by the standard. For use by
  216. * hwaccels, so that a hardware decoder has the information to
  217. * start decoding at the macroblock layer.
  218. */
  219. struct {
  220. const uint8_t *input;
  221. uint32_t range;
  222. uint32_t value;
  223. int bit_count;
  224. } coder_state_at_header_end;
  225. int header_partition_size;
  226. /**
  227. * These are all of the updatable probabilities for binary decisions.
  228. * They are only implicitly reset on keyframes, making it quite likely
  229. * for an interframe to desync if a prior frame's header was corrupt
  230. * or missing outright!
  231. */
  232. struct {
  233. uint8_t segmentid[3];
  234. uint8_t mbskip;
  235. uint8_t intra;
  236. uint8_t last;
  237. uint8_t golden;
  238. uint8_t pred16x16[4];
  239. uint8_t pred8x8c[3];
  240. uint8_t token[4][16][3][NUM_DCT_TOKENS - 1];
  241. uint8_t mvc[2][19];
  242. uint8_t scan[16];
  243. } prob[2];
  244. VP8Macroblock *macroblocks_base;
  245. int invisible;
  246. int update_last; ///< update VP56_FRAME_PREVIOUS with the current one
  247. int update_golden; ///< VP56_FRAME_NONE if not updated, or which frame to copy if so
  248. int update_altref;
  249. /**
  250. * If this flag is not set, all the probability updates
  251. * are discarded after this frame is decoded.
  252. */
  253. int update_probabilities;
  254. /**
  255. * All coefficients are contained in separate arith coding contexts.
  256. * There can be 1, 2, 4, or 8 of these after the header context.
  257. */
  258. int num_coeff_partitions;
  259. VP56RangeCoder coeff_partition[8];
  260. int coeff_partition_size[8];
  261. VideoDSPContext vdsp;
  262. VP8DSPContext vp8dsp;
  263. H264PredContext hpc;
  264. vp8_mc_func put_pixels_tab[3][3][3];
  265. VP8Frame frames[5];
  266. uint8_t colorspace; ///< 0 is the only value allowed (meaning bt601)
  267. uint8_t fullrange; ///< whether we can skip clamping in dsp functions
  268. int num_jobs;
  269. /**
  270. * This describes the macroblock memory layout.
  271. * 0 -> Only width+height*2+1 macroblocks allocated (frame/single thread).
  272. * 1 -> Macroblocks for entire frame allocated (sliced thread).
  273. */
  274. int mb_layout;
  275. int (*decode_mb_row_no_filter)(AVCodecContext *avctx, void *tdata, int jobnr, int threadnr);
  276. void (*filter_mb_row)(AVCodecContext *avctx, void *tdata, int jobnr, int threadnr);
  277. int vp7;
  278. /**
  279. * Fade bit present in bitstream (VP7)
  280. */
  281. int fade_present;
  282. /**
  283. * Interframe DC prediction (VP7)
  284. * [0] VP56_FRAME_PREVIOUS
  285. * [1] VP56_FRAME_GOLDEN
  286. */
  287. uint16_t inter_dc_pred[2][2];
  288. /**
  289. * Macroblock features (VP7)
  290. */
  291. uint8_t feature_enabled[4];
  292. uint8_t feature_present_prob[4];
  293. uint8_t feature_index_prob[4][3];
  294. uint8_t feature_value[4][4];
  295. } VP8Context;
  296. int ff_vp8_decode_init(AVCodecContext *avctx);
  297. int ff_vp8_decode_frame(AVCodecContext *avctx, void *data, int *got_frame,
  298. AVPacket *avpkt);
  299. int ff_vp8_decode_free(AVCodecContext *avctx);
  300. #endif /* AVCODEC_VP8_H */