/* For information on usage and redistribution, and for a DISCLAIMER OF ALL
* WARRANTIES, see the file, "LICENSE.txt," in this distribution.

iem_tab written by Thomas Musil, Copyright (c) IEM KUG Graz Austria 2000 - 2005 */

#ifdef NT
#pragma warning( disable : 4244 )
#pragma warning( disable : 4305 )
#endif


#include "m_pd.h"
#include "iemlib.h"
#include "iem_tab.h"
#include <math.h>
#include <stdio.h>
#include <string.h>

#define TABDUMTAB1SIZE 256
#define TABDUMTAB2SIZE 1024

static t_float tab_sin_exptab[TABDUMTAB1SIZE], tab_sin_mantissatab[TABDUMTAB2SIZE];

static void init_tab_sin(void)
{
	int i;
	
	for (i=0; i<TABDUMTAB1SIZE; i++)
	{
		t_float f;
		long l = (i ? (i == TABDUMTAB1SIZE-1 ? TABDUMTAB1SIZE-2 : i) : 1)<< 23;

		*(long *)(&f) = l;
		tab_sin_exptab[i] = 1.0f/sqrt(f);	
	}

	for (i=0; i<TABDUMTAB2SIZE; i++)
	{
		t_float f = 1.0f + (1.0f / (t_float)TABDUMTAB2SIZE) * (t_float)i;

		tab_sin_mantissatab[i] = 1.0f / sqrt(f);	
	}
}


/* -------------------------- tab_sin ------------------------------ */

typedef struct _tab_sin
{
	t_object	x_obj;
	int				x_size_src1;
	int				x_size_dst;
	int				x_offset_src1;
	int				x_offset_dst;
	float			*x_beg_mem_src1;
	float			*x_beg_mem_dst;
	t_symbol	*x_sym_scr1;
	t_symbol	*x_sym_dst;
} t_tab_sin;

static t_class *tab_sin_class;

static void tab_sin_src(t_tab_sin *x, t_symbol *s)
{
	x->x_sym_scr1 = s;
}

static void tab_sin_dst(t_tab_sin *x, t_symbol *s)
{
	x->x_sym_dst = s;
}

static void tab_sin_bang(t_tab_sin *x)
{
	int i, n;
	int ok_src, ok_dst;
	t_float *vec_src, *vec_dst;

	ok_src = iem_tab_check_arrays(gensym("tab_sin"), x->x_sym_scr1, &x->x_beg_mem_src1, &x->x_size_src1, 0);
	ok_dst = iem_tab_check_arrays(gensym("tab_sin"), x->x_sym_dst, &x->x_beg_mem_dst, &x->x_size_dst, 0);

	if(ok_src && ok_dst)
	{
		if(x->x_size_src1 < x->x_size_dst)
			n = x->x_size_src1;
		else
			n = x->x_size_dst;
		vec_src = x->x_beg_mem_src1;
		vec_dst = x->x_beg_mem_dst;
		if(n)
		{
			t_garray *a;

			while(n--)
			{
				t_float f = *vec_src;
				long l = *(long *)(vec_src++);

				if(f < 0.0f)
					*vec_dst++ = 0.0f;
				else
				{
					t_float g = tab_sin_exptab[(l >> 23) & 0xff] * tab_sin_mantissatab[(l >> 13) & 0x3ff];

					*vec_dst++ = f*g*(1.5f - 0.5f * g * g * f);
				}
			}
			outlet_bang(x->x_obj.ob_outlet);
			a = (t_garray *)pd_findbyclass(x->x_sym_dst, garray_class);
			garray_redraw(a);
		}
	}
}

