adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: BTrack - A Real-Time Beat Tracker: src/OnsetDetectionFunction.h Source File adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23:
adamstark@23:
adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23:
adamstark@23:
BTrack - A Real-Time Beat Tracker adamstark@23:
adamstark@23:
adamstark@23:
adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23: adamstark@23:
adamstark@24:  All Classes Files Functions Enumerations Pages
adamstark@23: adamstark@23: adamstark@23:
adamstark@23: adamstark@23:
adamstark@23: adamstark@23: adamstark@23:
adamstark@23:
adamstark@23:
adamstark@23:
OnsetDetectionFunction.h
adamstark@23:
adamstark@23:
adamstark@23: Go to the documentation of this file.
1 //=======================================================================
adamstark@23:
20 //=======================================================================
adamstark@23:
21 
adamstark@23:
22 #ifndef __ONSETDETECTIONFUNCTION_H
adamstark@23:
23 #define __ONSETDETECTIONFUNCTION_H
adamstark@23:
24 
adamstark@23:
25 #include "fftw3.h"
adamstark@28:
26 #include <vector>
adamstark@28:
27 
adamstark@28:
28 //=======================================================================
adamstark@28:
30 enum OnsetDetectionFunctionType
adamstark@28:
31 {
adamstark@28:
32  EnergyEnvelope,
adamstark@28:
33  EnergyDifference,
adamstark@28:
34  SpectralDifference,
adamstark@28:
35  SpectralDifferenceHWR,
adamstark@28:
36  PhaseDeviation,
adamstark@28:
37  ComplexSpectralDifference,
adamstark@28:
38  ComplexSpectralDifferenceHWR,
adamstark@28:
39  HighFrequencyContent,
adamstark@28:
40  HighFrequencySpectralDifference,
adamstark@28:
41  HighFrequencySpectralDifferenceHWR
adamstark@28:
42 };
adamstark@28:
43 
adamstark@28:
44 //=======================================================================
adamstark@28:
46 enum WindowType
adamstark@28:
47 {
adamstark@28:
48  RectangularWindow,
adamstark@28:
49  HanningWindow,
adamstark@28:
50  HammingWindow,
adamstark@28:
51  BlackmanWindow,
adamstark@28:
52  TukeyWindow
adamstark@28:
53 };
adamstark@28:
54 
adamstark@28:
55 //=======================================================================
adamstark@28:
57 class OnsetDetectionFunction
adamstark@28:
58 {
adamstark@28:
59 public:
adamstark@28:
60 
adamstark@30:
66  OnsetDetectionFunction(int hopSize_,int frameSize_);
adamstark@30:
67 
adamstark@28:
68 
adamstark@30:
75  OnsetDetectionFunction(int hopSize_,int frameSize_,int onsetDetectionFunctionType_,int windowType_);
adamstark@30:
76 
adamstark@30:
78  ~OnsetDetectionFunction();
adamstark@28:
79 
adamstark@30:
85  void initialise(int hopSize_,int frameSize_);
adamstark@30:
86 
adamstark@30:
93  void initialise(int hopSize_,int frameSize_,int onsetDetectionFunctionType_,int windowType_);
adamstark@30:
94 
adamstark@30:
99  double calculateOnsetDetectionFunctionSample(double *buffer);
adamstark@30:
100 
adamstark@30:
104  void setOnsetDetectionFunctionType(int onsetDetectionFunctionType_);
adamstark@28:
105 
adamstark@30:
106 private:
adamstark@30:
107 
adamstark@30:
109  void performFFT();
adamstark@30:
110 
adamstark@30:
111  //=======================================================================
adamstark@30:
113  double energyEnvelope();
adamstark@28:
114 
adamstark@30:
116  double energyDifference();
adamstark@28:
117 
adamstark@30:
119  double spectralDifference();
adamstark@28:
120 
adamstark@30:
122  double spectralDifferenceHWR();
adamstark@28:
123 
adamstark@30:
125  double phaseDeviation();
adamstark@30:
126 
adamstark@30:
128  double complexSpectralDifference();
adamstark@30:
129 
adamstark@30:
131  double complexSpectralDifferenceHWR();
adamstark@30:
132 
adamstark@30:
134  double highFrequencyContent();
adamstark@30:
135 
adamstark@30:
137  double highFrequencySpectralDifference();
adamstark@30:
138 
adamstark@30:
140  double highFrequencySpectralDifferenceHWR();
adamstark@30:
141 
adamstark@30:
142  //=======================================================================
adamstark@30:
144  void calculateRectangularWindow();
adamstark@30:
145 
adamstark@30:
147  void calculateHanningWindow();
adamstark@30:
148 
adamstark@30:
150  void calclulateHammingWindow();
adamstark@30:
151 
adamstark@30:
153  void calculateBlackmanWindow();
adamstark@30:
154 
adamstark@30:
156  void calculateTukeyWindow();
adamstark@30:
157 
adamstark@30:
158  //=======================================================================
adamstark@30:
163  double princarg(double phaseVal);
adamstark@30:
164 
adamstark@30:
165 
adamstark@30:
166  double pi;
adamstark@30:
168  int frameSize;
adamstark@30:
169  int hopSize;
adamstark@30:
170  int onsetDetectionFunctionType;
adamstark@30:
171  int windowType;
adamstark@30:
173  fftw_plan p;
adamstark@30:
174  fftw_complex *complexIn;
adamstark@30:
175  fftw_complex *complexOut;
adamstark@30:
177  bool initialised;
adamstark@30:
179  std::vector<double> frame;
adamstark@30:
180  std::vector<double> window;
adamstark@30:
182  double prevEnergySum;
adamstark@30:
184  std::vector<double> magSpec;
adamstark@30:
185  std::vector<double> prevMagSpec;
adamstark@30:
187  std::vector<double> phase;
adamstark@30:
188  std::vector<double> prevPhase;
adamstark@30:
189  std::vector<double> prevPhase2;
adamstark@30:
191 };
adamstark@30:
192 
adamstark@30:
193 
adamstark@30:
194 #endif
adamstark@23:
adamstark@23: adamstark@23:
adamstark@30: Generated on Tue Jan 28 2014 01:06:53 for BTrack - A Real-Time Beat Tracker by   adamstark@23: doxygen adamstark@23: 1.8.2 adamstark@23:
adamstark@23: adamstark@23: