/*
 * framespect~ : Weighted alpha comparison, block-by-block.
 * Copyright (c) 2005, Dr Edward Kelly <morph_2016@yahoo.co.uk>
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without modification, 
 * are permitted provided that the following conditions are met:
 * 
 * Redistributions of source code must retain the above copyright notice, 
 * this list of conditions and the following disclaimer.
 * 
 * Redistributions in binary form must reproduce the above copyright notice, this list 
 * of conditions and the following disclaimer in the documentation and/or other materials 
 * 
 * provided with the distribution.
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY 
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT 
 * SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "m_pd.h"
#include <math.h>

static t_class *framespect_tilde_class;

typedef struct _framespect_tilde 
{
  t_object x_obj;
  t_float f;
  t_float f_max, f_win, f_accum;
  t_outlet *f_score;
} t_framespect_tilde;

t_int *framespect_tilde_perform(t_int *w)
{
  t_framespect_tilde *x = (t_framespect_tilde *)(w[1]);  
  t_sample       *real1 =           (t_sample *)(w[2]);
  t_sample       *imag1 =           (t_sample *)(w[3]);
  t_sample       *real2 =           (t_sample *)(w[4]);
  t_sample       *imag2 =           (t_sample *)(w[5]);
  int                 n =                  (int)(w[6]);
  float r1, i1, r2, i2;
  float vector1;
  float vector2;
  x->f_accum = 0;
  float block_accum = 0;
  x->f_max = 0;
  float score = 0;
  float avg = 0;
  int block = n;
  x->f_win = x->f_win > 0 ? x->f_win : 0.01;

  while (n--)
    {
      r1 = (*real1++);
      i1 = (*imag1++);
      r2 = (*real2++);
      i2 = (*imag2++);
      vector1 = sqrt((r1 * r1) + (i1 * i1));
      vector2 = sqrt((r2 * r2) + (i2 * i2));
      vector1 = vector1 > 0 ? vector1 : 0 - vector1;
      vector2 = vector2 > 0 ? vector2 : 0 - vector2;
      block_accum += vector2;
      x->f_max = vector2 > x->f_max ? vector2 : x->f_max;
      float diff = vector1 > vector2 ? vector1 - vector2 : vector2 - vector1;
      x->f_accum += (1/((diff/x->f_win)+1)) * vector2;
    }
  score = x->f_accum / x->f_max;
  block_accum /= x->f_max;
  avg = score / block_accum;
  outlet_float(x->f_score, avg);
  
  return(w+7);
}

void framespect_tilde_dsp(t_framespect_tilde *x, t_signal **sp)
{
  dsp_add(framespect_tilde_perform, 6, x,
	  sp[0]->s_vec, sp[1]->s_vec, sp[2]->s_vec, sp[3]->s_vec, sp[0]->s_n);
}

void *framespect_tilde_new(t_floatarg f)
{
  t_framespect_tilde *x = (t_framespect_tilde *)pd_new(framespect_tilde_class);

  x->f_win = f;

  inlet_new (&x->x_obj, &x->x_obj.ob_pd, &s_signal, &s_signal);
  inlet_new (&x->x_obj, &x->x_obj.ob_pd, &s_signal, &s_signal);
  inlet_new (&x->x_obj, &x->x_obj.ob_pd, &s_signal, &s_signal);
  floatinlet_new (&x->x_obj, &x->f_win);
  x->f_score = outlet_new(&x->x_obj, gensym("float"));

  return (void *)x;
}


void framespect_tilde_setup(void)
{
  framespect_tilde_class = class_new(gensym("framespect~"),
				     (t_newmethod)framespect_tilde_new,
				     0, sizeof(t_framespect_tilde),
				     CLASS_DEFAULT, A_DEFFLOAT, 0);

  post("|'''''''''''framespect~'''''''''''''|");
  post("|'''''weighted alpha comparison'''''|");
  post("|'''edward'''''''kelly'''''''2005'''|");

  class_addmethod(framespect_tilde_class, (t_method)framespect_tilde_dsp,
		  gensym("dsp"), 0);

  CLASS_MAINSIGNALIN(framespect_tilde_class, t_framespect_tilde, f);
}