Line data    Source code

       1             : /*
       2             :  * Copyright (c) 2016, Alliance for Open Media. All rights reserved
       3             :  *
       4             :  * This source code is subject to the terms of the BSD 2 Clause License and
       5             :  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
       6             :  * was not distributed with this source code in the LICENSE file, you can
       7             :  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
       8             :  * Media Patent License 1.0 was not distributed with this source code in the
       9             :  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
      10             :  */
      11             : 
      12             : #ifndef AV1_ENCODER_TREEWRITER_H_
      13             : #define AV1_ENCODER_TREEWRITER_H_
      14             : 
      15             : #include "aom_dsp/bitwriter.h"
      16             : 
      17             : #ifdef __cplusplus
      18             : extern "C" {
      19             : #endif
      20             : 
      21             : void av1_tree_probs_from_distribution(aom_tree tree,
      22             :                                       unsigned int branch_ct[/* n - 1 */][2],
      23             :                                       const unsigned int num_events[/* n */]);
      24             : 
      25             : struct av1_token {
      26             :   int value;
      27             :   int len;
      28             : };
      29             : 
      30             : void av1_tokens_from_tree(struct av1_token *, const aom_tree_index *);
      31             : 
      32           0 : static INLINE void av1_write_token(aom_writer *w, const aom_tree_index *tree,
      33             :                                    const aom_prob *probs,
      34             :                                    const struct av1_token *token) {
      35           0 :   aom_write_tree(w, tree, probs, token->value, token->len, 0);
      36           0 : }
      37             : 
      38             : #ifdef __cplusplus
      39             : }  // extern "C"
      40             : #endif
      41             : 
      42             : #endif  // AV1_ENCODER_TREEWRITER_H_