]> git.sesse.net Git - vlc/blob - modules/visualization/galaktos/beat_detect.c
Removes trailing spaces. Removes tabs.
[vlc] / modules / visualization / galaktos / beat_detect.c
1 /*****************************************************************************
2  * beat_detect.c: basic beat detection algorithm
3  *****************************************************************************
4  * Copyright (C) 2004 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Cyril Deguet <asmax@videolan.org>
8  *          code from projectM http://xmms-projectm.sourceforge.net
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23  *****************************************************************************/
24
25 //
26 //by Peter Sperl
27 //
28 //Takes sound data from wherever and returns beat detection values
29 //Uses statistical Energy-Based methods. Very simple
30 //
31 //Some stuff was taken from Frederic Patin's beat-detection article, you'll find it online
32
33 #include "engine_vars.h"
34
35 double beat_buffer[32][80],beat_instant[32],beat_history[32];
36 double *beat_val,*beat_att,*beat_variance;
37 int beat_buffer_pos;
38
39 double vol_buffer[80],vol_instant,vol_history;
40
41 void initBeatDetect()
42 {
43
44   int x,y;
45
46   vol_instant=0;
47   vol_history=0;
48
49   for (y=0;y<80;y++)
50     {
51       vol_buffer[y]=0;
52     }
53
54   beat_buffer_pos=0;
55
56   beat_val=(double *)malloc(32*sizeof(double));
57   beat_att=(double *)malloc(32*sizeof(double));
58   beat_variance=(double *)malloc(32*sizeof(double));
59
60   for (x=0;x<32;x++)
61     {
62       beat_instant[x]=0;
63       beat_history[x]=0;
64       beat_val[x]=1.0;
65       beat_att[x]=1.0;
66       beat_variance[x]=0;
67       for (y=0;y<80;y++)
68     {
69       beat_buffer[x][y]=0;
70     }
71     }
72
73 }
74
75 void getBeatVals(double *vdataL,double *vdataR, double *vol)
76 {
77   int linear=0;
78   int x,y;
79
80   vol_instant=0;
81
82       for ( x=0;x<16;x++)
83     {
84     
85       beat_instant[x]=0;
86       for ( y=linear*2;y<(linear+8+x)*2;y++)
87         {
88           beat_instant[x]+=((vdataL[y]*vdataL[y])+(vdataR[y]*vdataR[y]))*(1.0/(8+x));
89           vol_instant+=((vdataL[y]*vdataL[y])+(vdataR[y]*vdataR[y]))*(1.0/512.0);
90
91         }
92     
93       linear=y/2;
94       beat_history[x]-=(beat_buffer[x][beat_buffer_pos])*.0125;
95       beat_buffer[x][beat_buffer_pos]=beat_instant[x];
96       beat_history[x]+=(beat_instant[x])*.0125;
97     
98       beat_val[x]=(beat_instant[x])/(beat_history[x]);
99     
100       beat_att[x]+=(beat_instant[x])/(beat_history[x]);
101
102
103      
104     }
105  
106       vol_history-=(vol_buffer[beat_buffer_pos])*.0125;
107       vol_buffer[beat_buffer_pos]=vol_instant;
108       vol_history+=(vol_instant)*.0125;
109
110       double temp2=0;
111       mid=0;
112       for(x=1;x<10;x++)
113     {
114      mid+=(beat_instant[x]);
115       temp2+=(beat_history[x]);
116     
117     }
118     
119      mid=mid/(1.5*temp2);
120      temp2=0;
121      treb=0;
122        for(x=10;x<16;x++)
123         {
124           treb+=(beat_instant[x]);
125           temp2+=(beat_history[x]);
126         }
127       treb=treb/(1.5*temp2);
128       *vol=vol_instant/(1.5*vol_history);
129  
130       bass=(beat_instant[0])/(1.5*beat_history[0]);
131
132       treb_att=.6 * treb_att + .4 * treb;
133       mid_att=.6 * mid_att + .4 * mid;
134       bass_att=.6 * bass_att + .4 * bass;
135       //printf("%f %f %f %f\n",bass,mid,treb,*vol);
136        // *vol=(beat_instant[3])/(beat_history[3]);
137       beat_buffer_pos++;
138       if( beat_buffer_pos>79)beat_buffer_pos=0;
139     
140 }
141 void freeBeatDetect()
142 {
143   free(beat_att);
144   free(beat_val);
145   free(beat_variance);
146 }