vp10/encoder/cost.h

   1 /*
   2  *  Copyright (c) 2014 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 #ifndef VP9_ENCODER_VP9_COST_H_
  12 #define VP9_ENCODER_VP9_COST_H_
  13
  14 #include "vpx_dsp/prob.h"
  15
  16 #ifdef __cplusplus
  17 extern "C" {
  18 #endif
  19
  20 extern const unsigned int vp10_prob_cost[256];
  21
  22 #define vp10_cost_zero(prob) (vp10_prob_cost[prob])
  23
  24 #define vp10_cost_one(prob) vp10_cost_zero(vpx_complement(prob))
  25
  26 #define vp10_cost_bit(prob, bit) vp10_cost_zero((bit) ? vpx_complement(prob) \
  27                                                     : (prob))
  28
  29 static INLINE unsigned int cost_branch256(const unsigned int ct[2],
  30                                           vpx_prob p) {
  31   return ct[0] * vp10_cost_zero(p) + ct[1] * vp10_cost_one(p);
  32 }
  33
  34 static INLINE int treed_cost(vpx_tree tree, const vpx_prob *probs,
  35                              int bits, int len) {
  36   int cost = 0;
  37   vpx_tree_index i = 0;
  38
  39   do {
  40     const int bit = (bits >> --len) & 1;
  41     cost += vp10_cost_bit(probs[i >> 1], bit);
  42     i = tree[i + bit];
  43   } while (len);
  44
  45   return cost;
  46 }
  47
  48 void vp10_cost_tokens(int *costs, const vpx_prob *probs, vpx_tree tree);
  49 void vp10_cost_tokens_skip(int *costs, const vpx_prob *probs, vpx_tree tree);
  50
  51 #ifdef __cplusplus
  52 }  // extern "C"
  53 #endif
  54
  55 #endif  // VP9_ENCODER_VP9_COST_H_