| /*M/////////////////////////////////////////////////////////////////////////////////////// |
| // |
| // IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. |
| // |
| // By downloading, copying, installing or using the software you agree to this license. |
| // If you do not agree to this license, do not download, install, |
| // copy or use the software. |
| // |
| // |
| // Intel License Agreement |
| // For Open Source Computer Vision Library |
| // |
| // Copyright (C) 2000, Intel Corporation, all rights reserved. |
| // Third party copyrights are property of their respective owners. |
| // |
| // Redistribution and use in source and binary forms, with or without modification, |
| // are permitted provided that the following conditions are met: |
| // |
| // * Redistribution's of source code must retain the above copyright notice, |
| // this list of conditions and the following disclaimer. |
| // |
| // * Redistribution's in binary form must reproduce the above copyright notice, |
| // this list of conditions and the following disclaimer in the documentation |
| // and/or other materials provided with the distribution. |
| // |
| // * The name of Intel Corporation may not be used to endorse or promote products |
| // derived from this software without specific prior written permission. |
| // |
| // This software is provided by the copyright holders and contributors "as is" and |
| // any express or implied warranties, including, but not limited to, the implied |
| // warranties of merchantability and fitness for a particular purpose are disclaimed. |
| // In no event shall the Intel Corporation or contributors be liable for any direct, |
| // indirect, incidental, special, exemplary, or consequential damages |
| // (including, but not limited to, procurement of substitute goods or services; |
| // loss of use, data, or profits; or business interruption) however caused |
| // and on any theory of liability, whether in contract, strict liability, |
| // or tort (including negligence or otherwise) arising in any way out of |
| // the use of this software, even if advised of the possibility of such damage. |
| // |
| //M*/ |
| |
| #include "_highgui.h" |
| #include "grfmt_jpeg.h" |
| |
| // JPEG filter factory |
| |
| GrFmtJpeg::GrFmtJpeg() |
| { |
| m_sign_len = 3; |
| m_signature = "\xFF\xD8\xFF"; |
| m_description = "JPEG files (*.jpeg;*.jpg;*.jpe)"; |
| } |
| |
| |
| GrFmtJpeg::~GrFmtJpeg() |
| { |
| } |
| |
| |
| GrFmtReader* GrFmtJpeg::NewReader( const char* filename ) |
| { |
| return new GrFmtJpegReader( filename ); |
| } |
| |
| |
| GrFmtWriter* GrFmtJpeg::NewWriter( const char* filename ) |
| { |
| return new GrFmtJpegWriter( filename ); |
| } |
| |
| |
| #ifdef HAVE_JPEG |
| |
| /****************************************************************************************\ |
| This part of the file implements JPEG codec on base of IJG libjpeg library, |
| in particular, this is the modified example.doc from libjpeg package. |
| See otherlibs/_graphics/readme.txt for copyright notice. |
| \****************************************************************************************/ |
| |
| #include <stdio.h> |
| #include <setjmp.h> |
| |
| #ifdef WIN32 |
| |
| #define XMD_H // prevent redefinition of INT32 |
| #undef FAR // prevent FAR redefinition |
| |
| #endif |
| |
| #if defined WIN32 && defined __GNUC__ |
| typedef unsigned char boolean; |
| #endif |
| |
| extern "C" { |
| #include "jpeglib.h" |
| } |
| |
| /////////////////////// Error processing ///////////////////// |
| |
| typedef struct GrFmtJpegErrorMgr |
| { |
| struct jpeg_error_mgr pub; /* "parent" structure */ |
| jmp_buf setjmp_buffer; /* jump label */ |
| } |
| GrFmtJpegErrorMgr; |
| |
| |
| METHODDEF(void) |
| error_exit( j_common_ptr cinfo ) |
| { |
| GrFmtJpegErrorMgr* err_mgr = (GrFmtJpegErrorMgr*)(cinfo->err); |
| |
| /* Return control to the setjmp point */ |
| longjmp( err_mgr->setjmp_buffer, 1 ); |
| } |
| |
| |
| /////////////////////// GrFmtJpegReader /////////////////// |
| |
| |
| GrFmtJpegReader::GrFmtJpegReader( const char* filename ) : GrFmtReader( filename ) |
| { |
| m_cinfo = 0; |
| m_f = 0; |
| } |
| |
| |
| GrFmtJpegReader::~GrFmtJpegReader() |
| { |
| } |
| |
| |
| void GrFmtJpegReader::Close() |
| { |
| if( m_f ) |
| { |
| fclose( m_f ); |
| m_f = 0; |
| } |
| |
| if( m_cinfo ) |
| { |
| jpeg_decompress_struct* cinfo = (jpeg_decompress_struct*)m_cinfo; |
| GrFmtJpegErrorMgr* jerr = (GrFmtJpegErrorMgr*)m_jerr; |
| |
| jpeg_destroy_decompress( cinfo ); |
| delete cinfo; |
| delete jerr; |
| m_cinfo = 0; |
| m_jerr = 0; |
| } |
| GrFmtReader::Close(); |
| } |
| |
| |
| bool GrFmtJpegReader::ReadHeader() |
| { |
| bool result = false; |
| Close(); |
| |
| jpeg_decompress_struct* cinfo = new jpeg_decompress_struct; |
| GrFmtJpegErrorMgr* jerr = new GrFmtJpegErrorMgr; |
| |
| cinfo->err = jpeg_std_error(&jerr->pub); |
| jerr->pub.error_exit = error_exit; |
| |
| m_cinfo = cinfo; |
| m_jerr = jerr; |
| |
| if( setjmp( jerr->setjmp_buffer ) == 0 ) |
| { |
| jpeg_create_decompress( cinfo ); |
| |
| m_f = fopen( m_filename, "rb" ); |
| if( m_f ) |
| { |
| jpeg_stdio_src( cinfo, m_f ); |
| jpeg_read_header( cinfo, TRUE ); |
| |
| m_width = cinfo->image_width; |
| m_height = cinfo->image_height; |
| m_iscolor = cinfo->num_components > 1; |
| |
| result = true; |
| } |
| } |
| |
| if( !result ) |
| Close(); |
| |
| return result; |
| } |
| |
| /*************************************************************************** |
| * following code is for supporting MJPEG image files |
| * based on a message of Laurent Pinchart on the video4linux mailing list |
| ***************************************************************************/ |
| |
| /* JPEG DHT Segment for YCrCb omitted from MJPEG data */ |
| static |
| unsigned char my_jpeg_odml_dht[0x1a4] = { |
| 0xff, 0xc4, 0x01, 0xa2, |
| |
| 0x00, 0x00, 0x01, 0x05, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x00, 0x00, |
| 0x00, 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, |
| |
| 0x01, 0x00, 0x03, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, |
| 0x00, 0x00, 0x00, 0x00, 0x00, |
| 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, |
| |
| 0x10, 0x00, 0x02, 0x01, 0x03, 0x03, 0x02, 0x04, 0x03, 0x05, 0x05, 0x04, |
| 0x04, 0x00, 0x00, 0x01, 0x7d, |
| 0x01, 0x02, 0x03, 0x00, 0x04, 0x11, 0x05, 0x12, 0x21, 0x31, 0x41, 0x06, |
| 0x13, 0x51, 0x61, 0x07, |
| 0x22, 0x71, 0x14, 0x32, 0x81, 0x91, 0xa1, 0x08, 0x23, 0x42, 0xb1, 0xc1, |
| 0x15, 0x52, 0xd1, 0xf0, |
| 0x24, 0x33, 0x62, 0x72, 0x82, 0x09, 0x0a, 0x16, 0x17, 0x18, 0x19, 0x1a, |
| 0x25, 0x26, 0x27, 0x28, |
| 0x29, 0x2a, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x43, 0x44, 0x45, |
| 0x46, 0x47, 0x48, 0x49, |
| 0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x63, 0x64, 0x65, |
| 0x66, 0x67, 0x68, 0x69, |
| 0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79, 0x7a, 0x83, 0x84, 0x85, |
| 0x86, 0x87, 0x88, 0x89, |
| 0x8a, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, 0xa2, 0xa3, |
| 0xa4, 0xa5, 0xa6, 0xa7, |
| 0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, |
| 0xc2, 0xc3, 0xc4, 0xc5, |
| 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8, |
| 0xd9, 0xda, 0xe1, 0xe2, |
| 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xf1, 0xf2, 0xf3, 0xf4, |
| 0xf5, 0xf6, 0xf7, 0xf8, |
| 0xf9, 0xfa, |
| |
| 0x11, 0x00, 0x02, 0x01, 0x02, 0x04, 0x04, 0x03, 0x04, 0x07, 0x05, 0x04, |
| 0x04, 0x00, 0x01, 0x02, 0x77, |
| 0x00, 0x01, 0x02, 0x03, 0x11, 0x04, 0x05, 0x21, 0x31, 0x06, 0x12, 0x41, |
| 0x51, 0x07, 0x61, 0x71, |
| 0x13, 0x22, 0x32, 0x81, 0x08, 0x14, 0x42, 0x91, 0xa1, 0xb1, 0xc1, 0x09, |
| 0x23, 0x33, 0x52, 0xf0, |
| 0x15, 0x62, 0x72, 0xd1, 0x0a, 0x16, 0x24, 0x34, 0xe1, 0x25, 0xf1, 0x17, |
| 0x18, 0x19, 0x1a, 0x26, |
| 0x27, 0x28, 0x29, 0x2a, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x43, 0x44, |
| 0x45, 0x46, 0x47, 0x48, |
| 0x49, 0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x63, 0x64, |
| 0x65, 0x66, 0x67, 0x68, |
| 0x69, 0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79, 0x7a, 0x82, 0x83, |
| 0x84, 0x85, 0x86, 0x87, |
| 0x88, 0x89, 0x8a, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, |
| 0xa2, 0xa3, 0xa4, 0xa5, |
| 0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, 0xb7, 0xb8, |
| 0xb9, 0xba, 0xc2, 0xc3, |
| 0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, |
| 0xd7, 0xd8, 0xd9, 0xda, |
| 0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xf2, 0xf3, 0xf4, |
| 0xf5, 0xf6, 0xf7, 0xf8, |
| 0xf9, 0xfa |
| }; |
| |
| /* |
| * Parse the DHT table. |
| * This code comes from jpeg6b (jdmarker.c). |
| */ |
| static |
| int my_jpeg_load_dht (struct jpeg_decompress_struct *info, unsigned char *dht, |
| JHUFF_TBL *ac_tables[], JHUFF_TBL *dc_tables[]) |
| { |
| unsigned int length = (dht[2] << 8) + dht[3] - 2; |
| unsigned int pos = 4; |
| unsigned int count, i; |
| int index; |
| |
| JHUFF_TBL **hufftbl; |
| unsigned char bits[17]; |
| unsigned char huffval[256]; |
| |
| while (length > 16) |
| { |
| bits[0] = 0; |
| index = dht[pos++]; |
| count = 0; |
| for (i = 1; i <= 16; ++i) |
| { |
| bits[i] = dht[pos++]; |
| count += bits[i]; |
| } |
| length -= 17; |
| |
| if (count > 256 || count > length) |
| return -1; |
| |
| for (i = 0; i < count; ++i) |
| huffval[i] = dht[pos++]; |
| length -= count; |
| |
| if (index & 0x10) |
| { |
| index -= 0x10; |
| hufftbl = &ac_tables[index]; |
| } |
| else |
| hufftbl = &dc_tables[index]; |
| |
| if (index < 0 || index >= NUM_HUFF_TBLS) |
| return -1; |
| |
| if (*hufftbl == NULL) |
| *hufftbl = jpeg_alloc_huff_table ((j_common_ptr)info); |
| if (*hufftbl == NULL) |
| return -1; |
| |
| memcpy ((*hufftbl)->bits, bits, sizeof (*hufftbl)->bits); |
| memcpy ((*hufftbl)->huffval, huffval, sizeof (*hufftbl)->huffval); |
| } |
| |
| if (length != 0) |
| return -1; |
| |
| return 0; |
| } |
| |
| /*************************************************************************** |
| * end of code for supportting MJPEG image files |
| * based on a message of Laurent Pinchart on the video4linux mailing list |
| ***************************************************************************/ |
| |
| bool GrFmtJpegReader::ReadData( uchar* data, int step, int color ) |
| { |
| bool result = false; |
| |
| color = color > 0 || (m_iscolor && color < 0); |
| |
| if( m_cinfo && m_jerr && m_width && m_height ) |
| { |
| jpeg_decompress_struct* cinfo = (jpeg_decompress_struct*)m_cinfo; |
| GrFmtJpegErrorMgr* jerr = (GrFmtJpegErrorMgr*)m_jerr; |
| JSAMPARRAY buffer = 0; |
| |
| if( setjmp( jerr->setjmp_buffer ) == 0 ) |
| { |
| /* check if this is a mjpeg image format */ |
| if ( cinfo->ac_huff_tbl_ptrs[0] == NULL && |
| cinfo->ac_huff_tbl_ptrs[1] == NULL && |
| cinfo->dc_huff_tbl_ptrs[0] == NULL && |
| cinfo->dc_huff_tbl_ptrs[1] == NULL ) |
| { |
| /* yes, this is a mjpeg image format, so load the correct |
| huffman table */ |
| my_jpeg_load_dht( cinfo, |
| my_jpeg_odml_dht, |
| cinfo->ac_huff_tbl_ptrs, |
| cinfo->dc_huff_tbl_ptrs ); |
| } |
| |
| if( color > 0 || (m_iscolor && color < 0) ) |
| { |
| color = 1; |
| if( cinfo->num_components != 4 ) |
| { |
| cinfo->out_color_space = JCS_RGB; |
| cinfo->out_color_components = 3; |
| } |
| else |
| { |
| cinfo->out_color_space = JCS_CMYK; |
| cinfo->out_color_components = 4; |
| } |
| } |
| else |
| { |
| color = 0; |
| if( cinfo->num_components != 4 ) |
| { |
| cinfo->out_color_space = JCS_GRAYSCALE; |
| cinfo->out_color_components = 1; |
| } |
| else |
| { |
| cinfo->out_color_space = JCS_CMYK; |
| cinfo->out_color_components = 4; |
| } |
| } |
| |
| jpeg_start_decompress( cinfo ); |
| |
| buffer = (*cinfo->mem->alloc_sarray)((j_common_ptr)cinfo, |
| JPOOL_IMAGE, m_width*4, 1 ); |
| |
| for( ; m_height--; data += step ) |
| { |
| jpeg_read_scanlines( cinfo, buffer, 1 ); |
| if( color ) |
| { |
| if( cinfo->out_color_components == 3 ) |
| icvCvt_RGB2BGR_8u_C3R( buffer[0], 0, data, 0, cvSize(m_width,1) ); |
| else |
| icvCvt_CMYK2BGR_8u_C4C3R( buffer[0], 0, data, 0, cvSize(m_width,1) ); |
| } |
| else |
| { |
| if( cinfo->out_color_components == 1 ) |
| memcpy( data, buffer[0], m_width ); |
| else |
| icvCvt_CMYK2Gray_8u_C4C1R( buffer[0], 0, data, 0, cvSize(m_width,1) ); |
| } |
| } |
| result = true; |
| jpeg_finish_decompress( cinfo ); |
| } |
| } |
| |
| Close(); |
| return result; |
| } |
| |
| |
| /////////////////////// GrFmtJpegWriter /////////////////// |
| |
| GrFmtJpegWriter::GrFmtJpegWriter( const char* filename ) : GrFmtWriter( filename ) |
| { |
| } |
| |
| |
| GrFmtJpegWriter::~GrFmtJpegWriter() |
| { |
| } |
| |
| |
| bool GrFmtJpegWriter::WriteImage( const uchar* data, int step, |
| int width, int height, int /*depth*/, int _channels ) |
| { |
| const int default_quality = 95; |
| struct jpeg_compress_struct cinfo; |
| GrFmtJpegErrorMgr jerr; |
| |
| bool result = false; |
| FILE* f = 0; |
| int channels = _channels > 1 ? 3 : 1; |
| uchar* buffer = 0; // temporary buffer for row flipping |
| |
| cinfo.err = jpeg_std_error(&jerr.pub); |
| jerr.pub.error_exit = error_exit; |
| |
| if( setjmp( jerr.setjmp_buffer ) == 0 ) |
| { |
| jpeg_create_compress(&cinfo); |
| f = fopen( m_filename, "wb" ); |
| |
| if( f ) |
| { |
| jpeg_stdio_dest( &cinfo, f ); |
| |
| cinfo.image_width = width; |
| cinfo.image_height = height; |
| cinfo.input_components = channels; |
| cinfo.in_color_space = channels > 1 ? JCS_RGB : JCS_GRAYSCALE; |
| |
| jpeg_set_defaults( &cinfo ); |
| jpeg_set_quality( &cinfo, default_quality, |
| TRUE /* limit to baseline-JPEG values */ ); |
| jpeg_start_compress( &cinfo, TRUE ); |
| |
| if( channels > 1 ) |
| buffer = new uchar[width*channels]; |
| |
| for( ; height--; data += step ) |
| { |
| uchar* ptr = (uchar*)data; |
| |
| if( _channels == 3 ) |
| { |
| icvCvt_BGR2RGB_8u_C3R( data, 0, buffer, 0, cvSize(width,1) ); |
| ptr = buffer; |
| } |
| else if( _channels == 4 ) |
| { |
| icvCvt_BGRA2BGR_8u_C4C3R( data, 0, buffer, 0, cvSize(width,1), 2 ); |
| ptr = buffer; |
| } |
| |
| jpeg_write_scanlines( &cinfo, &ptr, 1 ); |
| } |
| |
| jpeg_finish_compress( &cinfo ); |
| result = true; |
| } |
| } |
| |
| if(f) fclose(f); |
| jpeg_destroy_compress( &cinfo ); |
| |
| delete[] buffer; |
| return result; |
| } |
| |
| #else |
| |
| ////////////////////// JPEG-oriented two-level bitstream //////////////////////// |
| |
| RJpegBitStream::RJpegBitStream() |
| { |
| } |
| |
| RJpegBitStream::~RJpegBitStream() |
| { |
| Close(); |
| } |
| |
| |
| bool RJpegBitStream::Open( const char* filename ) |
| { |
| Close(); |
| Allocate(); |
| |
| m_is_opened = m_low_strm.Open( filename ); |
| if( m_is_opened ) SetPos(0); |
| return m_is_opened; |
| } |
| |
| |
| void RJpegBitStream::Close() |
| { |
| m_low_strm.Close(); |
| m_is_opened = false; |
| } |
| |
| |
| void RJpegBitStream::ReadBlock() |
| { |
| uchar* end = m_start + m_block_size; |
| uchar* current = m_start; |
| |
| if( setjmp( m_low_strm.JmpBuf()) == 0 ) |
| { |
| int sz = m_unGetsize; |
| memmove( current - sz, m_end - sz, sz ); |
| while( current < end ) |
| { |
| int val = m_low_strm.GetByte(); |
| if( val != 0xff ) |
| { |
| *current++ = (uchar)val; |
| } |
| else |
| { |
| val = m_low_strm.GetByte(); |
| if( val == 0 ) |
| *current++ = 0xFF; |
| else if( !(0xD0 <= val && val <= 0xD7) ) |
| { |
| m_low_strm.SetPos( m_low_strm.GetPos() - 2 ); |
| goto fetch_end; |
| } |
| } |
| } |
| fetch_end: ; |
| } |
| else |
| { |
| if( current == m_start && m_jmp_set ) |
| longjmp( m_jmp_buf, RBS_THROW_EOS ); |
| } |
| m_current = m_start; |
| m_end = m_start + (((current - m_start) + 3) & -4); |
| if( !bsIsBigEndian() ) |
| bsBSwapBlock( m_start, m_end ); |
| } |
| |
| |
| void RJpegBitStream::Flush() |
| { |
| m_end = m_start + m_block_size; |
| m_current = m_end - 4; |
| m_bit_idx = 0; |
| } |
| |
| void RJpegBitStream::AlignOnByte() |
| { |
| m_bit_idx &= -8; |
| } |
| |
| int RJpegBitStream::FindMarker() |
| { |
| int code = m_low_strm.GetWord(); |
| while( (code & 0xFF00) != 0xFF00 || (code == 0xFFFF || code == 0xFF00 )) |
| { |
| code = ((code&255) << 8) | m_low_strm.GetByte(); |
| } |
| return code; |
| } |
| |
| |
| /****************************** JPEG (JFIF) reader ***************************/ |
| |
| // zigzag & IDCT prescaling (AAN algorithm) tables |
| static const uchar zigzag[] = |
| { |
| 0, 8, 1, 2, 9, 16, 24, 17, 10, 3, 4, 11, 18, 25, 32, 40, |
| 33, 26, 19, 12, 5, 6, 13, 20, 27, 34, 41, 48, 56, 49, 42, 35, |
| 28, 21, 14, 7, 15, 22, 29, 36, 43, 50, 57, 58, 51, 44, 37, 30, |
| 23, 31, 38, 45, 52, 59, 60, 53, 46, 39, 47, 54, 61, 62, 55, 63, |
| 63, 63, 63, 63, 63, 63, 63, 63, 63, 63, 63, 63, 63, 63, 63, 63 |
| }; |
| |
| |
| static const int idct_prescale[] = |
| { |
| 16384, 22725, 21407, 19266, 16384, 12873, 8867, 4520, |
| 22725, 31521, 29692, 26722, 22725, 17855, 12299, 6270, |
| 21407, 29692, 27969, 25172, 21407, 16819, 11585, 5906, |
| 19266, 26722, 25172, 22654, 19266, 15137, 10426, 5315, |
| 16384, 22725, 21407, 19266, 16384, 12873, 8867, 4520, |
| 12873, 17855, 16819, 15137, 12873, 10114, 6967, 3552, |
| 8867, 12299, 11585, 10426, 8867, 6967, 4799, 2446, |
| 4520, 6270, 5906, 5315, 4520, 3552, 2446, 1247 |
| }; |
| |
| |
| #define fixb 14 |
| #define fix(x, n) (int)((x)*(1 << (n)) + .5) |
| #define fix1(x, n) (x) |
| #define fixmul(x) (x) |
| |
| #define C0_707 fix( 0.707106781f, fixb ) |
| #define C0_924 fix( 0.923879533f, fixb ) |
| #define C0_541 fix( 0.541196100f, fixb ) |
| #define C0_382 fix( 0.382683432f, fixb ) |
| #define C1_306 fix( 1.306562965f, fixb ) |
| |
| #define C1_082 fix( 1.082392200f, fixb ) |
| #define C1_414 fix( 1.414213562f, fixb ) |
| #define C1_847 fix( 1.847759065f, fixb ) |
| #define C2_613 fix( 2.613125930f, fixb ) |
| |
| #define fixc 12 |
| #define b_cb fix( 1.772, fixc ) |
| #define g_cb -fix( 0.34414, fixc ) |
| #define g_cr -fix( 0.71414, fixc ) |
| #define r_cr fix( 1.402, fixc ) |
| |
| #define y_r fix( 0.299, fixc ) |
| #define y_g fix( 0.587, fixc ) |
| #define y_b fix( 0.114, fixc ) |
| |
| #define cb_r -fix( 0.1687, fixc ) |
| #define cb_g -fix( 0.3313, fixc ) |
| #define cb_b fix( 0.5, fixc ) |
| |
| #define cr_r fix( 0.5, fixc ) |
| #define cr_g -fix( 0.4187, fixc ) |
| #define cr_b -fix( 0.0813, fixc ) |
| |
| |
| // IDCT without prescaling |
| static void aan_idct8x8( int *src, int *dst, int step ) |
| { |
| int workspace[64], *work = workspace; |
| int i; |
| |
| /* Pass 1: process rows */ |
| for( i = 8; i > 0; i--, src += 8, work += 8 ) |
| { |
| /* Odd part */ |
| int x0 = src[5], x1 = src[3]; |
| int x2 = src[1], x3 = src[7]; |
| |
| int x4 = x0 + x1; x0 -= x1; |
| |
| x1 = x2 + x3; x2 -= x3; |
| x3 = x1 + x4; x1 -= x4; |
| |
| x4 = (x0 + x2)*C1_847; |
| x0 = descale( x4 - x0*C2_613, fixb); |
| x2 = descale( x2*C1_082 - x4, fixb); |
| x1 = descale( x1*C1_414, fixb); |
| |
| x0 -= x3; |
| x1 -= x0; |
| x2 += x1; |
| |
| work[7] = x3; work[6] = x0; |
| work[5] = x1; work[4] = x2; |
| |
| /* Even part */ |
| x2 = src[2]; x3 = src[6]; |
| x0 = src[0]; x1 = src[4]; |
| |
| x4 = x2 + x3; |
| x2 = descale((x2-x3)*C1_414, fixb) - x4; |
| |
| x3 = x0 + x1; x0 -= x1; |
| x1 = x3 + x4; x3 -= x4; |
| x4 = x0 + x2; x0 -= x2; |
| |
| x2 = work[7]; |
| x1 -= x2; x2 = 2*x2 + x1; |
| work[7] = x1; work[0] = x2; |
| |
| x2 = work[6]; |
| x1 = x4 + x2; x4 -= x2; |
| work[1] = x1; work[6] = x4; |
| |
| x1 = work[5]; x2 = work[4]; |
| x4 = x0 + x1; x0 -= x1; |
| x1 = x3 + x2; x3 -= x2; |
| |
| work[2] = x4; work[5] = x0; |
| work[3] = x3; work[4] = x1; |
| } |
| |
| /* Pass 2: process columns */ |
| work = workspace; |
| for( i = 8; i > 0; i--, dst += step, work++ ) |
| { |
| /* Odd part */ |
| int x0 = work[8*5], x1 = work[8*3]; |
| int x2 = work[8*1], x3 = work[8*7]; |
| |
| int x4 = x0 + x1; x0 -= x1; |
| x1 = x2 + x3; x2 -= x3; |
| x3 = x1 + x4; x1 -= x4; |
| |
| x4 = (x0 + x2)*C1_847; |
| x0 = descale( x4 - x0*C2_613, fixb); |
| x2 = descale( x2*C1_082 - x4, fixb); |
| x1 = descale( x1*C1_414, fixb); |
| |
| x0 -= x3; |
| x1 -= x0; |
| x2 += x1; |
| |
| dst[7] = x3; dst[6] = x0; |
| dst[5] = x1; dst[4] = x2; |
| |
| /* Even part */ |
| x2 = work[8*2]; x3 = work[8*6]; |
| x0 = work[8*0]; x1 = work[8*4]; |
| |
| x4 = x2 + x3; |
| x2 = descale((x2-x3)*C1_414, fixb) - x4; |
| |
| x3 = x0 + x1; x0 -= x1; |
| x1 = x3 + x4; x3 -= x4; |
| x4 = x0 + x2; x0 -= x2; |
| |
| x2 = dst[7]; |
| x1 -= x2; x2 = 2*x2 + x1; |
| x1 = descale(x1,3); |
| x2 = descale(x2,3); |
| |
| dst[7] = x1; dst[0] = x2; |
| |
| x2 = dst[6]; |
| x1 = descale(x4 + x2,3); |
| x4 = descale(x4 - x2,3); |
| dst[1] = x1; dst[6] = x4; |
| |
| x1 = dst[5]; x2 = dst[4]; |
| |
| x4 = descale(x0 + x1,3); |
| x0 = descale(x0 - x1,3); |
| x1 = descale(x3 + x2,3); |
| x3 = descale(x3 - x2,3); |
| |
| dst[2] = x4; dst[5] = x0; |
| dst[3] = x3; dst[4] = x1; |
| } |
| } |
| |
| |
| static const int max_dec_htable_size = 1 << 12; |
| static const int first_table_bits = 9; |
| |
| GrFmtJpegReader::GrFmtJpegReader( const char* filename ) : GrFmtReader( filename ) |
| { |
| m_planes= -1; |
| m_offset= -1; |
| |
| int i; |
| for( i = 0; i < 4; i++ ) |
| { |
| m_td[i] = new short[max_dec_htable_size]; |
| m_ta[i] = new short[max_dec_htable_size]; |
| } |
| } |
| |
| |
| GrFmtJpegReader::~GrFmtJpegReader() |
| { |
| for( int i = 0; i < 4; i++ ) |
| { |
| delete[] m_td[i]; |
| m_td[i] = 0; |
| delete[] m_ta[i]; |
| m_ta[i] = 0; |
| } |
| } |
| |
| |
| void GrFmtJpegReader::Close() |
| { |
| m_strm.Close(); |
| GrFmtReader::Close(); |
| } |
| |
| |
| bool GrFmtJpegReader::ReadHeader() |
| { |
| char buffer[16]; |
| int i; |
| bool result = false, is_sof = false, |
| is_qt = false, is_ht = false, is_sos = false; |
| |
| assert( strlen(m_filename) != 0 ); |
| if( !m_strm.Open( m_filename )) return false; |
| |
| memset( m_is_tq, 0, sizeof(m_is_tq)); |
| memset( m_is_td, 0, sizeof(m_is_td)); |
| memset( m_is_ta, 0, sizeof(m_is_ta)); |
| m_MCUs = 0; |
| |
| if( setjmp( m_strm.JmpBuf()) == 0 ) |
| { |
| RMByteStream& lstrm = m_strm.m_low_strm; |
| |
| lstrm.Skip( 2 ); // skip SOI marker |
| |
| for(;;) |
| { |
| int marker = m_strm.FindMarker() & 255; |
| |
| // check for standalone markers |
| if( marker != 0xD8 /* SOI */ && marker != 0xD9 /* EOI */ && |
| marker != 0x01 /* TEM */ && !( 0xD0 <= marker && marker <= 0xD7 )) |
| { |
| int pos = lstrm.GetPos(); |
| int length = lstrm.GetWord(); |
| |
| switch( marker ) |
| { |
| case 0xE0: // APP0 |
| lstrm.GetBytes( buffer, 5 ); |
| if( strcmp(buffer, "JFIF") == 0 ) // JFIF identification |
| { |
| m_version = lstrm.GetWord(); |
| //is_jfif = true; |
| } |
| break; |
| |
| case 0xC0: // SOF0 |
| m_precision = lstrm.GetByte(); |
| m_height = lstrm.GetWord(); |
| m_width = lstrm.GetWord(); |
| m_planes = lstrm.GetByte(); |
| |
| if( m_width == 0 || m_height == 0 || // DNL not supported |
| (m_planes != 1 && m_planes != 3)) goto parsing_end; |
| |
| m_iscolor = m_planes == 3; |
| |
| memset( m_ci, -1, sizeof(m_ci)); |
| |
| for( i = 0; i < m_planes; i++ ) |
| { |
| int idx = lstrm.GetByte(); |
| |
| if( idx < 1 || idx > m_planes ) // wrong index |
| { |
| idx = i+1; // hack |
| } |
| cmp_info& ci = m_ci[idx-1]; |
| |
| if( ci.tq > 0 /* duplicated description */) goto parsing_end; |
| |
| ci.h = (char)lstrm.GetByte(); |
| ci.v = (char)(ci.h & 15); |
| ci.h >>= 4; |
| ci.tq = (char)lstrm.GetByte(); |
| if( !((ci.h == 1 || ci.h == 2 || ci.h == 4) && |
| (ci.v == 1 || ci.v == 2 || ci.v == 4) && |
| ci.tq < 3) || |
| // chroma mcu-parts should have equal sizes and |
| // be non greater then luma sizes |
| !( i != 2 || (ci.h == m_ci[1].h && ci.v == m_ci[1].v && |
| ci.h <= m_ci[0].h && ci.v <= m_ci[0].v))) |
| goto parsing_end; |
| } |
| is_sof = true; |
| m_type = marker - 0xC0; |
| break; |
| |
| case 0xDB: // DQT |
| if( !LoadQuantTables( length )) goto parsing_end; |
| is_qt = true; |
| break; |
| |
| case 0xC4: // DHT |
| if( !LoadHuffmanTables( length )) goto parsing_end; |
| is_ht = true; |
| break; |
| |
| case 0xDA: // SOS |
| is_sos = true; |
| m_offset = pos - 2; |
| goto parsing_end; |
| |
| case 0xDD: // DRI |
| m_MCUs = lstrm.GetWord(); |
| break; |
| } |
| lstrm.SetPos( pos + length ); |
| } |
| } |
| parsing_end: ; |
| } |
| |
| result = /*is_jfif &&*/ is_sof && is_qt && is_ht && is_sos; |
| if( !result ) |
| { |
| m_width = m_height = -1; |
| m_offset = -1; |
| m_strm.Close(); |
| } |
| return result; |
| } |
| |
| |
| bool GrFmtJpegReader::LoadQuantTables( int length ) |
| { |
| uchar buffer[128]; |
| int i, tq_size; |
| |
| RMByteStream& lstrm = m_strm.m_low_strm; |
| length -= 2; |
| |
| while( length > 0 ) |
| { |
| int tq = lstrm.GetByte(); |
| int size = tq >> 4; |
| tq &= 15; |
| |
| tq_size = (64<<size) + 1; |
| if( tq > 3 || size > 1 || length < tq_size ) return false; |
| length -= tq_size; |
| |
| lstrm.GetBytes( buffer, tq_size - 1 ); |
| |
| if( size == 0 ) // 8 bit quant factors |
| { |
| for( i = 0; i < 64; i++ ) |
| { |
| int idx = zigzag[i]; |
| m_tq[tq][idx] = buffer[i] * 16 * idct_prescale[idx]; |
| } |
| } |
| else // 16 bit quant factors |
| { |
| for( i = 0; i < 64; i++ ) |
| { |
| int idx = zigzag[i]; |
| m_tq[tq][idx] = ((unsigned short*)buffer)[i] * idct_prescale[idx]; |
| } |
| } |
| m_is_tq[tq] = true; |
| } |
| |
| return true; |
| } |
| |
| |
| bool GrFmtJpegReader::LoadHuffmanTables( int length ) |
| { |
| const int max_bits = 16; |
| uchar buffer[1024]; |
| int buffer2[1024]; |
| |
| int i, ht_size; |
| RMByteStream& lstrm = m_strm.m_low_strm; |
| length -= 2; |
| |
| while( length > 0 ) |
| { |
| int t = lstrm.GetByte(); |
| int hclass = t >> 4; |
| t &= 15; |
| |
| if( t > 3 || hclass > 1 || length < 17 ) return false; |
| length -= 17; |
| |
| lstrm.GetBytes( buffer, max_bits ); |
| for( i = 0, ht_size = 0; i < max_bits; i++ ) ht_size += buffer[i]; |
| |
| if( length < ht_size ) return false; |
| length -= ht_size; |
| |
| lstrm.GetBytes( buffer + max_bits, ht_size ); |
| |
| if( !::bsCreateDecodeHuffmanTable( |
| ::bsCreateSourceHuffmanTable( |
| buffer, buffer2, max_bits, first_table_bits ), |
| hclass == 0 ? m_td[t] : m_ta[t], |
| max_dec_htable_size )) return false; |
| if( hclass == 0 ) |
| m_is_td[t] = true; |
| else |
| m_is_ta[t] = true; |
| } |
| return true; |
| } |
| |
| |
| bool GrFmtJpegReader::ReadData( uchar* data, int step, int color ) |
| { |
| if( m_offset < 0 || !m_strm.IsOpened()) |
| return false; |
| |
| if( setjmp( m_strm.JmpBuf()) == 0 ) |
| { |
| RMByteStream& lstrm = m_strm.m_low_strm; |
| lstrm.SetPos( m_offset ); |
| |
| for(;;) |
| { |
| int marker = m_strm.FindMarker() & 255; |
| |
| if( marker == 0xD8 /* SOI */ || marker == 0xD9 /* EOI */ ) |
| goto decoding_end; |
| |
| // check for standalone markers |
| if( marker != 0x01 /* TEM */ && !( 0xD0 <= marker && marker <= 0xD7 )) |
| { |
| int pos = lstrm.GetPos(); |
| int length = lstrm.GetWord(); |
| |
| switch( marker ) |
| { |
| case 0xC4: // DHT |
| if( !LoadHuffmanTables( length )) goto decoding_end; |
| break; |
| |
| case 0xDA: // SOS |
| // read scan header |
| { |
| int idx[3] = { -1, -1, -1 }; |
| int i, ns = lstrm.GetByte(); |
| int sum = 0, a; // spectral selection & approximation |
| |
| if( ns != m_planes ) goto decoding_end; |
| for( i = 0; i < ns; i++ ) |
| { |
| int td, ta, c = lstrm.GetByte() - 1; |
| if( c < 0 || m_planes <= c ) |
| { |
| c = i; // hack |
| } |
| |
| if( idx[c] != -1 ) goto decoding_end; |
| idx[i] = c; |
| td = lstrm.GetByte(); |
| ta = td & 15; |
| td >>= 4; |
| if( !(ta <= 3 && m_is_ta[ta] && |
| td <= 3 && m_is_td[td] && |
| m_is_tq[m_ci[c].tq]) ) |
| goto decoding_end; |
| |
| m_ci[c].td = (char)td; |
| m_ci[c].ta = (char)ta; |
| |
| sum += m_ci[c].h*m_ci[c].v; |
| } |
| |
| if( sum > 10 ) goto decoding_end; |
| |
| m_ss = lstrm.GetByte(); |
| m_se = lstrm.GetByte(); |
| |
| a = lstrm.GetByte(); |
| m_al = a & 15; |
| m_ah = a >> 4; |
| |
| ProcessScan( idx, ns, data, step, color ); |
| goto decoding_end; // only single scan case is supported now |
| } |
| |
| //m_offset = pos - 2; |
| //break; |
| |
| case 0xDD: // DRI |
| m_MCUs = lstrm.GetWord(); |
| break; |
| } |
| |
| if( marker != 0xDA ) lstrm.SetPos( pos + length ); |
| } |
| } |
| decoding_end: ; |
| } |
| |
| return true; |
| } |
| |
| |
| void GrFmtJpegReader::ResetDecoder() |
| { |
| m_ci[0].dc_pred = m_ci[1].dc_pred = m_ci[2].dc_pred = 0; |
| } |
| |
| void GrFmtJpegReader::ProcessScan( int* idx, int ns, uchar* data, int step, int color ) |
| { |
| int i, s = 0, mcu, x1 = 0, y1 = 0; |
| int temp[64]; |
| int blocks[10][64]; |
| int pos[3], h[3], v[3]; |
| int x_shift = 0, y_shift = 0; |
| int nch = color ? 3 : 1; |
| |
| assert( ns == m_planes && m_ss == 0 && m_se == 63 && |
| m_al == 0 && m_ah == 0 ); // sequental & single scan |
| |
| assert( idx[0] == 0 && (ns ==1 || (idx[1] == 1 && idx[2] == 2))); |
| |
| for( i = 0; i < ns; i++ ) |
| { |
| int c = idx[i]; |
| h[c] = m_ci[c].h*8; |
| v[c] = m_ci[c].v*8; |
| pos[c] = s >> 6; |
| s += h[c]*v[c]; |
| } |
| |
| if( ns == 3 ) |
| { |
| x_shift = h[0]/(h[1]*2); |
| y_shift = v[0]/(v[1]*2); |
| } |
| |
| m_strm.Flush(); |
| ResetDecoder(); |
| |
| for( mcu = 0;; mcu++ ) |
| { |
| int x2, y2, x, y, xc; |
| int* cmp; |
| uchar* data1; |
| |
| if( mcu == m_MCUs && m_MCUs != 0 ) |
| { |
| ResetDecoder(); |
| m_strm.AlignOnByte(); |
| mcu = 0; |
| } |
| |
| // Get mcu |
| for( i = 0; i < ns; i++ ) |
| { |
| int c = idx[i]; |
| cmp = blocks[pos[c]]; |
| for( y = 0; y < v[c]; y += 8, cmp += h[c]*8 ) |
| for( x = 0; x < h[c]; x += 8 ) |
| { |
| GetBlock( temp, c ); |
| if( i < (color ? 3 : 1)) |
| { |
| aan_idct8x8( temp, cmp + x, h[c] ); |
| } |
| } |
| } |
| |
| y2 = v[0]; |
| x2 = h[0]; |
| |
| if( y1 + y2 > m_height ) y2 = m_height - y1; |
| if( x1 + x2 > m_width ) x2 = m_width - x1; |
| |
| cmp = blocks[0]; |
| data1 = data + x1*nch; |
| |
| if( ns == 1 ) |
| for( y = 0; y < y2; y++, data1 += step, cmp += h[0] ) |
| { |
| if( color ) |
| { |
| for( x = 0; x < x2; x++ ) |
| { |
| int val = descale( cmp[x] + 128*4, 2 ); |
| data1[x*3] = data1[x*3 + 1] = data1[x*3 + 2] = saturate( val ); |
| } |
| } |
| else |
| { |
| for( x = 0; x < x2; x++ ) |
| { |
| int val = descale( cmp[x] + 128*4, 2 ); |
| data1[x] = saturate( val ); |
| } |
| } |
| } |
| else |
| { |
| for( y = 0; y < y2; y++, data1 += step, cmp += h[0] ) |
| { |
| if( color ) |
| { |
| int shift = h[1]*(y >> y_shift); |
| int* cmpCb = blocks[pos[1]] + shift; |
| int* cmpCr = blocks[pos[2]] + shift; |
| x = 0; |
| if( x_shift == 0 ) |
| { |
| for( ; x < x2; x++ ) |
| { |
| int Y = (cmp[x] + 128*4) << fixc; |
| int Cb = cmpCb[x]; |
| int Cr = cmpCr[x]; |
| int t = (Y + Cb*b_cb) >> (fixc + 2); |
| data1[x*3] = saturate(t); |
| t = (Y + Cb*g_cb + Cr*g_cr) >> (fixc + 2); |
| data1[x*3 + 1] = saturate(t); |
| t = (Y + Cr*r_cr) >> (fixc + 2); |
| data1[x*3 + 2] = saturate(t); |
| } |
| } |
| else if( x_shift == 1 ) |
| { |
| for( xc = 0; x <= x2 - 2; x += 2, xc++ ) |
| { |
| int Y = (cmp[x] + 128*4) << fixc; |
| int Cb = cmpCb[xc]; |
| int Cr = cmpCr[xc]; |
| int t = (Y + Cb*b_cb) >> (fixc + 2); |
| data1[x*3] = saturate(t); |
| t = (Y + Cb*g_cb + Cr*g_cr) >> (fixc + 2); |
| data1[x*3 + 1] = saturate(t); |
| t = (Y + Cr*r_cr) >> (fixc + 2); |
| data1[x*3 + 2] = saturate(t); |
| Y = (cmp[x+1] + 128*4) << fixc; |
| t = (Y + Cb*b_cb) >> (fixc + 2); |
| data1[x*3 + 3] = saturate(t); |
| t = (Y + Cb*g_cb + Cr*g_cr) >> (fixc + 2); |
| data1[x*3 + 4] = saturate(t); |
| t = (Y + Cr*r_cr) >> (fixc + 2); |
| data1[x*3 + 5] = saturate(t); |
| } |
| } |
| for( ; x < x2; x++ ) |
| { |
| int Y = (cmp[x] + 128*4) << fixc; |
| int Cb = cmpCb[x >> x_shift]; |
| int Cr = cmpCr[x >> x_shift]; |
| int t = (Y + Cb*b_cb) >> (fixc + 2); |
| data1[x*3] = saturate(t); |
| t = (Y + Cb*g_cb + Cr*g_cr) >> (fixc + 2); |
| data1[x*3 + 1] = saturate(t); |
| t = (Y + Cr*r_cr) >> (fixc + 2); |
| data1[x*3 + 2] = saturate(t); |
| } |
| } |
| else |
| { |
| for( x = 0; x < x2; x++ ) |
| { |
| int val = descale( cmp[x] + 128*4, 2 ); |
| data1[x] = saturate(val); |
| } |
| } |
| } |
| } |
| |
| x1 += h[0]; |
| if( x1 >= m_width ) |
| { |
| x1 = 0; |
| y1 += v[0]; |
| data += v[0]*step; |
| if( y1 >= m_height ) break; |
| } |
| } |
| } |
| |
| |
| void GrFmtJpegReader::GetBlock( int* block, int c ) |
| { |
| memset( block, 0, 64*sizeof(block[0]) ); |
| |
| assert( 0 <= c && c < 3 ); |
| const short* td = m_td[m_ci[c].td]; |
| const short* ta = m_ta[m_ci[c].ta]; |
| const int* tq = m_tq[m_ci[c].tq]; |
| |
| // Get DC coefficient |
| int i = 0, cat = m_strm.GetHuff( td ); |
| int mask = bs_bit_mask[cat]; |
| int val = m_strm.Get( cat ); |
| |
| val -= (val*2 <= mask ? mask : 0); |
| m_ci[c].dc_pred = val += m_ci[c].dc_pred; |
| |
| block[0] = descale(val * tq[0],16); |
| |
| // Get AC coeffs |
| for(;;) |
| { |
| cat = m_strm.GetHuff( ta ); |
| if( cat == 0 ) break; // end of block |
| |
| i += (cat >> 4) + 1; |
| cat &= 15; |
| mask = bs_bit_mask[cat]; |
| val = m_strm.Get( cat ); |
| cat = zigzag[i]; |
| val -= (val*2 <= mask ? mask : 0); |
| block[cat] = descale(val * tq[cat], 16); |
| assert( i <= 63 ); |
| if( i >= 63 ) break; |
| } |
| } |
| |
| ////////////////////// WJpegStream /////////////////////// |
| |
| WJpegBitStream::WJpegBitStream() |
| { |
| } |
| |
| |
| WJpegBitStream::~WJpegBitStream() |
| { |
| Close(); |
| m_is_opened = false; |
| } |
| |
| |
| |
| bool WJpegBitStream::Open( const char* filename ) |
| { |
| Close(); |
| Allocate(); |
| |
| m_is_opened = m_low_strm.Open( filename ); |
| if( m_is_opened ) |
| { |
| m_block_pos = 0; |
| ResetBuffer(); |
| } |
| return m_is_opened; |
| } |
| |
| |
| void WJpegBitStream::Close() |
| { |
| if( m_is_opened ) |
| { |
| Flush(); |
| m_low_strm.Close(); |
| m_is_opened = false; |
| } |
| } |
| |
| |
| void WJpegBitStream::Flush() |
| { |
| Put( -1, m_bit_idx & 31 ); |
| *((ulong*&)m_current)++ = m_val; |
| WriteBlock(); |
| ResetBuffer(); |
| } |
| |
| |
| void WJpegBitStream::WriteBlock() |
| { |
| uchar* ptr = m_start; |
| if( !bsIsBigEndian() ) |
| bsBSwapBlock( m_start, m_current ); |
| |
| while( ptr < m_current ) |
| { |
| int val = *ptr++; |
| m_low_strm.PutByte( val ); |
| if( val == 0xff ) |
| { |
| m_low_strm.PutByte( 0 ); |
| } |
| } |
| |
| m_current = m_start; |
| } |
| |
| |
| /////////////////////// GrFmtJpegWriter /////////////////// |
| |
| GrFmtJpegWriter::GrFmtJpegWriter( const char* filename ) : GrFmtWriter( filename ) |
| { |
| } |
| |
| GrFmtJpegWriter::~GrFmtJpegWriter() |
| { |
| } |
| |
| // Standard JPEG quantization tables |
| static const uchar jpegTableK1_T[] = |
| { |
| 16, 12, 14, 14, 18, 24, 49, 72, |
| 11, 12, 13, 17, 22, 35, 64, 92, |
| 10, 14, 16, 22, 37, 55, 78, 95, |
| 16, 19, 24, 29, 56, 64, 87, 98, |
| 24, 26, 40, 51, 68, 81, 103, 112, |
| 40, 58, 57, 87, 109, 104, 121, 100, |
| 51, 60, 69, 80, 103, 113, 120, 103, |
| 61, 55, 56, 62, 77, 92, 101, 99 |
| }; |
| |
| |
| static const uchar jpegTableK2_T[] = |
| { |
| 17, 18, 24, 47, 99, 99, 99, 99, |
| 18, 21, 26, 66, 99, 99, 99, 99, |
| 24, 26, 56, 99, 99, 99, 99, 99, |
| 47, 66, 99, 99, 99, 99, 99, 99, |
| 99, 99, 99, 99, 99, 99, 99, 99, |
| 99, 99, 99, 99, 99, 99, 99, 99, |
| 99, 99, 99, 99, 99, 99, 99, 99, |
| 99, 99, 99, 99, 99, 99, 99, 99 |
| }; |
| |
| |
| // Standard Huffman tables |
| |
| // ... for luma DCs. |
| static const uchar jpegTableK3[] = |
| { |
| 0, 1, 5, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, |
| 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 |
| }; |
| |
| |
| // ... for chroma DCs. |
| static const uchar jpegTableK4[] = |
| { |
| 0, 3, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, |
| 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 |
| }; |
| |
| |
| // ... for luma ACs. |
| static const uchar jpegTableK5[] = |
| { |
| 0, 2, 1, 3, 3, 2, 4, 3, 5, 5, 4, 4, 0, 0, 1, 125, |
| 0x01, 0x02, 0x03, 0x00, 0x04, 0x11, 0x05, 0x12, |
| 0x21, 0x31, 0x41, 0x06, 0x13, 0x51, 0x61, 0x07, |
| 0x22, 0x71, 0x14, 0x32, 0x81, 0x91, 0xa1, 0x08, |
| 0x23, 0x42, 0xb1, 0xc1, 0x15, 0x52, 0xd1, 0xf0, |
| 0x24, 0x33, 0x62, 0x72, 0x82, 0x09, 0x0a, 0x16, |
| 0x17, 0x18, 0x19, 0x1a, 0x25, 0x26, 0x27, 0x28, |
| 0x29, 0x2a, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, |
| 0x3a, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, 0x49, |
| 0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, |
| 0x5a, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69, |
| 0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79, |
| 0x7a, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88, 0x89, |
| 0x8a, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, |
| 0x99, 0x9a, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7, |
| 0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, |
| 0xb7, 0xb8, 0xb9, 0xba, 0xc2, 0xc3, 0xc4, 0xc5, |
| 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xd2, 0xd3, 0xd4, |
| 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda, 0xe1, 0xe2, |
| 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, |
| 0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8, |
| 0xf9, 0xfa |
| }; |
| |
| // ... for chroma ACs |
| static const uchar jpegTableK6[] = |
| { |
| 0, 2, 1, 2, 4, 4, 3, 4, 7, 5, 4, 4, 0, 1, 2, 119, |
| 0x00, 0x01, 0x02, 0x03, 0x11, 0x04, 0x05, 0x21, |
| 0x31, 0x06, 0x12, 0x41, 0x51, 0x07, 0x61, 0x71, |
| 0x13, 0x22, 0x32, 0x81, 0x08, 0x14, 0x42, 0x91, |
| 0xa1, 0xb1, 0xc1, 0x09, 0x23, 0x33, 0x52, 0xf0, |
| 0x15, 0x62, 0x72, 0xd1, 0x0a, 0x16, 0x24, 0x34, |
| 0xe1, 0x25, 0xf1, 0x17, 0x18, 0x19, 0x1a, 0x26, |
| 0x27, 0x28, 0x29, 0x2a, 0x35, 0x36, 0x37, 0x38, |
| 0x39, 0x3a, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48, |
| 0x49, 0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, |
| 0x59, 0x5a, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, |
| 0x69, 0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, |
| 0x79, 0x7a, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87, |
| 0x88, 0x89, 0x8a, 0x92, 0x93, 0x94, 0x95, 0x96, |
| 0x97, 0x98, 0x99, 0x9a, 0xa2, 0xa3, 0xa4, 0xa5, |
| 0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4, |
| 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba, 0xc2, 0xc3, |
| 0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xd2, |
| 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8, 0xd9, 0xda, |
| 0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, |
| 0xea, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8, |
| 0xf9, 0xfa |
| }; |
| |
| |
| static const char jpegHeader[] = |
| "\xFF\xD8" // SOI - start of image |
| "\xFF\xE0" // APP0 - jfif extention |
| "\x00\x10" // 2 bytes: length of APP0 segment |
| "JFIF\x00" // JFIF signature |
| "\x01\x02" // version of JFIF |
| "\x00" // units = pixels ( 1 - inch, 2 - cm ) |
| "\x00\x01\x00\x01" // 2 2-bytes values: x density & y density |
| "\x00\x00"; // width & height of thumbnail: ( 0x0 means no thumbnail) |
| |
| #define postshift 14 |
| |
| // FDCT with postscaling |
| static void aan_fdct8x8( int *src, int *dst, |
| int step, const int *postscale ) |
| { |
| int workspace[64], *work = workspace; |
| int i; |
| |
| // Pass 1: process rows |
| for( i = 8; i > 0; i--, src += step, work += 8 ) |
| { |
| int x0 = src[0], x1 = src[7]; |
| int x2 = src[3], x3 = src[4]; |
| |
| int x4 = x0 + x1; x0 -= x1; |
| x1 = x2 + x3; x2 -= x3; |
| |
| work[7] = x0; work[1] = x2; |
| x2 = x4 + x1; x4 -= x1; |
| |
| x0 = src[1]; x3 = src[6]; |
| x1 = x0 + x3; x0 -= x3; |
| work[5] = x0; |
| |
| x0 = src[2]; x3 = src[5]; |
| work[3] = x0 - x3; x0 += x3; |
| |
| x3 = x0 + x1; x0 -= x1; |
| x1 = x2 + x3; x2 -= x3; |
| |
| work[0] = x1; work[4] = x2; |
| |
| x0 = descale((x0 - x4)*C0_707, fixb); |
| x1 = x4 + x0; x4 -= x0; |
| work[2] = x4; work[6] = x1; |
| |
| x0 = work[1]; x1 = work[3]; |
| x2 = work[5]; x3 = work[7]; |
| |
| x0 += x1; x1 += x2; x2 += x3; |
| x1 = descale(x1*C0_707, fixb); |
| |
| x4 = x1 + x3; x3 -= x1; |
| x1 = (x0 - x2)*C0_382; |
| x0 = descale(x0*C0_541 + x1, fixb); |
| x2 = descale(x2*C1_306 + x1, fixb); |
| |
| x1 = x0 + x3; x3 -= x0; |
| x0 = x4 + x2; x4 -= x2; |
| |
| work[5] = x1; work[1] = x0; |
| work[7] = x4; work[3] = x3; |
| } |
| |
| work = workspace; |
| // pass 2: process columns |
| for( i = 8; i > 0; i--, work++, postscale += 8, dst += 8 ) |
| { |
| int x0 = work[8*0], x1 = work[8*7]; |
| int x2 = work[8*3], x3 = work[8*4]; |
| |
| int x4 = x0 + x1; x0 -= x1; |
| x1 = x2 + x3; x2 -= x3; |
| |
| work[8*7] = x0; work[8*0] = x2; |
| x2 = x4 + x1; x4 -= x1; |
| |
| x0 = work[8*1]; x3 = work[8*6]; |
| x1 = x0 + x3; x0 -= x3; |
| work[8*4] = x0; |
| |
| x0 = work[8*2]; x3 = work[8*5]; |
| work[8*3] = x0 - x3; x0 += x3; |
| |
| x3 = x0 + x1; x0 -= x1; |
| x1 = x2 + x3; x2 -= x3; |
| |
| dst[0] = descale(x1*postscale[0], postshift); |
| dst[4] = descale(x2*postscale[4], postshift); |
| |
| x0 = descale((x0 - x4)*C0_707, fixb); |
| x1 = x4 + x0; x4 -= x0; |
| |
| dst[2] = descale(x4*postscale[2], postshift); |
| dst[6] = descale(x1*postscale[6], postshift); |
| |
| x0 = work[8*0]; x1 = work[8*3]; |
| x2 = work[8*4]; x3 = work[8*7]; |
| |
| x0 += x1; x1 += x2; x2 += x3; |
| x1 = descale(x1*C0_707, fixb); |
| |
| x4 = x1 + x3; x3 -= x1; |
| x1 = (x0 - x2)*C0_382; |
| x0 = descale(x0*C0_541 + x1, fixb); |
| x2 = descale(x2*C1_306 + x1, fixb); |
| |
| x1 = x0 + x3; x3 -= x0; |
| x0 = x4 + x2; x4 -= x2; |
| |
| dst[5] = descale(x1*postscale[5], postshift); |
| dst[1] = descale(x0*postscale[1], postshift); |
| dst[7] = descale(x4*postscale[7], postshift); |
| dst[3] = descale(x3*postscale[3], postshift); |
| } |
| } |
| |
| |
| bool GrFmtJpegWriter::WriteImage( const uchar* data, int step, |
| int width, int height, int /*depth*/, int _channels ) |
| { |
| assert( data && width > 0 && height > 0 ); |
| |
| if( !m_strm.Open( m_filename ) ) return false; |
| |
| // encode the header and tables |
| // for each mcu: |
| // convert rgb to yuv with downsampling (if color). |
| // for every block: |
| // calc dct and quantize |
| // encode block. |
| int x, y; |
| int i, j; |
| const int max_quality = 12; |
| int quality = max_quality; |
| WMByteStream& lowstrm = m_strm.m_low_strm; |
| int fdct_qtab[2][64]; |
| ulong huff_dc_tab[2][16]; |
| ulong huff_ac_tab[2][256]; |
| int channels = _channels > 1 ? 3 : 1; |
| int x_scale = channels > 1 ? 2 : 1, y_scale = x_scale; |
| int dc_pred[] = { 0, 0, 0 }; |
| int x_step = x_scale * 8; |
| int y_step = y_scale * 8; |
| int block[6][64]; |
| int buffer[1024]; |
| int luma_count = x_scale*y_scale; |
| int block_count = luma_count + channels - 1; |
| int Y_step = x_scale*8; |
| const int UV_step = 16; |
| double inv_quality; |
| |
| if( quality < 3 ) quality = 3; |
| if( quality > max_quality ) quality = max_quality; |
| |
| inv_quality = 1./quality; |
| |
| // Encode header |
| lowstrm.PutBytes( jpegHeader, sizeof(jpegHeader) - 1 ); |
| |
| // Encode quantization tables |
| for( i = 0; i < (channels > 1 ? 2 : 1); i++ ) |
| { |
| const uchar* qtable = i == 0 ? jpegTableK1_T : jpegTableK2_T; |
| int chroma_scale = i > 0 ? luma_count : 1; |
| |
| lowstrm.PutWord( 0xffdb ); // DQT marker |
| lowstrm.PutWord( 2 + 65*1 ); // put single qtable |
| lowstrm.PutByte( 0*16 + i ); // 8-bit table |
| |
| // put coefficients |
| for( j = 0; j < 64; j++ ) |
| { |
| int idx = zigzag[j]; |
| int qval = cvRound(qtable[idx]*inv_quality); |
| if( qval < 1 ) |
| qval = 1; |
| if( qval > 255 ) |
| qval = 255; |
| fdct_qtab[i][idx] = cvRound((1 << (postshift + 9))/ |
| (qval*chroma_scale*idct_prescale[idx])); |
| lowstrm.PutByte( qval ); |
| } |
| } |
| |
| // Encode huffman tables |
| for( i = 0; i < (channels > 1 ? 4 : 2); i++ ) |
| { |
| const uchar* htable = i == 0 ? jpegTableK3 : i == 1 ? jpegTableK5 : |
| i == 2 ? jpegTableK4 : jpegTableK6; |
| int is_ac_tab = i & 1; |
| int idx = i >= 2; |
| int tableSize = 16 + (is_ac_tab ? 162 : 12); |
| |
| lowstrm.PutWord( 0xFFC4 ); // DHT marker |
| lowstrm.PutWord( 3 + tableSize ); // define one huffman table |
| lowstrm.PutByte( is_ac_tab*16 + idx ); // put DC/AC flag and table index |
| lowstrm.PutBytes( htable, tableSize ); // put table |
| |
| bsCreateEncodeHuffmanTable( bsCreateSourceHuffmanTable( |
| htable, buffer, 16, 9 ), is_ac_tab ? huff_ac_tab[idx] : |
| huff_dc_tab[idx], is_ac_tab ? 256 : 16 ); |
| } |
| |
| // put frame header |
| lowstrm.PutWord( 0xFFC0 ); // SOF0 marker |
| lowstrm.PutWord( 8 + 3*channels ); // length of frame header |
| lowstrm.PutByte( 8 ); // sample precision |
| lowstrm.PutWord( height ); |
| lowstrm.PutWord( width ); |
| lowstrm.PutByte( channels ); // number of components |
| |
| for( i = 0; i < channels; i++ ) |
| { |
| lowstrm.PutByte( i + 1 ); // (i+1)-th component id (Y,U or V) |
| if( i == 0 ) |
| lowstrm.PutByte(x_scale*16 + y_scale); // chroma scale factors |
| else |
| lowstrm.PutByte(1*16 + 1); |
| lowstrm.PutByte( i > 0 ); // quantization table idx |
| } |
| |
| // put scan header |
| lowstrm.PutWord( 0xFFDA ); // SOS marker |
| lowstrm.PutWord( 6 + 2*channels ); // length of scan header |
| lowstrm.PutByte( channels ); // number of components in the scan |
| |
| for( i = 0; i < channels; i++ ) |
| { |
| lowstrm.PutByte( i+1 ); // component id |
| lowstrm.PutByte( (i>0)*16 + (i>0) );// selection of DC & AC tables |
| } |
| |
| lowstrm.PutWord(0*256 + 63);// start and end of spectral selection - for |
| // sequental DCT start is 0 and end is 63 |
| |
| lowstrm.PutByte( 0 ); // successive approximation bit position |
| // high & low - (0,0) for sequental DCT |
| |
| // encode data |
| for( y = 0; y < height; y += y_step, data += y_step*step ) |
| { |
| for( x = 0; x < width; x += x_step ) |
| { |
| int x_limit = x_step; |
| int y_limit = y_step; |
| const uchar* rgb_data = data + x*_channels; |
| int* Y_data = block[0]; |
| |
| if( x + x_limit > width ) x_limit = width - x; |
| if( y + y_limit > height ) y_limit = height - y; |
| |
| memset( block, 0, block_count*64*sizeof(block[0][0])); |
| |
| if( channels > 1 ) |
| { |
| int* UV_data = block[luma_count]; |
| |
| for( i = 0; i < y_limit; i++, rgb_data += step, Y_data += Y_step ) |
| { |
| for( j = 0; j < x_limit; j++, rgb_data += _channels ) |
| { |
| int r = rgb_data[2]; |
| int g = rgb_data[1]; |
| int b = rgb_data[0]; |
| |
| int Y = descale( r*y_r + g*y_g + b*y_b, fixc - 2) - 128*4; |
| int U = descale( r*cb_r + g*cb_g + b*cb_b, fixc - 2 ); |
| int V = descale( r*cr_r + g*cr_g + b*cr_b, fixc - 2 ); |
| int j2 = j >> (x_scale - 1); |
| |
| Y_data[j] = Y; |
| UV_data[j2] += U; |
| UV_data[j2 + 8] += V; |
| } |
| |
| rgb_data -= x_limit*_channels; |
| if( ((i+1) & (y_scale - 1)) == 0 ) |
| { |
| UV_data += UV_step; |
| } |
| } |
| } |
| else |
| { |
| for( i = 0; i < y_limit; i++, rgb_data += step, Y_data += Y_step ) |
| { |
| for( j = 0; j < x_limit; j++ ) |
| Y_data[j] = rgb_data[j]*4 - 128*4; |
| } |
| } |
| |
| for( i = 0; i < block_count; i++ ) |
| { |
| int is_chroma = i >= luma_count; |
| int src_step = x_scale * 8; |
| int run = 0, val; |
| int* src_ptr = block[i & -2] + (i & 1)*8; |
| const ulong* htable = huff_ac_tab[is_chroma]; |
| |
| aan_fdct8x8( src_ptr, buffer, src_step, fdct_qtab[is_chroma] ); |
| |
| j = is_chroma + (i > luma_count); |
| val = buffer[0] - dc_pred[j]; |
| dc_pred[j] = buffer[0]; |
| |
| { |
| float a = (float)val; |
| int cat = (((int&)a >> 23) & 255) - (126 & (val ? -1 : 0)); |
| |
| assert( cat <= 11 ); |
| m_strm.PutHuff( cat, huff_dc_tab[is_chroma] ); |
| m_strm.Put( val - (val < 0 ? 1 : 0), cat ); |
| } |
| |
| for( j = 1; j < 64; j++ ) |
| { |
| val = buffer[zigzag[j]]; |
| |
| if( val == 0 ) |
| { |
| run++; |
| } |
| else |
| { |
| while( run >= 16 ) |
| { |
| m_strm.PutHuff( 0xF0, htable ); // encode 16 zeros |
| run -= 16; |
| } |
| |
| { |
| float a = (float)val; |
| int cat = (((int&)a >> 23) & 255) - (126 & (val ? -1 : 0)); |
| |
| assert( cat <= 10 ); |
| m_strm.PutHuff( cat + run*16, htable ); |
| m_strm.Put( val - (val < 0 ? 1 : 0), cat ); |
| } |
| |
| run = 0; |
| } |
| } |
| |
| if( run ) |
| { |
| m_strm.PutHuff( 0x00, htable ); // encode EOB |
| } |
| } |
| } |
| } |
| |
| // Flush |
| m_strm.Flush(); |
| |
| lowstrm.PutWord( 0xFFD9 ); // EOI marker |
| m_strm.Close(); |
| |
| return true; |
| } |
| |
| #endif |
| |
| /* End of file. */ |
| |
| |
