File: sitkTransformCompare.cxx

package info (click to toggle)
simpleitk 1.0.1-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 75,056 kB
  • sloc: cpp: 25,403; python: 3,060; sh: 1,131; ansic: 369; java: 260; cs: 215; makefile: 51; ruby: 47; tcl: 22
file content (79 lines) | stat: -rw-r--r-- 2,848 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
/*=========================================================================
*
*  Copyright Insight Software Consortium
*
*  Licensed under the Apache License, Version 2.0 (the "License");
*  you may not use this file except in compliance with the License.
*  You may obtain a copy of the License at
*
*         http://www.apache.org/licenses/LICENSE-2.0.txt
*
*  Unless required by applicable law or agreed to in writing, software
*  distributed under the License is distributed on an "AS IS" BASIS,
*  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*  See the License for the specific language governing permissions and
*  limitations under the License.
*
*=========================================================================*/
#include <SimpleITK.h>
#include <memory>
#include "sitkTransformCompare.h"
#include "sitkTransformToDisplacementFieldFilter.h"

TransformCompare::TransformCompare()
  : m_Tolerance(0.0)
{
}



bool TransformCompare::Compare( const itk::simple::Transform &transform,
                                const itk::simple::Transform &baselineTransform,
                                const itk::simple::Image &fixedImage )
{
  namespace sitk = itk::simple;

  sitk::TransformToDisplacementFieldFilter toDisplacementField;
  toDisplacementField.SetOutputPixelType( sitk::sitkVectorFloat64 );
  toDisplacementField.SetReferenceImage( fixedImage );
  sitk::Image baselineDisplacement = toDisplacementField.Execute( baselineTransform );

  return this->Compare( transform, baselineDisplacement );
}


bool TransformCompare::Compare( const itk::simple::Transform &transform,
                                const itk::simple::Image &baselineDisplacement )
{
  namespace sitk = itk::simple;

  sitk::TransformToDisplacementFieldFilter toDisplacementField;
  toDisplacementField.SetOutputPixelType( baselineDisplacement.GetPixelID() );
  toDisplacementField.SetReferenceImage( baselineDisplacement );
  sitk::Image displacement = toDisplacementField.Execute( transform );

  sitk::Image diff =  sitk::Subtract( displacement, baselineDisplacement );

  diff = sitk::VectorMagnitude(diff);
  diff = sitk::Multiply(diff, diff);

  sitk::StatisticsImageFilter stats;
  stats.Execute ( diff );
  double rms = std::sqrt ( stats.GetMean() );

  std::cout << "<DartMeasurement name=\"RMSDifference\" type=\"numeric/float\">" << rms << "</DartMeasurement>" << std::endl;

  if ( rms > m_Tolerance )
    {
    std::ostringstream msg;
    msg << "TransformCompare: transformed image points Root Mean Square (RMS) difference was " << rms << " which exceeds the tolerance of " << m_Tolerance;
    msg << "\n";
    m_Message = msg.str();

    std::cout << "<DartMeasurement name=\"Tolerance\" type=\"numeric/float\">" << m_Tolerance << "</DartMeasurement>" << std::endl;

    return false;
    }

  return true;
}