Merge pull request #603 from ANTsX/AddMattes

ENH: Add Mattes.
ANTsX · Apr 9, 2024 · f79e9f3 · f79e9f3
2 parents 6b218c8 + 6c0488e
commit f79e9f3
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/ants/lib/LOCAL_antsImageToImageMetric.h b/ants/lib/LOCAL_antsImageToImageMetric.h
@@ -290,6 +290,17 @@ ANTsImageToImageMetric<MetricType> new_ants_metric( std::string precision, unsig
  MetricBasePointerType basePointer = dynamic_cast<MetricBaseType *>( metricPointer.GetPointer() );
  return wrap_metric< MetricType >( basePointer );
  }
+ else if ( metrictype == "MattesMutualInformation" )
+ {
+ typedef itk::Image<float, Dimension> ImageType;
+ typedef itk::MattesMutualInformationImageToImageMetricv4<ImageType,ImageType> SpecificMetricType;
+ typename SpecificMetricType::Pointer metricPointer = SpecificMetricType::New();
+ typedef itk::ImageToImageMetricv4<itk::Image<float, Dimension>,itk::Image<float,Dimension>> MetricBaseType;
+ typedef typename MetricBaseType::Pointer MetricBasePointerType;
+
+ MetricBasePointerType basePointer = dynamic_cast<MetricBaseType *>( metricPointer.GetPointer() );
+ return wrap_metric< MetricType >( basePointer );
+ }
  // should never reach this
  typedef itk::Image<float, Dimension> ImageType;
  typedef itk::MeanSquaresImageToImageMetricv4<itk::Image<float, Dimension>,itk::Image<float,Dimension>> SpecificMetricType;