git.sesse.net Git - vlc/blob - 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 <stdio.h>
  34 #include "engine_vars.h"
  35
  36 double beat_buffer[32][80],beat_instant[32],beat_history[32];
  37 double *beat_val,*beat_att,*beat_variance;
  38 int beat_buffer_pos;
  39
  40 double vol_buffer[80],vol_instant,vol_history;
  41
  42 void initBeatDetect()
  43 {
  44
  45   int x,y;
  46
  47   vol_instant=0;
  48   vol_history=0;
  49
  50   for (y=0;y<80;y++)
  51     {
  52       vol_buffer[y]=0;
  53     }
  54
  55   beat_buffer_pos=0;
  56
  57   beat_val=(double *)malloc(32*sizeof(double));
  58   beat_att=(double *)malloc(32*sizeof(double));
  59   beat_variance=(double *)malloc(32*sizeof(double));
  60
  61   for (x=0;x<32;x++)
  62     {
  63       beat_instant[x]=0;
  64       beat_history[x]=0;
  65       beat_val[x]=1.0;
  66       beat_att[x]=1.0;
  67       beat_variance[x]=0;
  68       for (y=0;y<80;y++)
  69         {
  70           beat_buffer[x][y]=0;
  71         }
  72     }
  73
  74 }
  75
  76 void getBeatVals(double *vdataL,double *vdataR, double *vol)
  77 {
  78   int linear=0;
  79   int x,y;
  80
  81   vol_instant=0;
  82
  83       for ( x=0;x<16;x++)
  84         {
  85
  86           beat_instant[x]=0;
  87           for ( y=linear*2;y<(linear+8+x)*2;y++)
  88             {
  89               beat_instant[x]+=((vdataL[y]*vdataL[y])+(vdataR[y]*vdataR[y]))*(1.0/(8+x));
  90               vol_instant+=((vdataL[y]*vdataL[y])+(vdataR[y]*vdataR[y]))*(1.0/512.0);
  91
  92             }
  93
  94           linear=y/2;
  95           beat_history[x]-=(beat_buffer[x][beat_buffer_pos])*.0125;
  96           beat_buffer[x][beat_buffer_pos]=beat_instant[x];
  97           beat_history[x]+=(beat_instant[x])*.0125;
  98
  99           beat_val[x]=(beat_instant[x])/(beat_history[x]);
 100
 101           beat_att[x]+=(beat_instant[x])/(beat_history[x]);
 102
 103
 104
 105         }
 106
 107       vol_history-=(vol_buffer[beat_buffer_pos])*.0125;
 108       vol_buffer[beat_buffer_pos]=vol_instant;
 109       vol_history+=(vol_instant)*.0125;
 110
 111       double temp2=0;
 112       mid=0;
 113       for(x=1;x<10;x++)
 114         {
 115          mid+=(beat_instant[x]);
 116           temp2+=(beat_history[x]);
 117
 118         }
 119
 120          mid=mid/(1.5*temp2);
 121          temp2=0;
 122          treb=0;
 123           for(x=10;x<16;x++)
 124             {
 125               treb+=(beat_instant[x]);
 126               temp2+=(beat_history[x]);
 127             }
 128           treb=treb/(1.5*temp2);
 129           *vol=vol_instant/(1.5*vol_history);
 130
 131           bass=(beat_instant[0])/(1.5*beat_history[0]);
 132
 133           treb_att=.6 * treb_att + .4 * treb;
 134           mid_att=.6 * mid_att + .4 * mid;
 135           bass_att=.6 * bass_att + .4 * bass;
 136           //printf("%f %f %f %f\n",bass,mid,treb,*vol);
 137            // *vol=(beat_instant[3])/(beat_history[3]);
 138           beat_buffer_pos++;
 139           if( beat_buffer_pos>79)beat_buffer_pos=0;
 140
 141 }
 142 void freeBeatDetect()
 143 {
 144   free(beat_att);
 145   free(beat_val);
 146   free(beat_variance);
 147 }