/*
 * CINELERRA
 * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include "asset.h"
#include "bcsignals.h"
#include "edit.h"
#include "file.h"
#include "filejpeg.h"
#include "interlacemodes.h"
#include "jpegwrapper.h"
#include "language.h"
#include "libmjpeg.h"
#include "mwindow.inc"
#include "vframe.h"
#include "videodevice.inc"
#include "mainerror.h"


FileJPEG::FileJPEG(Asset *asset, File *file)
 : FileList(asset, file, "JPEGLIST", ".jpg", FILE_JPEG, FILE_JPEG_LIST)
{
	decompressor = 0;
}

FileJPEG::~FileJPEG()
{
	if(decompressor) mjpeg_delete((mjpeg_t*)decompressor);
}


int FileJPEG::check_sig(Asset *asset)
{
	FILE *fp = fopen(asset->path, "r");
	if( !fp ) return 0;
	char test[10];
	int result = -1;
	if( fread(test, 1, sizeof(test), fp) == sizeof(test) ) {
		if( test[6] == 'J' && test[7] == 'F' && test[8] == 'I' && test[9] == 'F' ) {
			fseek(fp, 0, SEEK_SET);
			int w = 0, h = 0;
			result = read_header(fp, w, h);
		}
		else if(test[0] == 'J' && test[1] == 'P' && test[2] == 'E' && test[3] == 'G' &&
			test[4] == 'L' && test[5] == 'I' && test[6] == 'S' && test[7] == 'T') {
			result = 0;
		}
	}
	fclose(fp);

	if( result < 0 ) {
		int i = strlen(asset->path) - 4;
		if( i >= 0 && !strcasecmp(asset->path+i, ".jpg") )
			result = 0;
	}
	return !result ? 1 : 0;
}


void FileJPEG::get_parameters(BC_WindowBase *parent_window,
	Asset *asset, BC_WindowBase* &format_window,
	int audio_options, int video_options, EDL *edl)
{
	if(video_options)
	{
		JPEGConfigVideo *window = new JPEGConfigVideo(parent_window, asset);
		format_window = window;
		window->create_objects();
		window->run_window();
		delete window;
	}
}


int FileJPEG::can_copy_from(Asset *asset, int64_t position)
{
//printf("FileJPEG::can_copy_from %d %s\n", asset->format, asset->vcodec);
	if(asset->format == FILE_JPEG ||
		asset->format == FILE_JPEG_LIST)
		return 1;

	return 0;
}

int FileJPEG::colormodel_supported(int colormodel)
{
	return colormodel;
}


int FileJPEG::get_best_colormodel(Asset *asset, int driver)
{
	switch(driver)
	{
		case PLAYBACK_X11:
			return BC_RGB888;
			break;
		case PLAYBACK_X11_XV:
		case PLAYBACK_DV1394:
		case PLAYBACK_FIREWIRE:
		case PLAYBACK_ASYNCHRONOUS:
			return BC_YUV420P;
			break;
		case PLAYBACK_X11_GL:
			return BC_YUV888;
			break;
		case VIDEO4LINUX2:
			return BC_YUV420P;
			break;
		case VIDEO4LINUX2JPEG:
			return BC_YUV422;
			break;
		case CAPTURE_FIREWIRE:
		case CAPTURE_IEC61883:
			return BC_YUV420P;
			break;
		case CAPTURE_DVB:
		case VIDEO4LINUX2MPEG:
			return BC_YUV422P;
			break;
	}
	return BC_YUV420P;
}


int FileJPEG::write_frame(VFrame *frame, VFrame *data, FrameWriterUnit *unit)
{
	int result = 0;
	JPEGUnit *jpeg_unit = (JPEGUnit*)unit;

	if(!jpeg_unit->compressor)
		jpeg_unit->compressor = mjpeg_new(asset->width,
			asset->height,
			1);

	mjpeg_set_quality((mjpeg_t*)jpeg_unit->compressor, asset->jpeg_quality);


	mjpeg_compress((mjpeg_t*)jpeg_unit->compressor,
		frame->get_rows(),
		frame->get_y(),
		frame->get_u(),
		frame->get_v(),
		frame->get_color_model(),
		1);

// insert spherical tag
	if(asset->jpeg_sphere)
	{
		const char *sphere_tag = 
			"http://ns.adobe.com/xap/1.0/\x00<?xpacket begin='\xef\xbb\xbf' id='W5M0MpCehiHzreSzNTczkc9d'?>\n"
			"<x:xmpmeta xmlns:x='adobe:ns:meta/' x:xmptk='Image::Cinelerra'>\n"
			"<rdf:RDF xmlns:rdf='http://www.w3.org/1999/02/22-rdf-syntax-ns#'>\n"
			"\n"
			" <rdf:Description rdf:about=''\n"
			"  xmlns:GPano='http://ns.google.com/photos/1.0/panorama/'>\n"
			"  <GPano:ProjectionType>equirectangular</GPano:ProjectionType>\n"
			" </rdf:Description>\n"
			"</rdf:RDF>\n"
			"</x:xmpmeta>\n"
			"<?xpacket end='w'?>";

// calculate length by skipping the \x00 byte
		int skip = 32;
		int tag_len = strlen(sphere_tag + skip) + skip;
		int tag_len2 = tag_len + 2;
		int tag_len3 = tag_len + 4;
		
		data->allocate_compressed_data(
			mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor) + tag_len3);
		data->set_compressed_size(
			mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor) + tag_len3);
			
		int jfif_size = 0x14;
		uint8_t *ptr = data->get_data();
		memcpy(ptr, 
			mjpeg_output_buffer((mjpeg_t*)jpeg_unit->compressor), 
			jfif_size);
		ptr += jfif_size;
		*ptr++ = 0xff;
		*ptr++ = 0xe1;
		*ptr++ = (tag_len2 >> 8) & 0xff;
		*ptr++ = tag_len2 & 0xff;
		memcpy(ptr,
			sphere_tag,
			tag_len);
		ptr += tag_len;
		memcpy(ptr,
			mjpeg_output_buffer((mjpeg_t*)jpeg_unit->compressor) + jfif_size,
			mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor) - jfif_size);
	}
	else
	{
		data->allocate_compressed_data(mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));
		data->set_compressed_size(mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));
		memcpy(data->get_data(), 
			mjpeg_output_buffer((mjpeg_t*)jpeg_unit->compressor), 
			mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));
	}
	data->allocate_compressed_data(mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));
	data->set_compressed_size(mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));
	memcpy(data->get_data(),
		mjpeg_output_buffer((mjpeg_t*)jpeg_unit->compressor),
		mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));

	return result;
}










int FileJPEG::read_frame_header(char *path)
{
	FILE *fp = fopen(path, "rb");
	if( !fp ) {
		eprintf("FileJPEG::read_frame_header %s: %m\n", path);
		return 1;
	}
	int w = 0, h = 0, result = 1;
	unsigned char test[2];
	if( fread(test, 1, sizeof(test), fp) == sizeof(test) &&
	    test[0] == 0xff && test[1] == 0xd8 ) {
		fseek(fp, 0, SEEK_SET);
		result = read_header(fp, w, h);
	}
	fclose(fp);
	if( !result ) {
		asset->width = w;  asset->height = h;
		asset->interlace_mode = ILACE_MODE_NOTINTERLACED;
	}
	else
		eprintf("FileJPEG::read_frame_header %s bad header\n", path);
	return result;
}

int FileJPEG::read_header(FILE *fp, int &w, int &h)
{
	int result = 0;
	struct jpeg_error_mgr jpeg_error;
	struct jpeg_decompress_struct jpeg_decompress;
	jpeg_decompress.err = jpeg_std_error(&jpeg_error);
	jpeg_create_decompress(&jpeg_decompress);
	jpeg_stdio_src(&jpeg_decompress, fp);
	if( jpeg_read_header(&jpeg_decompress, TRUE) != JPEG_HEADER_OK ) result = 1;
	if( !result && jpeg_decompress.jpeg_color_space != JCS_YCbCr ) result = 1;
	if( !result && jpeg_decompress.comp_info[0].h_samp_factor > 2 ) result = 1;
	if( !result && jpeg_decompress.comp_info[0].v_samp_factor > 2 ) result = 1;
	if( !result ) {
		w = jpeg_decompress.image_width;
		h = jpeg_decompress.image_height;
	}
	jpeg_destroy((j_common_ptr)&jpeg_decompress);
	return result;
}


int FileJPEG::read_frame(VFrame *output, VFrame *input)
{
	if( input->get_compressed_size() < 2 ||
	    input->get_data()[0] != 0xff ||
	    input->get_data()[1] != 0xd8 )
		return 1;

	if( !decompressor )
		decompressor = mjpeg_new(asset->width, asset->height, 1);
// printf("FileJPEG::read_frame %d %p %d %d %d %p %p %p %p %d\n", __LINE__,
// input->get_data(), input->get_compressed_size(), output->get_w(), output->get_h(),
// output->get_rows(), output->get_y(), output->get_u(), output->get_v(),
// output->get_color_model());
	mjpeg_decompress((mjpeg_t*)decompressor, input->get_data(), input->get_compressed_size(), 0,
		output->get_rows(), output->get_y(), output->get_u(), output->get_v(),
		output->get_color_model(), 1);
//printf("FileJPEG::read_frame %d\n", __LINE__);
//
	return 0;
}

FrameWriterUnit* FileJPEG::new_writer_unit(FrameWriter *writer)
{
	return new JPEGUnit(this, writer);
}






JPEGUnit::JPEGUnit(FileJPEG *file, FrameWriter *writer)
 : FrameWriterUnit(writer)
{
	this->file = file;
	compressor = 0;
}
JPEGUnit::~JPEGUnit()
{
	if(compressor) mjpeg_delete((mjpeg_t*)compressor);
}







JPEGConfigVideo::JPEGConfigVideo(BC_WindowBase *parent_window, Asset *asset)
 : BC_Window(_(PROGRAM_NAME ": Video Compression"),
 	parent_window->get_abs_cursor_x(1),
 	parent_window->get_abs_cursor_y(1),
	xS(400), yS(200))
{
	this->parent_window = parent_window;
	this->asset = asset;
}

JPEGConfigVideo::~JPEGConfigVideo()
{
}

void JPEGConfigVideo::create_objects()
{
	int xs10 = xS(10), ys10 = yS(10);
	int x = xs10, y = ys10;
	lock_window("JPEGConfigVideo::create_objects");
	add_subwindow(new BC_Title(x, y, _("Quality:")));
	BC_ISlider *slider;
	add_subwindow(slider = new BC_ISlider(x + xS(80), y,
		0, xS(200), xS(200), 0, xS(100), asset->jpeg_quality, 0, 0,
		&asset->jpeg_quality));
	y += slider->get_h() + ys10;
	add_subwindow(new BC_CheckBox(x, y, 
		&asset->jpeg_sphere, _("Tag for spherical playback")));

	add_subwindow(new BC_OKButton(this));
	show_window(1);
	unlock_window();
}

int JPEGConfigVideo::close_event()
{
	set_done(0);
	return 1;
}