/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * Compressed jpeg
 *
 * Copyright 2012 Jay Sorg <jay.sorg@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <freerdp/config.h>

#include <winpr/stream.h>

#include <freerdp/codec/color.h>

#include <freerdp/codec/jpeg.h>

#ifdef WITH_JPEG

#define XMD_H

#include <jpeglib.h>

struct mydata_decomp
{
	char* data;
	int data_bytes;
};

/*****************************************************************************/
static void my_init_source(j_decompress_ptr cinfo)
{
}

/*****************************************************************************/
static boolean my_fill_input_buffer(j_decompress_ptr cinfo)
{
	struct mydata_decomp* md;

	md = (struct mydata_decomp*)(cinfo->client_data);
	cinfo->src->next_input_byte = (unsigned char*)(md->data);
	cinfo->src->bytes_in_buffer = md->data_bytes;
	return 1;
}

/*****************************************************************************/
static void my_skip_input_data(j_decompress_ptr cinfo, long num_bytes)
{
}

/*****************************************************************************/
static boolean my_resync_to_restart(j_decompress_ptr cinfo, int desired)
{
	return 1;
}

/*****************************************************************************/
static void my_term_source(j_decompress_ptr cinfo)
{
}

/*****************************************************************************/
static int do_decompress(char* comp_data, int comp_data_bytes, int* width, int* height, int* bpp,
                         char* decomp_data, int* decomp_data_bytes)
{
	struct jpeg_decompress_struct cinfo = { 0 };
	struct jpeg_error_mgr jerr;
	struct jpeg_source_mgr src_mgr = { 0 };
	struct mydata_decomp md = { 0 };
	JSAMPROW row_pointer[1] = { 0 };

	cinfo.err = jpeg_std_error(&jerr);
	jpeg_create_decompress(&cinfo);

	cinfo.src = &src_mgr;
	src_mgr.init_source = my_init_source;
	src_mgr.fill_input_buffer = my_fill_input_buffer;
	src_mgr.skip_input_data = my_skip_input_data;
	src_mgr.resync_to_restart = my_resync_to_restart;
	src_mgr.term_source = my_term_source;

	md.data = comp_data;
	md.data_bytes = comp_data_bytes;
	cinfo.client_data = &md;

	jpeg_read_header(&cinfo, 1);

	cinfo.out_color_space = JCS_RGB;

	*width = cinfo.image_width;
	*height = cinfo.image_height;
	*bpp = cinfo.num_components * 8;

	jpeg_start_decompress(&cinfo);

	while (cinfo.output_scanline < cinfo.image_height)
	{
		row_pointer[0] = (JSAMPROW)decomp_data;
		jpeg_read_scanlines(&cinfo, row_pointer, 1);
		decomp_data += cinfo.image_width * cinfo.num_components;
	}
	*decomp_data_bytes = cinfo.output_width * cinfo.output_height * cinfo.num_components;
	jpeg_finish_decompress(&cinfo);
	jpeg_destroy_decompress(&cinfo);
	return 0;
}

/* jpeg decompress */
BOOL jpeg_decompress(BYTE* input, BYTE* output, int width, int height, int size, int bpp)
{
	int lwidth;
	int lheight;
	int lbpp;
	int ldecomp_data_bytes;

	if (bpp != 24)
	{
		return 0;
	}
	if (do_decompress((char*)input, size, &lwidth, &lheight, &lbpp, (char*)output,
	                  &ldecomp_data_bytes) != 0)
	{
		return 0;
	}
	if (lwidth != width || lheight != height || lbpp != bpp)
	{
		return 0;
	}
	return 1;
}

#else

BOOL jpeg_decompress(BYTE* input, BYTE* output, int width, int height, int size, int bpp)
{
	return 0;
}

#endif