// Filename:-	JPGFile.cpp

#include "stdafx.h"
#include "oddbits.h"
#include "jpgfile.h"
#include "..\libs\pakstuff.h"

#include "jpeg-6\jpeglib.h"

// bool return really, but needs an int to match header proto which is included by a C file, and MS C compiler 
//	doesn't like bool types
//
int LoadJPG( const char *filename, unsigned char **pic, int *width, int *height ) 
{
	bool bReturn = true;
	*pic = NULL;

	/* This struct contains the JPEG decompression parameters and pointers to
   * working space (which is allocated as needed by the JPEG library).
   */
  struct jpeg_decompress_struct cinfo;
  /* We use our private extension JPEG error handler.
   * Note that this struct must live as long as the main JPEG parameter
   * struct, to avoid dangling-pointer problems.
   */
  /* This struct represents a JPEG error handler.  It is declared separately
   * because applications often want to supply a specialized error handler
   * (see the second half of this file for an example).  But here we just
   * take the easy way out and use the standard error handler, which will
   * print a message on stderr and call exit() if compression fails.
   * Note that this struct must live as long as the main JPEG parameter
   * struct, to avoid dangling-pointer problems.
   */
  struct jpeg_error_mgr jerr;
  /* More stuff */
  JSAMPARRAY buffer;		/* Output row buffer */
  int row_stride;		/* physical row width in output buffer */
  unsigned char *out;
  byte	*fbuffer;
  byte  *bbuf;

  /* In this example we want to open the input file before doing anything else,
   * so that the setjmp() error recovery below can assume the file is open.
   * VERY IMPORTANT: use "b" option to fopen() if you are on a machine that
   * requires it in order to read binary files.
   */

//stefix
#if 0
  ri.FS_ReadFile ( ( char * ) filename, (void **)&fbuffer);
  if (!fbuffer) 
	return false;
#else
	int iLoadedLen = LoadFile (filename, (void **)&fbuffer);
	if (iLoadedLen == -1)
	{
		iLoadedLen = PakLoadAnyFile(filename, (void **)&fbuffer);
		if (iLoadedLen == -1)
		{
			return false;
		}
	}
#endif


	try
	{

		  /* Step 1: allocate and initialize JPEG decompression object */

		  /* We have to set up the error handler first, in case the initialization
		   * step fails.  (Unlikely, but it could happen if you are out of memory.)
		   * This routine fills in the contents of struct jerr, and returns jerr's
		   * address which we place into the link field in cinfo.
		   */
		  cinfo.err = jpeg_std_error(&jerr);

		  /* Now we can initialize the JPEG decompression object. */
		  jpeg_create_decompress(&cinfo);

		  /* Step 2: specify data source (eg, a file) */

		  jpeg_stdio_src(&cinfo, fbuffer);

		  /* Step 3: read file parameters with jpeg_read_header() */

		  (void) jpeg_read_header(&cinfo, TRUE);
		  /* We can ignore the return value from jpeg_read_header since
		   *   (a) suspension is not possible with the stdio data source, and
		   *   (b) we passed TRUE to reject a tables-only JPEG file as an error.
		   * See libjpeg.doc for more info.
		   */

		  /* Step 4: set parameters for decompression */

		  /* In this example, we don't need to change any of the defaults set by
		   * jpeg_read_header(), so we do nothing here.
		   */

		  /* Step 5: Start decompressor */

		  (void) jpeg_start_decompress(&cinfo);
		  /* We can ignore the return value since suspension is not possible
		   * with the stdio data source.
		   */

		  /* We may need to do some setup of our own at this point before reading
		   * the data.  After jpeg_start_decompress() we have the correct scaled
		   * output image dimensions available, as well as the output colormap
		   * if we asked for color quantization.
		   * In this example, we need to make an output work buffer of the right size.
		   */ 
		  /* JSAMPLEs per row in output buffer */
		  row_stride = cinfo.output_width * cinfo.output_components;

		  if (cinfo.output_components!=4 && cinfo.output_components!=1 ) 
		  {
			throw(va("JPG %s is unsupported color depth (%d)\n", filename, cinfo.output_components));
		  }
		  out = (byte *)malloc(cinfo.output_width*cinfo.output_height*4);

		  *pic = out;
		  *width = cinfo.output_width;
		  *height = cinfo.output_height;

		  /* Step 6: while (scan lines remain to be read) */
		  /*           jpeg_read_scanlines(...); */

		  /* Here we use the library's state variable cinfo.output_scanline as the
		   * loop counter, so that we don't have to keep track ourselves.
		   */
		  while (cinfo.output_scanline < cinfo.output_height) {
			/* jpeg_read_scanlines expects an array of pointers to scanlines.
			 * Here the array is only one element long, but you could ask for
			 * more than one scanline at a time if that's more convenient.
			 */
			bbuf = ((out+(row_stride*cinfo.output_scanline)));
			buffer = &bbuf;
			(void) jpeg_read_scanlines(&cinfo, buffer, 1);
		  }

		  // if we've just loaded a greyscale, then adjust it from 8-bit to 32bit by stretch-copying it over itself...
		  //  (this also does the alpha stuff as well)
		  //
		  if (cinfo.output_components == 1)
		  {
			  byte *pbDest = (*pic + (cinfo.output_width * cinfo.output_height * 4))-1;
			  byte *pbSrc  = (*pic + (cinfo.output_width * cinfo.output_height    ))-1;
			  int  iPixels = cinfo.output_width * cinfo.output_height;

			  for (int i=0; i<iPixels; i++)
			  {
				  byte b = *pbSrc--;
				  *pbDest-- = 255;
				  *pbDest-- = b;
				  *pbDest-- = b;
				  *pbDest-- = b;
			  }
		  }
		  else	  
		  {// clear all the alphas to 255
			  int	i, j;
				byte	*buf;

				buf = *pic;

			  j = cinfo.output_width * cinfo.output_height * 4;
			  for ( i = 3 ; i < j ; i+=4 ) {
				  buf[i] = 255;
			  }
		  }

		  /* Step 7: Finish decompression */

		  (void) jpeg_finish_decompress(&cinfo);
		  /* We can ignore the return value since suspension is not possible
		   * with the stdio data source.
		   */

		  /* Step 8: Release JPEG decompression object */

		  /* This is an important step since it will release a good deal of memory. */
		  jpeg_destroy_decompress(&cinfo);
	}

	catch(LPCSTR psMessage)
	{			
		//ErrorBox
		Sys_Printf(va("JPEG read error: %s",psMessage));
		jpeg_destroy_decompress(&cinfo);
		bReturn = false;
		if (*pic)
		{
			free(*pic);
			*pic = NULL;
		}
	}

  /* After finish_decompress, we can close the input file.
   * Here we postpone it until after no more JPEG errors are possible,
   * so as to simplify the setjmp error logic above.  (Actually, I don't
   * think that jpeg_destroy can do an error exit, but why assume anything...)
   */
//stefix
#if 0
  ri.FS_FreeFile (fbuffer);
#else
  free(fbuffer);
#endif

  /* At this point you may want to check to see whether any corrupt-data
   * warnings occurred (test whether jerr.pub.num_warnings is nonzero).
   */

  /* And we're done! */

  return bReturn;
}


void JPG_ErrorThrow(LPCSTR message)
{
	throw (message);
}

void JPG_MessageOut(LPCSTR message)
{
	OutputDebugString(va("%s\n",message));
}



//////////////// eof ////////////