bitstreams.hpp

"More for C++" is a class library that provides some features that are usually common for object ori

//
//  This file is part of the "More for C++" library
//
//  Copyright (c) 1999-2003 by Thorsten Goertz ([e-mail])
//
//  The "More for C++" library is free software; you can redistribute it and/or
//  modify it under the terms of the license that comes with this package.
//
//  Read "license.txt" for more details.
//
//  THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED
//  WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED WARRANTIES
//  OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.

////////////////////////////////////////////////////////////////////////////////

#ifndef MORE_UTIL_BITSTREAMS_HPP
#define MORE_UTIL_BITSTREAMS_HPP

////////////////////////////////////////////////////////////////////////////////

#include <more/array.hpp>
#include <more/numbers.hpp>

////////////////////////////////////////////////////////////////////////////////

namespace more
{
  namespace util
  {
    class BitInputStream
    {
      public:
        typedef more::unsigned8 Byte;
        typedef more::unsigned32 Bits;

        BitInputStream( const Array<Byte>& bitBuffer, size_t nNoOfBits );
        size_t getNoOfBits( ) const;
        Bits getBits( size_t nNoOfBits ) const;
        void forgetBits( size_t nNoOfBits );

      private:
        void loadAccumulator( );

        Array<Byte> m_bitBuffer;
        size_t      m_nNoOfBitsLeft;
        Bits        m_nBitAccumulator;
        size_t      m_nNoOfLoadedBits;
    };

    class BitOutputStream
    {
      public:
        typedef more::unsigned8 Byte;
        typedef more::unsigned32 Bits;

        BitOutputStream( const Array<Byte>& bitBuffer, size_t nMaxNoOfBits );
        size_t getNoOfBits( ) const;
        void setBits( Bits nBits, size_t nNoOfBits, bool bBigEndianOrder = true );

      private:
        void setBitsInBigEndianOrder( Bits nBits, size_t nNoOfBits );
        void setBitsInLittleEndianOrder( Bits nBits, size_t nNoOfBits );

        Array<Byte> m_bitBuffer;
        size_t      m_nMaxNoOfBits;
        size_t      m_nNoOfBitsSet;
    };
  }
}

////////////////////////////////////////////////////////////////////////////////

#endif