| OLD | NEW |
| (Empty) | |
| 1 /* |
| 2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved. |
| 3 * |
| 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ |
| 10 |
| 11 |
| 12 #ifndef VP9_COMMON_VP9_BLOCKD_H_ |
| 13 #define VP9_COMMON_VP9_BLOCKD_H_ |
| 14 |
| 15 #include "./vpx_config.h" |
| 16 |
| 17 #include "vpx_ports/mem.h" |
| 18 #include "vpx_scale/yv12config.h" |
| 19 |
| 20 #include "vp9/common/vp9_common_data.h" |
| 21 #include "vp9/common/vp9_filter.h" |
| 22 #include "vp9/common/vp9_mv.h" |
| 23 #include "vp9/common/vp9_scale.h" |
| 24 |
| 25 #ifdef __cplusplus |
| 26 extern "C" { |
| 27 #endif |
| 28 |
| 29 #define BLOCK_SIZE_GROUPS 4 |
| 30 #define SKIP_CONTEXTS 3 |
| 31 #define INTER_MODE_CONTEXTS 7 |
| 32 |
| 33 /* Segment Feature Masks */ |
| 34 #define MAX_MV_REF_CANDIDATES 2 |
| 35 |
| 36 #define INTRA_INTER_CONTEXTS 4 |
| 37 #define COMP_INTER_CONTEXTS 5 |
| 38 #define REF_CONTEXTS 5 |
| 39 |
| 40 typedef enum { |
| 41 PLANE_TYPE_Y = 0, |
| 42 PLANE_TYPE_UV = 1, |
| 43 PLANE_TYPES |
| 44 } PLANE_TYPE; |
| 45 |
| 46 #define MAX_MB_PLANE 3 |
| 47 |
| 48 typedef char ENTROPY_CONTEXT; |
| 49 |
| 50 static INLINE int combine_entropy_contexts(ENTROPY_CONTEXT a, |
| 51 ENTROPY_CONTEXT b) { |
| 52 return (a != 0) + (b != 0); |
| 53 } |
| 54 |
| 55 typedef enum { |
| 56 KEY_FRAME = 0, |
| 57 INTER_FRAME = 1, |
| 58 FRAME_TYPES, |
| 59 } FRAME_TYPE; |
| 60 |
| 61 typedef enum { |
| 62 DC_PRED, // Average of above and left pixels |
| 63 V_PRED, // Vertical |
| 64 H_PRED, // Horizontal |
| 65 D45_PRED, // Directional 45 deg = round(arctan(1/1) * 180/pi) |
| 66 D135_PRED, // Directional 135 deg = 180 - 45 |
| 67 D117_PRED, // Directional 117 deg = 180 - 63 |
| 68 D153_PRED, // Directional 153 deg = 180 - 27 |
| 69 D207_PRED, // Directional 207 deg = 180 + 27 |
| 70 D63_PRED, // Directional 63 deg = round(arctan(2/1) * 180/pi) |
| 71 TM_PRED, // True-motion |
| 72 NEARESTMV, |
| 73 NEARMV, |
| 74 ZEROMV, |
| 75 NEWMV, |
| 76 MB_MODE_COUNT |
| 77 } PREDICTION_MODE; |
| 78 |
| 79 static INLINE int is_inter_mode(PREDICTION_MODE mode) { |
| 80 return mode >= NEARESTMV && mode <= NEWMV; |
| 81 } |
| 82 |
| 83 #define INTRA_MODES (TM_PRED + 1) |
| 84 |
| 85 #define INTER_MODES (1 + NEWMV - NEARESTMV) |
| 86 |
| 87 #define INTER_OFFSET(mode) ((mode) - NEARESTMV) |
| 88 |
| 89 /* For keyframes, intra block modes are predicted by the (already decoded) |
| 90 modes for the Y blocks to the left and above us; for interframes, there |
| 91 is a single probability table. */ |
| 92 |
| 93 typedef struct { |
| 94 PREDICTION_MODE as_mode; |
| 95 int_mv as_mv[2]; // first, second inter predictor motion vectors |
| 96 } b_mode_info; |
| 97 |
| 98 // Note that the rate-distortion optimization loop, bit-stream writer, and |
| 99 // decoder implementation modules critically rely on the enum entry values |
| 100 // specified herein. They should be refactored concurrently. |
| 101 typedef enum { |
| 102 NONE = -1, |
| 103 INTRA_FRAME = 0, |
| 104 LAST_FRAME = 1, |
| 105 GOLDEN_FRAME = 2, |
| 106 ALTREF_FRAME = 3, |
| 107 MAX_REF_FRAMES = 4 |
| 108 } MV_REFERENCE_FRAME; |
| 109 |
| 110 // This structure now relates to 8x8 block regions. |
| 111 typedef struct { |
| 112 // Common for both INTER and INTRA blocks |
| 113 BLOCK_SIZE sb_type; |
| 114 PREDICTION_MODE mode; |
| 115 TX_SIZE tx_size; |
| 116 int8_t skip; |
| 117 int8_t segment_id; |
| 118 int8_t seg_id_predicted; // valid only when temporal_update is enabled |
| 119 |
| 120 // Only for INTRA blocks |
| 121 PREDICTION_MODE uv_mode; |
| 122 |
| 123 // Only for INTER blocks |
| 124 MV_REFERENCE_FRAME ref_frame[2]; |
| 125 int_mv mv[2]; |
| 126 int_mv ref_mvs[MAX_REF_FRAMES][MAX_MV_REF_CANDIDATES]; |
| 127 uint8_t mode_context[MAX_REF_FRAMES]; |
| 128 INTERP_FILTER interp_filter; |
| 129 |
| 130 } MB_MODE_INFO; |
| 131 |
| 132 typedef struct MODE_INFO { |
| 133 MB_MODE_INFO mbmi; |
| 134 b_mode_info bmi[4]; |
| 135 } MODE_INFO; |
| 136 |
| 137 static INLINE PREDICTION_MODE get_y_mode(const MODE_INFO *mi, int block) { |
| 138 return mi->mbmi.sb_type < BLOCK_8X8 ? mi->bmi[block].as_mode |
| 139 : mi->mbmi.mode; |
| 140 } |
| 141 |
| 142 static INLINE int is_inter_block(const MB_MODE_INFO *mbmi) { |
| 143 return mbmi->ref_frame[0] > INTRA_FRAME; |
| 144 } |
| 145 |
| 146 static INLINE int has_second_ref(const MB_MODE_INFO *mbmi) { |
| 147 return mbmi->ref_frame[1] > INTRA_FRAME; |
| 148 } |
| 149 |
| 150 PREDICTION_MODE vp9_left_block_mode(const MODE_INFO *cur_mi, |
| 151 const MODE_INFO *left_mi, int b); |
| 152 |
| 153 PREDICTION_MODE vp9_above_block_mode(const MODE_INFO *cur_mi, |
| 154 const MODE_INFO *above_mi, int b); |
| 155 |
| 156 enum mv_precision { |
| 157 MV_PRECISION_Q3, |
| 158 MV_PRECISION_Q4 |
| 159 }; |
| 160 |
| 161 struct buf_2d { |
| 162 uint8_t *buf; |
| 163 int stride; |
| 164 }; |
| 165 |
| 166 struct macroblockd_plane { |
| 167 tran_low_t *dqcoeff; |
| 168 PLANE_TYPE plane_type; |
| 169 int subsampling_x; |
| 170 int subsampling_y; |
| 171 struct buf_2d dst; |
| 172 struct buf_2d pre[2]; |
| 173 const int16_t *dequant; |
| 174 ENTROPY_CONTEXT *above_context; |
| 175 ENTROPY_CONTEXT *left_context; |
| 176 }; |
| 177 |
| 178 #define BLOCK_OFFSET(x, i) ((x) + (i) * 16) |
| 179 |
| 180 typedef struct RefBuffer { |
| 181 // TODO(dkovalev): idx is not really required and should be removed, now it |
| 182 // is used in vp9_onyxd_if.c |
| 183 int idx; |
| 184 YV12_BUFFER_CONFIG *buf; |
| 185 struct scale_factors sf; |
| 186 } RefBuffer; |
| 187 |
| 188 typedef struct macroblockd { |
| 189 struct macroblockd_plane plane[MAX_MB_PLANE]; |
| 190 |
| 191 int mi_stride; |
| 192 |
| 193 MODE_INFO **mi; |
| 194 MODE_INFO *left_mi; |
| 195 MODE_INFO *above_mi; |
| 196 MB_MODE_INFO *left_mbmi; |
| 197 MB_MODE_INFO *above_mbmi; |
| 198 |
| 199 int up_available; |
| 200 int left_available; |
| 201 |
| 202 /* Distance of MB away from frame edges */ |
| 203 int mb_to_left_edge; |
| 204 int mb_to_right_edge; |
| 205 int mb_to_top_edge; |
| 206 int mb_to_bottom_edge; |
| 207 |
| 208 /* pointers to reference frames */ |
| 209 RefBuffer *block_refs[2]; |
| 210 |
| 211 /* pointer to current frame */ |
| 212 const YV12_BUFFER_CONFIG *cur_buf; |
| 213 |
| 214 ENTROPY_CONTEXT *above_context[MAX_MB_PLANE]; |
| 215 ENTROPY_CONTEXT left_context[MAX_MB_PLANE][16]; |
| 216 |
| 217 PARTITION_CONTEXT *above_seg_context; |
| 218 PARTITION_CONTEXT left_seg_context[8]; |
| 219 |
| 220 /* mc buffer */ |
| 221 DECLARE_ALIGNED(16, uint8_t, mc_buf[80 * 2 * 80 * 2]); |
| 222 |
| 223 #if CONFIG_VP9_HIGHBITDEPTH |
| 224 /* Bit depth: 8, 10, 12 */ |
| 225 int bd; |
| 226 DECLARE_ALIGNED(16, uint16_t, mc_buf_high[80 * 2 * 80 * 2]); |
| 227 #endif |
| 228 |
| 229 /* dqcoeff are shared by all the planes. So planes must be decoded serially */ |
| 230 DECLARE_ALIGNED(16, tran_low_t, dqcoeff[64 * 64]); |
| 231 |
| 232 int lossless; |
| 233 int corrupted; |
| 234 |
| 235 struct vpx_internal_error_info *error_info; |
| 236 } MACROBLOCKD; |
| 237 |
| 238 static INLINE BLOCK_SIZE get_subsize(BLOCK_SIZE bsize, |
| 239 PARTITION_TYPE partition) { |
| 240 return subsize_lookup[partition][bsize]; |
| 241 } |
| 242 |
| 243 extern const TX_TYPE intra_mode_to_tx_type_lookup[INTRA_MODES]; |
| 244 |
| 245 static INLINE TX_TYPE get_tx_type(PLANE_TYPE plane_type, |
| 246 const MACROBLOCKD *xd) { |
| 247 const MB_MODE_INFO *const mbmi = &xd->mi[0]->mbmi; |
| 248 |
| 249 if (plane_type != PLANE_TYPE_Y || xd->lossless || is_inter_block(mbmi)) |
| 250 return DCT_DCT; |
| 251 |
| 252 return intra_mode_to_tx_type_lookup[mbmi->mode]; |
| 253 } |
| 254 |
| 255 static INLINE TX_TYPE get_tx_type_4x4(PLANE_TYPE plane_type, |
| 256 const MACROBLOCKD *xd, int ib) { |
| 257 const MODE_INFO *const mi = xd->mi[0]; |
| 258 |
| 259 if (plane_type != PLANE_TYPE_Y || xd->lossless || is_inter_block(&mi->mbmi)) |
| 260 return DCT_DCT; |
| 261 |
| 262 return intra_mode_to_tx_type_lookup[get_y_mode(mi, ib)]; |
| 263 } |
| 264 |
| 265 void vp9_setup_block_planes(MACROBLOCKD *xd, int ss_x, int ss_y); |
| 266 |
| 267 static INLINE TX_SIZE get_uv_tx_size_impl(TX_SIZE y_tx_size, BLOCK_SIZE bsize, |
| 268 int xss, int yss) { |
| 269 if (bsize < BLOCK_8X8) { |
| 270 return TX_4X4; |
| 271 } else { |
| 272 const BLOCK_SIZE plane_bsize = ss_size_lookup[bsize][xss][yss]; |
| 273 return MIN(y_tx_size, max_txsize_lookup[plane_bsize]); |
| 274 } |
| 275 } |
| 276 |
| 277 static INLINE TX_SIZE get_uv_tx_size(const MB_MODE_INFO *mbmi, |
| 278 const struct macroblockd_plane *pd) { |
| 279 return get_uv_tx_size_impl(mbmi->tx_size, mbmi->sb_type, pd->subsampling_x, |
| 280 pd->subsampling_y); |
| 281 } |
| 282 |
| 283 static INLINE BLOCK_SIZE get_plane_block_size(BLOCK_SIZE bsize, |
| 284 const struct macroblockd_plane *pd) { |
| 285 return ss_size_lookup[bsize][pd->subsampling_x][pd->subsampling_y]; |
| 286 } |
| 287 |
| 288 typedef void (*foreach_transformed_block_visitor)(int plane, int block, |
| 289 BLOCK_SIZE plane_bsize, |
| 290 TX_SIZE tx_size, |
| 291 void *arg); |
| 292 |
| 293 void vp9_foreach_transformed_block_in_plane( |
| 294 const MACROBLOCKD *const xd, BLOCK_SIZE bsize, int plane, |
| 295 foreach_transformed_block_visitor visit, void *arg); |
| 296 |
| 297 |
| 298 void vp9_foreach_transformed_block( |
| 299 const MACROBLOCKD* const xd, BLOCK_SIZE bsize, |
| 300 foreach_transformed_block_visitor visit, void *arg); |
| 301 |
| 302 static INLINE void txfrm_block_to_raster_xy(BLOCK_SIZE plane_bsize, |
| 303 TX_SIZE tx_size, int block, |
| 304 int *x, int *y) { |
| 305 const int bwl = b_width_log2_lookup[plane_bsize]; |
| 306 const int tx_cols_log2 = bwl - tx_size; |
| 307 const int tx_cols = 1 << tx_cols_log2; |
| 308 const int raster_mb = block >> (tx_size << 1); |
| 309 *x = (raster_mb & (tx_cols - 1)) << tx_size; |
| 310 *y = (raster_mb >> tx_cols_log2) << tx_size; |
| 311 } |
| 312 |
| 313 void vp9_set_contexts(const MACROBLOCKD *xd, struct macroblockd_plane *pd, |
| 314 BLOCK_SIZE plane_bsize, TX_SIZE tx_size, int has_eob, |
| 315 int aoff, int loff); |
| 316 |
| 317 #ifdef __cplusplus |
| 318 } // extern "C" |
| 319 #endif |
| 320 |
| 321 #endif // VP9_COMMON_VP9_BLOCKD_H_ |
| OLD | NEW |