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//                        Intel License Agreement
11//                For Open Source Computer Vision Library
12//
13// Copyright (C) 2000, Intel Corporation, all rights reserved.
14// Third party copyrights are property of their respective owners.
15//
16// Redistribution and use in source and binary forms, with or without modification,
17// are permitted provided that the following conditions are met:
18//
19//   * Redistribution's of source code must retain the above copyright notice,
20//     this list of conditions and the following disclaimer.
21//
22//   * Redistribution's in binary form must reproduce the above copyright notice,
23//     this list of conditions and the following disclaimer in the documentation
24//     and/or other materials provided with the distribution.
25//
26//   * The name of Intel Corporation may not be used to endorse or promote products
27//     derived from this software without specific prior written permission.
28//
29// This software is provided by the copyright holders and contributors "as is" and
30// any express or implied warranties, including, but not limited to, the implied
31// warranties of merchantability and fitness for a particular purpose are disclaimed.
32// In no event shall the Intel Corporation or contributors be liable for any direct,
33// indirect, incidental, special, exemplary, or consequential damages
34// (including, but not limited to, procurement of substitute goods or services;
35// loss of use, data, or profits; or business interruption) however caused
36// and on any theory of liability, whether in contract, strict liability,
37// or tort (including negligence or otherwise) arising in any way out of
38// the use of this software, even if advised of the possibility of such damage.
39//
40//M*/
41
42#ifndef _GRFMT_JPEG_H_
43#define _GRFMT_JPEG_H_
44
45#include "grfmt_base.h"
46#include "bitstrm.h"
47
48#ifdef HAVE_JPEG
49
50/* IJG-based version */
51
52class GrFmtJpegReader : public GrFmtReader
53{
54public:
55
56    GrFmtJpegReader( const char* filename );
57    ~GrFmtJpegReader();
58
59    bool  ReadData( uchar* data, int step, int color );
60    bool  ReadHeader();
61    void  Close();
62
63protected:
64
65    void* m_cinfo; // pointer to IJG JPEG codec structure
66    void* m_jerr; // pointer to error processing manager state
67    FILE* m_f;
68};
69
70
71class GrFmtJpegWriter : public GrFmtWriter
72{
73public:
74
75    GrFmtJpegWriter( const char* filename );
76    ~GrFmtJpegWriter();
77
78    bool  WriteImage( const uchar* data, int step,
79                      int width, int height, int depth, int channels );
80};
81
82#else
83
84/* hand-crafted implementation */
85
86class RJpegBitStream : public RMBitStream
87{
88public:
89    RMByteStream  m_low_strm;
90
91    RJpegBitStream();
92    ~RJpegBitStream();
93
94    virtual bool  Open( const char* filename );
95    virtual void  Close();
96
97    void  Flush(); // flushes high-level bit stream
98    void  AlignOnByte();
99    int   FindMarker();
100
101protected:
102    virtual void  ReadBlock();
103};
104
105
106//////////////////// JPEG reader /////////////////////
107
108class GrFmtJpegReader : public GrFmtReader
109{
110public:
111
112    GrFmtJpegReader( const char* filename );
113    ~GrFmtJpegReader();
114
115    bool  ReadData( uchar* data, int step, int color );
116    bool  ReadHeader();
117    void  Close();
118
119protected:
120
121    int   m_offset; // offset of first scan
122    int   m_version; // JFIF version
123    int   m_planes; // 3 (YCrCb) or 1 (Gray)
124    int   m_precision; // 8 or 12-bit per sample
125    int   m_type; // SOF type
126    int   m_MCUs; // MCUs in restart interval
127    int   m_ss, m_se, m_ah, m_al; // progressive JPEG parameters
128
129    // information about each component
130    struct cmp_info
131    {
132        char h;  // horizontal sampling factor
133        char v;  // vertical   sampling factor
134        char tq; // quantization table index
135        char td, ta; // DC & AC huffman tables
136        int  dc_pred; // DC predictor
137    };
138
139    cmp_info m_ci[3];
140
141    int     m_tq[4][64];
142    bool    m_is_tq[4];
143
144    short*  m_td[4];
145    bool    m_is_td[4];
146
147    short*  m_ta[4];
148    bool    m_is_ta[4];
149
150    RJpegBitStream  m_strm;
151
152protected:
153
154    bool  LoadQuantTables( int length );
155    bool  LoadHuffmanTables( int length );
156    void  ProcessScan( int* idx, int ns, uchar* data, int step, int color );
157    void  ResetDecoder();
158    void  GetBlock( int* block, int c );
159};
160
161
162//////////////////// JPEG-specific output bitstream ///////////////////////
163
164class WJpegBitStream : public WMBitStream
165{
166public:
167    WMByteStream  m_low_strm;
168
169    WJpegBitStream();
170    ~WJpegBitStream();
171
172    virtual void  Flush();
173    virtual bool  Open( const char* filename );
174    virtual void  Close();
175
176protected:
177    virtual void  WriteBlock();
178};
179
180
181//////////////////// JPEG reader /////////////////////
182
183class GrFmtJpegWriter : public GrFmtWriter
184{
185public:
186
187    GrFmtJpegWriter( const char* filename );
188    ~GrFmtJpegWriter();
189
190    bool  WriteImage( const uchar* data, int step,
191                      int width, int height, int depth, int channels );
192
193protected:
194
195    WJpegBitStream  m_strm;
196};
197
198#endif /* HAVE_JPEG */
199
200
201// JPEG filter factory
202class GrFmtJpeg : public GrFmtFilterFactory
203{
204public:
205
206    GrFmtJpeg();
207    ~GrFmtJpeg();
208
209    GrFmtReader* NewReader( const char* filename );
210    GrFmtWriter* NewWriter( const char* filename );
211};
212
213
214#endif/*_GRFMT_JPEG_H_*/
215