ENH: Add m_RandomVariateGenerator

Author: N-Dekker

Common/CostFunctions/itkAdvancedImageToImageMetric.h

@@ -317,6 +317,8 @@ class ITK_TEMPLATE_EXPORT AdvancedImageToImageMetric : public ImageToImageMetric
   virtual void
   BeforeThreadedGetValueAndDerivative(const TransformParametersType & parameters) const;
 
+  itkSetMacro(RandomVariateGenerator, Statistics::MersenneTwisterRandomVariateGenerator *);
+
 protected:
   /** Constructor. */
   AdvancedImageToImageMetric();
@@ -598,6 +600,12 @@ class ITK_TEMPLATE_EXPORT AdvancedImageToImageMetric : public ImageToImageMetric
     itkExceptionMacro("Intentionally left unimplemented!");
   }
 
+  auto *
+  GetRandomVariateGenerator()
+  {
+    return m_RandomVariateGenerator;
+  }
+
   // Protected using-declaration, to avoid `-Woverloaded-virtual` warnings from GCC (GCC 11.4) or clang (macos-12).
   using Superclass::SetTransform;
 
@@ -627,6 +635,8 @@ class ITK_TEMPLATE_EXPORT AdvancedImageToImageMetric : public ImageToImageMetric
 
   MovingImageDerivativeScalesType m_MovingImageDerivativeScales{ MovingImageDerivativeScalesType::Filled(1.0) };
 
+  Statistics::MersenneTwisterRandomVariateGenerator * m_RandomVariateGenerator{ nullptr };
+
   // Private using-declarations, to avoid `-Woverloaded-virtual` warnings from GCC (GCC 11.4) or clang (macos-12).
   using Superclass::TransformPoint;
 

Common/ImageSamplers/itkImageRandomSamplerBase.h

@@ -94,6 +94,8 @@ class ITK_TEMPLATE_EXPORT ImageRandomSamplerBase : public ImageSamplerBase<TInpu
     return m_OptionalSeed;
   }
 
+  itkSetMacro(RandomVariateGenerator, Statistics::MersenneTwisterRandomVariateGenerator *);
+
   /** The input image dimension. */
   itkStaticConstMacro(InputImageDimension, unsigned int, Superclass::InputImageDimension);
 
@@ -108,6 +110,12 @@ class ITK_TEMPLATE_EXPORT ImageRandomSamplerBase : public ImageSamplerBase<TInpu
   void
   GenerateRandomNumberList();
 
+  auto *
+  GetRandomVariateGenerator()
+  {
+    return m_RandomVariateGenerator;
+  }
+
   /** PrintSelf. */
   void
   PrintSelf(std::ostream & os, Indent indent) const override;
@@ -117,6 +125,8 @@ class ITK_TEMPLATE_EXPORT ImageRandomSamplerBase : public ImageSamplerBase<TInpu
 
 private:
   std::optional<SeedIntegerType> m_OptionalSeed{};
+
+  Statistics::MersenneTwisterRandomVariateGenerator * m_RandomVariateGenerator{ nullptr };
 };
 
 } // end namespace itk

Core/Kernel/elxElastixMain.h

@@ -19,7 +19,8 @@
 #define elxElastixMain_h
 
 #include "elxMainBase.h"
-
+#include "elxDefaultConstruct.h"
+#include <itkMersenneTwisterRandomVariateGenerator.h>
 
 namespace elastix
 {
@@ -194,6 +195,8 @@ class ElastixMain : public MainBase
   ParameterMapType m_TransformParameterMap{};
 
   FlatDirectionCosinesType m_OriginalFixedImageDirectionFlat{};
+
+  elx::DefaultConstruct<itk::Statistics::MersenneTwisterRandomVariateGenerator> m_RandomVariateGenerator{};
 };
 
 } // end namespace elastix