File: FFTImageFilter.py

package info (click to toggle)
insighttoolkit 3.20.1%2Bgit20120521-5
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 80,672 kB
  • ctags: 85,253
  • sloc: cpp: 458,133; ansic: 196,222; fortran: 28,000; python: 3,839; tcl: 1,811; sh: 1,184; java: 583; makefile: 428; csh: 220; perl: 193; xml: 20
file content (35 lines) | stat: -rw-r--r-- 1,145 bytes parent folder | download | duplicates (4)
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
#
#  Example on the use of the MeanImageFilter
#

import itk
from sys import argv
itk.auto_progress(2)

dim = 2
PixelType = itk.F
ImageType = itk.Image[PixelType, dim]

reader = itk.ImageFileReader[ImageType].New( FileName=argv[1] )
fftFilter = itk.VnlFFTRealToComplexConjugateImageFilter[PixelType, dim].New(reader)

# why this Update() ?
fftFilter.Update()

ComplexImageType  = itk.Image[itk.complex[PixelType], dim]

complexWriter = itk.ImageFileWriter[ComplexImageType].New( fftFilter, FileName="complexImage.mhd" )
complexWriter.Update()

realFilter = itk.ComplexToRealImageFilter[ComplexImageType, ImageType].New( fftFilter )

WritePixelType = itk.UC
WriteImageType = itk.Image[WritePixelType, dim]
intensityRescaler = itk.RescaleIntensityImageFilter[ImageType, WriteImageType].New( realFilter, OutputMinimum=0, OutputMaximum=255 )
writer = itk.ImageFileWriter[WriteImageType].New( intensityRescaler, FileName=argv[2] )
writer.Update()

imaginaryFilter = itk.ComplexToImaginaryImageFilter[ComplexImageType, ImageType].New( fftFilter )
intensityRescaler.SetInput( imaginaryFilter.GetOutput() )
writer.SetFileName( argv[3] )
writer.Update()