added library for audio-processing
This commit is contained in:
1 parent
7975382705
commit
6eabb15f07
20 files changed
+4174
No files matched your search
@@ -0,0 +1,135 @@
|
||||
/*
|
||||
* Copyright (c) 2007 - 2008 by Damien Di Fede <ddf@compartmental.net>
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Library General Public License as published
|
||||
* by the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program 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 Library General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Library General Public
|
||||
* License along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||
*
|
||||
* This Code is taken from the source provided with a tutorial by
|
||||
* http://www.badlogicgames.com (as of 16.05.2013).
|
||||
* The original source can be found here:
|
||||
* http://code.google.com/p/audio-analysis/source/browse/trunk/src/com/badlogic/audio/analysis/DFT.java
|
||||
*/
|
||||
package org.wyrez.audio.analysis;
|
||||
|
||||
/**
|
||||
* DFT stands for Discrete Fourier Transform and is the most widely used Fourier
|
||||
* Transform. You will never want to use this class due to the fact that it is a
|
||||
* brute force implementation of the DFT and as such is quite slow. Use an FFT
|
||||
* instead. This exists primarily as a way to ensure that other implementations
|
||||
* of the DFT are working properly. This implementation expects an even
|
||||
* <code>timeSize</code> and will throw and IllegalArgumentException if this is
|
||||
* not the case.
|
||||
*
|
||||
* @author Damien Di Fede
|
||||
*
|
||||
* @see FourierTransform
|
||||
* @see FFT
|
||||
* @see <a href="http://www.dspguide.com/ch8.htm">The Discrete Fourier
|
||||
* Transform</a>
|
||||
*
|
||||
*/
|
||||
public class DFT extends FourierTransform {
|
||||
|
||||
/**
|
||||
* Constructs a DFT that expects audio buffers of length
|
||||
* <code>timeSize</code> that have been recorded with a sample rate of
|
||||
* <code>sampleRate</code>. Will throw an IllegalArgumentException if
|
||||
* <code>timeSize</code> is not even.
|
||||
*
|
||||
* @param timeSize the length of the audio buffers you plan to analyze
|
||||
* @param sampleRate the sample rate of the audio samples you plan to
|
||||
* analyze
|
||||
*/
|
||||
public DFT(int timeSize, float sampleRate) {
|
||||
super(timeSize, sampleRate);
|
||||
if (timeSize % 2 != 0) {
|
||||
throw new IllegalArgumentException("DFT: timeSize must be even.");
|
||||
}
|
||||
buildTrigTables();
|
||||
}
|
||||
|
||||
protected void allocateArrays() {
|
||||
spectrum = new float[timeSize / 2 + 1];
|
||||
real = new float[timeSize / 2 + 1];
|
||||
imag = new float[timeSize / 2 + 1];
|
||||
}
|
||||
|
||||
/**
|
||||
* Not currently implemented.
|
||||
*/
|
||||
public void scaleBand(int i, float s) {
|
||||
}
|
||||
|
||||
/**
|
||||
* Not currently implemented.
|
||||
*/
|
||||
public void setBand(int i, float a) {
|
||||
}
|
||||
|
||||
public void forward(float[] samples) {
|
||||
if (samples.length != timeSize) {
|
||||
throw new IllegalArgumentException("DFT.forward: The length of the passed sample buffer must be equal to DFT.timeSize().");
|
||||
}
|
||||
doWindow(samples);
|
||||
int N = samples.length;
|
||||
for (int f = 0; f <= N / 2; f++) {
|
||||
real[f] = 0.0f;
|
||||
imag[f] = 0.0f;
|
||||
for (int t = 0; t < N; t++) {
|
||||
real[f] += samples[t] * cos(t * f);
|
||||
imag[f] += samples[t] * -sin(t * f);
|
||||
}
|
||||
}
|
||||
fillSpectrum();
|
||||
}
|
||||
|
||||
public void inverse(float[] buffer) {
|
||||
int N = buffer.length;
|
||||
real[0] /= N;
|
||||
imag[0] = -imag[0] / (N / 2);
|
||||
real[N / 2] /= N;
|
||||
imag[N / 2] = -imag[0] / (N / 2);
|
||||
for (int i = 0; i < N / 2; i++) {
|
||||
real[i] /= (N / 2);
|
||||
imag[i] = -imag[i] / (N / 2);
|
||||
}
|
||||
for (int t = 0; t < N; t++) {
|
||||
buffer[t] = 0.0f;
|
||||
for (int f = 0; f < N / 2; f++) {
|
||||
buffer[t] += real[f] * cos(t * f) + imag[f] * sin(t * f);
|
||||
}
|
||||
}
|
||||
}
|
||||
// lookup table data and functions
|
||||
private float[] sinlookup;
|
||||
private float[] coslookup;
|
||||
|
||||
private void buildTrigTables() {
|
||||
int N = spectrum.length * timeSize;
|
||||
sinlookup = new float[N];
|
||||
coslookup = new float[N];
|
||||
for (int i = 0; i < N; i++) {
|
||||
sinlookup[i] = (float) Math.sin(i * TWO_PI / timeSize);
|
||||
coslookup[i] = (float) Math.cos(i * TWO_PI / timeSize);
|
||||
}
|
||||
}
|
||||
|
||||
private float sin(int i) {
|
||||
return sinlookup[i];
|
||||
}
|
||||
|
||||
private float cos(int i) {
|
||||
return coslookup[i];
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,238 @@
|
||||
/*
|
||||
* Copyright (c) 2007 - 2008 by Damien Di Fede <ddf@compartmental.net>
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Library General Public License as published
|
||||
* by the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program 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 Library General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Library General Public
|
||||
* License along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||
*
|
||||
* This Code is taken from the source provided with a tutorial by
|
||||
* http://www.badlogicgames.com (as of 16.05.2013).
|
||||
* The original source can be found here:
|
||||
* http://code.google.com/p/audio-analysis/source/browse/trunk/src/com/badlogic/audio/analysis/FFT.java
|
||||
*/
|
||||
package org.wyrez.audio.analysis;
|
||||
|
||||
/**
|
||||
* FFT stands for Fast Fourier Transform. It is an efficient way to calculate
|
||||
* the Complex Discrete Fourier Transform. There is not much to say about this
|
||||
* class other than the fact that when you want to analyze the spectrum of an
|
||||
* audio buffer you will almost always use this class. One restriction of this
|
||||
* class is that the audio buffers you want to analyze must have a length that
|
||||
* is a power of two. If you try to construct an FFT with a
|
||||
* <code>timeSize</code> that is not a power of two, an IllegalArgumentException
|
||||
* will be thrown.
|
||||
*
|
||||
* @see FourierTransform
|
||||
* @see <a href="http://www.dspguide.com/ch12.htm">The Fast Fourier
|
||||
* Transform</a>
|
||||
*
|
||||
* @author Damien Di Fede
|
||||
*
|
||||
*/
|
||||
public class FFT extends FourierTransform {
|
||||
|
||||
/**
|
||||
* Constructs an FFT that will accept sample buffers that are
|
||||
* <code>timeSize</code> long and have been recorded with a sample rate of
|
||||
* <code>sampleRate</code>.
|
||||
* <code>timeSize</code> <em>must</em> be a power of two. This will throw an
|
||||
* exception if it is not.
|
||||
*
|
||||
* @param timeSize the length of the sample buffers you will be analyzing
|
||||
* @param sampleRate the sample rate of the audio you will be analyzing
|
||||
*/
|
||||
public FFT(int timeSize, float sampleRate) {
|
||||
super(timeSize, sampleRate);
|
||||
if ((timeSize & (timeSize - 1)) != 0) {
|
||||
throw new IllegalArgumentException(
|
||||
"FFT: timeSize must be a power of two.");
|
||||
}
|
||||
buildReverseTable();
|
||||
buildTrigTables();
|
||||
}
|
||||
|
||||
protected void allocateArrays() {
|
||||
spectrum = new float[timeSize / 2 + 1];
|
||||
real = new float[timeSize];
|
||||
imag = new float[timeSize];
|
||||
}
|
||||
|
||||
public void scaleBand(int i, float s) {
|
||||
if (s < 0) {
|
||||
throw new IllegalArgumentException("Can't scale a frequency band by a negative value.");
|
||||
}
|
||||
if (spectrum[i] != 0) {
|
||||
real[i] /= spectrum[i];
|
||||
imag[i] /= spectrum[i];
|
||||
spectrum[i] *= s;
|
||||
real[i] *= spectrum[i];
|
||||
imag[i] *= spectrum[i];
|
||||
}
|
||||
if (i != 0 && i != timeSize / 2) {
|
||||
real[timeSize - i] = real[i];
|
||||
imag[timeSize - i] = -imag[i];
|
||||
}
|
||||
}
|
||||
|
||||
public void setBand(int i, float a) {
|
||||
if (a < 0) {
|
||||
throw new IllegalArgumentException("Can't set a frequency band to a negative value.");
|
||||
}
|
||||
if (real[i] == 0 && imag[i] == 0) {
|
||||
real[i] = a;
|
||||
spectrum[i] = a;
|
||||
} else {
|
||||
real[i] /= spectrum[i];
|
||||
imag[i] /= spectrum[i];
|
||||
spectrum[i] = a;
|
||||
real[i] *= spectrum[i];
|
||||
imag[i] *= spectrum[i];
|
||||
}
|
||||
if (i != 0 && i != timeSize / 2) {
|
||||
real[timeSize - i] = real[i];
|
||||
imag[timeSize - i] = -imag[i];
|
||||
}
|
||||
}
|
||||
|
||||
// performs an in-place fft on the data in the real and imag arrays
|
||||
// bit reversing is not necessary as the data will already be bit reversed
|
||||
private void fft() {
|
||||
for (int halfSize = 1; halfSize < real.length; halfSize *= 2) {
|
||||
// float k = -(float)Math.PI/halfSize;
|
||||
// phase shift step
|
||||
// float phaseShiftStepR = (float)Math.cos(k);
|
||||
// float phaseShiftStepI = (float)Math.sin(k);
|
||||
// using lookup table
|
||||
float phaseShiftStepR = cos(halfSize);
|
||||
float phaseShiftStepI = sin(halfSize);
|
||||
// current phase shift
|
||||
float currentPhaseShiftR = 1.0f;
|
||||
float currentPhaseShiftI = 0.0f;
|
||||
for (int fftStep = 0; fftStep < halfSize; fftStep++) {
|
||||
for (int i = fftStep; i < real.length; i += 2 * halfSize) {
|
||||
int off = i + halfSize;
|
||||
float tr = (currentPhaseShiftR * real[off]) - (currentPhaseShiftI * imag[off]);
|
||||
float ti = (currentPhaseShiftR * imag[off]) + (currentPhaseShiftI * real[off]);
|
||||
real[off] = real[i] - tr;
|
||||
imag[off] = imag[i] - ti;
|
||||
real[i] += tr;
|
||||
imag[i] += ti;
|
||||
}
|
||||
float tmpR = currentPhaseShiftR;
|
||||
currentPhaseShiftR = (tmpR * phaseShiftStepR) - (currentPhaseShiftI * phaseShiftStepI);
|
||||
currentPhaseShiftI = (tmpR * phaseShiftStepI) + (currentPhaseShiftI * phaseShiftStepR);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public void forward(float[] buffer) {
|
||||
if (buffer.length != timeSize) {
|
||||
throw new IllegalArgumentException("FFT.forward: The length of the passed sample buffer must be equal to timeSize().");
|
||||
}
|
||||
doWindow(buffer);
|
||||
// copy samples to real/imag in bit-reversed order
|
||||
bitReverseSamples(buffer);
|
||||
// perform the fft
|
||||
fft();
|
||||
// fill the spectrum buffer with amplitudes
|
||||
fillSpectrum();
|
||||
}
|
||||
|
||||
/**
|
||||
* Performs a forward transform on the passed buffers.
|
||||
*
|
||||
* @param buffReal the real part of the time domain signal to transform
|
||||
* @param buffImag the imaginary part of the time domain signal to transform
|
||||
*/
|
||||
public void forward(float[] buffReal, float[] buffImag) {
|
||||
if (buffReal.length != timeSize || buffImag.length != timeSize) {
|
||||
throw new IllegalArgumentException("FFT.forward: The length of the passed buffers must be equal to timeSize().");
|
||||
}
|
||||
setComplex(buffReal, buffImag);
|
||||
bitReverseComplex();
|
||||
fft();
|
||||
fillSpectrum();
|
||||
}
|
||||
|
||||
public void inverse(float[] buffer) {
|
||||
if (buffer.length > real.length) {
|
||||
throw new IllegalArgumentException("FFT.inverse: the passed array's length must equal FFT.timeSize().");
|
||||
}
|
||||
// conjugate
|
||||
for (int i = 0; i < timeSize; i++) {
|
||||
imag[i] *= -1;
|
||||
}
|
||||
bitReverseComplex();
|
||||
fft();
|
||||
// copy the result in real into buffer, scaling as we do
|
||||
for (int i = 0; i < buffer.length; i++) {
|
||||
buffer[i] = real[i] / real.length;
|
||||
}
|
||||
}
|
||||
private int[] reverse;
|
||||
|
||||
private void buildReverseTable() {
|
||||
int N = timeSize;
|
||||
reverse = new int[N];
|
||||
|
||||
// set up the bit reversing table
|
||||
reverse[0] = 0;
|
||||
for (int limit = 1, bit = N / 2; limit < N; limit <<= 1, bit >>= 1) {
|
||||
for (int i = 0; i < limit; i++) {
|
||||
reverse[i + limit] = reverse[i] + bit;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// copies the values in the samples array into the real array
|
||||
// in bit reversed order. the imag array is filled with zeros.
|
||||
private void bitReverseSamples(float[] samples) {
|
||||
for (int i = 0; i < samples.length; i++) {
|
||||
real[i] = samples[reverse[i]];
|
||||
imag[i] = 0.0f;
|
||||
}
|
||||
}
|
||||
|
||||
// bit reverse real[] and imag[]
|
||||
private void bitReverseComplex() {
|
||||
float[] revReal = new float[real.length];
|
||||
float[] revImag = new float[imag.length];
|
||||
for (int i = 0; i < real.length; i++) {
|
||||
revReal[i] = real[reverse[i]];
|
||||
revImag[i] = imag[reverse[i]];
|
||||
}
|
||||
real = revReal;
|
||||
imag = revImag;
|
||||
}
|
||||
// lookup tables
|
||||
private float[] sinlookup;
|
||||
private float[] coslookup;
|
||||
|
||||
private float sin(int i) {
|
||||
return sinlookup[i];
|
||||
}
|
||||
|
||||
private float cos(int i) {
|
||||
return coslookup[i];
|
||||
}
|
||||
|
||||
private void buildTrigTables() {
|
||||
int N = timeSize;
|
||||
sinlookup = new float[N];
|
||||
coslookup = new float[N];
|
||||
for (int i = 0; i < N; i++) {
|
||||
sinlookup[i] = (float) Math.sin(-(float) Math.PI / i);
|
||||
coslookup[i] = (float) Math.cos(-(float) Math.PI / i);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,662 @@
|
||||
/*
|
||||
* Copyright (c) 2007 - 2008 by Damien Di Fede <ddf@compartmental.net>
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Library General Public License as published
|
||||
* by the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program 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 Library General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Library General Public
|
||||
* License along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||
*
|
||||
* This Code is taken from the source provided with a tutorial by
|
||||
* http://www.badlogicgames.com (as of 16.05.2013).
|
||||
* The original source can be found here:
|
||||
* http://code.google.com/p/audio-analysis/source/browse/trunk/src/com/badlogic/audio/analysis/FourierTransform.java
|
||||
*/
|
||||
//########## Added Code ##########
|
||||
package org.wyrez.audio.analysis;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import org.wyrez.audio.util.Band;
|
||||
import org.wyrez.audio.util.SampleHelper;
|
||||
//#######End of Added Code #######
|
||||
|
||||
/**
|
||||
* A Fourier Transform is an algorithm that transforms a signal in the time
|
||||
* domain, such as a sample buffer, into a signal in the frequency domain, often
|
||||
* called the spectrum. The spectrum does not represent individual frequencies,
|
||||
* but actually represents frequency bands centered on particular frequencies.
|
||||
* The center frequency of each band is usually expressed as a fraction of the
|
||||
* sampling rate of the time domain signal and is equal to the index of the
|
||||
* frequency band divided by the total number of bands. The total number of
|
||||
* frequency bands is usually equal to the length of the time domain signal, but
|
||||
* access is only provided to frequency bands with indices less than half the
|
||||
* length, because they correspond to frequencies below the <a
|
||||
* href="http://en.wikipedia.org/wiki/Nyquist_frequency">Nyquist frequency</a>.
|
||||
* In other words, given a signal of length
|
||||
* <code>N</code>, there will be
|
||||
* <code>N/2</code> frequency bands in the spectrum.
|
||||
* <p>
|
||||
* As an example, if you construct a FourierTransform with a
|
||||
* <code>timeSize</code> of 1024 and and a
|
||||
* <code>sampleRate</code> of 44100 Hz, then the spectrum will contain values
|
||||
* for frequencies below 22010 Hz, which is the Nyquist frequency (half the
|
||||
* sample rate). If you ask for the value of band number 5, this will correspond
|
||||
* to a frequency band centered on
|
||||
* <code>5/1024 * 44100 = 0.0048828125 * 44100 = 215 Hz</code>. The width of
|
||||
* that frequency band is equal to
|
||||
* <code>2/1024</code>, expressed as a fraction of the total bandwidth of the
|
||||
* spectrum. The total bandwith of the spectrum is equal to the Nyquist
|
||||
* frequency, which in this case is 22100, so the bandwidth is equal to about 50
|
||||
* Hz. It is not necessary for you to remember all of these relationships,
|
||||
* though it is good to be aware of them. The function
|
||||
* <code>getFreq()</code> allows you to query the spectrum with a frequency in
|
||||
* Hz and the function
|
||||
* <code>getBandWidth()</code> will return the bandwidth in Hz of each frequency
|
||||
* band in the spectrum.
|
||||
* <p>
|
||||
* <b>Usage</b>
|
||||
* <p>
|
||||
* A typical usage of a FourierTransform is to analyze a signal so that the
|
||||
* frequency spectrum may be represented in some way, typically with vertical
|
||||
* lines. You could do this in Processing with the following code, where
|
||||
* <code>audio</code> is an AudioSource and
|
||||
* <code>fft</code> is an FFT (one of the derived classes of FourierTransform).
|
||||
*
|
||||
* <pre>
|
||||
* fft.forward(audio.left);
|
||||
* for (int i = 0; i < fft.specSize(); i++)
|
||||
* {
|
||||
* // draw the line for frequency band i, scaling it by 4 so we can see it a bit better
|
||||
* line(i, height, i, height - fft.getBand(i) * 4);
|
||||
* }
|
||||
* </pre>
|
||||
*
|
||||
* <b>Windowing</b>
|
||||
* <p>
|
||||
* Windowing is the process of shaping the audio samples before transforming
|
||||
* them to the frequency domain. If you call the
|
||||
* <code>window()</code> function with an appropriate constant, such as
|
||||
* FourierTransform.HAMMING, the sample buffers passed to the object for
|
||||
* analysis will be shaped by the current window before being transformed. The
|
||||
* result of using a window is to reduce the noise in the spectrum somewhat.
|
||||
* <p>
|
||||
* <b>Averages</b>
|
||||
* <p>
|
||||
* FourierTransform also has functions that allow you to request the creation of
|
||||
* an average spectrum. An average spectrum is simply a spectrum with fewer
|
||||
* bands than the full spectrum where each average band is the average of the
|
||||
* amplitudes of some number of contiguous frequency bands in the full spectrum.
|
||||
* <p>
|
||||
* <code>linAverages()</code> allows you to specify the number of averages that
|
||||
* you want and will group frequency bands into groups of equal number. So if
|
||||
* you have a spectrum with 512 frequency bands and you ask for 64 averages,
|
||||
* each average will span 8 bands of the full spectrum.
|
||||
* <p>
|
||||
* <code>logAverages()</code> will group frequency bands by octave and allows
|
||||
* you to specify the size of the smallest octave to use (in Hz) and also how
|
||||
* many bands to split each octave into. So you might ask for the smallest
|
||||
* octave to be 60 Hz and to split each octave into two bands. The result is
|
||||
* that the bandwidth of each average is different. One frequency is an octave
|
||||
* above another when it's frequency is twice that of the lower frequency. So,
|
||||
* 120 Hz is an octave above 60 Hz, 240 Hz is an octave above 120 Hz, and so on.
|
||||
* When octaves are split, they are split based on Hz, so if you split the
|
||||
* octave 60-120 Hz in half, you will get 60-90Hz and 90-120Hz. You can see how
|
||||
* these bandwidths increase as your octave sizes grow. For instance, the last
|
||||
* octave will always span
|
||||
* <code>sampleRate/4 - sampleRate/2</code>, which in the case of audio sampled
|
||||
* at 44100 Hz is 11025-22010 Hz. These logarithmically spaced averages are
|
||||
* usually much more useful than the full spectrum or the linearly spaced
|
||||
* averages because they map more directly to how humans perceive sound.
|
||||
* <p>
|
||||
* <code>calcAvg()</code> allows you to specify the frequency band you want an
|
||||
* average calculated for. You might ask for 60-500Hz and this function will
|
||||
* group together the bands from the full spectrum that fall into that range and
|
||||
* average their amplitudes for you.
|
||||
* <p>
|
||||
* If you don't want any averages calculated, then you can call
|
||||
* <code>noAverages()</code>. This will not impact your ability to use
|
||||
* <code>calcAvg()</code>, it will merely prevent the object from calculating an
|
||||
* average array every time you use
|
||||
* <code>forward()</code>.
|
||||
* <p>
|
||||
* <b>Inverse Transform</b>
|
||||
* <p>
|
||||
* FourierTransform also supports taking the inverse transform of a spectrum.
|
||||
* This means that a frequency spectrum will be transformed into a time domain
|
||||
* signal and placed in a provided sample buffer. The length of the time domain
|
||||
* signal will be
|
||||
* <code>timeSize()</code> long. The
|
||||
* <code>set</code> and
|
||||
* <code>scale</code> functions allow you the ability to shape the spectrum
|
||||
* already stored in the object before taking the inverse transform. You might
|
||||
* use these to filter frequencies in a spectrum or modify it in some other way.
|
||||
*
|
||||
* @author Damien Di Fede
|
||||
* @see <a href="http://www.dspguide.com/ch8.htm">The Discrete Fourier
|
||||
* Transform</a>
|
||||
*/
|
||||
public abstract class FourierTransform {
|
||||
|
||||
/**
|
||||
* A constant indicating no window should be used on sample buffers.
|
||||
*/
|
||||
public static final int NONE = 0;
|
||||
/**
|
||||
* A constant indicating a Hamming window should be used on sample buffers.
|
||||
*/
|
||||
public static final int HAMMING = 1;
|
||||
protected static final int LINAVG = 2;
|
||||
protected static final int LOGAVG = 3;
|
||||
protected static final int NOAVG = 4;
|
||||
protected static final float TWO_PI = (float) (2 * Math.PI);
|
||||
protected int timeSize;
|
||||
protected int sampleRate;
|
||||
protected float bandWidth;
|
||||
protected int whichWindow;
|
||||
protected float[] real;
|
||||
protected float[] imag;
|
||||
protected float[] spectrum;
|
||||
protected float[] averages;
|
||||
protected int whichAverage;
|
||||
protected int octaves;
|
||||
protected int avgPerOctave;
|
||||
|
||||
/**
|
||||
* Construct a FourierTransform that will analyze sample buffers that are
|
||||
* <code>ts</code> samples long and contain samples with a
|
||||
* <code>sr</code> sample rate.
|
||||
*
|
||||
* @param ts the length of the buffers that will be analyzed
|
||||
* @param sr the sample rate of the samples that will be analyzed
|
||||
*/
|
||||
FourierTransform(int ts, float sr) {
|
||||
timeSize = ts;
|
||||
sampleRate = (int) sr;
|
||||
bandWidth = (2f / timeSize) * ((float) sampleRate / 2f);
|
||||
noAverages();
|
||||
allocateArrays();
|
||||
whichWindow = NONE;
|
||||
}
|
||||
|
||||
// allocating real, imag, and spectrum are the responsibility of derived
|
||||
// classes
|
||||
// because the size of the arrays will depend on the implementation being used
|
||||
// this enforces that responsibility
|
||||
protected abstract void allocateArrays();
|
||||
|
||||
protected void setComplex(float[] r, float[] i) {
|
||||
if (real.length != r.length && imag.length != i.length) {
|
||||
throw new IllegalArgumentException("This won't work");
|
||||
} else {
|
||||
System.arraycopy(r, 0, real, 0, r.length);
|
||||
System.arraycopy(i, 0, imag, 0, i.length);
|
||||
}
|
||||
}
|
||||
|
||||
// fill the spectrum array with the amps of the data in real and imag
|
||||
// used so that this class can handle creating the average array
|
||||
// and also do spectrum shaping if necessary
|
||||
protected void fillSpectrum() {
|
||||
for (int i = 0; i < spectrum.length; i++) {
|
||||
spectrum[i] = (float) Math.sqrt(real[i] * real[i] + imag[i] * imag[i]);
|
||||
}
|
||||
|
||||
if (whichAverage == LINAVG) {
|
||||
int avgWidth = (int) spectrum.length / averages.length;
|
||||
for (int i = 0; i < averages.length; i++) {
|
||||
float avg = 0;
|
||||
int j;
|
||||
for (j = 0; j < avgWidth; j++) {
|
||||
int offset = j + i * avgWidth;
|
||||
if (offset < spectrum.length) {
|
||||
avg += spectrum[offset];
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
avg /= j + 1;
|
||||
averages[i] = avg;
|
||||
}
|
||||
} else if (whichAverage == LOGAVG) {
|
||||
for (int i = 0; i < octaves; i++) {
|
||||
float lowFreq, hiFreq, freqStep;
|
||||
if (i == 0) {
|
||||
lowFreq = 0;
|
||||
} else {
|
||||
lowFreq = (sampleRate / 2) / (float) Math.pow(2, octaves - i);
|
||||
}
|
||||
hiFreq = (sampleRate / 2) / (float) Math.pow(2, octaves - i - 1);
|
||||
freqStep = (hiFreq - lowFreq) / avgPerOctave;
|
||||
float f = lowFreq;
|
||||
for (int j = 0; j < avgPerOctave; j++) {
|
||||
int offset = j + i * avgPerOctave;
|
||||
averages[offset] = calcAvg(f, f + freqStep);
|
||||
f += freqStep;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the object to not compute averages.
|
||||
*
|
||||
*/
|
||||
public void noAverages() {
|
||||
averages = new float[0];
|
||||
whichAverage = NOAVG;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the number of averages used when computing the spectrum and spaces
|
||||
* the averages in a linear manner. In other words, each average band will
|
||||
* be
|
||||
* <code>specSize() / numAvg</code> bands wide.
|
||||
*
|
||||
* @param numAvg how many averages to compute
|
||||
*/
|
||||
public void linAverages(int numAvg) {
|
||||
if (numAvg > spectrum.length / 2) {
|
||||
throw new IllegalArgumentException("The number of averages for this transform can be at most " + spectrum.length / 2 + ".");
|
||||
} else {
|
||||
averages = new float[numAvg];
|
||||
}
|
||||
whichAverage = LINAVG;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the number of averages used when computing the spectrum based on the
|
||||
* minimum bandwidth for an octave and the number of bands per octave. For
|
||||
* example, with audio that has a sample rate of 44100 Hz,
|
||||
* <code>logAverages(11, 1)</code> will result in 12 averages, each
|
||||
* corresponding to an octave, the first spanning 0 to 11 Hz. To ensure that
|
||||
* each octave band is a full octave, the number of octaves is computed by
|
||||
* dividing the Nyquist frequency by two, and then the result of that by
|
||||
* two, and so on. This means that the actual bandwidth of the lowest octave
|
||||
* may not be exactly the value specified.
|
||||
*
|
||||
* @param minBandwidth the minimum bandwidth used for an octave
|
||||
* @param bandsPerOctave how many bands to split each octave into
|
||||
*/
|
||||
public void logAverages(int minBandwidth, int bandsPerOctave) {
|
||||
float nyq = (float) sampleRate / 2f;
|
||||
octaves = 1;
|
||||
while ((nyq /= 2) > minBandwidth) {
|
||||
octaves++;
|
||||
}
|
||||
avgPerOctave = bandsPerOctave;
|
||||
averages = new float[octaves * bandsPerOctave];
|
||||
whichAverage = LOGAVG;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the window to use on the samples before taking the forward
|
||||
* transform. If an invalid window is asked for, an error will be reported
|
||||
* and the current window will not be changed.
|
||||
*
|
||||
* @param which FourierTransform.HAMMING or FourierTransform.NONE
|
||||
*/
|
||||
public void window(int which) {
|
||||
if (which < 0 || which > 1) {
|
||||
throw new IllegalArgumentException("Invalid window type.");
|
||||
} else {
|
||||
whichWindow = which;
|
||||
}
|
||||
}
|
||||
|
||||
protected void doWindow(float[] samples) {
|
||||
switch (whichWindow) {
|
||||
case HAMMING:
|
||||
hamming(samples);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// windows the data in samples with a Hamming window
|
||||
protected void hamming(float[] samples) {
|
||||
for (int i = 0; i < samples.length; i++) {
|
||||
samples[i] *= (0.54f - 0.46f * Math.cos(TWO_PI * i / (samples.length - 1)));
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the length of the time domain signal expected by this transform.
|
||||
*
|
||||
* @return the length of the time domain signal expected by this transform
|
||||
*/
|
||||
public int timeSize() {
|
||||
return timeSize;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the size of the spectrum created by this transform. In other
|
||||
* words, the number of frequency bands produced by this transform. This is
|
||||
* typically equal to
|
||||
* <code>timeSize()/2 + 1</code>, see above for an explanation.
|
||||
*
|
||||
* @return the size of the spectrum
|
||||
*/
|
||||
public int specSize() {
|
||||
return spectrum.length;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the amplitude of the requested frequency band.
|
||||
*
|
||||
* @param i the index of a frequency band
|
||||
* @return the amplitude of the requested frequency band
|
||||
*/
|
||||
public float getBand(int i) {
|
||||
if (i < 0) {
|
||||
i = 0;
|
||||
}
|
||||
if (i > spectrum.length - 1) {
|
||||
i = spectrum.length - 1;
|
||||
}
|
||||
return spectrum[i];
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the width of each frequency band in the spectrum (in Hz). It
|
||||
* should be noted that the bandwidth of the first and last frequency bands
|
||||
* is half as large as the value returned by this function.
|
||||
*
|
||||
* @return the width of each frequency band in Hz.
|
||||
*/
|
||||
public float getBandWidth() {
|
||||
return bandWidth;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the amplitude of the
|
||||
* <code>i<sup>th</sup></code> frequency band to
|
||||
* <code>a</code>. You can use this to shape the spectrum before using
|
||||
* <code>inverse()</code>.
|
||||
*
|
||||
* @param i the frequency band to modify
|
||||
* @param a the new amplitude
|
||||
*/
|
||||
public abstract void setBand(int i, float a);
|
||||
|
||||
/**
|
||||
* Scales the amplitude of the
|
||||
* <code>i<sup>th</sup></code> frequency band by
|
||||
* <code>s</code>. You can use this to shape the spectrum before using
|
||||
* <code>inverse()</code>.
|
||||
*
|
||||
* @param i the frequency band to modify
|
||||
* @param s the scaling factor
|
||||
*/
|
||||
public abstract void scaleBand(int i, float s);
|
||||
|
||||
/**
|
||||
* Returns the index of the frequency band that contains the requested
|
||||
* frequency.
|
||||
*
|
||||
* @param freq the frequency you want the index for (in Hz)
|
||||
* @return the index of the frequency band that contains freq
|
||||
*/
|
||||
public int freqToIndex(float freq) {
|
||||
// special case: freq is lower than the bandwidth of spectrum[0]
|
||||
if (freq < getBandWidth() / 2) {
|
||||
return 0;
|
||||
}
|
||||
// special case: freq is within the bandwidth of spectrum[spectrum.length - 1]
|
||||
if (freq > sampleRate / 2 - getBandWidth() / 2) {
|
||||
return spectrum.length - 1;
|
||||
}
|
||||
// all other cases
|
||||
float fraction = freq / (float) sampleRate;
|
||||
int i = Math.round(timeSize * fraction);
|
||||
return i;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the middle frequency of the i<sup>th</sup> band.
|
||||
*
|
||||
* @param i the index of the band you want to middle frequency of
|
||||
*/
|
||||
public float indexToFreq(int i) {
|
||||
float bw = getBandWidth();
|
||||
// special case: the width of the first bin is half that of the others.
|
||||
// so the center frequency is a quarter of the way.
|
||||
if (i == 0) {
|
||||
return bw * 0.25f;
|
||||
}
|
||||
// special case: the width of the last bin is half that of the others.
|
||||
if (i == spectrum.length - 1) {
|
||||
float lastBinBeginFreq = (sampleRate / 2) - (bw / 2);
|
||||
float binHalfWidth = bw * 0.25f;
|
||||
return lastBinBeginFreq + binHalfWidth;
|
||||
}
|
||||
// the center frequency of the ith band is simply i*bw
|
||||
// because the first band is half the width of all others.
|
||||
// treating it as if it wasn't offsets us to the middle
|
||||
// of the band.
|
||||
return i * bw;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the center frequency of the i<sup>th</sup> average band.
|
||||
*
|
||||
* @param i which average band you want the center frequency of.
|
||||
*/
|
||||
public float getAverageCenterFrequency(int i) {
|
||||
if (whichAverage == LINAVG) {
|
||||
// an average represents a certain number of bands in the spectrum
|
||||
int avgWidth = (int) spectrum.length / averages.length;
|
||||
// the "center" bin of the average, this is fudgy.
|
||||
int centerBinIndex = i * avgWidth + avgWidth / 2;
|
||||
return indexToFreq(centerBinIndex);
|
||||
|
||||
} else if (whichAverage == LOGAVG) {
|
||||
// which "octave" is this index in?
|
||||
int octave = i / avgPerOctave;
|
||||
// which band within that octave is this?
|
||||
int offset = i % avgPerOctave;
|
||||
float lowFreq, hiFreq, freqStep;
|
||||
// figure out the low frequency for this octave
|
||||
if (octave == 0) {
|
||||
lowFreq = 0;
|
||||
} else {
|
||||
lowFreq = (sampleRate / 2) / (float) Math.pow(2, octaves - octave);
|
||||
}
|
||||
// and the high frequency for this octave
|
||||
hiFreq = (sampleRate / 2) / (float) Math.pow(2, octaves - octave - 1);
|
||||
// each average band within the octave will be this big
|
||||
freqStep = (hiFreq - lowFreq) / avgPerOctave;
|
||||
// figure out the low frequency of the band we care about
|
||||
float f = lowFreq + offset * freqStep;
|
||||
// the center of the band will be the low plus half the width
|
||||
return f + freqStep / 2;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the amplitude of the requested frequency in the spectrum.
|
||||
*
|
||||
* @param freq the frequency in Hz
|
||||
* @return the amplitude of the frequency in the spectrum
|
||||
*/
|
||||
public float getFreq(float freq) {
|
||||
return getBand(freqToIndex(freq));
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the amplitude of the requested frequency in the spectrum to
|
||||
* <code>a</code>.
|
||||
*
|
||||
* @param freq the frequency in Hz
|
||||
* @param a the new amplitude
|
||||
*/
|
||||
public void setFreq(float freq, float a) {
|
||||
setBand(freqToIndex(freq), a);
|
||||
}
|
||||
|
||||
/**
|
||||
* Scales the amplitude of the requested frequency by
|
||||
* <code>a</code>.
|
||||
*
|
||||
* @param freq the frequency in Hz
|
||||
* @param s the scaling factor
|
||||
*/
|
||||
public void scaleFreq(float freq, float s) {
|
||||
scaleBand(freqToIndex(freq), s);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the number of averages currently being calculated.
|
||||
*
|
||||
* @return the length of the averages array
|
||||
*/
|
||||
public int avgSize() {
|
||||
return averages.length;
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the value of the
|
||||
* <code>i<sup>th</sup></code> average.
|
||||
*
|
||||
* @param i the average you want the value of
|
||||
* @return the value of the requested average band
|
||||
*/
|
||||
public float getAvg(int i) {
|
||||
float ret;
|
||||
if (averages.length > 0) {
|
||||
ret = averages[i];
|
||||
} else {
|
||||
ret = 0;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
/**
|
||||
* Calculate the average amplitude of the frequency band bounded by
|
||||
* <code>lowFreq</code> and
|
||||
* <code>hiFreq</code>, inclusive.
|
||||
*
|
||||
* @param lowFreq the lower bound of the band
|
||||
* @param hiFreq the upper bound of the band
|
||||
* @return the average of all spectrum values within the bounds
|
||||
*/
|
||||
public float calcAvg(float lowFreq, float hiFreq) {
|
||||
int lowBound = freqToIndex(lowFreq);
|
||||
int hiBound = freqToIndex(hiFreq);
|
||||
float avg = 0;
|
||||
for (int i = lowBound; i <= hiBound; i++) {
|
||||
avg += spectrum[i];
|
||||
}
|
||||
avg /= (hiBound - lowBound + 1);
|
||||
return avg;
|
||||
}
|
||||
|
||||
/**
|
||||
* Performs a forward transform on
|
||||
* <code>buffer</code>.
|
||||
*
|
||||
* @param buffer the buffer to analyze
|
||||
*/
|
||||
public abstract void forward(float[] buffer);
|
||||
|
||||
/**
|
||||
* Performs a forward transform on values in
|
||||
* <code>buffer</code>.
|
||||
*
|
||||
* @param buffer the buffer of samples
|
||||
* @param startAt the index to start at in the buffer. there must be at
|
||||
* least timeSize() samples between the starting index and the end of the
|
||||
* buffer. If there aren't, an error will be issued and the operation will
|
||||
* not be performed.
|
||||
*
|
||||
*/
|
||||
public void forward(float[] buffer, int startAt) {
|
||||
if (buffer.length - startAt < timeSize) {
|
||||
throw new IllegalArgumentException("FourierTransform.forward: not enough samples in the buffer between " + startAt + " and " + buffer.length + " to perform a transform.");
|
||||
}
|
||||
|
||||
// copy the section of samples we want to analyze
|
||||
float[] section = new float[timeSize];
|
||||
System.arraycopy(buffer, startAt, section, 0, section.length);
|
||||
forward(section);
|
||||
}
|
||||
|
||||
/**
|
||||
* Performs an inverse transform of the frequency spectrum and places the
|
||||
* result in
|
||||
* <code>buffer</code>.
|
||||
*
|
||||
* @param buffer the buffer to place the result of the inverse transform in
|
||||
*/
|
||||
public abstract void inverse(float[] buffer);
|
||||
|
||||
/**
|
||||
* Performs an inverse transform of the frequency spectrum represented by
|
||||
* freqReal and freqImag and places the result in buffer.
|
||||
*
|
||||
* @param freqReal the real part of the frequency spectrum
|
||||
* @param freqImag the imaginary part the frequency spectrum
|
||||
* @param buffer the buffer to place the inverse transform in
|
||||
*/
|
||||
public void inverse(float[] freqReal, float[] freqImag, float[] buffer) {
|
||||
setComplex(freqReal, freqImag);
|
||||
inverse(buffer);
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the spectrum of the last FourierTransform.forward() call.
|
||||
*/
|
||||
public float[] getSpectrum() {
|
||||
return spectrum;
|
||||
}
|
||||
|
||||
//########## Added Code ##########
|
||||
/**
|
||||
* @return the spectrum of the last FourierTransform.forward() call for the
|
||||
* specified range.
|
||||
*/
|
||||
public float[] getSpectrum(Band band) {
|
||||
if (band == null) {
|
||||
return spectrum;
|
||||
}
|
||||
ArrayList<Float> spec = new ArrayList<Float>();
|
||||
for (float f = band.getLowFrequenzy(); f < band.getHighFrequenzy() + getBandWidth(); f += getBandWidth()) {
|
||||
spec.add(getFreq(f));
|
||||
}
|
||||
return SampleHelper.convertToFloatArray(spec);
|
||||
}
|
||||
|
||||
public int getBandSize(Band band) {
|
||||
if (band == null) {
|
||||
return spectrum.length;
|
||||
}
|
||||
int counter = 0;
|
||||
for (float f = band.getLowFrequenzy(); f < band.getHighFrequenzy() + getBandWidth(); f += getBandWidth()) {
|
||||
counter++;
|
||||
}
|
||||
return counter;
|
||||
}
|
||||
//#######End of Added Code #######
|
||||
|
||||
/**
|
||||
* @return the real part of the last FourierTransform.forward() call.
|
||||
*/
|
||||
public float[] getRealPart() {
|
||||
return real;
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the imaginary part of the last FourierTransform.forward() call.
|
||||
*/
|
||||
public float[] getImaginaryPart() {
|
||||
return imag;
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user