1 /* 2 * Copyright (c) 2012 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 #ifndef VP9_COMMON_VP9_SEG_COMMON_H_ 12 #define VP9_COMMON_VP9_SEG_COMMON_H_ 13 14 #include "vpx_dsp/prob.h" 15 16 #ifdef __cplusplus 17 extern "C" { 18 #endif 19 20 #define SEGMENT_DELTADATA 0 21 #define SEGMENT_ABSDATA 1 22 23 #define MAX_SEGMENTS 8 24 #define SEG_TREE_PROBS (MAX_SEGMENTS - 1) 25 26 #define PREDICTION_PROBS 3 27 28 // Segment level features. 29 typedef enum { 30 SEG_LVL_ALT_Q = 0, // Use alternate Quantizer .... 31 SEG_LVL_ALT_LF = 1, // Use alternate loop filter value... 32 SEG_LVL_REF_FRAME = 2, // Optional Segment reference frame 33 SEG_LVL_SKIP = 3, // Optional Segment (0,0) + skip mode 34 SEG_LVL_MAX = 4 // Number of features supported 35 } SEG_LVL_FEATURES; 36 37 struct segmentation { 38 uint8_t enabled; 39 uint8_t update_map; 40 uint8_t update_data; 41 uint8_t abs_delta; 42 uint8_t temporal_update; 43 44 vpx_prob tree_probs[SEG_TREE_PROBS]; 45 vpx_prob pred_probs[PREDICTION_PROBS]; 46 47 int16_t feature_data[MAX_SEGMENTS][SEG_LVL_MAX]; 48 uint32_t feature_mask[MAX_SEGMENTS]; 49 int aq_av_offset; 50 }; 51 52 static INLINE int segfeature_active(const struct segmentation *seg, 53 int segment_id, 54 SEG_LVL_FEATURES feature_id) { 55 return seg->enabled && (seg->feature_mask[segment_id] & (1 << feature_id)); 56 } 57 58 void vp9_clearall_segfeatures(struct segmentation *seg); 59 60 void vp9_enable_segfeature(struct segmentation *seg, int segment_id, 61 SEG_LVL_FEATURES feature_id); 62 63 int vp9_seg_feature_data_max(SEG_LVL_FEATURES feature_id); 64 65 int vp9_is_segfeature_signed(SEG_LVL_FEATURES feature_id); 66 67 void vp9_set_segdata(struct segmentation *seg, int segment_id, 68 SEG_LVL_FEATURES feature_id, int seg_data); 69 70 static INLINE int get_segdata(const struct segmentation *seg, int segment_id, 71 SEG_LVL_FEATURES feature_id) { 72 return seg->feature_data[segment_id][feature_id]; 73 } 74 75 extern const vpx_tree_index vp9_segment_tree[TREE_SIZE(MAX_SEGMENTS)]; 76 77 #ifdef __cplusplus 78 } // extern "C" 79 #endif 80 81 #endif // VP9_COMMON_VP9_SEG_COMMON_H_ 82