From 03b8c02ed0f008d529c3079769dfc8dcec08ef8f Mon Sep 17 00:00:00 2001
From: mathdugre <16450132+mathdugre@users.noreply.github.com>
Date: Tue, 4 Feb 2025 11:50:30 -0500
Subject: [PATCH] ENH: use prefix from --output for --write-interval-volumes

---
 ...tAndVelocityFieldRegistrationCommandIterationUpdate.h | 9 ++++++++-
 Examples/itkantsRegistrationHelper.hxx                   | 1 +
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/Examples/antsDisplacementAndVelocityFieldRegistrationCommandIterationUpdate.h b/Examples/antsDisplacementAndVelocityFieldRegistrationCommandIterationUpdate.h
index c21ff9f0b..1bb04a338 100644
--- a/Examples/antsDisplacementAndVelocityFieldRegistrationCommandIterationUpdate.h
+++ b/Examples/antsDisplacementAndVelocityFieldRegistrationCommandIterationUpdate.h
@@ -196,6 +196,12 @@ class antsDisplacementAndVelocityFieldRegistrationCommandIterationUpdate final :
 
   itkSetMacro(CurrentStageNumber, unsigned int);
 
+  void
+  SetOutputPrefix(const std::string & outputPrefix)
+  {
+    this->m_OutputPrefix = outputPrefix;
+  }
+
   void
   SetNumberOfIterations(const std::vector<unsigned int> & iterations)
   {
@@ -446,7 +452,7 @@ class antsDisplacementAndVelocityFieldRegistrationCommandIterationUpdate final :
     const unsigned int curLevel = filter->GetCurrentLevel();
     const unsigned int curIter = filter->GetCurrentIteration();
     std::stringstream  currentFileName;
-    currentFileName << "Stage" << this->m_CurrentStageNumber + 1 << "_level" << curLevel + 1;
+    currentFileName << this->m_OutputPrefix << "Stage" << this->m_CurrentStageNumber + 1 << "_level" << curLevel + 1;
     /*
      The name arrangement of written files are important to us.
      To prevent: "Iter1 Iter10 Iter2 Iter20" we use the following style.
@@ -500,6 +506,7 @@ class antsDisplacementAndVelocityFieldRegistrationCommandIterationUpdate final :
    */
   // itk::WeakPointer<OptimizerType>   m_Optimizer;
 
+  std::string                       m_OutputPrefix;
   std::vector<unsigned int>         m_NumberOfIterations;
   std::ostream *                    m_LogStream;
   itk::TimeProbe                    m_clock;
diff --git a/Examples/itkantsRegistrationHelper.hxx b/Examples/itkantsRegistrationHelper.hxx
index 6c821a684..25770156a 100644
--- a/Examples/itkantsRegistrationHelper.hxx
+++ b/Examples/itkantsRegistrationHelper.hxx
@@ -2165,6 +2165,7 @@ RegistrationHelper<TComputeType, VImageDimension>::DoRegistration()
         displacementFieldRegistrationObserver2->SetNumberOfIterations(currentStageIterations);
         displacementFieldRegistrationObserver2->SetOrigFixedImage(this->m_Metrics[0].m_FixedImage);
         displacementFieldRegistrationObserver2->SetOrigMovingImage(this->m_Metrics[0].m_MovingImage);
+        displacementFieldRegistrationObserver2->SetOutputPrefix(this->m_OutputPrefix);
         if (this->m_PrintSimilarityMeasureInterval != 0)
         {
           displacementFieldRegistrationObserver2->SetComputeFullScaleCCInterval(this->m_PrintSimilarityMeasureInterval);