vp9_seg_common.c 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  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. #include <assert.h>
  11. #include "vp9/common/vp9_blockd.h"
  12. #include "vp9/common/vp9_loopfilter.h"
  13. #include "vp9/common/vp9_seg_common.h"
  14. #include "vp9/common/vp9_quant_common.h"
  15. static const int seg_feature_data_signed[SEG_LVL_MAX] = { 1, 1, 0, 0 };
  16. static const int seg_feature_data_max[SEG_LVL_MAX] = {
  17. MAXQ, MAX_LOOP_FILTER, 3, 0 };
  18. // These functions provide access to new segment level features.
  19. // Eventually these function may be "optimized out" but for the moment,
  20. // the coding mechanism is still subject to change so these provide a
  21. // convenient single point of change.
  22. void vp9_clearall_segfeatures(struct segmentation *seg) {
  23. vp9_zero(seg->feature_data);
  24. vp9_zero(seg->feature_mask);
  25. seg->aq_av_offset = 0;
  26. }
  27. void vp9_enable_segfeature(struct segmentation *seg, int segment_id,
  28. SEG_LVL_FEATURES feature_id) {
  29. seg->feature_mask[segment_id] |= 1 << feature_id;
  30. }
  31. int vp9_seg_feature_data_max(SEG_LVL_FEATURES feature_id) {
  32. return seg_feature_data_max[feature_id];
  33. }
  34. int vp9_is_segfeature_signed(SEG_LVL_FEATURES feature_id) {
  35. return seg_feature_data_signed[feature_id];
  36. }
  37. void vp9_set_segdata(struct segmentation *seg, int segment_id,
  38. SEG_LVL_FEATURES feature_id, int seg_data) {
  39. assert(seg_data <= seg_feature_data_max[feature_id]);
  40. if (seg_data < 0) {
  41. assert(seg_feature_data_signed[feature_id]);
  42. assert(-seg_data <= seg_feature_data_max[feature_id]);
  43. }
  44. seg->feature_data[segment_id][feature_id] = seg_data;
  45. }
  46. const vpx_tree_index vp9_segment_tree[TREE_SIZE(MAX_SEGMENTS)] = {
  47. 2, 4, 6, 8, 10, 12,
  48. 0, -1, -2, -3, -4, -5, -6, -7
  49. };
  50. // TBD? Functions to read and write segment data with range / validity checking