You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdepim/indexlib/bitstream.cpp

109 lines
2.9 KiB

/* 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 "bitstream.h"
#include "logfile.h"
#include "format.h"
#include <assert.h>
bitstream::bitstream( unsigned char* data, unsigned size )
:bytes_( data ),
writeable_( true ),
size_( size ),
cur_( 0 ) { }
bitstream::bitstream( const unsigned char* data, unsigned size )
:bytes_( const_cast<unsigned char*>( data ) ),
writeable_( false ),
size_( size ),
cur_( 0 ) { }
bool bit( unsigned v, unsigned idx ) {
return ( v >> idx ) & 1;
}
bitstream& bitstream::operator << ( uint16_t x ) {
for ( unsigned i = 0; i != 16; ++i ) putbit( bit( x, i ) );
return *this;
}
bitstream& bitstream::operator << ( uint32_t x ) {
for ( unsigned i = 0; i != 32; ++i ) putbit( bit( x, i ) );
return *this;
}
bitstream& bitstream::operator >> ( uint16_t& v ) {
v = 0;
for ( int i = 0; i != 16; ++i ) {
v <<= 1;
v |= getbit();
}
return *this;
}
bitstream& bitstream::operator >> ( uint32_t& v ) {
v = 0;
for ( int i = 0; i != 32; ++i ) {
v <<= 1;
v |= getbit();
}
return *this;
}
void bitstream::putback( uint16_t ) {
assert( cur_ >= 16 );
cur_ -= 16;
}
void bitstream::putback( uint32_t ) {
assert( cur_ >= 32 );
cur_ -= 32;
}
bool bitstream::getbit() {
unsigned inbyte = cur_ % 8;
unsigned byte = cur_ / 8;
++cur_;
return ( bytes_[ byte ] >> inbyte ) & 1;
}
void bitstream::putbit( bool value ) {
unsigned inbyte = cur_ % 8;
unsigned byte = cur_ / 8;
assert( byte < size_ );
if ( value ) bytes_[ byte ] |= ( 1 << inbyte );
else bytes_[ byte ] &= ~( 1 << inbyte );
++cur_;
}