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 "vp9/common/vp9_prob.h"
15
16#ifdef __cplusplus
17extern "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.
29typedef 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
38struct segmentation {
39  uint8_t enabled;
40  uint8_t update_map;
41  uint8_t update_data;
42  uint8_t abs_delta;
43  uint8_t temporal_update;
44
45  vp9_prob tree_probs[SEG_TREE_PROBS];
46  vp9_prob pred_probs[PREDICTION_PROBS];
47
48  int16_t feature_data[MAX_SEGMENTS][SEG_LVL_MAX];
49  unsigned int feature_mask[MAX_SEGMENTS];
50};
51
52int vp9_segfeature_active(const struct segmentation *seg,
53                          int segment_id,
54                          SEG_LVL_FEATURES feature_id);
55
56void vp9_clearall_segfeatures(struct segmentation *seg);
57
58void vp9_enable_segfeature(struct segmentation *seg,
59                           int segment_id,
60                           SEG_LVL_FEATURES feature_id);
61
62int vp9_seg_feature_data_max(SEG_LVL_FEATURES feature_id);
63
64int vp9_is_segfeature_signed(SEG_LVL_FEATURES feature_id);
65
66void vp9_set_segdata(struct segmentation *seg,
67                     int segment_id,
68                     SEG_LVL_FEATURES feature_id,
69                     int seg_data);
70
71int vp9_get_segdata(const struct segmentation *seg,
72                    int segment_id,
73                    SEG_LVL_FEATURES feature_id);
74
75extern const vp9_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
83