/* 
 * 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"

typedef struct
{
        uint8_t *tmpy;
} bg_t;

livido_init_f	init_instance( livido_port_t *my_instance )
{
	int w=0;
        int h=0;

        lvd_extract_dimensions( my_instance, "out_channels", &w, &h );
     
        bg_t *b = (bg_t*) livido_malloc( sizeof( bg_t ));
        b->tmpy = (uint8_t*) livido_malloc(sizeof(uint8_t) * w * h);
	livido_memset( b->tmpy,0, w * h * sizeof(uint8_t));
    	int error = livido_property_set( my_instance, "PLUGIN_private", LIVIDO_ATOM_TYPE_VOIDPTR, 1, &b );

	return error;
}

static	void	blurmask_1x3(
		uint8_t *map,
		uint8_t *src,
		int width,
		int height,
		uint8_t *out,
		const double k)
{
    int r, c;
    uint8_t d;
    
    const int len = (width*height);

    for (r = 0; r < len; r += width) {
        for (c = 1; c < width-1; c++) {
                map[c + r] = (src[r + c - 1] +
                                  src[r + c] +
                                  src[r + c + 1]
                    ) / 3;
        }
    }
    
    for(c=0; c < len; c++) {
          d = (map[c] - src[c]) * k;
          out[c] = d;
        }
}


static	void	blurmask_3x3(
		uint8_t *map,
		uint8_t *src,
		int width,
		int height,
		uint8_t *out,
		const double k)
{
	int r, c;
	uint8_t d;
    
	const int len = (width*height)-width;


	for (r = width; r < len; r += width)
	{
		for (c = 1; c < width-1; c++)
		{
                	map[c + r] = (
				  src[r - width + c - 1] +
                                  src[r - width + c] +
                                  src[r - width + c + 1] +
                                  src[r + width + c - 1] +
                                  src[r + width + c] +
                                  src[r + width + c + 1] +
                                  src[r + c] +
                                  src[r + c + 1] +
                                  src[r + c - 1]  
                    ) / 9;
        	}
    	}

    	for(c=width; c < len; c++)
	{
          d = (map[c] - src[c]) * k;
          out[c] = d;
        }
}
static	void	negmask_3x3(
		uint8_t *map,
		uint8_t *src,
		int width,
		int height,
		uint8_t *out,
		const double k)
{
	int r, c;
	uint8_t d;
    
	const int len = (width*height)-width;


	for (r = width; r < len; r += width)
	{
		for (c = 1; c < width-1; c++)
		{
                	map[c + r] = 0xff - ((
				  src[r - width + c - 1] +
                                  src[r - width + c] +
                                  src[r - width + c + 1] +
                                  src[r + width + c - 1] +
                                  src[r + width + c] +
                                  src[r + width + c + 1] +
                                  src[r + c] +
                                  src[r + c + 1] +
                                  src[r + c - 1]  
                    ) / 9);
        	}
    	}

    	for(c=width; c < len; c++)
	{
          d = (map[c] - src[c]) * k;
          out[c] = d;
        }
}

livido_deinit_f	deinit_instance( livido_port_t *my_instance )
{
	bg_t *b = NULL;
        int error = livido_property_get( my_instance, "PLUGIN_private",
                        0, &b );
#ifdef STRICT_CHECKING
        assert( b != NULL );
#endif
        livido_free( b->tmpy );
	livido_free( b );

	return LIVIDO_NO_ERROR;
}

livido_process_f		process_instance( livido_port_t *my_instance, double timecode )
{
	int len =0;
	int i = 0;
	uint8_t *A[4] = {NULL,NULL,NULL,NULL};
	uint8_t *O[4]= {NULL,NULL,NULL,NULL};
	int x,y;
	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, "out_channels", 0, &w[1],&h[1], O,&palette[1] );
	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( palette[0] == palette[1] );
	assert( A[0] != NULL );
	assert( A[1] != NULL );
	assert( A[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];
	const int wid = w[0];
	const int hei = w[1];

        bg_t *b = NULL;
        error     = livido_property_get( my_instance, "PLUGIN_private", 0, &b );
#ifdef STRICT_CHECKING
        assert( error == NULL );
#endif
        double noise  = lvd_extract_param_number(
                                my_instance,
                                "in_parameters",
                                0 );
	int    type = lvd_extract_param_index(
				my_instance,
				"in_parameters",
				1 );

	switch(type)
	{
		case 0:
			negmask_3x3(
				b->tmpy,
				A[0],
				w[0],
				h[0],
				O[0],
				noise );
		break;
		case 1:
			blurmask_1x3(
				b->tmpy,
				A[0],
				w[0],
				h[0],
				O[0],
				noise );
		break;
		case 2:
			blurmask_3x3(
				b->tmpy,
				A[0],
				w[0],
				h[0],
				O[0],
				noise );
		default:
		break;
	}
	
	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[2];
	livido_port_t *in_chans[3];
	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", "Low Noise");	
		livido_set_string_value( port, "description", "Noisify the Image");
		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_YUV420P,
                       	LIVIDO_PALETTE_YUV422P,
			LIVIDO_PALETTE_YUV444P,
               		0
	};
	
        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", 3, 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", 3, palettes0);
		livido_set_int_value( port, "flags", 0);

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

                livido_set_string_value(port, "name", "Coefficient" );
                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.02 );
                livido_set_string_value( port, "description" ,"Coefficient");
	in_params[1] = livido_port_new( LIVIDO_PORT_TYPE_PARAMETER_TEMPLATE );
        port = in_params[1];

                livido_set_string_value(port, "name", "Noise method" );
                livido_set_string_value(port, "kind", "INDEX" );
                livido_set_int_value( port, "min", 0 );
                livido_set_int_value( port, "max", 2 );
                livido_set_int_value( port, "default", 0 );
                livido_set_string_value( port, "description" ,"Noise method");


	livido_set_portptr_array( filter, "in_channel_templates", 1 , in_chans );
	livido_set_portptr_array( filter, "out_parameter_templates",0, NULL );
	livido_set_portptr_array( filter, "in_parameter_templates",2, in_params );
	livido_set_portptr_array( filter, "out_channel_templates", 1, out_chans );
	//livido_set_portptr_array( filter, "out_channel_templates", 0, NULL );


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