#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