from sys import argv

import numpy as np

​

if len(argv) != 4:

    print("Usage: " + argv[0] + " <fixed2DImage> " + " <moving3DVolume> " + " <outputImage> " )

    exit(1)

​

​

# ---------- User-defined parameters ----------------

movInitTranslx = 0.

movInitTransly = 0.

movInitTranslz = 0.

​

# parameters for the Ray cast interpolator (same ones used to generate the FixedImage)

focal_length = 1000.

threshold = 0 # threshold above which the volume's intensities will be integrated.

​

# parameters for the resampled DRR (same ones used to generate the FixedImage)

dx = 1000 # size (# pixels) of the output image

dy = 1000 # size (# pixels) of the output image

sx = 0.5 # spacing x [mm] of the output image

sy = 0.5 # spacing y [mm] of the output image

o2Dx = 0 # principal points coordinate x [mm] (usually the 2D projection normal is assumed to be perpendicular )

o2Dy = 0 # principal points coordinate y [mm]

​

# ---------- Initializations ----------------

fixedImageFile = argv[1]

movingImageFile = argv[2]

​

PixelType = itk.F

Dimension = 3

ScalarType = itk.D

​

FixedImageType = itk.Image[PixelType, Dimension]

MovingImageType = itk.Image[PixelType, Dimension]

​

fixedImageReader  = itk.ImageFileReader[FixedImageType].New()

movingImageReader = itk.ImageFileReader[MovingImageType].New()

​

OptimizerType = itk.RegularStepGradientDescentOptimizerv4[ScalarType]

​

RegistrationType = itk.ImageRegistrationMethodv4[FixedImageType,

                                                 MovingImageType]

​

TransformType = itk.TranslationTransform[ScalarType, Dimension]

​

MetricType = itk.MeanSquaresImageToImageMetricv4[FixedImageType,

                                                 MovingImageType]

​

FixedInterpolatorType = itk.LinearInterpolateImageFunction[FixedImageType, ScalarType]

MovingInterpolatorType = itk.RayCastInterpolateImageFunction[MovingImageType, ScalarType]

​

#WriterTypeOutput = itk.ImageFileWriter[OutputImageType]

​

​

# ---------- Instantiate the classes for the registration framework ----------------