eessppeelllloo/veejay
1
0
Fork 0
mirror of https://github.com/game-stop/veejay.git synced 2026-06-16 04:31:23 +02:00
Code Issues Packages Projects Releases Wiki Activity

use lookup table for sqrt, remove gauss mask, remove min/max levels (can be achieved by using chained fx)

Browse Source
This commit is contained in:
c0ntrol
2017-01-02 16:26:30 +01:00
parent 106a18e2e3
commit a1a1db0fdf
1 changed files with 21 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files
veejay-current/veejay-server/libvje/effects/rgbkey.c
+21 -109
View File
@@ -30,7 +30,7 @@ vj_effect *rgbkey_init(int w,int h)
{
vj_effect *ve;
ve = (vj_effect *) vj_calloc(sizeof(vj_effect));
ve->num_params = 8;
ve->num_params = 6;
ve->defaults = (int *) vj_calloc(sizeof(int) * ve->num_params); /* default values */
ve->limits[0] = (int *) vj_calloc(sizeof(int) * ve->num_params); /* min */
ve->limits[1] = (int *) vj_calloc(sizeof(int) * ve->num_params); /* max */
@@ -39,9 +39,7 @@ vj_effect *rgbkey_init(int w,int h)
ve->defaults[2] = 255; /* g */
ve->defaults[3] = 0; /* b */
ve->defaults[4] = 1; /* tolerance far */
ve->defaults[5] = 0; /* level min */
ve->defaults[6] = 0xff; /* level max */
ve->defaults[7] = 0; /* alpha-in operator */
ve->defaults[5] = 0; /* alpha-in operator */
ve->limits[0][0] = 1;
ve->limits[1][0] = 255;
@@ -59,18 +57,12 @@ vj_effect *rgbkey_init(int w,int h)
ve->limits[1][4] = 255;
ve->limits[0][5] = 0;
ve->limits[1][5] = 255;
ve->limits[0][6] = 0;
ve->limits[1][6] = 255;
ve->limits[0][7] = 0;
ve->limits[1][7] = 4;
ve->limits[1][5] = 4;
ve->alpha = FLAG_ALPHA_SRC_A | FLAG_ALPHA_SRC_B | FLAG_ALPHA_OPTIONAL | FLAG_ALPHA_IN_OPERATOR;
ve->param_description = vje_build_param_list(ve->num_params,
"Tolerance Near", "Red", "Green", "Blue", "Tolerance Far","Level Min", "Level Max", "Alpha-IN operator");
"Tolerance Near", "Red", "Green", "Blue", "Tolerance Far","Alpha-IN operator");
ve->hints = vje_init_value_hint_list( ve->num_params );
@@ -79,100 +71,22 @@ vj_effect *rgbkey_init(int w,int h)
ve->extra_frame = 1;
ve->sub_format = 1;
ve->rgb_conv = 1;
ve->parallel = 0;
ve->parallel = 1;
vje_build_value_hint_list( ve->hints, ve->limits[1][7],7,
vje_build_value_hint_list( ve->hints, ve->limits[1][5],5,
"Ignore Alpha-IN", "Alpha-IN A", "Alpha-IN B", "Alpha-IN A or B", "Alpha-In A and B" );
return ve;
}
typedef struct {
float radius;
float strength;
int threshold;
float quality;
struct SwsContext *filter_context;
} FilterParam;
static int alloc_sws_context(FilterParam *f, int width, int height, unsigned int flags)
{
SwsVector *vec;
SwsFilter sws_filter;
vec = sws_getGaussianVec(f->radius, f->quality);
if (!vec)
return 0;
sws_scaleVec(vec, f->strength);
vec->coeff[vec->length / 2] += 1.0 - f->strength;
sws_filter.lumH = sws_filter.lumV = vec;
sws_filter.chrH = sws_filter.chrV = NULL;
f->filter_context = sws_getCachedContext(NULL,
width, height, AV_PIX_FMT_GRAY8,
width, height, AV_PIX_FMT_GRAY8,
flags, &sws_filter, NULL, NULL);
sws_freeVec(vec);
if (!f->filter_context)
return 0;
return 1;
}
static uint8_t *temp[2];
static FilterParam *gaussfilter = NULL;
static uint8_t __lookup_table[256];
int rgbkey_malloc(int w, int h)
{
gaussfilter = (FilterParam*) vj_calloc(sizeof(FilterParam));
gaussfilter->radius = 1.3f;
gaussfilter->strength = 1.0f;
gaussfilter->quality = 3.0f;
if(!alloc_sws_context( gaussfilter,w,h,yuv_sws_get_cpu_flags() ) )
return 0;
temp[0] = (uint8_t*) vj_malloc( sizeof(uint8_t) * RUP8(w*h*2));
if(temp[0] == NULL)
return 0;
temp[1] = temp[0] + RUP8(w*h);
return 1;
}
void rgbkey_free()
{
if(temp[0]) {
free(temp[0]);
temp[0] = NULL;
temp[1] = NULL;
}
if( gaussfilter->filter_context ) {
sws_freeContext( gaussfilter->filter_context );
gaussfilter->filter_context = NULL;
}
if( gaussfilter ) {
free(gaussfilter);
gaussfilter = NULL;
}
}
static void gaussblur(uint8_t *dst, const int dst_linesize,const uint8_t *src, const int src_linesize,
const int w, const int h,struct SwsContext *filter_context)
{
const uint8_t* const src_array[4] = {src};
uint8_t *dst_array[4] = {dst};
int src_linesize_array[4] = {src_linesize};
int dst_linesize_array[4] = {dst_linesize};
sws_scale(filter_context, src_array, src_linesize_array,
0, h, dst_array, dst_linesize_array);
}
/*
@@ -180,9 +94,11 @@ static void gaussblur(uint8_t *dst, const int dst_linesize,const uint8_t *src, c
*/
static inline double color_distance( uint8_t Cb, uint8_t Cr, int Cbk, int Crk, double dA, double dB )
{
double tmp = 0.0;
fast_sqrt( tmp, (Cbk - Cb) * (Cbk-Cb) + (Crk - Cr) * (Crk - Cr) );
// double tmp = 0.0;
// fast_sqrt( tmp, (Cbk - Cb) * (Cbk-Cb) + (Crk - Cr) * (Crk - Cr) );
double tmp = sqrt_table_get_pixel( (Cbk-Cb), (Crk-Cr) );
if( tmp < dA ) { /* near color key == bg */
return 0.0; /* near */
}
@@ -193,7 +109,7 @@ static inline double color_distance( uint8_t Cb, uint8_t Cr, int Cbk, int Crk, d
}
void rgbkey_apply(VJFrame *frame, VJFrame *frame2, int tola, int r, int g,int b,
int tolb, int min, int max, int operator)
int tolb, int operator)
{
unsigned int pos;
uint8_t *Y = frame->data[0];
@@ -206,7 +122,6 @@ void rgbkey_apply(VJFrame *frame, VJFrame *frame2, int tola, int r, int g,int b,
uint8_t *B = frame2->data[3];
const int len = frame->len;
int iy=0,iu=128,iv=128;
uint8_t *T = temp[0];
uint8_t op0,op1;
double dtola = (double) tola + 0.5f;
@@ -221,53 +136,50 @@ void rgbkey_apply(VJFrame *frame, VJFrame *frame2, int tola, int r, int g,int b,
case ALPHA_IGNORE:
//ignore alpha-in
for (pos = len; pos != 0; pos--) {
T[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
A[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
}
break;
case ALPHA_IN_A:
for (pos = len; pos != 0; pos--) {
if(A[pos] == 0)
continue;
T[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
A[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
}
break;
case ALPHA_IN_A_OR_B:
for (pos = len; pos != 0; pos--) {
if(A[pos] == 0 || B[pos] == 0)
continue;
T[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
A[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
}
break;
case ALPHA_IN_A_AND_B:
for (pos = len; pos != 0; pos--) {
if(A[pos] == 0 && B[pos] == 0)
continue;
T[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
A[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
}
break;
case ALPHA_IN_B:
for (pos = len; pos != 0; pos--) {
if(B[pos] == 0)
continue;
T[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
A[pos] = (uint8_t)( 255.0 * color_distance( Cb[pos],Cr[pos],iu,iv,dtola,dtolb ) );
}
break;
}
/* choke matte */
// reduces detail
gaussblur( A, frame->width, temp[0], frame->width, frame->width,frame->height,gaussfilter->filter_context );
/* level correction table */
__init_lookup_table( __lookup_table, 256, (float)min, (float)max, 0, 0xff );
/* composite bg onto fg */
for( pos = 0; pos < len; pos ++ ) {
op0 = __lookup_table[A[pos]];
op0 = A[pos];
op1 = 0xff - op0;
Y[pos] = ((op0 * Y[pos]) + (op1 * Y2[pos]))>> 8;
Cb[pos] = ((op0 * Cb[pos]) + (op1 * Cb2[pos]))>> 8;
Cr[pos] = ((op0 * Cr[pos]) + (op1 * Cr2[pos]))>>8;
}
/* Set by color key
Gaussblur on alpha channel
Mixdown */
}