/*
 * 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 "clip.h"
#include "bccmodels.h"
#include "effecttv.h"
#include "filexml.h"
#include "holo.h"
#include "holowindow.h"
#include "language.h"
#include "bccolors.h"

#include <stdint.h>
#include <stdio.h>
#include <string.h>


REGISTER_PLUGIN(HoloMain)










HoloConfig::HoloConfig()
{
	threshold = 40;
	recycle = 1.0;
}





HoloMain::HoloMain(PluginServer *server)
 : PluginVClient(server)
{
	effecttv = 0;
	bgimage = 0;
	do_reconfigure = 1;
}

HoloMain::~HoloMain()
{
	if(effecttv) {
		delete holo_server;
		delete effecttv;
	}

	if(bgimage)
		delete bgimage;
}

const char* HoloMain::plugin_title() { return N_("HolographicTV"); }
int HoloMain::is_realtime() { return 1; }

NEW_WINDOW_MACRO(HoloMain, HoloWindow)

int HoloMain::load_configuration()
{
	return 0;
}


void HoloMain::save_data(KeyFrame *keyframe)
{
}

void HoloMain::read_data(KeyFrame *keyframe)
{
}

void HoloMain::reconfigure()
{
	do_reconfigure = 0;

	effecttv->image_set_threshold_y(config.threshold);
}


#define ADD_FRAMES(type, components) \
{ \
	type **input_rows = (type**)input->get_rows(); \
	type **output_rows = (type**)output->get_rows(); \
	int w = input->get_w(); \
	int h = input->get_h(); \
 \
	for(int i = 0; i < h; i++) \
	{ \
		type *output_row = (type*)output_rows[i]; \
		type *input_row = (type*)input_rows[i]; \
 \
		for(int j = 0; j < w; j++) \
		{ \
			for(int k = 0; k < 3; k++) \
			{ \
				if(sizeof(type) == 4) \
				{ \
					int in_temp = (int)(*input_row * 0xffff); \
					int out_temp = (int)(*output_row * 0xffff); \
					int temp = (in_temp & out_temp) + \
						((in_temp ^ out_temp) >> 1); \
					*output_row = (type)temp / 0xffff; \
				} \
				else \
				{ \
					*output_row = ((uint16_t)*input_row & (uint16_t)*output_row) + \
						(((uint16_t)*input_row ^ (uint16_t)*output_row) >> 1); \
				} \
				output_row++; \
				input_row++; \
			} \
 \
			if(components == 4) \
			{ \
				output_row++; \
				input_row++; \
			} \
		} \
	} \
}


// Add input to output and store result in output
void HoloMain::add_frames(VFrame *output, VFrame *input)
{
	switch(output->get_color_model())
	{
		case BC_RGB888:
		case BC_YUV888:
			ADD_FRAMES(uint8_t, 3);
			break;
		case BC_RGB_FLOAT:
			ADD_FRAMES(float, 3);
			break;
		case BC_RGBA_FLOAT:
			ADD_FRAMES(float, 4);
			break;
		case BC_RGBA8888:
		case BC_YUVA8888:
			ADD_FRAMES(uint8_t, 4);
			break;
		case BC_RGB161616:
		case BC_YUV161616:
			ADD_FRAMES(uint16_t, 3);
			break;
		case BC_RGBA16161616:
		case BC_YUVA16161616:
			ADD_FRAMES(uint16_t, 4);
			break;
	}
}

void HoloMain::set_background()
{
/*
 * grab 4 frames and composite them to get a quality background image
 */
/**
 * For Cinelerra, we make every frame a holograph and expect the user to
 * provide a matte.
 **/
total = 0;

	switch(total)
	{
		case 0:
/* step 1: grab frame-1 to buffer-1 */
// 			tmp = new VFrame(input_ptr->get_w(),input_ptr->get_h(),
// 				project_color_model, 0);
			bgimage->copy_from(input_ptr);
			break;

		case 1:
/* step 2: add frame-2 to buffer-1 */
			add_frames(bgimage, input_ptr);
			break;

		case 2:
/* step 3: grab frame-3 to buffer-2 */
			tmp->copy_from(input_ptr);
			break;

		case 3:
/* step 4: add frame-4 to buffer-2 */
			add_frames(tmp, input_ptr);



/* step 5: add buffer-3 to buffer-1 */
			add_frames(bgimage, tmp);

			effecttv->image_bgset_y(bgimage);


			delete tmp;
			break;
	}
}


