ENH: Add convenience methods for writing parameter files

SuperElastix · Oct 7, 2017 · 9ba1683 · 9ba1683
1 parent 2ac905a
commit 9ba1683
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 5 deletions.
diff --git a/Core/Main/elxParameterObject.cxx b/Core/Main/elxParameterObject.cxx
@@ -138,6 +138,25 @@ ::AddParameterFile( const ParameterFileNameType & parameterFileName )
 }
 
 
+/**
+ * ********************* WriteParameterFile *********************
+ */
+
+
+void
+ParameterObject
+::WriteParameterFile( void )
+{
+  ParameterFileNameVectorType parameterFileNameVector;
+  for( unsigned int i = 0; i < m_ParameterMap.size(); ++i )
+  {
+    parameterFileNameVector.push_back( "ParametersFile." + ParameterObject::ToString( i ) + ".txt" );
+  }
+
+  this->WriteParameterFile( this->m_ParameterMap, parameterFileNameVector );
+}
+
+
 /**
  * ********************* WriteParameterFile *********************
  */
@@ -234,25 +253,46 @@ ::WriteParameterFile( const ParameterFileNameType & parameterFileName )
 
 void
 ParameterObject
-::WriteParameterFile( const ParameterFileNameVectorType & parameterFileNameVector )
+::WriteParameterFile( const ParameterMapVectorType & parameterMapVector, const ParameterFileNameVectorType & parameterFileNameVector )
 {
-  if( this->m_ParameterMap.size() != parameterFileNameVector.size() )
+  if( parameterMapVector.size() != parameterFileNameVector.size() )
   {
     itkExceptionMacro(
       << "Error writing to disk: The number of parameter maps ("
-      << this->m_ParameterMap.size() << ")"
+      << parameterMapVector.size() << ")"
       << " does not match the number of provided filenames ("
       << parameterFileNameVector.size()
       << ")." );
   }
 
-  for( unsigned int i = 0; i < this->m_ParameterMap.size(); ++i )
+  // Add initial transform parameter file names. Do not touch the first one,
+  // since it may have one already
+  for( unsigned int i = 1; i < parameterMapVector.size(); ++i )
   {
-    this->WriteParameterFile( this->m_ParameterMap[ i ], parameterFileNameVector[ i ] );
+    ParameterMapType parameterMap = parameterMapVector[ i ];
+    if( parameterMap.find( "TransformParameters" ) != parameterMap.end() )
+    {
+      parameterMap[ "InitialTransformParameterFileName" ][ 0 ] = parameterFileNameVector[ i - 1 ];
+    }
+
+    this->WriteParameterFile( parameterMap, parameterFileNameVector[ i ] );
   }
 }
 
 
+/**
+ * ********************* WriteParameterFile *********************
+ */
+
+
+void
+ParameterObject
+::WriteParameterFile( const ParameterFileNameVectorType & parameterFileNameVector )
+{
+  this->WriteParameterFile( this->m_ParameterMap, parameterFileNameVector );
+}
+
+
 /**
  * ********************* GetParameterMap *********************
  */

diff --git a/Core/Main/elxParameterObject.h b/Core/Main/elxParameterObject.h
@@ -75,12 +75,16 @@ class ParameterObject : public itk::DataObject
 
   void AddParameterFile( const ParameterFileNameType & parameterFileName );
 
+  void WriteParameterFile( void );
+
   void WriteParameterFile( const ParameterMapType & parameterMap, const ParameterFileNameType & parameterFileName );
 
   void WriteParameterFile( const ParameterFileNameType & parameterFileName );
 
   void WriteParameterFile( const ParameterFileNameVectorType & parameterFileNameVector );
 
+  void WriteParameterFile( const ParameterMapVectorType & parameterMapVector, const ParameterFileNameVectorType & parameterFileNameVector );
+
   /* Get preconfigured parameter maps. */
   static const ParameterMapType GetDefaultParameterMap( const std::string & transformName,
     const unsigned int & numberOfResolutions = 4u,