mirror of
https://github.com/CloudCompare/PoissonRecon.git
synced 2026-08-29 08:34:27 +08:00
mkazhdan
131 lines
3.6 KiB
C++
131 lines
3.6 KiB
C++
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
|
|
inline METHODDEF( void )
|
|
my_error_exit (j_common_ptr cinfo)
|
|
{
|
|
// cinfo->err really points to a my_error_mgr struct, so coerce pointer
|
|
my_error_ptr myerr = (my_error_ptr) cinfo->err;
|
|
|
|
// Always display the message.
|
|
// We could postpone this until after returning, if we chose.
|
|
(*cinfo->err->output_message) (cinfo);
|
|
|
|
// Return control to the setjmp point
|
|
longjmp(myerr->setjmp_buffer, 1);
|
|
}
|
|
|
|
inline bool JPEGReader::GetInfo( const char* fileName , unsigned int& width , unsigned int& height , unsigned int& channels )
|
|
{
|
|
FILE* fp = fopen( fileName , "rb" );
|
|
if( !fp ) ERROR_OUT( "Failed to open: " , fileName );
|
|
|
|
struct jpeg_decompress_struct cInfo;
|
|
struct my_error_mgr jErr;
|
|
|
|
cInfo.err = jpeg_std_error( &jErr.pub );
|
|
jErr.pub.error_exit = my_error_exit;
|
|
if( setjmp( jErr.setjmp_buffer ) )
|
|
{
|
|
jpeg_destroy_decompress( &cInfo );
|
|
ERROR_OUT( "JPEG error occured" );
|
|
}
|
|
|
|
jpeg_create_decompress( &cInfo );
|
|
jpeg_stdio_src( &cInfo , fp );
|
|
|
|
(void) jpeg_read_header( &cInfo , TRUE );
|
|
|
|
channels = cInfo.num_components;
|
|
width = cInfo.image_width;
|
|
height = cInfo.image_height;
|
|
jpeg_destroy_decompress( &cInfo );
|
|
|
|
fclose( fp );
|
|
return true;
|
|
}
|
|
|
|
inline JPEGReader::JPEGReader( const char* fileName , unsigned int& width , unsigned int& height , unsigned int& channels )
|
|
{
|
|
_currentRow = 0;
|
|
_fp = fopen( fileName , "rb" );
|
|
if( !_fp ) ERROR_OUT( "Failed to open: " , fileName );
|
|
|
|
_cInfo.err = jpeg_std_error( &_jErr.pub );
|
|
_jErr.pub.error_exit = my_error_exit;
|
|
if( setjmp( _jErr.setjmp_buffer ) )
|
|
{
|
|
jpeg_destroy_decompress( &_cInfo );
|
|
ERROR_OUT( "JPEG error occured" );
|
|
}
|
|
|
|
jpeg_create_decompress( &_cInfo );
|
|
jpeg_stdio_src( &_cInfo , _fp );
|
|
|
|
(void) jpeg_read_header( &_cInfo , TRUE );
|
|
(void) jpeg_start_decompress( &_cInfo );
|
|
|
|
channels = _cInfo.output_components;
|
|
width = _cInfo.output_width;
|
|
height = _cInfo.output_height;
|
|
}
|
|
inline JPEGReader::~JPEGReader( void )
|
|
{
|
|
(void) jpeg_finish_decompress( &_cInfo );
|
|
jpeg_destroy_decompress( &_cInfo );
|
|
fclose( _fp );
|
|
}
|
|
inline unsigned int JPEGReader::nextRow( unsigned char* row )
|
|
{
|
|
JSAMPROW row_pointers[1];
|
|
row_pointers[0] = row;
|
|
jpeg_read_scanlines( &_cInfo , row_pointers, 1 );
|
|
return _currentRow++;
|
|
}
|
|
|
|
inline JPEGWriter::JPEGWriter( const char* fileName , unsigned int width , unsigned int height , unsigned int channels , unsigned int quality )
|
|
{
|
|
_currentRow = 0;
|
|
_fp = fopen( fileName , "wb" );
|
|
if( !_fp ) ERROR_OUT( "Failed to open: " , fileName );
|
|
|
|
_cInfo.err = jpeg_std_error( &_jErr.pub );
|
|
jpeg_create_compress( &_cInfo );
|
|
|
|
jpeg_stdio_dest( &_cInfo , _fp );
|
|
|
|
_cInfo.image_width = width;
|
|
_cInfo.image_height = height;
|
|
_cInfo.input_components = channels;
|
|
_cInfo.in_color_space = JCS_RGB; /* colorspace of input image */
|
|
|
|
jpeg_set_defaults( &_cInfo );
|
|
jpeg_set_quality( &_cInfo , quality , TRUE );
|
|
|
|
jpeg_start_compress( &_cInfo , TRUE );
|
|
}
|
|
inline JPEGWriter::~JPEGWriter( void )
|
|
{
|
|
jpeg_finish_compress( &_cInfo );
|
|
jpeg_destroy_compress( &_cInfo );
|
|
fclose( _fp );
|
|
}
|
|
inline unsigned int JPEGWriter::nextRow( const unsigned char* row )
|
|
{
|
|
JSAMPROW row_pointer[1];
|
|
row_pointer[0] = ( unsigned char* )row;
|
|
(void) jpeg_write_scanlines( &_cInfo , row_pointer , 1 );
|
|
return _currentRow++;
|
|
}
|
|
|
|
inline unsigned int JPEGWriter::nextRows( const unsigned char* rows , unsigned int rowNum )
|
|
{
|
|
JSAMPROW* row_pointers = new JSAMPROW[ rowNum ];
|
|
for( unsigned int r=0 ; r<rowNum ; r++ ) row_pointers[r] = (unsigned char*)( rows + r * 3 * sizeof( unsigned char ) * _cInfo.image_width );
|
|
(void) jpeg_write_scanlines( &_cInfo , row_pointers , rowNum );
|
|
delete[] row_pointers;
|
|
return _currentRow += rowNum;
|
|
}
|
|
|