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 VPX_VP9_COMMON_VP9_SEG_COMMON_H_
12 #define VPX_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
segfeature_active(const struct segmentation * seg,int segment_id,SEG_LVL_FEATURES feature_id)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
get_segdata(const struct segmentation * seg,int segment_id,SEG_LVL_FEATURES feature_id)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 // VPX_VP9_COMMON_VP9_SEG_COMMON_H_
82