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

#include "m_pd.h"
#include "iemlib.h"
#include "iem_tab.h"

/* -------------------------- tab_eq_scalar ------------------------------ */
/*  if(x_beg_mem_src1[i] == compare)   */
/*    x_beg_mem_dst[i] = 1.0f;      */
/*  else                            */
/*    x_beg_mem_dst[i] += 0.0f;     */

typedef struct _tab_eq_scalar
{
  t_object  x_obj;
  int       x_size_src1;
  int       x_size_dst;
  int       x_offset_src1;
  int       x_offset_dst;
  iemarray_t   *x_beg_mem_src1;
  iemarray_t   *x_beg_mem_dst;
  t_symbol  *x_sym_scr1;
  t_symbol  *x_sym_dst;
} t_tab_eq_scalar;

static t_class *tab_eq_scalar_class;

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

static void tab_eq_scalar_float(t_tab_eq_scalar *x, t_floatarg compare)
{
  int i, n;
  int ok_src1, ok_dst;
  iemarray_t *vec_src1, *vec_dst;
  
  ok_src1 = iem_tab_check_arrays(gensym("tab_eq_scalar"), x->x_sym_scr1, &x->x_beg_mem_src1, &x->x_size_src1, 0);
  ok_dst = iem_tab_check_arrays(gensym("tab_eq_scalar"), x->x_sym_dst, &x->x_beg_mem_dst, &x->x_size_dst, 0);
  
  if(ok_src1 && ok_dst)
  {
    if(x->x_size_src1 < x->x_size_dst)
      n = x->x_size_src1;
    else
      n = x->x_size_dst;
    
    vec_src1 = x->x_beg_mem_src1;
    vec_dst = x->x_beg_mem_dst;
    if(n)
    {
      t_garray *a;
      
      for(i=0; i<n; i++)
      {
        if(iemarray_getfloat(vec_src1, i) == compare)
          iemarray_setfloat(vec_dst, i, 1.0f);
        else
          iemarray_setfloat(vec_dst, i, 0.0f);
      }
      outlet_bang(x->x_obj.ob_outlet);
      a = (t_garray *)pd_findbyclass(x->x_sym_dst, garray_class);
      garray_redraw(a);
    }
  }
}

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

static void tab_eq_scalar_list(t_tab_eq_scalar *x, t_symbol *s, int argc, t_atom *argv)
{
  int beg_src1, beg_dst;
  int i, n;
  int ok_src1, ok_dst;
  iemarray_t *vec_src1, *vec_dst;
  t_float compare;
  
  if((argc >= 4) &&
    IS_A_FLOAT(argv,0) &&
    IS_A_FLOAT(argv,1) &&
    IS_A_FLOAT(argv,2) &&
    IS_A_FLOAT(argv,3))
  {
    beg_src1 = (int)atom_getintarg(0, argc, argv);
    beg_dst = (int)atom_getintarg(1, argc, argv);
    n = (int)atom_getintarg(2, argc, argv);
    compare = (t_float)atom_getfloatarg(3, argc, argv);
    if(beg_src1 < 0)
      beg_src1 = 0;
    if(beg_dst < 0)
      beg_dst = 0;
    if(n < 0)
      n = 0;
    
    ok_src1 = iem_tab_check_arrays(gensym("tab_eq_scalar"), x->x_sym_scr1, &x->x_beg_mem_src1, &x->x_size_src1, beg_src1+n);
    ok_dst = iem_tab_check_arrays(gensym("tab_eq_scalar"), x->x_sym_dst, &x->x_beg_mem_dst, &x->x_size_dst, beg_dst+n);
    
    if(ok_src1 && ok_dst)
    {
      vec_src1 = x->x_beg_mem_src1 + beg_src1;
      vec_dst = x->x_beg_mem_dst + beg_dst;
      if(n)
      {
        t_garray *a;
        
        for(i=0; i<n; i++)
        {
          if(iemarray_getfloat(vec_src1, i) == compare)
            iemarray_setfloat(vec_dst, i, 1.0f);
          else
            iemarray_setfloat(vec_dst, i, 0.0f);
        }
        outlet_bang(x->x_obj.ob_outlet);
        a = (t_garray *)pd_findbyclass(x->x_sym_dst, garray_class);
        garray_redraw(a);
      }
    }
  }
  else
  {
    post("tab_eq_scalar-ERROR: list need 4 float arguments:");
    post("  source1_offset + destination_offset + number_of_samples_to_compare + compare_scalar");
  }
}

static void tab_eq_scalar_free(t_tab_eq_scalar *x)
{
}

static void *tab_eq_scalar_new(t_symbol *s, int argc, t_atom *argv)
{
  t_tab_eq_scalar *x = (t_tab_eq_scalar *)pd_new(tab_eq_scalar_class);
  t_symbol  *src1, *dst;
  
  if((argc >= 2) &&
    IS_A_SYMBOL(argv,0) &&
    IS_A_SYMBOL(argv,1))
  {
    src1 = (t_symbol *)atom_getsymbolarg(0, argc, argv);
    dst = (t_symbol *)atom_getsymbolarg(1, argc, argv);
  }
  else if((argc >= 1) &&
    IS_A_SYMBOL(argv,0))
  {
    src1 = (t_symbol *)atom_getsymbolarg(0, argc, argv);
    dst = src1;
  }
  else
  {
    post("tab_eq_scalar-ERROR: need 2 symbol arguments:");
    post("  source_array_name + destination_array_name");
    return(0);
  }
  
  x->x_sym_scr1 = src1;
  x->x_sym_dst = dst;
  outlet_new(&x->x_obj, &s_bang);
  return(x);
}

void tab_eq_scalar_setup(void)
{
  tab_eq_scalar_class = class_new(gensym("tab_eq_scalar"), (t_newmethod)tab_eq_scalar_new, (t_method)tab_eq_scalar_free,
    sizeof(t_tab_eq_scalar), 0, A_GIMME, 0);
  class_addfloat(tab_eq_scalar_class, (t_method)tab_eq_scalar_float);
  class_addlist(tab_eq_scalar_class, (t_method)tab_eq_scalar_list);
  class_addmethod(tab_eq_scalar_class, (t_method)tab_eq_scalar_src, gensym("src1"), A_DEFSYMBOL, 0);
  class_addmethod(tab_eq_scalar_class, (t_method)tab_eq_scalar_src, gensym("src"), A_DEFSYMBOL, 0);
  class_addmethod(tab_eq_scalar_class, (t_method)tab_eq_scalar_dst, gensym("dst"), A_DEFSYMBOL, 0);
//  class_sethelpsymbol(tab_eq_scalar_class, gensym("iemhelp2/tab_eq_scalar-help"));
}
