ITK  5.2.0
Insight Toolkit
SphinxExamples/src/Filtering/ImageGrid/WarpAnImageUsingADeformationField/Code.py
1 #!/usr/bin/env python
2 
3 # Copyright NumFOCUS
4 #
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
8 #
9 # http://www.apache.org/licenses/LICENSE-2.0.txt
10 #
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
16 
17 import sys
18 import itk
19 
20 if len(sys.argv) != 4:
21  print('Usage: ' + sys.argv[0] +
22  ' <InputFileName> <DisplacementFieldFileName> <OutputFileName>')
23  sys.exit(1)
24 
25 inputFileName = sys.argv[1]
26 displacementFieldFileName = sys.argv[2]
27 outputFileName = sys.argv[3]
28 
29 Dimension = 2
30 
31 VectorComponentType = itk.F
32 VectorPixelType = itk.Vector[VectorComponentType, Dimension]
33 
34 DisplacementFieldType = itk.Image[VectorPixelType, Dimension]
35 
36 PixelType = itk.UC
37 ImageType = itk.Image[PixelType, Dimension]
38 
39 reader = itk.ImageFileReader[ImageType].New()
40 reader.SetFileName(inputFileName)
41 
42 fieldReader = itk.ImageFileReader[DisplacementFieldType].New()
43 fieldReader.SetFileName(displacementFieldFileName)
44 fieldReader.Update()
45 
46 deformationField = fieldReader.GetOutput()
47 
48 warpFilter = \
49  itk.WarpImageFilter[ImageType, ImageType, DisplacementFieldType].New()
50 
51 interpolator = itk.LinearInterpolateImageFunction[ImageType, itk.D].New()
52 
53 warpFilter.SetInterpolator(interpolator)
54 
55 warpFilter.SetOutputSpacing(deformationField.GetSpacing())
56 warpFilter.SetOutputOrigin(deformationField.GetOrigin())
57 warpFilter.SetOutputDirection(deformationField.GetDirection())
58 
59 warpFilter.SetDisplacementField(deformationField)
60 
61 warpFilter.SetInput(reader.GetOutput())
62 
63 writer = itk.ImageFileWriter[ImageType].New()
64 writer.SetInput(warpFilter.GetOutput())
65 writer.SetFileName(outputFileName)
66 
67 writer.Update()
itk::Vector
A templated class holding a n-Dimensional vector.
Definition: itkVector.h:62
itk::ImageFileReader
Data source that reads image data from a single file.
Definition: itkImageFileReader.h:75
itk::LinearInterpolateImageFunction
Linearly interpolate an image at specified positions.
Definition: itkLinearInterpolateImageFunction.h:50
itk::ImageFileWriter
Writes image data to a single file.
Definition: itkImageFileWriter.h:87
itk::WarpImageFilter
Warps an image using an input displacement field.
Definition: itkWarpImageFilter.h:85
itk::Image
Templated n-dimensional image class.
Definition: itkImage.h:86