/*****************************************************************************
* Preamble
*****************************************************************************/
-#include <stdlib.h> /* malloc(), free() */
-#include <string.h>
-#include <vlc/vlc.h>
+#ifdef HAVE_CONFIG_H
+# include "config.h"
+#endif
+
+#include <vlc_common.h>
+#include <vlc_plugin.h>
#include <vlc_vout.h>
#include "vlc_filter.h"
+#include "filter_picture.h"
+
+#include "math.h"
/*****************************************************************************
* Local prototypes
static void Destroy ( vlc_object_t * );
static picture_t *Filter( filter_t *, picture_t * );
+static int ExtractCallback( vlc_object_t *, char const *,
+ vlc_value_t, vlc_value_t, void * );
+
+static void get_red_from_yuv420( picture_t *, picture_t *, int, int, int );
+static void get_green_from_yuv420( picture_t *, picture_t *, int, int, int );
+static void get_blue_from_yuv420( picture_t *, picture_t *, int, int, int );
+static void get_red_from_yuv422( picture_t *, picture_t *, int, int, int );
+static void get_green_from_yuv422( picture_t *, picture_t *, int, int, int );
+static void get_blue_from_yuv422( picture_t *, picture_t *, int, int, int );
+static void make_projection_matrix( filter_t *, int color, int *matrix );
+static void get_custom_from_yuv420( picture_t *, picture_t *, int, int, int, int * );
+static void get_custom_from_yuv422( picture_t *, picture_t *, int, int, int, int * );
+static void get_custom_from_packedyuv422( picture_t *, picture_t *, int * );
+
#define COMPONENT_TEXT N_("RGB component to extract")
#define COMPONENT_LONGTEXT N_("RGB component to extract. 0 for Red, 1 for Green and 2 for Blue.")
#define FILTER_PREFIX "extract-"
-static int pi_component_values[] = { 0, 1, 2 };
-static const char *ppsz_component_descriptions[] = { "Red", "Green", "Blue" };
+static const int pi_component_values[] = { 0xFF0000, 0x00FF00, 0x0000FF };
+static const char *const ppsz_component_descriptions[] = {
+ "Red", "Green", "Blue" };
/*****************************************************************************
* Module descriptor
*****************************************************************************/
vlc_module_begin();
- set_description( _("Extract RGB component video filter") );
- set_shortname( _("Extract" ));
+ set_description( N_("Extract RGB component video filter") );
+ set_shortname( N_("Extract" ));
set_category( CAT_VIDEO );
set_subcategory( SUBCAT_VIDEO_VFILTER );
set_capability( "video filter2", 0 );
add_shortcut( "extract" );
- add_integer( FILTER_PREFIX "component", 0, NULL,
- COMPONENT_TEXT, COMPONENT_LONGTEXT, VLC_FALSE );
+ add_integer_with_range( FILTER_PREFIX "component", 0xFF0000, 1, 0xFFFFFF,
+ NULL, COMPONENT_TEXT, COMPONENT_LONGTEXT, false );
change_integer_list( pi_component_values, ppsz_component_descriptions, 0 );
set_callbacks( Create, Destroy );
vlc_module_end();
-static const char *ppsz_filter_options[] = {
+static const char *const ppsz_filter_options[] = {
"component", NULL
};
-enum { RED, GREEN, BLUE };
+enum { RED=0xFF0000, GREEN=0x00FF00, BLUE=0x0000FF };
struct filter_sys_t
{
int i_color;
+ int *projection_matrix;
};
/*****************************************************************************
{
filter_t *p_filter = (filter_t *)p_this;
+ switch( p_filter->fmt_in.video.i_chroma )
+ {
+ case VLC_FOURCC('I','4','2','0'):
+ case VLC_FOURCC('I','Y','U','V'):
+ case VLC_FOURCC('J','4','2','0'):
+ case VLC_FOURCC('Y','V','1','2'):
+
+ case VLC_FOURCC('I','4','2','2'):
+ case VLC_FOURCC('J','4','2','2'):
+
+ CASE_PACKED_YUV_422
+ break;
+
+ default:
+ /* We only want planar YUV 4:2:0 or 4:2:2 */
+ msg_Err( p_filter, "Unsupported input chroma (%4s)",
+ (char*)&(p_filter->fmt_in.video.i_chroma) );
+ return VLC_EGENERIC;
+ }
+
/* Allocate structure */
p_filter->p_sys = malloc( sizeof( filter_sys_t ) );
if( p_filter->p_sys == NULL )
msg_Err( p_filter, "out of memory" );
return VLC_ENOMEM;
}
+ p_filter->p_sys->projection_matrix = malloc( 9 * sizeof( int ) );
+ if( !p_filter->p_sys->projection_matrix )
+ {
+ free( p_filter->p_sys );
+ msg_Err( p_filter, "out of memory" );
+ return VLC_ENOMEM;
+ }
config_ChainParse( p_filter, FILTER_PREFIX, ppsz_filter_options,
p_filter->p_cfg );
- var_Create( p_filter, FILTER_PREFIX "component",
- VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
- p_filter->p_sys->i_color = var_GetInteger( p_filter,
+ p_filter->p_sys->i_color = var_CreateGetIntegerCommand( p_filter,
FILTER_PREFIX "component" );
+ var_AddCallback( p_filter, FILTER_PREFIX "component",
+ ExtractCallback, p_filter->p_sys );
+
+ /* Matrix won't be used for RED, GREEN or BLUE in planar formats */
+ make_projection_matrix( p_filter, p_filter->p_sys->i_color,
+ p_filter->p_sys->projection_matrix );
p_filter->pf_video_filter = Filter;
{
filter_t *p_filter = (filter_t *)p_this;
+ free( p_filter->p_sys->projection_matrix );
free( p_filter->p_sys );
}
/*****************************************************************************
* Render
*****************************************************************************/
-static void get_red_from_yuv420( picture_t *, picture_t *, int, int, int );
-static void get_green_from_yuv420( picture_t *, picture_t *, int, int, int );
-static void get_blue_from_yuv420( picture_t *, picture_t *, int, int, int );
-static void get_red_from_yuv422( picture_t *, picture_t *, int, int, int );
-static void get_green_from_yuv422( picture_t *, picture_t *, int, int, int );
-static void get_blue_from_yuv422( picture_t *, picture_t *, int, int, int );
-
static picture_t *Filter( filter_t *p_filter, picture_t *p_pic )
{
picture_t *p_outpic;
get_blue_from_yuv420( p_pic, p_outpic,
Y_PLANE, U_PLANE, V_PLANE );
break;
+ default:
+ get_custom_from_yuv420( p_pic, p_outpic,
+ Y_PLANE, U_PLANE, V_PLANE,
+ p_filter->p_sys->projection_matrix);
+ break;
}
break;
get_blue_from_yuv422( p_pic, p_outpic,
Y_PLANE, U_PLANE, V_PLANE );
break;
+ default:
+ get_custom_from_yuv422( p_pic, p_outpic,
+ Y_PLANE, U_PLANE, V_PLANE,
+ p_filter->p_sys->projection_matrix);
+ break;
}
break;
+ CASE_PACKED_YUV_422
+ get_custom_from_packedyuv422( p_pic, p_outpic,
+ p_filter->p_sys->projection_matrix );
+ break;
+
default:
msg_Warn( p_filter, "Unsupported input chroma (%4s)",
(char*)&(p_pic->format.i_chroma) );
return NULL;
}
- p_outpic->date = p_pic->date;
- p_outpic->b_force = p_pic->b_force;
- p_outpic->i_nb_fields = p_pic->i_nb_fields;
- p_outpic->b_progressive = p_pic->b_progressive;
- p_outpic->b_top_field_first = p_pic->b_top_field_first;
-
- if( p_pic->pf_release )
- p_pic->pf_release( p_pic );
-
- return p_outpic;
+ return CopyInfoAndRelease( p_outpic, p_pic );
}
-inline uint8_t crop( int a );
-inline uint8_t crop( int a )
+static inline uint8_t crop( int a )
{
if( a < 0 ) return 0;
if( a > 255 ) return 255;
#define U 128
#define V 128
+static void mmult( double *res, double *a, double *b )
+{
+ int i, j, k;
+ for( i = 0; i < 3; i++ )
+ {
+ for( j = 0; j < 3; j++ )
+ {
+ res[ i*3 + j ] = 0.;
+ for( k = 0; k < 3; k++ )
+ {
+ res[ i*3 + j ] += a[ i*3 + k ] * b[ k*3 + j ];
+ }
+ }
+ }
+}
+static void make_projection_matrix( filter_t *p_filter, int color, int *matrix )
+{
+ double left_matrix[9] =
+ { 76.24500, 149.68500, 29.07000,
+ -43.02765, -84.47235, 127.50000,
+ 127.50000, -106.76534, -20.73466 };
+ double right_matrix[9] =
+ { 257.00392, 0.00000, 360.31950,
+ 257.00392, -88.44438, -183.53583,
+ 257.00392, 455.41095, 0.00000 };
+ double red = ((double)(( 0xFF0000 & color )>>16))/255.;
+ double green = ((double)(( 0x00FF00 & color )>>8))/255.;
+ double blue = ((double)( 0x0000FF & color ))/255.;
+ double norm = sqrt( red*red + green*green + blue*blue );
+ red /= norm;
+ green /= norm;
+ blue /= norm;
+ /* XXX: We might still need to norm the rgb_matrix */
+ double rgb_matrix[9] =
+ { red*red, red*green, red*blue,
+ red*green, green*green, green*blue,
+ red*blue, green*blue, blue*blue };
+ double result1[9];
+ double result[9];
+ int i;
+ msg_Dbg( p_filter, "red: %f", red );
+ msg_Dbg( p_filter, "green: %f", green );
+ msg_Dbg( p_filter, "blue: %f", blue );
+ mmult( result1, rgb_matrix, right_matrix );
+ mmult( result, left_matrix, result1 );
+ for( i = 0; i < 9; i++ )
+ {
+ matrix[i] = (int)result[i];
+ }
+ msg_Dbg( p_filter, "Projection matrix:" );
+ msg_Dbg( p_filter, "%6d %6d %6d", matrix[0], matrix[1], matrix[2] );
+ msg_Dbg( p_filter, "%6d %6d %6d", matrix[3], matrix[4], matrix[5] );
+ msg_Dbg( p_filter, "%6d %6d %6d", matrix[6], matrix[7], matrix[8] );
+}
+
+static void get_custom_from_yuv420( picture_t *p_inpic, picture_t *p_outpic,
+ int yp, int up, int vp, int *m )
+{
+ uint8_t *y1in = p_inpic->p[yp].p_pixels;
+ uint8_t *y2in;
+ uint8_t *uin = p_inpic->p[up].p_pixels;
+ uint8_t *vin = p_inpic->p[vp].p_pixels;
+
+ uint8_t *y1out = p_outpic->p[yp].p_pixels;
+ uint8_t *y2out;
+ uint8_t *uout = p_outpic->p[up].p_pixels;
+ uint8_t *vout = p_outpic->p[vp].p_pixels;
+
+ const int i_pitch = p_inpic->p[yp].i_pitch;
+ const int i_visible_pitch = p_inpic->p[yp].i_visible_pitch;
+ const int i_visible_lines = p_inpic->p[yp].i_visible_lines;
+
+ const int i_uv_pitch = p_inpic->p[up].i_pitch;
+ const int i_uv_visible_pitch = p_inpic->p[up].i_visible_pitch;
+
+ const uint8_t *yend = y1in + i_visible_lines * i_pitch;
+ while( y1in < yend )
+ {
+ const uint8_t *y1end = y1in + i_visible_pitch;
+ y2in = y1in + i_pitch;
+ y2out = y1out + i_pitch;
+ while( y1in < y1end )
+ {
+ *uout++ = crop( (*y1in * m[3] + (*uin-U) * m[4] + (*vin-V) * m[5])
+ / 65536 + U );
+ *vout++ = crop( (*y1in * m[6] + (*uin-U) * m[7] + (*vin-V) * m[8])
+ / 65536 + V );
+ *y1out++ = crop( (*y1in++ * m[0] + (*uin-U) * m[1] + (*vin-V) * m[2])
+ / 65536 );
+ *y1out++ = crop( (*y1in++ * m[0] + (*uin-U) * m[1] + (*vin-V) * m[2])
+ / 65536 );
+ *y2out++ = crop( (*y2in++ * m[0] + (*uin-U) * m[1] + (*vin-V) * m[2])
+ / 65536 );
+ *y2out++ = crop( (*y2in++ * m[0] + (*uin++ - U) * m[1] + (*vin++ -V) * m[2])
+ / 65536 );
+ }
+ y1in += 2*i_pitch - i_visible_pitch;
+ y1out += 2*i_pitch - i_visible_pitch;
+ uin += i_uv_pitch - i_uv_visible_pitch;
+ uout += i_uv_pitch - i_uv_visible_pitch;
+ vin += i_uv_pitch - i_uv_visible_pitch;
+ vout += i_uv_pitch - i_uv_visible_pitch;
+ }
+}
+static void get_custom_from_yuv422( picture_t *p_inpic, picture_t *p_outpic,
+ int yp, int up, int vp, int *m )
+{
+ uint8_t *y1in = p_inpic->p[yp].p_pixels;
+ uint8_t *uin = p_inpic->p[up].p_pixels;
+ uint8_t *vin = p_inpic->p[vp].p_pixels;
+
+ uint8_t *y1out = p_outpic->p[yp].p_pixels;
+ uint8_t *uout = p_outpic->p[up].p_pixels;
+ uint8_t *vout = p_outpic->p[vp].p_pixels;
+
+ const int i_pitch = p_inpic->p[yp].i_pitch;
+ const int i_visible_pitch = p_inpic->p[yp].i_visible_pitch;
+ const int i_visible_lines = p_inpic->p[yp].i_visible_lines;
+
+ const int i_uv_pitch = p_inpic->p[up].i_pitch;
+ const int i_uv_visible_pitch = p_inpic->p[up].i_visible_pitch;
+
+ const uint8_t *yend = y1in + i_visible_lines * i_pitch;
+ while( y1in < yend )
+ {
+ const uint8_t *y1end = y1in + i_visible_pitch;
+ while( y1in < y1end )
+ {
+ *uout++ = crop( (*y1in * m[3] + (*uin-U) * m[4] + (*vin-V) * m[5])
+ / 65536 + U );
+ *vout++ = crop( (*y1in * m[6] + (*uin-U) * m[7] + (*vin-V) * m[8])
+ / 65536 + V );
+ *y1out++ = crop( (*y1in++ * m[0] + (*uin-U) * m[1] + (*vin-V) * m[2])
+ / 65536 );
+ *y1out++ = crop( (*y1in++ * m[0] + (*uin++ -U) * m[1] + (*vin++ -V) * m[2])
+ / 65536 );
+ }
+ y1in += i_pitch - i_visible_pitch;
+ y1out += i_pitch - i_visible_pitch;
+ uin += i_uv_pitch - i_uv_visible_pitch;
+ uout += i_uv_pitch - i_uv_visible_pitch;
+ vin += i_uv_pitch - i_uv_visible_pitch;
+ vout += i_uv_pitch - i_uv_visible_pitch;
+ }
+}
+
+static void get_custom_from_packedyuv422( picture_t *p_inpic,
+ picture_t *p_outpic,
+ int *m )
+{
+ int i_y_offset, i_u_offset, i_v_offset;
+ if( GetPackedYuvOffsets( p_inpic->format.i_chroma, &i_y_offset,
+ &i_u_offset, &i_v_offset ) != VLC_SUCCESS )
+ return;
+
+ uint8_t *yin = p_inpic->p->p_pixels + i_y_offset;
+ uint8_t *uin = p_inpic->p->p_pixels + i_u_offset;
+ uint8_t *vin = p_inpic->p->p_pixels + i_v_offset;
+
+ uint8_t *yout = p_outpic->p->p_pixels + i_y_offset;
+ uint8_t *uout = p_outpic->p->p_pixels + i_u_offset;
+ uint8_t *vout = p_outpic->p->p_pixels + i_v_offset;
+
+ const int i_pitch = p_inpic->p->i_pitch;
+ const int i_visible_pitch = p_inpic->p->i_visible_pitch;
+ const int i_visible_lines = p_inpic->p->i_visible_lines;
+
+ const uint8_t *yend = yin + i_visible_lines * i_pitch;
+ while( yin < yend )
+ {
+ const uint8_t *ylend = yin + i_visible_pitch;
+ while( yin < ylend )
+ {
+ *uout = crop( (*yin * m[3] + (*uin-U) * m[4] + (*vin-V) * m[5])
+ / 65536 + U );
+ uout += 4;
+ *vout = crop( (*yin * m[6] + (*uin-U) * m[7] + (*vin-V) * m[8])
+ / 65536 + V );
+ vout += 4;
+ *yout = crop( (*yin * m[0] + (*uin-U) * m[1] + (*vin-V) * m[2])
+ / 65536 );
+ yin += 2;
+ yout += 2;
+ *yout = crop( (*yin * m[0] + (*uin-U) * m[1] + (*vin-V) * m[2])
+ / 65536 );
+ yin += 2;
+ yout += 2;
+ uin += 4;
+ vin += 4;
+ }
+ yin += i_pitch - i_visible_pitch;
+ yout += i_pitch - i_visible_pitch;
+ uin += i_pitch - i_visible_pitch;
+ uout += i_pitch - i_visible_pitch;
+ vin += i_pitch - i_visible_pitch;
+ vout += i_pitch - i_visible_pitch;
+ }
+}
+
static void get_red_from_yuv420( picture_t *p_inpic, picture_t *p_outpic,
int yp, int up, int vp )
{
/ 65536 );
*y2out++ = crop( (*y2in++ * 38470 + (*uin-U) * -13239 + (*vin-V) * -27473)
/ 65536 );
- *y2out++ = crop( (*y2in++ * 38470 + (*uin++ - U) * -13239 + (*vin++ -128) * -27473)
+ *y2out++ = crop( (*y2in++ * 38470 + (*uin++ - U) * -13239 + (*vin++ -V) * -27473)
/ 65536 );
}
y1in += 2*i_pitch - i_visible_pitch;
vout += i_uv_pitch - i_uv_visible_pitch;
}
}
+
+static int ExtractCallback( vlc_object_t *p_this, char const *psz_var,
+ vlc_value_t oldval, vlc_value_t newval,
+ void *p_data )
+{
+ VLC_UNUSED(oldval);
+ filter_sys_t *p_sys = (filter_sys_t *)p_data;
+
+ if( !strcmp( psz_var, FILTER_PREFIX "component" ) )
+ {
+ p_sys->i_color = newval.i_int;
+ /* Matrix won't be used for RED, GREEN or BLUE in planar formats */
+ make_projection_matrix( (filter_t *)p_this, p_sys->i_color,
+ p_sys->projection_matrix );
+ }
+ else
+ {
+ msg_Warn( p_this, "Unknown callback command." );
+ }
+ return VLC_SUCCESS;
+}