bitstream.h

Go to the documentation of this file.

/*
  Copyright (c) Mircea Neacsu (2014-2025) Licensed under MIT License.
  This file is part of MLIB project. See LICENSE file for full license terms.
*/

 
#pragma once
#include <iostream>
 
#if __has_include("defs.h")
#include "defs.h"
#endif
 
namespace mlib {

class bitstream
{
public:
  bitstream (std::iostream& str, unsigned int pack);

  bool read ();

  int read (size_t sz, bool is_signed = false);

  void write (bool val);

  void write (int val, size_t sz);
 
  void flush ();
 
protected:
  const unsigned int packing; 
  std::iostream& s;           

  virtual char encode (unsigned char bits);

  virtual unsigned char decode (char chr);
 
private:
  unsigned char buffer;
  int nbits;
};

inline 
char bitstream::encode (unsigned char bits)
{
  return (bits & (1 << packing) - 1);
}

inline unsigned char bitstream::decode (char chr)
{
  return (chr & (1 << packing) - 1);
}
 
}; // namespace mlib