File: ResampleImageFilter.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 (105 lines) | stat: -rw-r--r-- 2,586 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
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

#     INPUTS: {BrainProtonDensitySlice.png}
#     OUTPUTS: {ResampleImageFilterOutput1.png}
#     0

#     INPUTS: {BrainProtonDensitySlice.png}
#     OUTPUTS: {ResampleImageFilterOutput2.png}
#     1

#     INPUTS: {BrainProtonDensitySlice.png}
#     OUTPUTS: {ResampleImageFilterOutput3.png}
#     2

#     INPUTS: {BrainProtonDensitySlice.png}
#     OUTPUTS: {ResampleImageFilterOutput4.png}
#     3

import itk
from sys import argv, stderr, exit
itk.auto_progress(2)


if( len(argv) < 3 ):
  print >> stderr, """Missing Parameters
Usage: ResampleImageFilter.py inputImageFile  outputImageFile [exampleAction={0,1,2,3}]"""
  exit(1)

exampleAction = 0

if( len(argv) >= 4 ):
  exampleAction = int( argv[3] )

Dimension = 2
InputPixelType = itk.US
OutputPixelType = itk.UC
InputImageType = itk.Image[ InputPixelType,  Dimension ]
OutputImageType = itk.Image[ OutputPixelType, Dimension ]

ReaderType = itk.ImageFileReader[ InputImageType  ]
CastType = itk.CastImageFilter[ InputImageType, OutputImageType ]
WriterType = itk.ImageFileWriter[ OutputImageType ]

reader = ReaderType.New()
cast = CastType.New()
writer = WriterType.New()

reader.SetFileName( argv[1] )
writer.SetFileName( argv[2] )

FilterType = itk.ResampleImageFilter[InputImageType,InputImageType] 
filter = FilterType.New()

TransformType = itk.AffineTransform[ itk.D, Dimension ]

transform = TransformType.New()
filter.SetTransform( transform.GetPointer() )

InterpolatorType = itk.NearestNeighborInterpolateImageFunction[ 
                      InputImageType, itk.D ]

interpolator = InterpolatorType.New()
filter.SetInterpolator( interpolator.GetPointer() )

filter.SetDefaultPixelValue( 0 )

# spacing = itk.Vector[itk.D, Dimension]()
# spacing.SetElement(0, 1.0)
# spacing.SetElement(1, 1.0)
# filter.SetOutputSpacing( spacing )
filter.SetOutputSpacing( 1.0 )

# origin = itk.Point[itk.D, Dimension]()
# origin.SetElement(0, 0.0)
# origin.SetElement(1, 0.0)
# filter.SetOutputOrigin( origin )
filter.SetOutputOrigin( [0.0, 0.0] )

# size = itk.Size[Dimension]()
# size.SetElement(0, 300)
# size.SetElement(1, 300)
# filter.SetSize( size )
filter.SetSize( 300 )

filter.SetInput( reader.GetOutput() )
cast.SetInput( filter.GetOutput() )
writer.SetInput( cast.GetOutput() )
writer.Update()

# translation = itk.Vector[itk.D, Dimension]()
# translation.SetElement(0, -30)
# translation.SetElement(1, -50)
# transform.Translate( translation, 0 )
transform.Translate( [-30, -50], 0 )

if( exampleAction == 1 ):
  writer.Update()



filter.SetDefaultPixelValue( 100 )

if( exampleAction == 2 ):
  writer.Update()