// Copyright (C) 2005 Dave Griffiths
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

#include <fftw3.h>
#include <pthread.h>
#include <string>
#include <math.h>

#ifndef AUDIO_COLLECTOR
#define AUDIO_COLLECTOR

using namespace std;

static const int NUM_BARS = 16;
//static const int NUM_BARS = 32;


class FFT
{
public:
    FFT(int length);
    ~FFT();
	void Impulse2Freq(float *imp, float *out);
private:	
	fftw_plan m_Plan;
	unsigned int m_FFTLength;
	double *m_In;
	fftw_complex *m_Spectrum;
};

class AudioCollector
{
public:
	AudioCollector(int BufferLength, int FFTBuffers = 1);
	~AudioCollector();
	
	float *GetFFT();
	float *GetAudioBuffer() { return m_AudioBuffer; }
    float GetHarmonic(int h);
    bool  IsConnected();
    void  SetGain(float s) { m_Gain=s; }
    void  Process(const string &filename);
	int GetNumBars();
	
private:

    void AudioCallback_i(unsigned int);
	static void AudioCallback(void *, unsigned int);

	float m_Gain;
	
	int m_BufferLength;
	FFT m_FFT;
	pthread_mutex_t* m_Mutex;
	float *m_Buffer;
	float *m_AudioBuffer;
	float *m_FFTBuffer;
	float *m_FFTOutput;
	int    m_FFTBuffers;
	int    m_InputPort;
	
	float *m_JackBuffer;
	
	int    m_Dspfd;
	short *m_OSSBuffer;
	float  m_OneOverSHRT_MAX;
	bool   m_Processing;
	float *m_ProcessBuffer;
	int    m_ProcessPos;
	int	   m_ProcessLength;
};

#endif