File: itkVectorImageReadWriteTest.cxx

package info (click to toggle)
insighttoolkit 3.18.0-5
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 110,432 kB
  • ctags: 74,559
  • sloc: cpp: 412,627; ansic: 196,210; fortran: 28,000; python: 3,852; tcl: 2,005; sh: 1,186; java: 583; makefile: 458; csh: 220; perl: 193; xml: 20
file content (145 lines) | stat: -rw-r--r-- 4,793 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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkVectorImageReadWriteTest.cxx,v $
  Language:  C++
  Date:      $Date: 2009-03-27 11:30:57 $
  Version:   $Revision: 1.4 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/

#include "itkVector.h"
#include "itkImage.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkImageLinearConstIteratorWithIndex.h"
#include "itkImageLinearIteratorWithIndex.h"

#include <iostream>

int itkVectorImageReadWriteTest(int argc, char * argv [])
{

  if ( argc < 2 )
    {
    itkGenericOutputMacro(<<"Need a file to process");
    return EXIT_FAILURE;
    }

  // Test for vector pixel type.
  
  const unsigned int Dimension = 2;
  
  // Create image of vector pixels
  typedef itk::Vector< double, 4 >          PixelType;
  typedef itk::Image < PixelType, Dimension>        ImageType;
  typedef itk::ImageFileReader< ImageType > ReaderType;
  typedef itk::ImageFileWriter< ImageType > WriterType;
  
  ImageType::Pointer   inputImage  = ImageType::New();
  ReaderType::Pointer  reader      = ReaderType::New();
  WriterType::Pointer  writer      = WriterType::New();

  // In this test, we will create a 9x9 image of vectors with pixels (4,4) 
  // and (1,6) set to 'vector1'. We will filter it using 
  // RecursiveGaussianImageFilter and compare a few filtered pixels.
  // 
  //Create ON and OFF vectors
  PixelType vector0(0.0);
  PixelType vector1;
  vector1[0] = 1.0; 
  vector1[1] =  2.0; 
  vector1[2] =  3.0; 
  vector1[3] =  4.0; 
   
  typedef itk::ImageLinearIteratorWithIndex< ImageType >      IteratorType;
  typedef itk::ImageLinearConstIteratorWithIndex< ImageType > ConstIteratorType;
  
  //Create the 9x9 input image
  ImageType::SizeType size;
  size.Fill( 9 );
  ImageType::IndexType index;
  index.Fill( 0 );
  ImageType::RegionType region;
  region.SetSize( size );
  region.SetIndex( index );
  inputImage->SetLargestPossibleRegion( region );
  inputImage->SetBufferedRegion( region );
  inputImage->SetRequestedRegion( region );
  inputImage->Allocate();
  inputImage->FillBuffer( vector0);
  
  std::cout << "Create image of 9x9 image of vector pixels. IO Read and write it. " << std::endl;
  
  /* Set pixel (4,4) with the value 1 
   * and pixel (1,6) with the value 2 
   */
  index[0] = 4;
  index[1] = 4;
  inputImage->SetPixel( index, vector1);
  index[0] = 1;
  index[1] = 6;
  inputImage->SetPixel( index, vector1);

  writer->SetInput(inputImage);
  writer->SetFileName(argv[1]);
  writer->Update();

  reader->SetFileName(argv[1]);
  reader->Update();
  ImageType::Pointer outputImage = reader->GetOutput();
  
  ConstIteratorType cit( outputImage, outputImage->GetLargestPossibleRegion() );
  index[0] = 4;
  index[1] = 4;
  cit.SetIndex(index);
  if( cit.Get() != vector1 ) 
    {
    std::cout << "Vector Image Write-Read failed. Tried to write " << vector1 << 
      " But read " << cit.Get() << std::endl;
    return EXIT_FAILURE;
    }
  index[0] = 0;
  index[1] = 0;
  cit.SetIndex(index);
  if( cit.Get() != vector0 ) 
    {
    std::cout << "Vector Image Write-Read failed. Tried to write " << vector0 << 
      " But read " << cit.Get() << std::endl;
    return EXIT_FAILURE;
    }

  itk::ImageIOBase::Pointer io = reader->GetImageIO();
   

  std::cout << "ImageIO Pixel Information: " 
            << io->GetPixelTypeAsString( io->GetPixelType() ) << " " 
            << io->GetComponentTypeAsString( io->GetComponentType() ) << " "
            << io->GetNumberOfComponents() << std::endl;
  if ( io->GetNumberOfComponents() != 4 ||
       io->GetComponentType() != itk::ImageIOBase::DOUBLE ||
       io->GetPixelType() != itk::ImageIOBase::VECTOR) 
    {
    std::cout << "Unexpected pixel information" << std::endl;
    std::cout << "Expected: vector double 4" << std::endl;
    return EXIT_FAILURE;
    }

  std::cout << "Image of vector pixels write-read [PASSED]" << std::endl;

  std::cout << "Test << operator:: Vector1 = " << vector1 << "[PASSED]" << std::endl;
  std::cout << "Test NumericTraits<Vector<double,4>>::Zero " << 
                        itk::NumericTraits< PixelType >::Zero << std::endl;
  std::cout << "Test NumericTraits <Vector <double,4 > >::One " << 
                        itk::NumericTraits< PixelType >::One << std::endl;
   
  return EXIT_SUCCESS; 
}