vp9_seg_common.c 2.44 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
/*
 *  Copyright (c) 2010 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.
 */

John Koleszar's avatar
John Koleszar committed
11
#include <assert.h>
12

John Koleszar's avatar
John Koleszar committed
13
#include "vp9/common/vp9_blockd.h"
14
#include "vp9/common/vp9_loopfilter.h"
15
#include "vp9/common/vp9_seg_common.h"
16
#include "vp9/common/vp9_quant_common.h"
17

Dmitry Kovalev's avatar
Dmitry Kovalev committed
18
static const int seg_feature_data_signed[SEG_LVL_MAX] = { 1, 1, 0, 0 };
Dmitry Kovalev's avatar
Dmitry Kovalev committed
19
20
21

static const int seg_feature_data_max[SEG_LVL_MAX] = {
  MAXQ, MAX_LOOP_FILTER, 3, 0 };
22
23
24
25
26
27

// These functions provide access to new segment level features.
// Eventually these function may be "optimized out" but for the moment,
// the coding mechanism is still subject to change so these provide a
// convenient single point of change.

28
int vp9_segfeature_active(const struct segmentation *seg, int segment_id,
29
                          SEG_LVL_FEATURES feature_id) {
30
31
  return seg->enabled &&
         (seg->feature_mask[segment_id] & (1 << feature_id));
32
33
}

34
void vp9_clearall_segfeatures(struct segmentation *seg) {
35
36
  vp9_zero(seg->feature_data);
  vp9_zero(seg->feature_mask);
37
38
}

39
void vp9_enable_segfeature(struct segmentation *seg, int segment_id,
40
                           SEG_LVL_FEATURES feature_id) {
41
  seg->feature_mask[segment_id] |= 1 << feature_id;
42
43
}

John Koleszar's avatar
John Koleszar committed
44
45
int vp9_seg_feature_data_max(SEG_LVL_FEATURES feature_id) {
  return seg_feature_data_max[feature_id];
46
47
}

48
int vp9_is_segfeature_signed(SEG_LVL_FEATURES feature_id) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
49
  return seg_feature_data_signed[feature_id];
50
51
}

52
void vp9_set_segdata(struct segmentation *seg, int segment_id,
Dmitry Kovalev's avatar
Dmitry Kovalev committed
53
                     SEG_LVL_FEATURES feature_id, int seg_data) {
John Koleszar's avatar
John Koleszar committed
54
55
  assert(seg_data <= seg_feature_data_max[feature_id]);
  if (seg_data < 0) {
Dmitry Kovalev's avatar
Dmitry Kovalev committed
56
    assert(seg_feature_data_signed[feature_id]);
John Koleszar's avatar
John Koleszar committed
57
58
59
    assert(-seg_data <= seg_feature_data_max[feature_id]);
  }

60
  seg->feature_data[segment_id][feature_id] = seg_data;
61
62
}

63
int vp9_get_segdata(const struct segmentation *seg, int segment_id,
64
                    SEG_LVL_FEATURES feature_id) {
65
  return seg->feature_data[segment_id][feature_id];
66
}
67

68

69
const vp9_tree_index vp9_segment_tree[TREE_SIZE(MAX_SEGMENTS)] = {
70
71
72
73
74
  2,  4,  6,  8, 10, 12,
  0, -1, -2, -3, -4, -5, -6, -7
};


75
// TBD? Functions to read and write segment data with range / validity checking