summaryrefslogtreecommitdiffstats
path: root/indexlib/bitstream.h
blob: 5533d262c58613a49f1156dc0298f22f5ad8f96c (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
#ifndef LPC_BITSTREAM_H1102530057_INCLUDE_GUARD_
#define LPC_BITSTREAM_H1102530057_INCLUDE_GUARD_
/* This file is part of indexlib.
 * Copyright (C) 2005 Luís Pedro Coelho <luis@luispedro.org>
 *
 * Indexlib is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License, version 2, as
 * published by the Free Software Foundation and available as file
 * GPL_V2 which is distributed along with indexlib.
 * 
 * Indexlib is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301, USA
 * 
 * In addition, as a special exception, the copyright holders give
 * permission to link the code of this program with any edition of
 * the TQt library by Trolltech AS, Norway (or with modified versions
 * of TQt that use the same license as TQt), and distribute linked
 * combinations including the two.  You must obey the GNU General
 * Public License in all respects for all of the code used other than
 * TQt.  If you modify this file, you may extend this exception to
 * your version of the file, but you are not obligated to do so.  If
 * you do not wish to do so, delete this exception statement from
 * your version.
 */
#include <vector>
#include <inttypes.h>

struct bitstream {
	public:
		bitstream( const unsigned char* data, unsigned size );
		bitstream( unsigned char* data, unsigned size );

		bitstream& operator << ( uint16_t );
		bitstream& operator << ( uint32_t );

		bitstream& operator >> ( uint16_t& );
		bitstream& operator >> ( uint32_t& );

		void putback( uint16_t );
		void putback( uint32_t );

		unsigned nbytes() const { return size_ / 8 + bool( size_ % 8 ); }
		unsigned nbits() const { return size_; }
		const unsigned char* as_byte_vector() const { return &bytes_[ 0 ]; }
	private:
		bool getbit();
		unsigned char* bytes_;
		const bool writeable_;
		const unsigned size_;
		unsigned cur_;
		void putbit( bool );
};

template <typename T>
	unsigned bits_for();

#endif /* LPC_BITSTREAM_H1102530057_INCLUDE_GUARD_ */