adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: BTrack - A Real-Time Beat Tracker: src/OnsetDetectionFunction.h Source File adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60:
adamstark@60:
adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60:
adamstark@60:
BTrack - A Real-Time Beat Tracker adamstark@60:
adamstark@60:
adamstark@60:
adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60: adamstark@60:
adamstark@61:  All Classes Files Functions Enumerations Pages
adamstark@60: adamstark@60: adamstark@60:
adamstark@60: adamstark@60:
adamstark@60: adamstark@60: adamstark@60:
adamstark@60:
adamstark@60:
adamstark@60:
OnsetDetectionFunction.h
adamstark@60:
adamstark@60:
adamstark@60: Go to the documentation of this file.
1 //=======================================================================
adamstark@60:
20 //=======================================================================
adamstark@60:
21 
adamstark@60:
22 #ifndef __ONSETDETECTIONFUNCTION_H
adamstark@60:
23 #define __ONSETDETECTIONFUNCTION_H
adamstark@60:
24 
adamstark@60:
25 #include "fftw3.h"
adamstark@65:
26 #include <vector>
adamstark@65:
27 
adamstark@65:
28 //=======================================================================
adamstark@65:
30 enum OnsetDetectionFunctionType
adamstark@65:
31 {
adamstark@65:
32  EnergyEnvelope,
adamstark@65:
33  EnergyDifference,
adamstark@65:
34  SpectralDifference,
adamstark@65:
35  SpectralDifferenceHWR,
adamstark@65:
36  PhaseDeviation,
adamstark@65:
37  ComplexSpectralDifference,
adamstark@65:
38  ComplexSpectralDifferenceHWR,
adamstark@65:
39  HighFrequencyContent,
adamstark@65:
40  HighFrequencySpectralDifference,
adamstark@65:
41  HighFrequencySpectralDifferenceHWR
adamstark@65:
42 };
adamstark@65:
43 
adamstark@65:
44 //=======================================================================
adamstark@65:
46 enum WindowType
adamstark@65:
47 {
adamstark@65:
48  RectangularWindow,
adamstark@65:
49  HanningWindow,
adamstark@65:
50  HammingWindow,
adamstark@65:
51  BlackmanWindow,
adamstark@65:
52  TukeyWindow
adamstark@65:
53 };
adamstark@65:
54 
adamstark@65:
55 //=======================================================================
adamstark@65:
57 class OnsetDetectionFunction
adamstark@65:
58 {
adamstark@65:
59 public:
adamstark@65:
60 
adamstark@65:
67  OnsetDetectionFunction(int hopSize_,int frameSize_,int onsetDetectionFunctionType_,int windowType);
adamstark@65:
68 
adamstark@65:
70  ~OnsetDetectionFunction();
adamstark@65:
71 
adamstark@65:
78  void initialise(int hopSize_,int frameSize_,int onsetDetectionFunctionType_,int windowType);
adamstark@65:
79 
adamstark@65:
84  double calculateOnsetDetectionFunctionSample(double *buffer);
adamstark@65:
85 
adamstark@65:
89  void setOnsetDetectionFunctionType(int onsetDetectionFunctionType_);
adamstark@65:
90 
adamstark@65:
91 private:
adamstark@65:
92 
adamstark@65:
94  void performFFT();
adamstark@65:
95 
adamstark@65:
96  //=======================================================================
adamstark@65:
98  double energyEnvelope();
adamstark@65:
99 
adamstark@65:
101  double energyDifference();
adamstark@65:
102 
adamstark@65:
104  double spectralDifference();
adamstark@65:
105 
adamstark@65:
107  double spectralDifferenceHWR();
adamstark@65:
108 
adamstark@65:
110  double phaseDeviation();
adamstark@65:
111 
adamstark@65:
113  double complexSpectralDifference();
adamstark@65:
114 
adamstark@65:
116  double complexSpectralDifferenceHWR();
adamstark@65:
117 
adamstark@65:
119  double highFrequencyContent();
adamstark@65:
120 
adamstark@65:
122  double highFrequencySpectralDifference();
adamstark@65:
123 
adamstark@65:
125  double highFrequencySpectralDifferenceHWR();
adamstark@65:
126 
adamstark@65:
127  //=======================================================================
adamstark@65:
129  void calculateRectangularWindow();
adamstark@65:
130 
adamstark@65:
132  void calculateHanningWindow();
adamstark@65:
133 
adamstark@65:
135  void calclulateHammingWindow();
adamstark@65:
136 
adamstark@65:
138  void calculateBlackmanWindow();
adamstark@65:
139 
adamstark@65:
141  void calculateTukeyWindow();
adamstark@65:
142 
adamstark@65:
143  //=======================================================================
adamstark@65:
148  double princarg(double phaseVal);
adamstark@61:
149 
adamstark@65:
150 
adamstark@65:
151  double pi;
adamstark@65:
153  int frameSize;
adamstark@65:
154  int hopSize;
adamstark@65:
155  int onsetDetectionFunctionType;
adamstark@65:
157  fftw_plan p;
adamstark@65:
158  fftw_complex *complexIn;
adamstark@65:
159  fftw_complex *complexOut;
adamstark@65:
161  bool initialised;
adamstark@65:
163  std::vector<double> frame;
adamstark@65:
164  std::vector<double> window;
adamstark@65:
166  double prevEnergySum;
adamstark@65:
168  std::vector<double> magSpec;
adamstark@65:
169  std::vector<double> prevMagSpec;
adamstark@65:
171  std::vector<double> phase;
adamstark@65:
172  std::vector<double> prevPhase;
adamstark@65:
173  std::vector<double> prevPhase2;
adamstark@65:
175 };
adamstark@61:
176 
adamstark@65:
177 
adamstark@65:
178 #endif
adamstark@60:
adamstark@60: adamstark@60:
adamstark@65: Generated on Tue Jan 28 2014 00:29:23 for BTrack - A Real-Time Beat Tracker by   adamstark@60: doxygen adamstark@60: 1.8.2 adamstark@60:
adamstark@60: adamstark@60: