Files
veejay/veejay-ng/pluginpack/lvd_erode.c
Niels Elburg c03570924b added 10 more livido plugins
git-svn-id: svn://code.dyne.org/veejay/trunk@627 eb8d1916-c9e9-0310-b8de-cf0c9472ead5
2006-10-09 21:02:02 +00:00

220 lines
6.5 KiB
C

/*
* 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);
int error = livido_property_set( my_instance, "PLUGIN_private", LIVIDO_ATOM_TYPE_VOIDPTR, 1, &b );
return error;
}
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 thres = lvd_extract_param_number(
my_instance,
"in_parameters",
0 );
int threshold = (int)(255.0 * thres);
int kernel = lvd_extract_param_index(
my_instance,
"in_parameters",
1 );
uint8_t kernels[4][9] ={
{ 1,1,1, 1,1,1 ,1,1,1 },//0
{ 0,1,0, 1,1,1, 0,1,0 },//1
{ 0,0,0, 1,1,1, 0,0,0 },//2
{ 0,1,0, 0,1,0, 0,1,0 },//3
{ 0,0,1, 0,1,0, 1,0,0 },//4
{ 1,0,0, 0,1,0, 0,0,1 },
{ 1,1,1, 0,0,0, 0,0,0 },
{ 0,0,0, 0,0,0, 1,1,1 }
};
for( i = 0; i < len ;i ++ )
b->tmpy[i] = ( A[0][i] > threshold ? 0xff: 0 );
const uint8_t *bg = b->tmpy;
for( y = wid; y < (len-wid); y += wid )
{
for( x = 1 ; x < (wid-1); x ++ )
{
uint8_t matrix[9] = { bg[x-1+y-wid],
bg[x+y-wid],
bg[x+1+y-wid],
bg[x-1+y],
bg[x+y],
bg[x+1+y],
bg[x-1+y+wid],
bg[x+y+wid],
bg[x+1+y+wid] };
O[0][x + y] = _erode_kernel3x3( kernels[kernel], matrix );
}
}
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", "Erode");
livido_set_string_value( port, "description", "Erodes 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", "Threshold" );
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.8 );
livido_set_string_value( port, "description" ,"Threshold");
in_params[1] = livido_port_new( LIVIDO_PORT_TYPE_PARAMETER_TEMPLATE );
port = in_params[1];
livido_set_string_value(port, "name", "Kernel type" );
livido_set_string_value(port, "kind", "INDEX" );
livido_set_int_value( port, "min", 1 );
livido_set_int_value( port, "max", 8 );
livido_set_int_value( port, "default", 1 );
livido_set_string_value( port, "description" ,"Type of kernel 3x3");
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;
}