vp9_seg_common.c 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  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. int vp9_segfeature_active(const struct segmentation *seg, int segment_id,
  23. SEG_LVL_FEATURES feature_id) {
  24. return seg->enabled &&
  25. (seg->feature_mask[segment_id] & (1 << feature_id));
  26. }
  27. void vp9_clearall_segfeatures(struct segmentation *seg) {
  28. vp9_zero(seg->feature_data);
  29. vp9_zero(seg->feature_mask);
  30. }
  31. void vp9_enable_segfeature(struct segmentation *seg, int segment_id,
  32. SEG_LVL_FEATURES feature_id) {
  33. seg->feature_mask[segment_id] |= 1 << feature_id;
  34. }
  35. int vp9_seg_feature_data_max(SEG_LVL_FEATURES feature_id) {
  36. return seg_feature_data_max[feature_id];
  37. }
  38. int vp9_is_segfeature_signed(SEG_LVL_FEATURES feature_id) {
  39. return seg_feature_data_signed[feature_id];
  40. }
  41. void vp9_set_segdata(struct segmentation *seg, int segment_id,
  42. SEG_LVL_FEATURES feature_id, int seg_data) {
  43. assert(seg_data <= seg_feature_data_max[feature_id]);
  44. if (seg_data < 0) {
  45. assert(seg_feature_data_signed[feature_id]);
  46. assert(-seg_data <= seg_feature_data_max[feature_id]);
  47. }
  48. seg->feature_data[segment_id][feature_id] = seg_data;
  49. }
  50. int vp9_get_segdata(const struct segmentation *seg, int segment_id,
  51. SEG_LVL_FEATURES feature_id) {
  52. return seg->feature_data[segment_id][feature_id];
  53. }
  54. const vp9_tree_index vp9_segment_tree[TREE_SIZE(MAX_SEGMENTS)] = {
  55. 2, 4, 6, 8, 10, 12,
  56. 0, -1, -2, -3, -4, -5, -6, -7
  57. };
  58. // TBD? Functions to read and write segment data with range / validity checking