diff options
| -rw-r--r-- | modules/pdp_ocanvas.c | 438 |
1 files changed, 438 insertions, 0 deletions
diff --git a/modules/pdp_ocanvas.c b/modules/pdp_ocanvas.c new file mode 100644 index 0000000..e06a500 --- /dev/null +++ b/modules/pdp_ocanvas.c @@ -0,0 +1,438 @@ +/* + * PiDiP module. + * Copyright (c) by Yves Degoyon (ydegoyon@free.fr) + * + * 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., 675 Mass Ave, Cambridge, MA 02139, USA. + * + */ + +/* This object is an object allowing juxtaposition of frames from two inlets + * Written by Yves Degoyon + */ + + + +#include "pdp.h" +#include <math.h> + +static char *pdp_ocanvas_version = "pdp_ocanvas: version 0.1, display for several video sources, written by Yves Degoyon (ydegoyon@free.fr)"; + +#define MAX_CANVAS_INPUT 10 + +typedef struct pdp_ocanvas_struct +{ + t_object x_obj; + t_float x_f; + + t_outlet *x_outlet0; + t_int x_dropped; + t_int x_queue_id; + + t_int x_opacket; + + t_int x_current; + t_float x_xmouse; + t_float x_ymouse; + + t_int x_average; + + t_int *x_packets; + t_int *x_widths; + t_int *x_heights; + t_float *x_xoffsets; + t_float *x_yoffsets; + t_int *x_sizes; + + t_int x_owidth; + t_int x_oheight; + t_int x_osize; + t_int x_nbinputs; + +} t_pdp_ocanvas; + +static void pdp_ocanvas_process_yv12(t_pdp_ocanvas *x) +{ + t_int px, py, ppx, ppy, ii, nbs; + short int *pY, *pU, *pV; + short int *piY, *piU, *piV; + t_pdp *oheader; + short int *odata; + t_pdp *iheader; + short int *idata; + + x->x_opacket = pdp_packet_new_image_YCrCb( x->x_owidth, x->x_oheight ); + oheader = pdp_packet_header(x->x_opacket); + odata = (short int *)pdp_packet_data(x->x_opacket); + + oheader->info.image.encoding = PDP_IMAGE_YV12; + oheader->info.image.width = x->x_owidth; + oheader->info.image.height = x->x_oheight; + + memset( odata, 0x00, (x->x_osize + (x->x_osize>>1))<<1 ); + + pY = odata; + pV = odata+x->x_osize; + pU = odata+x->x_osize+(x->x_osize>>2); + for ( py=0; py<x->x_oheight; py++ ) + { + for ( px=0; px<x->x_owidth; px++ ) + { + nbs=0; + for ( ii=0; ii<x->x_nbinputs; ii++) + { + if ( x->x_packets[ii] != -1 ) + { + if ( (px >= (int)x->x_xoffsets[ii]) && ( px < (int)x->x_xoffsets[ii] + x->x_widths[ii] ) + && (py >= (int)x->x_yoffsets[ii]) && ( py < (int)x->x_yoffsets[ii] + x->x_heights[ii] ) + ) + { + nbs++; + idata = (short int *)pdp_packet_data(x->x_packets[ii]); + piY = idata; + piV = idata+x->x_sizes[ii]; + piU = idata+x->x_sizes[ii]+(x->x_sizes[ii]>>2); + ppx = px-(int)x->x_xoffsets[ii]; + ppy = py-(int)x->x_yoffsets[ii]; + *(pY+py*x->x_owidth+px) += *(piY+ppy*x->x_widths[ii]+ppx); + if ( (px%2==0) && (py%2==0) ) + { + *(pU+(py>>1)*(x->x_owidth>>1)+(px>>1)) += + *(piU+(ppy>>1)*(x->x_widths[ii]>>1)+(ppx>>1)); + *(pV+(py>>1)*(x->x_owidth>>1)+(px>>1)) += + *(piV+(ppy>>1)*(x->x_widths[ii]>>1)+(ppx>>1)); + } + } + } + } + if ( ( nbs != 0 ) && x->x_average ) + { + *(pY+py*x->x_owidth+px) /= nbs; + if ( (px%2==0) && (py%2==0) ) + { + *(pU+(py>>1)*(x->x_owidth>>1)+(px>>1)) /= nbs; + *(pV+(py>>1)*(x->x_owidth>>1)+(px>>1)) /= nbs; + } + } + } + } + + return; +} + +static void pdp_ocanvas_sendpacket(t_pdp_ocanvas *x) +{ + /* unregister and propagate if valid dest packet */ + pdp_packet_pass_if_valid(x->x_outlet0, &x->x_opacket); +} + +static void pdp_ocanvas_process(t_pdp_ocanvas *x, t_int ni) +{ + int encoding; + t_pdp *header = 0; + + /* check if image data packets are compatible */ + if ( (header = pdp_packet_header(x->x_packets[ni])) + && (PDP_IMAGE == header->type)){ + + /* pdp_ocanvas_process inputs and write into active inlet */ + switch(pdp_packet_header(x->x_packets[ni])->info.image.encoding){ + + case PDP_IMAGE_YV12: + pdp_queue_add(x, pdp_ocanvas_process_yv12, pdp_ocanvas_sendpacket, &x->x_queue_id); + break; + + case PDP_IMAGE_GREY: + break; + + default: + /* don't know the type, so dont pdp_ocanvas_process */ + break; + + } + } +} + +static void pdp_ocanvas_offset(t_pdp_ocanvas *x, t_floatarg ni, t_floatarg xoffset, t_floatarg yoffset) +{ + if ( ( ni < 1 ) || ( ni > x->x_nbinputs ) ) + { + post( "pdp_ocanvas : offset : wrong source : %d : must be between 1 and %d", ni, x->x_nbinputs ); + return; + } + x->x_xoffsets[(int)ni-1] = xoffset; + x->x_yoffsets[(int)ni-1] = yoffset; +} + +static void pdp_ocanvas_select(t_pdp_ocanvas *x, t_floatarg X, t_floatarg Y) +{ + t_int ii; + + x->x_current = -1; + X = X*x->x_owidth; + Y = Y*x->x_oheight; + // post( "pdp_ocanvas : select %f %f", X, Y ); + for ( ii=0; ii<x->x_nbinputs; ii++) + { + if ( x->x_packets[ii] != -1 ) + { + if ( (X >= x->x_xoffsets[ii]) && ( X < x->x_xoffsets[ii] + x->x_widths[ii] ) + && (Y >= x->x_yoffsets[ii]) && ( Y < x->x_yoffsets[ii] + x->x_heights[ii] ) + ) + { + x->x_current = ii; + x->x_xmouse = X; + x->x_ymouse = Y; + } + } + } +} + +static void pdp_ocanvas_drag(t_pdp_ocanvas *x, t_floatarg X, t_floatarg Y) +{ + X = X*x->x_owidth; + Y = Y*x->x_oheight; + // post( "pdp_ocanvas : drag %f %f", X, Y ); + if ( x->x_current != -1 ) + { + x->x_xoffsets[ x->x_current ] += (X-x->x_xmouse); + x->x_yoffsets[ x->x_current ] += (Y-x->x_ymouse); + x->x_xmouse = X; + x->x_ymouse = Y; + } +} + +static void pdp_ocanvas_unselect(t_pdp_ocanvas *x) +{ + x->x_current = -1; +} + +static void pdp_ocanvas_average(t_pdp_ocanvas *x, t_floatarg bvalue) +{ + if ( ( bvalue == 0.0 ) || ( bvalue == 1.0 ) ) + { + x->x_average = (int) bvalue; + } +} + +static void pdp_ocanvas_input(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f, t_int ni) +{ + t_pdp *header; + short int *data; + + /* if this is a register_ro message or register_rw message, register with packet factory */ + + if (s== gensym("register_rw")) + { + /* release the packet */ + if ( x->x_packets[ni] != -1 ) + { + pdp_packet_mark_unused(x->x_packets[ni]); + x->x_packets[ni] = -1; + } + x->x_dropped = pdp_packet_convert_ro_or_drop(&x->x_packets[ni], (int)f, pdp_gensym("image/YCrCb/*") ); + if ( x->x_packets[ni] != -1 ) + { + header = pdp_packet_header(x->x_packets[ni]); + x->x_widths[ni] = header->info.image.width; + x->x_heights[ni] = header->info.image.height; + x->x_sizes[ni] = x->x_widths[ni]*x->x_heights[ni]; + } + } + + if ((s == gensym("process")) && (-1 != x->x_packets[ni]) && (!x->x_dropped)) + { + /* add the process method and callback to the process queue */ + pdp_ocanvas_process(x, ni); + } +} + +static void pdp_ocanvas_input0(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 0); +} + +static void pdp_ocanvas_input1(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 1); +} + +static void pdp_ocanvas_input2(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 2); +} + +static void pdp_ocanvas_input3(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 3); +} + +static void pdp_ocanvas_input4(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 4); +} + +static void pdp_ocanvas_input5(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 5); +} + +static void pdp_ocanvas_input6(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 6); +} + +static void pdp_ocanvas_input7(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 7); +} + +static void pdp_ocanvas_input8(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 8); +} + +static void pdp_ocanvas_input9(t_pdp_ocanvas *x, t_symbol *s, t_floatarg f) +{ + pdp_ocanvas_input(x, s, f, 9); +} + +static void pdp_ocanvas_free(t_pdp_ocanvas *x) +{ + t_int ii; + + pdp_queue_finish(x->x_queue_id); + for ( ii=0; ii<x->x_nbinputs; ii++) + { + pdp_packet_mark_unused(x->x_packets[ii]); + } + pdp_packet_mark_unused(x->x_opacket); + if ( x->x_packets ) freebytes( x->x_packets, x->x_nbinputs*sizeof(t_int) ); + if ( x->x_widths ) freebytes( x->x_widths, x->x_nbinputs*sizeof(t_int) ); + if ( x->x_heights ) freebytes( x->x_heights, x->x_nbinputs*sizeof(t_int) ); + if ( x->x_sizes ) freebytes( x->x_sizes, x->x_nbinputs*sizeof(t_int) ); + if ( x->x_xoffsets ) freebytes( x->x_xoffsets, x->x_nbinputs*sizeof(t_float) ); + if ( x->x_yoffsets ) freebytes( x->x_yoffsets, x->x_nbinputs*sizeof(t_float) ); +} + +t_class *pdp_ocanvas_class; + +void *pdp_ocanvas_new(t_symbol *s, int argc, t_atom *argv) +{ + t_pdp_ocanvas *x = (t_pdp_ocanvas *)pd_new(pdp_ocanvas_class); + t_int ii; + char *imes[32]; + + if ( argc != 3 ) + { + post( "pdp_ocanvas : wrong constructor : pdp_ocanvas <width> <height> <nb inputs> (argc=%d)", argc); + return NULL; + } + if ( argv[0].a_type != A_FLOAT || argv[1].a_type != A_FLOAT || + argv[2].a_type != A_FLOAT ) + { + post( "pdp_ocanvas : wrong constructor : pdp_ocanvas <width> <height> <nb inputs>"); + return NULL; + } + + x->x_owidth = ( (int) argv[0].a_w.w_float / 8 ) * 8; // round to a multiple of 8 + x->x_oheight = ( (int) argv[1].a_w.w_float / 8 ) * 8; // round to a multiple of 8 + x->x_osize = x->x_owidth*x->x_oheight; + x->x_nbinputs = (int) argv[2].a_w.w_float; + + if ( x->x_owidth < 0 ) + { + post( "pdp_ocanvas : wrong width : %d", x->x_owidth); + return NULL; + } + if ( x->x_oheight < 0 ) + { + post( "pdp_ocanvas : wrong height : %d", x->x_oheight); + return NULL; + } + if ( x->x_nbinputs < 0 ) + { + post( "pdp_ocanvas : wrong number of inputs : %d", x->x_nbinputs); + return NULL; + } + if ( x->x_nbinputs > MAX_CANVAS_INPUT ) + { + post( "pdp_ocanvas : number of inputs is too high : %d : only %d supported", + x->x_nbinputs, MAX_CANVAS_INPUT); + return NULL; + } + + post ( "pdp_ocanvas : new %dx%d canvas with %d inputs", x->x_owidth, x->x_oheight, x->x_nbinputs ); + + x->x_packets = ( t_int* ) getbytes( x->x_nbinputs*sizeof(t_int) ); + x->x_widths = ( t_int* ) getbytes( x->x_nbinputs*sizeof(t_int) ); + x->x_heights = ( t_int* ) getbytes( x->x_nbinputs*sizeof(t_int) ); + x->x_sizes = ( t_int* ) getbytes( x->x_nbinputs*sizeof(t_int) ); + x->x_xoffsets = ( t_float* ) getbytes( x->x_nbinputs*sizeof(t_float) ); + x->x_yoffsets = ( t_float* ) getbytes( x->x_nbinputs*sizeof(t_float) ); + + x->x_opacket = pdp_packet_new_image_YCrCb( x->x_owidth, x->x_oheight ); + + for ( ii=0; ii<x->x_nbinputs; ii++) + { + sprintf( (char*)imes, "pdp%d", ii ); + inlet_new(&x->x_obj, &x->x_obj.ob_pd, gensym("pdp"), gensym((char*)imes) ); + x->x_packets[ii] = -1; + x->x_xoffsets[ii] = 0.; + x->x_yoffsets[ii] = 0.; + } + x->x_current = -1; + x->x_average = 0; + x->x_outlet0 = outlet_new(&x->x_obj, &s_anything); + + return (void *)x; +} + + +#ifdef __cplusplus +extern "C" +{ +#endif + + +void pdp_ocanvas_setup(void) +{ + char *imes[32]; + + // post( pdp_ocanvas_version ); + pdp_ocanvas_class = class_new(gensym("pdp_ocanvas"), (t_newmethod)pdp_ocanvas_new, + (t_method)pdp_ocanvas_free, sizeof(t_pdp_ocanvas), 0, A_GIMME, A_NULL); + class_sethelpsymbol( pdp_ocanvas_class, gensym("pdp_ocanvas.pd") ); + + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input0, gensym("pdp0"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input1, gensym("pdp1"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input2, gensym("pdp2"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input3, gensym("pdp3"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input4, gensym("pdp4"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input5, gensym("pdp5"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input6, gensym("pdp6"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input7, gensym("pdp7"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input8, gensym("pdp8"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_input9, gensym("pdp9"), A_SYMBOL, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_offset, gensym("offset"), A_DEFFLOAT, A_DEFFLOAT, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_select, gensym("select"), A_DEFFLOAT, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_drag, gensym("drag"), A_DEFFLOAT, A_DEFFLOAT, A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_unselect, gensym("unselect"), A_NULL); + class_addmethod(pdp_ocanvas_class, (t_method)pdp_ocanvas_average, gensym("average"), A_DEFFLOAT, A_NULL); + +} + +#ifdef __cplusplus +} +#endif |