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 #include <string.h> 13 #include "aom_dsp/bitwriter.h" 14 aom_start_encode(aom_writer * w,uint8_t * source)15void aom_start_encode(aom_writer *w, uint8_t *source) { 16 w->buffer = source; 17 w->pos = 0; 18 od_ec_enc_init(&w->ec, 62025); 19 } 20 aom_stop_encode(aom_writer * w)21int aom_stop_encode(aom_writer *w) { 22 int nb_bits; 23 uint32_t bytes; 24 unsigned char *data; 25 data = od_ec_enc_done(&w->ec, &bytes); 26 if (!data) { 27 od_ec_enc_clear(&w->ec); 28 return -1; 29 } 30 nb_bits = od_ec_enc_tell(&w->ec); 31 memcpy(w->buffer, data, bytes); 32 w->pos = bytes; 33 od_ec_enc_clear(&w->ec); 34 return nb_bits; 35 } 36 aom_tell_size(aom_writer * w)37int aom_tell_size(aom_writer *w) { 38 const int nb_bits = od_ec_enc_tell(&w->ec); 39 return nb_bits; 40 } 41