treewriter.c 954 B

123456789101112131415161718192021222324252627282930313233
  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 "treewriter.h"
  11. static void cost(int *const C, vp8_tree T, const vp8_prob *const P, int i,
  12. int c) {
  13. const vp8_prob p = P[i >> 1];
  14. do {
  15. const vp8_tree_index j = T[i];
  16. const int d = c + vp8_cost_bit(p, i & 1);
  17. if (j <= 0) {
  18. C[-j] = d;
  19. } else {
  20. cost(C, T, P, j, d);
  21. }
  22. } while (++i & 1);
  23. }
  24. void vp8_cost_tokens(int *c, const vp8_prob *p, vp8_tree t) {
  25. cost(c, t, p, 0, 0);
  26. }
  27. void vp8_cost_tokens2(int *c, const vp8_prob *p, vp8_tree t, int start) {
  28. cost(c, t, p, start, 0);
  29. }