static void tab_sin_list(t_tab_sin *x, t_symbol *s, int argc, t_atom *argv)
{
	int beg_src, beg_dst;
	int i, n;
	int ok_src, ok_dst;
	t_float *vec_src, *vec_dst;

	if((argc >= 3) &&
		IS_A_FLOAT(argv,0) &&
		IS_A_FLOAT(argv,1) &&
		IS_A_FLOAT(argv,2))
	{
		beg_src = (int)atom_getintarg(0, argc, argv);
		beg_dst = (int)atom_getintarg(1, argc, argv);
		n = (int)atom_getintarg(2, argc, argv);
		if(beg_src < 0)
			beg_src = 0;
		if(beg_dst < 0)
			beg_dst = 0;
		if(n < 0)
			n = 0;

		ok_src = iem_tab_check_arrays(gensym("tab_sin"), x->x_sym_scr1, &x->x_beg_mem_src1, &x->x_size_src1, beg_src+n);
		ok_dst = iem_tab_check_arrays(gensym("tab_sin"), x->x_sym_dst, &x->x_beg_mem_dst, &x->x_size_dst, beg_dst+n);

		if(ok_src && ok_dst)
		{
			vec_src = x->x_beg_mem_src1 + beg_src;
			vec_dst = x->x_beg_mem_dst + beg_dst;
			if(n)
			{
				t_garray *a;

				while(n--)
				{
					t_float f = *vec_src;
					long l = *(long *)(vec_src++);

					if(f < 0.0f)
						*vec_dst++ = 0.0f;
					else
					{
						t_float g = tab_sin_exptab[(l >> 23) & 0xff] * tab_sin_mantissatab[(l >> 13) & 0x3ff];

						*vec_dst++ = f*g*(1.5f - 0.5f * g * g * f);
					}
				}
				outlet_bang(x->x_obj.ob_outlet);
				a = (t_garray *)pd_findbyclass(x->x_sym_dst, garray_class);
				garray_redraw(a);
			}
		}
	}
	else
	{
		post("tab_sin-ERROR: list need 3 float arguments:");
		post("  source_offset + destination_offset + number_of_samples_to_sqrt");
	}
}

static void tab_sin_free(t_tab_sin *x)
{
}

static void *tab_sin_new(t_symbol *s, int argc, t_atom *argv)
{
	t_tab_sin *x = (t_tab_sin *)pd_new(tab_sin_class);
	t_symbol	*src, *dst;

	if((argc >= 2) &&
		IS_A_SYMBOL(argv,0) &&
		IS_A_SYMBOL(argv,1))
	{
		src = (t_symbol *)atom_getsymbolarg(0, argc, argv);
		dst = (t_symbol *)atom_getsymbolarg(1, argc, argv);
	}
	else if((argc >= 1) &&
		IS_A_SYMBOL(argv,0))
	{
		src = (t_symbol *)atom_getsymbolarg(0, argc, argv);
		dst = src;
	}
	else
	{
		post("tab_sin-ERROR: need 2 symbols arguments:");
		post("  source_array_name + destination_array_name");
		return(0);
	}

	x->x_sym_scr1 = src;
	x->x_sym_dst = dst;
	outlet_new(&x->x_obj, &s_bang);
	return(x);
}

void tab_sin_setup(void)
{
	init_tab_sin();
	tab_sin_class = class_new(gensym("tab_sin"), (t_newmethod)tab_sin_new, (t_method)tab_sin_free,
					 sizeof(t_tab_sin), 0, A_GIMME, 0);
	class_addbang(tab_sin_class, (t_method)tab_sin_bang);
	class_addlist(tab_sin_class, (t_method)tab_sin_list);
	class_addmethod(tab_sin_class, (t_method)tab_sin_src, gensym("src"), A_DEFSYMBOL, 0);
	class_addmethod(tab_sin_class, (t_method)tab_sin_src, gensym("src1"), A_DEFSYMBOL, 0);
	class_addmethod(tab_sin_class, (t_method)tab_sin_dst, gensym("dst"), A_DEFSYMBOL, 0);
	class_sethelpsymbol(tab_sin_class, gensym("iemhelp2/tab_sin-help"));
}