Bugfix: Handle treatment plans that contain number of protons

[clitk.git] / registration / clitkMultiResolutionPDEDeformableRegistration.txx

diff --git a/registration/clitkMultiResolutionPDEDeformableRegistration.txx b/registration/clitkMultiResolutionPDEDeformableRegistration.txx
index a10942c1db04dbc1c150d9d6ad66c4159c0b683f..43fd978b92e8e39bcc8020a30ef33ded70c86855 100644 (file)
--- a/registration/clitkMultiResolutionPDEDeformableRegistration.txx
+++ b/registration/clitkMultiResolutionPDEDeformableRegistration.txx
@@ -335,7 +335,7 @@ MultiResolutionPDEDeformableRegistration<TFixedImage,TMovingImage,TDeformationFi
       
       if( tempField.IsNull() )
        {
-         m_RegistrationFilter->SetInitialDeformationField( NULL );
+         m_RegistrationFilter->SetInitialDisplacementField( NULL );
        }
       else
        {
@@ -357,8 +357,7 @@ MultiResolutionPDEDeformableRegistration<TFixedImage,TMovingImage,TDeformationFi
       tempField = m_FieldExpander->GetOutput();
       tempField->DisconnectPipeline();
 
-      m_RegistrationFilter->SetInitialDeformationField( tempField );
-
+      m_RegistrationFilter->SetInitialDisplacementField( tempField );
       }
 
     // setup registration filter and pyramids