File: fourier.h

package info (click to toggle)
ctsim 4.5.2-1.1
  • links: PTS
  • area: main
  • in suites: etch, etch-m68k
  • size: 6,536 kB
  • ctags: 3,720
  • sloc: cpp: 26,768; sh: 3,691; ansic: 1,254; perl: 296; makefile: 263
file content (114 lines) | stat: -rw-r--r-- 3,830 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/*****************************************************************************
** FILE IDENTIFICATION
**
**   Name:          fourier.h
**   Purpose:       Header for Fourier transform functions
**   Programmer:    Kevin Rosenberg
**   Date Started:  Dec 2000
**
**  This is part of the CTSim program
**  Copyright (c) 1983-2001 Kevin Rosenberg
**
**  $Id: fourier.h 7939 2003-10-05 05:47:22Z kevin $
**
**  This program is free software; you can redistribute it and/or modify
**  it under the terms of the GNU General Public License (version 2) as
**  published by the Free Software Foundation.
**
**  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 <complex>
#ifdef HAVE_FFTW
#include <fftw3.h>
#endif

class ImageFile;

class Fourier {
public:
    static void shuffleFourierToNaturalOrder (ImageFile& im);
    static void shuffleNaturalToFourierOrder (ImageFile& im);

#ifdef HAVE_FFTW
    static void shuffleFourierToNaturalOrder (fftw_complex* pc, const int n);
    static void shuffleNaturalToFourierOrder (fftw_complex* pc, const int n);
#endif
    
// Odd Number of Points
//   Natural Frequency Order: -(n-1)/2...-1,0,1...(n-1)/2
//   Fourier Frequency Order: 0, 1..(n-1)/2,-(n-1)/2...-1
// Even Number of Points
//   Natural Frequency Order: -n/2...-1,0,1...((n/2)-1)
//   Fourier Frequency Order: 0,1...((n/2)-1),-n/2...-1
    template<class T>
    static void shuffleNaturalToFourierOrder (T* pVector, const int n) 
    {
      T* pTemp = new T [n];
      int i;
      if (isOdd(n)) { // Odd
        int iHalfN = (n - 1) / 2;

        pTemp[0] = pVector[iHalfN];
        for (i = 0; i < iHalfN; i++)
          pTemp[i + 1] = pVector[i + 1 + iHalfN];
        for (i = 0; i < iHalfN; i++)
          pTemp[i + iHalfN + 1] = pVector[i];
      } else {     // Even
        int iHalfN = n / 2;
        pTemp[0] = pVector[iHalfN];
        for (i = 0; i < iHalfN - 1; i++)
        pTemp[i + 1] = pVector[i + iHalfN + 1];
        for (i = 0; i < iHalfN; i++)
          pTemp[i + iHalfN] = pVector[i];
      }

    for (i = 0; i < n; i++)
      pVector[i] = pTemp[i];
    delete pTemp;
  }

  template<class T>
  static void shuffleFourierToNaturalOrder (T* pVector, const int n)
  {
    T* pTemp = new T [n];
    int i;
    if (isOdd(n)) { // Odd
      int iHalfN = (n - 1) / 2;
    
      pTemp[iHalfN] = pVector[0];
      for (i = 0; i < iHalfN; i++)
        pTemp[i + 1 + iHalfN] = pVector[i + 1];
      for (i = 0; i < iHalfN; i++)
        pTemp[i] = pVector[i + iHalfN + 1];
    } else {     // Even
      int iHalfN = n / 2;
      pTemp[iHalfN] = pVector[0];
      for (i = 0; i < iHalfN; i++)
        pTemp[i] = pVector[i + iHalfN];
      for (i = 0; i < iHalfN - 1; i++)
        pTemp[i + iHalfN + 1] = pVector[i+1];
    }
  
    for (i = 0; i < n; i++)
      pVector[i] = pTemp[i];
    delete pTemp;
  }
#if 0
    static void shuffleNaturalToFourierOrder (float* pdVector, const int n);
    static void shuffleNaturalToFourierOrder (double* pdVector, const int n);
    static void shuffleNaturalToFourierOrder (std::complex<double>* pdVector, const int n);
    static void shuffleFourierToNaturalOrder (float* pdVector, const int n);
    static void shuffleFourierToNaturalOrder (double* pdVector, const int n);
    static void shuffleFourierToNaturalOrder (std::complex<double>* pdVector, const int n);
#endif
};