Real2DFFTData.H

Go to the documentation of this file.

/* Copyright 2000-2018 Matt Flax <flatmax@flatmax.org>
   This file is part of GTK+ IOStream class set

   GTK+ IOStream 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.

   GTK+ IOStream 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 have received a copy of the GNU General Public License
   along with GTK+ IOStream
*/
#ifndef REAL2DFFTDATA_H_
#define REAL2DFFTDATA_H_

#include "fft/FFTCommon.H"
//#include "fft/Real2DFFT.H"

class Real2DFFTData {
  int x, y;
  fftw_real *mem;
  void memDeInit(void);
public:
  fftw_real *in, *power;
  fftw_complex *out;
  fftw_real *xSum, *ySum;
  fftw_real *timeXSum;
  fftw_real *realXSum, *imagXSum;

  double totalPower, maxPower, minPower;
  double xSumMin, xSumMax, ySumMin, ySumMax;
  int maxXSumIndex, maxYSumIndex;

  Real2DFFTData(int r, int c);
  ~Real2DFFTData();

  int getXSize(){return x;}
  int getYSize(){return y;}
  int getXHalfSize(){ if (!(x%2)) return x/2; else return x/2+1;}
  int getYHalfSize(){ if (!(y%2)) return y/2; else return y/2+1;}

  void reScale(void);
  void compPowerSpec(); // Find the power spectrum
  int sqrtPowerSpec();
  void compLogPowerSpec(); // Find the log power spectrum

  void timeSpecAverage();
  void complexSpecAverage();
  void powerSpecAverage();
  void findYSum(int start, int stop);
  void findYMax(void);

  void clearInput(void);
  void clearOutput(void);
};
#endif // REAL2DFFTDATA_H_