ENH: Enable specification of target Size for image resampling (#109)

InsightSoftwareConsortium · Apr 19, 2022 · 549e2c5 · 549e2c5
1 parent c33ba00
commit 549e2c5
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 1 deletion.
diff --git a/include/tubeResampleImage.h b/include/tubeResampleImage.h
@@ -100,6 +100,9 @@ class ResampleImage:
   /** Set Output Index */
   tubeWrapForceSetMacro( Index, std::vector<int>, Filter );
 
+  /** Set Output Index */
+  tubeWrapForceSetMacro( Size, std::vector<int>, Filter );
+
   /** Set Output Resample Factor */
   tubeWrapForceSetMacro( ResampleFactor, std::vector<double>, Filter );
 

diff --git a/src/Filtering/itktubeReResampleImageFilter.h b/src/Filtering/itktubeReResampleImageFilter.h
@@ -96,6 +96,9 @@ class ReResampleImageFilter
   /** Set Output Index */
   void SetIndex( std::vector<int> i );
 
+  /** Set Output Size */
+  void SetSize( std::vector<int> i );
+
   /** Set Output Resample Factor */
   void SetResampleFactor( std::vector<double> rf );
 
@@ -127,6 +130,7 @@ class ReResampleImageFilter
   std::vector<double>                    m_Spacing;
   std::vector<double>                    m_Origin;
   std::vector<int>                       m_Index;
+  std::vector<int>                       m_Size;
   std::vector<double>                    m_ResampleFactor;
   bool                                   m_MakeIsotropic;
   bool                                   m_MakeHighResIso;

diff --git a/src/Filtering/itktubeReResampleImageFilter.hxx b/src/Filtering/itktubeReResampleImageFilter.hxx
@@ -40,6 +40,7 @@ ReResampleImageFilter( void )
   m_Spacing.clear();
   m_Origin.clear();
   m_Index.clear();
+  m_Size.clear();
   m_ResampleFactor.clear();
   m_MakeIsotropic = false;
   m_MakeHighResIso = false;
@@ -85,13 +86,15 @@ Update( void )
     }
   outDirection = inDirection;
 
+  bool updateSize = true;
   if( m_MatchImage )
     {
     outSpacing = m_MatchImage->GetSpacing();
     outOrigin = m_MatchImage->GetOrigin();
     outDirection = m_MatchImage->GetDirection();
     outSize = m_MatchImage->GetLargestPossibleRegion().GetSize();
     outIndex = m_MatchImage->GetLargestPossibleRegion().GetIndex();
+    updateSize = false;
     }
 
   if( m_Origin.size() > 0 )
@@ -108,6 +111,14 @@ Update( void )
       outIndex[i] = m_Index[i];
       }
     }
+  if( m_Size.size() > 0 )
+    {
+    for( unsigned int i = 0; i < VDimension; i++ )
+      {
+      outSize[i] = m_Size[i];
+      }
+    updateSize = false;
+    }
   if( m_Spacing.size() > 0 )
     {
     for( unsigned int i = 0; i < VDimension; i++ )
@@ -163,7 +174,7 @@ Update( void )
       }
     }
 
-  if( !m_MatchImage )
+  if( updateSize )
     {
     std::vector< double > outResampleFactor;
     outResampleFactor.resize( VDimension );
@@ -253,6 +264,14 @@ SetIndex( std::vector<int> i )
   m_Index = i;
 }
 
+template< class TPixel, unsigned int VDimension >
+void
+ReResampleImageFilter< TPixel, VDimension >::
+SetSize( std::vector<int> i )
+{
+  m_Size = i;
+}
+
 template< class TPixel, unsigned int VDimension >
 void
 ReResampleImageFilter< TPixel, VDimension >::