#include "api__jpeg.h"
#include "avi_mjpg_decoder.h"
#include "../Winamp/wa_ipc.h"
#include <limits.h>
#include <intsafe.h>
#include <setjmp.h>
extern "C"
{
#undef FAR
#include "jpeglib.h"
};
uint8_t mjpeg_header[0x1A4] =
{
/* JPEG DHT Segment for YCrCb omitted from MJPG data */
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
};
int AVIDecoderCreator::CreateVideoDecoder( const nsavi::AVIH *avi_header, const nsavi::STRH *stream_header, const nsavi::STRF *stream_format, const nsavi::STRD *stream_data, ifc_avivideodecoder **decoder )
{
nsavi::video_format *format = (nsavi::video_format *)stream_format;
if ( format )
{
if ( format->compression == 'GPJM' ) // mjpg
{
*decoder = AVIMJPEG::CreateDecoder( format );
if ( *decoder )
return CREATEDECODER_SUCCESS;
else
return CREATEDECODER_FAILURE;
}
}
return CREATEDECODER_NOT_MINE;
}
#define CBCLASS AVIDecoderCreator
START_DISPATCH;
CB( CREATE_VIDEO_DECODER, CreateVideoDecoder )
END_DISPATCH;
#undef CBCLASS
AVIMJPEG *AVIMJPEG::CreateDecoder( nsavi::video_format *stream_format )
{
AVIMJPEG *decoder = new AVIMJPEG( stream_format );
if ( !decoder )
{
return 0;
}
return decoder;
}
AVIMJPEG::AVIMJPEG( nsavi::video_format *stream_format ) : stream_format( stream_format )
{
cinfo.err = jpeg_std_error( &jerr );
jpeg_create_decompress( &cinfo );
jpegLoader = 0;
width = 0;
height = 0;
decoded_image = 0;
image_size = 0;
image_outputted = true;
}
int AVIMJPEG::GetOutputProperties( int *x, int *y, int *color_format, double *aspect_ratio, int *flip )
{
*x = stream_format->width;
*y = stream_format->height;
*color_format = '23GR'; // RGB32
*aspect_ratio = 1.0;
*flip = 0;
return AVI_SUCCESS;
}
struct mjpeg_source_mgr
{
jpeg_source_mgr src;
int state;
const JOCTET *input_buffer;
size_t input_buffer_bytes;
};
static void init_source( j_decompress_ptr cinfo )
{}
static const JOCTET jpeg_eof[] = {(JOCTET) 0xFF, (JOCTET) JPEG_EOI};
static boolean fill_input_buffer( j_decompress_ptr cinfo )
{
mjpeg_source_mgr *msrc = (mjpeg_source_mgr *)cinfo->src;
jpeg_source_mgr *src = &msrc->src;
switch ( msrc->state++ )
{
case 0:
src->next_input_byte = mjpeg_header;
src->bytes_in_buffer = sizeof( mjpeg_header );
return TRUE;
case 1:
src->next_input_byte = msrc->input_buffer + 2;
src->bytes_in_buffer = msrc->input_buffer_bytes - 2;
return TRUE;
case 2:
src->next_input_byte = jpeg_eof;
src->bytes_in_buffer = 2;
return TRUE;
}
return TRUE;
}
void skip_input_data( j_decompress_ptr cinfo, long num_bytes )
{
if ( num_bytes > 0 )
{
if ( num_bytes > (long)cinfo->src->bytes_in_buffer )
{
fill_input_buffer( cinfo );
}
else
{
cinfo->src->next_input_byte += (size_t)num_bytes;
cinfo->src->bytes_in_buffer -= (size_t)num_bytes;
}
}
}
void term_source( j_decompress_ptr cinfo )
{}
static void wasabi_jpgload_error_exit( j_common_ptr cinfo )
{
jmp_buf *stack_env = (jmp_buf *)cinfo->client_data;
longjmp( *stack_env, 1 );
}
int AVIMJPEG::DecodeChunk( uint16_t type, const void *inputBuffer, size_t inputBufferBytes )
{
mjpeg_source_mgr src = { {(const JOCTET *)inputBuffer,2,init_source,fill_input_buffer,skip_input_data,jpeg_resync_to_restart,term_source}, 0, (const JOCTET *)inputBuffer, inputBufferBytes };
cinfo.src = &src.src;
/* set up error handling. basically C style exceptions :) */
jmp_buf stack_env;
cinfo.client_data = &stack_env;
cinfo.err->error_exit = wasabi_jpgload_error_exit;
if ( setjmp( stack_env ) )
{
// longjmp will goto here
jpeg_destroy_decompress( &cinfo );
return AVI_FAILURE;
}
if ( jpeg_read_header( &cinfo, TRUE ) == JPEG_HEADER_OK )
{
cinfo.out_color_space = JCS_RGB;
jpeg_start_decompress( &cinfo );
int this_image_size = cinfo.output_width * cinfo.output_height * 4;
if ( this_image_size > image_size )
{
image_size = this_image_size;
decoded_image = malloc( image_size );
}
ARGB32 *p = (ARGB32 *)decoded_image;// + (cinfo.output_width * cinfo.output_height);
while ( cinfo.output_scanline < cinfo.output_height )
{
//p -= cinfo.output_width;
jpeg_read_scanlines( &cinfo, (JSAMPARRAY)&p, 1 );
// this is pretty homo, but it has to be done
for ( unsigned int i = 0; i < cinfo.output_width; i++ )
// full alpha shift red to correct place leave green alone shift blue to correct place
p[ i ] = ( 0xff000000 | ( ( p[ i ] & 0x0000ff ) << 16 ) | ( p[ i ] & 0x00ff00 ) | ( ( p[ i ] & 0xff0000 ) >> 16 ) );
p += cinfo.output_width;
}
//if (w) *w = cinfo.output_width;
//if (h) *h = cinfo.output_height;
jpeg_finish_decompress( &cinfo );
}
//jpeg_destroy_decompress(&cinfo);
image_outputted = false;
return AVI_SUCCESS;
}
void AVIMJPEG::Flush()
{
}
int AVIMJPEG::GetPicture( void **data, void **decoder_data )
{
if ( image_outputted )
return AVI_FAILURE;
*data = decoded_image;
*decoder_data = 0;
image_outputted = true;
return AVI_SUCCESS;
}
void AVIMJPEG::FreePicture( void *data, void *decoder_data )
{
}
void AVIMJPEG::Close()
{
delete jpegLoader;
delete this;
}
#define CBCLASS AVIMJPEG
START_DISPATCH;
CB( GET_OUTPUT_PROPERTIES, GetOutputProperties )
CB( DECODE_CHUNK, DecodeChunk )
VCB( FLUSH, Flush )
VCB( CLOSE, Close )
CB( GET_PICTURE, GetPicture )
VCB( FREE_PICTURE, FreePicture )
END_DISPATCH;
#undef CBCLASS