#include "MSPd.h"

static t_class *arrayfilt_class;

#define OBJECT_NAME "arrayfilt~"
typedef struct _arrayfilt
{
  t_object x_obj;
  t_float x_f;
  t_word *a_samples;
  int a_frames;
  t_symbol *arrayname;
} t_arrayfilt;

static void *arrayfilt_new(t_symbol *msg, int argc, t_atom *argv);
static void arrayfilt_dsp(t_arrayfilt *x, t_signal **sp);
static void arrayfilt_setarray(t_arrayfilt *x);

void arrayfilt_tilde_setup(void) {
  arrayfilt_class = class_new(gensym("arrayfilt~"), (t_newmethod)arrayfilt_new,
                              0, sizeof(t_arrayfilt),0,A_GIMME,0);
  CLASS_MAINSIGNALIN(arrayfilt_class, t_arrayfilt, x_f);
  class_addmethod(arrayfilt_class, (t_method)arrayfilt_dsp, gensym("dsp"), A_CANT, 0);
  potpourri_announce(OBJECT_NAME);
}

void *arrayfilt_new(t_symbol *msg, int argc, t_atom *argv)
{
  t_arrayfilt *x = (t_arrayfilt *)pd_new(arrayfilt_class);
  // t_symbol *arrayname;
  inlet_new(&x->x_obj, &x->x_obj.ob_pd,gensym("signal"), gensym("signal"));
  outlet_new(&x->x_obj, gensym("signal"));
  outlet_new(&x->x_obj, gensym("signal"));
  x->arrayname = atom_getsymbolarg(0, argc, argv);
  arrayfilt_setarray(x);
  return x;
}

void arrayfilt_setarray(t_arrayfilt *x)
{
  t_garray *a;
  t_symbol *arrayname = x->arrayname;
  if (!(a = (t_garray *)pd_findbyclass(arrayname, garray_class))) {
    if (*arrayname->s_name) pd_error(x, "arrayflt~: %s: no such array", arrayname->s_name);
  }
  else  {
    garray_usedindsp(a);
    if (!garray_getfloatwords(a, &x->a_frames, &x->a_samples))
    {
      pd_error(x, "%s: bad template for player~", arrayname->s_name);
    }
  }
}

t_int *arrayfilt_perform(t_int *w)
{
  int i;
  t_arrayfilt *x = (t_arrayfilt *) w[1];
  t_float *mag_in = (t_float *) w[2];
  t_float *phase_in = (t_float *) w[3];
  t_float *mag_out = (t_float *) w[4];
  t_float *phase_out = (t_float *) w[5];
  t_float mag, phase;
  t_word *a_samples = x->a_samples;
  int a_frames = x->a_frames;
  int n = (int) w[6];
  int N2 = n / 2;

  arrayfilt_setarray(x);

  if(a_frames < N2+1) {
    goto exit;
  }
  for(i = 0; i < N2 + 1; i++) {
    mag = mag_in[i];
    phase = phase_in[i];
    mag_out[i] = mag * a_samples[i].w_float;
    phase_out[i] = phase;
  }
exit:
  return (w + 7);
}

void arrayfilt_dsp(t_arrayfilt *x, t_signal **sp)
{
  dsp_add(arrayfilt_perform,6, x,
          sp[0]->s_vec, sp[1]->s_vec, sp[2]->s_vec, sp[3]->s_vec, (t_int)sp[0]->s_n);
}
