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 ();         
  void write (int val); 
  void flush ();

  int mread (unsigned int sz, bool is_signed = false);

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

  virtual void encode (unsigned char bits, char& chr);

  virtual void decode (unsigned char& bits, char chr);
 
private:
  unsigned char buffer;
  int nbits;
};

inline void bitstream::encode (unsigned char bits, char& chr)
{
  chr = (bits & (1 << packing) - 1);
}

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