ENH: Add SubTransforms to WeightedCombinationTransform parameter map

Anticipating pull request #385 "Sync Transform WriteToFile with CreateTransformParametersMap".
SuperElastix · Jan 13, 2021 · 61df9d5 · 61df9d5
1 parent e2d2970
commit 61df9d5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Common/GTesting/elxTransformIOGTest.cxx b/Common/GTesting/elxTransformIOGTest.cxx
@@ -517,7 +517,7 @@ struct WithDimension
         { "StackSpacing", { expectedOne } } });
     WithElastixTransform<TranslationTransformElastix>::Test_CreateTransformParametersMap_for_default_transform({});
     WithElastixTransform<WeightedCombinationTransformElastix>::Test_CreateTransformParametersMap_for_default_transform(
-      { { "NormalizeCombinationWeights", { expectedFalse } } });
+      { { "NormalizeCombinationWeights", { expectedFalse } }, { "SubTransforms", {} } });
   }
 };
 

diff --git a/Components/Transforms/WeightedCombinationTransform/elxWeightedCombinationTransform.hxx b/Components/Transforms/WeightedCombinationTransform/elxWeightedCombinationTransform.hxx
@@ -158,7 +158,8 @@ WeightedCombinationTransformElastix<TElastix>::CreateDerivedTransformParametersM
 {
   const auto & itkTransform = *m_WeightedCombinationTransform;
 
-  return { { "NormalizeCombinationWeights", { BaseComponent::ToString(itkTransform.GetNormalizeWeights()) } } };
+  return { { "NormalizeCombinationWeights", { BaseComponent::ToString(itkTransform.GetNormalizeWeights()) } },
+           { "SubTransforms", m_SubTransformFileNames } };
 
 } // end CustomizeTransformParametersMap()