modules/highgui/src/bitstrm.hpp

   1 /*M///////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
   4 //
   5 //  By downloading, copying, installing or using the software you agree to this license.
   6 //  If you do not agree to this license, do not download, install,
   7 //  copy or use the software.
   8 //
   9 //
  10 //                           License Agreement
  11 //                For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
  14 // Copyright (C) 2009, Willow Garage Inc., all rights reserved.
  15 // Third party copyrights are property of their respective owners.
  16 //
  17 // Redistribution and use in source and binary forms, with or without modification,
  18 // are permitted provided that the following conditions are met:
  19 //
  20 //   * Redistribution's of source code must retain the above copyright notice,
  21 //     this list of conditions and the following disclaimer.
  22 //
  23 //   * Redistribution's in binary form must reproduce the above copyright notice,
  24 //     this list of conditions and the following disclaimer in the documentation
  25 //     and/or other materials provided with the distribution.
  26 //
  27 //   * The name of the copyright holders may not be used to endorse or promote products
  28 //     derived from this software without specific prior written permission.
  29 //
  30 // This software is provided by the copyright holders and contributors "as is" and
  31 // any express or implied warranties, including, but not limited to, the implied
  32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  33 // In no event shall the Intel Corporation or contributors be liable for any direct,
  34 // indirect, incidental, special, exemplary, or consequential damages
  35 // (including, but not limited to, procurement of substitute goods or services;
  36 // loss of use, data, or profits; or business interruption) however caused
  37 // and on any theory of liability, whether in contract, strict liability,
  38 // or tort (including negligence or otherwise) arising in any way out of
  39 // the use of this software, even if advised of the possibility of such damage.
  40 //
  41 //M*/
  42
  43 #ifndef _BITSTRM_H_
  44 #define _BITSTRM_H_
  45
  46 #include <stdio.h>
  47
  48 namespace cv
  49 {
  50
  51 #define DECLARE_RBS_EXCEPTION(name) \
  52 class RBS_ ## name ## _Exception : public cv::Exception \
  53 { \
  54 public: \
  55     RBS_ ## name ## _Exception(int code_, const String& err_, const String& func_, const String& file_, int line_) : \
  56         cv::Exception(code_, err_, func_, file_, line_) \
  57     {} \
  58 };
  59 DECLARE_RBS_EXCEPTION(THROW_EOS)
  60 #define RBS_THROW_EOS RBS_THROW_EOS_Exception(CV_StsError, "Unexpected end of input stream", CV_Func, __FILE__, __LINE__)
  61 DECLARE_RBS_EXCEPTION(THROW_FORB)
  62 #define RBS_THROW_FORB RBS_THROW_FORB_Exception(CV_StsError, "Forrbidden huffman code", CV_Func, __FILE__, __LINE__)
  63 DECLARE_RBS_EXCEPTION(BAD_HEADER)
  64 #define RBS_BAD_HEADER RBS_BAD_HEADER_Exception(CV_StsError, "Invalid header", CV_Func, __FILE__, __LINE__)
  65
  66 typedef unsigned long ulong;
  67
  68 // class RBaseStream - base class for other reading streams.
  69 class RBaseStream
  70 {
  71 public:
  72     //methods
  73     RBaseStream();
  74     virtual ~RBaseStream();
  75
  76     virtual bool  open( const string& filename );
  77     virtual bool  open( const Mat& buf );
  78     virtual void  close();
  79     bool          isOpened();
  80     void          setPos( int pos );
  81     int           getPos();
  82     void          skip( int bytes );
  83
  84 protected:
  85
  86     bool    m_allocated;
  87     uchar*  m_start;
  88     uchar*  m_end;
  89     uchar*  m_current;
  90     FILE*   m_file;
  91     int     m_block_size;
  92     int     m_block_pos;
  93     bool    m_is_opened;
  94
  95     virtual void  readBlock();
  96     virtual void  release();
  97     virtual void  allocate();
  98 };
  99
 100
 101 // class RLByteStream - uchar-oriented stream.
 102 // l in prefix means that the least significant uchar of a multi-uchar value goes first
 103 class RLByteStream : public RBaseStream
 104 {
 105 public:
 106     virtual ~RLByteStream();
 107
 108     int     getByte();
 109     int     getBytes( void* buffer, int count );
 110     int     getWord();
 111     int     getDWord();
 112 };
 113
 114 // class RMBitStream - uchar-oriented stream.
 115 // m in prefix means that the most significant uchar of a multi-uchar value go first
 116 class RMByteStream : public RLByteStream
 117 {
 118 public:
 119     virtual ~RMByteStream();
 120
 121     int     getWord();
 122     int     getDWord();
 123 };
 124
 125 // WBaseStream - base class for output streams
 126 class WBaseStream
 127 {
 128 public:
 129     //methods
 130     WBaseStream();
 131     virtual ~WBaseStream();
 132
 133     virtual bool  open( const string& filename );
 134     virtual bool  open( vector<uchar>& buf );
 135     virtual void  close();
 136     bool          isOpened();
 137     int           getPos();
 138
 139 protected:
 140
 141     uchar*  m_start;
 142     uchar*  m_end;
 143     uchar*  m_current;
 144     int     m_block_size;
 145     int     m_block_pos;
 146     FILE*   m_file;
 147     bool    m_is_opened;
 148     vector<uchar>* m_buf;
 149
 150     virtual void  writeBlock();
 151     virtual void  release();
 152     virtual void  allocate();
 153 };
 154
 155
 156 // class WLByteStream - uchar-oriented stream.
 157 // l in prefix means that the least significant uchar of a multi-byte value goes first
 158 class WLByteStream : public WBaseStream
 159 {
 160 public:
 161     virtual ~WLByteStream();
 162
 163     void  putByte( int val );
 164     void  putBytes( const void* buffer, int count );
 165     void  putWord( int val );
 166     void  putDWord( int val );
 167 };
 168
 169
 170 // class WLByteStream - uchar-oriented stream.
 171 // m in prefix means that the least significant uchar of a multi-byte value goes last
 172 class WMByteStream : public WLByteStream
 173 {
 174 public:
 175     virtual ~WMByteStream();
 176     void  putWord( int val );
 177     void  putDWord( int val );
 178 };
 179
 180 inline unsigned BSWAP(unsigned v)
 181 {
 182     return (v<<24)|((v&0xff00)<<8)|((v>>8)&0xff00)|((unsigned)v>>24);
 183 }
 184
 185 bool bsIsBigEndian( void );
 186
 187 }
 188
 189 #endif/*_BITSTRM_H_*/