vtkImageFourierFilter.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Visualization Toolkit
  Module:    vtkImageFourierFilter.h
 
  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
 
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
#ifndef vtkImageFourierFilter_h
#define vtkImageFourierFilter_h
 
#include "vtkImageDecomposeFilter.h"
#include "vtkImagingFourierModule.h" // For export macro
 
/*******************************************************************
                        COMPLEX number stuff
*******************************************************************/
 
struct vtkImageComplex_t
{
  double Real;
  double Imag;
};
using vtkImageComplex = struct vtkImageComplex_t;
 
#define vtkImageComplexEuclidSet(C, R, I)                                                          \
  (C).Real = (R);                                                                                  \
  (C).Imag = (I)
 
#define vtkImageComplexPolarSet(C, M, P)                                                           \
  (C).Real = (M)*cos(P);                                                                           \
  (C).Imag = (M)*sin(P)
 
#define vtkImageComplexPrint(C) printf("(%.3f, %.3f)", (C).Real, (C).Imag)
 
#define vtkImageComplexScale(cOut, S, cIn)                                                         \
  (cOut).Real = (cIn).Real * (S);                                                                  \
  (cOut).Imag = (cIn).Imag * (S)
 
#define vtkImageComplexConjugate(cIn, cOut)                                                        \
  (cOut).Imag = (cIn).Imag * -1.0;                                                                 \
  (cOut).Real = (cIn).Real
 
#define vtkImageComplexAdd(C1, C2, cOut)                                                           \
  (cOut).Real = (C1).Real + (C2).Real;                                                             \
  (cOut).Imag = (C1).Imag + (C2).Imag
 
#define vtkImageComplexSubtract(C1, C2, cOut)                                                      \
  (cOut).Real = (C1).Real - (C2).Real;                                                             \
  (cOut).Imag = (C1).Imag - (C2).Imag
 
#define vtkImageComplexMultiply(C1, C2, cOut)                                                      \
  {                                                                                                \
    vtkImageComplex vtkImageComplex_tMultiplyTemp;                                                 \
    vtkImageComplex_tMultiplyTemp.Real = (C1).Real * (C2).Real - (C1).Imag * (C2).Imag;            \
    vtkImageComplex_tMultiplyTemp.Imag = (C1).Real * (C2).Imag + (C1).Imag * (C2).Real;            \
    cOut = vtkImageComplex_tMultiplyTemp;                                                          \
  }
 
// This macro calculates exp(cIn) and puts the result in cOut
#define vtkImageComplexExponential(cIn, cOut)                                                      \
  {                                                                                                \
    double tmp = exp(cIn.Real);                                                                    \
    cOut.Real = tmp * cos(cIn.Imag);                                                               \
    cOut.Imag = tmp * sin(cIn.Imag);                                                               \
  }
 
/******************* End of COMPLEX number stuff ********************/
 
class VTKIMAGINGFOURIER_EXPORT vtkImageFourierFilter : public vtkImageDecomposeFilter
{
public:
  vtkTypeMacro(vtkImageFourierFilter, vtkImageDecomposeFilter);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  // public for templated functions of this object
 
  void ExecuteFft(vtkImageComplex* in, vtkImageComplex* out, int N);
 
  void ExecuteRfft(vtkImageComplex* in, vtkImageComplex* out, int N);
 
protected:
  vtkImageFourierFilter() = default;
  ~vtkImageFourierFilter() override = default;
 
  void ExecuteFftStep2(vtkImageComplex* p_in, vtkImageComplex* p_out, int N, int bsize, int fb);
  void ExecuteFftStepN(
    vtkImageComplex* p_in, vtkImageComplex* p_out, int N, int bsize, int n, int fb);
  void ExecuteFftForwardBackward(vtkImageComplex* in, vtkImageComplex* out, int N, int fb);
 
  int RequestData(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
 
private:
  vtkImageFourierFilter(const vtkImageFourierFilter&) = delete;
  void operator=(const vtkImageFourierFilter&) = delete;
};
 
#endif