rangeencoder.h

来自「对浮点型数据进行压缩」· C头文件 代码 · 共 93 行

/*
===============================================================================

  FILE:  rangeencoder.h
  
  CONTENTS:
      
  PROGRAMMERS:
  
    martin [e-mail]
  
  COPYRIGHT:
  
    Copyright (C) 2003 Martin Isenburg ([e-mail])
    
    This software 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.
  
  CHANGE HISTORY:
  
    28 June 2004 -- added an option for NOT storing the code characters at all 
    14 January 2003 -- adapted from michael schindler's code before SIGGRAPH
  
===============================================================================
*/
#ifndef RANGEENCODER_H
#define RANGEENCODER_H

#include <stdio.h>

#include <iostream>

using namespace std;


#include "rangemodel.h"

class RangeEncoder
{
public:

/* Start the encoder                                         */
  RangeEncoder(FILE* fp, bool store_chars = true);
  ~RangeEncoder();

/* Encode with modelling                                     */
  void encode(RangeModel* rm, unsigned int sym);

/* Encode a range without modelling                          */
  void encode(unsigned int range, unsigned int sym);

/* Encode an unsigned char without modelling                 */
  void encodeByte(unsigned char b);

/* Encode an unsigned short without modelling                */
  void encodeShort(unsigned short s);

/* Encode an unsigned int without modelling                  */
  void encodeInt(unsigned int i);

/* Encode a float without modelling                          */
  void encodeFloat(float f);

/* Finish encoding, returns number of bytes written          */
  unsigned int done();

  unsigned char* getChars();
  int getNumberChars();

  long getNumberBits();
  int getNumberBytes();

private:
  inline void normalize();
  inline void outbyte(unsigned int byte);

  FILE* fp;

  unsigned char* chars;
  int number_chars;
  int allocated_chars;

  unsigned int low;           /* low end of interval */
  unsigned int range;         /* length of interval */
  unsigned int help;          /* bytes_to_follow resp. intermediate value */
  unsigned char buffer;       /* buffer for input/output */
  /* the following is used only when encoding */
  unsigned int bytecount;     /* counter for outputed bytes  */
};

#endif