////////////////////////////////////////////////////////////////////////
// This file is part of the SndObj library
//
// 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 
//
// Copyright (c)Victor Lazzarini, 1997-2004
// See License.txt for a disclaimer of all warranties
// and licensing information

//////////////////////////////////////////////////////
// FFT.h: interface of the FFT class: 
//        short-time fast fourier transform
//        using the FFTW library (v. 2.1.3) 
//        Victor Lazzarini, 2003 
/////////////////////////////////////////////////////////

#ifndef _FFT_H
#define _FFT_H
#include "SndObj.h"
#include "Table.h"
#include <fftw3.h>

class FFT : public SndObj {
 protected:
  // m_vecsize is FFT size
  // m_hopsize should always be set to the time-domain
  // vector size
  int m_fftsize;
  int m_hopsize;  // hopsize
  int m_halfsize; // 1/2 fftsize
  int *m_counter; // counter 
  double* m_fftIn;
  fftw_complex* m_fftOut;
  fftw_plan m_plan;
  double m_fund;

  double m_scale; // scaling factor
  double m_norm;  // norm factor
  int m_frames;  // frame overlaps
  double** m_sigframe; // signal frames
  int m_cur;    // index into current frame 

  Table*  m_table; // window

 private:
  // fft wrapper method
  void inline fft(double* signal);
  // reset memory and initialisation
  void ReInit();

 public:
  FFT();
  FFT(Table* window, SndObj* input, double scale=1.f, 
      int fftsize=DEF_FFTSIZE, int hopsize=DEF_VECSIZE,
      double m_sr=DEF_SR);
  ~FFT();
  
  int GetFFTSize() { return m_fftsize; }
  int GetHopSize() { return m_hopsize; }
  void SetWindow(Table* window){ m_table = window;}
  int Connect(const char* mess, void* input);
  int Set(const char* mess, double value);
  void SetScale(double scale){ m_scale = scale; m_norm = m_fftsize/m_scale;}
  virtual void SetFFTSize(int fftsize);
  virtual void SetHopSize(int hopsize);

  short DoProcess();
};

#endif