rcencoder.h

#ifndef RC_ENCODER_H
#define RC_ENCODER_H

#include "types.h"
#include "rcmodel.h"

class RCencoder {
public:
  RCencoder() : error(false), low(0), range(-1u) {}
  virtual ~RCencoder() {}

  // finish encoding
  void finish();

  // encode a bit s
  void encode(bool s);

  // encode a number s : 0 <= s < 2^n
  template <typename UINT>
  void encode(UINT s, uint n);

  // encode a number s : l <= s < h
  template <typename UINT>
  void encode(UINT s, UINT l, UINT h);

  // encode a symbol s using probability modeling
  void encode(uint s, RCmodel* rm);

  // virtual function for writing byte stream
  virtual void putbyte(uint byte) = 0;

  // flush out any buffered bytes
  virtual void flush() {}

  // number of bytes written
  virtual size_t bytes() const = 0;

  bool error;

private:
  void encode_shift(uint s, uint n);
  void encode_ratio(uint s, uint n);
  void put(uint n);
  void normalize();

  uint low;   // low end of interval
  uint range; // length of interval
};

#include "rcencoder.inl"

#endif