/// <summary>
/// Start transformation process.
/// </summary>
public void StartTransformation()
{
sitk.VectorOfParameterMap parameterMaps = PrepareTransformationParameters();
foreach (string imageFilename in TransformImages)
{
using (sitk.Image img = ReadWriteUtils.ReadITKImageFromFile(imageFilename))
{
List <sitk.Image> channels = TransformationUtils.SplitColorChannels(img);
List <sitk.Image> transformedChannels = new List <sitk.Image>();
foreach (sitk.Image channel in channels)
{
transformedChannels.Add(ExecuteTransform(parameterMaps, channel));
channel.Dispose();
}
sitk.VectorOfImage vectorImages = new sitk.VectorOfImage();
foreach (sitk.Image tChannel in transformedChannels)
{
vectorImages.Add(TransformationUtils.InterpolateImage(tChannel, GetInterpolatorEnum(), sitk.PixelIDValueEnum.sitkUInt8));
}
sitk.ComposeImageFilter composeImageFilter = new sitk.ComposeImageFilter();
sitk.Image composedImage = composeImageFilter.Execute(vectorImages);
TransformationUtils.WriteTransformedImage(composedImage, ApplicationContext.OutputPath + "\\" + Path.GetFileNameWithoutExtension(imageFilename) + "_transformed.png");
}
}
}
public NonRigidRegistration(sitk.Image fixedImage, sitk.Image movingImage, RegistrationParameters parameters) : base(parameters)
{
// cast images to from pixel type uint to vector of float
sitk.CastImageFilter castImageFilter = new sitk.CastImageFilter();
castImageFilter.SetOutputPixelType(sitk.PixelIDValueEnum.sitkVectorFloat32);
sitk.Image vector1 = castImageFilter.Execute(fixedImage);
sitk.Image vector2 = castImageFilter.Execute(movingImage);
sitk.Image tempImage1 = TransformationUtils.GetColorChannelAsImage(vector1, ColorChannel.R);
sitk.Image tempImage2 = TransformationUtils.GetColorChannelAsImage(vector2, ColorChannel.R);
this.fixedImage = tempImage1;
this.movingImage = tempImage2;
// initiate elastix and set default registration params
elastix = new sitk.ElastixImageFilter();
if (parameterMap == null)
{
parameterMap = RegistrationUtils.GetDefaultParameterMap(parameters.RegistrationDefaultParams);
}
// coefficient map is used for penalty term
if (parameters.CoefficientMapFilename != null && parameters.NonRigidOptions == MaskedNonRigidRegistrationOptions.BsplineWithPenaltyTermAndCoefficientMap)
{
RegistrationUtils.ChangeOrAddParamIfNotExist(ref parameterMap, "DilateRigidityImages", RegistrationUtils.GetVectorString("true"));
RegistrationUtils.ChangeOrAddParamIfNotExist(ref parameterMap, "FixedRigidityImageName", RegistrationUtils.GetVectorString(parameters.CoefficientMapFilename));
}
// coefficient map is used for diffuse registration
if (parameters.CoefficientMapFilename != null && parameters.NonRigidOptions == MaskedNonRigidRegistrationOptions.DiffuseRegistration)
{
RegistrationUtils.ChangeOrAddParamIfNotExist(ref parameterMap, "UseMovingSegmentation", RegistrationUtils.GetVectorString("true"));
RegistrationUtils.ChangeOrAddParamIfNotExist(ref parameterMap, "MovingSegmentationFileName", RegistrationUtils.GetVectorString(parameters.CoefficientMapFilename));
}
// set output dir and log file
outputDirectory = Path.Combine(registrationParameters.OutputDirectory, registrationParameters.Iteration.ToString());
if (!Directory.Exists(outputDirectory))
{
Directory.CreateDirectory(outputDirectory);
}
elastix.SetOutputDirectory(outputDirectory);
elastix.SetLogFileName(outputDirectory + registrationParameters.ElastixLogFileName);
elastix.LogToConsoleOn();
}
/// <summary>
/// Execute transformation of rgb image.
/// </summary>
public void Execute()
{
string outputDir = ReadWriteUtils.GetOutputDirectory(registrationParameters, registrationParameters.Iteration);
if (Directory.Exists(outputDir))
{
Directory.CreateDirectory(outputDir);
}
// split rgb channels
sitk.VectorIndexSelectionCastImageFilter rgbVector = new sitk.VectorIndexSelectionCastImageFilter();
sitk.Image redChannel = rgbVector.Execute(movingImage, 0, sitk.PixelIDValueEnum.sitkFloat32);
sitk.Image greenChannel = rgbVector.Execute(movingImage, 1, sitk.PixelIDValueEnum.sitkFloat32);
sitk.Image blueChannel = rgbVector.Execute(movingImage, 2, sitk.PixelIDValueEnum.sitkFloat32);
foreach (var parameter in parameterMaps[0])
{
parameter["DefaultPixelValue"][0] = "255.0";
parameter["ResultImagePixelType"][0] = "short";
if (parameter.ContainsKey("UseBinaryFormatForTransformationParameters"))
{
parameter.Remove("UseBinaryFormatForTransformationParameters");
}
if (interpolationOrder != -1)
{
parameter["FinalBSplineInterpolationOrder"][0] = interpolationOrder.ToString();
}
else
{
parameter["FinalBSplineInterpolationOrder"][0] = "3";
}
if (parameter["Transform"][0] == "SimilarityTransform" ||
registrationParameters.RegistrationDefaultParams == RegistrationDefaultParameters.similarity)
{
sitk.VectorString vec = new sitk.VectorString();
vec.Add((registrationParameters.LargestImageWidth / 2).ToString()); // fixed image width / 2
vec.Add((registrationParameters.LargestImageHeight / 2).ToString()); // fixed image height / 2
parameter.Add("CenterOfRotationPoint", vec);
}
}
// initialize transformix
transformix.SetOutputDirectory(outputDir);
transformix.SetTransformParameterMap(parameterMaps.First()[0]);
// add further transform parameters
if (parameterMaps[0].Count > 1)
{
for (int i = 1; i < parameterMaps[0].Count; i++)
{
var vectorParameterMap = parameterMaps[0][i];
transformix.AddTransformParameterMap(vectorParameterMap);
}
}
transformix.ComputeDeformationFieldOn();
transformix.LogToFileOn();
if (registrationParameters.ComputeJaccobian)
{
transformix.ComputeSpatialJacobianOn();
transformix.ComputeDeterminantOfSpatialJacobianOn();
}
transformix.PrintParameterMap();
// red
transformix.SetMovingImage(redChannel);
sitk.Image resultRedChannel = transformix.Execute();
resultRedChannel = TransformationUtils.InterpolateImage(resultRedChannel, sitk.InterpolatorEnum.sitkBSplineResamplerOrder3, sitk.PixelIDValueEnum.sitkUInt8);
//ReadWriteUtils.WriteSitkImageWithPreCast(resultRedChannel, registrationParameters.OutputDirectory + "\\red_channel.png");
// green
transformix.SetMovingImage(greenChannel);
sitk.Image resultGreenChannel = transformix.Execute();
resultGreenChannel = TransformationUtils.InterpolateImage(resultGreenChannel, sitk.InterpolatorEnum.sitkBSplineResamplerOrder3, sitk.PixelIDValueEnum.sitkUInt8);
//ReadWriteUtils.WriteSitkImageWithPreCast(resultGreenChannel, registrationParameters.OutputDirectory + "\\green_channel.png");
// blue
transformix.SetMovingImage(blueChannel);
sitk.Image resultBlueChannel = transformix.Execute();
resultBlueChannel = TransformationUtils.InterpolateImage(resultBlueChannel, sitk.InterpolatorEnum.sitkBSplineResamplerOrder3, sitk.PixelIDValueEnum.sitkUInt8);
//ReadWriteUtils.WriteSitkImageWithPreCast(resultBlueChannel, registrationParameters.OutputDirectory + "\\blue_channel.png");
// compose image channels
sitk.VectorOfImage vectorImages = new sitk.VectorOfImage();
vectorImages.Add(resultRedChannel);
vectorImages.Add(resultGreenChannel);
vectorImages.Add(resultBlueChannel);
sitk.ComposeImageFilter composeImageFilter = new sitk.ComposeImageFilter();
sitk.Image composedImage = composeImageFilter.Execute(vectorImages);
if (registrationParameters.Type == RegistrationType.NonRigid)
{
transformedImage = TransformationUtils.InterpolateImage(composedImage, sitk.InterpolatorEnum.sitkBSplineResamplerOrder5, composedImage.GetPixelID());
}
else
{
transformedImage = composedImage;
}
transformedImage = composedImage;
// interpolation of output image (needs to be improved -> currently theres a little loss in quality)
// possible solution: interpolate grayscale images and compose afterwards
/*sitk.ExpandImageFilter expandImageFilter = new sitk.ExpandImageFilter();
* expandImageFilter.SetInterpolator(sitk.InterpolatorEnum.sitkLinear);
* transformedImage = expandImageFilter.Execute(composedImage);*/
//transformedImage = composedImage;
}
