Clementine-audio-player-Mac.../3rdparty/libprojectm/Renderer/BeatDetect.cpp

189 lines
4.2 KiB
C++

/**
* 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"
using namespace std;
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;
}