mirror of
https://github.com/game-stop/veejay.git
synced 2025-12-23 16:20:03 +01:00
Initial checkin of veejay 1.4
git-svn-id: svn://code.dyne.org/veejay/trunk@1172 eb8d1916-c9e9-0310-b8de-cf0c9472ead5
This commit is contained in:
Niels Elburg
327
veejay-current/veejay-server/libvje/effects/lumakey.c
Normal file
327
veejay-current/veejay-server/libvje/effects/lumakey.c
Normal file
@@ -0,0 +1,327 @@
|
||||
/*
|
||||
* Linux VeeJay
|
||||
*
|
||||
* Copyright(C)2002 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 <libvjmem/vjmem.h>
|
||||
#include "lumakey.h"
|
||||
|
||||
vj_effect *lumakey_init(int width, int height)
|
||||
{
|
||||
vj_effect *ve = (vj_effect *) vj_calloc(sizeof(vj_effect));
|
||||
ve->num_params = 5;
|
||||
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->limits[0][0] = 0; /* feather */
|
||||
ve->limits[1][0] = 255;
|
||||
ve->limits[0][1] = 0; /* threshold min */
|
||||
ve->limits[1][1] = 255;
|
||||
ve->limits[0][2] = 0; /* threshold max */
|
||||
ve->limits[1][2] = 255;
|
||||
ve->limits[0][3] = 1; /* distance */
|
||||
ve->limits[1][3] = width;
|
||||
ve->limits[0][4] = 0; /* type */
|
||||
ve->limits[1][4] = 2;
|
||||
ve->defaults[0] = 255;
|
||||
ve->defaults[1] = 150;
|
||||
ve->defaults[2] = 200;
|
||||
ve->defaults[3] = 1;
|
||||
ve->defaults[4] = 1;
|
||||
ve->description = "Luma Key";
|
||||
ve->extra_frame = 1;
|
||||
ve->sub_format = 1;
|
||||
ve->has_user = 0;
|
||||
ve->param_description = vje_build_param_list(ve->num_params, "Feather", "Min Threshold","Max Threshold","Distance", "Mode" );
|
||||
|
||||
return ve;
|
||||
}
|
||||
|
||||
|
||||
|
||||
void lumakey_simple(uint8_t * yuv1[3], uint8_t * yuv2[3], int width,
|
||||
int height, int threshold, int threshold2, int opacity)
|
||||
{
|
||||
|
||||
unsigned int x, y, len = width * height;
|
||||
uint8_t a1, a2;
|
||||
unsigned int op0, op1;
|
||||
uint8_t Y, Cb, Cr;
|
||||
op1 = (opacity > 255) ? 255 : opacity;
|
||||
op0 = 255 - op1;
|
||||
|
||||
for (y = 0; y < len; y += width) {
|
||||
for (x = 0; x < width; x++) {
|
||||
a1 = yuv1[0][x + y];
|
||||
a2 = yuv2[0][x + y];
|
||||
/*
|
||||
if ( a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2 )/255;
|
||||
Cb = (op0 * yuv1[1][x+y] + op1 * yuv2[1][x+y])/255;
|
||||
Cr =(op0 * yuv1[2][x+y] + op1 * yuv2[2][x+y])/255;
|
||||
|
||||
}
|
||||
*/
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][x + y] + op1 * yuv2[1][x + y]) >> 1;
|
||||
Cr = (op0 * yuv1[2][x + y] + op1 * yuv2[2][x + y]) >> 1;
|
||||
yuv1[0][x + y] = Y; // < 16 ? 16 : Y > 235 ? 235 : Y;
|
||||
yuv1[1][x + y] = Cb; // < 16 ? 16 : Cb > 240 ? 240 : Cb;
|
||||
yuv1[2][x + y] = Cr; // < 16 ? 16 : Cr > 240 ? 240 : Cr;
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void lumakey_smooth(uint8_t * yuv1[3], uint8_t * yuv2[3], int width,
|
||||
int height, int threshold, int threshold2, int opacity,
|
||||
int distance)
|
||||
{
|
||||
|
||||
unsigned int x, y = 0, len = width * height;
|
||||
uint8_t a1, a2;
|
||||
unsigned int op0, op1;
|
||||
uint8_t Y, Cb, Cr;
|
||||
unsigned int soft0, soft1;
|
||||
unsigned int t2, t3;
|
||||
uint8_t p1, p2;
|
||||
op1 = (opacity > 255) ? 255 : opacity;
|
||||
op0 = 255 - op1;
|
||||
|
||||
soft0 = 255 / distance;
|
||||
soft1 = 255 - soft0;
|
||||
|
||||
t2 = threshold - distance; // 0 - 4
|
||||
t3 = threshold2 + distance; // 3 + 4
|
||||
|
||||
/* first row */
|
||||
|
||||
if (t2 < 0)
|
||||
t2 = 0;
|
||||
if (t3 > 255)
|
||||
t3 = 255;
|
||||
|
||||
for (x = 0; x < width; x++) {
|
||||
a1 = yuv1[0][x];
|
||||
a2 = yuv2[0][x];
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][x] + op1 * yuv2[1][x]) >> 8;
|
||||
Cr = (op0 * yuv1[2][x] + op1 * yuv2[2][x]) >> 8;
|
||||
|
||||
yuv1[0][x] = Y; //< 16 ? 16 : Y > 235 ? 235 : Y;
|
||||
yuv1[1][x] = Cb; // < 16 ? 16 : Cb > 240 ? 240 : Cb;
|
||||
yuv1[2][x] = Cr; // < 16 ? 16 : Cr > 240 ? 240 : Cr;
|
||||
}
|
||||
}
|
||||
|
||||
for (y = width; y < len - width; y += width) {
|
||||
/* first pixel in column */
|
||||
a1 = yuv1[0][y];
|
||||
a2 = yuv2[0][y];
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][y] + op1 * yuv2[1][y]) >> 8;
|
||||
Cr = (op0 * yuv1[2][y] + op1 * yuv2[2][y]) >> 8;
|
||||
}
|
||||
/* rest of pixels in column */
|
||||
for (x = 1; x < width - 1; x++) {
|
||||
a1 = yuv1[0][x + y];
|
||||
a2 = yuv2[0][x + y];
|
||||
|
||||
if ((a1 >= t2 && a1 < threshold)
|
||||
|| (a1 > threshold2 && a1 <= t3)) {
|
||||
/* special case */
|
||||
p1 = ( /* calculate mean of a1 */
|
||||
yuv1[0][y - width + x - 1] +
|
||||
yuv1[0][y - width + x + 1] +
|
||||
yuv1[0][y - width + x] +
|
||||
yuv1[0][y + x] +
|
||||
yuv1[0][y + x - 1] +
|
||||
yuv1[0][y + x + 1] +
|
||||
yuv1[0][y + width + x] +
|
||||
yuv1[0][y + width + x + 1] +
|
||||
yuv1[0][y + width + x - 1]
|
||||
) / 9;
|
||||
p2 = ( /* calculate mean of a1 */
|
||||
yuv2[0][y - width + x - 1] +
|
||||
yuv2[0][y - width + x + 1] +
|
||||
yuv2[0][y - width + x] +
|
||||
yuv2[0][y + x] +
|
||||
yuv2[0][y + x - 1] +
|
||||
yuv2[0][y + x + 1] +
|
||||
yuv2[0][y + width + x] +
|
||||
yuv2[0][y + width + x + 1] +
|
||||
yuv2[0][y + width + x - 1]
|
||||
) / 9;
|
||||
|
||||
yuv1[0][x + y] = (op0 * p1 + op1 * p2) >> 8;
|
||||
yuv1[1][x + y] =
|
||||
(op0 * yuv1[1][x + y] + op1 * yuv2[1][x + y]) >> 8;
|
||||
yuv1[2][x + y] =
|
||||
(op0 * yuv1[2][x + y] + op1 * yuv2[2][x + y]) >> 8;
|
||||
|
||||
} else {
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][x + y] +
|
||||
op1 * yuv2[1][x + y]) >> 8;
|
||||
Cr = (op0 * yuv1[2][x + y] +
|
||||
op1 * yuv2[2][x + y]) >> 8;
|
||||
|
||||
yuv1[0][x + y] = Y; // < 16 ? 16 : Y > 235 ? 235 : Y;
|
||||
yuv1[1][x + y] = Cb; // < 16 ? 16 : Cb > 240 ? 240 : Cb;
|
||||
yuv1[2][x + y] = Cr; // < 16 ? 16 : Cr > 240 ? 240 : Cr;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
/* last row */
|
||||
for (x = len - width; x < len; x++) {
|
||||
a1 = yuv1[0][x];
|
||||
a2 = yuv2[0][x];
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][x] + op1 * yuv2[1][x]) >> 8;
|
||||
Cr = (op0 * yuv1[2][x] + op1 * yuv2[2][x]) >> 8;
|
||||
|
||||
yuv1[0][x] = Y; // < 16 ? 16 : Y > 235 ? 235 : Y;
|
||||
yuv1[1][x] = Cb; // < 16 ? 16 : Cb > 240 ? 240 : Cb;
|
||||
yuv1[2][x] = Cr; // < 16 ? 16 : Cr > 240 ? 240 : Cr;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
void lumakey_smooth_white(uint8_t * yuv1[3], uint8_t * yuv2[3], int width,
|
||||
int height, int threshold, int threshold2,
|
||||
int opacity, int distance)
|
||||
{
|
||||
|
||||
unsigned int x, y = 0, len = width * height;
|
||||
uint8_t a1, a2;
|
||||
unsigned int op0, op1;
|
||||
uint8_t Y, Cb, Cr;
|
||||
unsigned int soft0, soft1;
|
||||
unsigned int t2, t3;
|
||||
op1 = (opacity > 255) ? 255 : opacity;
|
||||
op0 = 255 - op1;
|
||||
|
||||
soft0 = 255 / distance;
|
||||
soft1 = 255 - soft0;
|
||||
|
||||
t2 = threshold - distance;
|
||||
t3 = threshold2 + distance;
|
||||
|
||||
/* first row */
|
||||
|
||||
for (x = 0; x < width; x++) {
|
||||
a1 = yuv1[0][x];
|
||||
a2 = yuv2[0][x];
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][x] + op1 * yuv2[1][x]) >> 8;
|
||||
Cr = (op0 * yuv1[2][x] + op1 * yuv2[2][x]) >> 8;
|
||||
|
||||
yuv1[0][x] = Y; // < 16 ? 16 : Y > 235 ? 235 : Y;
|
||||
yuv1[1][x] = Cb; // < 16 ? 16 : Cb > 240 ? 240 : Cb;
|
||||
yuv1[2][x] = Cr; // < 16 ? 16 : Cr > 240 ? 240 : Cr;
|
||||
}
|
||||
}
|
||||
|
||||
for (y = width; y < len - width; y += width) {
|
||||
/* first pixel in column */
|
||||
a1 = yuv1[0][y];
|
||||
a2 = yuv2[0][y];
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][y] + op1 * yuv2[1][y]) >> 8;
|
||||
Cr = (op0 * yuv1[2][y] + op1 * yuv2[2][y]) >> 8;
|
||||
}
|
||||
/* rest of pixels in column */
|
||||
/* rest of pixels in column */
|
||||
for (x = 1; x < width - 1; x++) {
|
||||
a1 = yuv1[0][x + y];
|
||||
a2 = yuv2[0][x + y];
|
||||
|
||||
if ((a1 >= t2 && a1 < threshold)
|
||||
|| (a1 > threshold2 && a1 <= t3)) {
|
||||
/* special case */
|
||||
yuv1[0][x + y] = 235;
|
||||
yuv1[1][x + y] = 128;
|
||||
yuv1[2][x + y] = 128;
|
||||
} else {
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][x + y] +
|
||||
op1 * yuv2[1][x + y]) >> 8;
|
||||
Cr = (op0 * yuv1[2][x + y] +
|
||||
op1 * yuv2[2][x + y]) >> 8;
|
||||
|
||||
yuv1[0][x + y] = Y; // < 16 ? 16 : Y > 235 ? 235 : Y;
|
||||
yuv1[1][x + y] = Cb; // < 16 ? 16 : Cb > 240 ? 240 : Cb;
|
||||
yuv1[2][x + y] = Cr; // < 16 ? 16 : Cr > 240 ? 240 : Cr;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
/* last row */
|
||||
for (x = len - width; x < len; x++) {
|
||||
a1 = yuv1[0][x];
|
||||
a2 = yuv2[0][x];
|
||||
if (a1 >= threshold && a1 <= threshold2) {
|
||||
Y = (op0 * a1 + op1 * a2) >> 8;
|
||||
Cb = (op0 * yuv1[1][x] + op1 * yuv2[1][x]) >> 8;
|
||||
Cr = (op0 * yuv1[2][x] + op1 * yuv2[2][x]) >> 8;
|
||||
|
||||
yuv1[0][x] = Y; // < 16 ? 16 : Y > 235 ? 235 : Y;
|
||||
yuv1[1][x] = Cb; // < 16 ? 16 : Cb > 240 ? 240 : Cb;
|
||||
yuv1[2][x] = Cr; // < 16 ? 16 : Cr > 240 ? 240 : Cr;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
void lumakey_apply( VJFrame *frame, VJFrame *frame2, int width,
|
||||
int height, int type, int threshold, int threshold2,
|
||||
int feather, int d)
|
||||
{
|
||||
|
||||
switch (type) {
|
||||
case 0:
|
||||
/* normal overlay */
|
||||
lumakey_simple(frame->data, frame2->data, width, height, threshold, threshold2,
|
||||
feather);
|
||||
break;
|
||||
case 1:
|
||||
/* threshold */
|
||||
lumakey_smooth_white(frame->data, frame2->data, width, height, threshold,
|
||||
threshold2, feather, d);
|
||||
break;
|
||||
case 2:
|
||||
lumakey_smooth(frame->data, frame2->data, width, height, threshold, threshold2,
|
||||
feather, d);
|
||||
break;
|
||||
}
|
||||
|
||||
}
|
||||
void lumakey_free(){}
|
||||
Reference in New Issue
Block a user