Home | History | Annotate | Download | only in common 
      1 /*
      2  *  Copyright (c) 2010 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 #include <assert.h>
     12 
     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 
     18 static const int seg_feature_data_signed[SEG_LVL_MAX] = { 1, 1, 0, 0 };
     19 
     20 static const int seg_feature_data_max[SEG_LVL_MAX] = {
     21   MAXQ, MAX_LOOP_FILTER, 3, 0 };
     22 
     23 // These functions provide access to new segment level features.
     24 // Eventually these function may be "optimized out" but for the moment,
     25 // the coding mechanism is still subject to change so these provide a
     26 // convenient single point of change.
     27 
     28 int vp9_segfeature_active(const struct segmentation *seg, int segment_id,
     29                           SEG_LVL_FEATURES feature_id) {
     30   return seg->enabled &&
     31          (seg->feature_mask[segment_id] & (1 << feature_id));
     32 }
     33 
     34 void vp9_clearall_segfeatures(struct segmentation *seg) {
     35   vp9_zero(seg->feature_data);
     36   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 
     44 void vp9_disable_segfeature(struct segmentation *seg, int segment_id,
     45                             SEG_LVL_FEATURES feature_id) {
     46   seg->feature_mask[segment_id] &= ~(1 << feature_id);
     47 }
     48 
     49 int vp9_seg_feature_data_max(SEG_LVL_FEATURES feature_id) {
     50   return seg_feature_data_max[feature_id];
     51 }
     52 
     53 int vp9_is_segfeature_signed(SEG_LVL_FEATURES feature_id) {
     54   return seg_feature_data_signed[feature_id];
     55 }
     56 
     57 void vp9_clear_segdata(struct segmentation *seg, int segment_id,
     58                        SEG_LVL_FEATURES feature_id) {
     59   seg->feature_data[segment_id][feature_id] = 0;
     60 }
     61 
     62 void vp9_set_segdata(struct segmentation *seg, int segment_id,
     63                      SEG_LVL_FEATURES feature_id, int seg_data) {
     64   assert(seg_data <= seg_feature_data_max[feature_id]);
     65   if (seg_data < 0) {
     66     assert(seg_feature_data_signed[feature_id]);
     67     assert(-seg_data <= seg_feature_data_max[feature_id]);
     68   }
     69 
     70   seg->feature_data[segment_id][feature_id] = seg_data;
     71 }
     72 
     73 int vp9_get_segdata(const struct segmentation *seg, int segment_id,
     74                     SEG_LVL_FEATURES feature_id) {
     75   return seg->feature_data[segment_id][feature_id];
     76 }
     77 
     78 
     79 const vp9_tree_index vp9_segment_tree[TREE_SIZE(MAX_SEGMENTS)] = {
     80   2,  4,  6,  8, 10, 12,
     81   0, -1, -2, -3, -4, -5, -6, -7
     82 };
     83 
     84 
     85 // TBD? Functions to read and write segment data with range / validity checking
     86