/* 
 * LIBVJE - veejay fx library
 *
 * Copyright(C)2006 Niels Elburg <nelburg@looze.net>
 * See COPYING for software license and distribution details
 */


#define IS_LIVIDO_PLUGIN 
#include 	<livido.h>
LIVIDO_PLUGIN
#include	"utils.h"
#include	"livido-utils.c"


livido_init_f	init_instance( livido_port_t *my_instance )
{
	return LIVIDO_NO_ERROR;
}


livido_deinit_f	deinit_instance( livido_port_t *my_instance )
{
	return LIVIDO_NO_ERROR;
}

livido_process_f		process_instance( livido_port_t *my_instance, double timecode )
{
	int len =0;
	int x,y;
	uint8_t *A[4] = {NULL,NULL,NULL,NULL};
	uint8_t *O[4]= {NULL,NULL,NULL,NULL};
	uint8_t *B[4] = { NULL,NULL,NULL,NULL };
	int palette[3];
	int w[3];
	int h[3];

	int error = lvd_extract_channel_values( my_instance, "in_channels", 0, &w[0], &h[0], A, &palette[0] );
	if( error != LIVIDO_NO_ERROR )
		return LIVIDO_ERROR_HARDWARE; //@ error codes in livido flanky

	error = lvd_extract_channel_values( my_instance, "in_channels", 1, &w[1], &h[1], B, &palette[1] );
	if( error != LIVIDO_NO_ERROR )
		return LIVIDO_ERROR_HARDWARE; //@ error codes in livido flanky


	
	error	  = lvd_extract_channel_values( my_instance, "out_channels", 0, &w[2],&h[2], O,&palette[2] );
	if( error != LIVIDO_NO_ERROR )
		return LIVIDO_ERROR_HARDWARE; //@ error codes in livido flanky

#ifdef STRICT_CHECKING
	assert( w[0] == w[1] );
	assert( h[0] == h[1] );
	assert( w[2] == w[1] );
	assert( h[2] == h[1] );
	assert( palette[0] == palette[1] );
	assert( palette[2] == palette[1] );
	assert( A[0] != NULL );
	assert( A[1] != NULL );
	assert( A[2] != NULL );
	assert( B[0] != NULL );
	assert( B[1] != NULL );
	assert( B[2] != NULL );
	assert( O[0] != NULL );
	assert( O[1] != NULL );
	assert( O[2] != NULL );
#endif

	int uv_len = lvd_uv_plane_len( palette[0],w[0],h[0] );
	len = w[0] * h[0];

	double h_scale = lvd_extract_param_number( my_instance, "in_parameters", 0 );	
	double v_scale = lvd_extract_param_number( my_instance, "in_parameters", 1 );	

	int	black_border = lvd_extract_param_boolean( my_instance, "in_parameters", 2 );	



	const int wid = w[0];
	const int hei = h[0];

	double	tmp = 0;
	int  	dx,dy,nx,ny;

	if( black_border)
	{
		for(y=0; y < hei; y++)
		{
			for(x=0; x < wid ; x++)
			{
				// calculate new location of pixel
				tmp = (double)(B[0][(y*wid+x)] - 128);
				// new x offset 
				dx = (int)(v_scale * tmp); //@rat
				// new y offset 
				dy = (int)(h_scale * tmp); //@ra
				// new pixel coordinates
				nx = x + dx;
				ny = y + dy;

				if( nx < 0 || ny < 0 || nx > wid || ny >= hei )
				{
					O[0][y*wid+x] = 16;
					O[1][y*wid+x] = 128;
					O[2][y*wid+x] = 128;
				}
				else
				{
					O[0][y*wid+x] = A[0][ny * wid + nx];
					O[1][y*wid+x] = A[1][ny * wid + nx];
					O[2][y*wid+x] = A[2][ny * wid + nx];
				}
			}
		}

	}
	else
	{
		
		for(y=0; y < hei; y++)
		{
			for(x=0; x < wid ; x++)
			{
				// calculate new location of pixel
				tmp = (double)(B[0][(y*wid+x)] - 128);
				// new x offset 
				dx = (int)(v_scale * tmp); //@rat
				// new y offset 
				dy = (int)(h_scale * tmp); //@ra
				// new pixel coordinates
				nx = x + dx;
				ny = y + dy;
				while(nx < 0)
					nx+=wid;	
				while(ny < 0)
					ny+=hei;
				if( ny >= hei ) ny = hei -1;
				if( nx > wid ) nx = wid;
				// put pixels from local copy
				O[0][y*wid+x] = A[0][ny * wid + nx];
				O[1][y*wid+x] = A[1][ny * wid + nx];
				O[2][y*wid+x] = A[2][ny * wid + nx];
			}
		}
	}
	return 	LIVIDO_NO_ERROR;
}

