veejay/veejay-current/veejay-server/libvje/effects/reflection.c

/* 
 * 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.
 */

/* Copyright (C) 2002 W.P. van Paassen - peter@paassen.tmfweb.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; see the file COPYING.  If not, write to the Free
   Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */

/* This effect was inspired by an article by Sqrt(-1) */
/* 08-22-02 Optimized by WP */
/* note that the code has not been fully optimized */

/* orignal code for RGB, define INTENSITY( r + b + g / 3 ),
   this effect works in YCbCr space now. */

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <libvjmem/vjmem.h>
#include "reflection.h"
#include <math.h>

static short reflect_aSin[2048];
static int reflection_map[2048][256];
static int sin_index = 0;
static int sin_index2 = 20;
static uint8_t *reflection_buffer;

vj_effect *reflection_init(int width,int height)
{
      vj_effect *ve = (vj_effect *) vj_calloc(sizeof(vj_effect));
    ve->num_params = 3;
    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] = 2;
    ve->defaults[1] = 5;
    ve->defaults[2] = 1;
    ve->limits[0][0] = 1;
    ve->limits[1][0] = 256;
    ve->limits[0][1] = 1;
    ve->limits[1][1] = 256;
    ve->limits[0][2] = 0;
    ve->limits[1][2] = 1;
    ve->description = "Bump 2D";
    ve->sub_format = 0;
    ve->extra_frame = 0;
	ve->has_user = 0;
	ve->param_description = vje_build_param_list( ve->num_params, "Value 1", "Value 2", "Mode");
    return ve;
}

int reflection_malloc(int width, int height)
{ 
  int i, x, y;
    float rad;
 
    for (i = 0; i < width; i++) {
	rad = (float) i * 0.0174532 * 0.703125;
	reflect_aSin[i] = (short) ((sin(rad) * 100.0) + 256.0);
    }
    for (x = 0; x < width; ++x) {
	for (y = 0; y < 256; ++y) {
	    float xx = (x - 128) / 128.0;
	    float yy = (y - 128) / 128.0;
	    float zz = 1.0 - sqrt(xx * xx + yy * yy);
	    zz *= 255.0;
	    if (zz < 0.0)
		zz = 0.0;
	    reflection_map[x][y] = (int) zz;
	}
    }
    reflection_buffer = (uint8_t *) vj_malloc(sizeof(uint8_t) * width + 1);	/* fixme */
    if(!reflection_buffer) return 0;

    return 1;


}



void reflection_free() {
  if(reflection_buffer) free(reflection_buffer);
  reflection_buffer = NULL;
}


void reflection_apply(VJFrame *frame, int width, int height, int index1,
		      int index2, int move)
{
    unsigned int normalx, normaly, x, y;
    unsigned int lightx, lighty, temp;
	int uv_height = frame->uv_height;
    int uv_width = frame->uv_width;
    uint8_t *row = frame->data[0] + width + 1;
    uint8_t *cbrow = frame->data[1] + uv_width + 1;
	uint8_t *crrow = frame->data[2] + uv_width + 1;
    uint8_t *Y = frame->data[0];
	uint8_t *Cb= frame->data[1];
	uint8_t *Cr= frame->data[2];

    lightx = reflect_aSin[sin_index];
    lighty = reflect_aSin[sin_index2];


    if (!move) {
		sin_index = index1;
		sin_index2 = index2;
    } else {
		sin_index += index1;
		sin_index2 += index2;
    }
    sin_index &= 511;
    sin_index2 &= 511;

    for (x = 0; x < width; x++) {
		reflection_buffer[x]= Y[x];
    }


    for (y = 1; y < height - 1; y++) {
		uint8_t p;
		temp = lighty - y;
		p = Y[x + (y * width)];

		for (x = 0; x < width; x++) {
		    int i1 = (int) p;
		    int i2 = Y[x + 1 + (y * width)];	/* deviate */
		    int i3 = (int) reflection_buffer[x];
		    normalx = i2 - i1 + lightx - x;
		    normaly = i1 - i3 + temp;
		    if (normalx < 0)
				normalx = 0;
		    else if (normalx > 255)
				normalx = 255;
		    if (normaly < 0)
				normaly = 0;
		    else if (normaly > 255)
				normaly = 255;
		    *row++ = reflection_map[normalx][normaly];
		    p = i2;
		    reflection_buffer[x] = i2;
		}
		*row += 2;
    }

    for (y = 1; y < uv_height - 1; y++) {
		uint8_t p;
		temp = lighty - y;
		p = Y[(x<<frame->shift_h) + (y << frame->shift_v) * width];

		for (x = 0; x < uv_width; x++) {
		    int i1 = (int) p;
		    int i2 = Y[(x<<frame->shift_h) + 1 + ((y<<frame->shift_v) * width)];	
		    int i3 = (int) reflection_buffer[(x<<frame->shift_h)];
		    normalx = i2 - i1 + lightx - x;
		    normaly = i1 - i3 + temp;
		    if (normalx < 0)
				normalx = 0;
		    else if (normalx > 255)
				normalx = 255;
		    if (normaly < 0)
				normaly = 0;
		    else if (normaly > 255)
				normaly = 255;
		    *cbrow++ = ((reflection_map[normalx][normaly] * (Cb[x + 1 + (y * uv_width)]-128)) >> 8) +128;
			*crrow++ = ((reflection_map[normalx][normaly] * (Cr[x + 1 + (y * uv_width)]-128)) >> 8) +128;
		    p = i2;
		    reflection_buffer[x] = i2;
		}
		*row += 2;
    }


    //sin_index+=n;
    //sin_index &= 511;
    //sin_index2 +=n-2;
    //sin_index2 &= 511;

}