| /* |
| * Copyright (c) 2014 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. |
| */ |
| |
| #ifndef VP9_ENCODER_VP9_CONTEXT_TREE_H_ |
| #define VP9_ENCODER_VP9_CONTEXT_TREE_H_ |
| |
| #include "vp9/common/vp9_blockd.h" |
| #include "vp9/encoder/vp9_block.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| struct VP9_COMP; |
| struct VP9Common; |
| struct ThreadData; |
| |
| // Structure to hold snapshot of coding context during the mode picking process |
| typedef struct { |
| MODE_INFO mic; |
| MB_MODE_INFO_EXT mbmi_ext; |
| uint8_t *zcoeff_blk; |
| tran_low_t *coeff[MAX_MB_PLANE][3]; |
| tran_low_t *qcoeff[MAX_MB_PLANE][3]; |
| tran_low_t *dqcoeff[MAX_MB_PLANE][3]; |
| uint16_t *eobs[MAX_MB_PLANE][3]; |
| |
| // dual buffer pointers, 0: in use, 1: best in store |
| tran_low_t *coeff_pbuf[MAX_MB_PLANE][3]; |
| tran_low_t *qcoeff_pbuf[MAX_MB_PLANE][3]; |
| tran_low_t *dqcoeff_pbuf[MAX_MB_PLANE][3]; |
| uint16_t *eobs_pbuf[MAX_MB_PLANE][3]; |
| |
| int is_coded; |
| int num_4x4_blk; |
| int skip; |
| int pred_pixel_ready; |
| // For current partition, only if all Y, U, and V transform blocks' |
| // coefficients are quantized to 0, skippable is set to 0. |
| int skippable; |
| uint8_t skip_txfm[MAX_MB_PLANE << 2]; |
| int best_mode_index; |
| int hybrid_pred_diff; |
| int comp_pred_diff; |
| int single_pred_diff; |
| int64_t best_filter_diff[SWITCHABLE_FILTER_CONTEXTS]; |
| |
| // TODO(jingning) Use RD_COST struct here instead. This involves a boarder |
| // scope of refactoring. |
| int rate; |
| int64_t dist; |
| |
| #if CONFIG_VP9_TEMPORAL_DENOISING |
| unsigned int newmv_sse; |
| unsigned int zeromv_sse; |
| unsigned int zeromv_lastref_sse; |
| PREDICTION_MODE best_sse_inter_mode; |
| int_mv best_sse_mv; |
| MV_REFERENCE_FRAME best_reference_frame; |
| MV_REFERENCE_FRAME best_zeromv_reference_frame; |
| #endif |
| |
| // motion vector cache for adaptive motion search control in partition |
| // search loop |
| MV pred_mv[MAX_REF_FRAMES]; |
| INTERP_FILTER pred_interp_filter; |
| } PICK_MODE_CONTEXT; |
| |
| typedef struct PC_TREE { |
| int index; |
| PARTITION_TYPE partitioning; |
| BLOCK_SIZE block_size; |
| PICK_MODE_CONTEXT none; |
| PICK_MODE_CONTEXT horizontal[2]; |
| PICK_MODE_CONTEXT vertical[2]; |
| union { |
| struct PC_TREE *split[4]; |
| PICK_MODE_CONTEXT *leaf_split[4]; |
| }; |
| } PC_TREE; |
| |
| void vp9_setup_pc_tree(struct VP9Common *cm, struct ThreadData *td); |
| void vp9_free_pc_tree(struct ThreadData *td); |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif /* VP9_ENCODER_VP9_CONTEXT_TREE_H_ */ |