livido_port_t	*livido_setup(livido_setup_t list[], int version)

{
	LIVIDO_IMPORT(list);

	livido_port_t *port = NULL;
	livido_port_t *in_params[3];
	livido_port_t *in_chans[2];
	livido_port_t *out_chans[1];
	livido_port_t *info = NULL;
	livido_port_t *filter = NULL;

	info = livido_port_new( LIVIDO_PORT_TYPE_PLUGIN_INFO );
	port = info;

		livido_set_string_value( port, "maintainer", "Niels");
		livido_set_string_value( port, "version","1");
	
	filter = livido_port_new( LIVIDO_PORT_TYPE_FILTER_CLASS );
	livido_set_int_value( filter, "api_version", LIVIDO_API_VERSION );
	livido_set_voidptr_value( filter, "deinit_func", &deinit_instance );
	livido_set_voidptr_value( filter, "init_func", &init_instance );
	livido_set_voidptr_value( filter, "process_func", &process_instance );

	port = filter;

		livido_set_string_value( port, "name", "Displacement Mapping");	
		livido_set_string_value( port, "description", "Displacement Mapping");
		livido_set_string_value( port, "author", "Niels Elburg");
		
		livido_set_int_value( port, "flags", 0);
		livido_set_string_value( port, "license", "GPL2");
		livido_set_int_value( port, "version", 1);
	
	int palettes0[] = {
			LIVIDO_PALETTE_YUV444P,
               		0
	};

	in_params[0] = livido_port_new( LIVIDO_PORT_TYPE_PARAMETER_TEMPLATE );
        port = in_params[0];

                livido_set_string_value(port, "name", "X displace" );
                livido_set_string_value(port, "kind", "NUMBER" );
                livido_set_double_value( port, "min", 0.0 );
                livido_set_double_value( port, "max", 1.0 );
                livido_set_double_value( port, "default", 0.1 );
                livido_set_string_value( port, "description" ,"Displacement in horizontal direction");
	in_params[1] = livido_port_new( LIVIDO_PORT_TYPE_PARAMETER_TEMPLATE );
        port = in_params[1];

                livido_set_string_value(port, "name", "Y displace" );
                livido_set_string_value(port, "kind", "NUMBER" );
                livido_set_double_value( port, "min", 0.0 );
                livido_set_double_value( port, "max", 1.0 );
                livido_set_double_value( port, "default", 0.1 );
                livido_set_string_value( port, "description" ,"Displacement in vertical direction");

	in_params[2] = livido_port_new( LIVIDO_PORT_TYPE_PARAMETER_TEMPLATE );
        port = in_params[2];

                livido_set_string_value(port, "name", "Drop" );
                livido_set_string_value(port, "kind", "SWITCH" );
                livido_set_boolean_value( port, "default", 0 );
                livido_set_string_value( port, "description" ,"Make pixels that drop outside black");


        in_chans[0] = livido_port_new( LIVIDO_PORT_TYPE_CHANNEL_TEMPLATE );
	port = in_chans[0];
            
                livido_set_string_value( port, "name", "Channel A");
           	livido_set_int_array( port, "palette_list", 1, palettes0);
		livido_set_int_value( port, "flags", 0);

	in_chans[1] = livido_port_new( LIVIDO_PORT_TYPE_CHANNEL_TEMPLATE );
	port = in_chans[1];
            
                livido_set_string_value( port, "name", "Greyscale Channel B");
           	livido_set_int_array( port, "palette_list", 1, palettes0);
		livido_set_int_value( port, "flags", 0);

	out_chans[0] = livido_port_new( LIVIDO_PORT_TYPE_CHANNEL_TEMPLATE );
	port = out_chans[0];
	
	        livido_set_string_value( port, "name", "Output Channel");
		livido_set_int_array( port, "palette_list", 1, palettes0);
		livido_set_int_value( port, "flags", 0);
	
	livido_set_portptr_array( filter, "in_channel_templates", 2 , in_chans );
	livido_set_portptr_array( filter, "out_parameter_templates",0, NULL );
	livido_set_portptr_array( filter, "in_parameter_templates",3, in_params );
	livido_set_portptr_array( filter, "out_channel_templates", 1, out_chans );

	livido_set_portptr_value(info, "filters", filter);
	return info;
}