/**
 * projectM -- Milkdrop-esque visualisation SDK
 * Copyright (C)2003-2004 projectM Team
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 * See 'LICENSE.txt' included within this release
 *
 */
/**
 * Takes sound data from wherever and returns beat detection values
 * Uses statistical Energy-Based methods. Very simple
 *
 * Some stuff was taken from Frederic Patin's beat-detection article,
 * you'll find it online
 */
#include <stdlib.h>
#include <stdio.h>

#include "wipemalloc.h"

#include "Common.hpp"
#include "PCM.hpp"
#include <cmath>
#include "BeatDetect.hpp"


BeatDetect::BeatDetect (PCM *pcm)
{
  int x,y;

  this->pcm=pcm;

  this->vol_instant = 0;
  this->vol_history = 0;

  for (y = 0;  y < 80;  y++)
  {
    this->vol_buffer[y] = 0;
  }

  this->beat_buffer_pos = 0;

  for (x = 0;  x < 32;  x++) {
    this->beat_instant[x] = 0;
    this->beat_history[x] = 0;
    this->beat_val[x] = 1.0;
    this->beat_att[x] = 1.0;
    this->beat_variance[x] = 0;
    for (y = 0;  y < 80;  y++) {
      this->beat_buffer[x][y] = 0;
    }
  }

  this->treb = 0;
  this->mid = 0;
  this->bass = 0;
  this->vol_old = 0;
  this->beat_sensitivity = 10.00;
  this->treb_att = 0;
  this->mid_att = 0;
  this->bass_att = 0;
  this->vol = 0;
}

BeatDetect::~BeatDetect()
{

}

void
BeatDetect::reset()
{
  this->treb      = 0;
  this->mid       = 0;
  this->bass      = 0;
  this->treb_att  = 0;
  this->mid_att   = 0;
  this->bass_att  = 0;
}

void
BeatDetect::detectFromSamples()
{
  vol_old = vol;
  bass = 0;  mid = 0;  treb = 0;

  getBeatVals (pcm->pcmdataL, pcm->pcmdataR);
}

void
BeatDetect::getBeatVals (float *vdataL, float *vdataR)
{
  int   linear=0;
  int   x,y;
  float temp2=0;

  vol_instant = 0;
  for (x = 0;  x < 16;  x++)
  {
    beat_instant[x]=0;
    for (y = linear * 2;  y < (linear + 8 + x) * 2;  y++)
    {
      beat_instant[x] += ((vdataL[y] * vdataL[y])
                       + (vdataR[y] * vdataR[y])) * (1.0 / (8 + x));
      //        printf ("beat_instant[%d]: %f %f %f\n", x, beat_instant[x], vdataL[y], vdataR[y]);
      vol_instant += ((vdataL[y] * vdataL[y])
                   + (vdataR[y] * vdataR[y])) * (1.0 / 512.0);

    }
    //printf ("1");
    linear = y / 2;
    beat_history[x] -= (beat_buffer[x][beat_buffer_pos])*.0125;
    beat_buffer[x][beat_buffer_pos] = beat_instant[x];
    beat_history[x] += (beat_instant[x]) * .0125;

    beat_val[x] = (beat_instant[x]) / (beat_history[x]);

    beat_att[x] += (beat_instant[x]) / (beat_history[x]);

    //printf ("2\n");

  }
  //printf ("b\n");
  vol_history -= (vol_buffer[beat_buffer_pos]) * .0125;
  vol_buffer[beat_buffer_pos] = vol_instant;
  vol_history += (vol_instant) * .0125;

  mid = 0;
  for (x = 1;  x < 10;  x++)
  {
    mid += (beat_instant[x]);
    temp2 += (beat_history[x]);
  }

  mid = mid / (1.5 * temp2);
  temp2 = 0;
  treb = 0;
  for (x = 10;  x < 16;  x++)
  {
    treb += (beat_instant[x]);
    temp2 += (beat_history[x]);
  }
  //printf ("c\n");
  treb = treb / (1.5 * temp2);
  //    *vol=vol_instant/(1.5*vol_history);
  vol = vol_instant / (1.5 * vol_history);

  bass = (beat_instant[0]) / (1.5 * beat_history[0]);


  if (projectM_isnan (treb)) {
    treb = 0.0;
  }
  if (projectM_isnan (mid)) {
    mid = 0.0;
  }
  if (projectM_isnan (bass)) {
    bass = 0.0;
  }
  treb_att = .6 * treb_att + .4 * treb;
  mid_att = .6 * mid_att + .4 * mid;
  bass_att = .6 * bass_att + .4 * bass;

  if (bass_att > 100) bass_att =100;
  if (bass > 100)     bass = 100;
  if (mid_att > 100)  mid_att = 100;
  if (mid > 100)      mid = 100;
  if (treb_att > 100) treb_att = 100;
  if (treb > 100)     treb = 100;
  if (vol > 100)      vol = 100;

  // *vol=(beat_instant[3])/(beat_history[3]);
  beat_buffer_pos++;
  if (beat_buffer_pos > 79) beat_buffer_pos = 0;
}