int HoloMain::process_realtime(VFrame *input_ptr, VFrame *output_ptr)
{
	this->input_ptr = input_ptr;
	this->output_ptr = output_ptr;




	load_configuration();



	if(do_reconfigure)
	{
		if(!effecttv)
		{
			effecttv = new EffectTV(input_ptr->get_w(), input_ptr->get_h());
			bgimage = new VFrame(input_ptr->get_w(), input_ptr->get_h(),
				input_ptr->get_color_model(), 0);

			for(int i = 0; i < 256; i++)
			{
				noisepattern[i] = (i * i * i / 40000)* i / 256;
			}

			holo_server = new HoloServer(this, 1, 1);
		}

		reconfigure();
	}

	set_background();

	holo_server->process_packages();

	total++;
	if(total >= config.recycle * project_frame_rate)
		total = 0;

	return 0;
}





HoloServer::HoloServer(HoloMain *plugin, int total_clients, int total_packages)
 : LoadServer(total_clients, total_packages)
{
	this->plugin = plugin;
}


LoadClient* HoloServer::new_client()
{
	return new HoloClient(this);
}




LoadPackage* HoloServer::new_package()
{
	return new HoloPackage;
}



void HoloServer::init_packages()
{
	for(int i = 0; i < get_total_packages(); i++)
	{
		HoloPackage *package = (HoloPackage*)get_package(i);
		package->row1 = plugin->input_ptr->get_h() * i / get_total_packages();
		package->row2 = plugin->input_ptr->get_h() * (i + 1) / get_total_packages();
	}
}








HoloClient::HoloClient(HoloServer *server)
 : LoadClient(server)
{
	this->plugin = server->plugin;
	phase = 0;
}


