aboutsummaryrefslogtreecommitdiffstats
path: root/libtoolame-dab/bitstream.c
blob: 28e64082a04c726ce51ff84d64a11c0b3307709b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "common.h"
#include "mem.h"
#include "bitstream.h"

/*****************************************************************************
 *
 *  bit_stream.c package
 *  Author:  Jean-Georges Fritsch, C-Cube Microsystems
 *           Matthias P. Braendli, www.opendigitalradio.org
 *  Changes
 *       Apr 2000 - removed all the file input routines. MFC
 *       Feb 2016 - removed all sort of things to make Toolame a library. mpb
 *****************************************************************************/

/********************************************************************
  This package provides functions to write (exclusive or read)
  information from (exclusive or to) the bit stream.

  If the bit stream is opened in read mode only the get functions are
  available. If the bit stream is opened in write mode only the put
  functions are available.
 ********************************************************************/

/*open_bit_stream_w(); open the device to write the bit stream into it    */
/*close_bit_stream();  close the device containing the bit stream         */
/*alloc_buffer();      open and initialize the buffer;                    */
/*desalloc_buffer();   empty and close the buffer                         */
/*put1bit(); write 1 bit from the bit stream  */
/*put1bit(); write 1 bit from the bit stream  */
/*putbits(); write N bits from the bit stream */

/* You must have one frame in memory if you are in DAB mode                 */
/* in conformity of the norme ETS 300 401 http://www.etsi.org               */
/* see toollame.c                                                           */
int minimum = MINIMUM;

void bs_set_minimum(int min)
{
    minimum = min;
}

/* empty the buffer to the output device when the buffer becomes full */
void empty_buffer (Bit_stream_struc * bs, int minimum)
{
    int j = 0;
    if (bs->output_buffer_written != 0) {
        fprintf(stderr, "ERROR: libtoolame output buffer was not emptied\n");
    }

    for (int i = bs->buf_size - 1; i >= minimum; i--) {
        if (j >= bs->output_buffer_size) {
            fprintf(stderr, "ERROR: libtoolame output buffer too small (%d vs %d)!\n",
                    bs->output_buffer_size, bs->buf_size - minimum);
            break;
        }

        bs->output_buffer[j] = bs->buf[i];
        j++;
    }
    bs->output_buffer_written = j;

    for (int i = minimum - 1; i >= 0; i--) {
        bs->buf[bs->buf_size - minimum + i] = bs->buf[i];
    }

    bs->buf_byte_idx = bs->buf_size - 1 - minimum;
    bs->buf_bit_idx = 8;
}


/* open the device to write the bit stream into it */
void open_bit_stream_w (Bit_stream_struc * bs, int size)
{
    alloc_buffer (bs, size);
    bs->buf_byte_idx = size - 1;
    bs->buf_bit_idx = 8;
    bs->totbit = 0;
    bs->mode = WRITE_MODE;
    bs->eob = FALSE;
    bs->eobs = FALSE;
}

/*close the device containing the bit stream after a write process*/
void close_bit_stream_w (Bit_stream_struc * bs)
{
    putbits (bs, 0, 7);
    empty_buffer (bs, bs->buf_byte_idx + 1);
    desalloc_buffer (bs);
}

/*open and initialize the buffer; */
void alloc_buffer (Bit_stream_struc * bs, int size)
{
    bs->buf =
        (unsigned char *) mem_alloc (size * sizeof (unsigned char), "buffer");
    bs->buf_size = size;
}

/*empty and close the buffer */
void desalloc_buffer (Bit_stream_struc * bs)
{
    free (bs->buf);
}

const int putmask[9] = { 0x0, 0x1, 0x3, 0x7, 0xf, 0x1f, 0x3f, 0x7f, 0xff };

/*write 1 bit from the bit stream */
void put1bit (Bit_stream_struc * bs, int bit)
{
    bs->totbit++;

    bs->buf[bs->buf_byte_idx] |= (bit & 0x1) << (bs->buf_bit_idx - 1);
    bs->buf_bit_idx--;
    if (!bs->buf_bit_idx) {
        bs->buf_bit_idx = 8;
        bs->buf_byte_idx--;
        if (bs->buf_byte_idx < 0)
            empty_buffer (bs, minimum);
        bs->buf[bs->buf_byte_idx] = 0;
    }
}

/*write N bits into the bit stream */
void putbits (Bit_stream_struc * bs, unsigned int val, int N)
{
    register int j = N;
    register int k, tmp;

    /* if (N > MAX_LENGTH)
       fprintf(stderr, "Cannot read or write more than %d bits at a time.\n", MAX_LENGTH); ignore check!! MFC Apr 00 */

    bs->totbit += N;
    while (j > 0) {
        k = MIN (j, bs->buf_bit_idx);
        tmp = val >> (j - k);
        bs->buf[bs->buf_byte_idx] |= (tmp & putmask[k]) << (bs->buf_bit_idx - k);
        bs->buf_bit_idx -= k;
        if (!bs->buf_bit_idx) {
            bs->buf_bit_idx = 8;
            bs->buf_byte_idx--;
            if (bs->buf_byte_idx < 0)
                empty_buffer (bs, minimum);
            bs->buf[bs->buf_byte_idx] = 0;
        }
        j -= k;
    }
}