| /** |
| * 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; |
| |
| } |
| |
| |