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;