void HoloClient::process_package(LoadPackage *package)
{
	int x, y;
	HoloPackage *local_package = (HoloPackage*)package;
	unsigned char **input_rows = plugin->input_ptr->get_rows() + local_package->row1;
	unsigned char **output_rows = plugin->output_ptr->get_rows() + local_package->row1;
	int width = plugin->input_ptr->get_w();
	int height = local_package->row2 - local_package->row1;
	unsigned char *diff;
	uint32_t s, t;
	int r, g, b;

	diff = plugin->effecttv->image_diff_filter(plugin->effecttv->image_bgsubtract_y(input_rows,
		plugin->input_ptr->get_color_model()));

	diff += width;
	output_rows++;
	input_rows++;


// Convert discrete channels to a single 24 bit number
#define STORE_PIXEL(type, components, dest, src, is_yuv) \
if(sizeof(type) == 4) \
{ \
	int r = (int)(src[0] * 0xff); \
	int g = (int)(src[1] * 0xff); \
	int b = (int)(src[2] * 0xff); \
	CLAMP(r, 0, 0xff); \
	CLAMP(g, 0, 0xff); \
	CLAMP(b, 0, 0xff); \
	dest = (r << 16) | (g << 8) | b; \
} \
else \
if(sizeof(type) == 2) \
{ \
	if(is_yuv) \
	{ \
		int r = (int)src[0] >> 8; \
		int g = (int)src[1] >> 8; \
		int b = (int)src[2] >> 8; \
		YUV::yuv.yuv_to_rgb_8(r, g, b); \
		dest = (r << 16) | (g << 8) | b; \
	} \
	else \
	{ \
		dest = (((uint32_t)src[0] << 8) & 0xff0000) | \
			((uint32_t)src[1] & 0xff00) | \
			((uint32_t)src[2]) >> 8; \
	} \
} \
else \
{ \
	if(is_yuv) \
	{ \
		int r = (int)src[0]; \
		int g = (int)src[1]; \
		int b = (int)src[2]; \
		YUV::yuv.yuv_to_rgb_8(r, g, b); \
		dest = (r << 16) | (g << 8) | b; \
	} \
	else \
	{ \
		dest = ((uint32_t)src[0] << 16) | \
			((uint32_t)src[1] << 8) | \
			(uint32_t)src[2]; \
	} \
}




#define HOLO_CORE(type, components, is_yuv) \
	for(y = 1; y < height - 1; y++) \
	{ \
		type *src = (type*)input_rows[y]; \
		type *bg = (type*)plugin->bgimage->get_rows()[y]; \
		type *dest = (type*)output_rows[y]; \
 \
 \
 \
		if(((y + phase) & 0x7f) < 0x58)  \
		{ \
			for(x = 0 ; x < width; x++)  \
			{ \
				if(*diff) \
				{ \
					STORE_PIXEL(type, components, s, src, is_yuv); \
 \
					t = (s & 0xff) +  \
						((s & 0xff00) >> 7) +  \
						((s & 0xff0000) >> 16); \
					t += plugin->noisepattern[EffectTV::fastrand() >> 24]; \
 \
					r = ((s & 0xff0000) >> 17) + t; \
					g = ((s & 0xff00) >> 8) + t; \
					b = (s & 0xff) + t; \
 \
					r = (r >> 1) - 100; \
					g = (g >> 1) - 100; \
					b = b >> 2; \
 \
					if(r < 20) r = 20; \
					if(g < 20) g = 20; \
 \
					STORE_PIXEL(type, components, s, bg, is_yuv); \
 \
					r += (s & 0xff0000) >> 17; \
					g += (s & 0xff00) >> 9; \
					b += ((s & 0xff) >> 1) + 40; \
 \
					if(r > 255) r = 255; \
					if(g > 255) g = 255; \
					if(b > 255) b = 255; \
 \
 					if(is_yuv) YUV::yuv.rgb_to_yuv_8(r, g, b); \
					if(sizeof(type) == 4) \
					{ \
						dest[0] = (type)r / 0xff; \
						dest[1] = (type)g / 0xff; \
						dest[2] = (type)b / 0xff; \
					} \
					else \
					if(sizeof(type) == 2) \
					{ \
						dest[0] = (r << 8) | r; \
						dest[1] = (g << 8) | g; \
						dest[2] = (b << 8) | b; \
					} \
					else \
					{ \
						dest[0] = r; \
						dest[1] = g; \
						dest[2] = b; \
					} \
				}  \
				else  \
				{ \
					dest[0] = bg[0]; \
					dest[1] = bg[1]; \
					dest[2] = bg[2]; \
				} \
 \
				diff++; \
				src += components; \
				dest += components; \
				bg += components; \
			} \
		}  \
		else  \
		{ \
			for(x = 0; x < width; x++) \
			{ \
				if(*diff) \
				{ \
					STORE_PIXEL(type, components, s, src, is_yuv); \
 \
 \
					t = (s & 0xff) + ((s & 0xff00) >> 6) + ((s & 0xff0000) >> 16); \
					t += plugin->noisepattern[EffectTV::fastrand() >> 24]; \
 \
					r = ((s & 0xff0000) >> 16) + t; \
					g = ((s & 0xff00) >> 8) + t; \
					b = (s & 0xff) + t; \
 \
					r = (r >> 1) - 100; \
					g = (g >> 1) - 100; \
					b = b >> 2; \
 \
					if(r < 0) r = 0; \
					if(g < 0) g = 0; \
 \
					STORE_PIXEL(type, components, s, bg, is_yuv); \
 \
					r += ((s & 0xff0000) >> 17) + 10; \
					g += ((s & 0xff00) >> 9) + 10; \
					b += ((s & 0xff) >> 1) + 40; \
 \
					if(r > 255) r = 255; \
					if(g > 255) g = 255; \
					if(b > 255) b = 255; \
 \
 					if(is_yuv) YUV::yuv.rgb_to_yuv_8(r, g, b); \
					if(sizeof(type) == 4) \
					{ \
						dest[0] = (type)r / 0xff; \
						dest[1] = (type)g / 0xff; \
						dest[2] = (type)b / 0xff; \
					} \
					else \
					if(sizeof(type) == 2) \
					{ \
						dest[0] = (r << 8) | r; \
						dest[1] = (g << 8) | g; \
						dest[2] = (b << 8) | b; \
					} \
					else \
					{ \
						dest[0] = r; \
						dest[1] = g; \
						dest[2] = b; \
					} \
				}  \
				else  \
				{ \
					dest[0] = bg[0]; \
					dest[1] = bg[1]; \
					dest[2] = bg[2]; \
				} \
 \
				diff++; \
				src += components; \
				dest += components; \
				bg += components; \
			} \
		} \
	}




	switch(plugin->input_ptr->get_color_model())
	{
		case BC_RGB888:
			HOLO_CORE(uint8_t, 3, 0);
			break;
		case BC_RGB_FLOAT:
			HOLO_CORE(float, 3, 0);
			break;
		case BC_YUV888:
			HOLO_CORE(uint8_t, 3, 1);
			break;
		case BC_RGBA_FLOAT:
			HOLO_CORE(float, 4, 0);
			break;
		case BC_RGBA8888:
			HOLO_CORE(uint8_t, 4, 0);
			break;
		case BC_YUVA8888:
			HOLO_CORE(uint8_t, 4, 1);
			break;
		case BC_RGB161616:
			HOLO_CORE(uint16_t, 3, 0);
			break;
		case BC_YUV161616:
			HOLO_CORE(uint16_t, 3, 1);
			break;
		case BC_RGBA16161616:
			HOLO_CORE(uint16_t, 4, 0);
			break;
		case BC_YUVA16161616:
			HOLO_CORE(uint16_t, 4, 1);
			break;
	}



	phase -= 37;
}



HoloPackage::HoloPackage()
{
}