+
+ //============================================================================
+ // Setting up the fixed image in a reference system
+ //============================================================================
+ const itk::Vector<double, InputImageType::ImageDimension> fixedResolution = fixedImage->GetSpacing();
+ typename InternalImageType::RegionType fixedRegion = fixedImage->GetLargestPossibleRegion();
+ typename InternalImageType::RegionType::SizeType fixedSize = fixedRegion.GetSize();
+
+ // Print the parameters of the moving image and the transform
+ if (m_Verbose) {
+ std::cout << "Target or Moving image:"<<std::endl;
+ std::cout << "Size: " << fixedSize[0] << ", " << fixedSize[1];
+ if (InputImageType::ImageDimension==3) std::cout<<", " << fixedSize[2];
+ std::cout << std::endl;
+
+ std::cout<< "Resolution: "<< fixedResolution[0] << ", " << fixedResolution[1];
+ if (InputImageType::ImageDimension==3) std::cout<< ", " << fixedResolution[2];
+ std::cout << std::endl;
+ }
+
+
+
+ //===========================================================================
+ // If given, we connect a mask to reference or target
+ //============================================================================
+ typedef itk::ImageMaskSpatialObject< InputImageType::ImageDimension > MaskType;
+ typename MaskType::Pointer fixedMask=NULL;
+ if (m_ArgsInfo.referenceMask_given) {
+ fixedMask= MaskType::New();
+ typedef itk::Image< unsigned char, InputImageType::ImageDimension > ImageMaskType;
+ typedef itk::ImageFileReader< ImageMaskType > MaskReaderType;
+ typename MaskReaderType::Pointer maskReader = MaskReaderType::New();
+ maskReader->SetFileName(m_ArgsInfo.referenceMask_arg);
+ try {
+ maskReader->Update();
+ } catch ( itk::ExceptionObject & err ) {
+ std::cerr << "ExceptionObject caught while reading mask !" << std::endl;
+ std::cerr << err << std::endl;
+ return;
+ }
+ if (m_Verbose)std::cout <<"Reference image mask was read..." <<std::endl;
+
+ // Set the image to the spatialObject
+ fixedMask->SetImage( maskReader->GetOutput() );
+ }
+
+ typedef itk::ImageMaskSpatialObject< InputImageType::ImageDimension > MaskType;
+ typename MaskType::Pointer movingMask=NULL;
+ if (m_ArgsInfo.targetMask_given) {
+ movingMask= MaskType::New();
+ typedef itk::Image< unsigned char, InputImageType::ImageDimension > ImageMaskType;
+ typedef itk::ImageFileReader< ImageMaskType > MaskReaderType;
+ typename MaskReaderType::Pointer maskReader = MaskReaderType::New();
+ maskReader->SetFileName(m_ArgsInfo.targetMask_arg);
+ try {
+ maskReader->Update();
+ } catch ( itk::ExceptionObject & err ) {
+ std::cerr << "ExceptionObject caught !" << std::endl;
+ std::cerr << err << std::endl;
+ }
+ if (m_Verbose)std::cout <<"Target image mask was read..." <<std::endl;
+
+ movingMask->SetImage( maskReader->GetOutput() );
+ }
+
+
+ //============================================================================
+ // The image pyramids
+ //============================================================================
+ typedef itk::RecursiveMultiResolutionPyramidImageFilter<InternalImageType,InternalImageType > FixedImagePyramidType;
+ typedef itk::RecursiveMultiResolutionPyramidImageFilter<InternalImageType,InternalImageType > MovingImagePyramidType;
+ typename FixedImagePyramidType::Pointer fixedImagePyramid = FixedImagePyramidType::New();
+ typename MovingImagePyramidType::Pointer movingImagePyramid = MovingImagePyramidType::New();
+ fixedImagePyramid->SetUseShrinkImageFilter(false);
+ fixedImagePyramid->SetInput(fixedImage);
+ fixedImagePyramid->SetNumberOfLevels(m_ArgsInfo.levels_arg);
+ movingImagePyramid->SetUseShrinkImageFilter(false);
+ movingImagePyramid->SetInput(movingImage);
+ movingImagePyramid->SetNumberOfLevels(m_ArgsInfo.levels_arg);
+ if (m_Verbose) std::cout<<"Creating the image pyramid..."<<std::endl;
+
+ fixedImagePyramid->Update();
+ movingImagePyramid->Update();
+
+
+
+ //============================================================================
+ // We retrieve the type of metric from the command line
+ //============================================================================
+ typedef clitk::GenericMetric<args_info_clitkAffineRegistration, InternalImageType, InternalImageType> GenericMetricType;
+ typename GenericMetricType::Pointer genericMetric=GenericMetricType::New();
+ genericMetric->SetArgsInfo(m_ArgsInfo);
+ genericMetric->SetFixedImage(fixedImagePyramid->GetOutput(0));
+ if (fixedMask) genericMetric->SetFixedImageMask(fixedMask);
+ typedef itk::ImageToImageMetric< InternalImageType, InternalImageType > MetricType;
+ typename MetricType::Pointer metric=genericMetric->GetMetricPointer();
+ if (movingMask) metric->SetMovingImageMask(movingMask);
+
+ if (threadsGiven) metric->SetNumberOfThreads( threads );
+
+ //============================================================================
+ // Initialize using image moments.
+ //============================================================================
+ if (m_ArgsInfo.moment_flag) {
+ typedef itk::ImageMomentsCalculator< InternalImageType > CalculatorType;
+ typename CalculatorType::Pointer fixedCalculator= CalculatorType::New();
+
+ typename InternalImageType::Pointer fixedThresh;
+ if (m_ArgsInfo.intThreshold_given) {
+ typedef itk::ThresholdImageFilter<InternalImageType> ThresholdImageFilterType;
+ typename ThresholdImageFilterType::Pointer thresholder = ThresholdImageFilterType::New();
+ thresholder->SetInput(fixedImage);
+ thresholder->SetLower(m_ArgsInfo.intThreshold_arg);
+ thresholder->Update();
+ fixedThresh=thresholder->GetOutput();
+ } else fixedThresh=fixedImage;
+
+ fixedCalculator->SetImage(fixedThresh);
+ fixedCalculator->Compute();
+ Vector<double, InputImageType::ImageDimension> fixedCenter=fixedCalculator->GetCenterOfGravity();
+ if (m_Verbose)std::cout<<"The fixed center of gravity is "<<fixedCenter<<"..."<<std::endl;
+
+ typedef itk::ImageMomentsCalculator< InternalImageType > CalculatorType;
+ typename CalculatorType::Pointer movingCalculator= CalculatorType::New();
+
+ typename InternalImageType::Pointer movingThresh;
+ if (m_ArgsInfo.intThreshold_given) {
+ typedef itk::ThresholdImageFilter<InternalImageType> ThresholdImageFilterType;
+ typename ThresholdImageFilterType::Pointer thresholder = ThresholdImageFilterType::New();
+ thresholder->SetInput(movingImage);
+ thresholder->SetLower(m_ArgsInfo.intThreshold_arg);
+ thresholder->Update();
+ movingThresh=thresholder->GetOutput();
+ } else movingThresh=movingImage;
+
+ movingCalculator->SetImage(movingThresh);
+ movingCalculator->Compute();
+ Vector<double, InputImageType::ImageDimension> movingCenter=movingCalculator->GetCenterOfGravity();
+ if (m_Verbose)std::cout<<"The moving center of gravity is "<<movingCenter<<"..."<<std::endl;
+
+ Vector<double, InputImageType::ImageDimension> shift= movingCenter-fixedCenter;
+ if (m_Verbose)std::cout<<"The initial shift applied is "<<shift<<"..."<<std::endl;
+
+ m_ArgsInfo.transX_arg= shift [0];
+ m_ArgsInfo.transY_arg= shift [1];
+ if (InputImageType::ImageDimension==3) m_ArgsInfo.transZ_arg=shift [2];
+ }
+
+ //============================================================================
+ // Transform
+ //============================================================================
+ typedef clitk::GenericAffineTransform<args_info_clitkAffineRegistration, TCoordRep, InputImageType::ImageDimension > GenericAffineTransformType;
+ typename GenericAffineTransformType::Pointer genericAffineTransform = GenericAffineTransformType::New();
+ genericAffineTransform->SetArgsInfo(m_ArgsInfo);
+ typedef itk::Transform< double, InputImageType::ImageDimension, InputImageType::ImageDimension > TransformType;
+ typename TransformType::Pointer transform = genericAffineTransform->GetTransform();
+ std::cout<<m_ArgsInfo.transform_arg<<std::endl;
+
+
+ //=======================================================
+ // Interpolator
+ //=======================================================
+ std::cout<<"setting Interpolator..."<<std::endl;
+ typedef clitk::GenericInterpolator<args_info_clitkAffineRegistration, InternalImageType,TCoordRep > GenericInterpolatorType;
+ typename GenericInterpolatorType::Pointer genericInterpolator=GenericInterpolatorType::New();
+ genericInterpolator->SetArgsInfo(m_ArgsInfo);
+ typedef itk::InterpolateImageFunction< InternalImageType, TCoordRep > InterpolatorType;
+ typename InterpolatorType::Pointer interpolator=genericInterpolator->GetInterpolatorPointer();
+ std::cout<<"end of interpolator"<<std::endl;
+
+ //============================================================================
+ // Optimizer
+ //============================================================================
+ typedef clitk::GenericOptimizer<args_info_clitkAffineRegistration> GenericOptimizerType;
+ unsigned int nParam = transform->GetNumberOfParameters();
+ typename GenericOptimizerType::Pointer genericOptimizer=GenericOptimizerType::New();
+ genericOptimizer->SetArgsInfo(m_ArgsInfo);
+ genericOptimizer->SetOutputIteration(m_Verbose);
+ genericOptimizer->SetOutputPosition(m_Verbose);
+ genericOptimizer->SetOutputValue(m_Verbose);
+ genericOptimizer->SetOutputGradient(m_ArgsInfo.gradient_flag);
+ genericOptimizer->SetMaximize(genericMetric->GetMaximize());
+ genericOptimizer->SetNumberOfParameters(nParam);
+ typedef itk::SingleValuedNonLinearOptimizer OptimizerType;
+ OptimizerType::Pointer optimizer=genericOptimizer->GetOptimizerPointer();
+
+ // Scales
+ itk::Optimizer::ScalesType scales( nParam );
+ for (unsigned int i=nParam-InputImageType::ImageDimension; i<nParam; i++) //Translations
+ scales[i] = m_ArgsInfo.tWeight_arg;
+ for (unsigned int i=0; i<nParam-InputImageType::ImageDimension; i++) //Rest
+ scales[i] = m_ArgsInfo.rWeight_arg*180./M_PI;
+ optimizer->SetScales(scales);
+ //============================================================================
+ // Multiresolution registration
+ //============================================================================
+ std::cout<<"start MultiResolution..."<<std::endl;
+ typedef itk::MultiResolutionImageRegistrationMethod< InternalImageType,InternalImageType > RegistrationType;
+ typename RegistrationType::Pointer registration = RegistrationType::New();
+ registration->SetFixedImage( fixedImage );
+ registration->SetFixedImageRegion(fixedImage->GetLargestPossibleRegion());
+ registration->SetMovingImage( movingImage );
+ registration->SetFixedImagePyramid( fixedImagePyramid );
+ registration->SetMovingImagePyramid( movingImagePyramid );
+ registration->SetTransform( transform );
+ registration->SetInitialTransformParameters( transform->GetParameters() );
+ registration->SetInterpolator( interpolator );
+ registration->SetMetric(metric);
+ registration->SetOptimizer(optimizer);
+ registration->SetNumberOfLevels( m_ArgsInfo.levels_arg );
+ if (m_Verbose) std::cout << "Setting "<< m_ArgsInfo.levels_arg <<" resolution levels..." << std::endl;
+ if (m_Verbose) std::cout << "Initial Transform: "<< registration->GetInitialTransformParameters()<<std::endl;
+
+ //============================================================================
+ // Connecting the commander to the registration to monitor it
+ //============================================================================
+ if (m_Verbose) {
+
+ // Output iteration info
+ CommandIterationUpdate::Pointer observer = CommandIterationUpdate::New();
+ observer->SetOptimizer(genericOptimizer);
+ optimizer->AddObserver( itk::IterationEvent(), observer );
+
+
+ // Output level info
+ typedef RegistrationInterfaceCommand<RegistrationType> CommandType;
+ typename CommandType::Pointer command = CommandType::New();
+ command->SetArgsInfo(m_ArgsInfo);
+ registration->AddObserver( itk::IterationEvent(), command );
+
+ }
+
+
+ //============================================================================
+ // Finally we can start the registration with the given amount of multiresolution levels
+ //============================================================================
+ if (m_Verbose) std::cout << "Starting the registration now..." << std::endl;
+
+ try {
+ registration->Update();
+ } catch ( itk::ExceptionObject & err ) {
+ std::cerr << "ExceptionObject caught !" << std::endl;
+ std::cerr << err << std::endl;
+ }
+
+
+ //============================================================================
+ // Processing the result of the registration
+ //============================================================================
+ OptimizerType::ParametersType finalParameters = registration->GetLastTransformParameters();
+ std::cout<< "Result : " <<std::setprecision(12)<<std::endl;
+
+ for (unsigned int i=nParam-InputImageType::ImageDimension; i<nParam; i++) //Translations
+ std::cout << " Translation " << i << " = " << finalParameters[i];
+ for (unsigned int i=0; i<nParam-InputImageType::ImageDimension; i++) //Rest
+ std::cout << " Other parameter " << i << " = " << finalParameters[i];
+
+ itk::Matrix<double,InputImageType::ImageDimension+1,InputImageType::ImageDimension+1> matrix;
+ if (m_ArgsInfo.transform_arg == 3) {
+ for (unsigned int i=0; i<InputImageType::ImageDimension; i++) {
+ matrix[i][3] = finalParameters[nParam-InputImageType::ImageDimension+i];
+ for (unsigned int j=0; j<InputImageType::ImageDimension; j++) {
+ matrix[i][j] = finalParameters[i*3+j];
+ }
+ matrix[3][3] = 1.0;
+ }
+ } else {
+ matrix = clitk::GetBackwardAffineMatrix<InputImageType::ImageDimension>(finalParameters);
+ std::cout<<"outside GetBackWardAffineMatrix...."<<std::endl;
+}
+
+ std::cout << " Affine transform matrix =" << std::endl;
+ std::cout << matrix <<std::setprecision(6)<< std::endl;
+ std::cout << " End of Registration" << std::endl;
+ // Write matrix to a file
+ if (m_ArgsInfo.matrix_given) {
+ std::ofstream mFile;
+ mFile.open(m_ArgsInfo.matrix_arg);
+ mFile<<std::setprecision(12)<<matrix<< std::setprecision(6)<<std::endl;
+ mFile.close();
+ }
+
+ //============================================================================
+ // Prepare the resampling filter in order to transform the moving image.
+ //============================================================================
+ // if (m_ArgsInfo.output_given || m_ArgsInfo.checker_after_given || m_ArgsInfo.after_given ) {
+ transform->SetParameters( finalParameters );
+ typedef itk::ResampleImageFilter< InternalImageType,InternalImageType > ResampleFilterType;
+ typename ResampleFilterType::Pointer resampler = ResampleFilterType::New();
+
+ resampler->SetTransform( transform );
+ resampler->SetInput( movingImage );
+ resampler->SetSize( fixedImage->GetLargestPossibleRegion().GetSize() );
+ resampler->SetOutputOrigin( fixedImage->GetOrigin() );
+ resampler->SetOutputSpacing( fixedImage->GetSpacing() );
+ resampler->SetDefaultPixelValue( 0 );
+ resampler->Update();
+ //Output?
+ // if (m_ArgsInfo.output_given) {
+ //We write an output in the same pixeltype then the input
+ /*typedef itk::ImageFileWriter< FixedImageType > WriterType;
+ typename WriterType::Pointer outputWriter = WriterType::New();
+ outputWriter->SetFileName(m_ArgsInfo.output_arg );
+ outputWriter->SetInput( resampler->GetOutput() );
+ outputWriter->Update();*/
+ typedef InternalImageType OutputImageType;
+ typename OutputImageType::Pointer outputImage = resampler->GetOutput();
+ std::cout<<"Writing Output....."<<std::endl;
+ this->template SetNextOutput<OutputImageType>(outputImage);
+ // }
+
+
+ //============================================================================
+ // Checker after?
+ //============================================================================
+ if (m_ArgsInfo.checker_after_given) {
+ //To display correctly the checkerboard image, the intensities must lie in the same range (normalized)
+ //We write the image in the internal image type
+ typedef itk::ResampleImageFilter< InternalImageType,InternalImageType > ResampleFilterType;
+ typename ResampleFilterType::Pointer internalResampler = ResampleFilterType::New();
+ internalResampler->SetTransform( transform );
+ internalResampler->SetInput( inputMovingImage );
+ internalResampler->SetSize( fixedImage->GetLargestPossibleRegion().GetSize() );
+ internalResampler->SetOutputOrigin( fixedImage->GetOrigin() );
+ internalResampler->SetOutputSpacing( fixedImage->GetSpacing() );
+ internalResampler->SetDefaultPixelValue( 0 );
+
+ //We pass the normalized images to the checker filter
+ typedef itk::CheckerBoardImageFilter< InternalImageType > CheckerBoardFilterType;
+ typename CheckerBoardFilterType::Pointer checkerFilter= CheckerBoardFilterType::New();
+
+ checkerFilter->SetInput1(inputFixedImage);
+ checkerFilter->SetInput2(internalResampler->GetOutput());
+ typedef itk::ImageFileWriter< InternalImageType > InternalWriterType;
+ typename InternalWriterType::Pointer checkerWriter = InternalWriterType::New();
+ checkerWriter->SetFileName(m_ArgsInfo.checker_after_arg);
+ checkerWriter->SetInput( checkerFilter->GetOutput() );
+ checkerWriter->Update();
+ }
+
+
+ //============================================================================
+ // Checker before?
+ //============================================================================
+ if (m_ArgsInfo.checker_before_given) {
+ //To display correctly the checkerboard image, the intensities must lie in the same range (normalized)
+ //We write the image in the internal image type
+ //We pass the normalized images to the checker filter
+ typedef itk::CheckerBoardImageFilter< InternalImageType > CheckerBoardFilterType;
+ typename CheckerBoardFilterType::Pointer checkerFilter= CheckerBoardFilterType::New();
+
+ checkerFilter->SetInput1(inputFixedImage);
+ checkerFilter->SetInput2(inputMovingImage);
+ typedef itk::ImageFileWriter< InternalImageType > InternalWriterType;
+ typename InternalWriterType::Pointer checkerWriter = InternalWriterType::New();
+ checkerWriter->SetFileName(m_ArgsInfo.checker_before_arg);
+ checkerWriter->SetInput( checkerFilter->GetOutput() );
+ checkerWriter->Update();
+ }
+
+
+ //============================================================================
+ // Difference After?
+ //============================================================================
+ if (m_ArgsInfo.after_given) {
+ typedef itk::SubtractImageFilter< InternalImageType, FixedImageType,FixedImageType > DifferenceImageFilterType;
+ typename DifferenceImageFilterType::Pointer differenceAfterFilter= DifferenceImageFilterType::New();
+
+ differenceAfterFilter->SetInput1(fixedImage);
+ differenceAfterFilter->SetInput2(resampler->GetOutput());
+
+ // Prepare a writer to write the difference image
+ typedef itk::ImageFileWriter< FixedImageType > WriterType;
+ typename WriterType::Pointer differenceAfterWriter = WriterType::New();
+ differenceAfterWriter->SetFileName(m_ArgsInfo.after_arg );
+ differenceAfterWriter->SetInput( differenceAfterFilter->GetOutput() );
+ differenceAfterWriter->Update();
+ }
+// }
+
+ //============================================================================
+ // Difference Before?
+ //============================================================================
+ if (m_ArgsInfo.before_given) {
+ typedef itk::CastImageFilter< InternalImageType,FixedImageType > CastFilterType;
+ typename CastFilterType::Pointer caster = CastFilterType::New();
+ caster->SetInput( movingImage );
+
+ typedef itk::SubtractImageFilter< InternalImageType, FixedImageType, FixedImageType > DifferenceImageFilterType;
+ typename DifferenceImageFilterType::Pointer differenceBeforeFilter= DifferenceImageFilterType::New();
+
+
+ differenceBeforeFilter->SetInput1(fixedImage);
+ differenceBeforeFilter->SetInput2(caster->GetOutput());
+
+ // Prepare a writer to write the difference image
+ typedef itk::ImageFileWriter< FixedImageType > WriterType;
+ typename WriterType::Pointer differenceBeforeWriter = WriterType::New();
+ differenceBeforeWriter->SetFileName(m_ArgsInfo.before_arg);
+ differenceBeforeWriter->SetInput( differenceBeforeFilter->GetOutput() );
+ differenceBeforeWriter->Update();
+ }
+
+}