eessppeelllloo/veejay
1
0
Fork 0
mirror of https://github.com/game-stop/veejay.git synced 2025-12-17 21:30:02 +01:00
Code Issues Packages Projects Releases Wiki Activity
Files
b96bb5fe2f316f764cd9da99b972981bf6c186a3
veejay/veejay-current/veejay-server/libvje/effects/rgbkey.c

330 lines
8.5 KiB
C
Raw Blame History

/*
* Linux VeeJay
*
* Copyright(C)2004 Niels Elburg <elburg@hio.hen.nl>
*
* 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307 , USA.
*/
#include <stdint.h>
#include <stdio.h>
#include <libvjmem/vjmem.h>
#include "rgbkey.h"
#include <stdlib.h>
#include <math.h>
#include "common.h"
/*
This effect is based on this small project:
http://www.cs.utah.edu/~michael/chroma/
The algorithm decides which pixels belong to resp. foreground
or background. Other effects that make use of this same algorithm are
Complex Invert, Complex Negation, Complex Saturation, Smooth RGB Key
Isolate by Color , Complex Threshold and Blend by Color Key,
*/
/*
(march,2005) fixed flaw (signed vs. unsigned) in algorithm
use selectable rgb -> yuv formula,
*/
vj_effect *rgbkey_init(int w,int h)
{
vj_effect *ve;
ve = (vj_effect *) vj_calloc(sizeof(vj_effect));
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 */
ve->defaults[0] = 4500; /* angle , 45 degrees*/
ve->defaults[1] = 0; /* r */
ve->defaults[2] = 0; /* g */
ve->defaults[3] = 255; /* b */
ve->defaults[4] = 1; /* type */
ve->defaults[5] = 3500; /* noise */
ve->limits[0][0] = 1;
ve->limits[1][0] = 9000;
ve->limits[0][1] = 0;
ve->limits[1][1] = 255;
ve->limits[0][2] = 0;
ve->limits[1][2] = 255;
ve->limits[0][3] = 0;
ve->limits[1][3] = 255;
ve->limits[0][4] = 0;
ve->limits[1][4] = 1; /* total noise suppression off */
ve->limits[0][5] = 1;
ve->limits[1][5] = 5500;
ve->param_description = vje_build_param_list(ve->num_params, "Angle", "Red", "Green", "Blue", "Mode", "Noise suppression");
ve->has_user = 0;
ve->description = "Chroma Key (RGB)";
ve->extra_frame = 1;
ve->sub_format = 1;
ve->rgb_conv = 1;
ve->parallel = 1;
return ve;
}
/*
void rgbkey_scan_fg(uint8_t * src2[3], int *r, int *g, int *b)
{
*r = (int) (+(1.0 * Y2[0]) + (0 * Cb2[0]) +
(1.402 * Cr2[0]));
*g = (int) (+(1.0 * Y2[0]) - (0.344136 * Cb2[0]) +
(-0.714136 * Cr2[0]));
*b = (int) (+(1.0 * Y2[0]) + (1.772 * Cb2[0]) +
(0 * Cr2[0]));
}
*/
void rgbkey_apply1(VJFrame *frame, VJFrame *frame2, int width,
int height, int i_angle, int r, int g,
int b, int i_noise)
{
uint8_t *fg_y, *fg_cb, *fg_cr;
uint8_t *bg_y, *bg_cb, *bg_cr;
int accept_angle_tg, accept_angle_ctg, one_over_kc;
int kfgy_scale, kg;
uint8_t cb, cr;
int kbg, x1, y1;
float kg1, tmp, aa = 255.0f, bb = 255.0f, _y = 0;
float angle = (float) i_angle / 100.0f;
float noise_level = (i_noise / 100.0f);
unsigned int pos;
uint8_t val, tmp1;
uint8_t *Y = frame->data[0];
uint8_t *Cb= frame->data[1];
uint8_t *Cr= frame->data[2];
uint8_t *Y2 = frame2->data[0];
uint8_t *Cb2= frame2->data[1];
uint8_t *Cr2= frame2->data[2];
uint8_t iy,iu,iv;
_rgb2yuv( r,g,b, iy,iu,iv );
_y = (float) iy;
aa = (float) iu;
bb = (float) iv;
tmp = sqrt(((aa * aa) + (bb * bb)));
cb = 0xff * (aa / tmp);
cr = 0xff * (bb / tmp);
kg1 = tmp;
/* obtain coordinate system for cb / cr */
accept_angle_tg = 0xf * tan(M_PI * angle / 180.0f);
accept_angle_ctg = 0xf / tan(M_PI * angle / 180.0f);
tmp = 1 / kg1;
one_over_kc = 0xff * 2 * tmp - 0xff;
kfgy_scale = 0xf * (float) (_y) / kg1;
kg = kg1;
/* intialize pointers */
fg_y = Y;
fg_cb = Cb;
fg_cr = Cr;
/* 2005: swap these !! */
bg_y = Y2;
bg_cb = Cb2;
bg_cr = Cr2;
for (pos = (width * height); pos != 0; pos--) {
short xx, yy;
/* convert foreground to xz coordinates where x direction is
defined by key color */
xx = (((fg_cb[pos]) * cb) + ((fg_cr[pos]) * cr)) >> 7;
yy = (((fg_cr[pos]) * cb) - ((fg_cb[pos]) * cr)) >> 7;
/* accept angle should not be > 90 degrees
reasonable results between 10 and 80 degrees.
*/
val = (xx * accept_angle_tg) >> 4;
if (abs(yy) < val) {
/* compute fg, suppress fg in xz according to kfg
*/
val = (yy * accept_angle_ctg) >> 4;
x1 = abs(val);
y1 = yy;
tmp1 = xx - x1;
kbg = (tmp1 * one_over_kc) >> 1;
if (kbg < 0)
kbg = 0;
if (kbg > 255)
kbg = 255;
val = (tmp1 * kfgy_scale) >> 4;
if( val > 255 )
val = 255;
val = fg_y[pos] - val;
Y[pos] = val;
// convert suppressed fg back to cbcr
Cb[pos] = ((x1 * cb) - (y1 * cr)) >> 7;
Cr[pos] = ((x1 * cr) - (y1 * cb)) >> 7;
// deal with noise
val = (yy * yy) + (kg * kg);
if (val < (noise_level * noise_level)) {
kbg = 255;
}
Y[pos] = (Y[pos] + (kbg * bg_y[pos])) >> 8;
Cb[pos] = (Cb[pos] + (kbg * bg_cb[pos])) >> 8;
Cr[pos] = (Cr[pos] + (kbg * bg_cr[pos])) >> 8;
}
}
}
void rgbkey_apply2(VJFrame *frame, VJFrame *frame2, int width,
int height, int i_angle,int r, int g,
int b, int i_noise)
{
uint8_t *fg_y, *fg_cb, *fg_cr;
uint8_t *bg_y, *bg_cb, *bg_cr;
int accept_angle_tg, accept_angle_ctg, one_over_kc;
int kfgy_scale, kg;
uint8_t cb, cr;
int kbg, x1, y1;
float kg1, tmp, aa = 255.0, bb = 255.0, _y = 0;
float angle = (float) i_angle / 100.0f;
float noise_level = (i_noise / 100.0f);
unsigned int pos;
uint8_t val, tmp1;
uint8_t *Y = frame->data[0];
uint8_t *Cb= frame->data[1];
uint8_t *Cr= frame->data[2];
uint8_t *Y2 = frame2->data[0];
int8_t *Cb2= frame2->data[1];
uint8_t *Cr2= frame2->data[2];
uint8_t iy,iu,iv;
_rgb2yuv( r,g,b, iy,iu,iv );
_y = (float) iy;
aa = (float) iu;
bb = (float) iv;
tmp = sqrt(((aa * aa) + (bb * bb)));
cb = 255 * (aa / tmp);
cr = 255 * (bb / tmp);
kg1 = tmp;
/* obtain coordinate system for cb / cr */
accept_angle_tg = 0xf * tan(M_PI * angle / 180.0);
accept_angle_ctg = 0xf / tan(M_PI * angle / 180.0);
tmp = 1 / kg1;
one_over_kc = 0xff * 2 * tmp - 0xff;
kfgy_scale = 0xf * (float) (_y) / kg1;
kg = kg1;
/* intialize pointers */
fg_y = Y;
fg_cb = Cb;
fg_cr = Cr;
bg_y = Y2;
bg_cb = Cb2;
bg_cr = Cr2;
int len = frame->len;
for (pos = 0; pos < len; pos++) {
short xx, yy;
/* convert foreground to xz coordinates where x direction is
defined by key color */
xx = (((fg_cb[pos]) * cb) + ((fg_cr[pos]) * cr)) >> 7;
yy = (((fg_cr[pos]) * cb) - ((fg_cb[pos]) * cr)) >> 7;
/* accept angle should not be > 90 degrees
reasonable results between 10 and 80 degrees.
*/
val = (xx * accept_angle_tg) >> 4;
if (abs(yy) < val) {
/* compute fg, suppress fg in xz according to kfg */
val = (yy * accept_angle_ctg) >> 4;
x1 = abs(val);
y1 = yy;
tmp1 = xx - x1;
kbg = (tmp1 * one_over_kc) >> 1;
if (kbg < 0)
kbg = 0;
if (kbg > 255)
kbg = 255;
val = (tmp1 * kfgy_scale) >> 4;
val = fg_y[pos] - val;
Y[pos] = val;
/* convert suppressed fg back to cbcr */
Cb[pos] = ((x1 * cb) - (y1 * cr)) >> 7;
Cr[pos] = ((x1 * cr) - (y1 * cb)) >> 7;
/* deal with noise */
val = (yy * yy) + (kg * kg);
if (val < (noise_level * noise_level)) {
Y[pos] = 0; Cb[pos] = 128; Cr[pos] = 128;
//kbg = 0xff;
} else {
Y[pos] = (Y[pos] + (kbg * bg_y[pos])) >> 8;
Cb[pos] = (Cb[pos] + (kbg * bg_cb[pos])) >> 8;
Cr[pos] = (Cr[pos] + (kbg * bg_cr[pos])) >> 8;
}
}
}
}
/* this is the same as rgbkey_apply1, but here we have total noise suppression
*/
void rgbkey_apply(VJFrame *frame, VJFrame *frame2, int width,
int height, int i_angle, int red, int green,
int blue, int type, int i_noise)
{
switch (type) {
case 0:
rgbkey_apply1(frame, frame2, width, height, i_angle, red,
green, blue, i_noise);
break;
case 1:
rgbkey_apply2(frame, frame2, width, height, i_angle, red,
green, blue, i_noise);
break;
}
}
void rgbkey_free(){}
Reference in New Issue View Git Blame Copy Permalink