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