| /* |
| * Copyright (c) 2016, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 2 Clause License and |
| * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License |
| * was not distributed with this source code in the LICENSE file, you can |
| * obtain it at www.aomedia.org/license/software. If the Alliance for Open |
| * Media Patent License 1.0 was not distributed with this source code in the |
| * PATENTS file, you can obtain it at www.aomedia.org/license/patent. |
| */ |
| |
| #ifndef AOM_AOM_DSP_BITWRITER_H_ |
| #define AOM_AOM_DSP_BITWRITER_H_ |
| |
| #include <assert.h> |
| |
| #include "config/aom_config.h" |
| |
| #include "aom_dsp/daalaboolwriter.h" |
| #include "aom_dsp/prob.h" |
| |
| #if CONFIG_RD_DEBUG |
| #include "av1/common/blockd.h" |
| #include "av1/encoder/cost.h" |
| #endif |
| #if CONFIG_INTRA_ENTROPY |
| #include "av1/common/entropymode.h" |
| #endif // CONFIG_INTRA_ENTROPY |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| typedef struct daala_writer aom_writer; |
| |
| typedef struct TOKEN_STATS { |
| int cost; |
| #if CONFIG_RD_DEBUG |
| int txb_coeff_cost_map[TXB_COEFF_COST_MAP_SIZE][TXB_COEFF_COST_MAP_SIZE]; |
| #endif |
| } TOKEN_STATS; |
| |
| static INLINE void init_token_stats(TOKEN_STATS *token_stats) { |
| #if CONFIG_RD_DEBUG |
| int r, c; |
| for (r = 0; r < TXB_COEFF_COST_MAP_SIZE; ++r) { |
| for (c = 0; c < TXB_COEFF_COST_MAP_SIZE; ++c) { |
| token_stats->txb_coeff_cost_map[r][c] = 0; |
| } |
| } |
| #endif |
| token_stats->cost = 0; |
| } |
| |
| static INLINE void aom_start_encode(aom_writer *bc, uint8_t *buffer) { |
| aom_daala_start_encode(bc, buffer); |
| } |
| |
| static INLINE int aom_stop_encode(aom_writer *bc) { |
| return aom_daala_stop_encode(bc); |
| } |
| |
| static INLINE void aom_write(aom_writer *br, int bit, int probability) { |
| aom_daala_write(br, bit, probability); |
| } |
| |
| static INLINE void aom_write_bit(aom_writer *w, int bit) { |
| aom_write(w, bit, 128); // aom_prob_half |
| } |
| |
| static INLINE void aom_write_literal(aom_writer *w, int data, int bits) { |
| int bit; |
| |
| for (bit = bits - 1; bit >= 0; bit--) aom_write_bit(w, 1 & (data >> bit)); |
| } |
| |
| static INLINE void aom_write_cdf(aom_writer *w, int symb, |
| const aom_cdf_prob *cdf, int nsymbs) { |
| daala_write_symbol(w, symb, cdf, nsymbs); |
| } |
| |
| static INLINE void aom_write_symbol(aom_writer *w, int symb, aom_cdf_prob *cdf, |
| int nsymbs) { |
| aom_write_cdf(w, symb, cdf, nsymbs); |
| if (w->allow_update_cdf) update_cdf(cdf, symb, nsymbs); |
| } |
| |
| #if CONFIG_INTRA_ENTROPY |
| // For neural network |
| static INLINE void aom_write_symbol_nn(aom_writer *w, int symb, |
| aom_cdf_prob *cdf, |
| NN_CONFIG_EM *nn_model, int nsymbs) { |
| aom_write_cdf(w, symb, cdf, nsymbs); |
| if (w->allow_update_cdf) { |
| av1_nn_backprop_em(nn_model, symb); |
| av1_nn_update_em(nn_model, nn_model->lr); |
| } |
| } |
| #endif // CONFIG_INTRA_ENTROPY |
| |
| #ifdef __cplusplus |
| } // extern "C" |
| #endif |
| |
| #endif // AOM_AOM_DSP_BITWRITER_H_ |